CbmStsSetup.h

Go to the documentation of this file.

 
#ifndef CBMSTSSETUP_H
#define CBMSTSSETUP_H 1
 
#include <Rtypes.h>      // for THashConsistencyHolder, ClassDef
#include <RtypesCore.h>  // for Int_t, Double_t, Bool_t, UInt_t, UChar_t
 
#include <FairLogger.h>  // for LOG
 
#include <map>      // for map, __map_const_iterator, operator!=
#include <set>      // for set
#include <string>   // for string
#include <utility>  // for pair
#include <vector>   // for vector
 
#include "CbmStsElement.h"  // for CbmStsElement
#include "CbmStsSensor.h"   // for CbmStsSensor
 
class CbmStsModule;
class CbmStsParSetModule;
class CbmStsParSetSensor;
class CbmStsParSetSensorCond;
class CbmStsStation;
class TGeoManager;
 
class CbmStsSetup : public CbmStsElement {
 
public:
  virtual ~CbmStsSetup() {};
 
 
  CbmStsElement* GetElement(Int_t address, Int_t level);
 
 
  const char* GetLevelName(Int_t level) const;
 
 
  CbmStsModule* GetModule(Int_t index) const { return fModuleVector.at(index); }
 
 
  CbmStsSensor* GetSensor(Int_t index) const { return fSensorVector.at(index); }
 
 
  Int_t GetNofModules() const { return fModules.size(); }
 
 
  Int_t GetNofSensors() const { return fSensors.size(); }
 
 
  Int_t GetNofStations() const { return fStations.size(); }
 
 
  CbmStsStation* GetStation(Int_t stationId) const {
    if (fStations.find(stationId) == fStations.end()) return nullptr;
    return fStations.at(stationId);
  }
 
 
  Int_t GetStationNumber(Int_t address);
 
 
  Bool_t Init(const char* geometryFile = nullptr);
 
 
  Bool_t IsInit() const { return fIsInitialised; }
 
 
  Bool_t IsModuleParsInit() const { return fIsModuleParsInit; }
 
 
  Bool_t IsSensorCondInit() const { return fIsSensorParsInit; }
 
 
  Bool_t IsSensorParsInit() const { return fIsSensorParsInit; }
 
 
  static CbmStsSetup* Instance();
 
 
  void ListModules() const;
 
 
  void ListSensors() const {
    for (auto it = fSensors.begin(); it != fSensors.end(); it++)
      LOG(info) << it->second->ToString();
  }
 
 
  UInt_t SetModuleParameters(CbmStsParSetModule* modulePars);
 
 
  UInt_t SetSensorConditions(CbmStsParSetSensorCond* conds);
 
 
  UInt_t SetSensorParameters(CbmStsParSetSensor* parSet);
 
 
private:
  CbmStsSetup();
 
 
  Int_t CreateStations();
 
 
  Bool_t ReadGeometry(TGeoManager* geoManager);
 
 
  Bool_t ReadGeometry(const char* fileName);
 
 
  CbmStsSetup(const CbmStsSetup&) = delete;
 
 
  CbmStsSetup operator=(const CbmStsSetup&) = delete;
 
 
private:
  static CbmStsSetup* fgInstance;     
  Bool_t fIsInitialised    = kFALSE;  
  Bool_t fIsModuleParsInit = kFALSE;  
  Bool_t fIsSensorParsInit = kFALSE;  
  Bool_t fIsSensorCondInit = kFALSE;  
  Bool_t fHasStations      = kFALSE;  
 
  // --- Map of sensors. Key is address.
  std::map<Int_t, CbmStsSensor*> fSensors;
 
  // --- Map of modules. Key is address.
  std::map<Int_t, CbmStsModule*> fModules;
 
  // --- Vector of modules. For convenient loops.
  std::vector<CbmStsModule*> fModuleVector;
 
  // --- Vector of sensors. For convenient loops.
  std::vector<CbmStsSensor*> fSensorVector;
 
  // --- Map of stations. Key is station number.
  // --- Stations are a special case needed for reconstruction;
  // --- they are not elements in the setup.
  std::map<Int_t, CbmStsStation*> fStations;  
 
 
  ClassDef(CbmStsSetup, 3);
};
 
#endif /* CBMSTSSETUP_H */