PWGHF/hfe/AliHFEreducedEventCreatorAOD.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 ALIHFEREDUCEDEVENTCREATORAOD_H
  20 #define ALIHFEREDUCEDEVENTCREATORAOD_H
  21
  22 #include "AliAnalysisTaskSE.h"
  23
  24 class TClonesArray;
  25 class TString;
  26 class TTree;
  27 class AliAODMCHeader;
  28 class AliAnalysisUtils;
  29 class AliPIDResponse;
  30 class AliHFEcuts;
  31 class AliHFEextraCuts;
  32 class AliHFEpidTPC;
  33 class AliHFEsignalCuts;
  34 class AliHFEreducedEvent;
  35
  36 class AliHFEreducedEventCreatorAOD : public AliAnalysisTaskSE{
  37   public:
  38     AliHFEreducedEventCreatorAOD();
  39     AliHFEreducedEventCreatorAOD(const char *name);
  40     virtual ~AliHFEreducedEventCreatorAOD();
  41
  42     virtual void UserCreateOutputObjects();
  43     virtual void UserExec(Option_t *);
  44     virtual void Terminate(Option_t *);
  45
  46     // Setters for cuts
  47     void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
  48     void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
  49     void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
  50     void SetNbOfTOFSigma(Float_t nbOfTOFSigma) { fNbOfTOFSigma = nbOfTOFSigma; };
  51     AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }
  52
  53     Bool_t IsTOFmismatch(const AliVTrack *const track, const AliPIDResponse *const pid) const;
  54
  55   private:
  56     AliHFEreducedEventCreatorAOD(const AliHFEreducedEventCreatorAOD &);
  57     AliHFEreducedEventCreatorAOD &operator=(const AliHFEreducedEventCreatorAOD &);
  58
  59     AliAODMCHeader *fAODMCHeader;     // ! MC info AOD
  60     TClonesArray *fAODArrayMCInfo;    // ! MC info particle AOD
  61     TTree *fHFEtree;                  // HFE tree
  62     AliAnalysisUtils *fAnalysisUtils; // Analysis Utils
  63     AliHFEreducedEvent *fHFEevent;    // hfe event
  64     AliHFEcuts *fTrackCuts;           // Track
  65     AliHFEextraCuts *fExtraCuts;      // HFE IP info
  66     AliHFEsignalCuts *fSignalCuts;    // Signal Cuts
  67     AliHFEpidTPC *fTPCpid;            // TPC PID
  68     Int_t fEventNumber;               // Event Number
  69     Int_t fNclustersTPC;              // Min Number of clusters in TPC
  70     Int_t fNclustersTPCPID;           // Min Number of clusters for TPC PID
  71     Int_t fNclustersITS;              // Min Number of clusters in ITS
  72     Float_t fNbOfTOFSigma;            // Nb of TOF Sigma
  73     Bool_t fRemoveFirstEvent;         // Remove first event from chunk
  74
  75     ClassDef(AliHFEreducedEventCreatorAOD, 2)
  76 };
  77 #endif
  78
  79