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

// #include "AliMuonForwardTrackAnalysis.h"
// #include "TDatabasePDG.h"
// #include "TGeoGlobalMagField.h"
// #include "TROOT.h"
// #include "AliMagF.h"

enum {kNoOption, kResonanceOnly, kCharmOnly, kBeautyOnly, kBackground1mu, kBackground2mu, kNoResonances};

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

void AliMuonForwardTrackAnalysis(const Char_t *readDir= ".",                       // the directory with the MuonGlobalTracks.root and geometry.root files
				 Int_t option = kNoOption,                         // for resonance analysis: kResonanceOnly 
				 Double_t massMin = 0.,                            // lower limit for the cut on dimuon mass
				 Double_t massMax = 10.,                           // upper limit for the cut on dimuon mass
				 Double_t maxChi2SingleMuons = 1.5,                // upper limit for the cut on the single muon chi2
				 Double_t maxOffsetSingleMuons = 250.,             // upper limit for the cut on the single muon offset w.r.t. the primary vtx
				 Bool_t correlateCutOnOffsetChi2 = kTRUE,          // if true, the cur region in the chi2-offset plane for single muons is a quadrant aorund the origin
				 Double_t maxWOffsetMuonPairsAtPrimaryVtx = 1.e9,  // upper limit for the cut on weighted offset of dimuond w.r.t. the primary vtx
				 Double_t maxWOffsetMuonPairsAtPCA = 1.e9,         // upper limit for the cut on weighted offset of dimuond w.r.t. their PCA
				 Double_t maxDistancePrimaryVtxPCA = 1.e9,         // upper limit for the cut on the distance between primary vtx and PCA
				 Double_t minPCAQuality = 0.,                      // lower limit for the cut on the PCA quality
				 Int_t triggerLevel = 1,                           // level of the trigger both muons must satisfy
				 Int_t maxNWrongClusters = 999,                    // maximum number of wrong MFT clusters for a global muon track
				 const Char_t *outDir = ".",                       // directory where the output file will be created
				 Bool_t singleMuonAnalysis = kTRUE,                // if true, the aalysis of single muons will be performed
				 Bool_t muonPairAnalysis = kTRUE,                  // if true, the aalysis of muon pairs will be performed
				 Int_t firstEvent = -1,
				 Int_t lastEvent = -1, 
				 Int_t myRandom = 0,                               // number which will tag the name of the output file
				 Double_t ptMinSingleMuons = 0.0,                  // lower limit for the cut on the single muon pt
				 Double_t trueMass = 0.0,                          // used to evaluate the pseudo proper decay length, usually for J/psi only
				 Bool_t evalDimuonVtxResolution=kFALSE,            // to be set true only if prompt dimuon sources are analyzed
				 Int_t nEventsToMix = 0) {                         // if <1 or >100, mixing is not performed
  
  const Double_t mJpsi = TDatabasePDG::Instance()->GetParticle("J/psi")->Mass();

  if (trueMass<0) trueMass = mJpsi;
 
  gROOT -> LoadMacro("./AliMuonForwardTrackAnalysis.cxx+");
  //  AliLog::SetClassDebugLevel("AliMuonForwardTrackPair", 1);
  //  AliLog::SetClassDebugLevel("AliMuonForwardTrackAnalysis", 2);

  TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG));
  
  AliMuonForwardTrackAnalysis *myAnalysis = new AliMuonForwardTrackAnalysis();
  myAnalysis->ReadEvents(firstEvent, lastEvent);
  myAnalysis->SetInputDir(readDir);
  myAnalysis->SetOutputDir(outDir);
  myAnalysis->SetMassRange(massMin, massMax);
  myAnalysis->SetTrueMass(trueMass);
  myAnalysis->SetSingleMuonAnalysis(singleMuonAnalysis);
  myAnalysis->SetMuonPairAnalysis(muonPairAnalysis);
  myAnalysis->SetOption(option);

  myAnalysis->SetMaxNWrongClustersMC(maxNWrongClusters);
  myAnalysis->SetMinPtSingleMuons(ptMinSingleMuons);
  myAnalysis->SetMaxChi2SingleMuons(maxChi2SingleMuons);
  myAnalysis->SetMaxOffsetSingleMuons(maxOffsetSingleMuons);
  myAnalysis->CorrelateCutOnOffsetChi2(correlateCutOnOffsetChi2);

  myAnalysis->SetMaxWOffsetMuonPairsAtPrimaryVtx(maxWOffsetMuonPairsAtPrimaryVtx);
  myAnalysis->SetMaxWOffsetMuonPairsAtPCA(maxWOffsetMuonPairsAtPCA);
  myAnalysis->SetMaxDistancePrimaryVtxPCA(maxDistancePrimaryVtxPCA);
  myAnalysis->SetMinPCAQuality(minPCAQuality);

  myAnalysis->SetMatchTrigger(triggerLevel);

  myAnalysis->EvalDimuonVtxResolution(evalDimuonVtxResolution);    // it should be true only with prompt dimuon sources

  myAnalysis->SetNEventsToMix(nEventsToMix);

  if (myAnalysis->Init("MuonGlobalTracks.root")) {
    while (myAnalysis->LoadNextEvent()) continue;
    myAnalysis->Terminate(Form("outFiles/outFile.%d.%d.%d.root", myAnalysis->GetFirstEvent(), myAnalysis->GetLastEvent(), myRandom));
  }

}

