[u/mrichter/AliRoot.git] / MFT / AliMuonForwardTrackAnalysis.h

#ifndef AliMuonForwardTrackAnalysis_H
#define AliMuonForwardTrackAnalysis_H 

//====================================================================================================================================================
//
//      Class for the analysis of the ALICE muon forward tracks (MUON + MFT)
//
//      Contact author: antonio.uras@cern.ch
//
//====================================================================================================================================================

#include "TObject.h"
#include "TClonesArray.h"
#include "AliMuonForwardTrack.h"
#include "TMatrixD.h"
#include "TTree.h"
#include "TH1D.h"
#include "TH2D.h"
#include "AliLog.h"
#include "TFile.h"
#include "TParticle.h"
#include "AliMUONTrackParam.h"
#include "AliMUONTrackExtrap.h"
#include "TDatabasePDG.h"
#include "TGraph.h"
#include "TAxis.h"

//====================================================================================================================================================

class AliMuonForwardTrackAnalysis : public TObject {

public:

  enum {kNoOption, kPionsKaons, kResonanceOnly};
  
  AliMuonForwardTrackAnalysis();
  
  virtual ~AliMuonForwardTrackAnalysis() {};  // destructor

  Bool_t Init(Char_t *inputFileName);
  Bool_t LoadNextEvent();
  void Terminate(Char_t *outputFileName);

  void BookHistos();

  void SetInputDir(Char_t *inputDir)   { fInputDir  = inputDir; }
  void SetOutputDir(Char_t *outputDir) { fOutputDir = outputDir; }

  Int_t GetNTracksAnalyzed() { return fNTracksAnalyzed; }

  void SetMassRange(Int_t nBins, Double_t min, Double_t max) { fNMassBins=nBins; fMassMin=min; fMassMax=max; }
  void SetPtDimuRange(Int_t nBins, Double_t min, Double_t max) { fNPtDimuBins=TMath::Min(nBins,fNMaxPtBinsDimuons) ; fPtDimuMin=min; fPtDimuMax=max; }
  void SetSingleMuonAnalysis(Bool_t singleMuonAnalysis) { fSingleMuonAnalysis = singleMuonAnalysis; }
  void SetMuonPairAnalysis(Bool_t muonPairAnalysis) { fMuonPairAnalysis = muonPairAnalysis; }
  void SetMatchTrigger(Bool_t matchTrigger) { fMatchTrigger = matchTrigger; }

  Bool_t AnalyzeSingleMuon();
  Bool_t AnalyzeMuonPair();
  void BuildMuonPairs();

  void EvalDimuonVtxResolution(Bool_t eval) { fEvalDimuonVtxResolution = eval; }

  Bool_t PassedCutSingleMuon(AliMuonForwardTrack *track);
  Bool_t PassedCutMuonPair(AliMuonForwardTrackPair *pair);

  void SetVertResMC(Double_t xRes, Double_t yRes, Double_t zRes) { fXVertResMC=xRes; fYVertResMC=yRes; fZVertResMC=zRes; }

  void SetOption(Int_t option) { fOption = option; }
  void SetMaxNWrongClustersMC(Int_t nClusters) { fMaxNWrongClustersMC = nClusters; }
  void SetPtMinSingleMuons(Double_t ptMin) { fPtMinSingleMuons = ptMin; }

  void ReadEvents(Int_t firstEv, Int_t lastEv) { fFirstEvent = firstEv; fLastEvent = lastEv; }
  Int_t GetFirstEvent() { return fFirstEvent; }
  Int_t GetLastEvent()  { return fLastEvent; }

  void UseBransonForCut(Bool_t useBranson) { fUseBransonForCut = useBranson; }
  void UseBransonForKinematics(Bool_t useBranson) { fUseBransonForKinematics = useBranson; }
  void UseCutOnOffsetChi2(Bool_t useCut) { fCutOnOffsetChi2 = useCut; }

private:

  static const Int_t fNMaxPtBinsDimuons = 50;

  TString fInputDir, fOutputDir;

  TTree *fInputTreeWithBranson, *fInputTreeWithoutBranson;  //!

  TClonesArray *fMuonForwardTracksWithBranson,    *fMuonForwardTrackPairsWithBranson;                //!
  TClonesArray *fMuonForwardTracksWithoutBranson, *fMuonForwardTrackPairsWithoutBranson;             //!
  AliMuonForwardTrack *fMFTTrackWithBranson, *fMFTTrackWithoutBranson, *fMFTTrack;                   //!
  AliMuonForwardTrackPair *fMFTTrackPairWithBranson, *fMFTTrackPairWithoutBranson, *fMFTTrackPair;   //!
  TParticle *fMCRefTrack;                                                                            //!

