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     if (cut) {
 138       return (cut->Selected());
 139     } else  {
 140       return 0;
 141     }
 142 }