PairAnalysisCutGroup.cxx

Go to the documentation of this file.

 
//                                                                      //
// Authors:
//   * Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
//   Julian Book   <Julian.Book@cern.ch>
//
//
//                                                                      //
//   Allow to define groups of cut conditions which are tested with     //
//      an OR condition between groups and an AND within groups         //
//                                                                      //
 
#include "PairAnalysisCutGroup.h"
 
#include "PairAnalysisVarManager.h"
 
ClassImp(PairAnalysisCutGroup)
 
  PairAnalysisCutGroup::PairAnalysisCutGroup(Bool_t compOperator /*=kCompOR*/)
  : AnalysisCuts()
  , fCutGroupList(0x0)
  , fCompOperator(compOperator)
{
  //
  // Default constructor
  //
}
 
//_____________________________________________________________________
PairAnalysisCutGroup::PairAnalysisCutGroup(const char* name, const char* title, Bool_t compOperator /*=kCompOR*/)
  : AnalysisCuts(name, title)
  , fCutGroupList(0x0)
  , fCompOperator(compOperator)
{
  //
  // Named Constructor
  //
}
 
//_____________________________________________________________________
PairAnalysisCutGroup::~PairAnalysisCutGroup()
{
  //
  //Default Destructor
  //
}
 
//_____________________________________________________________________
void PairAnalysisCutGroup::Init()
{
  // Loop over all cuts and call Init
  TIter next(&fCutGroupList);
  while (AnalysisCuts* thisCut = (AnalysisCuts*) next())
    thisCut->Init();
}
 
//________________________________________________________________________
Bool_t PairAnalysisCutGroup::IsSelected(Double_t* const values)
{
  //
  // Make cut decision
  //
 
  //Different init for and/or makes code shorter
  Bool_t selectionResult = fCompOperator;
 
  TIter listIterator(&fCutGroupList);
  while (AnalysisCuts* thisCut = (AnalysisCuts*) listIterator()) {
    if (fCompOperator == static_cast<Bool_t>(ETruthValues::kCompOR)) {
      selectionResult = (selectionResult || thisCut->IsSelected(values));
    }
    else {  //kCompAND
      selectionResult = (selectionResult && thisCut->IsSelected(values));
    }
  }
  return selectionResult;
}
 
//_____________________________________________________________________
Bool_t PairAnalysisCutGroup::IsSelected(TObject* track)
{
  //
  // Selection-finder handling different comparison operations
  //
 
  if (!track) return kFALSE;
 
  //Fill values
  Double_t* values = PairAnalysisVarManager::GetData();
  PairAnalysisVarManager::Fill(track, values);
 
  return (IsSelected(values));
}
 
//_____________________________________________________________________
 
void PairAnalysisCutGroup::AddCut(AnalysisCuts* fCut)
{
  //
  // Add a defined cut to the list
  //
 
  fCutGroupList.Add(fCut);
}
 
//_____________________________________________________________________
void PairAnalysisCutGroup::SetCompOperator(Bool_t compOperator)
{
  //
  // Switch between AND/OR
  //
 
  fCompOperator = compOperator;
}
 
//________________________________________________________________________
void PairAnalysisCutGroup::Print(const Option_t* /*option*/) const
{
  //
  // Print cuts and the range
  //
 
  printf("*****************************************************************\n");
  printf("cut group '%s'\n", GetTitle());
  if (fCompOperator == static_cast<Bool_t>(ETruthValues::kCompAND)) { printf("Cut are compared with AND \n"); }
  else {
    printf("Cut are compared with OR \n");
  }
  TIter listIterator(&fCutGroupList);
  while (AnalysisCuts* thisCut = (AnalysisCuts*) listIterator()) {
    thisCut->Print();
  }
  printf("*****************************************************************\n");
}