Adding Id to PWG3 classes for better tracking of the coverity defect fixes (Ivana)
[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEcutStep.cxx
1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 *                                                                        *
4 * Author: The ALICE Off-line Project.                                    *
5 * Contributors are mentioned in the code where appropriate.              *
6 *                                                                        *
7 * Permission to use, copy, modify and distribute this software and its   *
8 * documentation strictly for non-commercial purposes is hereby granted   *
9 * without fee, provided that the above copyright notice appears in all   *
10 * copies and that both the copyright notice and this permission notice   *
11 * appear in the supporting documentation. The authors make no claims     *
12 * about the suitability of this software for any purpose. It is          *
13 * provided "as is" without express or implied warranty.                  *
14 **************************************************************************/
15
16 /* $Id$ */
17
18 //
19 // Cut step class
20 // Select all tracks surviving cuts in one special cut step
21 // Used in AliHFEtrackFilter
22 // 
23 // Author:
24 //   Markus Fasel <M.Fasel@gsi.de>
25 //
26 #include <TObjArray.h>
27
28 #include "AliAnalysisCuts.h"
29 #include "AliCFCutBase.h"
30 #include "AliHFEcutStep.h"
31 #include "AliLog.h"
32 #include "AliMCEvent.h"
33
34 ClassImp(AliHFEcutStep)
35
36 //__________________________________________________________________
37 AliHFEcutStep::AliHFEcutStep(const Char_t *name):
38   TNamed(name, ""),
39   fCuts(NULL)
40 {
41   //
42   // Default Constructor
43   //
44   fCuts = new TObjArray;
45 }
46
47 //__________________________________________________________________
48 AliHFEcutStep::AliHFEcutStep(const AliHFEcutStep &o):
49   TNamed(o),
50   fCuts(NULL)
51 {
52   //
53   // Copy constructor
54   //
55   o.Copy(*this);
56 }
57
58 //__________________________________________________________________
59 AliHFEcutStep &AliHFEcutStep::operator=(const AliHFEcutStep &o){
60   //
61   // Assignment operator
62   //
63   if(this != &o)
64     o.Copy(*this);
65   return *this;
66 }
67  
68 //__________________________________________________________________
69 AliHFEcutStep::~AliHFEcutStep(){
70   //
71   // destructor
72   //
73   delete fCuts;
74 }
75
76 //__________________________________________________________________
77 void AliHFEcutStep::Copy(TObject &o) const{
78   //
79   // Copy into content into object o
80   //
81   TNamed::Copy(o);
82   AliHFEcutStep &target = dynamic_cast<AliHFEcutStep &>(o);
83
84   // Make copy
85   target.fCuts = dynamic_cast<TObjArray *>(fCuts->Clone());
86 }
87
88 //__________________________________________________________________
89 Bool_t AliHFEcutStep::IsSelected(TObject *o){
90   //
91   // Filter tracks in the given cut step
92   // Apply all cut objects
93   //
94   AliDebug(1, Form("Cut Step %s: Number of cut objects: %d", GetName(), fCuts->GetEntriesFast()));
95   if(!fCuts->GetEntriesFast()) return kTRUE;
96   Bool_t isSelected = kTRUE;
97   AliAnalysisCuts *cuts = NULL;
98   for(Int_t iCut = 0; iCut < fCuts->GetEntriesFast(); iCut++){
99     cuts = dynamic_cast<AliAnalysisCuts *>(fCuts->UncheckedAt(iCut));
100     if(cuts && (!cuts->IsSelected(o))) isSelected = kFALSE;
101   }
102   AliDebug(1, Form("Accepted: %s", isSelected ? "yes" : "no"));
103   return isSelected;
104
105
106 //__________________________________________________________________
107 AliAnalysisCuts *AliHFEcutStep::GetCut(const Char_t *cutName){
108   //
109   // return cut object
110   //
111   return dynamic_cast<AliAnalysisCuts *>(fCuts->FindObject(cutName));
112 }
113
114 //__________________________________________________________________
115 void AliHFEcutStep::AddCut(AliAnalysisCuts *cut){
116   //
117   // Add cut object to the cut step
118   //
119   fCuts->Add(cut);
120 }
121
122 //__________________________________________________________________
123 void AliHFEcutStep::SetMC(AliMCEvent *mc){
124   //
125   // Set MC information to the cuts in the cut step
126   //
127   AliCFCutBase *cfc = NULL;
128   for(Int_t icut = 0; icut < fCuts->GetEntriesFast(); icut++){
129     if((cfc = dynamic_cast<AliCFCutBase *>(fCuts->UncheckedAt(icut)))) cfc->SetMCEventInfo(mc);
130   }
131 }
132
133 //__________________________________________________________________
134 void AliHFEcutStep::SetRecEvent(AliVEvent *rec){
135   //
136   // Publish rec event to the cut step
137   //
138   AliCFCutBase *cfc = NULL;
139   for(Int_t icut = 0; icut < fCuts->GetEntriesFast(); icut++){
140     if((cfc = dynamic_cast<AliCFCutBase *>(fCuts->UncheckedAt(icut)))) cfc->SetRecEventInfo(rec);
141   }
142 }
143