AliVTrackSelection.h

Go to the documentation of this file.

#ifndef ALIVTRACKSELECTION_H_
#define ALIVTRACKSELECTION_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 AliVTrackSelection : public TObject {
public:
  AliVTrackSelection();
  AliVTrackSelection(const AliVTrackSelection &ref);
  AliVTrackSelection &operator=(const AliVTrackSelection &ref);
  virtual ~AliVTrackSelection();

  TObjArray *GetAcceptedTracks(const TClonesArray * const tracks);
  TObjArray *GetAcceptedTracks(const AliVEvent *const event);
  virtual bool IsTrackAccepted(AliVTrack * const trk) = 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;     

  ClassDef(AliVTrackSelection, 1);
};

#endif /* ALIVTRACKSELECTION_H_ */