fixing rule violations
[u/mrichter/AliRoot.git] / PWGGA / GammaConv / ConvCorrelations / AliAnaConvIsolation.h
1 /* This file is property of and copyright                                 *\r
2  * ALICE Experiment at CERN, All rights reserved.                         *\r
3  * See cxx source for full Copyright notice                               */\r
4 \r
5 /// @file   AliAnalysisTaskGammaJet.h\r
6 /// @author Svein Lindal\r
7 /// @brief  Class used to run isolation studies of conversion gamma/pions\r
8  \r
9 \r
10 #ifndef ALIANACONVISOLATION_H\r
11 #define ALIANACONVISOLATION_H\r
12 \r
13 #include <iostream>\r
14 #include "TObject.h"\r
15 #include "Rtypes.h"\r
16 #include <TMath.h>\r
17 #include "TString.h"\r
18 class TH2F;\r
19 class TF1;\r
20 class TH1F;\r
21 class AliAODConversionPhoton;\r
22 class AliAODConversionParticle;\r
23 class TClonesArray;\r
24 class TList;\r
25 \r
26 using namespace std;\r
27 \r
28 class AliAnaConvIsolation : public TObject {\r
29 \r
30 public:\r
31   \r
32   AliAnaConvIsolation(); \r
33   AliAnaConvIsolation(Float_t coneSize, Float_t maxPtThreshold, Float_t sumPtThreshold, Float_t maxPtFraction, Float_t sumPtFraction);\r
34   virtual ~AliAnaConvIsolation();\r
35   \r
36 \r
37   //void CreateHistograms();\r
38 \r
39 \r
40   ///Set And get cone size\r
41   void SetConeSize(Float_t cs) {fConeSize = cs*cs;}\r
42   Float_t GetConeSize() const {return TMath::Sqrt(fConeSize);}\r
43 \r
44 \r
45   //Set and get max pt threshold\r
46   void SetMaxPtThreshold(Float_t cs) {fMaxPtThreshold = cs;}\r
47   Float_t GetPtThreshold() const {return fMaxPtThreshold;}\r
48 \r
49 \r
50   //Set and get sum pt threshold\r
51   void SetSumPtThreshold(Float_t cs) {fSumPtThreshold = cs;}\r
52   Float_t GetPtSumThreshold() const {return fSumPtThreshold;}\r
53 \r
54   //Set and get max Pt fraction threshold\r
55   void SetMaxPtFraction(Float_t cs) {fMaxPtFraction = cs;}\r
56   Float_t GetPtFraction() const {return fMaxPtFraction;}\r
57 \r
58   //Set and get sum pt fraction threshold\r
59   void SetSumPtFraction(Float_t cs) {fSumPtFraction = cs;}\r
60   Float_t GetPtSumFraction() const {return fSumPtFraction;}\r
61 \r
62   //Set min pt for a particle to be counted in bg\r
63   void SetMinPt( Float_t minpt) {fMinPt = minpt; }\r
64 \r
65   //Get isolation curve\r
66   TF1 * GetIsolationCurve() const { return fIsoCurve; }\r
67 \r
68   //Set function of isolation curve\r
69   void SetIsolationCurve(TString  curve) { \r
70     fCurveFunction = curve;\r
71   }\r
72 \r
73   Bool_t IsLeading(AliAODConversionParticle  * particle, const TObjArray * tracks, const TObjArray * aodParticles);\r
74 \r
75 \r
76   //Fill histograms\r
77   //void FillHistograms(Float_t pt, Float_t ptMax, Float_t ptSum, Bool_t isolated, Int_t nTracks);\r
78 \r
79   //Get list of histograms\r
80   //TList * GetHistograms() const { return fHistograms;}\r
81 \r
82   //Is particle isolated\r
83   Bool_t IsIsolated( const AliAODConversionPhoton * const particle, const TClonesArray * const tracks, Bool_t &leading);\r
84   Bool_t IsIsolated( AliAODConversionPhoton * const particle, const TClonesArray * const tracks, const Int_t nSpawn, const Int_t * const spawn, Bool_t &leading );\r
85   Int_t IsLeading(const AliAODConversionParticle * particle, const TObjArray * tracks, const Int_t * tIDs) const;\r
86 \r
87   //Is eta - phi distance smaller than conesize ?\r
88   Bool_t IsInCone(Float_t dEta, Float_t dPhi, const Float_t coneSize) const {\r
89         dPhi = (TMath::Abs(dPhi) < TMath::Pi()) ? dPhi : TMath::TwoPi() - TMath::Abs(dPhi); \r
90     return ( (dEta*dEta + dPhi*dPhi) < coneSize);\r
91   }\r
92 \r
93  private:\r
94 \r
95   ///Evaluate whether particle is isolated according to criterie\r
96   Bool_t EvaluateIsolationCriteria(Float_t ptSum, Float_t pt) const;\r
97 \r
98   TF1 * fIsoCurve; ///Curve defining if particle is isolated or not \r
99   TString fCurveFunction; ///Funtion defining curve\r
100 \r
101   Float_t fConeSize; //Size of isolation cone\r
102   Float_t fMinPt; //min pt for bg particles\r
103   Float_t fMaxPtThreshold; //max pt threshold\r
104   Float_t fSumPtThreshold; //sum pt threhold\r
105   Float_t fMaxPtFraction;  //max pt fraction threshold\r
106   Float_t fSumPtFraction; //sum pt fraction threshold\r
107 \r
108   // TList * fHistograms; //list of histograms\r
109 \r
110   // TH2F * fhMaxPtInCone[2]; //histogram of max pt in cone\r
111   // TH2F * fhSumPtInCone[2]; //histogram of sum pt in cone\r
112   // TH2F * fhSumPtVsMaxPt[2];//sum pt vs max pt\r
113   \r
114   // // TH1F * fHistSumPt[2][2]; \r
115   // // TH1F * fHistMaxPt[2][2];\r
116   \r
117   // TH1F * fhPtCandidates[2]; //pt distribution of isolation candidates\r
118   // TH1F * fhTrackMult[2];    //Track multiplicity of events with / wo isolated particles\r
119   \r
120 \r
121   // Float_t fHistogramMaxPt; //Upper pt limit in histograms\r
122 \r
123 \r
124   AliAnaConvIsolation(const AliAnaConvIsolation&); // not implemented\r
125   AliAnaConvIsolation& operator=(const AliAnaConvIsolation&); // not implemented\r
126   ClassDef(AliAnaConvIsolation, 2); // example of analysis\r
127 };\r
128 \r
129 #endif\r