[u/mrichter/AliRoot.git] / PWG3 / muon / AliAnalysisTaskLinkToMC.h

#ifndef ALIANALYSISTASKLINKTOMC_H
#define ALIANALYSISTASKLINKTOMC_H
/* This file is property of and copyright by the ALICE HLT Project        *
 * ALICE Experiment at CERN, All rights reserved.                         *
 * See cxx source for full Copyright notice                               */

///
/// @file   AliAnalysisTaskLinkToMC.h
/// @author Artur Szostak <artursz@iafrica.com>
/// @date   27 Oct 2008
/// @brief  Definition of the AliAnalysisTaskLinkToMC task to connect reconstructed tracks with MC tracks.
///

#include "AliAnalysisTaskSE.h"

class TList;
class TH2D;
class TMap;
class AliESDMuonTrack;
class AliMCParticle;

/**
 * \class AliAnalysisTaskLinkToMC
 * Connects reconstructed tracks found in the ESD to their corresponding Monte
 * Carlo tracks via the proximity of their hit coordinates.
 */
class AliAnalysisTaskLinkToMC : public AliAnalysisTaskSE
{
public:
	AliAnalysisTaskLinkToMC();
	AliAnalysisTaskLinkToMC(const char* name);

	virtual void UserCreateOutputObjects();
	virtual void UserExec(Option_t* option);
	virtual void Terminate(Option_t* option);
	
	/// Returns the absolute maximum distance in centimetres between the
	/// cluster and track reference that is allowed in the X direction.
	Double_t HardCutLimitX() const { return fHardCutLimitX; }
	
	/// Returns the absolute maximum distance in centimetres between the
	/// cluster and track reference that is allowed in the Y direction.
	Double_t HardCutLimitY() const { return fHardCutLimitY; }
	
	/// Returns the absolute maximum distance in centimetres between the
	/// cluster and track reference that is allowed in the Z direction.
	Double_t HardCutLimitZ() const { return fHardCutLimitZ; }
	
	void HardCutLimitX(Double_t value);
	void HardCutLimitY(Double_t value);
	void HardCutLimitZ(Double_t value);
	
	Double_t SigmaCut() const;
	
	/// Sets the maximum sigma value to allow for each X and Y dimension
	/// between clusters and track references.
	void SigmaCut(Double_t value) { fSigmaCut2 = value*value; }
	
	/// Returns the minimum number of clusters that must match between the
	/// ESD track clusters and track references.
	Int_t MinClusters() const { return fMinClusters; }
	
	/// Sets the minimum number of clusters that must match between the ESD
	/// track clusters and track references.
	void MinClusters(Int_t value) { fMinClusters = value; }
	
	/// Returns the minimum number of clusters that must match between the
	/// ESD track clusters and track references in stations 1 and 2.
	Int_t MinClustersInSt12() const { return fMinClustersInSt12; }
	
	/// Sets the minimum number of clusters that must match between the ESD
	/// track clusters and track references in stations 1 and 2.
	void MinClustersInSt12(Int_t value) { fMinClustersInSt12 = value; }
	
	/// Returns the minimum number of clusters that must match between the
	/// ESD track clusters and track references in stations 4 and 5.
	Int_t MinClustersInSt45() const { return fMinClustersInSt45; }
	
	/// Sets the minimum number of clusters that must match between the ESD
	/// track clusters and track references in stations 4 and 5.
	void MinClustersInSt45(Int_t value) { fMinClustersInSt45 = value; }
	
	/// Returns the flag indicating if histograms should be generated for
	/// checking the quality of track matching.
	Bool_t GenerateHistograms() const { return fMakeHists; }
	
	/// Set the flag indicating if histograms should be generated for checking
	/// the quality of track matching.
	void GenerateHistograms(Bool_t value) { fMakeHists = value; }
	
	/// Returns the flag indicating if the Z coordinate is to be used for
	/// track matching or not.
	Bool_t UseZCoordinate() const { return fUseZCoordinate; }
	
	/// Set the flag indicating if the Z coordinate should be used for track
	/// matching.
	void UseZCoordinate(Bool_t value) { fUseZCoordinate = value; }
	
	Bool_t ChamberMustMatch(Int_t chamber) const;
	void ChamberMustMatch(Int_t chamber, Bool_t value);
	Bool_t StationMustMatch(Int_t station) const;
	void StationMustMatch(Int_t station, Bool_t value);

private:

	/// Do not allow copying of this object. Method is not implemented.
	AliAnalysisTaskLinkToMC(const AliAnalysisTaskLinkToMC& obj);
	/// Do not allow copying of this object. Method is not implemented.
	AliAnalysisTaskLinkToMC& operator = (const AliAnalysisTaskLinkToMC& obj);
	
	void FindLinksToMC(TMap& links);
	void CalculateTrackCompatibility(
			AliESDMuonTrack* esdTrack, AliMCParticle* mcTrack,
			bool& tracksCompatible, Double_t& chi2, Int_t& noOfClusters
		);
	
