Ref storage set in initialisation
[u/mrichter/AliRoot.git] / PWG2 / AliProtonCorrectionTask.h
CommitLineData
b5fc8a3e 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 : R. Vernet, Consorzio Cometa - Catania (it)
18//-----------------------------------------------------------------------
19
20#ifndef ALIPROTONCORRECTIONTASK_H
21#define ALIPROTONCORRECTIONTASK_H
22
23#include "AliAnalysisTaskSE.h"
24
25class TH1I;
26class TParticle ;
27class TFile ;
28class AliStack ;
29class AliCFManager;
30class AliESDtrack;
31class AliVParticle;
32
33class AliProtonCorrectionTask : public AliAnalysisTaskSE {
34 public:
35
36 enum {
37 kStepGenerated = 0,
38 kStepReconstructible = 1,
39 kStepReconstructed = 2,
40 kStepSelected = 3
41 };
42
43 AliProtonCorrectionTask();
44 AliProtonCorrectionTask(const Char_t* name);
45 AliProtonCorrectionTask& operator= (const AliProtonCorrectionTask& c);
46 AliProtonCorrectionTask(const AliProtonCorrectionTask& c);
47 virtual ~AliProtonCorrectionTask();
48
49 // ANALYSIS FRAMEWORK STUFF to loop on data and fill output objects
50 void UserCreateOutputObjects();
51 void UserExec(Option_t *option);
52 void Terminate(Option_t *);
53
54 // CORRECTION FRAMEWORK RELATED FUNCTIONS
251e4034 55 void SetCFManagerProtons(AliCFManager* io) {fCFManagerProtons = io;}
56 // global correction manager
57 AliCFManager * GetCFManagerProtons() const {return fCFManagerProtons;}
58 // get corr manager
59 void SetCFManagerAntiProtons(AliCFManager* io) {fCFManagerAntiProtons = io;} // global correction manager
60 AliCFManager * GetCFManagerAntiProtons() const {return fCFManagerAntiProtons;}// get corr manager
b5fc8a3e 61 void SetQAList(TList* list) {fQAHistList = list;}
62
63 // Data types
64 Bool_t IsReadTPCTracks() const {return fReadTPCTracks;}
65 Bool_t IsReadAODData() const {return fReadAODData;}
66 void SetReadTPCTracks (Bool_t flag=kTRUE) {fReadTPCTracks=flag;}
67 void SetReadAODData (Bool_t flag=kTRUE) {fReadAODData=flag;}
68
69 protected:
70 Double_t Rapidity(Double_t px, Double_t py, Double_t pz);
71
3079041e 72 Bool_t fReadTPCTracks; // flag to loop on TPC tracks only
73 Bool_t fReadAODData; // flag for AOD/ESD input files
74 AliCFManager *fCFManagerProtons; // pointer to the CF manager
75 AliCFManager *fCFManagerAntiProtons; // pointer to the CF manager
76 TList *fQAHistList; // list of QA histograms
b5fc8a3e 77
78 // Histograms
79 //Number of events
80 TH1I *fHistEventsProcessed; // simple histo for monitoring the number of events processed
81
82 ClassDef(AliProtonCorrectionTask,1);
83};
84
85#endif