[u/mrichter/AliRoot.git] / PWG4 / UserTasks / EmcalTasks / AliAnalysisTaskEMCALPi0PbPb.h

#ifndef AliAnalysisTaskEMCALPi0PbPb_h
#define AliAnalysisTaskEMCALPi0PbPb_h

// $Id$

class TAxis;
class TClonesArray;
class TH1;
class TH2;
class TNtuple;
class TObjArray;
class AliAODCaloCells;
class AliAODCaloCluster;
class AliAODEvent;
class AliAODTrack;
class AliAODVertex;
class AliEMCALGeometry;
class AliEMCALRecoUtils;
class AliESDCaloCells;
class AliESDCaloCluster;
class AliESDEvent;
class AliESDTrack;
class AliESDVertex;
class AliESDtrackCuts;
class AliMCEvent;
class AliMCParticle;
class AliStaHeader;
class AliStaVertex;

#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         SetAsymMax(Double_t asymMax)                   { fAsymMax       = asymMax;   }
  void         SetCentrality(const char *n)                   { fCentVar       = n;         }
  void         SetCentralityRange(Double_t from, Double_t to) { fCentFrom=from; fCentTo=to; }
  void         SetClusName(const char *n)                     { fClusName      = n;         }
  void         SetDoAfterburner(Bool_t b)                     { fDoAfterburner = b;         }
  void         SetDoTrackMatWithGeom(Bool_t b)                { fDoTrMatGeom   = b;         }
  void         SetFillNtuple(Bool_t b)                        { fDoNtuple      = b;         }
  void         SetGeoName(const char *n)                      { fGeoName       = n;         }
  void         SetGeoUtils(AliEMCALGeometry *geo)             { fGeom          = geo;       }
  void         SetIsoDist(Double_t d)                         { fIsoDist       = d;         }
  void         SetL0TimeRange(Int_t l, Int_t h)               { fMinL0Time=l; fMaxL0Time=h; }
  void         SetMarkCells(const char *n)                    { fMarkCells     = n;         }
  void         SetMcMode(Bool_t b)                            { fMcMode        = b;         }
  void         SetEmbedMode(Bool_t b)                         { fEmbedMode     = b;         }
  void         SetMinClusEnergy(Double_t e)                   { fMinE          = e;         }
  void         SetMinEcc(Double_t ecc)                        { fMinEcc        = ecc;       }
  void         SetMinErat(Double_t erat)                      { fMinErat       = erat;      }
  void         SetMinNClustersPerTrack(Double_t m)            { fMinNClusPerTr = m;         }
  void         SetNminCells(Int_t n)                          { fNminCells     = n;         }
  void         SetPrimTrackCuts(AliESDtrackCuts *c)           { fPrimTrCuts    = c;         }
  void         SetRecoUtils(AliEMCALRecoUtils *reco)          { fReco          = reco;      }
  void         SetTrClassNames(const char *n)                 { fTrClassNames  = n;         }
  void         SetTrackCuts(AliESDtrackCuts *c)               { fTrCuts        = c;         }
  void         SetTrainMode(Bool_t b)                         { fTrainMode     = b;         }
  void         SetUseQualFlag(Bool_t b)                       { fUseQualFlag   = b;         }
  void         SetVertexRange(Double_t z1, Double_t z2)       { fVtxZMin=z1; fVtxZMax=z2;   }
  void         SetDoPhysicsSelection(Bool_t b)                { fDoPSel        = b;         }

 protected:
  virtual void CalcCaloTriggers();
  virtual void CalcClusterProps();
  virtual void CalcPrimTracks();
  virtual void CalcMcInfo();
  virtual void CalcTracks();
  virtual void ClusterAfterburner();
  virtual void FillCellHists();
  virtual void FillClusHists();
  virtual void FillNtuple();
  virtual void FillOtherHists();
  virtual void FillPionHists();
  virtual void FillMcHists();
  virtual void FillTrackHists();
  void         FillVertex(AliStaVertex *v, const AliESDVertex *esdv);
  void         FillVertex(AliStaVertex *v, const AliAODVertex *aodv);
  Double_t     GetCellIsolation(Double_t cEta, Double_t cPhi, Double_t radius=0.2)                        const;
  Double_t     GetCellEnergy(const AliVCluster *c)    const;
  Double_t     GetMaxCellEnergy(const AliVCluster *c) const { Short_t id=-1; return GetMaxCellEnergy(c,id); }
  Double_t     GetMaxCellEnergy(const AliVCluster *c, Short_t &id)                                        const;
  Int_t        GetNCells(const AliVCluster *c, Double_t emin=0.)                                          const;
  void         GetSigma(const AliVCluster *c, Double_t &sigmaMax, Double_t &sigmaMin)                     const;
  Double_t     GetTrackIsolation(Double_t cEta, Double_t cPhi, Double_t radius=0.2, Double_t pt=0.)       const;
  Double_t     GetTrigEnergy(const AliVCluster *c)                                                        const;
  Bool_t       IsShared(const AliVCluster *c)                                                             const;
  void         PrintDaughters(const AliVParticle *p, const TObjArray *arr, Int_t level=0)                 const;
  void         PrintDaughters(const AliMCParticle *p, const AliMCEvent *arr, Int_t level=0)               const;
  void         PrintTrackRefs(AliMCParticle *p)                                                           const;
  void         ProcessDaughters(AliVParticle *p, Int_t index, const TObjArray *arr);
  void         ProcessDaughters(AliMCParticle *p, Int_t index, const AliMCEvent *arr);

    // 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=-10cm)
  Double_t               fVtxZMax;                // max primary vertex z (def=+10cm)
  Bool_t                 fUseQualFlag;            // if true use quality flag for centrality
  TString                fClusName;               // cluster branch name (def="")
  Bool_t                 fDoNtuple;               // if true write out ntuple
  Bool_t                 fDoAfterburner;          // if true run after burner
  Double_t               fAsymMax;                // maximum energy asymmetry (def=1)
  Int_t                  fNminCells;              // minimum number of cells attached to cluster (def=1)
  Double_t               fMinE;                   // minimum cluster energy (def=0.1 GeV/c)
  Double_t               fMinErat;                // minimum emax/ec ratio (def=0)
  Double_t               fMinEcc;                 // minimum eccentricity (def=0)
  TString                fGeoName;                // geometry name (def = EMCAL_FIRSTYEARV1)
  Double_t               fMinNClusPerTr;          // minimum number of cluster per track (def=50)
  Double_t               fIsoDist;                // isolation distance (def=0.2)
  TString                fTrClassNames;           // trigger class names
  AliESDtrackCuts       *fTrCuts;                 // track cuts
  AliESDtrackCuts       *fPrimTrCuts;             // track cuts
  Bool_t                 fDoTrMatGeom;            // track matching including geometry
  Bool_t                 fTrainMode;              // train mode with minimal number of resources
  TString                fMarkCells;              // list of mark cells to monitor
  Int_t                  fMinL0Time;              // minimum accepted time for trigger
  Int_t                  fMaxL0Time;              // maximum accepted time for trigger
  Bool_t                 fMcMode;                 // monte carlo mode
  Bool_t                 fEmbedMode;              // embedding mode
  AliEMCALGeometry      *fGeom;                   // geometry utils
  AliEMCALRecoUtils     *fReco;                   // reco utils
  Bool_t                 fDoPSel;                 // if false then accept all events
    // derived members (ie with ! after //)
  Bool_t                 fIsGeoMatsSet;           //!indicate that geo matrices are set 
  ULong64_t              fNEvs;                   //!accepted events 
  TList                 *fOutput;                 //!container of output histograms
  TObjArray             *fTrClassNamesArr;        //!array of trig class names  
  AliESDEvent           *fEsdEv;                  //!pointer to input esd event
  AliAODEvent           *fAodEv;                  //!pointer to input aod event
  const TObjArray             *fRecPoints;              //!pointer to rec points (AliAnalysisTaskEMCALClusterizeFast)
  const TClonesArray          *fDigits;                 //!pointer to digits     (AliAnalysisTaskEMCALClusterizeFast)
  TObjArray             *fEsdClusters;            //!pointer to esd clusters
  AliESDCaloCells       *fEsdCells;               //!pointer to esd cells
  TObjArray             *fAodClusters;            //!pointer to aod clusters
  AliAODCaloCells       *fAodCells;               //!pointer to aod cells
  TAxis                 *fPtRanges;               //!pointer to pt ranges
  TObjArray             *fSelTracks;              //!pointer to selected tracks
  TObjArray             *fSelPrimTracks;          //!pointer to selected primary tracks
  Int_t                  fNAmpInTrigger;          //!number of cells to keep trigger statistic
  Float_t               *fAmpInTrigger;           //!amplitude for calo cells which are part of trigger
    // ntuple
  TTree                 *fNtuple;                 //!pointer to ntuple
  AliStaHeader          *fHeader;                 //!pointer to header
  AliStaVertex          *fPrimVert;               //!pointer to primary vertex
  AliStaVertex          *fSpdVert;                //!pointer to SPD vertex
  AliStaVertex          *fTpcVert;                //!pointer to TPC vertex
  TClonesArray          *fClusters;               //!pointer to clusters
  TClonesArray          *fTriggers;               //!pointer to triggers
  TClonesArray          *fMcParts;                //!pointer to mc particles
    // histograms
  TH1                   *fHCuts;                  //!histo for cuts
  TH1                   *fHVertexZ;               //!histo for vtxz
  TH1                   *fHVertexZ2;              //!histo for vtxz after vtx cuts
  TH1                   *fHCent;                  //!histo for cent
  TH1                   *fHCentQual;              //!histo for cent after quality flag cut
  TH1                   *fHTclsBeforeCuts;        //!histo for trigger classes before cuts
  TH1                   *fHTclsAfterCuts;         //!histo for trigger classes after cuts

    // histograms for cells
  TH2                  **fHColuRow;               //!histo for cell column and row
  TH2                  **fHColuRowE;              //!histo for cell column and row weight energy
  TH1                  **fHCellMult;              //!histo for cell multiplicity in module
  TH1                   *fHCellE;                 //!histo for cell energy
  TH1                   *fHCellH;                 //!histo for highest cell energy
  TH1                   *fHCellM;                 //!histo for mean cell energy (normalized to hit cells)
  TH1                   *fHCellM2;                //!histo for mean cell energy (normalized to all cells)
  TH1                  **fHCellFreqNoCut;         //!histo for cell frequency without cut
  TH1                  **fHCellFreqCut100M;       //!histo for cell frequency with cut 100MeV
  TH1                  **fHCellFreqCut300M;       //!histo for cell frequency with cut 300MeV
  TH1                  **fHCellFreqE;             //!histo for cell frequency weighted with energy
  TH1                  **fHCellCheckE;            //!histo for cell E distribution for given channels
    // histograms for clusters
  TH1                   *fHClustEccentricity;     //!histo for cluster eccentricity
  TH2                   *fHClustEtaPhi;           //!histo for cluster eta vs. phi
  TH2                   *fHClustEnergyPt;         //!histo for cluster energy vs. pT
  TH2                   *fHClustEnergySigma;      //!histo for cluster energy vs. variance over long axis 
  TH2                   *fHClustSigmaSigma;       //!histo for sigma vs. lambda_0 comparison
  TH2                   *fHClustNCellEnergyRatio; //!histo for cluster n cells vs. energy ratio
  TH2			*fHClustEnergyNCell;      //!histo for cluster energy vs. cluster n cells
    // histograms for primary tracks
  TH1			*fHPrimTrackPt;           //!histo for primary track pt
  TH1			*fHPrimTrackEta;          //!histo for primary track eta
  TH1			*fHPrimTrackPhi;           //!histo for primary track phi
    // histograms for track matching
  TH1                   *fHMatchDr;               //!histo for dR track cluster matching
  TH1                   *fHMatchDz;               //!histo for dZ track cluster matching
  TH1                   *fHMatchEp;               //!histo for E/p track cluster matching
    // histograms for pion candidates
  TH2                   *fHPionEtaPhi;            //!histo for pion eta vs. phi
  TH2                   *fHPionMggPt;             //!histo for pion mass vs. pT
  TH2                   *fHPionMggAsym;           //!histo for pion mass vs. asym
  TH2                   *fHPionMggDgg;            //!histo for pion mass vs. opening angle
  TH1                   *fHPionInvMasses[21];     //!histos for invariant mass plots 
    // histograms for MC

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

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

