Modifications for the xlc compiler
[u/mrichter/AliRoot.git] / ANALYSIS / AliEventCut.cxx
1 #include "AliEventCut.h"
2 //________________________________
3 ///////////////////////////////////////////////////////////
4 //
5 // class AliRunAnalysis
6 //
7 //
8 //
9 //
10 ///////////////////////////////////////////////////////////
11
12 #include <TObjArray.h>
13
14 #include "AliEventBaseCut.h"
15
16 ClassImp(AliEventCut)
17
18 AliEventCut::AliEventCut():
19  fBaseCuts(10)
20 {
21 //costructor
22
23 }
24 /*********************************************************/
25 AliEventCut::AliEventCut(const AliEventCut& in):
26  TObject(in),
27  fBaseCuts(in.fBaseCuts)
28 {
29   //cpy ctor
30   fBaseCuts.SetOwner(kTRUE);
31 }
32 /*********************************************************/
33
34 AliEventCut::~AliEventCut()
35 {
36 //costructor
37 }
38
39 /*********************************************************/
40
41 Bool_t AliEventCut::Rejected(AliAOD* aod) const
42 {
43   //returns kTRUE if rejected
44   if (aod == 0x0)
45    {
46      Error("Pass","Pointer to AOD is NULL. Not passed the cut");
47      return kFALSE;
48    }
49    
50   TIter iter(&fBaseCuts);
51   AliEventBaseCut* becut;
52   while (( becut = (AliEventBaseCut*)iter() ))
53    {
54      if (becut->Rejected(aod)) return kTRUE;
55    }
56   return kFALSE;
57 }
58 /*********************************************************/
59 void AliEventCut::AddBasePartCut(AliEventBaseCut* ebcut)
60 {
61 //Adds a base cut
62  if (ebcut == 0x0)
63   {
64     Error("AddBasePartCut","Pointer to base cut is NULL");
65     return;
66   }
67  
68  if (ebcut->GetProperty() != AliEventBaseCut::kNone)
69   {
70     if (FindCut(ebcut->GetProperty()))
71      {
72        Warning("AddBasePartCut","Cut with this property is already in the list of base cuts");
73      }
74   }  
75   
76  fBaseCuts.Add(ebcut->Clone());
77  
78 }
79 /*********************************************************/
80
81 AliEventBaseCut* AliEventCut::FindCut(AliEventBaseCut::EEventCutProperty prop)
82 {
83 //Finds and returns pointer to the cut with given property
84  Int_t n = fBaseCuts.GetEntries();
85  for (Int_t i = 0; i<n; i++)
86   {
87     AliEventBaseCut* bcut = (AliEventBaseCut*)fBaseCuts.At(i);
88     if (bcut->GetProperty() == prop)
89        return bcut; //we found the cut we were searching for
90   }
91
92  return 0x0; //we did not found this cut
93
94 }
95 /*********************************************************/
96
97 void AliEventCut::SetNChargedRange(Int_t min,Int_t max,Double_t etamin,Double_t etamax)
98 {
99  //Sets renge of number of charged particles
100   AliNChargedCut* cut = dynamic_cast<AliNChargedCut*>(FindCut(AliEventBaseCut::kNChargedCut));
101   if(cut) 
102    { 
103      cut->SetRange(min,max);
104      cut->SetEtaRange(etamin,etamax);
105    }  
106   else fBaseCuts.Add(new AliNChargedCut(min,max,etamin,etamax));
107 }
108 /*********************************************************/
109
110 void AliEventCut::SetVertexXRange(Double_t min, Double_t max)
111 {
112   //Sets range of z coordinate of a primary vertex
113   AliEventBaseCut* cut = FindCut(AliEventBaseCut::kPrimVertexXCut);
114   if (cut) cut->SetRange(min,max);
115   else fBaseCuts.Add(new AliPrimVertexXCut(min,max));
116 }
117 /*********************************************************/
118
119 void AliEventCut::SetVertexYRange(Double_t min, Double_t max)
120 {
121   //Sets range of z coordinate of a primary vertex
122   AliEventBaseCut* cut = FindCut(AliEventBaseCut::kPrimVertexYCut);
123   if (cut) cut->SetRange(min,max);
124   else fBaseCuts.Add(new AliPrimVertexYCut(min,max));
125 }
126 /*********************************************************/
127
128 void AliEventCut::SetVertexZRange(Double_t min, Double_t max)
129 {
130   //Sets range of z coordinate of a primary vertex
131   AliEventBaseCut* cut = FindCut(AliEventBaseCut::kPrimVertexZCut);
132   if (cut) cut->SetRange(min,max);
133   else fBaseCuts.Add(new AliPrimVertexZCut(min,max));
134 }
135 /*********************************************************/
136 /*********************************************************/
137 /*********************************************************/
138
139 ClassImp(AliEventEmptyCut)