updated for e-h analysis
[u/mrichter/AliRoot.git] / CORRFW / AliCFManager.h
CommitLineData
563113d0 1#ifndef ALICFMANAGER_H
2#define ALICFMANAGER_H
3/**************************************************************************
4 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * *
6 * Author: The ALICE Off-line Project. *
7 * Contributors are mentioned in the code where appropriate. *
8 * *
9 * Permission to use, copy, modify and distribute this software and its *
10 * documentation strictly for non-commercial purposes is hereby granted *
11 * without fee, provided that the above copyright notice appears in all *
12 * copies and that both the copyright notice and this permission notice *
13 * appear in the supporting documentation. The authors make no claims *
14 * about the suitability of this software for any purpose. It is *
15 * provided "as is" without express or implied warranty. *
16 **************************************************************************/
17// Prototype class helping the user to handle event & particle-level
18// selections/correction containers inside the Analysis job
19// Author:S.Arcelli. Silvia.Arcelli@cern.ch
20
84df86a7 21//
22// updated by renaud.vernet@cern.ch (2008/10/08) :
23// removed predefined maximum number of steps
24// now the number of steps are fixed by the particle/event containers themselves.
25//
26
563113d0 27#include "TNamed.h"
00148604 28#include "AliCFContainer.h"
29#include "AliLog.h"
9f6be3a2 30
563113d0 31//____________________________________________________________________________
32class AliCFManager : public TNamed
33{
34 public :
35 AliCFManager() ;
36 AliCFManager(Char_t* name, Char_t* title) ;
37 AliCFManager(const AliCFManager& c) ;
38 AliCFManager& operator=(const AliCFManager& c) ;
39 virtual ~AliCFManager();
40
41 //
42 //Currently foreseen selection steps for event-level cuts:
43 //generator, trigger, reconstruction
44 //
45 enum{
46 kEvtGenCuts=0,
84df86a7 47 kEvtTrigCuts,
48 kEvtRecCuts
49 };
563113d0 50
51 //
52 //Currently foreseen selection steps for particle-level cuts:
53 //generator, acceptance, reconstruction, user selection
54 //
563113d0 55 enum{
84df86a7 56 kPartGenCuts=0,
57 kPartAccCuts,
58 kPartRecCuts,
59 kPartSelCuts
60 };
563113d0 61
62 //
63 // Setters:
64 //
563113d0 65
66 //pass the pointer to the correction container
00148604 67 virtual void SetEventContainer(AliCFContainer* c) {
68 fEvtContainer=c;
4b49fad2 69 SetNStepEvent(c->GetNStep());
00148604 70 AliWarning(Form("Please dont forget to set the cut list (event empty) for the %d event-selection step requested",fNStepEvt));
71 }
72
73 //pass the pointer to the correction container
74 virtual void SetParticleContainer(AliCFContainer* c) {
75 fPartContainer=c;
4b49fad2 76 SetNStepParticle(c->GetNStep());
00148604 77 AliWarning(Form("Please dont forget to set the cut list (even empty) for the %d particle-selection step requested",fNStepPart));
78 }
79
4b49fad2 80 //Set the number of steps (already done if you have defined your containers)
81 virtual void SetNStepEvent (Int_t nstep) {fNStepEvt = nstep;}
82 virtual void SetNStepParticle(Int_t nstep) {fNStepPart = nstep;}
83
563113d0 84 //Setter for event-level selection cut list at selection step isel
84df86a7 85 virtual void SetEventCutsList(Int_t isel, TObjArray* array) ;
86
563113d0 87 //Setter for particle-level selection cut list at selection step isel
84df86a7 88 virtual void SetParticleCutsList(Int_t isel, TObjArray* array) ;
563113d0 89
90 //
91 //Getters
92 //
93 // pointer to the Event-level correction container
94 virtual AliCFContainer* GetEventContainer() const {return fEvtContainer;} ;
95
96 // pointer to the Particle-level correction container
97 virtual AliCFContainer* GetParticleContainer() const {return fPartContainer;} ;
98
99 //pointer to the event-level cut list for event selection step isel
100 virtual TObjArray* GetEventCutsList(Int_t isel) const {return fEvtCutList[isel];};
101
102//pointer to the particle-level cut list for particle selection step isel
103 virtual TObjArray* GetParticleCutsList(Int_t isel) const {return fPartCutList[isel];};
104
105
106 //Pass the pointer to obj to the selections (used to access MC/rec global
107 //event info when requested
0c01ae65 108 virtual void SetMCEventInfo(const TObject *obj) const;
109 virtual void SetRecEventInfo(const TObject *obj) const;
16a98c19 110 virtual void SetEventInfo(TObject*) const {AliError("DEPRECATED !! -> use SetMCEventInfo or SetRecEventInfo instead");}
563113d0 111
112 //Cut Checkers: by default *all* the cuts of a given input list is checked
113 //(.and. of all cuts), but the user can select a subsample of cuts in the
114 //list via the string argument selcuts
115
116 virtual Bool_t CheckEventCuts(Int_t isel, TObject *obj, const TString &selcuts="all") const;
117 virtual Bool_t CheckParticleCuts(Int_t isel, TObject *obj, const TString &selcuts="all") const;
118
563113d0 119 private:
00148604 120
121 //number of steps
d98cee7b 122 Int_t fNStepEvt; // number of steps in event selection
123 Int_t fNStepPart; // number of steps in particle selection
563113d0 124 //the correction grid
125 AliCFContainer *fEvtContainer; //ptr to Evt-Level correction container
126 //the correction grid
127 AliCFContainer *fPartContainer; //ptr to Particle-level correction container
128 //Evt-Level Selections
00148604 129 TObjArray **fEvtCutList; //[fNStepEvt] arrays of cuts for each event-selection level
563113d0 130 //Particle-level selections
00148604 131 TObjArray **fPartCutList ; //[fNStepPart] arrays of cuts for each particle-selection level
563113d0 132
133 Bool_t CompareStrings(const TString &cutname,const TString &selcuts) const;
134
84df86a7 135 ClassDef(AliCFManager,2);
563113d0 136};
137
138
139#endif