[u/mrichter/AliRoot.git] / PWGLF / STRANGENESS / LambdaK0PbPb / runLukeAOD.C

// run.C
//
// Template run macro for AliAnalysisTaskLukeAOD.cxx/.h with example layout of
// physics selections and options, in macro and task.
//
// Author: Arvinder Palaha
// Edited by Luke Hanratty for AODSExAOD
//
class AliAnalysisGrid;

//______________________________________________________________________________
void runExAOD(
         const char* runtype = "proof", // local, proof or grid
         const char *gridmode = "full", // Set the run mode (can be "full", "test", "offline", "submit" or "terminate"). Full & Test work for proof
         const bool bMCtruth = 1, // 1 = MCEvent handler is on (MC truth), 0 = MCEvent handler is off (MC reconstructed/real data)
         const bool bMCphyssel = 1, // 1 = looking at MC truth or reconstructed, 0 = looking at real data
         const Long64_t nentries = 1000, // for local and proof mode, ignored in grid mode. Set to 1234567890 for all events.
         const Long64_t firstentry = 0, // for local and proof mode, ignored in grid mode
         const char *proofdataset = "/alice/sim/LHC11a10a_000138662_AOD048", //"/alice/sim/LHC11a10a_000138795_AOD048", //"/alice/data/LHC10h_000139507_p2_AOD049", //"/alice/data/LHC10e_000130375_p2", //"/alice/data/LHC10c_000120821_p1", // path to dataset on proof cluster, for proof analysis
         const char *proofcluster = "hanratty@alice-caf.cern.ch", //"alice-caf.cern.ch", //"hanratty@skaf.saske.sk", //"alice-caf.cern.ch", // which proof cluster to use in proof mode
         const char *taskname = "AOD_AllCent_1" // sets name of grid generated macros
         )
{
    // check run type
    if(runtype != "local" && runtype != "proof" && runtype != "grid"){
        Printf("\n\tIncorrect run option, check first argument of run macro");
        Printf("\tint runtype = local, proof or grid\n");
        return;
    }
    Printf("%s analysis chosen",runtype);
  
    // load libraries
    gSystem->Load("libCore.so");        
    gSystem->Load("libGeom.so");
    gSystem->Load("libVMC.so");
    gSystem->Load("libPhysics.so");
    gSystem->Load("libTree.so");
    gSystem->Load("libSTEERBase.so");
    gSystem->Load("libESD.so");
    gSystem->Load("libAOD.so");
    gSystem->Load("libANALYSIS.so");
    gSystem->Load("libANALYSISalice.so");
  
    // add aliroot indlude path
    gROOT->ProcessLine(Form(".include %s/include",gSystem->ExpandPathName("$ALICE_ROOT")));
    gROOT->SetStyle("Plain");
        
    // analysis manager
    AliAnalysisManager* mgr = new AliAnalysisManager(taskname);
    
    // create the alien handler and attach it to the manager
    AliAnalysisGrid *plugin = CreateAlienHandler(taskname, gridmode, proofcluster, proofdataset); 
    mgr->SetGridHandler(plugin);
    
    AliVEventHandler* aodH = new AliAODInputHandler();
    mgr->SetInputEventHandler(aodH);
        
    // mc event handler
    /*if(bMCtruth) {
        AliMCEventHandler* mchandler = new AliMCEventHandler();
        // Not reading track references
        mchandler->SetReadTR(kFALSE);
        mgr->SetMCtruthEventHandler(mchandler);
    }*/   

    //gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
    //AliPhysicsSelectionTask *physSelTask = AddTaskPhysicsSelection(bMCphyssel);
    //if(!physSelTask) { Printf("no physSelTask"); return; }
    //AliPhysicsSelection *physSel = physSelTask->GetPhysicsSelection();
    //physSel->AddCollisionTriggerClass("+CINT1B-ABCE-NOPF-ALL");// #3119 #769");
         
	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
    AliPhysicsSelectionTask *physSelTask = AddTaskPhysicsSelection(bMCphyssel);
    if(!physSelTask) { Printf("no physSelTask"); return; }
	
	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
    AliAnalysisTask *PIDTask = AddTaskPIDResponse(bMCtruth,kTRUE);
    if(!PIDTask) { Printf("no PIDtask"); return; }
	
	/*gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");
	AliCentralitySelectionTask *centralityTask = AddTaskCentrality();
    if(!centralityTask) { Printf("no centralityTask"); return; }
	*/
	
    // create task
    gROOT->LoadMacro("AliAnalysisTaskLukeAOD.cxx++g");
    AliAnalysisTaskSE* task = new AliAnalysisTaskLukeAOD(taskname);
    task->SelectCollisionCandidates(AliVEvent::kMB); // if physics selection performed in UserExec(), this line should be commented
	mgr->AddTask(task);
    
    // set output root file name for different analysis
    TString outfilename = Form("list.%s.root",runtype);
  
    // create containers for input/output
    AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
    AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("coutput1", TList::Class(), AliAnalysisManager::kOutputContainer, outfilename);
        
    // connect input/output
    mgr->ConnectInput(task, 0, cinput);
    mgr->ConnectOutput(task, 1, coutput1);
	//mgr->ConnectInput (centralityTask,0, mgr->GetCommonInputContainer());
        
    // enable debug printouts
    mgr->SetDebugLevel(2);
    if (!mgr->InitAnalysis()) return;
    mgr->PrintStatus();
  
    // start analysis
    Printf("Starting Analysis....");
    mgr->StartAnalysis(runtype,nentries,firstentry);
}

