PWGGA/EMCALTasks/AliAnalysisTaskEMCALIsoPhoton.h

   1 #ifndef AliAnalysisTaskEMCALIsoPhoton_h
   2 #define AliAnalysisTaskEMCALIsoPhoton_h
   3
   4 // $Id$
   5
   6 class TH1F;
   7 class TH2F;
   8 class TH3F;
   9 class THnSparse;
  10 class TList;
  11 class TObjArray;
  12 class AliEMCALGeometry;
  13 class AliOADBContainer;
  14 class AliESDCaloCells;
  15 class AliESDEvent;
  16 class AliESDtrack;
  17 class AliESDtrackCuts;
  18 class AliAODEvent;
  19 class AliAODCaloCells;
  20 class AliVCluster;
  21 class AliMCEvent;
  22 class AliStack;
  23 class TParticle;
  24 class AliAODMCParticle;
  25 class TGeoHMatrix;
  26
  27 #include "AliAnalysisTaskSE.h"
  28
  29 class AliAnalysisTaskEMCALIsoPhoton : public AliAnalysisTaskSE {
  30  public:
  31   AliAnalysisTaskEMCALIsoPhoton();
  32   AliAnalysisTaskEMCALIsoPhoton(const char *name);
  33   virtual ~AliAnalysisTaskEMCALIsoPhoton() {}
  34
  35   void                   UserCreateOutputObjects();
  36   void                   UserExec(Option_t *option);
  37   void                   Terminate(Option_t *);
  38
  39   void                   GetCeIso(TVector3 vec, Int_t maxid, Float_t &iso, Float_t &phiband, Float_t &core, Double_t EtCl);
  40   Double_t               GetCrossEnergy(const AliVCluster *cluster, Short_t &idmax);
  41   Double_t               GetMaxCellEnergy(const AliVCluster *cluster, Short_t &id) const;
  42   void                   GetTrIso(TVector3 vec, Float_t &iso, Float_t &phiband, Float_t &core);
  43   Double_t               GetTrackMatchedPt(Int_t matchIndex);
  44   void                   FillClusHists();
  45   void                   FillMcHists();
  46   void                   FillQA();
  47   Float_t                GetClusSource(const AliVCluster *cluster);
  48   void                   FollowGamma();
  49   void                   GetDaughtersInfo(int firstd, int lastd, int selfid, const char *indputindent);
  50   Float_t                GetMcPtSumInCone(Float_t etaclus, Float_t phiclus, Float_t R);
  51   void                   LoopOnCells();
  52   bool                   IsExotic(AliVCluster *c);
  53   void                   SetExotCut(Double_t c)                 { fExoticCut          = c;       }
  54   void                   SetGeoName(const char *n)              { fGeoName            = n;       }
  55   void                   SetIsoConeR(Double_t r)                { fIsoConeR           = r;       }
  56   void                   SetPeriod(const char *n)               { fPeriod             = n;       }
  57   void                   SetTriggerBit(const char *tb)          { fTrigBit            = tb;      }
  58   void                   SetPrimTrackCuts(AliESDtrackCuts *c)   { fPrTrCuts           = c;       }
  59   void                   SetComplTrackCuts(AliESDtrackCuts *c)  { fCompTrCuts         = c;       }
  60   void                   SetTrainMode(Bool_t t)                 { fIsTrain            = t;       }
  61   void                   SetMcMode(Bool_t mc)                   { fIsMc               = mc;      }
  62   void                   SetDebugOn(Bool_t d)                   { fDebug              = d;       }
  63   void                   SetPathStringSelect(char *p)           { fPathStrOpt         = p;       }
  64   void                   SetEtCut(Double_t ec)                  { fECut               = ec;      }
  65   void                   SetImportGeometryFromFile(Bool_t  im,
  66                                            TString pa = "")     { fImportGeometryFromFile = im ;
  67                                                                   fImportGeometryFilePath = pa ; }
  68   void                  SetTrackFilterBit(ULong_t bit)          { fFilterBit = bit;  }
  69   void                  SetHybridOn()                           { fSelHybrid = kTRUE; }
  70   void                  SetFillQA()                             { fFillQA = kTRUE; }
  71   void                  SelectCPVFromTrack(Bool_t b)            { fCpvFromTrack = b; }
  72   void                  SetEtPtHistoBinning(Int_t n,
  73                                             Double_t lowx,
  74                                             Double_t highx)     { fNBinsPt = n; fPtBinLowEdge = lowx; fPtBinHighEdge = highx; }
  75   void                  SetRemoveMatchClus(Bool_t b)            { fRemMatchClus       = b;       }
  76   void                  SetMinIsoClusE(Double_t emin)           { fMinIsoClusE        = emin;    }
  77   void                  SetTrCoreRemoval(Bool_t b)              { fTrCoreRem          = b;       }
  78   void                  SetClusTDiff(Double_t diff)             { fClusTDiff          = diff;    }
  79   void                  SetPileUpRejSPD()                       { fPileUpRejSPD       = kTRUE;   }
  80   void                  SetDistanceToBadCh(Double_t d)          { fDistToBadChan      = d;       }
  81  protected:
  82   TObjArray             *fESDClusters;           //!pointer to EMCal clusters
  83   TObjArray             *fAODClusters;           //!pointer to EMCal clusters
  84   TObjArray             *fSelPrimTracks;         //!pointer to ESD primary tracks
  85   TClonesArray          *fTracks;                //!track input array
  86   TClonesArray          *fAODMCParticles;        //!MC particles array for AOD analysis
  87   AliESDCaloCells       *fESDCells;              //!pointer to EMCal cells, esd
  88   AliAODCaloCells       *fAODCells;              //!pointer to EMCal cells, aod
  89   AliESDtrackCuts       *fPrTrCuts;              //pointer to hold the prim track cuts
  90   AliESDtrackCuts       *fCompTrCuts;            //pointer to hold complementary track cuts (a la Gustavo)
  91   AliEMCALGeometry      *fGeom;                  // geometry utils
  92   TString                fGeoName;               // geometry name (def = EMCAL_FIRSTYEARV1)
  93   AliOADBContainer      *fOADBContainer;         //!OADB container used to load misalignment matrices
  94   TVector3               fVecPv;                 // vector to hold the event's primary vertex
  95   TString                fPeriod;                // string to the LHC period
  96   TString                fTrigBit;               // string to the trigger bit name
  97   Bool_t                 fIsTrain;               // variable to set train mode
  98   Bool_t                 fIsMc;                  // variable to set mc mode
  99   Bool_t                 fDebug;                 // variable to set on/off debugging printouts
 100   TString                fPathStrOpt;            // variable to set the name of files to be analyzed (MC only)
 101   Double_t               fExoticCut;             // variable to set the cut on exotic clusters
 102   Double_t               fIsoConeR;              // variable to set the isolation cone radius
 103   Int_t                  fNDimensions;           // variable to set the number of dimensions of n-sparse
 104   Double_t               fECut;                  // variable to set the minimum E of a cluster
 105   Int_t                  fTrackMult;             // global variable with the event multiplicity
 106   TString                fMcIdFamily;            // string that holds the ids of all particles originated from the prompt photon
 107   Int_t                  fNClusForDirPho;        // number of clusters from prompt photon per event
 108   Float_t                fDirPhoPt;              // prompt photon pt (assumes only one per event)
 109   Float_t                fHigherPtCone;          // higher pt inside the cone around the candidate
 110   Bool_t                 fImportGeometryFromFile;  // Import geometry settings in geometry.root file
 111   TString                fImportGeometryFilePath;  // path fo geometry.root file
 112   Double_t               fMaxPtTrack;            //track with highest pt in event
 113   Double_t               fMaxEClus;              //cluster with highest energy in event
 114   Int_t                  fNCells50;              // variable to keep the number of cells with E>50 MeV
 115   ULong_t                fFilterBit;             // Track selection bit, for AODs
 116   Bool_t                 fSelHybrid;             // bool to select hybrid tracks
 117   Bool_t                 fFillQA;                // bool to fill the QA plots
 118   TString                fClusIdFromTracks;      // string to hold the list of cluster ids given by tracks
 119   Bool_t                 fCpvFromTrack;          // set the track-matching method to track->GetEMCALcluster()
 120   Int_t                  fNBinsPt;               // set the number of bins in axis of histograms filled with pt (or Et)
 121   Double_t               fPtBinLowEdge;          // low edge of the first pt (Et) bin
 122   Double_t               fPtBinHighEdge;         // high edge of the first pt (Et) bin
 123   Bool_t                 fRemMatchClus;          // flag to remove completely a cluster matched from the isolation
 124   Double_t               fMinIsoClusE;           // minimum energy for a cluster to be counted in the iso cone
 125   Int_t                  fNCuts;                 // number of cuts (QA purposes)
 126   Bool_t                 fTrCoreRem;             // flag to set the removal of the core in track isolation (true removes it, default)
 127   Double_t               fClusTDiff;             // variable to hold the time diff between the candidate cluster and the isolation clusters
 128   Bool_t                 fPileUpRejSPD;          // flag to set pile-up rejection via SPD (multiple vertices)
 129   Double_t               fDistToBadChan;         // distance to bad channel
 130   TString                fInConeInvMass;         // string to hold the array of inv. mass values of the candidate with isolation clusters
 131
 132  private:
 133   AliESDEvent *fESD;      //! ESD object
 134   AliAODEvent *fAOD;      //! AOD object
 135   AliVEvent   *fVEvent;   //! AliVEvent
 136   AliMCEvent  *fMCEvent;  //! MC event object
 137   AliStack    *fStack;    //!MC particles stack object
 138   TGeoHMatrix *fGeomMatrix[12];//! Geometry misalignment matrices for EMCal
 139   TList       *fOutputList; //! Output list
 140   //histograms for events with 1+ track pt>1
 141   TH1F        *fEvtSel;                    //!evt selection counter: 0=all trg, 1=pv cut
 142   TH1F        *fNClusEt10;                 //!number of clusters w/ Et>10 in the event
 143   TH1F        *fRecoPV;                    //!histogram to record if an event has a prim. vert.
 144   TH1F        *fPVtxZ;                     //!primary vertex Z before cut
 145   TH1F        *fTrMultDist;                //!track multiplicity distribution
 146   TH2F        *fClusEtCPVSBGISO;           //!iso-all vs. clusters Et after CPV and 0.1<M02<0.3
 147   TH2F        *fClusEtCPVBGISO;            //!iso-all vs. clusters Et after CPV and 0.5<M02<2.0
 148   TH3F        *fMCDirPhotonPtEtaPhi;       //!direct produced photon pt, eta, phi
 149   TH3F        *fMCIsoDirPhotonPtEtaPhi;    //!direct produced photon pt, eta, phi, isolated @ mc level
 150   TH2F        *fMCDirPhotonPtEtIso;        //!direct produced photon pt and isolation pt @ mc level
 151   TH1F        *fDecayPhotonPtMC;           //!decay photon pt
 152   TH2F        *fCellAbsIdVsAmpl;           //!cell abs id vs cell amplitude (energy)
 153   TH2F        *fNClusHighClusE;            //!total number of clusters vs. highest clus energy in the event
 154   TH2F        *fHigherPtConeM02;           //!M02 vs. the higher pt of a track inside the cone
 155   TH2F        *fClusEtMcPt;                //!cluster et x mc-pt
 156   TH2F        *fClusMcDetaDphi;            //!delta-eta x delta-phi(reco-mc)
 157   TH2F        *fNClusPerPho;               //!delta-eta x delta-phi(reco-mc)
 158   TH2F        *fMcPtInConeBG;              //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in BG template
 159   TH2F        *fMcPtInConeSBG;             //!sum of mc-pt of "primary" particles inside de cone, as a function of NET-ISO in SBG range
 160   TH2F        *fMcPtInConeBGnoUE;          //!sum of mc-pt of "primary" particles inside de cone, as a function of ISO in BG template no UE sub
 161   TH2F        *fMcPtInConeSBGnoUE;         //!sum of mc-pt of "primary" particles inside de cone, as a function of ISO in SBG range no UE sub
 162   TH2F        *fMcPtInConeTrBGnoUE;        //!sum of mc-pt of "primary" particles inside de cone, as a function of trk only ISO in BG template no UE sub
 163   TH2F        *fMcPtInConeTrSBGnoUE;       //!sum of mc-pt of "primary" particles inside de cone, as a function of trk only ISO in SBG range no UE sub
 164   TH2F        *fMcPtInConeMcPhoPt;         //!sum of mc-pt of "primary" particles inside de cone, as a function of prompt photon mc-pt
 165   TH2F        *fAllIsoEtMcGamma;           //!all iso distribution vs. Et clus for clusters comming from a MC prompt photon
 166   TH2F        *fAllIsoNoUeEtMcGamma;       //!all iso distribution (without UE subtraction) vs. Et clus for clusters comming from a MC prompt photon
 167   TH3F        *fMCDirPhotonPtEtaPhiNoClus; //!pt x eta x phi for prompt photons that didn't produce clusters
 168   TH3F        *fInvMassWithConeVsEtAndIso; //!Candidate Et vs. Inv Mass with in cone cluster vs. Iso(EMC+trk), only for 0.1<M02<0.3 clusters
 169   TH2F        *fInConePairedClusEtVsCandEt;//!Et of the paired cluster (0.11<m<0.165 GeV) vs. candidate Et
 170   THnSparse   *fHnOutput;                  //!Output matrix with 7 dimensions
 171
 172   //QA histos
 173   TList       *fQAList;           //!output list holding QA histos
 174   TH1F        *fNTracks;          //!number of tracks from Array->GetEntries()
 175   TH1F        *fEmcNCells;        //!number of emcal cells in the event
 176   TH1F        *fEmcNClus;         //!# of emcal clusters
 177   TH1F        *fEmcNClusCut;      //!# of clusters in an event with at least 1 clus with E > fECut ("triggered event")
 178   TH1F        *fNTracksECut;      //!number of tracks from Array->GetEntries() in "triggered event"
 179   TH1F        *fEmcNCellsCut;     //!number of emcal cells in a in "triggered event"
 180   TH1F        *fEmcClusETM1;      //!emcal track matched cluster energy (TracDx,z method)
 181   TH1F        *fEmcClusETM2;      //!emcal track matched cluster energy (track->GetEMCALcluster() method)
 182   TH1F        *fEmcClusNotExo;    //!cluster energy (exotics removed)
 183   TH2F        *fEmcClusEClusCuts; //!cluster E spectrum per cluster cut (none, exotic, exo+cpv1, exo+cpv1+time, exo+cpv1+time+m02)
 184   TH2F        *fEmcClusEPhi;      //!cluster E spectrum vs. phi
 185   TH2F        *fEmcClusEPhiCut;   //!cluster E spectrum vs. phi in "triggered event"
 186   TH2F        *fEmcClusEEta;      //!cluster E spectrum vs. eta
 187   TH2F        *fEmcClusEEtaCut;   //!cluster E spectrum vs. eta in "triggered event"
 188   TH2F        *fTrackPtPhi;       //!selected tracks pt vs. phi
 189   TH2F        *fTrackPtPhiCut;    //!selected tracks pt vs. phi in "triggered event"
 190   TH2F        *fTrackPtEta;       //!selected tracks pt vs. eta
 191   TH2F        *fTrackPtEtaCut;    //!selected tracks pt vs. eta in "triggered event"
 192   TH2F        *fMaxCellEPhi;      //!max cell energy vs. cell phi
 193   TH2F        *fDetaDphiFromTM;   //!dphi vs deta of track->GetEMCALcluster() clusters
 194   TH2F        *fEoverPvsE;        //!E/p for tracks with 80<TPCsignal<100 vs cluster E (check material)
 195
 196
 197   AliAnalysisTaskEMCALIsoPhoton(const AliAnalysisTaskEMCALIsoPhoton&); // not implemented
 198   AliAnalysisTaskEMCALIsoPhoton& operator=(const AliAnalysisTaskEMCALIsoPhoton&); // not implemented
 199
 200   ClassDef(AliAnalysisTaskEMCALIsoPhoton, 1); // Class to analyse isolated photons
 201 };
 202 #endif