BSEmcDetector.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                                    BSEmcDetector
// -----                    Based on FairDetector and PndSdsDetector
// -----               Created 12/03/2019 by B. Salisbury
// -------------------------------------------------------------------------
#ifndef BSEMCDETECTOR_HH
#define BSEMCDETECTOR_HH

#include <string>
#include <vector>

#include "Rtypes.h"
#include "RtypesCore.h"
#include "TLorentzVector.h"
#include "TString.h"
#include "TVector3.h"

#include "FairDetector.h"

#include "PndGeoHandling.h"

class PndSensorNameIdMap;
class TClonesArray;
class TGeoNode;
class BSEmcMCPoint;
class FairVolume;
class TGeoMatrix;
class TBuffer;
class TClass;
class TMemberInspector;

class BSEmcDetector : public FairDetector {
 public:
  BSEmcDetector();

  BSEmcDetector(BSEmcDetector &other)
    : FairDetector(other), fListOfSensitives(other.fListOfSensitives), fGeoH(PndGeoHandling::Instance()), fIdMap(other.fIdMap), fPosIndex(other.fPosIndex),
      fOutBranchName(other.fOutBranchName), fFolderName(other.fFolderName), fPndEmcCollection(other.fPndEmcCollection), fTransformMatrix(other.fTransformMatrix)

  {
    ResetParameters();
  };

  BSEmcDetector(const char *t_name, Bool_t t_active);

  virtual ~BSEmcDetector();

  // BSEmcDetector &operator=(BSEmcDetector &other)
  // {
  //   fListOfSensitives = other.fListOfSensitives;
  //   fPosIndex = other.fPosIndex;
  //   fOutBranchName = other.fOutBranchName;
  //   fFolderName = other.fFolderName;
  //   fPndEmcCollection = other.fPndEmcCollection;
  //   fTransformMatrix = other.fTransformMatrix;
  //   //fGeoH= PndGeoHandling::Instance();
  //   return *this;
  // };

  virtual void Initialize() /*override*/;

  virtual Bool_t ProcessHits(FairVolume *t_vol = nullptr) /*override*/;

  virtual void EndOfEvent() /*override*/;

  virtual void FinishRun() /*override*/;

  virtual void Register() /*override*/;

  virtual TClonesArray *GetCollection(Int_t t_iColl) const /*override*/;

  virtual void Print() const;

  virtual void SetSpecialPhysicsCuts() /*override*/;

  virtual void Reset() /*override*/;

  virtual void CopyClones(TClonesArray *t_origin, TClonesArray *t_target, Int_t t_offset) /*override*/;

  virtual void ConstructGeometry() /*override*/;
  // void ConstructRootGeometry();
  virtual void ConstructASCIIGeometry() /*override*/;

  void SetFolderName(const TString &t_folder) { fFolderName = t_folder; }
  void SetBranchOutName(const TString &t_branchout) { fOutBranchName = t_branchout; }
  const TString &GetFolderName() const { return fFolderName; }
  const TString &GetBranchOutName() const { return fOutBranchName; }

  void SetSensitiveNames(const std::vector<std::string> &t_sensNames) { fListOfSensitives = t_sensNames; };
  virtual void SetDefaultSensitiveNames() { fListOfSensitives = {"Crystal"}; } /*override*/;
  const std::vector<std::string> &GetListOfSensitives() { return fListOfSensitives; }
  void SetTransformMatrix(TGeoMatrix *t_transform) { fTransformMatrix = t_transform; }
  const TGeoMatrix *GetTransformMatrix() const { return fTransformMatrix; }

  void SetIdMap(PndSensorNameIdMap *t_map) { fIdMap = t_map; }
  PndSensorNameIdMap *GetIdMap() { return fIdMap; }

  void SetPersistency(Bool_t t_storePoints) { fPersistency = t_storePoints; }

 protected:
  virtual Int_t GetDetectorId();
  Bool_t CheckIfSensitive(std::string t_name) /*override*/;

  BSEmcMCPoint *
  AddHit(Int_t t_trackID, Int_t t_detID, Int_t t_evtID, TVector3 t_pos, TVector3 t_mom, Double_t t_tof, Double_t t_length, Double_t t_eLoss, Bool_t t_enterning, Bool_t t_exiting);

  void ResetParameters();

 protected:
  std::vector<std::string> fListOfSensitives{}; 
  PndGeoHandling *fGeoH{nullptr};               
  PndSensorNameIdMap *fIdMap{nullptr};          

 private:
  Int_t fPosIndex{-1};                      
  TString fOutBranchName{""};               
  TString fFolderName{""};                  
  TClonesArray *fPndEmcCollection{nullptr}; 

  Int_t fTrackID{-1};                    
  Int_t fVolumeID{-1};                   
  Int_t fCopyNumber{-1};                 
  Int_t fEventID{-1};                    
  TLorentzVector fPos{};                 
  TLorentzVector fMom{};                 
  Double32_t fTime{-1};                  
  Double32_t fLength{-1};                
  Double32_t fELoss{-1};                 
  TGeoMatrix *fTransformMatrix{nullptr}; 
  Int_t fNumberOfPoints{0};
  Bool_t fPersistency{kFALSE};
  ClassDef(BSEmcDetector, 7);
};

#endif /*BSEMCDETECTOR_HH*/