PndElectronCorrectionTest.h

Go to the documentation of this file.

//* $Id: */

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

#ifndef PndElectronCorrectionTest_H
#define PndElectronCorrectionTest_H 1

#include "PndPersistencyTask.h"

#include "FairGeanePro.h"

#include "TClonesArray.h"

#include <map>

class TH1;
class TH2;
class FairMCPoint;

class PndElectronCorrectionTest : public PndPersistencyTask {

 public:
  PndElectronCorrectionTest();

  PndElectronCorrectionTest(const char *name, Int_t iVerbose);

  virtual ~PndElectronCorrectionTest();

  void SetTrackBranchNames(TString branchName) { fTrackArrays[branchName] = nullptr; }
  // Test
  void SetEmcBranchName(TString val) { fEmcDetectorName = val; }

  virtual void Exec(Option_t *opt);

 protected:
  // TVector3 GetPhotonCorrection(int option, FairMCPoint *currentPoint = nullptr);
  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 *> fTrackArrays;
  TClonesArray *fEmcDetector = nullptr; 
  TString fEmcDetectorName = "EmcBump";

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

  ClassDef(PndElectronCorrectionTest, 1);
};

#endif