[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliPWG4HighPtQAMC.h

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

//-----------------------------------------------------------------------
// This class compares the global reconstruction with the MC information
// Author : Marta Verweij - UU
//-----------------------------------------------------------------------

#ifndef ALIPWG4HIGHPTQAMC_H
#define ALIPWG4HIGHPTQAMC_H

#include "AliAnalysisTask.h"

class TH1F;
class TH2F;
class TH3F;
class TProfile;
class TList;
class AliESDEvent;
class AliESDtrackCuts;
class AliMCEvent;
class AliStack;
class AliGenPythiaEventHeader;
//class AliAnalysisHelperJetTasks;

class AliPWG4HighPtQAMC: public AliAnalysisTask {

 public:
  AliPWG4HighPtQAMC();
  AliPWG4HighPtQAMC(const char *name);
  ~AliPWG4HighPtQAMC() {;}
 
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);
  virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2

  Bool_t SelectEvent();    //decides if event is used for analysis

  void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
  void SetCutsITS(AliESDtrackCuts* trackCutsITS) {fTrackCutsITS = trackCutsITS;}

  void SetTrackType(Int_t trackType) {fTrackType = trackType;}
  void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
  Float_t GetPtMax()           {return fPtMax;}

  static AliGenPythiaEventHeader*  GetPythiaEventHeader(AliMCEvent *mcEvent);
  static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root

 protected:

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

  AliESDEvent *fESD;              //! ESD object
  AliMCEvent  *fMC;               //! MC event object
  AliStack    *fStack;            //! stack object

  AliESDtrackCuts *fTrackCuts;    // TrackCuts for global reconstructed vs MC comparison
  AliESDtrackCuts *fTrackCutsITS; // TrackCuts including ITSrefit

  Int_t   fTrackType;             // 0: global track; 1:TPConly track

  Float_t fPtMax;                 // Maximum pT for histograms

  Float_t fAvgTrials;             // Average number of trials
  
  TH1F *fNEventAll;                            //! Event counter
  TH1F *fNEventSel;                            //! Event counter
  TH1F *fNEventReject;                         //! Book keeping of reason of rejecting events
 
  TProfile*     fh1Xsec;                       //! pythia cross section and trials
  TH1F*         fh1Trials;                     //! trials which are added
  TH1F*         fh1PtHard;                     //! pt hard of the event
  TH1F*         fh1PtHardTrials;               //! pt hard of the event


  TH1F *fPtAll;                                //! Pt spectrum all charged particles
  TH1F *fPtSel;                                //! Pt spectrum all selected charged particles by fTrackCuts
  TH2F *fPtAllminPtMCvsPtAll;                  //! Momentum resolution (global vs MC)
  TH3F *fPtAllminPtMCvsPtAllNPointTPC;         //! Momentum resolution vs NPointTPC
  TH3F *fPtAllminPtMCvsPtAllDCAR;              //! Momentum resolution vs DCAR
  TH3F *fPtAllminPtMCvsPtAllDCAZ;              //! Momentum resolution vs DCAZ
  TH3F *fPtAllminPtMCvsPtAllPhi;               //! Momentum resolution vs Phi
  TH3F *fPtAllminPtMCvsPtAllNPointITS;         //! Momentum resolution vs NPointITS
  TH3F *fPtAllminPtMCvsPtAllNSigmaToVertex;    //! Momentum resolution vs NSigmaToVertes
  TH3F *fPtAllminPtMCvsPtAllChi2C;             //! Momentum resolution vs Chi2Constrained
  TH3F *fPtAllminPtMCvsPtAllRel1PtUncertainty; //! Momentum resolution vs relUncertainty1Pt

  TH1F *fPtAllMC;     //! Pt spectrum all charged particles
  TH1F *fPtSelMC;     //! Pt spectrum all selected charged particles by fTrackCuts
  TH1F *fPtSelMCITS;  //! Pt spectrum all selected charged particles by fTrackCutsITS

  TList *fHistList; //! List of Histograms
  
  TH1F *fPtSelITS;                              //! Pt spectrum all selected charged particles by fTrackCutsITS
  TH2F *fPtITSminPtMCvsPtITS;                  //! Momentum resolution (global with ITSrefit vs MC)
  TH3F *fPtITSminPtMCvsPtITSNPointTPC;         //! Momentum resolution vs NPointTPC 
  TH3F *fPtITSminPtMCvsPtITSDCAR;              //! Momentum resolution vs DCAR
  TH3F *fPtITSminPtMCvsPtITSDCAZ;              //! Momentum resolution vs DCAZ
  TH3F *fPtITSminPtMCvsPtITSPhi;               //! Momentum resolution vs Phi
  TH3F *fPtITSminPtMCvsPtITSNPointITS;         //! Momentum resolution vs NPointITS
  TH3F *fPtITSminPtMCvsPtITSNSigmaToVertex;    //! Momentum resolution vs NSigmaToVertex
  TH3F *fPtITSminPtMCvsPtITSChi2C;             //! Momentum resolution vs Chi2Constrained
  TH3F *fPtITSminPtMCvsPtITSRel1PtUncertainty; //! Momentum resolution vs relUncertainty1Pt

  TList *fHistListITS; //! List of Histograms

 
  ClassDef(AliPWG4HighPtQAMC,1) 
  
};
#endif
Commit	Line	Data
fdceab34	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	//-----------------------------------------------------------------------
	17	// This class compares the global reconstruction with the MC information
	18	// Author : Marta Verweij - UU
	19	//-----------------------------------------------------------------------
	20
	21	#ifndef ALIPWG4HIGHPTQAMC_H
	22	#define ALIPWG4HIGHPTQAMC_H
	23
	24	#include "AliAnalysisTask.h"
	25
	26	class TH1F;
	27	class TH2F;
	28	class TH3F;
