]>
Commit | Line | Data |
---|---|---|
1 | #ifndef ALIANALYSISMUMUBINNING_H | |
2 | #define ALIANALYSISMUMUBINNING_H | |
3 | ||
4 | #include "TNamed.h" | |
5 | #include "TString.h" | |
6 | #include "TMath.h" | |
7 | ||
8 | // | |
9 | // AliAnalysisMuMuBinning : | |
10 | // helper class to store the various bins used | |
11 | // | |
12 | // author: L. Aphecetche (Subatech) | |
13 | // | |
14 | ||
15 | class TMap; | |
16 | class TObjArray; | |
17 | ||
18 | ||
19 | class AliAnalysisMuMuBinning : public TNamed | |
20 | { | |
21 | public: | |
22 | ||
23 | class Range; | |
24 | ||
25 | AliAnalysisMuMuBinning(const char* name="", const char* title=""); | |
26 | AliAnalysisMuMuBinning(const AliAnalysisMuMuBinning& rhs); | |
27 | AliAnalysisMuMuBinning& operator=(const AliAnalysisMuMuBinning& rhs); | |
28 | ||
29 | virtual ~AliAnalysisMuMuBinning(); | |
30 | ||
31 | void AddBin(const AliAnalysisMuMuBinning::Range& bin); | |
32 | ||
33 | void AddBin(const char* what, const char* quantity, | |
34 | Double_t xmin=TMath::Limits<Double_t>::Max(), | |
35 | Double_t xmax=TMath::Limits<Double_t>::Max(), | |
36 | const char* flavour="") | |
37 | { AddBin(what,quantity,xmin,xmax,TMath::Limits<Double_t>::Max(),TMath::Limits<Double_t>::Max(),flavour); } | |
38 | ||
39 | void AddBin(const char* what, const char* quantity, | |
40 | Double_t xmin, | |
41 | Double_t xmax, | |
42 | Double_t ymin, | |
43 | Double_t ymax, | |
44 | const char* flavour=""); | |
45 | ||
46 | TObjArray* CreateWhatArray() const; | |
47 | ||
48 | TObjArray* CreateQuantityArray() const; | |
49 | ||
50 | Double_t* CreateBinArray() const; | |
51 | Double_t* CreateBinArrayY() const; | |
52 | Double_t* CreateBinArrayX() const; | |
53 | ||
54 | TObjArray* CreateBinObjArray() const; | |
55 | TObjArray* CreateBinObjArray(const char* what) const; | |
56 | TObjArray* CreateBinObjArray(const char* what, const char* quantity, const char* flavour) const; | |
57 | ||
58 | Int_t GetNBinsX() const; | |
59 | Int_t GetNBinsY() const; | |
60 | ||
61 | AliAnalysisMuMuBinning* Project(const char* what, const char* quantity, const char* flavour="") const; | |
62 | ||
63 | virtual void Print(Option_t* opt="") const; | |
64 | ||
65 | void CreateMesh(const char* what, const char* quantity1, const char* quantity2, const char* flavour="", Bool_t remove12=kFALSE); | |
66 | ||
67 | Long64_t Merge(TCollection* list); | |
68 | ||
69 | Bool_t IsEqual(const TObject* obj) const; | |
70 | ||
71 | class Range : public TObject { | |
72 | ||
73 | public: | |
74 | ||
75 | Range(const char* what="",const char* quantity="", | |
76 | Double_t xmin=TMath::Limits<Double_t>::Max(), | |
77 | Double_t xmax=TMath::Limits<Double_t>::Max(), | |
78 | Double_t ymin=TMath::Limits<Double_t>::Max(), | |
79 | Double_t ymax=TMath::Limits<Double_t>::Max(), | |
80 | const char* version=""); | |
81 | ||
82 | virtual Int_t Compare(const TObject* obj) const; | |
83 | Bool_t IsEqual(const TObject* obj) const { return Compare(obj)==0; } | |
84 | Bool_t IsSortable() const { return kTRUE; } | |
85 | ||
86 | virtual TObject* Clone(const char* /*newname*/ = "") const { return new Range(*this); } | |
87 | ||
88 | bool operator==(const Range& other) const { return Compare(&other)==0; } | |
89 | ||
90 | bool operator!=(const Range& other) const { return !(*this==other); } | |
91 | ||
92 | Bool_t IsIntegrated() const; | |
93 | ||
94 | TString Quantity() const { return fQuantity; } | |
95 | TString What() const { return fWhat; } | |
96 | Double_t Xmin() const { return fXmin; } | |
97 | Double_t Xmax() const { return fXmax; } | |
98 | Double_t Ymin() const { return fYmin; } | |
99 | Double_t Ymax() const { return fYmax; } | |
100 | ||
101 | Double_t WidthX() const { return TMath::Abs(fXmin-fXmax); } | |
102 | ||
103 | Double_t WidthY() const { return TMath::Abs(fYmin-fYmax); } | |
104 | ||
105 | Bool_t Is2D() const { return fYmax > fYmin; } | |
106 | ||
107 | const char* GetName() const { return What().Data(); } | |
108 | ||
109 | TString AsString() const; | |
110 | ||
111 | virtual void Print(Option_t* opt="") const; | |
112 | ||
113 | Bool_t IsInRange(Double_t x, Double_t y=TMath::Limits<Double_t>::Max()) const; | |
114 | ||
115 | TString Flavour() const { return fFlavour; } | |
116 | ||
117 | private: | |
118 | TString fWhat; // what this range is about (e.g. J/psi particle, event, etc...) | |
119 | TString fQuantity; // binning type (e.g. pt, y, phi) | |
120 | Double_t fXmin; // x-min of the range | |
121 | Double_t fXmax; // x-max of the range | |
122 | Double_t fYmin; // x-min of the range | |
123 | Double_t fYmax; // x-max of the range | |
124 | TString fFlavour; // flavour (if any) this range, e.g. coarse, fine, etc... | |
125 | ||
126 | ClassDef(AliAnalysisMuMuBinning::Range,3) | |
127 | }; | |
128 | ||
129 | ||
130 | private: | |
131 | ||
132 | TMap* fBins; // list of bins (what -> list of bins) = (TObjString -> TObjArray) | |
133 | ||
134 | ClassDef(AliAnalysisMuMuBinning,2) // custom binning for MuMu analysis | |
135 | }; | |
136 | ||
137 | #endif |