//______________________________________________________________________________
AliAnalysisGrid* CreateAlienHandler(const char *taskname, const char *gridmode, const char *proofcluster, const char *proofdataset)
{
    AliAnalysisAlien *plugin = new AliAnalysisAlien();
    // Set the run mode (can be "full", "test", "offline", "submit" or "terminate")
    plugin->SetRunMode(gridmode);

    // Set versions of used packages
    plugin->SetAPIVersion("V1.1X");
    plugin->SetROOTVersion("v5-30-03");
    plugin->SetAliROOTVersion("v5-02-16-AN");

    // Declare input data to be processed.

    // Method 1: Create automatically XML collections using alien 'find' command.
    // Define production directory LFN
    plugin->SetGridDataDir("/alice/data/2010/LHC10h");
    // On real reconstructed data:
    // plugin->SetGridDataDir("/alice/data/2009/LHC09d");
    // Set data search pattern
    //plugin->SetDataPattern("*ESDs.root"); // THIS CHOOSES ALL PASSES
    // Data pattern for reconstructed data
    plugin->SetDataPattern("*ESDs/pass2/*AOD.root"); // CHECK LATEST PASS OF DATA SET IN ALIENSH
    plugin->SetRunPrefix("000");   // real data
    // ...then add run numbers to be considered
    plugin->AddRunNumber(139507);
    //plugin->SetRunRange(114917,115322);
    plugin->SetNrunsPerMaster(1);
    plugin->SetOutputToRunNo();
    // comment out the next line when using the "terminate" option, unless
    // you want separate merged files for each run
    plugin->SetMergeViaJDL();

    // Method 2: Declare existing data files (raw collections, xml collections, root file)
    // If no path mentioned data is supposed to be in the work directory (see SetGridWorkingDir())
    // XML collections added via this method can be combined with the first method if
    // the content is compatible (using or not tags)
    //   plugin->AddDataFile("tag.xml");
    //   plugin->AddDataFile("/alice/data/2008/LHC08c/000057657/raw/Run57657.Merged.RAW.tag.root");

    // Define alien work directory where all files will be copied. Relative to alien $HOME.
    plugin->SetGridWorkingDir(taskname);

    // Declare alien output directory. Relative to working directory.
    plugin->SetGridOutputDir("out"); // In this case will be $HOME/taskname/out

    // Declare the analysis source files names separated by blancs. To be compiled runtime
    // using ACLiC on the worker nodes.
    plugin->SetAnalysisSource("AliAnalysisTaskLukeAOD.cxx");

    // Declare all libraries (other than the default ones for the framework. These will be
    // loaded by the generated analysis macro. Add all extra files (task .cxx/.h) here.
    plugin->SetAdditionalLibs("AliAnalysisTaskLukeAOD.h AliAnalysisTaskLukeAOD.cxx");

    // Declare the output file names separated by blancs.
    // (can be like: file.root or file.root@ALICE::Niham::File)
    // To only save certain files, use SetDefaultOutputs(kFALSE), and then
    // SetOutputFiles("list.root other.filename") to choose which files to save
    plugin->SetDefaultOutputs();
    //plugin->SetOutputFiles("list.root");

    // Optionally set a name for the generated analysis macro (default MyAnalysis.C)
    plugin->SetAnalysisMacro(Form("%s.C",taskname));

    // Optionally set maximum number of input files/subjob (default 100, put 0 to ignore)
    plugin->SetSplitMaxInputFileNumber(100);

    // Optionally modify the executable name (default analysis.sh)
    plugin->SetExecutable(Form("%s.sh",taskname));

    // set number of test files to use in "test" mode
    plugin->SetNtestFiles(10);

    // Optionally resubmit threshold.
    plugin->SetMasterResubmitThreshold(90);

    // Optionally set time to live (default 30000 sec)
    plugin->SetTTL(30000);

    // Optionally set input format (default xml-single)
    plugin->SetInputFormat("xml-single");

    // Optionally modify the name of the generated JDL (default analysis.jdl)
    plugin->SetJDLName(Form("%s.jdl",taskname));

    // Optionally modify job price (default 1)
    plugin->SetPrice(1);      

    // Optionally modify split mode (default 'se')    
    plugin->SetSplitMode("se");
    
    //----------------------------------------------------------
    //---      PROOF MODE SPECIFIC SETTINGS         ------------
    //---------------------------------------------------------- 
    // Proof cluster
    plugin->SetProofCluster(proofcluster);
    // Dataset to be used   
    plugin->SetProofDataSet(proofdataset);
    // May need to reset proof. Supported modes: 0-no reset, 1-soft, 2-hard
    plugin->SetProofReset(0);
    // May limit number of workers
    plugin->SetNproofWorkers(10);
    // May limit the number of workers per slave
    plugin->SetNproofWorkersPerSlave(1);   
    // May use a specific version of root installed in proof
    plugin->SetRootVersionForProof("VO_ALICE@ROOT::v5-30-03-1");
    // May set the aliroot mode. Check http://aaf.cern.ch/node/83 
    plugin->SetAliRootMode("default"); // Loads AF libs by default
    // May request ClearPackages (individual ClearPackage not supported)
    plugin->SetClearPackages(kFALSE);
    // Plugin test mode works only providing a file containing test file locations, used in "local" mode also
    plugin->SetFileForTestMode("files.txt"); // file should contain path name to a local directory containg *ESDs.root etc
    // Request connection to alien upon connection to grid
    plugin->SetProofConnectGrid(kFALSE);

    return plugin;
}
Commit	Line	Data
12ca6c45	1	// run.C
12ca6c45	2	//
8a7cb9cc	3	// Template run macro for AliAnalysisTaskLukeAOD.cxx/.h with example layout of
12ca6c45	4	// physics selections and options, in macro and task.
	5	//
	6	// Author: Arvinder Palaha
