FTGeom.h

Go to the documentation of this file.

#pragma once
 
#include <tuple>
#include <vector>
 
using std::tuple;
using std::vector;
 
#include "FTLayer.h"
#include "FTSuperLayer.h"
#include "FTWire.h"
 
struct FTLayerGeomInfo {
  FTLayerGeomInfo( int layerId, int superLayerId, int wire_start, int wire_end )
      : layerId( layerId )
      , superLayerId( superLayerId )
      , wire_start( wire_start )
      , wire_end( wire_end ) {}
 
  const int layerId;
  const int superLayerId;
  const int wire_start;
  const int wire_end;
};
 
struct FTSuperLayerGeomInfo {
  FTSuperLayerGeomInfo( int superLayerId, int layer_start, int layer_end, int wire_start,
                        int wire_end )
      : superLayerId( superLayerId )
      , layer_start( layer_start )
      , layer_end( layer_end )
      , wire_start( wire_start )
      , wire_end( wire_end ) {}
 
  const int superLayerId;
  const int layer_start;
  const int layer_end;
  const int wire_start;
  const int wire_end;
};
 
class FTGeom {
public:
  static FTGeom* instance();
  void           init();
 
  FTWire*       getVirtualWire() { return m_virtualWire; }
  FTWire*       getWire( const int wire_id ) { return m_wires[wire_id]; }
  FTLayer*      getLayer( const int layer_id ) { return m_layers[layer_id]; }
  FTSuperLayer* getSuperLayer( const int superLayer_id ) {
    return m_superLayers[superLayer_id];
  }
 
  // ########################################## //
  // ############# input: wire_id ############# //
  // ########################################## //
  // get layer id from wire id
  inline const int wire2layer( const int wire_id ) {
    return std::get<1>( m_geom_relationship[wire_id] );
  }
 
  // get super layer id from wire id
  inline const int wire2superLayer( const int wire_id ) {
    return std::get<2>( m_geom_relationship[wire_id] );
  }
 
  // get first wire id in its layer
  inline const int wire2wireStartLayer( const int wire_id ) {
    return m_layerGeomInfo[wire2layer( wire_id )].wire_start;
  }
 
  // get last wire id in its layer
  inline const int wire2wireEndLayer( const int wire_id ) {
    return m_layerGeomInfo[wire2layer( wire_id )].wire_end;
  }
 
  // get index in its layer
  inline const int wire2wireIndexLayer( const int wire_id ) {
    return wire_id - wire2wireStartLayer( wire_id );
  }
 
  // get first wire id in its super layer
  inline const int wire2wireStartSuperLayer( const int wire_id ) {
    return m_superLayerGeomInfo[wire2superLayer( wire_id )].wire_start;
  }
 
  // get last wire id in its super layer
  inline const int wire2wireEndSuperLayer( const int wire_id ) {
    return m_superLayerGeomInfo[wire2superLayer( wire_id )].wire_end;
  }
 
  // ########################################### //
  // ############# input: layer_id ############# //
  // ########################################### //
  // get layer id from super layer id
  inline const int layer2superLayer( const int layer_id ) {
    return m_layerGeomInfo[layer_id].superLayerId;
  }
 
  // get first wire id in layer
  inline const int layer2wireStart( const int layer_id ) {
    return m_layerGeomInfo[layer_id].wire_start;
  }
 
  // get last wire id in layer
  inline const int layer2wireEnd( const int layer_id ) {
    return m_layerGeomInfo[layer_id].wire_end;
  }
 
  // get first layer id in its super layer
  inline const int layer2layerStart( const int layer_id ) {
    return m_superLayerGeomInfo[layer2superLayer( layer_id )].layer_start;
  }
 
  // get index in its super layer
  inline const int layer2layerIndex( const int layer_id ) {
    return layer_id - layer2layerStart( layer_id );
  }
 
  // get number of wires in layer
  inline const int layer2nWires( const int layer_id ) {
    return m_layerGeomInfo[layer_id].wire_end - m_layerGeomInfo[layer_id].wire_start + 1;
  }
 
  // ########################################## //
  // ########## input: superLayer_id ########## //
  // ########################################## //
 
  // get first layer id in super layer
  inline const int superLayer2layerStart( const int superLayer_id ) {
    return m_superLayerGeomInfo[superLayer_id].layer_start;
  }
 
  // get last layer id in super layer
  inline const int superLayer2layerEnd( const int superLayer_id ) {
    return m_superLayerGeomInfo[superLayer_id].layer_end;
  }
 
  // get first wire id in super layer
  inline const int superLayer2wireStart( const int superLayer_id ) {
    return m_superLayerGeomInfo[superLayer_id].wire_start;
  }
 
  // get last wire id in super layer
  inline const int superLayer2wireEnd( const int superLayer_id ) {
    return m_superLayerGeomInfo[superLayer_id].wire_end;
  }
 
  // get number of layers in super layer
  inline const int superLayer2nLayers( const int superLayer_id ) {
    return m_superLayerGeomInfo[superLayer_id].layer_end -
           m_superLayerGeomInfo[superLayer_id].layer_start + 1;
  }
 
  // get number of wires in super layer
  inline const int superLayer2nWires( const int superLayer_id ) {
    return m_superLayerGeomInfo[superLayer_id].wire_end -
           m_superLayerGeomInfo[superLayer_id].wire_start + 1;
  }
 
private:
  FTGeom();
  ~FTGeom() {
    for ( auto w : m_wires ) delete w;
    for ( auto l : m_layers ) delete l;
    for ( auto s : m_superLayers ) delete s;
    delete m_virtualWire;
  }
 
  // wire, layer, superlayer containers
  FTWire*               m_virtualWire;
  FTList<FTWire*>       m_wires;
  FTList<FTLayer*>      m_layers;
  FTList<FTSuperLayer*> m_superLayers;
 
  // geometry information
  FTList<tuple<int, int, int>> m_geom_relationship; // [wireId, layerId, superLayerId]
  FTList<FTLayerGeomInfo>      m_layerGeomInfo;
  FTList<FTSuperLayerGeomInfo> m_superLayerGeomInfo;
};