Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronV0Cuts.h

#ifndef ALIDIELECTRONV0CUTS_H
#define ALIDIELECTRONV0CUTS_H

/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#############################################################
//#                                                           # 
//#         Class AliDielectronV0Cuts                        #
//#         Provide cuts for all variables handled in         #
//#           AliDielectronV0Manager                         #
//#                                                           #
//#  Authors:                                                 #
//#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
//#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
//#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
//#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
//#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
//#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
//#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
//#                                                           #
//#############################################################

#include <Rtypes.h>

#include <AliDielectronVarCuts.h>

class AliDielectronV0Cuts : public AliDielectronVarCuts {
public:
  enum EV0finder {
    kAll = 0,
    kOffline,
    kOnTheFly
  };
  enum PIDCutType { kBoth=0, kAny };

  AliDielectronV0Cuts();
  AliDielectronV0Cuts(const char* name, const char* title);
  virtual ~AliDielectronV0Cuts();
  //TODO: make copy constructor and assignment operator public

  //
  //Analysis cuts interface
  //
  void InitEvent(AliVTrack *trk);
  Bool_t IsNewEvent(const AliVEvent *ev);
  virtual Bool_t IsSelected(TObject* track);
  virtual Bool_t IsSelected(TList*   /* list */ ) {return kFALSE;}
  void SetV0finder(EV0finder finder) {fV0finder=finder;}
  void SetPdgCodes(Int_t mother, Int_t negDaughter, Int_t posDaughter) {fMotherPdg=mother; fNegPdg=negDaughter; fPosPdg=posDaughter;}
  void SetExcludeTracks(Bool_t exclude) {fExcludeTracks=exclude;}
  void SetDefaultPID(Int_t def, PIDCutType type=kBoth) {fPID=def; fPIDCutType=type; }
  void ResetUniqueEventNumbers() { fOrbit=0; fPeriod=0; fBunchCross=0; }

  // cut information
  virtual void Print(const Option_t* option = "") const;

private:

  TBits fV0TrackArr;                        // array with booleans where TrackID corresponds to bitnumber
  Bool_t fExcludeTracks;                       // cut logic: exclude or include tracks corresponding to a V0 candidate
  EV0finder fV0finder;                      // which v0 finder

  Int_t fMotherPdg;                         // target pdg code of the mother
  Int_t fNegPdg;                            // target pdg code of the negative daughter
  Int_t fPosPdg;                            // target pdg code of the positive daughter
  Int_t fPID;                               // default PID usage (see AliDielectronPID)
  PIDCutType fPIDCutType;                   // apply PIDcuts to one or both daughter tracks

  // memebers needed to identify an event
  UInt_t fOrbit;                            // orbit number
  UInt_t fPeriod;                           // period number
  UShort_t fBunchCross;                     // bunch cross number

  AliDielectronV0Cuts(const AliDielectronV0Cuts &c);
  AliDielectronV0Cuts &operator=(const AliDielectronV0Cuts &c);

  ClassDef(AliDielectronV0Cuts,3)          // cut class for V0 candidates
};

#endif