	bool IsFindable(AliMCParticle* track) const;
	
	void FillHistsFromMC();
	void FillHistsFromLinks(TMap& links);
	void CreateAODTracks(TMap& links);

	TList* fHistos;  //! List of all histograms produced by this analysis task.
	TH2D* fFindableHist;  //! Histogram of reconstructible tracks (filled with MC information).
	TH2D* fFoundHistMC;  //! Histogram of reconstructed tracks (filled with MC information).
	TH2D* fFoundHist;  //! Histogram of reconstructed tracks (filled with reconstructed information).
	TH2D* fFakeHist;  //! Histogram of fake tracks (filled with reconstructed information).
	
	Double_t fHardCutLimitX; //! The absolute maximum difference between a cluster and reference track point in X direction in centimetres.
	Double_t fHardCutLimitY; //! The absolute maximum difference between a cluster and reference track point in Y direction in centimetres.
	Double_t fHardCutLimitZ; //! The absolute maximum difference between a cluster and reference track point in Z direction in centimetres.
	Double_t fVarX; //! The variance to use in the X direction if error value in the cluster was not set.
	Double_t fVarY; //! The variance to use in the Y direction if error value in the cluster was not set.
	Double_t fSigmaCut2; //! The cut on the chi2 X/Y direction component to make for each cluster fit.
	Int_t fMinClusters; //! Minimum number of clusters that must match between an ESD and MC track for it to be considered compatible.
	Int_t fMinClustersInSt12;  //! The minimum number of clusters that must match in stations 1 and 2.
	Int_t fMinClustersInSt45;  //! The minimum number of clusters that must match in stations 4 and 5.
	Bool_t fMakeHists;  //! Flag indicating if histogram filling and generation should be performed.
	Bool_t fUseZCoordinate;  //! Flag indicating if the Z coordinate should be used for comparrison.
	Bool_t fChamberMustMatch[14];  //! A flag for each chamber which must have a compatible hit matched.
	Bool_t fStationMustMatch[7];  //! A flag for each station which must have at least one compatible hit matched.

	ClassDef(AliAnalysisTaskLinkToMC, 0) // Analysis task to connect reconstructed tracks to their Monte Carlo counterparts.
};

