[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskChargedJetsPA.h

#ifndef ALIANALYSISTASKCHARGEDJETSPA_H
#define ALIANALYSISTASKCHARGEDJETSPA_H

//#define DEBUGMODE

class TList;
class TClonesArray;
class TString;
class AliEmcalJet;
class AliRhoParameter;
class AliVParticle;
class AliLog;
class AliAnalysisUtils;
class TRandom3;
class AliESDtrack;

#include "AliESDtrackCuts.h"
#include "THn.h"

const int kMaxMatch=5;
const double kMaxChi2 = 200;

// Class for the hybrid track cuts
class AliESDHybridTrackcuts {
 public:
  AliESDHybridTrackcuts() : fMainCuts(0), fAdditionalCuts(0) {};
  AliESDHybridTrackcuts(const AliESDHybridTrackcuts& pd) : fMainCuts(0), fAdditionalCuts(0) {((AliESDHybridTrackcuts &) pd).Copy(*this);};
  virtual ~AliESDHybridTrackcuts() {delete fAdditionalCuts; delete fMainCuts;};

  virtual void Copy(AliESDHybridTrackcuts &c) const
  {
    AliESDHybridTrackcuts& target = (AliESDHybridTrackcuts &) c;
    target.fAdditionalCuts = fAdditionalCuts;
    target.fMainCuts = fMainCuts;
  };

  // Meta function to accept hybrid tracks
  Int_t AcceptTrack (const AliESDtrack* esdTrack) {
    if (AcceptNormalTrack(esdTrack))
      return 1;
    else if (AcceptAdditionalTrack(esdTrack))
      return 2;
    else
      return 0;
  };

  Bool_t  AcceptNormalTrack (const AliESDtrack* esdTrack) {if(fMainCuts) return fMainCuts->AcceptTrack(esdTrack); else return kFALSE;};
  Bool_t  AcceptAdditionalTrack (const AliESDtrack* esdTrack) {if(fAdditionalCuts) return fAdditionalCuts->AcceptTrack(esdTrack); else return kFALSE;};
  void    SetMainCuts (AliESDtrackCuts* trackCuts) {fMainCuts = trackCuts;};
  void    SetAdditionalCuts (AliESDtrackCuts* trackCuts) {fAdditionalCuts = trackCuts;};
  AliESDtrackCuts* GetAdditionalCuts () {return fAdditionalCuts;};
  AliESDtrackCuts* GetMainCuts () {return fMainCuts;};

  AliESDHybridTrackcuts &operator=(const AliESDHybridTrackcuts &c) {if (this != &c) ((AliESDHybridTrackcuts &) c).Copy(*this); return *this;};
 protected:
  AliESDtrackCuts*  fMainCuts;          // trackcut object for the global tracks
  AliESDtrackCuts*  fAdditionalCuts;    // trackcut object for the complementary tracks

// ClassDef(AliESDHybridTrackcuts, 2); 
};

class AliAnalysisTaskChargedJetsPA : public AliAnalysisTaskSE {
 public:
  // ######### CONTRUCTORS/DESTRUCTORS AND STD FUNCTIONS
  AliAnalysisTaskChargedJetsPA() : AliAnalysisTaskSE(), fOutputLists(), fCurrentOutputList(0), fDoJetAnalysis(1), fAnalyzeJetProfile(0), fAnalyzeTrackcuts(0), fParticleLevel(0), fUseDefaultVertexCut(1), fUsePileUpCut(1), fSetCentralityToOne(0), fNoExternalBackground(0), fBackgroundForJetProfile(0), fPartialAnalysisNParts(1), fPartialAnalysisIndex(0), fJetArray(0), fTrackArray(0), fBackgroundJetArray(0), fJetArrayName(), fTrackArrayName(), fBackgroundJetArrayName(), fRhoTaskName(), fRandConeRadius(0.4), fSignalJetRadius(0.4), fBackgroundJetRadius(0.4), fNumberExcludedJets(-1), fMinEta(-0.9), fMaxEta(0.9), fMinJetEta(-0.5), fMaxJetEta(0.5), fMinTrackPt(0.150), fMinJetPt(5.0), fMinJetArea(0.5), fMinBackgroundJetPt(0.0), fMinNCrossedRows(70), fUsePtDepCrossedRowsCut(0), fNumberOfCentralityBins(20), fCentralityType("V0A"), fMatchTr(), fMatchChi(), fPrimaryVertex(0), fFirstLeadingJet(0), fSecondLeadingJet(0), fFirstLeadingKTJet(0), fSecondLeadingKTJet(0), fNumberSignalJets(0), fNumberSignalJetsAbove5GeV(0), fRandom(0), fHelperClass(0), fInitialized(0), fTaskInstanceCounter(0), fIsDEBUG(0), fIsPA(1), fNoTerminate(1), fEventCounter(0), fHybridESDtrackCuts(0), fHybridESDtrackCuts_variedPtDep(0), fHybridESDtrackCuts_variedPtDep2(0)
  {
  // dummy
  }

  AliAnalysisTaskChargedJetsPA(const char *name, const char* trackArrayName, const char* jetArrayName, const char* backgroundJetArrayName, Bool_t analyzeJetProfile, Bool_t analyzeTrackcuts);
  virtual ~AliAnalysisTaskChargedJetsPA();
  virtual void     UserCreateOutputObjects();
  virtual void     UserExec(Option_t *option);
  virtual Bool_t   UserNotify();
  virtual void     Terminate(Option_t *);

  // ######### SETTERS/GETTERS
  void        SetDoJetAnalysis(Bool_t val) {fDoJetAnalysis = val;}
  void        SetAnalyzeJetProfile(Bool_t val) {fAnalyzeJetProfile = val;}
  void        SetAnalyzeTrackcuts(Bool_t val) {fAnalyzeTrackcuts = val;}
  void        SetAnalyzePartialEvents(Int_t nParts, Int_t index) {fPartialAnalysisNParts = nParts; fPartialAnalysisIndex = index;}
  void        SetUseDefaultVertexCut (Bool_t val) {fUseDefaultVertexCut = val;}
  void        SetUsePileUpCut (Bool_t val) {fUsePileUpCut = val;}
  void        SetNoTerminate (Bool_t val) {fNoTerminate = val;}
  void        SetIsPA (Bool_t val) {fIsPA = val;}
  void        SetCentralityToOne (Bool_t val) {fSetCentralityToOne = val;}
  void        SetNoExternalBackground (Bool_t val) {fNoExternalBackground = val;}
  void        SetBackgroundForJetProfile (Bool_t val) {fBackgroundForJetProfile = val;}
  void        SetMinNCrossedRows(Int_t val) {fMinNCrossedRows = val;}
  void        SetUsePtDepCrossedRowsCut(Bool_t val) {fUsePtDepCrossedRowsCut = val;}

  void        SetNumberOfCentralityBins(Int_t val) {fNumberOfCentralityBins = val;} 
  void        SetTrackMinPt(Double_t minPt) {fMinJetPt = minPt;}
  void        SetSignalJetMinPt(Double_t minPt) {fMinJetPt = minPt;}
  void        SetSignalJetMinArea(Double_t minArea) {fMinJetArea = minArea;}
  void        SetBackgroundJetMinPt(Double_t minPt) {fMinBackgroundJetPt = minPt;}
  void        SetRandConeRadius(Double_t radius) {fRandConeRadius = radius;}
  void        SetSignalJetRadius(Double_t radius) {fSignalJetRadius = radius;}
  void        SetBackgroundJetRadius(Double_t radius) {fBackgroundJetRadius = radius;}
  void        SetCentralityType(const char* type) {fCentralityType = type;}
  void        SetExternalRhoTaskName(const char* name) {fRhoTaskName = name;}
  void        SetAcceptanceEta(Double_t minEta, Double_t maxEta) {fMinEta = minEta; fMaxEta = maxEta;}
  void        SetAcceptanceJetEta(Double_t minEta, Double_t maxEta) {fMinJetEta = minEta; fMaxJetEta = maxEta;}
  Int_t       GetInstanceCounter() {return fTaskInstanceCounter;}
  void        SetCurrentOutputList(Int_t i)
  {
    if(i==0)
    {
      fCurrentOutputList = fOutputLists[0];
    }
    else if(i==1)
    {
      if(fAnalyzeJetProfile)
        fCurrentOutputList = fOutputLists[1];
      else
        AliError("Non-existing output list demanded!");
    }
    else if(i==2)
    {
      if(!fAnalyzeJetProfile && fAnalyzeTrackcuts)
        fCurrentOutputList = fOutputLists[1];
      else if(fAnalyzeJetProfile && fAnalyzeTrackcuts)
        fCurrentOutputList = fOutputLists[2];
      else
        AliError("Non-existing output list demanded!");
    }
  }

 private:

  // ######### MAIN CALCULATION FUNCTIONS
  void        InitializeTrackcuts();
  void        GetLeadingJets();
  Double_t    GetCorrectedJetPt(AliEmcalJet* jet, Double_t background);
  Double_t    GetDeltaPt(Double_t rho, Double_t leadingJetExclusionProbability = 0);

  void        GetKTBackgroundDensityAll(Int_t numberExcludeLeadingJets, Double_t& rhoPbPb, Double_t& rhoPbPbWithGhosts, Double_t& rhoCMS, Double_t& rhoImprovedCMS, Double_t& rhoMean, Double_t& rhoTrackLike);
  void        GetTRBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoNoExclusion, Double_t& rhoConeExclusion02, Double_t& rhoConeExclusion04, Double_t& rhoConeExclusion06, Double_t& rhoConeExclusion08, Double_t& rhoExactExclusion);
  void        GetPPBackgroundDensity(Double_t& background);

  Double_t    GetConePt(Double_t eta, Double_t phi, Double_t radius);
  Double_t    GetCorrectedConePt(Double_t eta, Double_t phi, Double_t radius, Double_t background);
  Int_t       GetConeConstituentCount(Double_t eta, Double_t phi, Double_t radius);
  Double_t    GetExternalRho();
  void        CreateJetProfilePlots(Double_t bgrd);
  void        CreateCutHistograms();
  void        CreateITSTPCMatchingHistograms();
  void        GetPerpendicularCone(Double_t vecPhi, Double_t vecTheta, Double_t& conePt);

  // ######### CHECK FUNCTIONS
  Bool_t      IsTrackInAcceptance(AliVParticle* track);
  Bool_t      IsTrackInCone(AliVTrack* track, Double_t eta, Double_t phi, Double_t radius);
  Bool_t      IsTrackInJet(AliEmcalJet* jet, Int_t trackIndex);
  Bool_t      IsJetOverlapping(AliEmcalJet* jet1, AliEmcalJet* jet2);

  Bool_t      IsEventInAcceptance(AliVEvent* event);
  Bool_t      IsBackgroundJetInAcceptance(AliEmcalJet* jet);
  Bool_t      IsSignalJetInAcceptance(AliEmcalJet* jet, Bool_t usePtCut = kFALSE);
  
  // ######### HELPER FUNCTIONS
  Double_t    EtaToTheta(Double_t arg);
  Double_t    ThetaToEta(Double_t arg);
  Double_t    GetDeltaPhi(Double_t phi1, Double_t phi2);
  Double_t    MCGetOverlapCircleRectancle(Double_t cPosX, Double_t cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax);
  Double_t    MCGetOverlapMultipleCirclesRectancle(Int_t numCircles, std::vector<Double_t> cPosX, std::vector<Double_t> cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax);
  void        Match(AliESDtrack* tr0, AliESDtrack* tr1, Int_t& nmatch, Bool_t excludeMom = kFALSE, Double_t rotate=0);

  // ######### HISTOGRAM FUNCTIONS
  void        FillHistogram(const char * key, Double_t x);
  void        FillHistogram(const char * key, Double_t x, Double_t y);
  void        FillHistogram(const char * key, Double_t x, Double_t y, Double_t add);
  void        FillCutHistogram(const char * key, Double_t cut, Double_t pT, Double_t eta, Double_t phi, Int_t isAdditionalTrack);

  const char* GetHistoName(const char* name)
  {
    if (fParticleLevel)    
      return Form("%s_MC", name);
    return Form("%s", name);
  }
  template <class T> T* AddHistogram1D(const char* name = "CustomHistogram", const char* title = "NO_TITLE", const char* options = "", Int_t xBins = 100, Double_t xMin = 0.0, Double_t xMax = 20.0, const char* xTitle = "x axis", const char* yTitle = "y axis");
  template <class T> T* AddHistogram2D(const char* name = "CustomHistogram", const char* title = "NO_TITLE", const char* options = "", Int_t xBins = 100, Double_t xMin = 0.0, Double_t xMax = 20.0, Int_t yBins = 100, Double_t yMin = 0.0, Double_t yMax = 20.0, const char* xTitle = "x axis", const char* yTitle = "y axis", const char* zTitle = "z axis");
  THnF* AddCutHistogram(const char* name, const char* title, const char* cutName, Int_t nBins, Double_t xMin, Double_t xMax);
  void  BinLogAxis(const THn *h, Int_t axisNumber);

  // ######### STANDARD FUNCTIONS
  void      Calculate(AliVEvent* event);
  void      ExecOnce();
  void      Init ();

  std::vector<TList*> fOutputLists;           //! Output lists
  TList*              fCurrentOutputList;     //! Currently selected list where the histograms will be saved to
  // ########## USAGE TRIGGERS 
  Bool_t              fDoJetAnalysis;         // trigger if jets/tracks etc. should be analyzed
  Bool_t              fAnalyzeJetProfile;     // trigger if jet profile should be analyzed
  Bool_t              fAnalyzeTrackcuts;      // trigger if trackcuts should be analyzed
  Bool_t              fParticleLevel;         // trigger if data is kinematics only (for naming reasons)
  Bool_t              fUseDefaultVertexCut;   // trigger if automatic vertex cut from helper class should be done
  Bool_t              fUsePileUpCut;          // trigger if pileup cut should be done
  Bool_t              fSetCentralityToOne;    // trigger if centrality val. should be set to one for every event (failsafe)
  Bool_t              fNoExternalBackground;  // External background is set to 0 (e.g. for PP)
  Int_t               fBackgroundForJetProfile; // Which background will be subtracted for the profile calculation (0=External,1=Improved CMS,2=CMS,3=PP,4=TR,5=None)
  Int_t               fPartialAnalysisNParts; // take only every Nth event
  Int_t               fPartialAnalysisIndex;  // using e.g. only every 5th event, this specifies which one
  

  // ########## SOURCE INFORMATION
  TClonesArray*       fJetArray;              //! object containing the jets
  TClonesArray*       fTrackArray;            //! object containing the tracks
  TClonesArray*       fBackgroundJetArray;    //! object containing background jets
  TString             fJetArrayName;          // name of object containing the jets
  TString             fTrackArrayName;        // name of object containing the tracks
  TString             fBackgroundJetArrayName;// name of object containing event wise bckgrds
  TString             fRhoTaskName;           // name of rho task for this analysis
  // ########## JET/DIJET/RC PROPERTIES
  Double_t            fRandConeRadius;        // Radius for the random cones
  Double_t            fSignalJetRadius;       // Radius for the signal jets
  Double_t            fBackgroundJetRadius;   // Radius for the KT background jets
  Int_t               fNumberExcludedJets;    // Number of jets to be excluded from backgrounds
  // ########## CUTS 
  Double_t            fMinEta;                // min eta of tracks
  Double_t            fMaxEta;                // max eta of tracks
  Double_t            fMinJetEta;             // min eta of jets
  Double_t            fMaxJetEta;             // max eta of jets
  Double_t            fMinTrackPt;            // Min track pt to be accepted
  Double_t            fMinJetPt;              // Min jet pt to be accepted
  Double_t            fMinJetArea;            // Min jet area to be accepted
  Double_t            fMinBackgroundJetPt;    // Min jet pt to be accepted as background jet
  Int_t               fMinNCrossedRows;       // Min number of crossed TPC rows for trackcut analysis
  Bool_t              fUsePtDepCrossedRowsCut;// Trigger if linear pT dep. for crossed rows cut should be applied
  Int_t               fNumberOfCentralityBins;// Number of centrality bins used for histograms
  TString             fCentralityType;        // Used centrality estimate (V0A, V0C, V0M, ...)

  AliESDtrack*        fMatchTr[kMaxMatch];    //! Helper variables track matching
  Double_t            fMatchChi[kMaxMatch];   //! Helper variables track matching


  // ########## EVENT PROPERTIES
  const AliVVertex*   fPrimaryVertex;         //! Vertex found per event
  AliEmcalJet*        fFirstLeadingJet;       //! leading jet in event
  AliEmcalJet*        fSecondLeadingJet;      //! next to leading jet in event
  AliEmcalJet*        fFirstLeadingKTJet;     //! leading kT jet in event
  AliEmcalJet*        fSecondLeadingKTJet;    //! next to leading kT jet in event
  Int_t               fNumberSignalJets;      // Number of signal jets in event
  Int_t               fNumberSignalJetsAbove5GeV; // Number of signal jets in event > 5GeV
  // ########## GENERAL VARS
  TRandom3*           fRandom;                //! A random number
  AliAnalysisUtils*   fHelperClass;           //! Vertex selection helper
  Bool_t              fInitialized;           // trigger if tracks/jets are loaded
  Int_t               fTaskInstanceCounter;   // for naming reasons
  Bool_t              fIsDEBUG;               // Debug trigger
  Bool_t              fIsPA;                  // pPb trigger
  Bool_t              fNoTerminate;           // don't use terminate routines
  ULong_t             fEventCounter;          // Internal event counter
  AliESDHybridTrackcuts* fHybridESDtrackCuts; //! these trackcuts are applied
  AliESDHybridTrackcuts* fHybridESDtrackCuts_variedPtDep; //! these trackcuts are applied
  AliESDHybridTrackcuts* fHybridESDtrackCuts_variedPtDep2; //! these trackcuts are applied

  AliAnalysisTaskChargedJetsPA(const AliAnalysisTaskChargedJetsPA&);
  AliAnalysisTaskChargedJetsPA& operator=(const AliAnalysisTaskChargedJetsPA&);

  ClassDef(AliAnalysisTaskChargedJetsPA, 3); // Charged jet analysis for pA

};
#endif
Commit	Line	Data
8628b70c	1	#ifndef ALIANALYSISTASKCHARGEDJETSPA_H
	2	#define ALIANALYSISTASKCHARGEDJETSPA_H
	3
