PndPropagationTest.h

Go to the documentation of this file.

//* $Id: */

// -------------------------------------------------------------------------
// -----                    PndPropagationTest header file                 -----
// -----                  Created 05/12/2010 by R. Karabowicz          -----
// -------------------------------------------------------------------------

#ifndef PndPropagationTest_H
#define PndPropagationTest_H 1

#include "PndPersistencyTask.h"

#include "FairGeanePro.h"

#include "TClonesArray.h"

#include <map>

class TH1;
class TH2;
class FairMCPoint;

class PndPropagationTest : public PndPersistencyTask {

 public:
  PndPropagationTest();

  PndPropagationTest(Int_t propParticle, Int_t iVerbose = 0);

  PndPropagationTest(const char *name, Int_t iVerbose);

  virtual ~PndPropagationTest();

  void SetDectectorBranchNames(TString branchName) { fDetArrays[branchName] = nullptr; }
  // Test
  void SetPhotonBranchName(TString val) { fPhotonDetectorName = val; }
  void SetPropPdg(int val) { fPropPdg = val; }

  void SetInnerExtrapolationRadius(double val) { fInnerExtrapoRadius = val; }

  virtual void Exec(Option_t *opt);

 protected:
  // TVector3 GetPhotonCorrection(int option, FairMCPoint *currentPoint = nullptr);
  TVector3 GetMCMomentumCorrection();                                  // uses MC truth momentum of photon hit
  TVector3 GetEnergyHitCorrection(TVector3 &innerPos);                 // uses MC truth energy plus direction between photon hit and electron hit
  Bool_t IsPhotonFromMother(FairMCPoint *point, Int_t motherPdg = 11); // checks if the point comes from a photon and if this comes from the primary electron.
 private:
  virtual void SetParContainers();

  virtual InitStatus Init();

  virtual InitStatus ReInit();

  void Reset();

  virtual void Finish();

  TClonesArray *fMcTracks = nullptr;
  std::map<TString, TClonesArray *> fDetArrays;
  TClonesArray *fPhotonDetector = nullptr;        
  TString fPhotonDetectorName = "IdealTubePoint"; 
  TClonesArray *fPhotonStart = nullptr;           
  FairGeanePro *fGeanePropagator = nullptr;       
  int fPropPdg = -211;                            

  int fGeaneNotFitted = 0;
  int fGenfitNotFitted = 0;

  double fInnerExtrapoRadius = -1;

  std::map<TString, TH1 *> fHistos;
  std::map<TString, TH2 *> f2DHistos;

  ClassDef(PndPropagationTest, 1);
};

#endif