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"

class PndPidCandidate;

class PndPidBremCorrected4Mom;
class PndEmcCluster;
class PndEmcBump;

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

class PndPidBremCorrector: public FairTask
{

 public:

        PndPidBremCorrector();

        // 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 GetSepPhotonE(PndPidCandidate *, std::vector<Int_t>&);
        double GetMergPhotonE(PndPidCandidate *, std::vector<Int_t>&);

        void GetEmcPhiBumpList(int iClust);

        TClonesArray* fBumpArray;
        TClonesArray* fClusterArray;


        TClonesArray* fPhiBumpArray;

        TClonesArray* fChargedCandidateArray;
        TClonesArray* fNeutralCandidateArray;

        TClonesArray* fBremCorrected4MomArray;

        double fRecMomOfEle;
        double fRecThetaOfEle;
        double fRecPhiOfEle;
        int 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