12eb4ac1	4	//#define DEBUGMODE
8628b70c	5
8628b70c	6	class TList;
8628b70c	7	class TClonesArray;
	8	class TString;
	9	class AliEmcalJet;
	10	class AliRhoParameter;
	11	class AliVParticle;
	12	class AliLog;
3fa9cde0	13	class AliAnalysisUtils;
b95d0245	14	class TRandom3;
013103f5	15	class AliESDtrack;
	16
	17	#include "AliESDtrackCuts.h"
	18	#include "THn.h"
	19
0cf2a402	20	const int kMaxMatch=5;
565f94c4	21	const double kMaxChi2 = 200;
0cf2a402	22
013103f5	23	// Class for the hybrid track cuts
	24	class AliESDHybridTrackcuts {
	25	public:
	26	AliESDHybridTrackcuts() : fMainCuts(0), fAdditionalCuts(0) {};
	27	AliESDHybridTrackcuts(const AliESDHybridTrackcuts& pd) : fMainCuts(0), fAdditionalCuts(0) {((AliESDHybridTrackcuts &) pd).Copy(*this);};
	28	virtual ~AliESDHybridTrackcuts() {delete fAdditionalCuts; delete fMainCuts;};
	29
	30	virtual void Copy(AliESDHybridTrackcuts &c) const
	31	{
	32	AliESDHybridTrackcuts& target = (AliESDHybridTrackcuts &) c;
	33	target.fAdditionalCuts = fAdditionalCuts;
	34	target.fMainCuts = fMainCuts;
	35	};
	36
	37	// Meta function to accept hybrid tracks
	38	Int_t AcceptTrack (const AliESDtrack* esdTrack) {
	39	if (AcceptNormalTrack(esdTrack))
	40	return 1;
	41	else if (AcceptAdditionalTrack(esdTrack))
	42	return 2;
	43	else
	44	return 0;
	45	};
	46
	47	Bool_t AcceptNormalTrack (const AliESDtrack* esdTrack) {if(fMainCuts) return fMainCuts->AcceptTrack(esdTrack); else return kFALSE;};
	48	Bool_t AcceptAdditionalTrack (const AliESDtrack* esdTrack) {if(fAdditionalCuts) return fAdditionalCuts->AcceptTrack(esdTrack); else return kFALSE;};
	49	void SetMainCuts (AliESDtrackCuts* trackCuts) {fMainCuts = trackCuts;};
	50	void SetAdditionalCuts (AliESDtrackCuts* trackCuts) {fAdditionalCuts = trackCuts;};
	51	AliESDtrackCuts* GetAdditionalCuts () {return fAdditionalCuts;};
	52	AliESDtrackCuts* GetMainCuts () {return fMainCuts;};
	53
	54	AliESDHybridTrackcuts &operator=(const AliESDHybridTrackcuts &c) {if (this != &c) ((AliESDHybridTrackcuts &) c).Copy(this); return this;};
	55	protected:
	56	AliESDtrackCuts* fMainCuts; // trackcut object for the global tracks
	57	AliESDtrackCuts* fAdditionalCuts; // trackcut object for the complementary tracks
	58
	59	// ClassDef(AliESDHybridTrackcuts, 2);
	60	};
