[u/mrichter/AliRoot.git] / PWG1 / ITS / AliAnalysisTaskITSTrackingCheck.h

#ifndef AliAnalysisTaskITSTrackingCheck_cxx
#define AliAnalysisTaskITSTrackingCheck_cxx

/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//*************************************************************************
// Class AliAnalysiTaskITSTrackingCheck
// AliAnalysisTask to extract from ESD tracks the information on the
// ITS tracking efficiency and resolutions
//
// Author: A.Dainese, andrea.dainese@pd.infn.it
//*************************************************************************

class TNtuple;
class TParticle;
class TH1F;
class AliESDEvent;
class AliESDVertex;
class AliESDfriend;

#include "AliAnalysisTask.h"

class AliAnalysisTaskITSTrackingCheck : public AliAnalysisTask 
{
 public:

  AliAnalysisTaskITSTrackingCheck(const char *name = "AliAnalysisTaskITSTrackingCheck");
  virtual ~AliAnalysisTaskITSTrackingCheck(); 
  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);
  Bool_t         GetReadMC() const { return fReadMC; }
  void           SetReadMC(Bool_t flag=kTRUE) { fReadMC=flag; }
  void           SetReadRPLabels(Bool_t flag=kTRUE) { fReadRPLabels=flag; }
  void           SetFillNtuples(Bool_t flag=kTRUE) { fFillNtuples=flag; }

  
 protected:
  Bool_t       fReadMC; // read Monte Carlo
  Bool_t       fReadRPLabels; // read MC labels from ITS.RecPoints
  Bool_t       fFillNtuples; // fill expert ntuples
  AliESDEvent  *fESD;    // ESD object
  AliESDfriend *fESDfriend; // ESD friend object
  TList        *fOutput; //! list send on output slot 0
  TH1F         *fHistNtracks; //! output hist
  TH1F         *fHistNclsITSMI; //! output hist
  TH1F         *fHistNclsITSSA; //! output hist
  TH1F         *fHistNclsITSSAInAcc; //! output hist
  TH1F         *fHistClusterMapITSMI; //! output hist
  TH1F         *fHistClusterMapITSMIok; //! output hist
  TH1F         *fHistClusterMapITSMIbad; //! output hist
  TH1F         *fHistClusterMapITSMIskipped; //! output hist
  TH1F         *fHistClusterMapITSMIoutinz; //! output hist
  TH1F         *fHistClusterMapITSMInorefit; //! output hist
  TH1F         *fHistClusterMapITSMInocls; //! output hist
  TH1F         *fHistClusterMapITSMIokoutinzbad; //! output hist
  TH1F         *fHistClusterMapITSSA; //! output hist
  TH1F         *fHistClusterMapITSSAok; //! output hist
  TH1F         *fHistClusterMapITSSAbad; //! output hist
  TH1F         *fHistClusterMapITSSAskipped; //! output hist
  TH1F         *fHistClusterMapITSSAoutinz; //! output hist
  TH1F         *fHistClusterMapITSSAnorefit; //! output hist
  TH1F         *fHistClusterMapITSSAnocls; //! output hist
  TH1F         *fHistClusterMapITSSAokoutinzbad; //! output hist
  TH1F         *fHistClusterMapITSSAInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAokInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAbadInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAskippedInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAoutinzInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAnorefitInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAnoclsInAcc; //! output hist
  TH1F         *fHistClusterMapITSSAokoutinzbadInAcc; //! output hist
  TH1F         *fHistClusterMapModuleITSSAokInAcc; //! output hist
  TH1F         *fHistClusterMapModuleITSSAbadInAcc; //! output hist
  TH1F         *fHistClusterMapModuleITSSAnoclsInAcc; //! output hist
  TH1F         *fHistPhiTPC; //! output hist
  TH1F         *fHistPtTPC; //! output hist
  TH1F         *fHistPtITSMI2; //! output hist
  TH1F         *fHistPtITSMI3; //! output hist
  TH1F         *fHistPtITSMI4; //! output hist
  TH1F         *fHistPtITSMI5; //! output hist
  TH1F         *fHistPtITSMI6; //! output hist
  TH1F         *fHistPtITSMISPD; //! output hist
  TNtuple      *fNtupleESDTracks; //! output ntuple
  TNtuple      *fNtupleITSAlignExtra; //! output ntuple
  TNtuple      *fNtupleITSAlignSPDTracklets; //! output ntuple
  Int_t         fCountsPerPtBin[10]; // track per pt bin

 private:    

  AliAnalysisTaskITSTrackingCheck(const AliAnalysisTaskITSTrackingCheck&); // not implemented
  AliAnalysisTaskITSTrackingCheck& operator=(const AliAnalysisTaskITSTrackingCheck&); // not implemented
  

  Int_t NumberOfITSClustersMC(Int_t label) const;
  Double_t ParticleImpParMC(TParticle *part,AliESDVertex *vert,Double_t bzT) const;
  Bool_t SelectPt(Double_t pt);

  ClassDef(AliAnalysisTaskITSTrackingCheck,1); // ITS tracks analysis
};