#ifndef AliStaObjs_h
#define AliStaObjs_h
class AliStaHeader
{
 public:
  AliStaHeader() : fRun(0), fOrbit(0), fPeriod(0), fBx(0), fL0(0), fL1(0), fL2(0),
                   fTrClassMask(0), fTrCluster(0), fOffTriggers(0), fFiredTriggers(),
                   fTcls(0), fV0Cent(0), fV0(0), fCl1Cent(0), fCl1(0), fTrCent(0), fTr(0),
                   fCqual(-1), fPsi(0), fPsiRes(0), fNSelTr(0), fNSelPrimTr(0), fNSelPrimTr1(0),
                   fNSelPrimTr2(0), fNCells(0), fNCells1(0), fNCells2(0), fNCells5(0), 
                   fNClus(0), fNClus1(0), fNClus2(0), fNClus5(0), 
                   fMaxCellE(0), fMaxClusE(0) {;}
  ULong64_t     GetEventId() const {
                  return (((ULong64_t)fPeriod << 36) |
                          ((ULong64_t)fOrbit  << 12) |
                          (ULong64_t)fBx); 
                }
  virtual ~AliStaHeader() {;}

 public:
  Int_t         fRun;            //         run number
  UInt_t        fOrbit;          //         orbit number
  UInt_t        fPeriod;         //         period number
  UShort_t      fBx;             //         bunch crossing id
  UInt_t        fL0;             //         l0 trigger bits
  UInt_t        fL1;             //         l1 trigger bits
  UShort_t      fL2;             //         l2 trigger bits
  ULong64_t     fTrClassMask;    //         trigger class mask
  UChar_t       fTrCluster;      //         trigger cluster mask
  UInt_t        fOffTriggers;    //         fired offline triggers for this event
  TString       fFiredTriggers;  //         string with fired triggers
  UInt_t        fTcls;           //         custom trigger definition
  Double32_t    fV0Cent;         //[0,0,16] v0 cent
  Double32_t    fV0;             //[0,0,16] v0 result used for cent 
  Double32_t    fCl1Cent;        //[0,0,16] cl1 cent
  Double32_t    fCl1;            //[0,0,16] cl1 result used for cent 
  Double32_t    fTrCent;         //[0,0,16] tr cent
  Double32_t    fTr;             //[0,0,16] tr result used for cent 
  Int_t         fCqual;          //         centrality quality
  Double32_t    fPsi;            //[0,0,16] event-plane angle
  Double32_t    fPsiRes;         //[0,0,16] event-plane ange resolution
  UShort_t      fNSelTr;         //         # selected tracks         
  UShort_t      fNSelPrimTr;     //         # selected tracks (primary)
  UShort_t      fNSelPrimTr1;    //         # selected tracks (primary) pt > 1 GeV/c
  UShort_t      fNSelPrimTr2;    //         # selected tracks (primary) pt > 2 GeV/c
  UShort_t      fNCells;         //         # cells
  UShort_t      fNCells1;        //         # cells > 1 GeV
  UShort_t      fNCells2;        //         # cells > 2 GeV
  UShort_t      fNCells5;        //         # cells > 5 GeV
  UShort_t      fNClus;          //         # clus
  UShort_t      fNClus1;         //         # clus > 1 GeV
  UShort_t      fNClus2;         //         # clus > 2 GeV
  UShort_t      fNClus5;         //         # clus > 5 GeV
  Double32_t    fMaxCellE;       //[0,0,16] maximum cell energy
  Double32_t    fMaxClusE;       //[0,0,16] maximum clus energy

