PWGJE/AliPWG4HighPtTrackQA.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 //-----------------------------------------------------------------------
  17 // This class stores QA variables as function of pT for different type
  18 // of tracks and track selection criteria
  19 // Author : Marta Verweij - UU
  20 //-----------------------------------------------------------------------
  21
  22 #ifndef ALIPWG4HIGHPTTRACKQA_H
  23 #define ALIPWG4HIGHPTTRACKQA_H
  24
  25 #include "AliAnalysisTaskSE.h"
  26
  27 class TH1F;
  28 class TH2F;
  29 class TH3F;
  30 class TProfile;
  31 class TList;
  32 class TArrayF;
  33
  34 class AliVEvent;
  35 class AliESDEvent;
  36 class AliESDtrackCuts;
  37 class AliESDVertex;
  38 class AliAODTrack;
  39 class AliESDtrack;
  40
  41 class AliGenPythiaEventHeader;
  42 class AliMCEvent;
  43 //class AliAnalysisHelperJetTasks;
  44
  45 class AliPWG4HighPtTrackQA: public AliAnalysisTaskSE {
  46
  47  public:
  48   AliPWG4HighPtTrackQA();
  49   AliPWG4HighPtTrackQA(const char *name);
  50   virtual ~AliPWG4HighPtTrackQA() {;}
  51
  52   //  virtual void   ConnectInputData(Option_t *);
  53   virtual void   UserCreateOutputObjects();
  54   virtual void   UserExec(Option_t *option);
  55   virtual void   Terminate(Option_t *);
  56   virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
  57
  58   enum DataType {kESD,kAOD};
  59
  60   Bool_t IsPbPb() {return fIsPbPb;}  //is PbPb data?
  61   Bool_t SelectEvent();              //decides if event is used for analysis
  62   Int_t CalculateCentrality(AliVEvent *ev);
  63   Int_t CalculateCentrality(AliESDEvent *esd);
  64   Int_t CalculateCentrality(const AliAODEvent *aod);
  65   Int_t GetCentralityClass(Float_t cent=-1.) const;
  66   void DoAnalysisESD();
  67   void DoAnalysisAOD();
  68   void FillHistograms();
  69
  70   //Setters
  71   void SetDataType(DataType d)             {fDataType = d;}
  72   void SetIsPbPb(Bool_t cs)                {fIsPbPb = cs;}
  73   void SetCentralityClass(int cent)        {fCentClass=cent;}
  74   void SetCuts(AliESDtrackCuts* trackCuts)         {fTrackCuts         = trackCuts;}
  75   void SetCutsITSLoose(AliESDtrackCuts* trackCuts) {fTrackCutsITSLoose = trackCuts;}
  76   void SetCutsTPConly(AliESDtrackCuts* trackCuts)  {fTrackCutsTPConly  = trackCuts;}
  77   void SetTrackType(Int_t trackType) {fTrackType = trackType;}
  78   void SetFilterMask(UInt_t filterMask)    {fFilterMask    = filterMask ;}
  79   void SetIncludeNoITS(Bool_t f)           {fIncludeNoITS  = f          ; }
  80
  81   void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
  82   void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
  83   void SetPtBinEdges(Int_t region, Double_t ptmax, Double_t ptBinWidth);
  84   void SetNVariables(Int_t nv) {fNVariables = nv;}
  85
  86   Float_t GetPtMax()           {return fPtMax;}
  87   Float_t GetTPCClusterInfo(const AliAODTrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159, Bool_t useFitMap=kFALSE) const;
  88   Float_t GetTPCClusterInfoFitMap(const AliESDtrack *tr,Int_t nNeighbours=3, Int_t type=0, Int_t row0=0, Int_t row1=159) const;
  89   Int_t   GetTrackLengthTPC(const AliESDtrack *track) const;
  90   Int_t   GetTrackLengthTPC(const AliAODTrack *track) const;
  91   Float_t GetGoldenChi2(AliESDtrack *origtrack);
  92   Float_t GetGGCChi2(AliESDtrack *origtrack);
  93
  94   static AliGenPythiaEventHeader*  GetPythiaEventHeader(const AliMCEvent *mcEvent);
  95   static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root
  96
  97  protected:
  98
  99  private:
 100   AliPWG4HighPtTrackQA(const AliPWG4HighPtTrackQA&);
 101   AliPWG4HighPtTrackQA& operator=(const AliPWG4HighPtTrackQA&);
 102
 103   DataType fDataType;             // kESD or kAOD
 104
 105   AliVEvent   *fEvent;            //! AliVEvent object
 106   AliESDEvent *fESD;              //! ESD object
 107   const AliESDVertex   *fVtx;     //! vertex object
 108
 109   AliESDtrackCuts *fTrackCuts;         // TrackCuts
 110   AliESDtrackCuts *fTrackCutsITSLoose; // Loose ITS track cuts
 111   AliESDtrackCuts *fTrackCutsTPConly;  // TPC track cuts
 112   Int_t   fTrackType;                  // 0: global track; 1:TPConly track 2: TPConly constrained track 3: global ITSrefit 4: TPConly constrained track with QA selection based on global track
 113   UInt_t fFilterMask;                  // Select tracks from specific track cuts belonging to certain filter mask for AOD analysis
 114   Bool_t fIncludeNoITS;                // includes tracks with failed ITS refit
 115
 116   Double_t fSigmaConstrainedMax;  // max sigma on constrained fit
 117   Float_t fPtMax;                 // Maximum pT for histograms
 118   Float_t fPtBinEdges[3][2];      // 3 regions total with different binning for pT axis of histos
 119
 120   Bool_t   fIsPbPb;               // kTRUE if PbPb
 121   Int_t fCentClass;               // Select only events from predefined centrality class
 122
 123
 124   /*
 125     QA variables stored in TArrayF *fVariables
 126   0: pt
 127   1: phi
 128   2: eta
 129   3: dca2D
 130   4: dcaZ
 131   5: nClustersTPC
 132   6: nPointITS
 133   7: chi2C
 134   8: nSigmaToVertex
 135   9: relUncertainty1Pt
 136   10: chi2PerClusterTPC
 137   11: #crossed rows
 138   12: (#crossed rows)/(#findable clusters)
 139   13: SigmaY2
 140   14: SigmaZ2
 141   15: SigmaSnp2
 142   16: SigmaTgl2
 143   17: Sigma1Pt2
 144   18: NClustersTPCIter1
 145   19: TPCChi2Iter1
 146   20: NClustersTPCShared
 147   21: Chi2Gold (TPC constrained vs global)
 148   22: Chi2GGC (global constrained vs global)
 149   23: NCrossed rows from fit map
 150   24: (#crossed rows)/(#findable clusters) from fit map
 151   */
 152
 153   Int_t fNVariables;             // Number of variables
 154   TArrayF *fVariables;           // QA variables
 155
 156   UChar_t fITSClusterMap;  // map of clusters, one bit per a layer
 157
 158   Float_t fAvgTrials;             // Average number of trials
 159
 160   TH1F *fNEventAll;                            //! Event counter
 161   TH1F *fNEventSel;                            //! Event counter
 162   TH1F *fNEventReject;                         //! Book keeping of reason of rejecting events
 163
 164   TH1F *fh1Centrality;                         //! Centrality
 165
 166   TProfile*     fh1Xsec;                       //! pythia cross section and trials
 167   TH1F*         fh1Trials;                     //! trials which are added
 168   TH1F*         fh1PtHard;                     //! pt hard of the event
 169   TH1F*         fh1PtHardTrials;               //! pt hard of the event
 170
 171   TH1F *fh1NTracksAll;                         //! All tracks
 172   TH1F *fh1NTracksReject;                      //! Reason why track was rejected
 173   TH1F *fh1NTracksSel;                         //! Number of accepted tracks
 174
 175   TH1F *fPtAll;                                //! Pt spectrum all charged particles
 176   TH1F *fPtSel;                                //! Pt spectrum all selected charged particles by fTrackCuts
 177   TH2F *fPtPhi;                                //! Pt vs Phi
 178   TH2F *fPtEta;                                //! Pt vs Eta
 179   TH3F *fPtEtaPhi;                             //! Pt vs Eta vs Phi
 180   TH2F *fPtDCA2D;                              //! Pt vs DCA2D
 181   TH2F *fPtDCAZ;                               //! Pt vs DCAZ
 182   TH2F *fPtNClustersTPC;                       //! Pt vs nClustersTPC
 183   TH2F *fPtNClustersTPCIter1;                  //! Pt vs nClustersTPCIter1
 184   TH3F *fPtNClustersTPCIter1Phi;               //! Pt vs nClustersTPCIter1 vs Phi
 185   TH2F *fPtNClustersTPCShared;                 //! Pt vs nClustersTPCShared
 186   TH2F *fPtNClustersTPCSharedFrac;             //! Pt vs nClustersTPCSharedFrac
 187   TH2F *fPtNPointITS;                          //! Pt vs nPointITS
 188   TH3F *fPtNPointITSPhi;                       //! Pt vs nPointITS vs phi
 189   TH2F *fPtChi2C;                              //! Pt vs Chi2C
 190   TH2F *fPtNSigmaToVertex;                     //! Pt vs nSigmaToVertex
 191
 192   TH2F *fPtRelUncertainty1Pt;                  //! Pt vs relUncertainty1Pt
 193   TH3F *fPtRelUncertainty1PtNClus;             //! Pt vs relUncertainty1Pt vs NClustersTPC
 194   TH3F *fPtRelUncertainty1PtNClusIter1;        //! Pt vs relUncertainty1Pt vs NClustersTPCIter1
 195   TH3F *fPtRelUncertainty1PtNPointITS;         //! Pt vs relUncertainty1Pt vs NPointITS
 196   TH3F *fPtRelUncertainty1PtITSClusterMap;     //! Pt vs relUncertainty1Pt vs byte of ITS clustermap
 197
 198
 199   TH3F *fPtRelUncertainty1PtChi2;              //! Pt vs relUncertainty1Pt vs Chi2TPC/NClus
 200   TH3F *fPtRelUncertainty1PtChi2Iter1;         //! Pt vs relUncertainty1Pt vs Chi2TPC/NClusIter1
 201   TH3F *fPtRelUncertainty1PtPhi;               //! Pt vs relUncertainty1PtPhi
 202   TH2F *fPtUncertainty1Pt;                     //! Pt vs Uncertainty1Pt
 203   TH2F *fPtChi2PerClusterTPC;                  //! Pt vs Chi2PerClusterTPC
 204   TH2F *fPtChi2PerClusterTPCIter1;             //! Pt vs Chi2PerClusterTPCIter1
 205   TH2F *fPtNCrossedRows;                       //! Pt vs NCrossedRows
 206   TH3F *fPtNCrossedRowsPhi;                    //! Pt vs NCrossedRows vs Phi
 207   TH3F *fPtNCrossedRowsNClusFPhi;              //! Pt vs NCrossedRows/NClusF vs Phi
 208   TH3F *fPtNCrRNCrRNClusF;                     //! Pt vs NCrossedRows vs NCrossedRows/NClusF
 209   TH2F *fPtNCrossedRowsFit;                    //! Pt vs NCrossedRows from NClusterFitMap
 210   TH3F *fPtNCrossedRowsFitPhi;                 //! Pt vs NCrossedRows from FitMap vs Phi
 211   TH3F *fPtNCrossedRowsNClusFFitPhi;              //! Pt vs NCrossedRows/NClusF from NClusterFitMap
 212   TH2F *fNCrossedRowsNCrossedRowsFit;          //! NCrossedRows from ClusterMap vs NCrossedRows from NClusterFitMap
 213   TH2F *fNClustersNCrossedRows;                //! NClusters vs NCrossedRows
 214   TH2F *fNClustersNCrossedRowsFit;             //! NClusters vs NCrossedRows from fit map
 215   TH3F *fPtRelUncertainty1PtNCrossedRows;       //! Pt vs relUncertainty1Pt vs NCrossedRows
 216   TH3F *fPtRelUncertainty1PtNCrossedRowsFit;    //! Pt vs relUncertainty1Pt vs NCrossedRowsFitMap
 217
 218   TH2F *fPtChi2Gold;                           //! Pt vs Chi2 between global and TPC constrained track
 219   TH2F *fPtChi2GGC;                            //! Pt vs Chi2 between global and global constrained track
 220   TH3F *fPtChi2GoldPhi;                        //! Pt vs Chi2 between global and TPC constrained track vs phi
 221   TH3F *fPtChi2GGCPhi;                         //! Pt vs Chi2 between global and global constrained track vs phi
 222   TH2F *fChi2GoldChi2GGC;                      //! Correlations between gold chi2 and GGC chi2
 223
 224   //histos for covariance matrix elements
 225   TH2F *fPtSigmaY2;                            //! 1/Pt vs sigma(y) extCov[0]
 226   TH2F *fPtSigmaZ2;                            //! 1/Pt vs sigma(z) extCov[2]
 227   TH2F *fPtSigmaSnp2;                          //! 1/Pt vs sigma(Snp) extCov[5]
 228   TH2F *fPtSigmaTgl2;                          //! 1/Pt vs sigma(Tgl) extCov[9]
 229   TH2F *fPtSigma1Pt2;                          //! 1/Pt vs sigma(1/pT) extCov[14]
 230
 231   //profiles for covariance matrix elements
 232   TProfile *fProfPtSigmaY2;                    //! 1/Pt vs sigma(y) extCov[0]
 233   TProfile *fProfPtSigmaZ2;                    //! 1/Pt vs sigma(z) extCov[2]
 234   TProfile *fProfPtSigmaSnp2;                  //! 1/Pt vs sigma(Snp) extCov[5]
 235   TProfile *fProfPtSigmaTgl2;                  //! 1/Pt vs sigma(Tgl) extCov[9]
 236   TProfile *fProfPtSigma1Pt2;                  //! 1/Pt vs sigma(1/pT) extCov[14]
 237
 238   TProfile *fProfPtSigma1Pt;                   //! pT vs sigma(1/Pt)
 239   TProfile *fProfPtPtSigma1Pt;                 //! pT vs pT*sigma(1/Pt)
 240
 241   TList *fHistList; //! List of Histograms
 242
 243   ClassDef(AliPWG4HighPtTrackQA,4)
 244 };
 245 #endif