PairAnalysisObjectCuts.h

Go to the documentation of this file.

#ifndef PAIRANALYSISOBJECTCUTS_H
#define PAIRANALYSISOBJECTCUTS_H
 
//#############################################################
//#                                                           #
//#         Class PairAnalysisObjectCuts                     #
//#         Provide cuts for using objects                    #
//#                                                           #
//#  Authors:                                                 #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#                                                           #
//#############################################################
 
#include <Rtypes.h>
#include <TBits.h>
 
#include "AnalysisCuts.h"
#include "PairAnalysisVarManager.h"
 
class TGraph;
class THnBase;
class PairAnalysisObjectCuts : public AnalysisCuts {
public:
  // Whether all cut criteria have to be fulfilled of just any
  enum CutType { kAll = 0, kAny };
  enum { kNMaxCuts = 10 };
 
  PairAnalysisObjectCuts();
  PairAnalysisObjectCuts(const char* name, const char* title);
  virtual ~PairAnalysisObjectCuts();
  //TODO: make copy constructor and assignment operator public
  void AddCut(PairAnalysisVarManager::ValueTypes type,
              const char* formulaMin,
              const char* formulaMax,
              Bool_t excludeRange = kFALSE);
  void AddCut(const char* formula,
              const char* formulaMin,
              const char* formulaMax,
              Bool_t excludeRange = kFALSE);
 
  void AddCut(PairAnalysisVarManager::ValueTypes type,
              TGraph* const graphMin,
              TGraph* const graphMax,
              Bool_t excludeRange = kFALSE);
  void AddCut(const char* formula,
              TGraph* const graphMin,
              TGraph* const graphMax,
              Bool_t excludeRange = kFALSE);
 
  void AddCut(PairAnalysisVarManager::ValueTypes type,
              THnBase* const histMin,
              THnBase* const histMax,
              Bool_t excludeRange = kFALSE);
  void AddCut(const char* formula,
              THnBase* const histMin,
              THnBase* const histMax,
              Bool_t excludeRange = kFALSE);
 
  // setters
  void SetCutType(CutType type) { fCutType = type; }
 
  // getters
  CutType GetCutType() const { return fCutType; }
 
  Int_t GetNCuts() { return fNActiveCuts; }
 
  //
  //Analysis cuts interface
  //
  virtual Bool_t IsSelected(Double_t* const values);
  virtual Bool_t IsSelected(TObject* track);
  virtual Bool_t IsSelected(TList* /* list */) { return kFALSE; }
 
  //
  // Cut information
  //
  virtual UInt_t GetSelectedCutsMask() const { return fSelectedCutsMask; }
 
  virtual void Print(const Option_t* option = "") const;
 
 
private:
  TBits* fUsedVars;  // list of used variables
  UShort_t
    fActiveCuts[PairAnalysisObjectCuts::kNMaxCuts];  // list of activated cuts
  UShort_t fNActiveCuts  = 0;                        // number of acive cuts
  UInt_t fActiveCutsMask = 0;                        // mask of active cuts
 
  UInt_t fSelectedCutsMask =
    0;  // Maks of selected cuts, is available after calling IsSelected
  CutType fCutType = kAll;  // type of the cut: any, all
 
  Bool_t fCutExclude[PairAnalysisObjectCuts::kNMaxCuts];  // inverse cut logic?
  TObject*
    fCutMin[PairAnalysisObjectCuts::kNMaxCuts];  // use object as lower cut
  TObject*
    fCutMax[PairAnalysisObjectCuts::kNMaxCuts];  // use object as upper cut
  TFormula* fVarFormula
    [PairAnalysisObjectCuts::kNMaxCuts];  // use a formula for the variable
 
  PairAnalysisObjectCuts(const PairAnalysisObjectCuts& c);
  PairAnalysisObjectCuts& operator=(const PairAnalysisObjectCuts& c);
 
  ClassDef(PairAnalysisObjectCuts,
           1)  // Cut class for special cuts (formulas, graphs, histograms)
};
 
 
#endif