Adding charge of the muon track in the AOD (Roberta)

[u/mrichter/AliRoot.git] / PWG3 / muon / AnalysisTrainMuonCAF.C

void AnalysisTrainMuonCAF(char* fileout = "AliAOD.root", char *datasetname = "myDataSet", Int_t nev=1234567890)\r
{\r
// Macro to produce a generic AOD starting from an ESD file. \r
// The AOD is filled with two tasks: \r
// 1- with the first one (AliAnalysisTaskESDfilter), \r
//    all the branches of the AOD are filled apart from the muons. \r
// 2- with the second task (AliAnalysisTaskESDMuonFilter) \r
//    muons tracks are added to the tracks branch \r
// This macro works on the CAF\r
// R. Arnaldi 4/5/08\r
\r
  gSystem->Load("libTree.so");\r
  gSystem->Load("libGeom.so");\r
  gSystem->Load("libVMC.so");\r
  gSystem->Load("libPhysics.so");\r
    \r
  // Reset user processes if CAF if not responding anymore\r
  // TProof::Reset("lxb6046");\r
\r
  // Connect to proof\r
  TProof::Open("lxb6046"); // may be username@lxb6046 if user not the same as on local\r
\r
  // Clear packages if changing ROOT version on CAF or local\r
  // gProof->ClearPackages();\r
  // Enable proof debugging if needed\r
  // gProof->SetLogLevel(5);\r
\r
  // Common packages\r
  gProof->UploadPackage("STEERBase.par");\r
  gProof->EnablePackage("STEERBase");\r
  gProof->UploadPackage("ESD.par");\r
  gProof->EnablePackage("ESD");\r
  gProof->UploadPackage("AOD.par");\r
  gProof->EnablePackage("AOD");\r
  gProof->UploadPackage("ANALYSIS.par");\r
  gProof->EnablePackage("ANALYSIS");\r
  gProof->UploadPackage("ANALYSISalice.par");\r
  gProof->EnablePackage("ANALYSISalice");\r
  // Analysis-specific\r
  // --- Enable the PWG3base Package\r
  gProof->UploadPackage("PWG3muon.par");\r
  gProof->EnablePackage("PWG3muon");\r
\r
  // Chain from files staged on CAF\r
  // gROOT->LoadMacro("CreateESDChain.C");\r
  // TChain* chain = CreateESDChain("ESD1503X_v1.txt",3);\r
  // TChain* chain = CreateESDChain("ESD82XX_30Kshort.txt", 10);\r
  \r
  // Chain from datasets\r
  gROOT->LoadMacro("CreateChainFromDataSet.C");\r
  ds = gProof->GetDataSet(datasetname)->GetStagedSubset();\r
  chain = CreateChainFromDataSet(ds, "esdTree");   \r
  \r
  // Make the analysis manager\r
  AliAnalysisManager *mgr  = new AliAnalysisManager("Analysis Train", "Analysis train");\r
  \r
  // ESD input handler\r
  AliESDInputHandler *esdHandler = new AliESDInputHandler();\r
  esdHandler->SetInactiveBranches("FMD CaloCluster");\r
  // AOD output handler\r
  AliAODHandler* aodHandler   = new AliAODHandler();\r
  //aodHandler->SetOutputFileName(fileout);\r
  aodHandler->SetOutputFileName("AOD.root");\r
\r
  mgr->SetInputEventHandler(esdHandler);\r
  mgr->SetOutputEventHandler(aodHandler);\r
  \r
  // Set of cuts plugged into the ESD filter\r
  // \r
  // standard\r
  AliESDtrackCuts* esdTrackCutsL = new AliESDtrackCuts("AliESDtrackCuts", "Loose");\r
  esdTrackCutsL->SetMinNClustersTPC(50);\r
  esdTrackCutsL->SetMaxChi2PerClusterTPC(3.5);\r
  esdTrackCutsL->SetMaxCovDiagonalElements(2,2,0.5,0.5,2);\r
  esdTrackCutsL->SetRequireTPCRefit(kTRUE);\r
  esdTrackCutsL->SetMinNsigmaToVertex(3);\r
  esdTrackCutsL->SetRequireSigmaToVertex(kTRUE);\r
  esdTrackCutsL->SetAcceptKingDaughters(kFALSE);\r
  //\r
  // hard\r
  AliESDtrackCuts* esdTrackCutsH = new AliESDtrackCuts("AliESDtrackCuts", "Hard");\r
  esdTrackCutsH->SetMinNClustersTPC(100);\r
  esdTrackCutsH->SetMaxChi2PerClusterTPC(2.0);\r
  esdTrackCutsH->SetMaxCovDiagonalElements(2,2,0.5,0.5,2);\r
  esdTrackCutsH->SetRequireTPCRefit(kTRUE);\r
  esdTrackCutsH->SetMinNsigmaToVertex(2);\r
  esdTrackCutsH->SetRequireSigmaToVertex(kTRUE);\r
  esdTrackCutsH->SetAcceptKingDaughters(kFALSE);\r
  //\r
  AliAnalysisFilter* trackFilter = new AliAnalysisFilter("trackFilter");\r
  trackFilter->AddCuts(esdTrackCutsL);\r
  trackFilter->AddCuts(esdTrackCutsH);\r
\r
  // ESD filter task putting standard info to output generic AOD \r
  AliAnalysisTaskESDfilter *esdfilter = new AliAnalysisTaskESDfilter("ESD Filter");\r
  //esdfilter->SetTrackFilter(trackFilter);\r
  esdfilter->SetDebugLevel(10);\r
  mgr->AddTask(esdfilter);\r
  \r
  // ESD filter task putting muon info to output generic AOD \r
  AliAnalysisTaskESDMuonFilter *esdmuonfilter = new AliAnalysisTaskESDMuonFilter("ESD Muon Filter");\r
  mgr->AddTask(esdmuonfilter);\r
\r
  // Containers for input/output\r
  AliAnalysisDataContainer *cin_esd = mgr->CreateContainer("cESD",TChain::Class(), AliAnalysisManager::kInputContainer);\r
  // Output AOD container. \r
  AliAnalysisDataContainer *cout_aod = mgr->CreateContainer("cAOD", TTree::Class(), AliAnalysisManager::kOutputContainer, "default");\r
        						    \r
  // Connect containers to tasks slots\r
  mgr->ConnectInput  (esdfilter,  0, cin_esd  );\r
  mgr->ConnectOutput (esdfilter,  0, cout_aod );\r
\r
  mgr->ConnectInput  (esdmuonfilter,  0, cin_esd);\r
  mgr->ConnectOutput (esdmuonfilter,  0, cout_aod );\r
\r
  //\r
  // Run the analysis\r
  //	\r
  if (mgr->InitAnalysis()) {\r
      mgr->PrintStatus();\r
      mgr->StartAnalysis("proof",chain,nev);\r
  }   \r
}\r
\r