PndSdsDetector.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                         PndSdsDetector header file            -----
// -----                          Based on PndMvdDetector              -----
// -----                      Created 6/04/06  by T. Stockmanns        -----
// -------------------------------------------------------------------------

#ifndef PNDSDSDETECTOR_H
#define PNDSDSDETECTOR_H


//#include "TClonesArray.h"
#include "TLorentzVector.h"
#include "TVector3.h"
#include "FairDetector.h"
#include "TString.h"
#include "PndGeoHandling.h"
#include "PndDetectorList.h"

#include <string>
#include <vector>
#include <map>

class TClonesArray;
class TGeoNode;
class PndSdsMCPoint;
class FairVolume;



class PndSdsDetector : public FairDetector
{

 public:

  PndSdsDetector();

  PndSdsDetector(PndSdsDetector& other) :
    FairDetector(other),
    fPersistance(other.fPersistance),
    fTrackID(other.fTrackID),
    fVolumeID(other.fVolumeID),
    fPosIn(other.fPosIn),
    fPosOut(other.fPosOut),
    fMomIn(other.fMomIn),
    fMomOut(other.fMomOut),
    fTime(other.fTime),
    fLength(other.fLength),
    fELoss(other.fELoss),
    fGeoH(other.fGeoH),
    fPosIndex(other.fPosIndex),
    fPndSdsCollection(other.fPndSdsCollection),
    fUseRadDamOption(other.fUseRadDamOption),
    fOutBranchName(other.fOutBranchName),
    fFolderName(other.fFolderName),
    fDetectorID(other.fDetectorID),
    fListOfSensitives(other.fListOfSensitives)
  {};
  PndSdsDetector(const char* name, Bool_t active);

  virtual ~PndSdsDetector();

  PndSdsDetector& operator=(PndSdsDetector& other)
  {
    fPersistance=other.fPersistance;
    fTrackID=other.fTrackID;
    fVolumeID=other.fVolumeID;
    fPosIn=other.fPosIn;
    fPosOut=other.fPosOut;
    fMomIn=other.fMomIn;
    fMomOut=other.fMomOut;
    fTime=other.fTime;
    fLength=other.fLength;
    fELoss=other.fELoss;
    fGeoH=other.fGeoH;
    fPosIndex=other.fPosIndex;
    fPndSdsCollection=other.fPndSdsCollection;
    fUseRadDamOption=other.fUseRadDamOption;
    fOutBranchName=other.fOutBranchName;
    fFolderName=other.fFolderName;
    fDetectorID=other.fDetectorID;
    fListOfSensitives=other.fListOfSensitives;
    return *this;
  };

  virtual void Initialize();

  virtual void SetBranchNames()=0;
  
  virtual void SetDefaultSensorNames() = 0;
  void SetExclusiveSensorType(const TString sens);

  virtual Bool_t  ProcessHits(FairVolume* vol = 0);


  virtual void EndOfEvent();

  virtual void   FinishRun();


  virtual void Register();


  virtual TClonesArray* GetCollection(Int_t iColl) const;


  virtual void Print() const;


  virtual void SetSpecialPhysicsCuts();
  

  virtual void Reset();


  virtual void CopyClones(TClonesArray* cl1, TClonesArray* cl2,
                          Int_t offset);


  virtual void ConstructGeometry();
 // void ConstructRootGeometry();
  virtual void ConstructASCIIGeometry();
 // void ExpandNode(TGeoNode *fN);
  
  void SetDetectorID(DetectorId id){fDetectorID = id;fDetId =id;}

  void SetRadDamOption(bool val){fUseRadDamOption = val;};
  bool GetRadDamOption(){return fUseRadDamOption;};

  // overload function from FairModule
  void SetVerboseLevel(Int_t level){fVerboseLevel=level;fGeoH->SetVerbose(level);}
  
  void SetPersistance(Bool_t p = kTRUE) {fPersistance=p;};
  Bool_t GetPersistance() {return fPersistance;};
  
protected:
  
  Bool_t fPersistance; // switch to turn on/off storing the arrays to a file
  
  
  Int_t          fTrackID;           
  Int_t          fVolumeID;             
  TLorentzVector fPosIn;             
  TLorentzVector fPosOut;            
  TLorentzVector fMomIn;             
  TLorentzVector fMomOut;            
  Double32_t     fTime;              
  Double32_t     fLength;            
  Double32_t     fELoss;             
//  std::map<TString, Int_t> fVolumeIDMap;       //!  map to create a unique Detector ID
  PndGeoHandling* fGeoH;             
  Int_t fPosIndex;                   
  TClonesArray* fPndSdsCollection;      
  bool fUseRadDamOption;                        
  TString fOutBranchName;     
  TString fFolderName;        
  DetectorId fDetectorID;          

  std::vector<std::string> fListOfSensitives; 

  bool CheckIfSensitive(std::string name);

  PndSdsMCPoint* AddHit(Int_t trackID, Int_t detID, Int_t sensorID,
                      TVector3 posIn, TVector3 posOut,
                      TVector3 momIn, TVector3 momOut,
                      Double_t time, Double_t length, Double_t eLoss);


  void ResetParameters();


  ClassDef(PndSdsDetector,6);

};


inline void PndSdsDetector::ResetParameters() {
  fTrackID = 0;
  fVolumeID = 1;
  fPosIn.SetXYZM(0.0, 0.0, 0.0, 0.0);
  fPosOut.SetXYZM(0.0, 0.0, 0.0, 0.0);
  fMomIn.SetXYZM(0.0, 0.0, 0.0, 0.0);
  fMomOut.SetXYZM(0.0, 0.0, 0.0, 0.0);
  fTime = fLength = fELoss = 0;
  fPosIndex = 0;
}


#endif