AnalysisFilter.cxx

Go to the documentation of this file.

//
// Manager class for filter decisions based on cuts
// The filter contains a list of sets of cuts and controll histograms (optional).
// A bit field is filled in order to store the decision of each cut-set.
//
// Authors:
//   * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
//   Julian Book   <Julian.Book@cern.ch>
//
// Add cut object to list via AddCuts(AnalysisCuts* cuts)
// Add control histograms via AddHistos(PairAnalysisHistos* histos)
//
 
#include "AnalysisFilter.h"
#include "AnalysisCuts.h"
#include "PairAnalysisHistos.h"
#include <THashList.h>
#include <TList.h>
#include <TObject.h>
 
#include "PairAnalysisVarManager.h"
 
ClassImp(AnalysisFilter)
 
 
  AnalysisFilter::AnalysisFilter()
  : TNamed(), fCuts(0), fHistos() {
  // Default constructor
  fHistos.SetOwner(kTRUE);
  fHistos.SetName("PairAnalysis_FilterHistos");
}
 
AnalysisFilter::AnalysisFilter(const char* name, const char* title)
  : TNamed(name, title)
  , fCuts(new TList())
  , fHistos()
//    fHistos(new TList())
{
  // Constructor
  fHistos.SetOwner(kTRUE);
  fHistos.SetName(name);
}
 
AnalysisFilter::AnalysisFilter(const AnalysisFilter& obj)
  : TNamed(obj), fCuts(0), fHistos() {
  // Copy constructor
  fCuts = obj.fCuts;
  //    fHistos = obj.fHistos;
}
 
AnalysisFilter::~AnalysisFilter() {
  // Destructor
  if (fCuts) fCuts->Delete("slow");
  //if (fHistos) fHistos->Delete("slow");
  fHistos.Clear();
  delete fCuts;
}
 
AnalysisFilter& AnalysisFilter::operator=(const AnalysisFilter& other) {
  // Assignment
  if (&other != this) {
    TNamed::operator=(other);
    fCuts           = other.fCuts;
    //     fHistos = other.fHistos;
  }
  return *this;
}
 
UInt_t AnalysisFilter::IsSelected(TObject* obj) {
  //
  // Loop over all set of cuts
  // and store the decision
  UInt_t result = 0;
  UInt_t filterMask;
 
  TIter next(fCuts);
  AnalysisCuts* cuts;
  Int_t iCutB = 1;
 
 
  //Fill values - fill only once the variables
  // TODO: check carefully the case of pairlegcuts,
  //       before switching to IsSelected(values)
  // Double_t values[PairAnalysisVarManager::kNMaxValuesMC];
  // PairAnalysisVarManager::Fill(obj,values);
 
  while ((cuts = (AnalysisCuts*) next())) {
    //Bool_t acc = cuts->IsSelected(values);
    Bool_t acc = cuts->IsSelected(obj);  
    if ((filterMask = cuts->GetFilterMask()) > 0) {
      acc = (acc && (filterMask == result));
    }
    cuts->SetSelected(acc);
    if (acc) { result |= iCutB & 0x00ffffff; }
    iCutB *= 2;
  }
 
  return result;
}
 
UInt_t AnalysisFilter::IsSelected(Double_t* const values) {
  //
  // Loop over all set of cuts
  // and store the decision
  UInt_t result = 0;
  UInt_t filterMask;
 
  TIter next(fCuts);
  AnalysisCuts* cuts;
  Int_t iCutB = 1;
 
  while ((cuts = (AnalysisCuts*) next())) {
    Bool_t acc = cuts->IsSelected(values);
    if ((filterMask = cuts->GetFilterMask()) > 0) {
      acc = (acc && (filterMask == result));
    }
    cuts->SetSelected(acc);
    if (acc) { result |= iCutB & 0x00ffffff; }
    iCutB *= 2;
  }
 
  return result;
}
 
UInt_t AnalysisFilter::IsSelected(TList* list) {
  //
  // Loop over all set of cuts
  // and store the decision
  UInt_t result = 0;
  UInt_t filterMask;
 
  TIter next(fCuts);
  AnalysisCuts* cuts;
  Int_t iCutB = 1;
 
  while ((cuts = (AnalysisCuts*) next())) {
    Bool_t acc = cuts->IsSelected(list);
    if ((filterMask = cuts->GetFilterMask()) > 0) {
      acc = (acc && (filterMask & result));
    }
    cuts->SetSelected(acc);
    if (acc) { result |= iCutB & 0x00ffffff; }
    iCutB *= 2;
  }
 
  return result;
}
 
void AnalysisFilter::Init() {
  //
  // Loop over all set of cuts and call Init
  TIter next(fCuts);
  AnalysisCuts* cuts;
  while ((cuts = (AnalysisCuts*) next()))
    cuts->Init();
}
 
void AnalysisFilter::AddCuts(AnalysisCuts* cuts) {
  // Add a set of cuts
  fCuts->Add(cuts);
}
 
void AnalysisFilter::AddHistos(PairAnalysisHistos* histos) {
  //
  // add histos for each cut
  //
  fHistos.Clear();
  TIter next(fCuts);
  AnalysisCuts* cuts;
  Int_t iCut = 0;
  while ((cuts = (AnalysisCuts*) next())) {
    fHistos.AddAt(
      (PairAnalysisHistos*) histos->GetHistogramList()->Clone(cuts->GetName()),
      iCut++);
    //    printf("AnalysisFilter::AddHistos add histos for %s at %d \n",cuts->GetName(),iCut);
  }
}
 
Bool_t AnalysisFilter::IsSelected(char* name) {
  //
  // Returns current result for cut with name
  AnalysisCuts* cut = (AnalysisCuts*) (fCuts->FindObject(name));
  if (cut) {
    return (cut->Selected());
  } else {
    return 0;
  }
}