Various updates

[u/mrichter/AliRoot.git] / PWGPP / TRD / AliTRDinfoGen.h

#ifndef ALITRDINFOGEN_H\r
#define ALITRDINFOGEN_H\r
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 * See cxx source for full Copyright notice                               */\r
\r
/* $Id: AliTRDinfoGen.h 27496 2008-07-22 08:35:45Z cblume $ */\r
\r
////////////////////////////////////////////////////////////////////////////\r
//                                                                        //\r
//  TRD Performance tender wagon                                          //\r
//                                                                        //\r
////////////////////////////////////////////////////////////////////////////\r
\r
#ifndef ALIANALYSISTASKSE_H\r
#include "AliAnalysisTaskSE.h"\r
#endif\r
\r
class AliESDEvent;\r
class AliMCEvent;\r
class AliESDfriend;\r
class AliTRDtrackInfo;\r
class AliTRDeventInfo;\r
class AliTRDv0Info;\r
class AliTRDeventCuts;\r
class AliESDtrackCuts;\r
class TObjArray;\r
class TString;\r
class TTreeSRedirector;\r
class AliTRDReconstructor;\r
class AliTRDgeometry;\r
class AliTRDinfoGen : public AliAnalysisTaskSE\r
{\r
public:\r
  enum AliTRDinfoGenSteeringBits{\r
    kMCdata               = BIT(18)\r
   ,kUseLocalTrkSelection = BIT(19)\r
   ,kCollision            = BIT(20)\r
   ,kOCDB                 = BIT(21)\r
   ,kTrkPoints            = BIT(22)\r
  };\r
  enum AliTRDinfoGenObjects{\r
     kTracksESD =  0\r
    ,kTracksMC\r
    ,kV0\r
    ,kTPC\r
    ,kITS\r
    ,kTRDin\r
    ,kTRDout\r
    ,kBarrel\r
    ,kBarrelMC\r
    ,kSA\r
    ,kSAMC\r
    ,kKink\r
    ,kKinkMC\r
    ,kBarrelFriend\r
    ,kSAFriend\r
    ,kNObjects\r
  };\r
  enum AliTRDinfoGenClasses{\r
     kStatTrk = 0\r
    ,kEvType\r
//    ,kBCtrack\r
    ,kBC\r
    ,kTrigger\r
    ,kChmb\r
    ,kNclasses\r
  };\r
  AliTRDinfoGen();\r
  AliTRDinfoGen(char* name);\r
  virtual ~AliTRDinfoGen();\r
  \r
  static Float_t GetEndITS() { return fgkITS;}\r
  static Float_t GetEndTPC() { return fgkTPC;}\r
  static Float_t GetEndTRD() { return fgkTRD;}\r
  Int_t   GetNRefFigures() const  { return 1;} \r
  const char* GetOCDB() const {return fOCDB.Data();}\r
  Bool_t  GetRefFigure(Int_t ifig);\r
  Bool_t  Load(const Char_t *fn="AnalysisResults.root", const Char_t *dir="TRD_Performance", const Char_t *name=NULL);\r
\r
  Bool_t  HasMCdata() const       { return TestBit(kMCdata);};\r
  // temporary until check with AliAnalysisTaskSE collision selection mechanism\r
  Bool_t  IsInitOCDB() const {return TestBit(kOCDB);}\r
  Bool_t  IsCollision() const {return TestBit(kCollision);}\r
  Bool_t  HasTrackPoints() const {return TestBit(kTrkPoints);}\r
  void    MakeSummary();\r
  static const AliTRDReconstructor* Reconstructor() {return fgReconstructor;}\r
  static AliTRDgeometry*      Geometry() {return fgGeo;}\r
  void    SetInitOCDB(Bool_t set=kTRUE) {SetBit(kOCDB, set);}\r
  void    SetCollision(Bool_t set=kTRUE) {SetBit(kCollision, set);}\r
  //void    SetLocalEvSelection(const AliTRDeventCuts */*cut*/){;} \r
  void    SetLocalEvSelection(const AliTRDeventCuts &ec);\r
  //void    SetLocalTrkSelection(const AliESDtrackCuts */*cut*/){;} \r
  void    SetLocalTrkSelection(Bool_t use=kTRUE) {SetBit(kUseLocalTrkSelection, use);}\r
  void    SetLocalV0Selection(const AliTRDv0Info &v0);\r
  void    SetMCdata(Bool_t mc = kTRUE) {SetBit(kMCdata, mc);}\r
  void    SetOCDB(const char *ocdb) {fOCDB=ocdb;}\r
\r
  Bool_t  UseLocalEvSelection() const {return Bool_t(fEventCut);}\r
  Bool_t  UseLocalTrkSelection() const {return TestBit(kUseLocalTrkSelection);}\r
  void    UseTrackPoints(Bool_t use=kTRUE) {SetBit(kTrkPoints, use);}\r
  void    UserCreateOutputObjects();\r
  void    UserExec(Option_t *);\r
  void    Terminate(Option_t* option = "");\r
private:\r
  // rough radial limits for TRD\r
  static const Float_t fgkITS;      // end ITS\r
  static const Float_t fgkTPC;      // end TPC\r
  static const Float_t fgkTRD;      // end TRD\r
\r
  // Track selection\r
  static const Float_t fgkTrkDCAxy; // cm\r
  static const Float_t fgkTrkDCAz;  // cm\r
  static const Int_t   fgkNclTPC;   // N clusters TPC\r
  static const Float_t fgkPt;       // min. pt\r
  static const Float_t fgkEta;      // eta range\r
  static AliTRDReconstructor   *fgReconstructor; // single instance of TRD reconstructor used by all tasks\r
  static AliTRDgeometry        *fgGeo;           // single instance of TRD geometry used by all tasks\r
\r
  AliTRDinfoGen(const AliTRDinfoGen&);\r
  AliTRDinfoGen& operator=(const AliTRDinfoGen&);\r
  TTreeSRedirector* DebugStream();\r
  void              MakeChambers();\r
\r
//  UShort_t          fBF;             //! Bunch Fill of previous event\r
  AliESDEvent      *fESDev;          //! ESD event\r
  AliMCEvent       *fMCev;           //! MC event\r
  // event/track cuts OO - to be used\r
  AliTRDeventCuts  *fEventCut;       // event cut\r
  AliESDtrackCuts  *fTrackCut;       // track cut\r
  AliTRDv0Info     *fV0Cut;          // v0 cut\r
  TString           fOCDB;           // OCDB location\r
  AliTRDtrackInfo  *fTrackInfo;      //! Track info\r
  AliTRDeventInfo  *fEventInfo;	     //! Event info\r
  AliTRDv0Info     *fV0Info;         //! V0 info\r
  TObjArray        *fTracksBarrel;   //! Array of barrel tracks\r
  TObjArray        *fTracksSA;       //! Array of stand alone tracks\r
  TObjArray        *fTracksKink;     //! Array of kink tracks\r
  TObjArray        *fV0List;         //! V0 container\r
  TObjArray        *fClusters;       //! Clusters container\r
  TObjArray        *fContainer;      //! container to store results\r
  TObjArray        *fRecos;          //! array of reco params\r
  TTreeSRedirector *fDebugStream;    //! debug stream\r
\r
  ClassDef(AliTRDinfoGen, 7)         // entry to TRD analysis train\r
};\r
#endif\r