b4691ee7	29	class TProfile;
fdceab34	30	class TList;
	31	class AliESDEvent;
	32	class AliESDtrackCuts;
b1cd0099	33	class AliMCEvent;
cce400da	34	class AliStack;
b4691ee7	35	class AliGenPythiaEventHeader;
b4691ee7	36	//class AliAnalysisHelperJetTasks;
fdceab34	37
	38	class AliPWG4HighPtQAMC: public AliAnalysisTask {
	39
	40	public:
	41	AliPWG4HighPtQAMC();
	42	AliPWG4HighPtQAMC(const char *name);
	43	~AliPWG4HighPtQAMC() {;}
b4691ee7	44
fdceab34	45	virtual void ConnectInputData(Option_t *);
	46	virtual void CreateOutputObjects();
	47	virtual void Exec(Option_t *option);
	48	virtual void Terminate(Option_t *);
b4691ee7	49	virtual Bool_t Notify(); //Copied from AliAnalysisTaskJetSpectrum2
fdceab34	50
cce400da	51	Bool_t SelectEvent(); //decides if event is used for analysis
cce400da	52
fdceab34	53	void SetCuts(AliESDtrackCuts* trackCuts) {fTrackCuts = trackCuts;}
	54	void SetCutsITS(AliESDtrackCuts* trackCutsITS) {fTrackCutsITS = trackCutsITS;}
	55
71e77a79	56	void SetTrackType(Int_t trackType) {fTrackType = trackType;}
b1cd0099	57	void SetPtMax(Float_t ptmax) {fPtMax = ptmax;}
	58	Float_t GetPtMax() {return fPtMax;}
	59
b4691ee7	60	static AliGenPythiaEventHeader* GetPythiaEventHeader(AliMCEvent *mcEvent);
	61	static Bool_t PythiaInfoFromFile(const char* currFile,Float_t &fXsec,Float_t &fTrials);// get the cross section and the trails either from pyxsec.root or from pysec_hists.root
	62
fdceab34	63	protected:
	64
	65	private:
fdceab34	66	AliPWG4HighPtQAMC(const AliPWG4HighPtQAMC&);
	67	AliPWG4HighPtQAMC& operator=(const AliPWG4HighPtQAMC&);
	68
	69	AliESDEvent *fESD; //! ESD object
b1cd0099	70	AliMCEvent *fMC; //! MC event object
cce400da	71	AliStack *fStack; //! stack object
cce400da	72
fdceab34	73	AliESDtrackCuts *fTrackCuts; // TrackCuts for global reconstructed vs MC comparison
	74	AliESDtrackCuts *fTrackCutsITS; // TrackCuts including ITSrefit
	75
b4691ee7	76	Int_t fTrackType; // 0: global track; 1:TPConly track
71e77a79	77
b1cd0099	78	Float_t fPtMax; // Maximum pT for histograms
b1cd0099	79
b4691ee7	80	Float_t fAvgTrials; // Average number of trials
fdceab34	81
8f0faa80	82	TH1F *fNEventAll; //! Event counter
8f0faa80	83	TH1F *fNEventSel; //! Event counter
cb76764e	84	TH1F *fNEventReject; //! Book keeping of reason of rejecting events
b4691ee7	85
	86	TProfile* fh1Xsec; //! pythia cross section and trials
	87	TH1F* fh1Trials; //! trials which are added
	88	TH1F* fh1PtHard; //! pt hard of the event
	89	TH1F* fh1PtHardTrials; //! pt hard of the event
	90
	91
fdceab34	92	TH1F *fPtAll; //! Pt spectrum all charged particles
	93	TH1F *fPtSel; //! Pt spectrum all selected charged particles by fTrackCuts
	94	TH2F *fPtAllminPtMCvsPtAll; //! Momentum resolution (global vs MC)
	95	TH3F *fPtAllminPtMCvsPtAllNPointTPC; //! Momentum resolution vs NPointTPC
	96	TH3F *fPtAllminPtMCvsPtAllDCAR; //! Momentum resolution vs DCAR
	97	TH3F *fPtAllminPtMCvsPtAllDCAZ; //! Momentum resolution vs DCAZ
	98	TH3F *fPtAllminPtMCvsPtAllPhi; //! Momentum resolution vs Phi
	99	TH3F *fPtAllminPtMCvsPtAllNPointITS; //! Momentum resolution vs NPointITS
	100	TH3F *fPtAllminPtMCvsPtAllNSigmaToVertex; //! Momentum resolution vs NSigmaToVertes
	101	TH3F *fPtAllminPtMCvsPtAllChi2C; //! Momentum resolution vs Chi2Constrained
	102	TH3F *fPtAllminPtMCvsPtAllRel1PtUncertainty; //! Momentum resolution vs relUncertainty1Pt
	103
	104	TH1F *fPtAllMC; //! Pt spectrum all charged particles
	105	TH1F *fPtSelMC; //! Pt spectrum all selected charged particles by fTrackCuts
	106	TH1F *fPtSelMCITS; //! Pt spectrum all selected charged particles by fTrackCutsITS
	107
	108	TList *fHistList; //! List of Histograms
	109
	110	TH1F *fPtSelITS; //! Pt spectrum all selected charged particles by fTrackCutsITS
	111	TH2F *fPtITSminPtMCvsPtITS; //! Momentum resolution (global with ITSrefit vs MC)
	112	TH3F *fPtITSminPtMCvsPtITSNPointTPC; //! Momentum resolution vs NPointTPC
	113	TH3F *fPtITSminPtMCvsPtITSDCAR; //! Momentum resolution vs DCAR
	114	TH3F *fPtITSminPtMCvsPtITSDCAZ; //! Momentum resolution vs DCAZ
	115	TH3F *fPtITSminPtMCvsPtITSPhi; //! Momentum resolution vs Phi
	116	TH3F *fPtITSminPtMCvsPtITSNPointITS; //! Momentum resolution vs NPointITS
	117	TH3F *fPtITSminPtMCvsPtITSNSigmaToVertex; //! Momentum resolution vs NSigmaToVertex
	118	TH3F *fPtITSminPtMCvsPtITSChi2C; //! Momentum resolution vs Chi2Constrained
	119	TH3F *fPtITSminPtMCvsPtITSRel1PtUncertainty; //! Momentum resolution vs relUncertainty1Pt
	120
	121	TList *fHistListITS; //! List of Histograms
	122
	123
	124	ClassDef(AliPWG4HighPtQAMC,1)
	125
	126	};
	127	#endif