  Int_t fEv, fFirstEvent, fLastEvent, fNTracksOfEvent, fNTracksAnalyzedOfEvent, fNTracksAnalyzed, fNPairsOfEvent, fNPairsAnalyzedOfEvent;
  Int_t fNTracksAnalyzedOfEventAfterCut, fNPairsAnalyzedOfEventAfterCut;
  
  TH1D *fHistOffsetSingleMuonsX, *fHistOffsetSingleMuonsY, *fHistOffsetSingleMuons, *fHistWOffsetSingleMuons;      //!
  TH1D *fHistErrorSingleMuonsX, *fHistErrorSingleMuonsY;                                                           //!
  TH1D *fHistZOriginSingleMuonsMC;
  
  TH2D *fHistZROriginSingleMuonsMC, *fHistSingleMuonsPtRapidityMC;                                                //! 
  TH2D *fHistSingleMuonsOffsetChi2, *fHistSingleMuonsOffsetChi2_BeforeMFT, *fHistSingleMuonsOffsetChi2_AfterMFT;  //!
  TGraph *fGraphSingleMuonsOffsetChi2;                                                                            //!

  TH1D *fHistMassMuonPairs[fNMaxPtBinsDimuons+1];                                  //!
  TH1D *fHistMassMuonPairsWithoutMFT[fNMaxPtBinsDimuons+1];                        //!
  TH1D *fHistMassMuonPairsMC[fNMaxPtBinsDimuons+1];                                //!
  TH1D *fHistWOffsetMuonPairsAtPrimaryVtx[fNMaxPtBinsDimuons+1];                   //!
  TH1D *fHistWOffsetMuonPairsAtPCA[fNMaxPtBinsDimuons+1];                          //!
  TH1D *fHistDistancePrimaryVtxPCA[fNMaxPtBinsDimuons+1];                          //!
  TH2D *fHistDistancePrimaryVtxPCAvsWOffsetMuonPairsAtPCA[fNMaxPtBinsDimuons+1];   //!
  TH1D *fHistDimuonVtxResolutionX[fNMaxPtBinsDimuons+1];                           //!
  TH1D *fHistDimuonVtxResolutionY[fNMaxPtBinsDimuons+1];                           //!
  TH1D *fHistDimuonVtxResolutionZ[fNMaxPtBinsDimuons+1];                           //!

  TH2D *fHistRapidityPtMuonPairs;           //!
 
  TH2D *fHistMassMuonPairsVsPt;             //!
  TH2D *fHistMassMuonPairsWithoutMFTVsPt;   //!
	TH2D *fHistMassMuonPairsVsPtLSp;             //!
	TH2D *fHistMassMuonPairsWithoutMFTVsPtLSp;   //!
	TH2D *fHistMassMuonPairsVsPtLSm;             //!
	TH2D *fHistMassMuonPairsWithoutMFTVsPtLSm;   //!
	
  Bool_t fEvalDimuonVtxResolution;

  Int_t fNMassBins, fNPtDimuBins;
  Double_t fMassMin, fMassMax, fPtDimuMin, fPtDimuMax;
  TAxis *fPtAxisDimuons;

  Bool_t fSingleMuonAnalysis, fMuonPairAnalysis, fMatchTrigger;
  Int_t fOption;

  Double_t fXVertResMC, fYVertResMC, fZVertResMC;
  Double_t fPrimaryVtxX, fPrimaryVtxY, fPrimaryVtxZ;
  Int_t fMaxNWrongClustersMC;
  Double_t fPtMinSingleMuons;

  Bool_t fUseBransonForCut, fUseBransonForKinematics, fCutOnOffsetChi2;

  Double_t fCenterOffset, fCenterChi2, fScaleOffset, fScaleChi2, fRadiusCut;

  ClassDef(AliMuonForwardTrackAnalysis, 1)

};

//====================================================================================================================================================
	
#endif
Commit	Line	Data
e806e863	1	#ifndef AliMuonForwardTrackAnalysis_H
	2	#define AliMuonForwardTrackAnalysis_H
	3
	4	//====================================================================================================================================================
	5	//
	6	// Class for the analysis of the ALICE muon forward tracks (MUON + MFT)
	7	//
	8	// Contact author: antonio.uras@cern.ch
	9	//
	10	//====================================================================================================================================================
	11
	12	#include "TObject.h"
	13	#include "TClonesArray.h"
	14	#include "AliMuonForwardTrack.h"
	15	#include "TMatrixD.h"
	16	#include "TTree.h"
	17	#include "TH1D.h"
	18	#include "TH2D.h"
	19	#include "AliLog.h"
	20	#include "TFile.h"
	21	#include "TParticle.h"
	22	#include "AliMUONTrackParam.h"
	23	#include "AliMUONTrackExtrap.h"
	24	#include "TDatabasePDG.h"
