Fix to ignore ZN background check in MC
[u/mrichter/AliRoot.git] / PWGJE / AliPWG4HighPtSpectra.h
CommitLineData
fdceab34 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// Author : Marta Verweij - UU
18//-----------------------------------------------------------------------
19
20#ifndef ALIPWG4HIGHPTSPECTRA_H
21#define ALIPWG4HIGHPTSPECTRA_H
22
23#include "AliAnalysisTask.h"
24#include "AliCFManager.h"
25
26class TH1I;
b5cc0c6d 27class TH1F;
fdceab34 28class TH1D;
a337a5a9 29class TH2F;
b4691ee7 30class TProfile;
31class TFile;
32class TList;
33
67ebd013 34//class AliCFManager;
fdceab34 35class AliESDtrackCuts;
36class AliESDEvent;
34fc6450 37class AliESDVertex;
b4691ee7 38class AliMCEvent;
cce400da 39class AliStack;
b4691ee7 40class AliGenPythiaEventHeader;
fdceab34 41
42class AliPWG4HighPtSpectra : public AliAnalysisTask {
b5cc0c6d 43 public:
fdceab34 44
45 enum {
ec555a3c 46 kStepReconstructed = 0,
2b553e6f 47 kStepSecondaries = 1,
48 kStepReconstructedMC = 2,
49 kStepMCAcceptance = 3
fdceab34 50 };
51
52 AliPWG4HighPtSpectra();
53 AliPWG4HighPtSpectra(const Char_t* name);
67ebd013 54 // AliPWG4HighPtSpectra& operator= (const AliPWG4HighPtSpectra& c);
55 // AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra& c);
56 ~AliPWG4HighPtSpectra() {;};
fdceab34 57
58 // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
e5abcde9 59 virtual void LocalInit();
fdceab34 60 virtual void ConnectInputData(Option_t *);
61 virtual void CreateOutputObjects();
62 virtual void Exec(Option_t *option);
63 virtual void Terminate(Option_t *);
b4691ee7 64 virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
fdceab34 65
2b553e6f 66 Bool_t IsPbPb() {return fIsPbPb;} //is PbPb data?
cce400da 67 Bool_t SelectEvent(); //decides if event is used for analysis
2b553e6f 68 Int_t CalculateCentrality(AliESDEvent *esd);
cce400da 69
2b553e6f 70 //Setters
71 void SetIsPbPb(Bool_t cs) {fIsPbPb = cs;}
72 void SetCentralityClass(int cent) {fCentClass=cent;}
5d87a047 73 void SetTriggerMask(UInt_t t) {fTriggerMask=t;}
2b553e6f 74
fdceab34 75 // CORRECTION FRAMEWORK RELATED FUNCTIONS
67ebd013 76 void SetCFManagerPos(const AliCFManager* io1) {fCFManagerPos = io1;} // global correction manager
77 const AliCFManager * GetCFManagerPos() const {return fCFManagerPos;} // get corr manager
78 void SetCFManagerNeg(const AliCFManager* io2) {fCFManagerNeg = io2;} // global correction manager
79 const AliCFManager * GetCFManagerNeg() const {return fCFManagerNeg;} // get corr manager
2b553e6f 80
81 //if fTrackType=0 (Global)
82 //if fTrackType=1 (TPConly)
83 //if fTrackType=2 (TPConly constrained)
34fc6450 84 void SetTrackType(Int_t trackType) {fTrackType = trackType;}
fdceab34 85 //AliESDtrackCuts setters
86 void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
327d12da 87 void SetCutsReject(AliESDtrackCuts* trackCuts) {fTrackCutsReject = trackCuts;}
2b553e6f 88
89 void SetSigmaConstrainedMax(Double_t sigma) {fSigmaConstrainedMax=sigma;}
fdceab34 90
91 // Data types
92 Bool_t IsReadAODData() const {return fReadAODData;}
b5cc0c6d 93 void SetReadAODData(Bool_t flag=kTRUE) {fReadAODData=flag;}
b4691ee7 94
95 static AliGenPythiaEventHeader* GetPythiaEventHeader(AliMCEvent *mcEvent);
96 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
b5cc0c6d 97
fdceab34 98 protected:
67ebd013 99 Bool_t fReadAODData ; // flag for AOD/ESD input files
100 const AliCFManager *fCFManagerPos ; // pointer to the CF manager for positive charged particles
101 const AliCFManager *fCFManagerNeg ; // pointer to the CF manager for negative charged particles
102
cce400da 103 AliESDEvent *fESD; //! ESD object
104 AliMCEvent *fMC; //! MC event object
105 AliStack *fStack; //! stack object
106
34fc6450 107 const AliESDVertex *fVtx; //! vertex object
108
5d87a047 109 UInt_t fTriggerMask; // Trigger mask to select events
110 Bool_t fIsPbPb; // kTRUE if PbPb
111 Int_t fCentClass; // Select only events from predefined centrality class
34fc6450 112
113 Int_t fTrackType; // Type of track to be used in analysis
1f329128 114 //AliESDtrackCuts options. Must be setted in AddTaskPWG4HighPTSpectra.C. They correspond with different steps in container.
115 AliESDtrackCuts *fTrackCuts; // trackCuts applied to global tracks
327d12da 116 AliESDtrackCuts *fTrackCutsReject; // trackCuts to reject tracks (hybrid case)
2b553e6f 117
118 Double_t fSigmaConstrainedMax; // max sigma on constrained fit
fdceab34 119
120 private:
1f329128 121 AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra&);
122 AliPWG4HighPtSpectra& operator=(const AliPWG4HighPtSpectra&);
fdceab34 123
b4691ee7 124 Float_t fAvgTrials; // Average number of trials
125
fdceab34 126 // Histograms
127 //Number of events
b4691ee7 128 TList *fHistList; //! List of output histograms
1f329128 129 TH1F *fNEventAll; //! Event counter
130 TH1F *fNEventSel; //! Event counter: Selected events for analysis
2b553e6f 131 TH1F *fNEventReject; //! Book keeping of reason of rejecting events
132
133 TH1F *fh1Centrality; //! Centrality
b5cc0c6d 134
b4691ee7 135 TProfile* fh1Xsec; //! pythia cross section and trials
136 TH1F* fh1Trials; //! trials which are added
137 TH1F* fh1PtHard; //! pt hard of the event
138 TH1F* fh1PtHardTrials; //! pt hard of the event
139
a337a5a9 140 TH2F *fPtRelUncertainty1PtPrim; //! Pt vs relUncertainty1Pt for primary particles
141 TH2F *fPtRelUncertainty1PtSec; //! Pt vs relUncertainty1Pt for secondary particles
142
2b553e6f 143 ClassDef(AliPWG4HighPtSpectra,3);
fdceab34 144};
145
146#endif