//================================================================================================================================
Commit	Line	Data
6240bf36	1	// #include "AliMuonForwardTrackAnalysis.h"
	2	// #include "TDatabasePDG.h"
	3	// #include "TGeoGlobalMagField.h"
	4	// #include "TROOT.h"
	5	// #include "AliMagF.h"
cabfe25b	6
88ca07e7	7	enum {kNoOption, kResonanceOnly, kCharmOnly, kBeautyOnly, kBackground1mu, kBackground2mu, kNoResonances};
88ca07e7	8
cabfe25b	9	//=============================================================================================================================================================
e806e863	10
cabfe25b	11	void AliMuonForwardTrackAnalysis(const Char_t *readDir= ".", // the directory with the MuonGlobalTracks.root and geometry.root files
	12	Int_t option = kNoOption, // for resonance analysis: kResonanceOnly
	13	Double_t massMin = 0., // lower limit for the cut on dimuon mass
	14	Double_t massMax = 10., // upper limit for the cut on dimuon mass
	15	Double_t maxChi2SingleMuons = 1.5, // upper limit for the cut on the single muon chi2
	16	Double_t maxOffsetSingleMuons = 250., // upper limit for the cut on the single muon offset w.r.t. the primary vtx
	17	Bool_t correlateCutOnOffsetChi2 = kTRUE, // if true, the cur region in the chi2-offset plane for single muons is a quadrant aorund the origin
	18	Double_t maxWOffsetMuonPairsAtPrimaryVtx = 1.e9, // upper limit for the cut on weighted offset of dimuond w.r.t. the primary vtx
	19	Double_t maxWOffsetMuonPairsAtPCA = 1.e9, // upper limit for the cut on weighted offset of dimuond w.r.t. their PCA
	20	Double_t maxDistancePrimaryVtxPCA = 1.e9, // upper limit for the cut on the distance between primary vtx and PCA
	21	Double_t minPCAQuality = 0., // lower limit for the cut on the PCA quality
	22	Int_t triggerLevel = 1, // level of the trigger both muons must satisfy
	23	Int_t maxNWrongClusters = 999, // maximum number of wrong MFT clusters for a global muon track
	24	const Char_t *outDir = ".", // directory where the output file will be created
	25	Bool_t singleMuonAnalysis = kTRUE, // if true, the aalysis of single muons will be performed
	26	Bool_t muonPairAnalysis = kTRUE, // if true, the aalysis of muon pairs will be performed
e806e863	27	Int_t firstEvent = -1,
e806e863	28	Int_t lastEvent = -1,
cabfe25b	29	Int_t myRandom = 0, // number which will tag the name of the output file
cabfe25b	30	Double_t ptMinSingleMuons = 0.0, // lower limit for the cut on the single muon pt
6240bf36	31	Double_t trueMass = 0.0, // used to evaluate the pseudo proper decay length, usually for J/psi only
	32	Bool_t evalDimuonVtxResolution=kFALSE, // to be set true only if prompt dimuon sources are analyzed
	33	Int_t nEventsToMix = 0) { // if <1 or >100, mixing is not performed
e806e863	34
6240bf36	35	const Double_t mJpsi = TDatabasePDG::Instance()->GetParticle("J/psi")->Mass();
	36
	37	if (trueMass<0) trueMass = mJpsi;
	38
e806e863	39	gROOT -> LoadMacro("./AliMuonForwardTrackAnalysis.cxx+");
e806e863	40	// AliLog::SetClassDebugLevel("AliMuonForwardTrackPair", 1);
6240bf36	41	// AliLog::SetClassDebugLevel("AliMuonForwardTrackAnalysis", 2);
e806e863	42
e806e863	43	TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG));
	44
	45	AliMuonForwardTrackAnalysis *myAnalysis = new AliMuonForwardTrackAnalysis();
	46	myAnalysis->ReadEvents(firstEvent, lastEvent);
	47	myAnalysis->SetInputDir(readDir);
	48	myAnalysis->SetOutputDir(outDir);
