Stefan Rosegger and Jim Thomas
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / 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;
29class TFile ;
67ebd013 30//class AliCFManager;
fdceab34 31class AliESDtrackCuts;
32class AliESDEvent;
33
34class AliPWG4HighPtSpectra : public AliAnalysisTask {
b5cc0c6d 35 public:
fdceab34 36
37 enum {
67ebd013 38 kStepReconstructed = 0,
fdceab34 39 kStepReconstructedTPCOnly = 1,
67ebd013 40 kStepSecondaries = 2,
41 kStepMCtrackable = 3,
e7b49043 42 kStepReconstructedMC = 4,
43 kStepMCAcceptance = 5
fdceab34 44 };
45
46 AliPWG4HighPtSpectra();
47 AliPWG4HighPtSpectra(const Char_t* name);
67ebd013 48 // AliPWG4HighPtSpectra& operator= (const AliPWG4HighPtSpectra& c);
49 // AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra& c);
50 ~AliPWG4HighPtSpectra() {;};
fdceab34 51
52 // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
e5abcde9 53 virtual void LocalInit();
fdceab34 54 virtual void ConnectInputData(Option_t *);
55 virtual void CreateOutputObjects();
56 virtual void Exec(Option_t *option);
57 virtual void Terminate(Option_t *);
58
59 // CORRECTION FRAMEWORK RELATED FUNCTIONS
67ebd013 60 void SetCFManagerPos(const AliCFManager* io1) {fCFManagerPos = io1;} // global correction manager
61 const AliCFManager * GetCFManagerPos() const {return fCFManagerPos;} // get corr manager
62 void SetCFManagerNeg(const AliCFManager* io2) {fCFManagerNeg = io2;} // global correction manager
63 const AliCFManager * GetCFManagerNeg() const {return fCFManagerNeg;} // get corr manager
64
fdceab34 65 //AliESDtrackCuts setters
66 void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
b5cc0c6d 67 //Select the trigger
68 void SelectTrigger(Int_t trig) { fTrigger = trig; }
fdceab34 69
70 // Data types
71 Bool_t IsReadAODData() const {return fReadAODData;}
b5cc0c6d 72 void SetReadAODData(Bool_t flag=kTRUE) {fReadAODData=flag;}
73
fdceab34 74 protected:
67ebd013 75 Bool_t fReadAODData ; // flag for AOD/ESD input files
76 const AliCFManager *fCFManagerPos ; // pointer to the CF manager for positive charged particles
77 const AliCFManager *fCFManagerNeg ; // pointer to the CF manager for negative charged particles
78
b5cc0c6d 79 AliESDEvent *fESD; //! ESD object
80 //AliESDtrackCuts options. Must be setted in AddTaskPWG4HighPtQAMC.C. They correspond with different steps in container.
81 AliESDtrackCuts *fTrackCuts; // trackCuts applied
82 Int_t fTrigger; //Trigger flag as defined in AliAnalysisHelperJetTasks.h
fdceab34 83
84 private:
67ebd013 85 AliPWG4HighPtSpectra(const AliPWG4HighPtSpectra&);
86 AliPWG4HighPtSpectra& operator=(const AliPWG4HighPtSpectra&);
87
fdceab34 88
89 // Histograms
90 //Number of events
67ebd013 91 TList *fHistList; //! List of output histograms
b5cc0c6d 92 TH1F *fNEventAll; //! Event counter
93 TH1F *fNEventSel; //! Event counter: Selected events for analysis
94
fdceab34 95 ClassDef(AliPWG4HighPtSpectra,1);
96};
97
98#endif