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 "AliVEvent.h"
#include <TString.h>

class TArrayI;
class TTree;
class TObjArray;

class AliEmcalTrackSelection;
class AliVCluster;
class AliVEvent;
class AliVParticle;

namespace HighPtTracks {

class AliReducedHighPtEvent;
class AliReducedPatchContainer;

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

  void SetIndex(Int_t index) { fIndex = index; }
  void SetTrackSelection(AliEmcalTrackSelection *trackSelection) { fTrackSelection = trackSelection;}

  Int_t GetIndex() const { return fIndex; }
  AliEmcalTrackSelection *GetTrackSelection() const { return fTrackSelection; }

protected:
  Int_t                                                           fIndex;               
  AliEmcalTrackSelection                                          *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(AliEmcalTrackSelection * 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;
  }
  void SetMinBiasSelection(AliVEvent::EOfflineTriggerTypes minbiasbit) { fMinBiasSelection = minbiasbit; }

  void SetApplyCentralitySelection(Bool_t doApply = kTRUE) { fApplyCentralitySelection = doApply; }

  void SetCentraltityRange(Float_t mincent, Float_t maxcent) {
    fSelectCentralityRange[0] = mincent;
    fSelectCentralityRange[1] = maxcent;
  }
  void SetTriggerSetup(const char *triggersetup){
    fTriggerSetup = triggersetup;
  }
  void SetBasicEventSelection(UInt_t selection) { fEventSelectionBits = selection; }

  void SetFractionOfEventsToKeep(Double_t frac){
    if(frac < 0) fKeepFractionEvents = 0.;
    else if(frac > 1) fKeepFractionEvents = 1.;
    else fKeepFractionEvents = frac;
  }

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;              

  UInt_t                    fEventSelectionBits;          
  Bool_t                    fSwapTriggerThresholds;       
  Double_t                  fMinClusterE;                 
  Double_t                  fMaxClusterE;                 
  Double_t                  fMinPt;                       
  Double_t                  fMaxPt;                       
  Double_t                  fMinEta;                      
  Double_t                  fMaxEta;                      
  Float_t                   fKeepFractionEvents;          
  Bool_t                    fApplyCentralitySelection;    
  TString                   fCentralityMethod;            
  TString                   fTriggerSetup;                
  Float_t                   fSelectCentralityRange[2];    // Range for centraltity selection
  AliVEvent::EOfflineTriggerTypes fMinBiasSelection;      

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

  ClassDef(AliReducedHighPtEventCreator, 1)
};

} /* namespace HighPtTracks */

#endif /* ALIREDUCEDHIGHPTEVENTCREATOR_H */