bbfb6fac	7	// Edited by Luke Hanratty for AODSExAOD
12ca6c45	8	//
	9	class AliAnalysisGrid;
	10
	11	//______________________________________________________________________________
bbfb6fac	12	void runExAOD(
12ca6c45	13	const char* runtype = "proof", // local, proof or grid
	14	const char *gridmode = "full", // Set the run mode (can be "full", "test", "offline", "submit" or "terminate"). Full & Test work for proof
	15	const bool bMCtruth = 1, // 1 = MCEvent handler is on (MC truth), 0 = MCEvent handler is off (MC reconstructed/real data)
	16	const bool bMCphyssel = 1, // 1 = looking at MC truth or reconstructed, 0 = looking at real data
	17	const Long64_t nentries = 1000, // for local and proof mode, ignored in grid mode. Set to 1234567890 for all events.
	18	const Long64_t firstentry = 0, // for local and proof mode, ignored in grid mode
bbfb6fac	19	const char *proofdataset = "/alice/sim/LHC11a10a_000138662_AOD048", //"/alice/sim/LHC11a10a_000138795_AOD048", //"/alice/data/LHC10h_000139507_p2_AOD049", //"/alice/data/LHC10e_000130375_p2", //"/alice/data/LHC10c_000120821_p1", // path to dataset on proof cluster, for proof analysis
bbfb6fac	20	const char *proofcluster = "hanratty@alice-caf.cern.ch", //"alice-caf.cern.ch", //"hanratty@skaf.saske.sk", //"alice-caf.cern.ch", // which proof cluster to use in proof mode
12ca6c45	21	const char *taskname = "AOD_AllCent_1" // sets name of grid generated macros
	22	)
	23	{
	24	// check run type
	25	if(runtype != "local" && runtype != "proof" && runtype != "grid"){
	26	Printf("\n\tIncorrect run option, check first argument of run macro");
	27	Printf("\tint runtype = local, proof or grid\n");
	28	return;
	29	}
	30	Printf("%s analysis chosen",runtype);
	31
	32	// load libraries
	33	gSystem->Load("libCore.so");
	34	gSystem->Load("libGeom.so");
	35	gSystem->Load("libVMC.so");
	36	gSystem->Load("libPhysics.so");
	37	gSystem->Load("libTree.so");
	38	gSystem->Load("libSTEERBase.so");
	39	gSystem->Load("libESD.so");
	40	gSystem->Load("libAOD.so");
	41	gSystem->Load("libANALYSIS.so");
	42	gSystem->Load("libANALYSISalice.so");
	43
	44	// add aliroot indlude path
	45	gROOT->ProcessLine(Form(".include %s/include",gSystem->ExpandPathName("$ALICE_ROOT")));
	46	gROOT->SetStyle("Plain");
	47
	48	// analysis manager
	49	AliAnalysisManager* mgr = new AliAnalysisManager(taskname);
	50
	51	// create the alien handler and attach it to the manager
	52	AliAnalysisGrid *plugin = CreateAlienHandler(taskname, gridmode, proofcluster, proofdataset);
	53	mgr->SetGridHandler(plugin);
	54
	55	AliVEventHandler* aodH = new AliAODInputHandler();
	56	mgr->SetInputEventHandler(aodH);
	57
	58	// mc event handler
	59	/*if(bMCtruth) {
	60	AliMCEventHandler* mchandler = new AliMCEventHandler();
	61	// Not reading track references
	62	mchandler->SetReadTR(kFALSE);
	63	mgr->SetMCtruthEventHandler(mchandler);
	64	}*/
	65
	66	//gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
	67	//AliPhysicsSelectionTask *physSelTask = AddTaskPhysicsSelection(bMCphyssel);
	68	//if(!physSelTask) { Printf("no physSelTask"); return; }
	69	//AliPhysicsSelection *physSel = physSelTask->GetPhysicsSelection();
	70	//physSel->AddCollisionTriggerClass("+CINT1B-ABCE-NOPF-ALL");// #3119 #769");
	71
	72	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
	73	AliPhysicsSelectionTask *physSelTask = AddTaskPhysicsSelection(bMCphyssel);
	74	if(!physSelTask) { Printf("no physSelTask"); return; }
	75
	76	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
	77	AliAnalysisTask *PIDTask = AddTaskPIDResponse(bMCtruth,kTRUE);
	78	if(!PIDTask) { Printf("no PIDtask"); return; }
	79
	80	/*gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");
	81	AliCentralitySelectionTask *centralityTask = AddTaskCentrality();
	82	if(!centralityTask) { Printf("no centralityTask"); return; }
	83	*/
	84
85	// create task
8a7cb9cc	86	gROOT->LoadMacro("AliAnalysisTaskLukeAOD.cxx++g");
8a7cb9cc	87	AliAnalysisTaskSE* task = new AliAnalysisTaskLukeAOD(taskname);
12ca6c45	88	task->SelectCollisionCandidates(AliVEvent::kMB); // if physics selection performed in UserExec(), this line should be commented
	89	mgr->AddTask(task);
	90
	91	// set output root file name for different analysis
	92	TString outfilename = Form("list.%s.root",runtype);
	93
	94	// create containers for input/output
	95	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	96	AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("coutput1", TList::Class(), AliAnalysisManager::kOutputContainer, outfilename);
	97
	98	// connect input/output
	99	mgr->ConnectInput(task, 0, cinput);
	100	mgr->ConnectOutput(task, 1, coutput1);
	101	//mgr->ConnectInput (centralityTask,0, mgr->GetCommonInputContainer());
	102
	103	// enable debug printouts
	104	mgr->SetDebugLevel(2);
	105	if (!mgr->InitAnalysis()) return;
	106	mgr->PrintStatus();
	107
	108	// start analysis
	109	Printf("Starting Analysis....");
	110	mgr->StartAnalysis(runtype,nentries,firstentry);
	111	}
	112
	113	//______________________________________________________________________________
	114	AliAnalysisGrid* CreateAlienHandler(const char taskname, const char gridmode, const char proofcluster, const char proofdataset)
	115	{
	116	AliAnalysisAlien *plugin = new AliAnalysisAlien();
	117	// Set the run mode (can be "full", "test", "offline", "submit" or "terminate")
	118	plugin->SetRunMode(gridmode);
	119
	120	// Set versions of used packages
	121	plugin->SetAPIVersion("V1.1X");
	122	plugin->SetROOTVersion("v5-30-03");
bbfb6fac	123	plugin->SetAliROOTVersion("v5-02-16-AN");
12ca6c45	124
	125	// Declare input data to be processed.
	126
	127	// Method 1: Create automatically XML collections using alien 'find' command.
	128	// Define production directory LFN
	129	plugin->SetGridDataDir("/alice/data/2010/LHC10h");
	130	// On real reconstructed data:
	131	// plugin->SetGridDataDir("/alice/data/2009/LHC09d");
	132	// Set data search pattern
	133	//plugin->SetDataPattern("*ESDs.root"); // THIS CHOOSES ALL PASSES
	134	// Data pattern for reconstructed data
	135	plugin->SetDataPattern("ESDs/pass2/AOD.root"); // CHECK LATEST PASS OF DATA SET IN ALIENSH
	136	plugin->SetRunPrefix("000"); // real data
	137	// ...then add run numbers to be considered
	138	plugin->AddRunNumber(139507);
	139	//plugin->SetRunRange(114917,115322);
	140	plugin->SetNrunsPerMaster(1);
	141	plugin->SetOutputToRunNo();
	142	// comment out the next line when using the "terminate" option, unless
	143	// you want separate merged files for each run
	144	plugin->SetMergeViaJDL();
	145
	146	// Method 2: Declare existing data files (raw collections, xml collections, root file)
	147	// If no path mentioned data is supposed to be in the work directory (see SetGridWorkingDir())
	148	// XML collections added via this method can be combined with the first method if
	149	// the content is compatible (using or not tags)
	150	// plugin->AddDataFile("tag.xml");
	151	// plugin->AddDataFile("/alice/data/2008/LHC08c/000057657/raw/Run57657.Merged.RAW.tag.root");
	152
	153	// Define alien work directory where all files will be copied. Relative to alien $HOME.
	154	plugin->SetGridWorkingDir(taskname);
	155
	156	// Declare alien output directory. Relative to working directory.
	157	plugin->SetGridOutputDir("out"); // In this case will be $HOME/taskname/out
	158
	159	// Declare the analysis source files names separated by blancs. To be compiled runtime
	160	// using ACLiC on the worker nodes.
8a7cb9cc	161	plugin->SetAnalysisSource("AliAnalysisTaskLukeAOD.cxx");
12ca6c45	162
	163	// Declare all libraries (other than the default ones for the framework. These will be
	164	// loaded by the generated analysis macro. Add all extra files (task .cxx/.h) here.
8a7cb9cc	165	plugin->SetAdditionalLibs("AliAnalysisTaskLukeAOD.h AliAnalysisTaskLukeAOD.cxx");
12ca6c45	166
	167	// Declare the output file names separated by blancs.
	168	// (can be like: file.root or file.root@ALICE::Niham::File)
	169	// To only save certain files, use SetDefaultOutputs(kFALSE), and then
	170	// SetOutputFiles("list.root other.filename") to choose which files to save
	171	plugin->SetDefaultOutputs();
	172	//plugin->SetOutputFiles("list.root");
	173
	174	// Optionally set a name for the generated analysis macro (default MyAnalysis.C)
	175	plugin->SetAnalysisMacro(Form("%s.C",taskname));
	176
	177	// Optionally set maximum number of input files/subjob (default 100, put 0 to ignore)
	178	plugin->SetSplitMaxInputFileNumber(100);
	179
	180	// Optionally modify the executable name (default analysis.sh)
	181	plugin->SetExecutable(Form("%s.sh",taskname));
	182
	183	// set number of test files to use in "test" mode
	184	plugin->SetNtestFiles(10);
	185
	186	// Optionally resubmit threshold.
	187	plugin->SetMasterResubmitThreshold(90);
	188
	189	// Optionally set time to live (default 30000 sec)
	190	plugin->SetTTL(30000);
	191
	192	// Optionally set input format (default xml-single)
	193	plugin->SetInputFormat("xml-single");
	194
	195	// Optionally modify the name of the generated JDL (default analysis.jdl)
	196	plugin->SetJDLName(Form("%s.jdl",taskname));
	197
	198	// Optionally modify job price (default 1)
	199	plugin->SetPrice(1);
	200
	201	// Optionally modify split mode (default 'se')
	202	plugin->SetSplitMode("se");
	203
	204	//----------------------------------------------------------
	205	//--- PROOF MODE SPECIFIC SETTINGS ------------
	206	//----------------------------------------------------------
	207	// Proof cluster
	208	plugin->SetProofCluster(proofcluster);
	209	// Dataset to be used
	210	plugin->SetProofDataSet(proofdataset);
	211	// May need to reset proof. Supported modes: 0-no reset, 1-soft, 2-hard
	212	plugin->SetProofReset(0);
	213	// May limit number of workers
	214	plugin->SetNproofWorkers(10);
	215	// May limit the number of workers per slave
	216	plugin->SetNproofWorkersPerSlave(1);
	217	// May use a specific version of root installed in proof
	218	plugin->SetRootVersionForProof("VO_ALICE@ROOT::v5-30-03-1");
	219	// May set the aliroot mode. Check http://aaf.cern.ch/node/83
	220	plugin->SetAliRootMode("default"); // Loads AF libs by default
	221	// May request ClearPackages (individual ClearPackage not supported)
	222	plugin->SetClearPackages(kFALSE);
	223	// Plugin test mode works only providing a file containing test file locations, used in "local" mode also
	224	plugin->SetFileForTestMode("files.txt"); // file should contain path name to a local directory containg *ESDs.root etc
	225	// Request connection to alien upon connection to grid
	226	plugin->SetProofConnectGrid(kFALSE);
	227
	228	return plugin;
	229	}
230