-Move cent, z axis into sparse
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliGammaConversionAODBGHandler.h
1 //-*- Mode: C++ -*-
2 #ifndef ALIGAMMACONVERSIONAODBGHANDLER_H
3 #define ALIGAMMACONVERSIONAODBGHANDLER_H
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice     */
6
7 ////////////////////////////////////////////////
8 //--------------------------------------------- 
9 // Class  for handling of background calculation
10 //---------------------------------------------
11 ////////////////////////////////////////////////
12
13 #include <vector>
14
15
16 // --- ROOT system ---
17 #include <TObject.h> 
18 #include "AliKFParticle.h"
19 #include "AliAODConversionPhoton.h"
20 #include "AliAODConversionMother.h"
21 #include "TClonesArray.h"
22 #include "AliESDVertex.h"
23
24 #if __GNUC__ >= 3
25 using namespace std;
26 #endif
27
28 typedef vector<AliAODConversionPhoton*> AliGammaConversionAODVector;
29
30 class AliGammaConversionAODBGHandler : public TObject {
31
32 public: 
33    struct GammaConversionVertex
34    {
35       Double_t fX;
36       Double_t fY;
37       Double_t fZ;
38    };
39    typedef struct GammaConversionVertex GammaConversionVertex; //!
40
41    typedef vector<AliGammaConversionAODVector> AliGammaConversionBGEventVector;
42    typedef vector<AliGammaConversionBGEventVector> AliGammaConversionMultipicityVector;
43    typedef vector<AliGammaConversionMultipicityVector> AliGammaConversionBGVector;
44
45    AliGammaConversionAODBGHandler();                                                        //constructor
46    AliGammaConversionAODBGHandler(UInt_t binsZ,UInt_t binsMultiplicity,UInt_t nEvents); // constructor
47    AliGammaConversionAODBGHandler(UInt_t collisionSystem,UInt_t centMin,UInt_t centMax,UInt_t nEvents, Bool_t useTrackMult);
48    AliGammaConversionAODBGHandler(const AliGammaConversionAODBGHandler & g);                   //copy constructor
49    AliGammaConversionAODBGHandler & operator = (const AliGammaConversionAODBGHandler & g);     //assignment operator
50    virtual ~AliGammaConversionAODBGHandler();                                               //virtual destructor
51
52    void Initialize(Double_t * const zBinLimitsArray, Double_t * const multiplicityBinLimitsArray);
53
54    Int_t GetZBinIndex(Double_t z) const;
55
56    Int_t GetMultiplicityBinIndex(Int_t mult) const;
57
58    void AddEvent(TList* const eventGammas, Double_t xvalue,Double_t yvalue,Double_t zvalue, Int_t multiplicity);
59    void AddElectronEvent(TClonesArray* const eventENeg, Double_t zvalue, Int_t multiplicity);
60
61    Int_t GetNBGEvents()const {return fNEvents;}
62
63    AliGammaConversionAODVector* GetBGGoodV0s(Int_t zbin, Int_t mbin, Int_t event);
64    AliGammaConversionAODVector* GetBGGoodENeg(Int_t event, Double_t zvalue, Int_t multiplicity);
65    void PrintBGArray();
66
67    GammaConversionVertex * GetBGEventVertex(Int_t zbin, Int_t mbin, Int_t event){return &fBGEventVertex[zbin][mbin][event];}
68
69    Double_t GetBGProb(Int_t z, Int_t m){return fBGProbability[z][m];}
70
71 private:
72
73    Int_t fNEvents; // number of events
74    Int_t ** fBGEventCounter; //! bg counter
75    Int_t ** fBGEventENegCounter;//! bg electron counter
76    Double_t ** fBGProbability; //! prob per bin
77    GammaConversionVertex *** fBGEventVertex;//! array of event vertex
78    Int_t fNBinsZ; //n z bins
79    Int_t fNBinsMultiplicity; //n bins multiplicity
80    Double_t *fBinLimitsArrayZ;//! bin limits z array
81    Double_t *fBinLimitsArrayMultiplicity;//! bin limit multiplicity array
82    AliGammaConversionBGVector fBGEvents; //background events
83    AliGammaConversionBGVector fBGEventsENeg; //background electron events
84    ClassDef(AliGammaConversionAODBGHandler,2)
85 };
86 #endif