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 #ifndef ALIHFECOLLECTION_H
  29 #include "AliHFEcollection.h"
  30 #endif
  31
  32 class TList;
  33
  34 class AliMCEvent;
  35
  36
  37 class AliHFEV0pidMC : public TObject {
  38
  39  public:
  40   AliHFEV0pidMC();
  41   ~AliHFEV0pidMC();
  42
  43   void    Init();
  44   Bool_t  Process(TObjArray * const array, Int_t type);
  45
  46   void     SetMCEvent(AliMCEvent * const mc) { fMC = mc; };
  47
  48   inline TList* GetListOfQAhistograms();
  49
  50  private:
  51   AliHFEV0pidMC(const AliHFEV0pidMC &);
  52   AliHFEV0pidMC &operator=(const AliHFEV0pidMC &);
  53   Int_t PDGtoPIDdaughter(Int_t pdg) const;    // convert the PDG code to local PID
  54   Int_t PDGtoPIDmother(Int_t pdg) const;      // convert the PDG code to local PID
  55
  56   AliMCEvent*         fMC;      // MC event
  57   AliHFEcollection*   fColl;    // Histogram collection
  58    ClassDef(AliHFEV0pidMC, 1)   // QA class for V0 PID
  59 };
  60 //____________________________________________________________
  61 TList *AliHFEV0pidMC::GetListOfQAhistograms(){
  62   //
  63   // Get QA histograms
  64   //
  65   if(fColl)
  66     return fColl->GetList();
  67   return NULL;
  68 }
  69 #endif