updated pPb tasks

[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskHFECal.h

diff --git a/PWGHF/hfe/AliAnalysisTaskHFECal.h b/PWGHF/hfe/AliAnalysisTaskHFECal.h
index 800018ee97314393b77621c81b4645bad8b52df7..e6b0aa419d957732974b899fe853a612258eca5d 100644 (file)
--- a/PWGHF/hfe/AliAnalysisTaskHFECal.h
+++ b/PWGHF/hfe/AliAnalysisTaskHFECal.h
@@ -19,6 +19,7 @@
 class THnSparse;
 class TH2F;
 class TLorentzVector;
+class TGraphErrors;
 
 class AliEMCALTrack;
 class AliMagF;
@@ -50,16 +51,24 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   
   void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
   void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
+  void SetMimpTassCut (Double_t MimpTassCut) {fMimpTassCut = MimpTassCut;};
+  void SetMimNsigassCut (Double_t MimNsigassCut) {fMimNsigassCut = MimNsigassCut;};
   void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
+  void SetMassConstraint       (Bool_t MassConstraint)         { fSetMassConstraint    = MassConstraint; };
+  void SetMassWidthCut         (Bool_t MassWidthCut)           { fSetMassWidthCut      = MassWidthCut; };
+  void SetMassCalMethod        (Bool_t KFpart)         { fSetKFpart = KFpart; };
   void SetQAHist (int qahist) {fqahist = qahist;};
   AliHFEpid *GetPID() const { return fPID; }
   void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
   void SelectPhotonicElectron(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta);
+  void SelectPhotonicElectron2(Int_t itrack, Double_t cent, AliESDtrack *track, Bool_t &fFlagPhotonicElec, Bool_t &fFlagConvinatElec, Double_t nSig, Double_t shower, Double_t ep, Double_t mce, Double_t w, Int_t ibgevent, Bool_t tagpi0, Bool_t tageta);
+  double SumpT(Int_t itrack, AliESDtrack *track);
   void FindMother(TParticle* part, int &label, int &pid);
   double GetMCweight(double mcPi0pT);
   double GetMCweightEta(double mcEtapT);
   void FindTriggerClusters();
-  double NsigCorr(float cent);
+  double MCEopMeanCorrection(double pTmc, float central);
+  double NsigmaCorrection(double tmpeta, float central);
  private:
   
   Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
@@ -83,7 +92,12 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   AliHFEpid            *fPID;                  //! PID
   AliHFEpidQAmanager   *fPIDqa;                //! PID QA manager
   Double_t             fOpeningAngleCut;       //openingAngle cut value
+  Double_t             fMimpTassCut;   //openingAngle cut value
+  Double_t             fMimNsigassCut; //openingAngle cut value
   Double_t             fInvmassCut;            //invariant mass cut value
+  Bool_t                fSetMassConstraint;            // Set mass constraint
+  Bool_t                fSetMassWidthCut;              // Set mass constraint
+  Bool_t                fSetKFpart;            // Set mass constraint
  
   int ftriggers[48][60];//!
   int ftriggersCut[48][60];//!
@@ -104,6 +118,8 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   THnSparseD           *fInvmassULS;           //! Inv mass of ULS (e,e)
   THnSparseD           *fInvmassLSmc;          //! Inv mass of LS (e,e)
   THnSparseD           *fInvmassULSmc;         //! Inv mass of ULS (e,e)
+  TH2D         *fInvmassLSreco;                //! Inv mass of LS (e,e)
+  TH2D         *fInvmassULSreco;               //! Inv mass of ULS (e,e)
   TH2D         *fInvmassLSmc0;         //! Inv mass of ULS (e,e)
   TH2D         *fInvmassLSmc1;         //! Inv mass of ULS (e,e)
   TH2D         *fInvmassLSmc2;         //! Inv mass of ULS (e,e)
@@ -126,6 +142,7 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   
   TH1F                 *fCent;                 //! centrality
   THnSparseD           *fEleInfo;              //! EMC acc
+  THnSparseD           *fElenSigma;            //! EMC acc
   /*
   //<---- trigger info
   TH1F       *fClsEBftTrigCut; //Cluster E before trigger selection
@@ -175,8 +192,21 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
  TH2D                   *fMomDtoE; 
  TH2D                   *fLabelCheck;
  TH2D                   *fgeoFake;
+ TH2D                   *fFakeTrk0;
+ TH2D                   *fFakeTrk1;
  TH2D                   *ftimingEle;
- 
+ TH2D                   *fIncMaxE;
+ TH2D                   *fIncReco;
+ TH2D                   *fPhoReco;
+ TH2D                   *fSamReco; 
+ TH2D                   *fIncRecoMaxE;
+ TH2D                   *fPhoRecoMaxE;
+ TH2D                   *fSamRecoMaxE; 
+
+ //<----- correction
+ TGraphErrors           *fnSigEtaCorr[7];
+
+
   AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
   AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented