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

#ifndef ALIJETRESPONSEMAKER_H
#define ALIJETRESPONSEMAKER_H

// $Id$

class AliGenPythiaEventHeader;
class TClonesArray;
class TH1;
class TH2;
class AliNamedArrayI;

#include "AliEmcalJet.h"
#include "AliAnalysisTaskEmcalJet.h"

class AliJetResponseMaker : public AliAnalysisTaskEmcalJet {
 public:
  AliJetResponseMaker();
  AliJetResponseMaker(const char *name);
  virtual ~AliJetResponseMaker();

  enum MatchingType{
    kNoMatching = 0,
    kGeometrical = 1,
    kMCLabel = 2,
    kSameCollections = 3
  };

  void                        UserCreateOutputObjects();
  Bool_t                      UserNotify();

  void                        SetJets2Name(const char *n)                                     { fJets2Name         = n         ; }
  void                        SetTracks2Name(const char *n)                                   { fTracks2Name       = n         ; }
  void                        SetClus2Name(const char *n)                                     { fCalo2Name         = n         ; }
  void                        SetJet2EtaLimits(Float_t min=-999, Float_t max=-999)            { fJet2MinEta = min, fJet2MaxEta = max ; }
  void                        SetJet2PhiLimits(Float_t min=-999, Float_t max=-999)            { fJet2MinPhi = min, fJet2MaxPhi = max ; }
  void                        SetRho2Name(const char *n)                                      { fRho2Name          = n         ; }
  void                        SetPtBiasJet2Clus(Float_t b)                                    { fPtBiasJet2Clus    = b         ; }
  void                        SetPtBiasJet2Track(Float_t b)                                   { fPtBiasJet2Track   = b         ; }
  void                        SetMatching(MatchingType t, Double_t p1=1, Double_t p2=1)       { fMatching = t; fMatchingPar1 = p1; fMatchingPar2 = p2; }
  void                        SetPtHardBin(Int_t b)                                           { fSelectPtHardBin   = b         ; }
  void                        SetAreMCCollections(Bool_t f1, Bool_t f2)                       { fAreCollections1MC = f1; fAreCollections2MC = f2; }
  void                        SetIsEmbedded(Bool_t i)                                         { fIsEmbedded        = i         ; }
  void                        SetIsPythia(Bool_t i)                                           { fIsPythia          = i         ; }

 protected:
  Bool_t                      PythiaInfoFromFile(const char* currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard);
  Bool_t                      IsEventSelected();
  Bool_t                      AcceptJet(AliEmcalJet* jet) const;
  Bool_t                      AcceptBiasJet2(AliEmcalJet *jet) const;
  void                        ExecOnce();
  void                        DoJetLoop(Bool_t order);
  Bool_t                      FillHistograms();
  Bool_t                      RetrieveEventObjects();
  Bool_t                      Run();
  Bool_t                      DoJetMatching();
  void                        SetMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, MatchingType matching);
  void                        GetGeometricalMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, Double_t &d) const;
  void                        GetMCLabelMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, Double_t &d1, Double_t &d2) const;
  void                        GetSameCollectionsMatchingLevel(AliEmcalJet *jet1, AliEmcalJet *jet2, Double_t &d1, Double_t &d2) const;

  TString                     fTracks2Name;                   // name of second track collection
  TString                     fCalo2Name;                     // name of second cluster collection
  TString                     fJets2Name;                     // name of second jet collection
  TString                     fRho2Name;                      // name of second jet collection
  Float_t                     fPtBiasJet2Track;               // select jets 2 with a minimum pt track
  Float_t                     fPtBiasJet2Clus;                // select jets 2 with a minimum pt cluster
  Bool_t                      fAreCollections1MC;             // collections 1 MC
  Bool_t                      fAreCollections2MC;             // collections 1 MC
  MatchingType                fMatching;                      // matching type
  Double_t                    fMatchingPar1;                  // matching parameter for jet1-jet2 matching
  Double_t                    fMatchingPar2;                  // matching parameter for jet2-jet1 matching
  Float_t                     fJet2MinEta;                    // minimum eta jet 2 acceptance
  Float_t                     fJet2MaxEta;                    // maximum eta jet 2 acceptance
  Float_t                     fJet2MinPhi;                    // minimum phi jet 2 acceptance
  Float_t                     fJet2MaxPhi;                    // maximum phi jet 2 acceptance  
  Int_t                       fSelectPtHardBin;               // select one pt hard bin for analysis
  Bool_t                      fIsEmbedded;                    // trigger, embedded signal
  Bool_t                      fIsPythia;                      // trigger, if it is a PYTHIA production

  AliGenPythiaEventHeader    *fPythiaHeader;                  //!event Pythia header
  Int_t                       fPtHardBin;                     //!event pt hard bin
  Int_t                       fNTrials;                       //!event trials
  TClonesArray               *fTracks2;                       //!Tracks 2
  TClonesArray               *fCaloClusters2;                 //!Clusters 2
  TClonesArray               *fJets2;                         //!Jets 2
  AliRhoParameter            *fRho2;                          //!Event rho 2
  Double_t                    fRho2Val;                       //!Event rho 2 value 
  AliNamedArrayI             *fTracks2Map;                    //!MC particle map
  // General histograms
  TH1                        *fHistTrialsAfterSel;            //!total number of trials per pt hard bin after selection
  TH1                        *fHistEventsAfterSel;            //!total number of events per pt hard bin after selection
  TH1                        *fHistTrials;                    //!trials from pyxsec.root
  TH1                        *fHistXsection;                  //!x section from pyxsec.root
  TH1                        *fHistEvents;                    //!total number of events per pt hard bin
  // Jets 1
  TH2                        *fHistJets1PhiEta;               //!phi-eta distribution of jets 1
  TH2                        *fHistJets1PtArea;               //!inclusive jet pt vs area histogram 1
  TH2                        *fHistJets1CorrPtArea;           //!inclusive jet pt vs. area histogram 1
  TH2                        *fHistLeadingJets1PtArea;        //!leading jet pt vs area histogram 1
  TH2                        *fHistLeadingJets1CorrPtArea;    //!leading jet pt vs. area histogram 1
  TH2                        *fHistJets1NEFvsPt;              //!Jet neutral energy fraction vs. jet pt 1
  TH2                        *fHistJets1CEFvsCEFPt;           //!Jet charged energy fraction vs. charged jet pt 1
  TH2                        *fHistJets1ZvsPt;                //!Constituent Pt over Jet Pt ratio vs. jet pt 1
  // Jets 2
  TH2                        *fHistJets2PhiEta;               //!phi-eta distribution of jets 2
  TH2                        *fHistJets2PtArea;               //!inclusive jet pt vs. area histogram 2
  TH2                        *fHistJets2CorrPtArea;           //!inclusive jet pt vs. area histogram 2
  TH2                        *fHistLeadingJets2PtArea;        //!leading jet pt vs. area histogram 2
  TH2                        *fHistLeadingJets2CorrPtArea;    //!leading jet pt vs. area histogram 2
  TH2                        *fHistJets2PhiEtaAcceptance;     //!phi-eta distribution of jets 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
  TH2                        *fHistJets2PtAreaAcceptance;     //!inclusive jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
  TH2                        *fHistJets2CorrPtAreaAcceptance; //!inclusive jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
  TH2                        *fHistLeadingJets2PtAreaAcceptance;     //!leading jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
  TH2                        *fHistLeadingJets2CorrPtAreaAcceptance; //!leading jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
  TH2                        *fHistJets2NEFvsPt;              //!Jet neutral energy fraction vs. jet pt 2
  TH2                        *fHistJets2CEFvsCEFPt;           //!Jet charged energy fraction vs. charged jet pt 2
  TH2                        *fHistJets2ZvsPt;                //!Constituent Pt over Jet Pt ratio vs. jet pt 2
  // Jet1-Jet2 matching
  TH2                        *fHistCommonEnergy1vsJet1Pt;              //!common energy 1 (%) vs jet 1 pt
  TH2                        *fHistCommonEnergy2vsJet2Pt;              //!common energy 2 (%) vs jet 2 pt
  TH2                        *fHistDistancevsJet1Pt;                   //!distance vs jet 1 pt
  TH2                        *fHistDistancevsJet2Pt;                   //!distance vs jet 2 pt
  TH2                        *fHistDistancevsCommonEnergy1;            //!distance vs common energy 1 (%)
  TH2                        *fHistDistancevsCommonEnergy2;            //!distance vs common energy 2 (%)
  TH2                        *fHistJet2PtOverJet1PtvsJet2Pt;           //!jet 2 pt over jet 1 pt vs jet 2 pt
  TH2                        *fHistJet1PtOverJet2PtvsJet1Pt;           //!jet 1 pt over jet 2 pt vs jet 1 pt
  TH2                        *fHistDeltaEtaPhi;                        //!delta eta-phi between matched jets
  TH2                        *fHistDeltaPtvsJet1Pt;                    //!delta pt between matched jets vs jet 1 pt
  TH2                        *fHistDeltaPtvsJet2Pt;                    //!delta pt between matched jets vs jet 2 pt
  TH2                        *fHistDeltaPtvsMatchingLevel;             //!delta pt between matched jets vs matching level
  TH2                        *fHistDeltaCorrPtvsJet1Pt;                //!delta pt corr between matched jets vs jet 1 pt
  TH2                        *fHistDeltaCorrPtvsJet2Pt;                //!delta pt corr between matched jets vs jet 2 pt
  TH2                        *fHistDeltaCorrPtvsMatchingLevel;         //!delta pt corr between matched jets vs matching level
  TH2                        *fHistNonMatchedJets1PtArea;              //!non-matched jet 1 pt distribution
  TH2                        *fHistNonMatchedJets2PtArea;              //!non-matched jet 2 pt distribution
  TH2                        *fHistNonMatchedJets1CorrPtArea;          //!non-matched jet pt distribution
  TH2                        *fHistNonMatchedJets2CorrPtArea;          //!non-matched jet pt distribution
  TH2                        *fHistJet1PtvsJet2Pt;                     //!correlation jet 1 pt vs jet 2 pt
  TH2                        *fHistJet1CorrPtvsJet2CorrPt;             //!correlation jet 1 corr pt vs jet 2 corr pt
  TH2                        *fHistJet1MCPtvsJet2Pt;                   //!correlation jet 1 MC pt vs jet 2 pt
  TH2                        *fHistMissedJets2PtArea;                  //!jets 2 not found in jet 1 collection


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

  ClassDef(AliJetResponseMaker, 13) // Jet response matrix producing task
};
#endif
Commit	Line	Data
2949a2ec	1	#ifndef ALIJETRESPONSEMAKER_H
	2	#define ALIJETRESPONSEMAKER_H
	3
