Corrections for Coverity warnings
[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 // Cut step class
17 // Select all tracks surviving cuts in one special cut step
18 // Used in AliHFEtrackFilter
19 // 
20 // Author:
21 //   Markus Fasel <M.Fasel@gsi.de>
22 //
23 #include <TObjArray.h>
24
25 #include "AliAnalysisCuts.h"
26 #include "AliCFCutBase.h"
27 #include "AliHFEcutStep.h"
28 #include "AliLog.h"
29 #include "AliMCEvent.h"
30
31 ClassImp(AliHFEcutStep)
32
33 //__________________________________________________________________
34 AliHFEcutStep::AliHFEcutStep(const Char_t *name):
35   TNamed(name, ""),
36   fCuts(NULL)
37 {
38   //
39   // Default Constructor
40   //
41   fCuts = new TObjArray;
42 }
43
44 //__________________________________________________________________
45 AliHFEcutStep::AliHFEcutStep(const AliHFEcutStep &o):
46   TNamed(o),
47   fCuts(NULL)
48 {
49   //
50   // Copy constructor
51   //
52   o.Copy(*this);
53 }
54
55 //__________________________________________________________________
56 AliHFEcutStep &AliHFEcutStep::operator=(const AliHFEcutStep &o){
57   //
58   // Assignment operator
59   //
60   if(this != &o)
61     o.Copy(*this);
62   return *this;
63 }
64  
65 //__________________________________________________________________
66 AliHFEcutStep::~AliHFEcutStep(){
67   //
68   // destructor
69   //
70   delete fCuts;
71 }
72
73 //__________________________________________________________________
74 void AliHFEcutStep::Copy(TObject &o) const{
75   //
76   // Copy into content into object o
77   //
78   TNamed::Copy(o);
79   AliHFEcutStep &target = dynamic_cast<AliHFEcutStep &>(o);
80
81   // Make copy
82   target.fCuts = dynamic_cast<TObjArray *>(fCuts->Clone());
83 }
84
85 //__________________________________________________________________
86 Bool_t AliHFEcutStep::IsSelected(TObject *o){
87   //
88   // Filter tracks in the given cut step
89   // Apply all cut objects
90   //
91   AliDebug(1, Form("Cut Step %s: Number of cut objects: %d", GetName(), fCuts->GetEntriesFast()));
92   if(!fCuts->GetEntriesFast()) return kTRUE;
93   Bool_t isSelected = kTRUE;
94   AliAnalysisCuts *cuts = NULL;
95   for(Int_t iCut = 0; iCut < fCuts->GetEntriesFast(); iCut++){
96     cuts = dynamic_cast<AliAnalysisCuts *>(fCuts->UncheckedAt(iCut));
97     if(cuts && (!cuts->IsSelected(o))) isSelected = kFALSE;
98   }
99   AliDebug(1, Form("Accepted: %s", isSelected ? "yes" : "no"));
100   return isSelected;
101
102
103 //__________________________________________________________________
104 AliAnalysisCuts *AliHFEcutStep::GetCut(const Char_t *cutName){
105   //
106   // return cut object
107   //
108   return dynamic_cast<AliAnalysisCuts *>(fCuts->FindObject(cutName));
109 }
110
111 //__________________________________________________________________
112 void AliHFEcutStep::AddCut(AliAnalysisCuts *cut){
113   //
114   // Add cut object to the cut step
115   //
116   fCuts->Add(cut);
117 }
118
119 //__________________________________________________________________
120 void AliHFEcutStep::SetMC(AliMCEvent *mc){
121   //
122   // Set MC information to the cuts in the cut step
123   //
124   AliCFCutBase *cfc = NULL;
125   for(Int_t icut = 0; icut < fCuts->GetEntriesFast(); icut++){
126     if((cfc = dynamic_cast<AliCFCutBase *>(fCuts->UncheckedAt(icut)))) cfc->SetMCEventInfo(mc);
127   }
128 }
129
130 //__________________________________________________________________
131 void AliHFEcutStep::SetRecEvent(AliVEvent *rec){
132   //
133   // Publish rec event to the cut step
134   //
135   AliCFCutBase *cfc = NULL;
136   for(Int_t icut = 0; icut < fCuts->GetEntriesFast(); icut++){
137     if((cfc = dynamic_cast<AliCFCutBase *>(fCuts->UncheckedAt(icut)))) cfc->SetRecEventInfo(rec);
138   }
139 }
140