#endif // ALIANALYSISTASKLINKTOMC_H
Commit	Line	Data
4c904d93	1	#ifndef ALIANALYSISTASKLINKTOMC_H
	2	#define ALIANALYSISTASKLINKTOMC_H
	3	/* This file is property of and copyright by the ALICE HLT Project *
	4	* ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	///
	8	/// @file AliAnalysisTaskLinkToMC.h
	9	/// @author Artur Szostak <artursz@iafrica.com>
	10	/// @date 27 Oct 2008
	11	/// @brief Definition of the AliAnalysisTaskLinkToMC task to connect reconstructed tracks with MC tracks.
	12	///
	13
	14	#include "AliAnalysisTaskSE.h"
	15
	16	class TList;
	17	class TH2D;
	18	class TMap;
	19	class AliESDMuonTrack;
	20	class AliMCParticle;
	21
	22	/**
	23	* \class AliAnalysisTaskLinkToMC
	24	* Connects reconstructed tracks found in the ESD to their corresponding Monte
	25	* Carlo tracks via the proximity of their hit coordinates.
	26	*/
	27	class AliAnalysisTaskLinkToMC : public AliAnalysisTaskSE
	28	{
	29	public:
	30	AliAnalysisTaskLinkToMC();
	31	AliAnalysisTaskLinkToMC(const char* name);
	32
	33	virtual void UserCreateOutputObjects();
	34	virtual void UserExec(Option_t* option);
	35	virtual void Terminate(Option_t* option);
	36
	37	/// Returns the absolute maximum distance in centimetres between the
	38	/// cluster and track reference that is allowed in the X direction.
	39	Double_t HardCutLimitX() const { return fHardCutLimitX; }
	40
	41	/// Returns the absolute maximum distance in centimetres between the
	42	/// cluster and track reference that is allowed in the Y direction.
	43	Double_t HardCutLimitY() const { return fHardCutLimitY; }
	44
	45	/// Returns the absolute maximum distance in centimetres between the
	46	/// cluster and track reference that is allowed in the Z direction.
	47	Double_t HardCutLimitZ() const { return fHardCutLimitZ; }
	48
	49	void HardCutLimitX(Double_t value);
	50	void HardCutLimitY(Double_t value);
	51	void HardCutLimitZ(Double_t value);
	52
	53	Double_t SigmaCut() const;
	54
	55	/// Sets the maximum sigma value to allow for each X and Y dimension
	56	/// between clusters and track references.
	57	void SigmaCut(Double_t value) { fSigmaCut2 = value*value; }
	58
	59	/// Returns the minimum number of clusters that must match between the
	60	/// ESD track clusters and track references.
	61	Int_t MinClusters() const { return fMinClusters; }
	62
	63	/// Sets the minimum number of clusters that must match between the ESD
	64	/// track clusters and track references.
65	void MinClusters(Int_t value) { fMinClusters = value; }
66
67	/// Returns the minimum number of clusters that must match between the
68	/// ESD track clusters and track references in stations 1 and 2.
69	Int_t MinClustersInSt12() const { return fMinClustersInSt12; }
70
71	/// Sets the minimum number of clusters that must match between the ESD
72	/// track clusters and track references in stations 1 and 2.
73	void MinClustersInSt12(Int_t value) { fMinClustersInSt12 = value; }
74
75	/// Returns the minimum number of clusters that must match between the
76	/// ESD track clusters and track references in stations 4 and 5.
77	Int_t MinClustersInSt45() const { return fMinClustersInSt45; }
78
79	/// Sets the minimum number of clusters that must match between the ESD
80	/// track clusters and track references in stations 4 and 5.
81	void MinClustersInSt45(Int_t value) { fMinClustersInSt45 = value; }
82
83	/// Returns the flag indicating if histograms should be generated for
84	/// checking the quality of track matching.
85	Bool_t GenerateHistograms() const { return fMakeHists; }
86
87	/// Set the flag indicating if histograms should be generated for checking
88	/// the quality of track matching.
89	void GenerateHistograms(Bool_t value) { fMakeHists = value; }
90
91	/// Returns the flag indicating if the Z coordinate is to be used for
92	/// track matching or not.
93	Bool_t UseZCoordinate() const { return fUseZCoordinate; }
94
95	/// Set the flag indicating if the Z coordinate should be used for track
96	/// matching.
97	void UseZCoordinate(Bool_t value) { fUseZCoordinate = value; }
98
99	Bool_t ChamberMustMatch(Int_t chamber) const;
100	void ChamberMustMatch(Int_t chamber, Bool_t value);
101	Bool_t StationMustMatch(Int_t station) const;
102	void StationMustMatch(Int_t station, Bool_t value);
103
104	private:
105
106	/// Do not allow copying of this object. Method is not implemented.
107	AliAnalysisTaskLinkToMC(const AliAnalysisTaskLinkToMC& obj);
108	/// Do not allow copying of this object. Method is not implemented.
109	AliAnalysisTaskLinkToMC& operator = (const AliAnalysisTaskLinkToMC& obj);
110
111	void FindLinksToMC(TMap& links);
112	void CalculateTrackCompatibility(
113	AliESDMuonTrack* esdTrack, AliMCParticle* mcTrack,
114	bool& tracksCompatible, Double_t& chi2, Int_t& noOfClusters
115	);
116
117	bool IsFindable(AliMCParticle* track) const;
118
119	void FillHistsFromMC();
120	void FillHistsFromLinks(TMap& links);
121	void CreateAODTracks(TMap& links);
122
123	TList* fHistos; //! List of all histograms produced by this analysis task.
124	TH2D* fFindableHist; //! Histogram of reconstructible tracks (filled with MC information).
125	TH2D* fFoundHistMC; //! Histogram of reconstructed tracks (filled with MC information).
126	TH2D* fFoundHist; //! Histogram of reconstructed tracks (filled with reconstructed information).
127	TH2D* fFakeHist; //! Histogram of fake tracks (filled with reconstructed information).
128
129	Double_t fHardCutLimitX; //! The absolute maximum difference between a cluster and reference track point in X direction in centimetres.
130	Double_t fHardCutLimitY; //! The absolute maximum difference between a cluster and reference track point in Y direction in centimetres.
131	Double_t fHardCutLimitZ; //! The absolute maximum difference between a cluster and reference track point in Z direction in centimetres.
132	Double_t fVarX; //! The variance to use in the X direction if error value in the cluster was not set.
133	Double_t fVarY; //! The variance to use in the Y direction if error value in the cluster was not set.
134	Double_t fSigmaCut2; //! The cut on the chi2 X/Y direction component to make for each cluster fit.
135	Int_t fMinClusters; //! Minimum number of clusters that must match between an ESD and MC track for it to be considered compatible.
136	Int_t fMinClustersInSt12; //! The minimum number of clusters that must match in stations 1 and 2.
137	Int_t fMinClustersInSt45; //! The minimum number of clusters that must match in stations 4 and 5.
138	Bool_t fMakeHists; //! Flag indicating if histogram filling and generation should be performed.
139	Bool_t fUseZCoordinate; //! Flag indicating if the Z coordinate should be used for comparrison.
140	Bool_t fChamberMustMatch[14]; //! A flag for each chamber which must have a compatible hit matched.
141	Bool_t fStationMustMatch[7]; //! A flag for each station which must have at least one compatible hit matched.
142
805e576c	143	ClassDef(AliAnalysisTaskLinkToMC, 0) // Analysis task to connect reconstructed tracks to their Monte Carlo counterparts.
4c904d93	144	};
	145
	146	#endif // ALIANALYSISTASKLINKTOMC_H
	147