CbmStsMC.h

Go to the documentation of this file.

 
#ifndef CBMSTSMC_H
#define CBMSTSMC_H
 
 
#include "CbmStsTrackStatus.h"
#include "FairDetector.h"
#include "FairRootManager.h"
#include "TClonesArray.h"
#include <map>
 
class FairVolume;
class CbmStsPoint;
class CbmStsSetup;
class TGeoNode;
class TGeoCombiTrans;
 
class CbmStsMC : public FairDetector {
public:
  CbmStsMC(Bool_t active = kTRUE, const char* name = "STSMC");
 
 
  virtual ~CbmStsMC();
 
 
  virtual Bool_t CheckIfSensitive(std::string name) {
    return (TString(name).Contains("Sensor") ? kTRUE : kFALSE);
  }
 
 
  virtual void ConstructGeometry();
 
 
  virtual void EndOfEvent();
 
 
  virtual void Initialize();
 
 
  virtual TClonesArray* GetCollection(Int_t iColl) const {
    return (iColl ? NULL : fStsPoints);
  }
 
 
  virtual void Print(Option_t* opt = "") const;
 
 
  virtual Bool_t ProcessHits(FairVolume* vol = 0);
 
 
  virtual void Register() {
    FairRootManager::Instance()->Register(
      "StsPoint", GetName(), fStsPoints, kTRUE);
  }
 
 
  void ProcessNeutrals(Bool_t choice = kTRUE) { fProcessNeutrals = choice; }
 
 
  virtual void Reset();
 
  virtual void ConstructRootGeometry(TGeoMatrix* shift = NULL);
 
private:
  CbmStsTrackStatus fStatusIn;   
  CbmStsTrackStatus fStatusOut;  
  Double_t fEloss;               
  std::map<TString, Int_t>
    fAddressMap;             
  TClonesArray* fStsPoints;  
  CbmStsSetup* fSetup;       
  TGeoCombiTrans*
    fCombiTrans;            
  Bool_t fProcessNeutrals;  
 
  CbmStsPoint* CreatePoint();
 
 
  void SetStatus(CbmStsTrackStatus& status);
 
 
  CbmStsMC(const CbmStsMC&);
 
 
  CbmStsMC operator=(const CbmStsMC&);
 
 
  ClassDef(CbmStsMC, 1);
};
 
#endif /* CBMSTSMC_H */