b5ab1ac4	25	#include "TGraph.h"
cd2f51d2	26	#include "TAxis.h"
e806e863	27
	28	//====================================================================================================================================================
	29
	30	class AliMuonForwardTrackAnalysis : public TObject {
	31
	32	public:
	33
d8c2cc3e	34	enum {kNoOption, kPionsKaons, kResonanceOnly};
e806e863	35
	36	AliMuonForwardTrackAnalysis();
	37
	38	virtual ~AliMuonForwardTrackAnalysis() {}; // destructor
	39
	40	Bool_t Init(Char_t *inputFileName);
	41	Bool_t LoadNextEvent();
	42	void Terminate(Char_t *outputFileName);
	43
	44	void BookHistos();
	45
	46	void SetInputDir(Char_t *inputDir) { fInputDir = inputDir; }
	47	void SetOutputDir(Char_t *outputDir) { fOutputDir = outputDir; }
	48
	49	Int_t GetNTracksAnalyzed() { return fNTracksAnalyzed; }
	50
	51	void SetMassRange(Int_t nBins, Double_t min, Double_t max) { fNMassBins=nBins; fMassMin=min; fMassMax=max; }
cd2f51d2	52	void SetPtDimuRange(Int_t nBins, Double_t min, Double_t max) { fNPtDimuBins=TMath::Min(nBins,fNMaxPtBinsDimuons) ; fPtDimuMin=min; fPtDimuMax=max; }
e806e863	53	void SetSingleMuonAnalysis(Bool_t singleMuonAnalysis) { fSingleMuonAnalysis = singleMuonAnalysis; }
	54	void SetMuonPairAnalysis(Bool_t muonPairAnalysis) { fMuonPairAnalysis = muonPairAnalysis; }
	55	void SetMatchTrigger(Bool_t matchTrigger) { fMatchTrigger = matchTrigger; }
	56
	57	Bool_t AnalyzeSingleMuon();
	58	Bool_t AnalyzeMuonPair();
	59	void BuildMuonPairs();
	60
789c2e43	61	void EvalDimuonVtxResolution(Bool_t eval) { fEvalDimuonVtxResolution = eval; }
789c2e43	62
7e3dd1af	63	Bool_t PassedCutSingleMuon(AliMuonForwardTrack *track);
	64	Bool_t PassedCutMuonPair(AliMuonForwardTrackPair *pair);
	65
e806e863	66	void SetVertResMC(Double_t xRes, Double_t yRes, Double_t zRes) { fXVertResMC=xRes; fYVertResMC=yRes; fZVertResMC=zRes; }
	67
	68	void SetOption(Int_t option) { fOption = option; }
	69	void SetMaxNWrongClustersMC(Int_t nClusters) { fMaxNWrongClustersMC = nClusters; }
	70	void SetPtMinSingleMuons(Double_t ptMin) { fPtMinSingleMuons = ptMin; }
	71
	72	void ReadEvents(Int_t firstEv, Int_t lastEv) { fFirstEvent = firstEv; fLastEvent = lastEv; }
	73	Int_t GetFirstEvent() { return fFirstEvent; }
	74	Int_t GetLastEvent() { return fLastEvent; }
	75
7e3dd1af	76	void UseBransonForCut(Bool_t useBranson) { fUseBransonForCut = useBranson; }
	77	void UseBransonForKinematics(Bool_t useBranson) { fUseBransonForKinematics = useBranson; }
	78	void UseCutOnOffsetChi2(Bool_t useCut) { fCutOnOffsetChi2 = useCut; }
	79
e806e863	80	private:
e806e863	81
cd2f51d2	82	static const Int_t fNMaxPtBinsDimuons = 50;
e806e863	83
	84	TString fInputDir, fOutputDir;
	85
7e3dd1af	86	TTree fInputTreeWithBranson, fInputTreeWithoutBranson; //!
e806e863	87
7e3dd1af	88	TClonesArray fMuonForwardTracksWithBranson, fMuonForwardTrackPairsWithBranson; //!
	89	TClonesArray fMuonForwardTracksWithoutBranson, fMuonForwardTrackPairsWithoutBranson; //!
	90	AliMuonForwardTrack fMFTTrackWithBranson, fMFTTrackWithoutBranson, *fMFTTrack; //!
	91	AliMuonForwardTrackPair fMFTTrackPairWithBranson, fMFTTrackPairWithoutBranson, *fMFTTrackPair; //!
	92	TParticle *fMCRefTrack; //!
e806e863	93
e806e863	94	Int_t fEv, fFirstEvent, fLastEvent, fNTracksOfEvent, fNTracksAnalyzedOfEvent, fNTracksAnalyzed, fNPairsOfEvent, fNPairsAnalyzedOfEvent;
d8c2cc3e	95	Int_t fNTracksAnalyzedOfEventAfterCut, fNPairsAnalyzedOfEventAfterCut;
e806e863	96
	97	TH1D fHistOffsetSingleMuonsX, fHistOffsetSingleMuonsY, fHistOffsetSingleMuons, fHistWOffsetSingleMuons; //!
	98	TH1D fHistErrorSingleMuonsX, fHistErrorSingleMuonsY; //!
d8c2cc3e	99	TH1D *fHistZOriginSingleMuonsMC;
	100
	101	TH2D fHistZROriginSingleMuonsMC, fHistSingleMuonsPtRapidityMC; //!
	102	TH2D fHistSingleMuonsOffsetChi2, fHistSingleMuonsOffsetChi2_BeforeMFT, *fHistSingleMuonsOffsetChi2_AfterMFT; //!
	103	TGraph *fGraphSingleMuonsOffsetChi2; //!
cd2f51d2	104
789c2e43	105	TH1D *fHistMassMuonPairs[fNMaxPtBinsDimuons+1]; //!
	106	TH1D *fHistMassMuonPairsWithoutMFT[fNMaxPtBinsDimuons+1]; //!
	107	TH1D *fHistMassMuonPairsMC[fNMaxPtBinsDimuons+1]; //!
	108	TH1D *fHistWOffsetMuonPairsAtPrimaryVtx[fNMaxPtBinsDimuons+1]; //!
	109	TH1D *fHistWOffsetMuonPairsAtPCA[fNMaxPtBinsDimuons+1]; //!
	110	TH1D *fHistDistancePrimaryVtxPCA[fNMaxPtBinsDimuons+1]; //!
	111	TH2D *fHistDistancePrimaryVtxPCAvsWOffsetMuonPairsAtPCA[fNMaxPtBinsDimuons+1]; //!
	112	TH1D *fHistDimuonVtxResolutionX[fNMaxPtBinsDimuons+1]; //!
	113	TH1D *fHistDimuonVtxResolutionY[fNMaxPtBinsDimuons+1]; //!
	114	TH1D *fHistDimuonVtxResolutionZ[fNMaxPtBinsDimuons+1]; //!
	115
	116	TH2D *fHistRapidityPtMuonPairs; //!
e806e863	117
789c2e43	118	TH2D *fHistMassMuonPairsVsPt; //!
	119	TH2D *fHistMassMuonPairsWithoutMFTVsPt; //!
	120	TH2D *fHistMassMuonPairsVsPtLSp; //!
	121	TH2D *fHistMassMuonPairsWithoutMFTVsPtLSp; //!
	122	TH2D *fHistMassMuonPairsVsPtLSm; //!
	123	TH2D *fHistMassMuonPairsWithoutMFTVsPtLSm; //!
	124
	125	Bool_t fEvalDimuonVtxResolution;
	126
cd2f51d2	127	Int_t fNMassBins, fNPtDimuBins;
	128	Double_t fMassMin, fMassMax, fPtDimuMin, fPtDimuMax;
	129	TAxis *fPtAxisDimuons;
e806e863	130
	131	Bool_t fSingleMuonAnalysis, fMuonPairAnalysis, fMatchTrigger;
	132	Int_t fOption;
	133
	134	Double_t fXVertResMC, fYVertResMC, fZVertResMC;
7e3dd1af	135	Double_t fPrimaryVtxX, fPrimaryVtxY, fPrimaryVtxZ;
e806e863	136	Int_t fMaxNWrongClustersMC;
	137	Double_t fPtMinSingleMuons;
	138
7e3dd1af	139	Bool_t fUseBransonForCut, fUseBransonForKinematics, fCutOnOffsetChi2;
	140
	141	Double_t fCenterOffset, fCenterChi2, fScaleOffset, fScaleChi2, fRadiusCut;
	142
e806e863	143	ClassDef(AliMuonForwardTrackAnalysis, 1)
	144
	145	};
	146
	147	//====================================================================================================================================================
	148
	149	#endif
	150