[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetFastSimulation.h

#ifndef ALIJETFASTSIMULATION_H
#define ALIJETFASTSIMULATION_H

// $Id$

class TClonesArray;
class TRandom3;
class AliVParticle;
class AliPicoTrack;

#include "AliAnalysisTaskEmcal.h"

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

  virtual void           LocalInit();
  virtual void           UserCreateOutputObjects();

  void                   SetTracksOutName(const char *n)          { fTracksOutName   = n;    }
  void                   SetNTrackClasses(Int_t i)                { fNTrackClasses   = i;    }
  void                   SetFixedTrackEfficiency(Double_t eff)    { fEfficiencyFixed = eff ; }

  void                   SetUseTrResolutionFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Resolution/PtResol_LHCh_Cent0-10_v1.root") {fUseTrPtResolutionFromOADB = b; fPathTrPtResolution=path;}
  void                   SetUseTrEfficiencyFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Efficiency/Efficiency_LHC11a2aj_Cent0_v1.root") {fUseTrEfficiencyFromOADB = b; fPathTrEfficiency=path;}
  void                   SetSmearResolution(Bool_t b)                               { fUseTrPtResolutionSmearing = b ;}
  void                   SetDiceEfficiency(Int_t b)                                 { fUseDiceEfficiency         = b ;}
  void                   SetDiceEfficiencyMinPt(Double_t pt)                        { fDiceEfficiencyMinPt       = pt;}

 protected:
  void                   ExecOnce();
  Bool_t                 Run();

  void                   SimulateTracks();
  Bool_t                 DiceEfficiency(AliPicoTrack *vp, Double_t eff[3], Double_t rnd);
  AliPicoTrack          *SmearPt(AliPicoTrack *vp, Double_t eff[3], Double_t rnd);
  Double_t               GetMomentumSmearing(Int_t cat, Double_t pt);
  void                   FitMomentumResolution();
  void                   LoadTrEfficiencyRootFileFromOADB();
  void                   LoadTrPtResolutionRootFileFromOADB();
  void                   SetMomentumResolutionHybrid(TProfile *p1, TProfile *p2, TProfile *p3);
  void                   SetEfficiencyHybrid(TH1 *h1, TH1 *h2, TH1 *h3);

  TString                fTracksOutName;       // name of output track collection
  TClonesArray          *fTracksOut;           //!output track collection
  Int_t                  fNTrackClasses;       // number of track classes
  TRandom3 *fRandom;                           //! random number generator
  Double_t  fEfficiencyFixed;                  // fixed efficiency for all pT and all types of tracks
  TProfile *fMomResH1;                         // Momentum resolution from TrackQA Hybrid Category 1
  TProfile *fMomResH2;                         // Momentum resolution from TrackQA Hybrid Category 2
  TProfile *fMomResH3;                         // Momentum resolution from TrackQA Hybrid Category 3
  TF1      *fMomResH1Fit;                      // fit to momentum resolution
  TF1      *fMomResH2Fit;                      // fit to momentum resolution
  TF1      *fMomResH3Fit;                      // fit to momentum resolution
  TH1      *fhEffH1;                           // Efficiency for Spectra Hybrid Category 1
  TH1      *fhEffH2;                           // Efficiency for Spectra Hybrid Category 2
  TH1      *fhEffH3;                           // Efficiency for Spectra Hybrid Category 3
  Bool_t    fUseTrPtResolutionSmearing;        // Apply momentum smearing on track level
  Int_t     fUseDiceEfficiency;                // Flag to apply efficiency on track level by dicing 0: no dicing; 1: dicing wrt to input branch;
  Double_t  fDiceEfficiencyMinPt;              // Only do efficiency dicing for tracks above this pt
  Bool_t    fUseTrPtResolutionFromOADB;        // Load track pt resolution root file from OADB path
  Bool_t    fUseTrEfficiencyFromOADB;          // Load tracking efficiency root file from OADB path
  TString   fPathTrPtResolution;               // OADB path to root file
  TString   fPathTrEfficiency;                 // OADB path to root file

  //Output objects
  TH1F     *fHistPtDet;                        //!pT spectrum of detector level particles
  TH2F     *fh2PtGenPtSmeared;                 //! Control histo smeared momentum
  TProfile *fp1Efficiency;                     //! Control profile efficiency
  TProfile *fp1PtResolution;                   //! Control profile for pT resolution

  
 private:
  AliJetFastSimulation(const AliJetFastSimulation&);            // not implemented
  AliJetFastSimulation &operator=(const AliJetFastSimulation&); // not implemented

  ClassDef(AliJetFastSimulation, 1) // Jet fast simulation task
};
#endif
Commit	Line	Data
b3821b05	1	#ifndef ALIJETFASTSIMULATION_H
	2	#define ALIJETFASTSIMULATION_H
	3
	4	// $Id$
	5
	6	class TClonesArray;
	7	class TRandom3;
	8	class AliVParticle;
	9	class AliPicoTrack;
	10
	11	#include "AliAnalysisTaskEmcal.h"
	12
	13	class AliJetFastSimulation : public AliAnalysisTaskEmcal {
	14	public:
	15	AliJetFastSimulation();
	16	AliJetFastSimulation(const char *name);
	17	virtual ~AliJetFastSimulation();
	18
	19	virtual void LocalInit();
	20	virtual void UserCreateOutputObjects();
	21
	22	void SetTracksOutName(const char *n) { fTracksOutName = n; }
	23	void SetNTrackClasses(Int_t i) { fNTrackClasses = i; }
	24	void SetFixedTrackEfficiency(Double_t eff) { fEfficiencyFixed = eff ; }
	25
	26	void SetUseTrResolutionFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Resolution/PtResol_LHCh_Cent0-10_v1.root") {fUseTrPtResolutionFromOADB = b; fPathTrPtResolution=path;}
	27	void SetUseTrEfficiencyFromOADB(Bool_t b=kTRUE, TString path="$ALICE_ROOT/OADB/PWGJE/Efficiency/Efficiency_LHC11a2aj_Cent0_v1.root") {fUseTrEfficiencyFromOADB = b; fPathTrEfficiency=path;}
	28	void SetSmearResolution(Bool_t b) { fUseTrPtResolutionSmearing = b ;}
	29	void SetDiceEfficiency(Int_t b) { fUseDiceEfficiency = b ;}
	30	void SetDiceEfficiencyMinPt(Double_t pt) { fDiceEfficiencyMinPt = pt;}
	31
	32	protected:
	33	void ExecOnce();
	34	Bool_t Run();
	35
	36	void SimulateTracks();
	37	Bool_t DiceEfficiency(AliPicoTrack *vp, Double_t eff[3], Double_t rnd);
	38	AliPicoTrack SmearPt(AliPicoTrack vp, Double_t eff[3], Double_t rnd);
	39	Double_t GetMomentumSmearing(Int_t cat, Double_t pt);
	40	void FitMomentumResolution();
	41	void LoadTrEfficiencyRootFileFromOADB();
	42	void LoadTrPtResolutionRootFileFromOADB();
	43	void SetMomentumResolutionHybrid(TProfile p1, TProfile p2, TProfile *p3);
	44	void SetEfficiencyHybrid(TH1 h1, TH1 h2, TH1 *h3);
	45
	46	TString fTracksOutName; // name of output track collection
	47	TClonesArray *fTracksOut; //!output track collection
	48	Int_t fNTrackClasses; // number of track classes
	49	TRandom3 *fRandom; //! random number generator
	50	Double_t fEfficiencyFixed; // fixed efficiency for all pT and all types of tracks
	51	TProfile *fMomResH1; // Momentum resolution from TrackQA Hybrid Category 1
	52	TProfile *fMomResH2; // Momentum resolution from TrackQA Hybrid Category 2
	53	TProfile *fMomResH3; // Momentum resolution from TrackQA Hybrid Category 3
	54	TF1 *fMomResH1Fit; // fit to momentum resolution
	55	TF1 *fMomResH2Fit; // fit to momentum resolution
	56	TF1 *fMomResH3Fit; // fit to momentum resolution
	57	TH1 *fhEffH1; // Efficiency for Spectra Hybrid Category 1
	58	TH1 *fhEffH2; // Efficiency for Spectra Hybrid Category 2
	59	TH1 *fhEffH3; // Efficiency for Spectra Hybrid Category 3
	60	Bool_t fUseTrPtResolutionSmearing; // Apply momentum smearing on track level
	61	Int_t fUseDiceEfficiency; // Flag to apply efficiency on track level by dicing 0: no dicing; 1: dicing wrt to input branch;
	62	Double_t fDiceEfficiencyMinPt; // Only do efficiency dicing for tracks above this pt
	63	Bool_t fUseTrPtResolutionFromOADB; // Load track pt resolution root file from OADB path
	64	Bool_t fUseTrEfficiencyFromOADB; // Load tracking efficiency root file from OADB path
65	TString fPathTrPtResolution; // OADB path to root file
66	TString fPathTrEfficiency; // OADB path to root file
67
68	//Output objects
69	TH1F *fHistPtDet; //!pT spectrum of detector level particles
70	TH2F *fh2PtGenPtSmeared; //! Control histo smeared momentum
71	TProfile *fp1Efficiency; //! Control profile efficiency
72	TProfile *fp1PtResolution; //! Control profile for pT resolution
73
74
75	private:
76	AliJetFastSimulation(const AliJetFastSimulation&); // not implemented
77	AliJetFastSimulation &operator=(const AliJetFastSimulation&); // not implemented
78
79	ClassDef(AliJetFastSimulation, 1) // Jet fast simulation task
80	};
81	#endif