PndGeanePro.h

Go to the documentation of this file.

#ifndef PNDGEANEPRO_H
#define PNDGEANEPRO_H

#include "FairLogger.h"
#include "FairPropagator.h"
#include "TGeant3.h"
#include "TString.h"
#include "TVector3.h"
#include "TDatabasePDG.h"
#include "FairGeaneApplication.h"
#include "PndPropagator.h"
#include "PndGeaneGeometryMethods.h"

class PndGeanePro : public PndPropagator
{
 private:
  Float_t fDestinationLengthArray[1];   
  Float_t fInErrorMatrix[15];           
  Double_t fTransportMatrix[5][5];      
  Float_t fInitialPlane[6];             
  Float_t fInitialPositionArray[3];     
  Float_t fInitialMomentumArray[3];     
  Float_t fFinalPlane[12];              
  Float_t fFinalPositionArray[3];       
  Float_t fFinalMomentumArray[3];       
  
  TGeant3* gMC3;                        
  Ertrio_t* afErtrio;                   
  TDatabasePDG* fDBpdg;                 

  TString fPropOption;                  
  Int_t fNrep;                          
  Int_t fGeantCode;                     
  Int_t fProMode;                       
  Float_t fTrackTime;                   

  PndGeaneGeometryMethods fGeoMethods;  
  
  PndGeanePro(const PndGeanePro&);      
  PndGeanePro& operator=(const PndGeanePro&);

 public:
  PndGeanePro();  
  ~PndGeanePro(); 

  void Init(FairTrackPar* TParam);
  
  Bool_t Propagate(Int_t PDG);
  
  virtual Bool_t Propagate(FairTrackPar* TStart, FairTrackPar* TEnd, int PDG);

  virtual Bool_t Propagate(Float_t* x1, Float_t* p1, Float_t* x2, Float_t* p2, Int_t PDG);

  virtual PndProp::PCAOutputStruct FindPCA(Int_t pca, Int_t PDGCode, TVector3 point, TVector3 wire1,
                                           TVector3 wire2, Double_t maxdistance);

  virtual Bool_t SetDestinationPlane(const TVector3& v0, const TVector3& v1, const TVector3& v2);

  virtual Bool_t SetOriginPlane(const TVector3& v0, const TVector3& v1);

  virtual Bool_t SetDestinationVolume(std::string volName, Int_t copyNo, Int_t option);

  virtual Bool_t SetDestinationLength(Float_t length);

  virtual Bool_t SetDestinationPoint(TVector3 point);

  virtual Bool_t SetDestinationWire(TVector3 wire1, TVector3 wire2);

  virtual Bool_t SetPropagateOnlyParameters();

  virtual bool SetPCAPropagation(Int_t pca, Int_t dir = 1, FairTrackPar* par = nullptr);

  void SetBackProp() { fPropOption = "BPE"; }
  
  Float_t GetLengthAtPCA() { return fPcaOutput.TrackLength; }
  
  void GetTransportMatrix(Double_t trm[5][5])
  {
    for (Int_t i = 0; i < 5; i++){
      for (Int_t j = 0; j < 5; j++) {
        trm[i][j] = fTransportMatrix[i][j];
      }
    }
  }

  ClassDef(PndGeanePro, 1);
};

#endif //PNDGEANEPRO_H