AliReducedHighPtEventCreator.h

Go to the documentation of this file.

#ifndef ALIREDUCEDHIGHPTEVENTCREATOR_H
#define ALIREDUCEDHIGHPTEVENTCREATOR_H
/* Copyright(c) 1998-2015, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include "AliAnalysisTaskEmcal.h"
#include <TString.h>

class TArrayI;
class TTree;
class TObjArray;

class AliVCluster;
class AliVEvent;
class AliVParticle;

namespace EMCalTriggerPtAnalysis{
class AliEMCalPtTaskVTrackSelection;
}

namespace HighPtTracks {

class AliReducedHighPtEvent;
class AliReducedPatchContainer;

class AliReducedTrackSelectionContainer : public TObject{
public:
  AliReducedTrackSelectionContainer();
  AliReducedTrackSelectionContainer(Int_t index, EMCalTriggerPtAnalysis::AliEMCalPtTaskVTrackSelection * sel);
  virtual ~AliReducedTrackSelectionContainer();

  void SetIndex(Int_t index) { fIndex = index; }
  void SetTrackSelection(EMCalTriggerPtAnalysis::AliEMCalPtTaskVTrackSelection *trackSelection) { fTrackSelection = trackSelection;}

  Int_t GetIndex() const { return fIndex; }
  EMCalTriggerPtAnalysis::AliEMCalPtTaskVTrackSelection *GetTrackSelection() const { return fTrackSelection; }

protected:
  Int_t                                                           fIndex;               
  EMCalTriggerPtAnalysis::AliEMCalPtTaskVTrackSelection           *fTrackSelection;     

private:
  AliReducedTrackSelectionContainer(AliReducedTrackSelectionContainer &);
  AliReducedTrackSelectionContainer &operator=(AliReducedTrackSelectionContainer &);

  ClassDef(AliReducedTrackSelectionContainer, 1);
};

class AliReducedHighPtEventCreator: public AliAnalysisTaskEmcal {
public:
  AliReducedHighPtEventCreator();
  AliReducedHighPtEventCreator(const char *name);
  virtual ~AliReducedHighPtEventCreator();

  virtual void UserCreateOutputObjects();
  virtual Bool_t Run();

  void AddVirtualTrackSelection(EMCalTriggerPtAnalysis::AliEMCalPtTaskVTrackSelection * sel, Int_t index);
  void SetSwapTriggerThresholds(Bool_t doswap) { fSwapTriggerThresholds = doswap; }
  void SetClusterEnergyCut(Double_t minE, Double_t maxE) {
    fMinClusterE = minE;
    fMaxClusterE = maxE;
  }
  void SetPtRange(Double_t minpt, Double_t maxpt){
    fMinPt = minpt;
    fMaxPt = maxpt;
  }
  void SetEtaRange(Double_t mineta, Double_t maxeta){
    fMinEta = mineta;
    fMaxEta = maxeta;
  }

  void SetCentralityMethod(const char *centmethod){
    fCentralityMethod = centmethod;
  }

protected:
  Bool_t SelectEvent(AliVEvent *event) const;
  Bool_t SelectCluster(const AliVCluster *clust) const;
  Int_t  SelectTrack(AliVTrack *track, TArrayI &cutindices) const;
  void ConvertTriggerPatches(TClonesArray *patches, AliReducedPatchContainer *cont);
  void FixTrackInputEvent(AliVTrack *trk);
  Int_t GetTPCCrossedRows(const AliVTrack *trk) const;
  void GetCellEnergies(AliVCluster *emccluster, TArrayD &energies) const;
  TTree                   *fOutputTree;                   
  AliReducedHighPtEvent   *fOutputEvent;                  
  TObjArray               *fTrackSelections;              

  Bool_t                    fSwapTriggerThresholds;       
  Double_t                  fMinClusterE;                 
  Double_t                  fMaxClusterE;                 
  Double_t                  fMinPt;                       
  Double_t                  fMaxPt;                       
  Double_t                  fMinEta;                      
  Double_t                  fMaxEta;                      
  TString                   fCentralityMethod;            

private:
  AliReducedHighPtEventCreator(const AliReducedHighPtEventCreator &);
  AliReducedHighPtEventCreator &operator=(const AliReducedHighPtEventCreator &);

  ClassDef(AliReducedHighPtEventCreator, 1)
};

} /* namespace HighPtTracks */

#endif /* ALIREDUCEDHIGHPTEVENTCREATOR_H */