Replacing surface pressure alias with new one, and removing cr5 pressure alias.

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

void AnalysisTrainMuonLocal(char* filein = "AliESDs.root", char* fileout = "AliAOD.root")\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 locally\r
// R. Arnaldi 5/5/08\r
\r
{\r
    gSystem->Load("libTree.so");\r
    gSystem->Load("libGeom.so");\r
    gSystem->Load("libVMC.so");\r
    gSystem->Load("libPhysics.so");\r
\r
    // If analysis is .par based:\r
\r
    // Common packages\r
    SetupPar("STEERBase");\r
    SetupPar("ESD");\r
    SetupPar("AOD");\r
    SetupPar("ANALYSIS");\r
    SetupPar("ANALYSISalice");\r
    // Analysis-specific packages\r
    SetupPar("PWG3muon");      \r
  \r
    // Input ESD file\r
    TChain* chain = new TChain("esdTree");  \r
    chain->Add(filein);\r
   \r
    // Define 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
    \r
    // AOD output handler\r
    AliAODHandler* aodHandler   = new AliAODHandler();\r
    aodHandler->SetOutputFileName(fileout);\r
\r
    mgr->SetInputEventHandler(esdHandler);\r
    mgr->SetOutputEventHandler(aodHandler);\r
    \r
    // Set of cuts for the ESD filter\r
    // \r
    // standard cuts\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 cuts\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 in the 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 in the 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(), \r
							     AliAnalysisManager::kInputContainer);\r
    // Output AOD container. \r
    AliAnalysisDataContainer *cout_aod = mgr->CreateContainer("cAOD", TTree::Class(),\r
							      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("local",chain);\r
    }   \r
}\r
\r
//______________________________________________________________________________\r
void SetupPar(char* pararchivename)\r
{\r
    if (pararchivename) {\r
	char processline[1024];\r
	sprintf(processline,".! tar xvzf %s.par",pararchivename);\r
	gROOT->ProcessLine(processline);\r
	TString ocwd = gSystem->WorkingDirectory();\r
	gSystem->ChangeDirectory(pararchivename);\r
	\r
	// check for BUILD.sh and execute\r
	if (!gSystem->AccessPathName("PROOF-INF/BUILD.sh")) {\r
	    printf("*******************************\n");\r
	    printf("*** Building PAR archive    ***\n");\r
	    printf("*******************************\n");\r
	    \r
	    if (gSystem->Exec("PROOF-INF/BUILD.sh")) {\r
		Error("runProcess","Cannot Build the PAR Archive! - Abort!");\r
		return -1;\r
	    }\r
	}\r
	// check for SETUP.C and execute\r
	if (!gSystem->AccessPathName("PROOF-INF/SETUP.C")) {\r
	    printf("*******************************\n");\r
	    printf("*** Setup PAR archive       ***\n");\r
	    printf("*******************************\n");\r
	    gROOT->Macro("PROOF-INF/SETUP.C");\r
	}\r
	\r
	gSystem->ChangeDirectory(ocwd.Data());\r
   printf("Current dir: %s\n", ocwd.Data());\r
    } \r
}\r