PWGHF/hfe/AliHFEdebugTreeTaskAOD.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 // Debug tree to look at the distribution of the variable we are cutting on
  17 //
  18 //
  19 #ifndef ALIHFEDEBUGTREETASKAOD_H
  20 #define ALIHFEDEBUGTREETASKAOD_H
  21
  22 #include "AliAnalysisTaskSE.h"
  23
  24 class AliHFEcuts;
  25 class TString;
  26 class TTreeSRedirector;
  27 class AliHFEpidTPC;
  28 class AliAODMCHeader;
  29 class TClonesArray;
  30 class AliHFEsignalCuts;
  31 class AliHFEextraCuts;
  32
  33 class AliHFEdebugTreeTaskAOD : public AliAnalysisTaskSE{
  34   public:
  35     AliHFEdebugTreeTaskAOD();
  36     AliHFEdebugTreeTaskAOD(const char *name);
  37     virtual ~AliHFEdebugTreeTaskAOD();
  38
  39     virtual void UserCreateOutputObjects();
  40     virtual void UserExec(Option_t *);
  41     virtual void Terminate(Option_t *);
  42
  43     // Setters for cuts
  44     void SetFileName(const char *filename);
  45     void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
  46     void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
  47     void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
  48     AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }
  49
  50   private:
  51     AliHFEdebugTreeTaskAOD(const AliHFEdebugTreeTaskAOD &);
  52     AliHFEdebugTreeTaskAOD &operator=(const AliHFEdebugTreeTaskAOD &);
  53
  54     AliAODMCHeader *fAODMCHeader;     // ! MC info AOD
  55     TClonesArray *fAODArrayMCInfo;    // ! MC info particle AOD
  56     AliHFEcuts *fTrackCuts;           // Track
  57     AliHFEextraCuts *fExtraCuts;      // HFE IP info
  58     AliHFEsignalCuts *fSignalCuts;    // Signal Cuts
  59     AliHFEpidTPC *fTPCpid;            // TPC PID
  60     Int_t fEventNumber;               // Event Number
  61     Int_t fNclustersTPC;              // Min Number of clusters in TPC
  62     Int_t fNclustersTPCPID;           // Min Number of clusters for TPC PID
  63     Int_t fNclustersITS;              // Min Number of clusters in ITS
  64     TString fFilename;                // file name for the debug tree
  65     TTreeSRedirector *fDebugTree;     // Debug Tree
  66
  67     ClassDef(AliHFEdebugTreeTaskAOD, 1)
  68 };
  69 #endif
  70