AliHighPtReconstructionEfficiency.h

Go to the documentation of this file.

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

#include "AliAnalysisTaskSE.h"

#include <map>
#include <vector>
#include <fastjet/PseudoJet.hh>
#include "AliParticleMap.h"

class AliESDtrackCuts;
class AliVParticle;
class AliVTrack;
class TList;
class TNtuple;

namespace HighPtTracks {

class AliReducedJetEvent;
class AliReducedJetInfo;

class AliHighPtReconstructionEfficiency : public AliAnalysisTaskSE {
public:
  enum CutType_t{
    kRJStandardCuts = 0,      
    kRJHybridCuts = 1         
  };
  AliHighPtReconstructionEfficiency();
  AliHighPtReconstructionEfficiency(const char *name);
  virtual ~AliHighPtReconstructionEfficiency();

  virtual void UserCreateOutputObjects();
  virtual bool UserNotify();
  virtual void UserExec(Option_t* /*option*/);
  void Terminate(Option_t* /*option*/) {}

  void SetMaxEtaParticles(double maxeta) { fMaxEtaParticles = maxeta; }

  void SetMaxEtaJets(double maxeta) { fMaxEtaJets = maxeta; }

  void SetMinPtTracks(double minpt) { fMinPtParticles = minpt; }

  void SetMaxDR(double maxdr) { fMaxDR = maxdr; }

  void SetStandardTrackCuts(AliESDtrackCuts *const cuts) { SetTrackCuts(cuts, kRJStandardCuts); }

  void SetHybridTrackCuts(AliESDtrackCuts *const cuts) { SetTrackCuts(cuts, kRJHybridCuts); }

  void SetTaskDebugMode() { fTaskDebugMode = true; }

protected:
  void SetTrackCuts(AliESDtrackCuts *const cuts, CutType_t cuttype) { fTrackCuts[cuttype] = cuts; }
  bool IsSelected(const AliVTrack * const track, CutType_t type) const;
  bool IsTrueSelected(const AliVParticle *const track) const;
  void SelectParticlesForJetfinding(TList &particles) const;
  void CreateRectrackLookup();
  std::vector<AliReconstructedParticlePair> SelectParticles() const;
  double GetDR(const fastjet::PseudoJet &recjet, const AliVParticle *const inputtrack) const;
  const AliParticleList * FindReconstructedParticleFast(int label) const;
  void ProcessJet(AliReducedJetInfo *const jet, const std::vector<AliReconstructedParticlePair> &particles) const;
  void ConvertConstituents(AliReducedJetInfo * const recjet, const fastjet::PseudoJet &inputjet);
  bool IsPhysicalPrimary(const AliVParticle *const part) const;
  bool PythiaInfoFromFile(const char* currFile, double &fXsec, double &fTrials, int &pthard) const ;
  unsigned short GetNumberOfTPCTrackReferences(AliVParticle *const trk) const;

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

  AliESDtrackCuts         *fTrackCuts[2];       
  AliParticleMap          *fParticleMap;        

  // Output objects
  TTree                     *fJetTree;          
  AliReducedJetEvent        *fJetEvent;         

  double              fMaxEtaJets;              
  double              fMaxEtaParticles;         
  double              fMinPtParticles;          
  double              fMaxDR;                   

  double              fCrossSection;            
  double              fNtrials;                 
  int               fPtHardBin;                 

  bool              fTaskDebugMode;             

  ClassDef(AliHighPtReconstructionEfficiency, 1);
};

}
#endif /* ALIHIGHPTRECONSTRUCTIONEFFICIENCY_H */