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 class AliHFEV0taginfo;
  36
  37 class AliHFEreducedEventCreatorAOD : public AliAnalysisTaskSE{
  38   public:
  39     AliHFEreducedEventCreatorAOD();
  40     AliHFEreducedEventCreatorAOD(const char *name);
  41     virtual ~AliHFEreducedEventCreatorAOD();
  42
  43     virtual void UserCreateOutputObjects();
  44     virtual void UserExec(Option_t *);
  45     virtual void Terminate(Option_t *);
  46
  47     // Setters for cuts
  48     void SetMinNclustersTPC(Int_t mincl) { fNclustersTPC = mincl; };
  49     void SetMinNclustersTPCPID(Int_t mincl) { fNclustersTPCPID = mincl; };
  50     void SetMinNclustersITS(Int_t mincl) { fNclustersITS = mincl; };
  51     void SetNbOfTOFSigma(Float_t nbOfTOFSigma) { fNbOfTOFSigma = nbOfTOFSigma; };
  52     void SetFlagPileupEvents() { fFlagPileupEvents = kTRUE; }
  53     void SetSelectSignalOnly(Bool_t select = kTRUE) { fSelectSignalOnly = select; }
  54     AliHFEpidTPC *GetTPCResponse() { return fTPCpid; }
  55
  56     Bool_t IsTOFmismatch(const AliVTrack *const track, const AliPIDResponse *const pid) const;
  57
  58   private:
  59     AliHFEreducedEventCreatorAOD(const AliHFEreducedEventCreatorAOD &);
  60     AliHFEreducedEventCreatorAOD &operator=(const AliHFEreducedEventCreatorAOD &);
  61
  62     AliAODMCHeader *fAODMCHeader;     // ! MC info AOD
  63     TClonesArray *fAODArrayMCInfo;    // ! MC info particle AOD
  64     TTree *fHFEtree;                  // HFE tree
  65     AliAnalysisUtils *fAnalysisUtils; // Analysis Utils
  66     AliHFEreducedEvent *fHFEevent;    // hfe event
  67     AliHFEcuts *fTrackCuts;           // Track
  68     AliHFEextraCuts *fExtraCuts;      // HFE IP info
  69     AliHFEsignalCuts *fSignalCuts;    // Signal Cuts
  70     AliHFEpidTPC *fTPCpid;            // TPC PID
  71     AliHFEV0taginfo *fV0Tagger;       // Tags v0 tracks per Event
  72     Int_t fEventNumber;               // Event Number
  73     Int_t fNclustersTPC;              // Min Number of clusters in TPC
  74     Int_t fNclustersTPCPID;           // Min Number of clusters for TPC PID
  75     Int_t fNclustersITS;              // Min Number of clusters in ITS
  76     Float_t fNbOfTOFSigma;            // Nb of TOF Sigma
  77     Bool_t fRemoveFirstEvent;         // Remove first event from chunk
  78     Bool_t fFlagPileupEvents;         // Flag pileup events
  79     Bool_t fSelectSignalOnly;         // Select signal-only tracks
  80
  81     ClassDef(AliHFEreducedEventCreatorAOD, 2)
  82 };
  83 #endif
  84
  85