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 = static_cast<int>(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