Bug fix in array size
[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisFilter.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 // Manager class for filter decisions based on cuts
20 // The filter contains a list of sets of cuts.
21 // A bit field is filled in order to store the decision of each cut-set. 
22 // Author: Andreas Morsch
23 // andreas.morsch@cern.ch
24
25 #include <TObject.h>
26 #include <TList.h>
27 #include "AliAnalysisFilter.h"
28 #include "AliAnalysisCuts.h"
29
30
31 ClassImp(AliAnalysisFilter)
32
33
34 ////////////////////////////////////////////////////////////////////////
35
36 AliAnalysisFilter::AliAnalysisFilter():
37     TNamed(),
38     fCuts(0)
39 {
40   // Default constructor
41 }
42
43 AliAnalysisFilter::AliAnalysisFilter(const char* name, const char* title):
44     TNamed(name, title),
45     fCuts(new TList())
46 {
47   // Constructor
48 }
49
50 AliAnalysisFilter::AliAnalysisFilter(const AliAnalysisFilter& obj):
51     TNamed(obj),
52     fCuts(0)
53 {
54 // Copy constructor
55     fCuts = obj.fCuts;
56 }
57
58
59 AliAnalysisFilter& AliAnalysisFilter::operator=(const AliAnalysisFilter& other)
60 {
61 // Assignment
62     TNamed::operator=(other);
63     fCuts = other.fCuts;
64     return *this;
65    }
66    
67 UInt_t AliAnalysisFilter::IsSelected(TObject* obj)
68 {
69     //
70     // Loop over all set of cuts
71     // and store the decision
72     UInt_t result = 0;
73     UInt_t filterMask;
74     
75     TIter next(fCuts);
76     AliAnalysisCuts *cuts;
77     Int_t iCutB = 1;
78         
79     while((cuts = (AliAnalysisCuts*)next())) {
80         Bool_t acc = cuts->IsSelected(obj);
81         if ((filterMask = cuts->GetFilterMask()) > 0) {
82             acc = (acc && (filterMask & result));
83         }
84         cuts->SetSelected(acc);
85         if (acc) {result |= iCutB & 0x00ffffff;}
86         iCutB *= 2;
87     }  
88
89     return result;
90 }
91
92 UInt_t AliAnalysisFilter::IsSelected(TList* list)
93 {
94     //
95     // Loop over all set of cuts
96     // and store the decision
97     UInt_t result = 0;
98     UInt_t filterMask;
99
100     TIter next(fCuts);
101     AliAnalysisCuts *cuts;
102     Int_t iCutB = 1;
103         
104     while((cuts = (AliAnalysisCuts*)next())) {
105         Bool_t acc = cuts->IsSelected(list);
106         if ((filterMask = cuts->GetFilterMask()) > 0) {
107             acc = (acc && (filterMask & result));
108         }
109         cuts->SetSelected(acc);
110         if (acc) {result |= iCutB & 0x00ffffff;}
111         iCutB *= 2;
112     }  
113
114     return result;
115 }
116
117 void AliAnalysisFilter::Init()
118 {
119     //
120     // Loop over all set of cuts and call Init
121     TIter next(fCuts);
122     AliAnalysisCuts *cuts;
123     while((cuts = (AliAnalysisCuts*)next())) cuts->Init();
124 }
125
126 void AliAnalysisFilter::AddCuts(AliAnalysisCuts* cuts)
127 {
128     // Add a set of cuts
129     fCuts->Add(cuts);
130 }
131
132 Bool_t AliAnalysisFilter::IsSelected(char* name)
133 {
134     //
135     // Returns current result for cut with name
136     AliAnalysisCuts* cut = (AliAnalysisCuts*) (fCuts->FindObject(name));
137     return (cut->Selected());
138 }