X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=MFT%2FAliMuonForwardTrackAnalysis.C;h=b070b2fd07cdac7807fdd665350b2be26c0c5936;hp=b81217564c500c00b285764753be0c5cb9c0540d;hb=64185694fcaf783dd7a1830be90968e8558919ba;hpb=3b6fd486b69b5f92a08fd193f869055f6be273bf

diff --git a/MFT/AliMuonForwardTrackAnalysis.C b/MFT/AliMuonForwardTrackAnalysis.C
index b81217564c5..b070b2fd07c 100644
--- a/MFT/AliMuonForwardTrackAnalysis.C
+++ b/MFT/AliMuonForwardTrackAnalysis.C
@@ -1,52 +1,82 @@
-enum {kNoOption, kOpenFlavor, kResonanceOnly};
+// #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= ".",
-				 Int_t option = kNoOption,
-				 Int_t nMassBin = 100, 
-				 Double_t massMin = 0.,
-				 Double_t massMax = 10.,
-				 const Char_t *outDir = ".",
-				 Bool_t singleMuonAnalysis = kTRUE,
-				 Bool_t muonPairAnalysis = kTRUE,
+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 = 1.e9.,            // upper limit for the cut on the single muon offset w.r.t. the primary vtx
+				 Bool_t correlateCutOnOffsetChi2 = kTRUE,          // if true, the cut 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,
-				 Int_t maxNWrongClusters = 999,
-				 Double_t ptMinSingleMuons = 0.0) {
+				 Int_t numTag = 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 = 3.097,                        // 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 Char_t *tag = "noTag",                      // tag added to the output file name
+				 Double_t etaMinSingleMuons = -3.6,                // lower limit for the cut on the single muon eta
+				 Double_t etaMaxSingleMuons = -2.5) {              // upper limit for the cut on the single muon eta
   
+  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", 1);
+  //  AliLog::SetClassDebugLevel("AliMuonForwardTrackAnalysis", 2);
 
-  //  TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG, AliMagF::kBeamTypeAA, 2750.));
-  //  TGeoGlobalMagField::Instance()->SetField(new AliMagF("Maps","Maps", -1., -1., AliMagF::k5kG, AliMagF::kBeamTypepp, 7000.));
   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(nMassBin, massMin, massMax);
-  myAnalysis->SetPtDimuRange(10, 0., 5.);
+  myAnalysis->SetMassRange(massMin, massMax);
+  myAnalysis->SetTrueMass(trueMass);
   myAnalysis->SetSingleMuonAnalysis(singleMuonAnalysis);
   myAnalysis->SetMuonPairAnalysis(muonPairAnalysis);
   myAnalysis->SetOption(option);
-  myAnalysis->SetMatchTrigger(kTRUE);
+
   myAnalysis->SetMaxNWrongClustersMC(maxNWrongClusters);
-  myAnalysis->SetPtMinSingleMuons(ptMinSingleMuons);
+  myAnalysis->SetMinPtSingleMuons(ptMinSingleMuons);
+  //  myAnalysis->SetEtaRangeSingleMuons(etaMinSingleMuons, etaMaxSingleMuons);
+  myAnalysis->SetMaxChi2SingleMuons(maxChi2SingleMuons);
+  myAnalysis->SetMaxOffsetSingleMuons(maxOffsetSingleMuons);
+  myAnalysis->CorrelateCutOnOffsetChi2(correlateCutOnOffsetChi2);
 
-  myAnalysis->UseCutOnOffsetChi2(kFALSE);        // cut on the single muons
+  myAnalysis->SetMaxWOffsetMuonPairsAtPrimaryVtx(maxWOffsetMuonPairsAtPrimaryVtx);
+  myAnalysis->SetMaxWOffsetMuonPairsAtPCA(maxWOffsetMuonPairsAtPCA);
+  myAnalysis->SetMaxDistancePrimaryVtxPCA(maxDistancePrimaryVtxPCA);
+  myAnalysis->SetMinPCAQuality(minPCAQuality);
 
-  myAnalysis->UseBransonForCut(kFALSE);
-  myAnalysis->UseBransonForKinematics(kFALSE);
+  myAnalysis->SetMatchTrigger(triggerLevel);
 
-  myAnalysis->Init("MuonGlobalTracks.root");
+  myAnalysis->EvalDimuonVtxResolution(evalDimuonVtxResolution);    // it should be true only with prompt dimuon sources
 
-  while (myAnalysis->LoadNextEvent()) continue;
+  myAnalysis->SetNEventsToMix(nEventsToMix);
 
-  myAnalysis->Terminate(Form("outFiles/outFile.%d.%d.%d.root", myAnalysis->GetFirstEvent(), myAnalysis->GetLastEvent(), myRandom));
+  if (myAnalysis->Init("MuonGlobalTracks.root")) {
+    while (myAnalysis->LoadNextEvent()) continue;
+    myAnalysis->Terminate(Form("outFiles/outFile.%d.%s.root", numTag, tag));
+  }
 
 }