Added new cuts on number of crossed rows and findable clusters

[u/mrichter/AliRoot.git] / PWG4 / CaloCalib / AliAnalysisTaskEMCALPi0PbPb.h

#ifndef AliAnalysisTaskEMCALPi0PbPb_cxx
#define AliAnalysisTaskEMCALPi0PbPb_cxx

// $Id$

class TAxis;
class TH1F;
class TH2F;
class TClonesArray;
class TObjArray;
class AliAODCaloCells;
class AliAODCaloCluster;
class AliAODEvent;
class AliESDCaloCells;
class AliESDCaloCluster;
class AliESDEvent;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskEMCALPi0PbPb : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskEMCALPi0PbPb();
  AliAnalysisTaskEMCALPi0PbPb(const char *name);
  virtual ~AliAnalysisTaskEMCALPi0PbPb(); 
  
  void         UserCreateOutputObjects();
  void         UserExec(Option_t *option);
  void         Terminate(Option_t *);

  void         SetCentrality(const char *name)                { fCentVar = name;            }
  void         SetCentralityRange(Double_t from, Double_t to) { fCentFrom=from; fCentTo=to; }
  void         SetClusName(const char *name)                  { fClusName = name;           }
  void         SetVertexRange(Double_t z1, Double_t z2)       { fVtxZMin=z1; fVtxZMax=z2;   }
  void         SetUseQualFlag(Bool_t b)                       { fUseQualFlag = b;           }

 protected:
  virtual void FillCellHists();
  virtual void FillClusHists();
  virtual void FillPionHists();

    // input members
  TString                fCentVar;              // variable for centrality determination
  Double_t               fCentFrom;             // min centrality (def=0)
  Double_t               fCentTo;               // max centrality (def=100)
  Double_t               fVtxZMin;              // min primary vertex z (def=-7cm)
  Double_t               fVtxZMax;              // max primary vertex z (def=+7cm)
  Bool_t                 fUseQualFlag;          // if true use quality flag for centrality
  TString                fClusName;             // cluster branch name (def="")
     // derived members (ie with ! after //)
  TList                 *fOutput;               //!container of output histograms
  AliESDEvent           *fEsdEv;                //!pointer to input esd event
  AliAODEvent           *fAodEv;                //!pointer to input aod event
  const TObjArray       *fRecPoints;            //!pointer to rec points (AliAnalysisTaskEMCALClusterizeFast)
  const TObjArray       *fEsdClusters;          //!pointer to esd clusters
  const AliESDCaloCells *fEsdCells;             //!pointer to esd cells
  const TObjArray       *fAodClusters;          //!pointer to aod clusters
  const AliAODCaloCells *fAodCells;             //!pointer to aod cells
  TAxis                 *fPtRanges;             //!pointer to pt ranges
    // histograms
  TH1F                  *fHCuts;                //!histo for cuts
  TH1F                  *fHVertexZ;             //!histo for vtxz
  TH1F                  *fHVertexZ2;            //!histo for vtxz after vtx cuts
  TH1F                  *fHCent;                //!histo for cent
  TH1F                  *fHCentQual;            //!histo for cent after quality flag cut
    // histograms for clusters
  TH2F                  *fHClustEtaPhi;         //!histo for cluster eta vs. phi
  TH2F                  *fHClustEnergyPt;       //!histo for cluster energy vs. pT
  TH2F                  *fHClustEnergySigma;    //!histo for cluster 
  TH2F                  *fHClustNTowEnergyRatio;//!histo for cluster n tow vs. energy ratio
    // histograms for pion candidates
  TH2F                  *fHPionEtaPhi;          //!histo for pion eta vs. phi
  TH2F                  *fHPionMggPt;           //!histo for pion mass vs. pT
  TH2F                  *fHPionMggAsym;         //!histo for pion mass vs. asym
  TH1F                  *fHPionInvMasses[20];   //!histos for invariant mass plots 

 private:
  AliAnalysisTaskEMCALPi0PbPb(const AliAnalysisTaskEMCALPi0PbPb&);            // not implemented
  AliAnalysisTaskEMCALPi0PbPb &operator=(const AliAnalysisTaskEMCALPi0PbPb&); // not implemented

  ClassDef(AliAnalysisTaskEMCALPi0PbPb, 1); // Analysis task for neutral pions in Pb+Pb
};

#endif