PndPidBremCorrector.h

Go to the documentation of this file.

#ifndef PID_BREMCORRECTOR_H
#define PID_BREMCORRECTOR_H

#include <vector>
#include <map>

#include "FairTask.h"
#include "TObject.h"

#include "TString.h"
#include "TClonesArray.h"
#include "TVector3.h"

class PndPidCandidate;

class PndPidBremCorrected4Mom;
class PndEmcCluster;
class PndEmcBump;

// Path of file:
//  ----- $pandaroot/pid/PidCorr

class PndPidBremCorrector : public FairTask {

 public:
  PndPidBremCorrector();
  PndPidBremCorrector(const char *name, const char *title = "Pnd Task");

  // Destructor
  virtual ~PndPidBremCorrector();

  // Methods
  virtual InitStatus Init();

  virtual void Exec(Option_t *opt);

  void SetStorageOfData(Bool_t p = kTRUE) { fPersistance = p; };

  virtual void FinishTask(){};
  void SetOutputBranch(TString branch) { fTrackBranchNamePidHypo = branch; };

 private:
  PndPidBremCorrected4Mom *AddBremCorrected4Mom();

  Double_t GetSepPhotonE(PndPidCandidate *, std::vector<Int_t> &);
  Double_t GetMergPhotonE(PndPidCandidate *, std::vector<Int_t> &);

  void GetEmcPhiBumpList(Int_t iClust);
  Double_t GetBz(const TVector3 &pos);

  TClonesArray *fBumpArray;
  TClonesArray *fClusterArray;

  TClonesArray *fPhiBumpArray;

  TClonesArray *fChargedCandidateArray;

  TClonesArray *fBremCorrected4MomArray;

  Double_t fRecMomOfEle;
  Double_t fRecThetaOfEle;
  Double_t fRecPhiOfEle;
  Int_t fCharge;

  Double_t fSepPhotonE;
  Double_t fMergPhotonE;

  std::vector<PndEmcBump *> fEmcPhiBumpList;
  TString fTrackBranchNamePidHypo;

  Bool_t fPersistance; // switch to turn on/off storing the arrays to a file
  // Data members

  PndPidBremCorrector(const PndPidBremCorrector &L);
  PndPidBremCorrector &operator=(const PndPidBremCorrector &) { return *this; };

  ClassDef(PndPidBremCorrector, 1);
};

#endif