PWG3/hfe/AliHFEV0pidMC.h

   1 /**************************************************************************
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3 *                                                                        *
   4 * Author: The ALICE Off-line Project.                                    *
   5 * Contributors are mentioned in the code where appropriate.              *
   6 *                                                                        *
   7 * Permission to use, copy, modify and distribute this software and its   *
   8 * documentation strictly for non-commercial purposes is hereby granted   *
   9 * without fee, provided that the above copyright notice appears in all   *
  10 * copies and that both the copyright notice and this permission notice   *
  11 * appear in the supporting documentation. The authors make no claims     *
  12 * about the suitability of this software for any purpose. It is          *
  13 * provided "as is" without express or implied warranty.                  *
  14 **************************************************************************/
  15 //
  16 // Benchmarking class for V0 finder and PID.
  17 // Relies on MC information
  18 // For more see source file
  19 //
  20
  21 #ifndef ALIHFEV0PIDMC_H
  22 #define ALIHFEV0PIDMC_H
  23
  24 #ifndef ROOT_TObject
  25 #include <TObject.h>
  26 #endif
  27
  28 class TList;
  29
  30 class AliMCEvent;
  31
  32 class AliHFEV0cuts;
  33 class AliHFEcollection;
  34
  35 class AliHFEV0pidMC : public TObject {
  36
  37  public:
  38   AliHFEV0pidMC();
  39   ~AliHFEV0pidMC();
  40
  41   void    Init();
  42   Bool_t  Process(TObjArray * const array, Int_t type);
  43
  44   void     SetMCEvent(AliMCEvent * const mc) { fMC = mc; };
  45
  46   inline TList* GetListOfQAhistograms();
  47
  48  private:
  49   AliHFEV0pidMC(const AliHFEV0pidMC &);
  50   AliHFEV0pidMC &operator=(const AliHFEV0pidMC &);
  51   Int_t PDGtoPIDdaughter(Int_t pdg) const;    // convert the PDG code to local PID
  52   Int_t PDGtoPIDmother(Int_t pdg) const;      // convert the PDG code to local PID
  53   Int_t PDGtoAliPID(Int_t pdg) const;         // convert PDG to AliPID
  54
  55   AliMCEvent*         fMC;      // MC event
  56   AliHFEcollection*   fColl;    // Histogram collection
  57   AliHFEV0cuts*       fV0cuts;  // V0 cut class
  58
  59    ClassDef(AliHFEV0pidMC, 1)   // QA class for V0 PID
  60 };
  61 //____________________________________________________________
  62 TList *AliHFEV0pidMC::GetListOfQAhistograms(){
  63   //
  64   // Get QA histograms
  65   //
  66   if(fColl)
  67     return fColl->GetList();
  68   return NULL;
  69 }
  70 #endif