#endif
Commit	Line	Data
8d63376d	1	#ifndef AliAnalysisTaskITSTrackingCheck_cxx
	2	#define AliAnalysisTaskITSTrackingCheck_cxx
	3
	4	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	//*************************************************************************
	8	// Class AliAnalysiTaskITSTrackingCheck
	9	// AliAnalysisTask to extract from ESD tracks the information on the
	10	// ITS tracking efficiency and resolutions
	11	//
	12	// Author: A.Dainese, andrea.dainese@pd.infn.it
	13	//*************************************************************************
	14
	15	class TNtuple;
	16	class TParticle;
	17	class TH1F;
	18	class AliESDEvent;
	19	class AliESDVertex;
	20	class AliESDfriend;
	21
	22	#include "AliAnalysisTask.h"
	23
	24	class AliAnalysisTaskITSTrackingCheck : public AliAnalysisTask
	25	{
	26	public:
	27
	28	AliAnalysisTaskITSTrackingCheck(const char *name = "AliAnalysisTaskITSTrackingCheck");
	29	virtual ~AliAnalysisTaskITSTrackingCheck();
	30
	31	virtual void ConnectInputData(Option_t *);
	32	virtual void CreateOutputObjects();
	33	virtual void Exec(Option_t *option);
	34	virtual void Terminate(Option_t *);
	35	Bool_t GetReadMC() const { return fReadMC; }
	36	void SetReadMC(Bool_t flag=kTRUE) { fReadMC=flag; }
	37	void SetReadRPLabels(Bool_t flag=kTRUE) { fReadRPLabels=flag; }
2308c814	38	void SetFillNtuples(Bool_t flag=kTRUE) { fFillNtuples=flag; }
8d63376d	39
	40
	41	protected:
	42	Bool_t fReadMC; // read Monte Carlo
	43	Bool_t fReadRPLabels; // read MC labels from ITS.RecPoints
2308c814	44	Bool_t fFillNtuples; // fill expert ntuples
8d63376d	45	AliESDEvent *fESD; // ESD object
	46	AliESDfriend *fESDfriend; // ESD friend object
	47	TList *fOutput; //! list send on output slot 0
	48	TH1F *fHistNtracks; //! output hist
	49	TH1F *fHistNclsITSMI; //! output hist
	50	TH1F *fHistNclsITSSA; //! output hist
2308c814	51	TH1F *fHistNclsITSSAInAcc; //! output hist
8d63376d	52	TH1F *fHistClusterMapITSMI; //! output hist
	53	TH1F *fHistClusterMapITSMIok; //! output hist
	54	TH1F *fHistClusterMapITSMIbad; //! output hist
	55	TH1F *fHistClusterMapITSMIskipped; //! output hist
	56	TH1F *fHistClusterMapITSMIoutinz; //! output hist
	57	TH1F *fHistClusterMapITSMInorefit; //! output hist
	58	TH1F *fHistClusterMapITSMInocls; //! output hist
2308c814	59	TH1F *fHistClusterMapITSMIokoutinzbad; //! output hist
8d63376d	60	TH1F *fHistClusterMapITSSA; //! output hist
	61	TH1F *fHistClusterMapITSSAok; //! output hist
	62	TH1F *fHistClusterMapITSSAbad; //! output hist
	63	TH1F *fHistClusterMapITSSAskipped; //! output hist
	64	TH1F *fHistClusterMapITSSAoutinz; //! output hist
	65	TH1F *fHistClusterMapITSSAnorefit; //! output hist
	66	TH1F *fHistClusterMapITSSAnocls; //! output hist
2308c814	67	TH1F *fHistClusterMapITSSAokoutinzbad; //! output hist
	68	TH1F *fHistClusterMapITSSAInAcc; //! output hist
	69	TH1F *fHistClusterMapITSSAokInAcc; //! output hist
	70	TH1F *fHistClusterMapITSSAbadInAcc; //! output hist
	71	TH1F *fHistClusterMapITSSAskippedInAcc; //! output hist
	72	TH1F *fHistClusterMapITSSAoutinzInAcc; //! output hist
	73	TH1F *fHistClusterMapITSSAnorefitInAcc; //! output hist
	74	TH1F *fHistClusterMapITSSAnoclsInAcc; //! output hist
	75	TH1F *fHistClusterMapITSSAokoutinzbadInAcc; //! output hist
	76	TH1F *fHistClusterMapModuleITSSAokInAcc; //! output hist
	77	TH1F *fHistClusterMapModuleITSSAbadInAcc; //! output hist
	78	TH1F *fHistClusterMapModuleITSSAnoclsInAcc; //! output hist
8d63376d	79	TH1F *fHistPhiTPC; //! output hist
	80	TH1F *fHistPtTPC; //! output hist
	81	TH1F *fHistPtITSMI2; //! output hist
	82	TH1F *fHistPtITSMI3; //! output hist
	83	TH1F *fHistPtITSMI4; //! output hist
	84	TH1F *fHistPtITSMI5; //! output hist
	85	TH1F *fHistPtITSMI6; //! output hist
	86	TH1F *fHistPtITSMISPD; //! output hist
	87	TNtuple *fNtupleESDTracks; //! output ntuple
	88	TNtuple *fNtupleITSAlignExtra; //! output ntuple
	89	TNtuple *fNtupleITSAlignSPDTracklets; //! output ntuple
	90	Int_t fCountsPerPtBin[10]; // track per pt bin
	91
	92	private:
	93
	94	AliAnalysisTaskITSTrackingCheck(const AliAnalysisTaskITSTrackingCheck&); // not implemented
	95	AliAnalysisTaskITSTrackingCheck& operator=(const AliAnalysisTaskITSTrackingCheck&); // not implemented
	96
	97
	98	Int_t NumberOfITSClustersMC(Int_t label) const;
	99	Double_t ParticleImpParMC(TParticle part,AliESDVertex vert,Double_t bzT) const;
	100	Bool_t SelectPt(Double_t pt);
	101
	102	ClassDef(AliAnalysisTaskITSTrackingCheck,1); // ITS tracks analysis
	103	};
	104
	105	#endif