7549c451	4	// $Id$
2949a2ec	5
1ee1b5b8	6	class AliGenPythiaEventHeader;
2949a2ec	7	class TClonesArray;
b16bb001	8	class TH1;
b16bb001	9	class TH2;
5be3857d	10	class AliNamedArrayI;
2949a2ec	11
cfc2ac24	12	#include "AliEmcalJet.h"
6fd5039f	13	#include "AliAnalysisTaskEmcalJet.h"
2949a2ec	14
6fd5039f	15	class AliJetResponseMaker : public AliAnalysisTaskEmcalJet {
2949a2ec	16	public:
	17	AliJetResponseMaker();
	18	AliJetResponseMaker(const char *name);
	19	virtual ~AliJetResponseMaker();
	20
cd6431de	21	enum MatchingType{
	22	kNoMatching = 0,
	23	kGeometrical = 1,
7f76e479	24	kMCLabel = 2,
7f76e479	25	kSameCollections = 3
cd6431de	26	};
cd6431de	27
a7ab01d5	28	void UserCreateOutputObjects();
ca5c29fa	29	Bool_t UserNotify();
a7ab01d5	30
7f76e479	31	void SetJets2Name(const char *n) { fJets2Name = n ; }
	32	void SetTracks2Name(const char *n) { fTracks2Name = n ; }
	33	void SetClus2Name(const char *n) { fCalo2Name = n ; }
	34	void SetJet2EtaLimits(Float_t min=-999, Float_t max=-999) { fJet2MinEta = min, fJet2MaxEta = max ; }
	35	void SetJet2PhiLimits(Float_t min=-999, Float_t max=-999) { fJet2MinPhi = min, fJet2MaxPhi = max ; }
	36	void SetRho2Name(const char *n) { fRho2Name = n ; }
	37	void SetPtBiasJet2Clus(Float_t b) { fPtBiasJet2Clus = b ; }
	38	void SetPtBiasJet2Track(Float_t b) { fPtBiasJet2Track = b ; }
	39	void SetMatching(MatchingType t, Double_t p1=1, Double_t p2=1) { fMatching = t; fMatchingPar1 = p1; fMatchingPar2 = p2; }
	40	void SetPtHardBin(Int_t b) { fSelectPtHardBin = b ; }
	41	void SetAreMCCollections(Bool_t f1, Bool_t f2) { fAreCollections1MC = f1; fAreCollections2MC = f2; }
4358e58a	42	void SetIsEmbedded(Bool_t i) { fIsEmbedded = i ; }
4358e58a	43	void SetIsPythia(Bool_t i) { fIsPythia = i ; }
2949a2ec	44
2949a2ec	45	protected:
ca5c29fa	46	Bool_t PythiaInfoFromFile(const char* currFile, Float_t &fXsec, Float_t &fTrials, Int_t &pthard);
4643d2e8	47	Bool_t IsEventSelected();
cd6431de	48	Bool_t AcceptJet(AliEmcalJet* jet) const;
cd6431de	49	Bool_t AcceptBiasJet2(AliEmcalJet *jet) const;
1ee1b5b8	50	void ExecOnce();
cfc2ac24	51	void DoJetLoop(Bool_t order);
6fd5039f	52	Bool_t FillHistograms();
6fd5039f	53	Bool_t RetrieveEventObjects();
1ee1b5b8	54	Bool_t Run();
f660c2d6	55	Bool_t DoJetMatching();
7f76e479	56	void SetMatchingLevel(AliEmcalJet jet1, AliEmcalJet jet2, MatchingType matching);
	57	void GetGeometricalMatchingLevel(AliEmcalJet jet1, AliEmcalJet jet2, Double_t &d) const;
	58	void GetMCLabelMatchingLevel(AliEmcalJet jet1, AliEmcalJet jet2, Double_t &d1, Double_t &d2) const;
	59	void GetSameCollectionsMatchingLevel(AliEmcalJet jet1, AliEmcalJet jet2, Double_t &d1, Double_t &d2) const;
2949a2ec	60
cd6431de	61	TString fTracks2Name; // name of second track collection
	62	TString fCalo2Name; // name of second cluster collection
	63	TString fJets2Name; // name of second jet collection
	64	TString fRho2Name; // name of second jet collection
	65	Float_t fPtBiasJet2Track; // select jets 2 with a minimum pt track
	66	Float_t fPtBiasJet2Clus; // select jets 2 with a minimum pt cluster
	67	Bool_t fAreCollections1MC; // collections 1 MC
	68	Bool_t fAreCollections2MC; // collections 1 MC
	69	MatchingType fMatching; // matching type
ca5c29fa	70	Double_t fMatchingPar1; // matching parameter for jet1-jet2 matching
ca5c29fa	71	Double_t fMatchingPar2; // matching parameter for jet2-jet1 matching
cd6431de	72	Float_t fJet2MinEta; // minimum eta jet 2 acceptance
	73	Float_t fJet2MaxEta; // maximum eta jet 2 acceptance
	74	Float_t fJet2MinPhi; // minimum phi jet 2 acceptance
	75	Float_t fJet2MaxPhi; // maximum phi jet 2 acceptance
b16bb001	76	Int_t fSelectPtHardBin; // select one pt hard bin for analysis
4358e58a	77	Bool_t fIsEmbedded; // trigger, embedded signal
4358e58a	78	Bool_t fIsPythia; // trigger, if it is a PYTHIA production
1ee1b5b8	79
b16bb001	80	AliGenPythiaEventHeader *fPythiaHeader; //!event Pythia header
b16bb001	81	Int_t fPtHardBin; //!event pt hard bin
b16bb001	82	Int_t fNTrials; //!event trials
cd6431de	83	TClonesArray *fTracks2; //!Tracks 2
cd6431de	84	TClonesArray *fCaloClusters2; //!Clusters 2
cfc2ac24	85	TClonesArray *fJets2; //!Jets 2
cd6431de	86	AliRhoParameter *fRho2; //!Event rho 2
cd6431de	87	Double_t fRho2Val; //!Event rho 2 value
5be3857d	88	AliNamedArrayI *fTracks2Map; //!MC particle map
1ee1b5b8	89	// General histograms
ca5c29fa	90	TH1 *fHistTrialsAfterSel; //!total number of trials per pt hard bin after selection
	91	TH1 *fHistEventsAfterSel; //!total number of events per pt hard bin after selection
	92	TH1 *fHistTrials; //!trials from pyxsec.root
	93	TH1 *fHistXsection; //!x section from pyxsec.root
b16bb001	94	TH1 *fHistEvents; //!total number of events per pt hard bin
cd6431de	95	// Jets 1
	96	TH2 *fHistJets1PhiEta; //!phi-eta distribution of jets 1
	97	TH2 *fHistJets1PtArea; //!inclusive jet pt vs area histogram 1
	98	TH2 *fHistJets1CorrPtArea; //!inclusive jet pt vs. area histogram 1
ca5c29fa	99	TH2 *fHistLeadingJets1PtArea; //!leading jet pt vs area histogram 1
ca5c29fa	100	TH2 *fHistLeadingJets1CorrPtArea; //!leading jet pt vs. area histogram 1
63fac07f	101	TH2 *fHistJets1NEFvsPt; //!Jet neutral energy fraction vs. jet pt 1
	102	TH2 *fHistJets1CEFvsCEFPt; //!Jet charged energy fraction vs. charged jet pt 1
	103	TH2 *fHistJets1ZvsPt; //!Constituent Pt over Jet Pt ratio vs. jet pt 1
cd6431de	104	// Jets 2
	105	TH2 *fHistJets2PhiEta; //!phi-eta distribution of jets 2
	106	TH2 *fHistJets2PtArea; //!inclusive jet pt vs. area histogram 2
	107	TH2 *fHistJets2CorrPtArea; //!inclusive jet pt vs. area histogram 2
ca5c29fa	108	TH2 *fHistLeadingJets2PtArea; //!leading jet pt vs. area histogram 2
ca5c29fa	109	TH2 *fHistLeadingJets2CorrPtArea; //!leading jet pt vs. area histogram 2
cfc2ac24	110	TH2 *fHistJets2PhiEtaAcceptance; //!phi-eta distribution of jets 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
	111	TH2 *fHistJets2PtAreaAcceptance; //!inclusive jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
	112	TH2 *fHistJets2CorrPtAreaAcceptance; //!inclusive jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
ca5c29fa	113	TH2 *fHistLeadingJets2PtAreaAcceptance; //!leading jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
ca5c29fa	114	TH2 *fHistLeadingJets2CorrPtAreaAcceptance; //!leading jet pt vs. area histogram 2 using jet 1 cuts (acceptance, leading hadron bias, ...)
63fac07f	115	TH2 *fHistJets2NEFvsPt; //!Jet neutral energy fraction vs. jet pt 2
	116	TH2 *fHistJets2CEFvsCEFPt; //!Jet charged energy fraction vs. charged jet pt 2
	117	TH2 *fHistJets2ZvsPt; //!Constituent Pt over Jet Pt ratio vs. jet pt 2
cd6431de	118	// Jet1-Jet2 matching
ca5c29fa	119	TH2 *fHistCommonEnergy1vsJet1Pt; //!common energy 1 (%) vs jet 1 pt
	120	TH2 *fHistCommonEnergy2vsJet2Pt; //!common energy 2 (%) vs jet 2 pt
	121	TH2 *fHistDistancevsJet1Pt; //!distance vs jet 1 pt
	122	TH2 *fHistDistancevsJet2Pt; //!distance vs jet 2 pt
6a20534a	123	TH2 *fHistDistancevsCommonEnergy1; //!distance vs common energy 1 (%)
6a20534a	124	TH2 *fHistDistancevsCommonEnergy2; //!distance vs common energy 2 (%)
c560b734	125	TH2 *fHistJet2PtOverJet1PtvsJet2Pt; //!jet 2 pt over jet 1 pt vs jet 2 pt
7030f36f	126	TH2 *fHistJet1PtOverJet2PtvsJet1Pt; //!jet 1 pt over jet 2 pt vs jet 1 pt
ca5c29fa	127	TH2 *fHistDeltaEtaPhi; //!delta eta-phi between matched jets
7030f36f	128	TH2 *fHistDeltaPtvsJet1Pt; //!delta pt between matched jets vs jet 1 pt
cfc2ac24	129	TH2 *fHistDeltaPtvsJet2Pt; //!delta pt between matched jets vs jet 2 pt
cfc2ac24	130	TH2 *fHistDeltaPtvsMatchingLevel; //!delta pt between matched jets vs matching level
7030f36f	131	TH2 *fHistDeltaCorrPtvsJet1Pt; //!delta pt corr between matched jets vs jet 1 pt
cfc2ac24	132	TH2 *fHistDeltaCorrPtvsJet2Pt; //!delta pt corr between matched jets vs jet 2 pt
cfc2ac24	133	TH2 *fHistDeltaCorrPtvsMatchingLevel; //!delta pt corr between matched jets vs matching level
cd6431de	134	TH2 *fHistNonMatchedJets1PtArea; //!non-matched jet 1 pt distribution
	135	TH2 *fHistNonMatchedJets2PtArea; //!non-matched jet 2 pt distribution
	136	TH2 *fHistNonMatchedJets1CorrPtArea; //!non-matched jet pt distribution
	137	TH2 *fHistNonMatchedJets2CorrPtArea; //!non-matched jet pt distribution
	138	TH2 *fHistJet1PtvsJet2Pt; //!correlation jet 1 pt vs jet 2 pt
	139	TH2 *fHistJet1CorrPtvsJet2CorrPt; //!correlation jet 1 corr pt vs jet 2 corr pt
4358e58a	140	TH2 *fHistJet1MCPtvsJet2Pt; //!correlation jet 1 MC pt vs jet 2 pt
cd6431de	141	TH2 *fHistMissedJets2PtArea; //!jets 2 not found in jet 1 collection
2949a2ec	142
4358e58a	143
2949a2ec	144	private:
	145	AliJetResponseMaker(const AliJetResponseMaker&); // not implemented
	146	AliJetResponseMaker &operator=(const AliJetResponseMaker&); // not implemented
	147
4358e58a	148	ClassDef(AliJetResponseMaker, 13) // Jet response matrix producing task
2949a2ec	149	};
2949a2ec	150	#endif