cabfe25b	49	myAnalysis->SetMassRange(massMin, massMax);
cabfe25b	50	myAnalysis->SetTrueMass(trueMass);
e806e863	51	myAnalysis->SetSingleMuonAnalysis(singleMuonAnalysis);
	52	myAnalysis->SetMuonPairAnalysis(muonPairAnalysis);
	53	myAnalysis->SetOption(option);
cabfe25b	54
e806e863	55	myAnalysis->SetMaxNWrongClustersMC(maxNWrongClusters);
cabfe25b	56	myAnalysis->SetMinPtSingleMuons(ptMinSingleMuons);
	57	myAnalysis->SetMaxChi2SingleMuons(maxChi2SingleMuons);
	58	myAnalysis->SetMaxOffsetSingleMuons(maxOffsetSingleMuons);
	59	myAnalysis->CorrelateCutOnOffsetChi2(correlateCutOnOffsetChi2);
	60
	61	myAnalysis->SetMaxWOffsetMuonPairsAtPrimaryVtx(maxWOffsetMuonPairsAtPrimaryVtx);
	62	myAnalysis->SetMaxWOffsetMuonPairsAtPCA(maxWOffsetMuonPairsAtPCA);
	63	myAnalysis->SetMaxDistancePrimaryVtxPCA(maxDistancePrimaryVtxPCA);
	64	myAnalysis->SetMinPCAQuality(minPCAQuality);
3b6fd486	65
cabfe25b	66	myAnalysis->SetMatchTrigger(triggerLevel);
e806e863	67
cabfe25b	68	myAnalysis->EvalDimuonVtxResolution(evalDimuonVtxResolution); // it should be true only with prompt dimuon sources
5f74a7f3	69
6240bf36	70	myAnalysis->SetNEventsToMix(nEventsToMix);
e806e863	71
6240bf36	72	if (myAnalysis->Init("MuonGlobalTracks.root")) {
	73	while (myAnalysis->LoadNextEvent()) continue;
	74	myAnalysis->Terminate(Form("outFiles/outFile.%d.%d.%d.root", myAnalysis->GetFirstEvent(), myAnalysis->GetLastEvent(), myRandom));
	75	}
e806e863	76
	77	}
	78
	79	//================================================================================================================================
	80