AliEmcalTrackSelection.h

Go to the documentation of this file.

#ifndef ALIEMCALTRACKSELECTION_H_
#define ALIEMCALTRACKSELECTION_H_
/* Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include <TObject.h>
#include <TBits.h>

class TClonesArray;
class TObjArray;
class AliVCuts;
class AliVEvent;
class AliVTrack;

class AliEmcalTrackSelection : public TObject {
public:
  enum ETrackFilterType_t {
    kNoTrackFilter = 0,
    kCustomTrackFilter,
    kHybridTracks,
    kTPCOnlyTracks
  };

  AliEmcalTrackSelection();
  AliEmcalTrackSelection(const AliEmcalTrackSelection &ref);
  AliEmcalTrackSelection &operator=(const AliEmcalTrackSelection &ref);
  virtual ~AliEmcalTrackSelection();

  TObjArray *GetAcceptedTracks(const TClonesArray * const tracks);
  TObjArray *GetAcceptedTracks(const AliVEvent *const event);
  virtual bool IsTrackAccepted(AliVTrack * const trk) = 0;

  virtual void GenerateTrackCuts(ETrackFilterType_t type, const char* period = "") = 0;
  void AddTrackCuts(AliVCuts *cuts);
  void AddTrackCuts(TObjArray *cuts);
  Int_t GetNumberOfCutObjects() const;
  AliVCuts *GetTrackCuts(Int_t icut);

  const TBits& GetTrackBitmap() const { return fTrackBitmap; }
  const TClonesArray* GetAcceptedTrackBitmaps() const { return fListOfTrackBitmaps; }

  void SetSelectionModeAny() { fSelectionModeAny = kTRUE ; }
  void SetSelectionModeAll() { fSelectionModeAny = kFALSE; }

protected:
  TObjArray    *fListOfTracks;         
  TClonesArray *fListOfTrackBitmaps;   
  TBits         fTrackBitmap;          
  TObjArray    *fListOfCuts;           
  Bool_t        fSelectionModeAny;     


};

#endif /* ALIEMCALTRACKSELECTION_H_ */