  ClassDef(AliStaHeader,3) // Header class
};

class AliStaVertex
{
 public:
  AliStaVertex(Double_t x=0, Double_t y=0, Double_t z=0) : fVx(x), fVy(y), fVz(z), fVc(-1), fDisp(0), fZres(0),
                                                           fChi2(0), fSt(0), fIs3D(0), fIsZ(0) {;}
  virtual ~AliStaVertex() {;}

 public:
  Double_t      fVx;          //[0,0,16] vertex x
  Double_t      fVy;          //[0,0,16] vertex y
  Double_t      fVz;          //[0,0,16] vertex z
  Double_t      fVc;          //[0,0,16] number of contributors to vertex
  Double_t      fDisp;        //[0,0,16] dispersion
  Double_t      fZres;        //[0,0,16] z-resolution
  Double_t      fChi2;        //[0,0,16] chi2 of fit
  Bool_t        fSt;          //         status bit
  Bool_t        fIs3D;        //         is vertex from 3D
  Bool_t        fIsZ;         //         is vertex from Z only

  ClassDef(AliStaVertex,1) // Vertex class
};

class AliStaCluster : public TObject
{
 public:
    AliStaCluster() : TObject(), fE(0), fR(0), fEta(0), fPhi(0), fN(0), fN1(0), fN3(0), fIdMax(0), fEmax(0), fTmax(0), 
                      fDbc(-1), fDisp(-1), fM20(0), fM02(0), fEcc(0), fSig(0), fIsTrackM(0), fTrDz(0), fTrDr(-1), 
                      fTrEp(0), fTrIso(0), fTrIso1(0), fTrIso2(0), fCeIso(0), fCeCore(0), fIsTrigM(0), fTrigE(-1), 
                      fTrigMaskE(-1), fIsShared(0), fMcLabel(-1), fEmbE(0) {;}

