update for mass study

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

diff --git a/PWGHF/hfe/AliAnalysisTaskHFECal.h b/PWGHF/hfe/AliAnalysisTaskHFECal.h
index f5f5de409738723d1e4082cca423324b2527227b..8e1a81f56e0bae889a7f4d331052fa8fb50d263b 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;
@@ -51,6 +52,7 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
   void SetOpeningAngleCut (Double_t openingAngle) {fOpeningAngleCut = openingAngle;};
   void SetInvariantMassCut (Double_t invmass) {fInvmassCut = invmass;};
+  void SetMassConstraint       (Bool_t MassConstraint)         { fSetMassConstraint    = MassConstraint; };
   void SetQAHist (int qahist) {fqahist = qahist;};
   AliHFEpid *GetPID() const { return fPID; }
   void SetRejectKinkMother(Bool_t rejectKinkMother = kFALSE) { fRejectKinkMother = rejectKinkMother; };
@@ -59,6 +61,8 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   double GetMCweight(double mcPi0pT);
   double GetMCweightEta(double mcEtapT);
   void FindTriggerClusters();
+  double MCEopMeanCorrection(double pTmc, float central);
+  double NsigmaCorrection(double tmpeta, float central);
  private:
   
   Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
@@ -83,6 +87,7 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
   AliHFEpidQAmanager   *fPIDqa;                //! PID QA manager
   Double_t             fOpeningAngleCut;       //openingAngle cut value
   Double_t             fInvmassCut;            //invariant mass cut value
+  Bool_t                fSetMassConstraint;            // Set mass constraint
  
   int ftriggers[48][60];//!
   int ftriggersCut[48][60];//!
@@ -91,6 +96,7 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
 
   TH1F                 *fNoEvents;             //! no of events
   THnSparseD           *fEMCAccE;              //! EMC acc
+  TH2F                 *hEMCAccE;              //! EMC acc
   TH1F                 *fTrkpt;                //! track pt
   TH2F                 *fTrkEovPBef;           //! track E/p before HFE pid
   TH2F                 *fTrkEovPAft;           //! track E/p after HFE pid
@@ -124,6 +130,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
@@ -164,8 +171,8 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
  THnSparseD            *fSameElecPtMC_pi0e_TPC;                //! Same inclusive ele pt
  TH1D                   *CheckNclust;  
  TH1D                   *CheckNits;  
- TH2D                   *Hpi0pTcheck; 
- TH2D                   *HETApTcheck; 
+ THnSparseD             *Hpi0pTcheck; 
+ THnSparseD             *HETApTcheck; 
  TH2D                   *HphopTcheck; 
  TH2D                   *HDpTcheck; 
  TH2D                   *HBpTcheck; 
@@ -173,7 +180,14 @@ class AliAnalysisTaskHFECal : public AliAnalysisTaskSE {
  TH2D                   *fMomDtoE; 
  TH2D                   *fLabelCheck;
  TH2D                   *fgeoFake;
+ TH2D                   *fFakeTrk0;
+ TH2D                   *fFakeTrk1;
+ TH2D                   *ftimingEle;
  
+ //<----- correction
+ TGraphErrors           *fnSigEtaCorr[7];
+
+
   AliAnalysisTaskHFECal(const AliAnalysisTaskHFECal&); // not implemented
   AliAnalysisTaskHFECal& operator=(const AliAnalysisTaskHFECal&); // not implemented