8628b70c	61
8628b70c	62	class AliAnalysisTaskChargedJetsPA : public AliAnalysisTaskSE {
8628b70c	63	public:
12eb4ac1	64	// ######### CONTRUCTORS/DESTRUCTORS AND STD FUNCTIONS
0cf2a402	65	AliAnalysisTaskChargedJetsPA() : AliAnalysisTaskSE(), fOutputLists(), fCurrentOutputList(0), fDoJetAnalysis(1), fAnalyzeJetProfile(0), fAnalyzeTrackcuts(0), fParticleLevel(0), fUseDefaultVertexCut(1), fUsePileUpCut(1), fSetCentralityToOne(0), fNoExternalBackground(0), fBackgroundForJetProfile(0), fPartialAnalysisNParts(1), fPartialAnalysisIndex(0), fJetArray(0), fTrackArray(0), fBackgroundJetArray(0), fJetArrayName(), fTrackArrayName(), fBackgroundJetArrayName(), fRhoTaskName(), fRandConeRadius(0.4), fSignalJetRadius(0.4), fBackgroundJetRadius(0.4), fNumberExcludedJets(-1), fMinEta(-0.9), fMaxEta(0.9), fMinJetEta(-0.5), fMaxJetEta(0.5), fMinTrackPt(0.150), fMinJetPt(5.0), fMinJetArea(0.5), fMinBackgroundJetPt(0.0), fMinNCrossedRows(70), fUsePtDepCrossedRowsCut(0), fNumberOfCentralityBins(20), fCentralityType("V0A"), fMatchTr(), fMatchChi(), fPrimaryVertex(0), fFirstLeadingJet(0), fSecondLeadingJet(0), fFirstLeadingKTJet(0), fSecondLeadingKTJet(0), fNumberSignalJets(0), fNumberSignalJetsAbove5GeV(0), fRandom(0), fHelperClass(0), fInitialized(0), fTaskInstanceCounter(0), fIsDEBUG(0), fIsPA(1), fNoTerminate(1), fEventCounter(0), fHybridESDtrackCuts(0), fHybridESDtrackCuts_variedPtDep(0), fHybridESDtrackCuts_variedPtDep2(0)
b95d0245	66	{
013103f5	67	// dummy
b95d0245	68	}
0983dd5b	69
013103f5	70	AliAnalysisTaskChargedJetsPA(const char name, const char trackArrayName, const char* jetArrayName, const char* backgroundJetArrayName, Bool_t analyzeJetProfile, Bool_t analyzeTrackcuts);
8628b70c	71	virtual ~AliAnalysisTaskChargedJetsPA();
	72	virtual void UserCreateOutputObjects();
	73	virtual void UserExec(Option_t *option);
392aa819	74	virtual Bool_t UserNotify();
8628b70c	75	virtual void Terminate(Option_t *);
efb9b161	76
12eb4ac1	77	// ######### SETTERS/GETTERS
013103f5	78	void SetDoJetAnalysis(Bool_t val) {fDoJetAnalysis = val;}
2a82f637	79	void SetAnalyzeJetProfile(Bool_t val) {fAnalyzeJetProfile = val;}
013103f5	80	void SetAnalyzeTrackcuts(Bool_t val) {fAnalyzeTrackcuts = val;}
c2b0976c	81	void SetAnalyzePartialEvents(Int_t nParts, Int_t index) {fPartialAnalysisNParts = nParts; fPartialAnalysisIndex = index;}
2a82f637	82	void SetUseDefaultVertexCut (Bool_t val) {fUseDefaultVertexCut = val;}
392aa819	83	void SetUsePileUpCut (Bool_t val) {fUsePileUpCut = val;}
0779c72f	84	void SetNoTerminate (Bool_t val) {fNoTerminate = val;}
013103f5	85	void SetIsPA (Bool_t val) {fIsPA = val;}
9f1e4f53	86	void SetCentralityToOne (Bool_t val) {fSetCentralityToOne = val;}
4bb1bca6	87	void SetNoExternalBackground (Bool_t val) {fNoExternalBackground = val;}
3dff28bc	88	void SetBackgroundForJetProfile (Bool_t val) {fBackgroundForJetProfile = val;}
2d6ed1a3	89	void SetMinNCrossedRows(Int_t val) {fMinNCrossedRows = val;}
e3ed0600	90	void SetUsePtDepCrossedRowsCut(Bool_t val) {fUsePtDepCrossedRowsCut = val;}
e3ed0600	91
fa866fc8	92	void SetNumberOfCentralityBins(Int_t val) {fNumberOfCentralityBins = val;}
12eb4ac1	93	void SetTrackMinPt(Double_t minPt) {fMinJetPt = minPt;}
	94	void SetSignalJetMinPt(Double_t minPt) {fMinJetPt = minPt;}
	95	void SetSignalJetMinArea(Double_t minArea) {fMinJetArea = minArea;}
	96	void SetBackgroundJetMinPt(Double_t minPt) {fMinBackgroundJetPt = minPt;}
12eb4ac1	97	void SetRandConeRadius(Double_t radius) {fRandConeRadius = radius;}
	98	void SetSignalJetRadius(Double_t radius) {fSignalJetRadius = radius;}
	99	void SetBackgroundJetRadius(Double_t radius) {fBackgroundJetRadius = radius;}
12eb4ac1	100	void SetCentralityType(const char* type) {fCentralityType = type;}
06b41974	101	void SetExternalRhoTaskName(const char* name) {fRhoTaskName = name;}
b836a932	102	void SetAcceptanceEta(Double_t minEta, Double_t maxEta) {fMinEta = minEta; fMaxEta = maxEta;}
b836a932	103	void SetAcceptanceJetEta(Double_t minEta, Double_t maxEta) {fMinJetEta = minEta; fMaxJetEta = maxEta;}
12eb4ac1	104	Int_t GetInstanceCounter() {return fTaskInstanceCounter;}
013103f5	105	void SetCurrentOutputList(Int_t i)
	106	{
	107	if(i==0)
	108	{
	109	fCurrentOutputList = fOutputLists[0];
	110	}
	111	else if(i==1)
	112	{
	113	if(fAnalyzeJetProfile)
	114	fCurrentOutputList = fOutputLists[1];
	115	else
	116	AliError("Non-existing output list demanded!");
	117	}
	118	else if(i==2)
	119	{
	120	if(!fAnalyzeJetProfile && fAnalyzeTrackcuts)
	121	fCurrentOutputList = fOutputLists[1];
	122	else if(fAnalyzeJetProfile && fAnalyzeTrackcuts)
	123	fCurrentOutputList = fOutputLists[2];
	124	else
	125	AliError("Non-existing output list demanded!");
	126	}
	127	}
8628b70c	128
8628b70c	129	private:
12eb4ac1	130
12eb4ac1	131	// ######### MAIN CALCULATION FUNCTIONS
013103f5	132	void InitializeTrackcuts();
013103f5	133	void GetLeadingJets();
e8df4140	134	Double_t GetCorrectedJetPt(AliEmcalJet* jet, Double_t background);
c2b0976c	135	Double_t GetDeltaPt(Double_t rho, Double_t leadingJetExclusionProbability = 0);
0cb49d99	136
0cb49d99	137	void GetKTBackgroundDensityAll(Int_t numberExcludeLeadingJets, Double_t& rhoPbPb, Double_t& rhoPbPbWithGhosts, Double_t& rhoCMS, Double_t& rhoImprovedCMS, Double_t& rhoMean, Double_t& rhoTrackLike);
b95d0245	138	void GetTRBackgroundDensity(Int_t numberExcludeLeadingJets, Double_t& rhoNoExclusion, Double_t& rhoConeExclusion02, Double_t& rhoConeExclusion04, Double_t& rhoConeExclusion06, Double_t& rhoConeExclusion08, Double_t& rhoExactExclusion);
3dff28bc	139	void GetPPBackgroundDensity(Double_t& background);
013103f5	140
12eb4ac1	141	Double_t GetConePt(Double_t eta, Double_t phi, Double_t radius);
271cbc5d	142	Double_t GetCorrectedConePt(Double_t eta, Double_t phi, Double_t radius, Double_t background);
a8b7cbf6	143	Int_t GetConeConstituentCount(Double_t eta, Double_t phi, Double_t radius);
06b41974	144	Double_t GetExternalRho();
013103f5	145	void CreateJetProfilePlots(Double_t bgrd);
013103f5	146	void CreateCutHistograms();
0cf2a402	147	void CreateITSTPCMatchingHistograms();
12eb4ac1	148	void GetPerpendicularCone(Double_t vecPhi, Double_t vecTheta, Double_t& conePt);
	149
	150	// ######### CHECK FUNCTIONS
	151	Bool_t IsTrackInAcceptance(AliVParticle* track);
	152	Bool_t IsTrackInCone(AliVTrack* track, Double_t eta, Double_t phi, Double_t radius);
b95d0245	153	Bool_t IsTrackInJet(AliEmcalJet* jet, Int_t trackIndex);
b95d0245	154	Bool_t IsJetOverlapping(AliEmcalJet* jet1, AliEmcalJet* jet2);
12eb4ac1	155
0cb49d99	156	Bool_t IsEventInAcceptance(AliVEvent* event);
12eb4ac1	157	Bool_t IsBackgroundJetInAcceptance(AliEmcalJet* jet);
ec676623	158	Bool_t IsSignalJetInAcceptance(AliEmcalJet* jet, Bool_t usePtCut = kFALSE);
8628b70c	159
12eb4ac1	160	// ######### HELPER FUNCTIONS
	161	Double_t EtaToTheta(Double_t arg);
	162	Double_t ThetaToEta(Double_t arg);
	163	Double_t GetDeltaPhi(Double_t phi1, Double_t phi2);
	164	Double_t MCGetOverlapCircleRectancle(Double_t cPosX, Double_t cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax);
b95d0245	165	Double_t MCGetOverlapMultipleCirclesRectancle(Int_t numCircles, std::vector<Double_t> cPosX, std::vector<Double_t> cPosY, Double_t cRadius, Double_t rPosXmin, Double_t rPosXmax, Double_t rPosYmin, Double_t rPosYmax);
0cf2a402	166	void Match(AliESDtrack* tr0, AliESDtrack* tr1, Int_t& nmatch, Bool_t excludeMom = kFALSE, Double_t rotate=0);
12eb4ac1	167
	168	// ######### HISTOGRAM FUNCTIONS
	169	void FillHistogram(const char * key, Double_t x);
	170	void FillHistogram(const char * key, Double_t x, Double_t y);
	171	void FillHistogram(const char * key, Double_t x, Double_t y, Double_t add);
013103f5	172	void FillCutHistogram(const char * key, Double_t cut, Double_t pT, Double_t eta, Double_t phi, Int_t isAdditionalTrack);
013103f5	173
8628b70c	174	const char* GetHistoName(const char* name)
8628b70c	175	{
013103f5	176	if (fParticleLevel)
7e413f22	177	return Form("%s_MC", name);
7e413f22	178	return Form("%s", name);
8628b70c	179	}
8628b70c	180	template <class T> T* AddHistogram1D(const char* name = "CustomHistogram", const char* title = "NO_TITLE", const char* options = "", Int_t xBins = 100, Double_t xMin = 0.0, Double_t xMax = 20.0, const char* xTitle = "x axis", const char* yTitle = "y axis");
8628b70c	181	template <class T> T* AddHistogram2D(const char* name = "CustomHistogram", const char* title = "NO_TITLE", const char* options = "", Int_t xBins = 100, Double_t xMin = 0.0, Double_t xMax = 20.0, Int_t yBins = 100, Double_t yMin = 0.0, Double_t yMax = 20.0, const char* xTitle = "x axis", const char* yTitle = "y axis", const char* zTitle = "z axis");
013103f5	182	THnF* AddCutHistogram(const char* name, const char* title, const char* cutName, Int_t nBins, Double_t xMin, Double_t xMax);
013103f5	183	void BinLogAxis(const THn *h, Int_t axisNumber);
8628b70c	184
12eb4ac1	185	// ######### STANDARD FUNCTIONS
8628b70c	186	void Calculate(AliVEvent* event);
	187	void ExecOnce();
	188	void Init ();
	189
013103f5	190	std::vector<TList*> fOutputLists; //! Output lists
013103f5	191	TList* fCurrentOutputList; //! Currently selected list where the histograms will be saved to
12eb4ac1	192	// ########## USAGE TRIGGERS
013103f5	193	Bool_t fDoJetAnalysis; // trigger if jets/tracks etc. should be analyzed
2a82f637	194	Bool_t fAnalyzeJetProfile; // trigger if jet profile should be analyzed
013103f5	195	Bool_t fAnalyzeTrackcuts; // trigger if trackcuts should be analyzed
013103f5	196	Bool_t fParticleLevel; // trigger if data is kinematics only (for naming reasons)
2a82f637	197	Bool_t fUseDefaultVertexCut; // trigger if automatic vertex cut from helper class should be done
392aa819	198	Bool_t fUsePileUpCut; // trigger if pileup cut should be done
9f1e4f53	199	Bool_t fSetCentralityToOne; // trigger if centrality val. should be set to one for every event (failsafe)
4bb1bca6	200	Bool_t fNoExternalBackground; // External background is set to 0 (e.g. for PP)
3dff28bc	201	Int_t fBackgroundForJetProfile; // Which background will be subtracted for the profile calculation (0=External,1=Improved CMS,2=CMS,3=PP,4=TR,5=None)
c2b0976c	202	Int_t fPartialAnalysisNParts; // take only every Nth event
	203	Int_t fPartialAnalysisIndex; // using e.g. only every 5th event, this specifies which one
	204
12eb4ac1	205
8628b70c	206	// ########## SOURCE INFORMATION
	207	TClonesArray* fJetArray; //! object containing the jets
	208	TClonesArray* fTrackArray; //! object containing the tracks
8628b70c	209	TClonesArray* fBackgroundJetArray; //! object containing background jets
013103f5	210	TString fJetArrayName; // name of object containing the jets
	211	TString fTrackArrayName; // name of object containing the tracks
	212	TString fBackgroundJetArrayName;// name of object containing event wise bckgrds
06b41974	213	TString fRhoTaskName; // name of rho task for this analysis
8628b70c	214	// ########## JET/DIJET/RC PROPERTIES
	215	Double_t fRandConeRadius; // Radius for the random cones
	216	Double_t fSignalJetRadius; // Radius for the signal jets
	217	Double_t fBackgroundJetRadius; // Radius for the KT background jets
8628b70c	218	Int_t fNumberExcludedJets; // Number of jets to be excluded from backgrounds
8628b70c	219	// ########## CUTS
b836a932	220	Double_t fMinEta; // min eta of tracks
	221	Double_t fMaxEta; // max eta of tracks
	222	Double_t fMinJetEta; // min eta of jets
	223	Double_t fMaxJetEta; // max eta of jets
8628b70c	224	Double_t fMinTrackPt; // Min track pt to be accepted
8628b70c	225	Double_t fMinJetPt; // Min jet pt to be accepted
	226	Double_t fMinJetArea; // Min jet area to be accepted
	227	Double_t fMinBackgroundJetPt; // Min jet pt to be accepted as background jet
2d6ed1a3	228	Int_t fMinNCrossedRows; // Min number of crossed TPC rows for trackcut analysis
e3ed0600	229	Bool_t fUsePtDepCrossedRowsCut;// Trigger if linear pT dep. for crossed rows cut should be applied
b95d0245	230	Int_t fNumberOfCentralityBins;// Number of centrality bins used for histograms
12eb4ac1	231	TString fCentralityType; // Used centrality estimate (V0A, V0C, V0M, ...)
8628b70c	232
565f94c4	233	AliESDtrack* fMatchTr[kMaxMatch]; //! Helper variables track matching
565f94c4	234	Double_t fMatchChi[kMaxMatch]; //! Helper variables track matching
0cf2a402	235
0cf2a402	236
8628b70c	237	// ########## EVENT PROPERTIES
565f94c4	238	const AliVVertex* fPrimaryVertex; //! Vertex found per event
8628b70c	239	AliEmcalJet* fFirstLeadingJet; //! leading jet in event
8628b70c	240	AliEmcalJet* fSecondLeadingJet; //! next to leading jet in event
013103f5	241	AliEmcalJet* fFirstLeadingKTJet; //! leading kT jet in event
	242	AliEmcalJet* fSecondLeadingKTJet; //! next to leading kT jet in event
	243	Int_t fNumberSignalJets; // Number of signal jets in event
	244	Int_t fNumberSignalJetsAbove5GeV; // Number of signal jets in event > 5GeV
8628b70c	245	// ########## GENERAL VARS
8628b70c	246	TRandom3* fRandom; //! A random number
3fa9cde0	247	AliAnalysisUtils* fHelperClass; //! Vertex selection helper
013103f5	248	Bool_t fInitialized; // trigger if tracks/jets are loaded
8628b70c	249	Int_t fTaskInstanceCounter; // for naming reasons
b95d0245	250	Bool_t fIsDEBUG; // Debug trigger
013103f5	251	Bool_t fIsPA; // pPb trigger
0779c72f	252	Bool_t fNoTerminate; // don't use terminate routines
c2b0976c	253	ULong_t fEventCounter; // Internal event counter
013103f5	254	AliESDHybridTrackcuts* fHybridESDtrackCuts; //! these trackcuts are applied
e3ed0600	255	AliESDHybridTrackcuts* fHybridESDtrackCuts_variedPtDep; //! these trackcuts are applied
e3ed0600	256	AliESDHybridTrackcuts* fHybridESDtrackCuts_variedPtDep2; //! these trackcuts are applied
013103f5	257
12eb4ac1	258	AliAnalysisTaskChargedJetsPA(const AliAnalysisTaskChargedJetsPA&);
12eb4ac1	259	AliAnalysisTaskChargedJetsPA& operator=(const AliAnalysisTaskChargedJetsPA&);
8628b70c	260
013103f5	261	ClassDef(AliAnalysisTaskChargedJetsPA, 3); // Charged jet analysis for pA
8628b70c	262
	263	};
	264	#endif