 public:
  Double32_t    fE;                //[0,0,16] energy
  Double32_t    fR;                //[0,0,16] radius (cylinder)
  Double32_t    fEta;              //[0,0,16] eta
  Double32_t    fPhi;              //[0,0,16] phi
  UChar_t       fN;                //         number of cells
  UChar_t       fN1;               //         number of cells > 100 MeV
  UChar_t       fN3;               //         number of cells > 300 MeV
  UShort_t      fIdMax;            //         id maximum cell
  Double32_t    fEmax;             //[0,0,16] energy of maximum cell
  Double32_t    fTmax;             //[0,0,16] time of maximum cell
  Double32_t    fDbc;              //[0,0,16] distance to nearest bad channel
  Double32_t    fDisp;             //[0,0,16] cluster dispersion, for shape analysis
  Double32_t    fM20;              //[0,0,16] 2-nd moment along the main eigen axis
  Double32_t    fM02;              //[0,0,16] 2-nd moment along the second eigen axis
  Double32_t    fEcc;              //[0,0,16] eccentricity
  Double32_t    fSig;              //[0,0,16] sigma
  Bool_t        fIsTrackM;         //         if true then track values are set
  Double32_t    fTrDz;             //[0,0,16] dZ to nearest track
  Double32_t    fTrDr;             //[0,0,16] dR to nearest track (in x,y)
  Double32_t    fTrEp;             //[0,0,16] E/P to nearest track 
  Double32_t    fTrIso;            //[0,0,16] track isolation
  Double32_t    fTrIso1;           //[0,0,16] track isolation (pt>1GeV/c)
  Double32_t    fTrIso2;           //[0,0,16] track isolation (pt>2GeV/c)
  Double32_t    fCeIso;            //[0,0,16] cell isolation
  Double32_t    fCeCore;           //[0,0,16] cell content in R=0.025
  Bool_t        fIsTrigM;          //         if true then trigger values are set
  Double32_t    fTrigE;            //[0,0,16] trigger tower energy
  Double32_t    fTrigMaskE;        //[0,0,16] masked trigger tower energy
  Bool_t        fIsShared;         //         =true then extends across more than one super module
  Short_t       fMcLabel;          //         index of closest MC particle
  Double32_t    fEmbE;             //[0,0,16] sum of energy of embedded (MC) cells in cluster

  ClassDef(AliStaCluster,6) // Cluster class
};

class AliStaTrigger : public TObject
{
 public:
  AliStaTrigger() : TObject(), fE(0), fEta(0), fPhi(0), fAmp(0), fMinTime(0), fMaxTime(0) {}

 public:
  Double32_t    fE;                //[0,0,16] energy
  Double32_t    fEta;              //[0,0,16] eta
  Double32_t    fPhi;              //[0,0,16] phi
  Double32_t    fAmp;              //[0,0,16] amplitude
  Short_t       fMinTime;          //        minimum L0 "time"
  Short_t       fMaxTime;          //        maximum L0 "time"

  ClassDef(AliStaTrigger,1) // Trigger class
};

class AliStaPart : public TObject
{
 public:
  AliStaPart() : TObject(), fPt(0), fEta(0), fPhi(0), fVR(0), fVEta(0), fVPhi(0), fPid(0), fMo(-1), fDet(-2), 
                 fLab(-1), fNs(0) { memset(fDs,-1,sizeof(Short_t)*9); }

  Int_t         OnEmcal() const { return (fDet==8);  }
  Int_t         IsSim()   const { return (fDet!=-2); }
    
 public:
  Double32_t    fPt;               //[0,0,16] pt
  Double32_t    fEta;              //[0,0,16] eta
  Double32_t    fPhi;              //[0,0,16] phi
  Double32_t    fVR;               //[0,0,16] prod r (cylinder)
  Double32_t    fVEta;             //[0,0,16] prod eta
  Double32_t    fVPhi;             //[0,0,16] prod phi
  Short_t       fPid;              //         pid
  Short_t       fMo;               //         index of mother
  Short_t       fDet;              //         detector in which particle left trace (8 for EMCAL, see AliTrackReference.h)
    // the following must be filled before first usage
  Short_t       fLab;              //!        label (index in array)
  Short_t       fNs;               //!        number of daughters
  Short_t       fDs[9];            //!        daughters

  ClassDef(AliStaPart,1) // Particle class
};
#endif
Commit	Line	Data
6bf90832	1	#ifndef AliAnalysisTaskEMCALPi0PbPb_h
6bf90832	2	#define AliAnalysisTaskEMCALPi0PbPb_h
ea3fd2d5	3
	4	// $Id$
	5
fa443410	6	class TAxis;
f5d4ab70	7	class TClonesArray;
296ea9b4	8	class TH1;
296ea9b4	9	class TH2;
f5d4ab70	10	class TNtuple;
717fe7de	11	class TObjArray;
	12	class AliAODCaloCells;
	13	class AliAODCaloCluster;
