EVE/EveBase/AliEveEventSelector.h

   1 /**************************************************************************
   2  * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
   3  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
   4  * full copyright notice.                                                 *
   5  **************************************************************************/
   6
   7 ////////////////////////////////////////////////////////////////////////////
   8 //
   9 //  AliEveEventSelector class
  10 //  selects events according to given criteria
  11 //
  12 //  origin: Mikolaj Krzewicki, Nikhef, Mikolaj.Krzewicki@cern.ch
  13 //
  14 ////////////////////////////////////////////////////////////////////////////
  15
  16 #ifndef AliEveEventSelector_H
  17 #define AliEveEventSelector_H
  18
  19 #include "AliEveEventManager.h"
  20 #include <TObject.h>
  21 #include <TString.h>
  22
  23 class TTree;
  24 class TList;
  25 class AliESDEvent;
  26 class TEntryList;
  27 class AliEveEventManager;
  28 class TString;
  29 class TObjArray;
  30
  31 class AliEveEventSelector : public TObject
  32 {
  33 public:
  34   AliEveEventSelector(AliEveEventManager* evman);
  35   void SetEventManager(AliEveEventManager* evman) { fPEventManager = evman; }
  36   AliEveEventManager* GetEventManager() { return fPEventManager; }
  37   TTree* GetESDTree() const { return fPEventManager->GetESDTree(); }
  38
  39   Bool_t FindNext(Int_t& i);
  40   Bool_t FindPrev(Int_t& i);
  41
  42   void SetSelectionString( const TString& str );
  43   void SetSelectionString( const char* str );
  44   TString GetSelectionString() {return fString;}
  45   Bool_t GetSelectOnString() {return fSelectOnString;}
  46   void SetSelectOnString( const Bool_t b = kTRUE ) {fSelectOnString = b;}
  47   void SetWrapAround( const Bool_t b = kTRUE ) {fWrapAround = b;}
  48   Bool_t GetWrapAround() const { return fWrapAround; }
  49   void Update();
  50
  51   void SetSelectOnTriggerType( const Bool_t b = kTRUE ) {fSelectOnTriggerType = b;}
  52   void SetTriggerType( const TString& type );
  53   void SetTriggerType( const char* type );
  54   TString GetTriggerType() const {return fTriggerType;}
  55   Bool_t GetSelectOnTriggerType() {return fSelectOnTriggerType;}
  56
  57   void SetSelectOnTriggerString( const Bool_t b=kTRUE ) { fSelectOnTriggerString = b; }
  58   void SetTriggerSelectionString( const TString m );
  59   TString GetTriggerSelectionString() const { return fTriggerSelectionString; }
  60   Bool_t GetSelectOnTriggerString() const { return fSelectOnTriggerString; }
  61   void SetTriggerMaskPatternString( TString str) { fTriggerMaskPatternString = str; }
  62   TString GetTriggerMaskPatternString() const { return fTriggerMaskPatternString; }
  63
  64   void SetSelectOnMultiplicity( const Bool_t b = kTRUE) { fSelectOnMultiplicity = b; }
  65   Bool_t GetSelectOnMultiplicity() { return fSelectOnMultiplicity; }
  66   void SetMultiplicityLow( const Int_t low ) { fMultiplicityLow = low; }
  67   void SetMultiplicityHigh( const Int_t high ) { fMultiplicityHigh = high; }
  68   Int_t GetMultiplicityLow() { return fMultiplicityLow; }
  69   Int_t GetMultiplicityHigh() { return fMultiplicityHigh; }
  70
  71 protected:
  72   void UpdateEntryList();
  73   Bool_t CheckOtherSelection(AliESDEvent* ESD);
  74
  75   Bool_t FindNextInTree(TTree* tree, Int_t& i);
  76   Bool_t FindPrevInTree(TTree* tree, Int_t& i);
  77
  78 private:
  79   AliEveEventSelector(const AliEveEventSelector&);
  80   AliEveEventSelector& operator=(const AliEveEventSelector&);
  81
  82   AliEveEventManager *fPEventManager; //event manager
  83   Bool_t fWrapAround;                 //go back to the first event after the last?
  84
  85   Bool_t  fSelectOnString;            //whether to select on string expression
  86   TString fString;                    //expression to select on
  87   TEntryList* fPEntryList;            //list of enry numbers after selection
  88   Int_t  fEntryListId;                //current entry no. in entry list
  89   Int_t  fLastTreeSize;               //stores the last treesize
  90
  91   Bool_t  fSelectOnTriggerType;       //whether to select on trigger type
  92   TString fTriggerType;               //trigger type
  93
  94   Bool_t  fSelectOnTriggerString;     //whether to select on trigger type with a formula
  95   TString fTriggerSelectionString;    //trigger selection formula
  96   TString fTriggerMaskPatternString;  //trigger mask placeholder for evaluation
  97
  98   Bool_t fSelectOnMultiplicity;       //whether to select on multiplicity
  99   Int_t  fMultiplicityLow;            //lower bound on multiplicity
 100   Int_t  fMultiplicityHigh;           //higher bound on multiplicity
 101
 102
 103   ClassDef(AliEveEventSelector, 2);
 104 };
 105
 106 #endif