PWG2/RESONANCES/AliRsnComparisonAT.h

   1 //
   2 // Class AliRsnComparisonAT
   3 //
   4 // TODO
   5 //
   6 // authors: Martin Vala (martin.vala@cern.ch)
   7 //          Alberto Pulvirenti (alberto.pulvirenti@ct.infn.it)
   8 //
   9 #ifndef ALIRSNMVCOMPARISONAT_H
  10 #define ALIRSNMVCOMPARISONAT_H
  11
  12 #include "TObjArray.h"
  13
  14 #include "AliRsnDaughter.h"
  15 #include "AliRsnEvent.h"
  16 #include "AliRsnBaseAT.h"
  17 #include "AliRsnCut.h"
  18 #include "AliRsnCutSet.h"
  19
  20 class AliRsnComparisonAT : public AliRsnBaseAT
  21 {
  22   public:
  23
  24     enum
  25     {
  26       kMyInputNum=1,
  27       kMyPIDInputNum=5
  28     };
  29
  30     AliRsnComparisonAT(const char*name="AliRsnComparisonAT");
  31
  32     virtual ~AliRsnComparisonAT() {}
  33
  34     virtual void   InitIOVars();
  35     virtual void   LocalInit();
  36     virtual void   CreateOutputObjects();
  37     virtual void   Exec(Option_t *option);
  38     virtual void   Terminate(Option_t *);
  39
  40     void        LoopOverESDtracks();
  41     void        LoopOverMCtracks();
  42     void        LoopOverRSNDaughters();
  43     void        PrintStat();
  44
  45     void AddMyInput(AliRsnComparisonObj *obj,const Int_t &index=0);
  46     void AddMyPIDInput(AliRsnComparisonObj *obj,const Int_t &index=0);
  47     void SetMyPIDInputName(TString name="default",const Int_t &index=0);
  48
  49     void SetNumberOfPIDInputs(const Int_t& theValue) { fMyPIDInputNum = theValue; }
  50
  51
  52   private:
  53
  54     AliRsnComparisonAT(const AliRsnComparisonAT&)
  55         : AliRsnBaseAT(""),fOutList(0x0),fMyInputNum(1),fMyPIDInputNum(0) {}
  56     AliRsnComparisonAT& operator=(const AliRsnComparisonAT&) {return *this;}
  57
  58     TList       *fOutList;                      // output list
  59
  60     Int_t       fMyInputNum;
  61     TObjArray   fMyInput[kMyInputNum];
  62     Int_t       fMyPIDInputNum;
  63     TObjArray   fMyPIDInput[kMyPIDInputNum];
  64
  65
  66
  67     ClassDef(AliRsnComparisonAT, 1)
  68 };
  69
  70 #endif