ea3fd2d5	14	class AliAODEvent;
296ea9b4	15	class AliAODTrack;
788ca675	16	class AliAODVertex;
2ef5608f	17	class AliEMCALGeometry;
296ea9b4	18	class AliEMCALRecoUtils;
717fe7de	19	class AliESDCaloCells;
ea3fd2d5	20	class AliESDCaloCluster;
717fe7de	21	class AliESDEvent;
0ec74551	22	class AliESDTrack;
788ca675	23	class AliESDVertex;
0ec74551	24	class AliESDtrackCuts;
807016ea	25	class AliMCEvent;
38727e64	26	class AliMCParticle;
788ca675	27	class AliStaHeader;
788ca675	28	class AliStaVertex;
ea3fd2d5	29
	30	#include "AliAnalysisTaskSE.h"
	31
	32	class AliAnalysisTaskEMCALPi0PbPb : public AliAnalysisTaskSE {
	33	public:
4ea96211	34	AliAnalysisTaskEMCALPi0PbPb();
4ea96211	35	AliAnalysisTaskEMCALPi0PbPb(const char *name);
ea3fd2d5	36	virtual ~AliAnalysisTaskEMCALPi0PbPb();
ea3fd2d5	37
286b47a5	38	void UserCreateOutputObjects();
	39	void UserExec(Option_t *option);
	40	void Terminate(Option_t *);
717fe7de	41
b6c599fe	42	void SetAsymMax(Double_t asymMax) { fAsymMax = asymMax; }
b6c599fe	43	void SetCentrality(const char *n) { fCentVar = n; }
286b47a5	44	void SetCentralityRange(Double_t from, Double_t to) { fCentFrom=from; fCentTo=to; }
b6c599fe	45	void SetClusName(const char *n) { fClusName = n; }
a49742b5	46	void SetDoAfterburner(Bool_t b) { fDoAfterburner = b; }
b6c599fe	47	void SetDoTrackMatWithGeom(Bool_t b) { fDoTrMatGeom = b; }
	48	void SetFillNtuple(Bool_t b) { fDoNtuple = b; }
	49	void SetGeoName(const char *n) { fGeoName = n; }
2ef5608f	50	void SetGeoUtils(AliEMCALGeometry *geo) { fGeom = geo; }
b6c599fe	51	void SetIsoDist(Double_t d) { fIsoDist = d; }
38727e64	52	void SetL0TimeRange(Int_t l, Int_t h) { fMinL0Time=l; fMaxL0Time=h; }
3a952328	53	void SetMarkCells(const char *n) { fMarkCells = n; }
38727e64	54	void SetMcMode(Bool_t b) { fMcMode = b; }
cfd7d5b2	55	void SetEmbedMode(Bool_t b) { fEmbedMode = b; }
b6c599fe	56	void SetMinClusEnergy(Double_t e) { fMinE = e; }
	57	void SetMinEcc(Double_t ecc) { fMinEcc = ecc; }
	58	void SetMinErat(Double_t erat) { fMinErat = erat; }
3a952328	59	void SetMinNClustersPerTrack(Double_t m) { fMinNClusPerTr = m; }
b6c599fe	60	void SetNminCells(Int_t n) { fNminCells = n; }
3a952328	61	void SetPrimTrackCuts(AliESDtrackCuts *c) { fPrimTrCuts = c; }
38727e64	62	void SetRecoUtils(AliEMCALRecoUtils *reco) { fReco = reco; }
b6c599fe	63	void SetTrClassNames(const char *n) { fTrClassNames = n; }
b6c599fe	64	void SetTrackCuts(AliESDtrackCuts *c) { fTrCuts = c; }
3a952328	65	void SetTrainMode(Bool_t b) { fTrainMode = b; }
b6c599fe	66	void SetUseQualFlag(Bool_t b) { fUseQualFlag = b; }
d595acbb	67	void SetVertexRange(Double_t z1, Double_t z2) { fVtxZMin=z1; fVtxZMax=z2; }
807016ea	68	void SetDoPhysicsSelection(Bool_t b) { fDoPSel = b; }
717fe7de	69
717fe7de	70	protected:
38727e64	71	virtual void CalcCaloTriggers();
296ea9b4	72	virtual void CalcClusterProps();
b6c599fe	73	virtual void CalcPrimTracks();
38727e64	74	virtual void CalcMcInfo();
3a952328	75	virtual void CalcTracks();
323834f0	76	virtual void ClusterAfterburner();
76332037	77	virtual void FillCellHists();
76332037	78	virtual void FillClusHists();
788ca675	79	virtual void FillNtuple();
323834f0	80	virtual void FillOtherHists();
788ca675	81	virtual void FillPionHists();
38727e64	82	virtual void FillMcHists();
f5e0f1e2	83	virtual void FillTrackHists();
788ca675	84	void FillVertex(AliStaVertex v, const AliESDVertex esdv);
788ca675	85	void FillVertex(AliStaVertex v, const AliAODVertex aodv);
0fbe8d4f	86	Double_t GetCellIsolation(Double_t cEta, Double_t cPhi, Double_t radius=0.2) const;
	87	Double_t GetCellEnergy(const AliVCluster *c) const;
	88	Double_t GetMaxCellEnergy(const AliVCluster *c) const { Short_t id=-1; return GetMaxCellEnergy(c,id); }
	89	Double_t GetMaxCellEnergy(const AliVCluster *c, Short_t &id) const;
	90	Int_t GetNCells(const AliVCluster *c, Double_t emin=0.) const;
	91	void GetSigma(const AliVCluster *c, Double_t &sigmaMax, Double_t &sigmaMin) const;
	92	Double_t GetTrackIsolation(Double_t cEta, Double_t cPhi, Double_t radius=0.2, Double_t pt=0.) const;
	93	Double_t GetTrigEnergy(const AliVCluster *c) const;
	94	Bool_t IsShared(const AliVCluster *c) const;
807016ea	95	void PrintDaughters(const AliVParticle p, const TObjArray arr, Int_t level=0) const;
807016ea	96	void PrintDaughters(const AliMCParticle p, const AliMCEvent arr, Int_t level=0) const;
38727e64	97	void PrintTrackRefs(AliMCParticle *p) const;
807016ea	98	void ProcessDaughters(AliVParticle p, Int_t index, const TObjArray arr);
807016ea	99	void ProcessDaughters(AliMCParticle p, Int_t index, const AliMCEvent arr);
286b47a5	100
717fe7de	101	// input members
6eb6260e	102	TString fCentVar; // variable for centrality determination
	103	Double_t fCentFrom; // min centrality (def=0)
	104	Double_t fCentTo; // max centrality (def=100)
	105	Double_t fVtxZMin; // min primary vertex z (def=-10cm)
	106	Double_t fVtxZMax; // max primary vertex z (def=+10cm)
	107	Bool_t fUseQualFlag; // if true use quality flag for centrality
	108	TString fClusName; // cluster branch name (def="")
	109	Bool_t fDoNtuple; // if true write out ntuple
a49742b5	110	Bool_t fDoAfterburner; // if true run after burner
f224d35b	111	Double_t fAsymMax; // maximum energy asymmetry (def=1)
f224d35b	112	Int_t fNminCells; // minimum number of cells attached to cluster (def=1)
296ea9b4	113	Double_t fMinE; // minimum cluster energy (def=0.1 GeV/c)
f224d35b	114	Double_t fMinErat; // minimum emax/ec ratio (def=0)
f224d35b	115	Double_t fMinEcc; // minimum eccentricity (def=0)
6bf90832	116	TString fGeoName; // geometry name (def = EMCAL_FIRSTYEARV1)
b6c599fe	117	Double_t fMinNClusPerTr; // minimum number of cluster per track (def=50)
296ea9b4	118	Double_t fIsoDist; // isolation distance (def=0.2)
b3ee6797	119	TString fTrClassNames; // trigger class names
b3ee6797	120	AliESDtrackCuts *fTrCuts; // track cuts
3a952328	121	AliESDtrackCuts *fPrimTrCuts; // track cuts
b6c599fe	122	Bool_t fDoTrMatGeom; // track matching including geometry
3a952328	123	Bool_t fTrainMode; // train mode with minimal number of resources
	124	TString fMarkCells; // list of mark cells to monitor
	125	Int_t fMinL0Time; // minimum accepted time for trigger
	126	Int_t fMaxL0Time; // maximum accepted time for trigger
38727e64	127	Bool_t fMcMode; // monte carlo mode
cfd7d5b2	128	Bool_t fEmbedMode; // embedding mode
2ef5608f	129	AliEMCALGeometry *fGeom; // geometry utils
38727e64	130	AliEMCALRecoUtils *fReco; // reco utils
807016ea	131	Bool_t fDoPSel; // if false then accept all events
f5d4ab70	132	// derived members (ie with ! after //)
27c2e3d9	133	Bool_t fIsGeoMatsSet; //!indicate that geo matrices are set
d9f26424	134	ULong64_t fNEvs; //!accepted events
6eb6260e	135	TList *fOutput; //!container of output histograms
b3ee6797	136	TObjArray *fTrClassNamesArr; //!array of trig class names
6eb6260e	137	AliESDEvent *fEsdEv; //!pointer to input esd event
6eb6260e	138	AliAODEvent *fAodEv; //!pointer to input aod event
c2fe5f0e	139	const TObjArray *fRecPoints; //!pointer to rec points (AliAnalysisTaskEMCALClusterizeFast)
c2fe5f0e	140	const TClonesArray *fDigits; //!pointer to digits (AliAnalysisTaskEMCALClusterizeFast)
6eb6260e	141	TObjArray *fEsdClusters; //!pointer to esd clusters
	142	AliESDCaloCells *fEsdCells; //!pointer to esd cells
	143	TObjArray *fAodClusters; //!pointer to aod clusters
	144	AliAODCaloCells *fAodCells; //!pointer to aod cells
	145	TAxis *fPtRanges; //!pointer to pt ranges
296ea9b4	146	TObjArray *fSelTracks; //!pointer to selected tracks
3a952328	147	TObjArray *fSelPrimTracks; //!pointer to selected primary tracks
	148	Int_t fNAmpInTrigger; //!number of cells to keep trigger statistic
	149	Float_t *fAmpInTrigger; //!amplitude for calo cells which are part of trigger
788ca675	150	// ntuple
	151	TTree *fNtuple; //!pointer to ntuple
	152	AliStaHeader *fHeader; //!pointer to header
	153	AliStaVertex *fPrimVert; //!pointer to primary vertex
	154	AliStaVertex *fSpdVert; //!pointer to SPD vertex
	155	AliStaVertex *fTpcVert; //!pointer to TPC vertex
	156	TClonesArray *fClusters; //!pointer to clusters
3a952328	157	TClonesArray *fTriggers; //!pointer to triggers
807016ea	158	TClonesArray *fMcParts; //!pointer to mc particles
717fe7de	159	// histograms
296ea9b4	160	TH1 *fHCuts; //!histo for cuts
	161	TH1 *fHVertexZ; //!histo for vtxz
	162	TH1 *fHVertexZ2; //!histo for vtxz after vtx cuts
	163	TH1 *fHCent; //!histo for cent
	164	TH1 *fHCentQual; //!histo for cent after quality flag cut
b3ee6797	165	TH1 *fHTclsBeforeCuts; //!histo for trigger classes before cuts
	166	TH1 *fHTclsAfterCuts; //!histo for trigger classes after cuts
	167
d595acbb	168	// histograms for cells
296ea9b4	169	TH2 **fHColuRow; //!histo for cell column and row
	170	TH2 **fHColuRowE; //!histo for cell column and row weight energy
	171	TH1 **fHCellMult; //!histo for cell multiplicity in module
	172	TH1 *fHCellE; //!histo for cell energy
	173	TH1 *fHCellH; //!histo for highest cell energy
	174	TH1 *fHCellM; //!histo for mean cell energy (normalized to hit cells)
	175	TH1 *fHCellM2; //!histo for mean cell energy (normalized to all cells)
	176	TH1 **fHCellFreqNoCut; //!histo for cell frequency without cut
2e4d8148	177	TH1 **fHCellFreqCut100M; //!histo for cell frequency with cut 100MeV
	178	TH1 **fHCellFreqCut300M; //!histo for cell frequency with cut 300MeV
	179	TH1 **fHCellFreqE; //!histo for cell frequency weighted with energy
296ea9b4	180	TH1 **fHCellCheckE; //!histo for cell E distribution for given channels
fa443410	181	// histograms for clusters
296ea9b4	182	TH1 *fHClustEccentricity; //!histo for cluster eccentricity
	183	TH2 *fHClustEtaPhi; //!histo for cluster eta vs. phi
	184	TH2 *fHClustEnergyPt; //!histo for cluster energy vs. pT
	185	TH2 *fHClustEnergySigma; //!histo for cluster energy vs. variance over long axis
	186	TH2 *fHClustSigmaSigma; //!histo for sigma vs. lambda_0 comparison
f5e0f1e2	187	TH2 *fHClustNCellEnergyRatio; //!histo for cluster n cells vs. energy ratio
	188	TH2 *fHClustEnergyNCell; //!histo for cluster energy vs. cluster n cells
	189	// histograms for primary tracks
	190	TH1 *fHPrimTrackPt; //!histo for primary track pt
	191	TH1 *fHPrimTrackEta; //!histo for primary track eta
	192	TH1 *fHPrimTrackPhi; //!histo for primary track phi
b6c599fe	193	// histograms for track matching
	194	TH1 *fHMatchDr; //!histo for dR track cluster matching
	195	TH1 *fHMatchDz; //!histo for dZ track cluster matching
	196	TH1 *fHMatchEp; //!histo for E/p track cluster matching
fa443410	197	// histograms for pion candidates
296ea9b4	198	TH2 *fHPionEtaPhi; //!histo for pion eta vs. phi
	199	TH2 *fHPionMggPt; //!histo for pion mass vs. pT
	200	TH2 *fHPionMggAsym; //!histo for pion mass vs. asym
	201	TH2 *fHPionMggDgg; //!histo for pion mass vs. opening angle
	202	TH1 *fHPionInvMasses[21]; //!histos for invariant mass plots
38727e64	203	// histograms for MC
ea3fd2d5	204
ea3fd2d5	205	private:
717fe7de	206	AliAnalysisTaskEMCALPi0PbPb(const AliAnalysisTaskEMCALPi0PbPb&); // not implemented
717fe7de	207	AliAnalysisTaskEMCALPi0PbPb &operator=(const AliAnalysisTaskEMCALPi0PbPb&); // not implemented
ea3fd2d5	208
c2fe5f0e	209	ClassDef(AliAnalysisTaskEMCALPi0PbPb, 11) // Analysis task for neutral pions in Pb+Pb
ea3fd2d5	210	};
ea3fd2d5	211	#endif
788ca675	212
	213	#ifndef AliStaObjs_h
	214	#define AliStaObjs_h
	215	class AliStaHeader
	216	{
	217	public:
	218	AliStaHeader() : fRun(0), fOrbit(0), fPeriod(0), fBx(0), fL0(0), fL1(0), fL2(0),
	219	fTrClassMask(0), fTrCluster(0), fOffTriggers(0), fFiredTriggers(),
5fe1ca23	220	fTcls(0), fV0Cent(0), fV0(0), fCl1Cent(0), fCl1(0), fTrCent(0), fTr(0),
f5e0f1e2	221	fCqual(-1), fPsi(0), fPsiRes(0), fNSelTr(0), fNSelPrimTr(0), fNSelPrimTr1(0),
f5e0f1e2	222	fNSelPrimTr2(0), fNCells(0), fNCells1(0), fNCells2(0), fNCells5(0),
5fe1ca23	223	fNClus(0), fNClus1(0), fNClus2(0), fNClus5(0),
5fe1ca23	224	fMaxCellE(0), fMaxClusE(0) {;}
807016ea	225	ULong64_t GetEventId() const {
	226	return (((ULong64_t)fPeriod << 36) \|
	227	((ULong64_t)fOrbit << 12) \|
	228	(ULong64_t)fBx);
	229	}
788ca675	230	virtual ~AliStaHeader() {;}
788ca675	231
	232	public:
	233	Int_t fRun; // run number
	234	UInt_t fOrbit; // orbit number
	235	UInt_t fPeriod; // period number
	236	UShort_t fBx; // bunch crossing id
	237	UInt_t fL0; // l0 trigger bits
	238	UInt_t fL1; // l1 trigger bits
	239	UShort_t fL2; // l2 trigger bits
	240	ULong64_t fTrClassMask; // trigger class mask
	241	UChar_t fTrCluster; // trigger cluster mask
	242	UInt_t fOffTriggers; // fired offline triggers for this event
	243	TString fFiredTriggers; // string with fired triggers
	244	UInt_t fTcls; // custom trigger definition
	245	Double32_t fV0Cent; //[0,0,16] v0 cent
5fe1ca23	246	Double32_t fV0; //[0,0,16] v0 result used for cent
788ca675	247	Double32_t fCl1Cent; //[0,0,16] cl1 cent
5fe1ca23	248	Double32_t fCl1; //[0,0,16] cl1 result used for cent
788ca675	249	Double32_t fTrCent; //[0,0,16] tr cent
5fe1ca23	250	Double32_t fTr; //[0,0,16] tr result used for cent
788ca675	251	Int_t fCqual; // centrality quality
b6c599fe	252	Double32_t fPsi; //[0,0,16] event-plane angle
b6c599fe	253	Double32_t fPsiRes; //[0,0,16] event-plane ange resolution
5fe1ca23	254	UShort_t fNSelTr; // # selected tracks
5fe1ca23	255	UShort_t fNSelPrimTr; // # selected tracks (primary)
f5e0f1e2	256	UShort_t fNSelPrimTr1; // # selected tracks (primary) pt > 1 GeV/c
f5e0f1e2	257	UShort_t fNSelPrimTr2; // # selected tracks (primary) pt > 2 GeV/c
5fe1ca23	258	UShort_t fNCells; // # cells
	259	UShort_t fNCells1; // # cells > 1 GeV
	260	UShort_t fNCells2; // # cells > 2 GeV
	261	UShort_t fNCells5; // # cells > 5 GeV
	262	UShort_t fNClus; // # clus
	263	UShort_t fNClus1; // # clus > 1 GeV
	264	UShort_t fNClus2; // # clus > 2 GeV
	265	UShort_t fNClus5; // # clus > 5 GeV
	266	Double32_t fMaxCellE; //[0,0,16] maximum cell energy
	267	Double32_t fMaxClusE; //[0,0,16] maximum clus energy
788ca675	268
5fe1ca23	269	ClassDef(AliStaHeader,3) // Header class
788ca675	270	};
	271
	272	class AliStaVertex
	273	{
	274	public:
	275	AliStaVertex(Double_t x=0, Double_t y=0, Double_t z=0) : fVx(x), fVy(y), fVz(z), fVc(-1), fDisp(0), fZres(0),
	276	fChi2(0), fSt(0), fIs3D(0), fIsZ(0) {;}
	277	virtual ~AliStaVertex() {;}
	278
	279	public:
	280	Double_t fVx; //[0,0,16] vertex x
	281	Double_t fVy; //[0,0,16] vertex y
	282	Double_t fVz; //[0,0,16] vertex z
	283	Double_t fVc; //[0,0,16] number of contributors to vertex
	284	Double_t fDisp; //[0,0,16] dispersion
	285	Double_t fZres; //[0,0,16] z-resolution
	286	Double_t fChi2; //[0,0,16] chi2 of fit
	287	Bool_t fSt; // status bit
	288	Bool_t fIs3D; // is vertex from 3D
	289	Bool_t fIsZ; // is vertex from Z only
	290
	291	ClassDef(AliStaVertex,1) // Vertex class
	292	};
	293
	294	class AliStaCluster : public TObject
	295	{
	296	public:
c2fe5f0e	297	AliStaCluster() : TObject(), fE(0), fR(0), fEta(0), fPhi(0), fN(0), fN1(0), fN3(0), fIdMax(0), fEmax(0), fTmax(0),
	298	fDbc(-1), fDisp(-1), fM20(0), fM02(0), fEcc(0), fSig(0), fIsTrackM(0), fTrDz(0), fTrDr(-1),
	299	fTrEp(0), fTrIso(0), fTrIso1(0), fTrIso2(0), fCeIso(0), fCeCore(0), fIsTrigM(0), fTrigE(-1),
	300	fTrigMaskE(-1), fIsShared(0), fMcLabel(-1), fEmbE(0) {;}
788ca675	301
	302	public:
	303	Double32_t fE; //[0,0,16] energy
38727e64	304	Double32_t fR; //[0,0,16] radius (cylinder)
788ca675	305	Double32_t fEta; //[0,0,16] eta
	306	Double32_t fPhi; //[0,0,16] phi
	307	UChar_t fN; // number of cells
	308	UChar_t fN1; // number of cells > 100 MeV
	309	UChar_t fN3; // number of cells > 300 MeV
	310	UShort_t fIdMax; // id maximum cell
	311	Double32_t fEmax; //[0,0,16] energy of maximum cell
c2fe5f0e	312	Double32_t fTmax; //[0,0,16] time of maximum cell
788ca675	313	Double32_t fDbc; //[0,0,16] distance to nearest bad channel
	314	Double32_t fDisp; //[0,0,16] cluster dispersion, for shape analysis
	315	Double32_t fM20; //[0,0,16] 2-nd moment along the main eigen axis
	316	Double32_t fM02; //[0,0,16] 2-nd moment along the second eigen axis
	317	Double32_t fEcc; //[0,0,16] eccentricity
	318	Double32_t fSig; //[0,0,16] sigma
f3582e89	319	Bool_t fIsTrackM; // if true then track values are set
788ca675	320	Double32_t fTrDz; //[0,0,16] dZ to nearest track
3a952328	321	Double32_t fTrDr; //[0,0,16] dR to nearest track (in x,y)
788ca675	322	Double32_t fTrEp; //[0,0,16] E/P to nearest track
788ca675	323	Double32_t fTrIso; //[0,0,16] track isolation
b6c599fe	324	Double32_t fTrIso1; //[0,0,16] track isolation (pt>1GeV/c)
b6c599fe	325	Double32_t fTrIso2; //[0,0,16] track isolation (pt>2GeV/c)
788ca675	326	Double32_t fCeIso; //[0,0,16] cell isolation
3a952328	327	Double32_t fCeCore; //[0,0,16] cell content in R=0.025
f3582e89	328	Bool_t fIsTrigM; // if true then trigger values are set
3a952328	329	Double32_t fTrigE; //[0,0,16] trigger tower energy
3a952328	330	Double32_t fTrigMaskE; //[0,0,16] masked trigger tower energy
0fbe8d4f	331	Bool_t fIsShared; // =true then extends across more than one super module
56fd6cb2	332	Short_t fMcLabel; // index of closest MC particle
c2fe5f0e	333	Double32_t fEmbE; //[0,0,16] sum of energy of embedded (MC) cells in cluster
3a952328	334
c2fe5f0e	335	ClassDef(AliStaCluster,6) // Cluster class
3a952328	336	};
	337
	338	class AliStaTrigger : public TObject
	339	{
	340	public:
	341	AliStaTrigger() : TObject(), fE(0), fEta(0), fPhi(0), fAmp(0), fMinTime(0), fMaxTime(0) {}
	342
	343	public:
	344	Double32_t fE; //[0,0,16] energy
	345	Double32_t fEta; //[0,0,16] eta
	346	Double32_t fPhi; //[0,0,16] phi
	347	Double32_t fAmp; //[0,0,16] amplitude
807016ea	348	Short_t fMinTime; // minimum L0 "time"
807016ea	349	Short_t fMaxTime; // maximum L0 "time"
788ca675	350
3a952328	351	ClassDef(AliStaTrigger,1) // Trigger class
788ca675	352	};
38727e64	353
38727e64	354	class AliStaPart : public TObject
	355	{
	356	public:
8c56d760	357	AliStaPart() : TObject(), fPt(0), fEta(0), fPhi(0), fVR(0), fVEta(0), fVPhi(0), fPid(0), fMo(-1), fDet(-2),
6e5c43d8	358	fLab(-1), fNs(0) { memset(fDs,-1,sizeof(Short_t)*9); }
8c56d760	359
	360	Int_t OnEmcal() const { return (fDet==8); }
	361	Int_t IsSim() const { return (fDet!=-2); }
807016ea	362
38727e64	363	public:
807016ea	364	Double32_t fPt; //[0,0,16] pt
38727e64	365	Double32_t fEta; //[0,0,16] eta
38727e64	366	Double32_t fPhi; //[0,0,16] phi
807016ea	367	Double32_t fVR; //[0,0,16] prod r (cylinder)
	368	Double32_t fVEta; //[0,0,16] prod eta
	369	Double32_t fVPhi; //[0,0,16] prod phi
	370	Short_t fPid; // pid
	371	Short_t fMo; // index of mother
8c56d760	372	Short_t fDet; // detector in which particle left trace (8 for EMCAL, see AliTrackReference.h)
	373	// the following must be filled before first usage
	374	Short_t fLab; //! label (index in array)
807016ea	375	Short_t fNs; //! number of daughters
8c56d760	376	Short_t fDs[9]; //! daughters
38727e64	377
56fd6cb2	378	ClassDef(AliStaPart,1) // Particle class
38727e64	379	};
788ca675	380	#endif