B2 analysis code

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / Nuclei / B2 / macros / LHC10bcde.C

/**************************************************************************\r
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
 *                                                                        *\r
 * Author: The ALICE Off-line Project.                                    *\r
 * Contributors are mentioned in the code where appropriate.              *\r
 *                                                                        *\r
 * Permission to use, copy, modify and distribute this software and its   *\r
 * documentation strictly for non-commercial purposes is hereby granted   *\r
 * without fee, provided that the above copyright notice appears in all   *\r
 * copies and that both the copyright notice and this permission notice   *\r
 * appear in the supporting documentation. The authors make no claims     *\r
 * about the suitability of this software for any purpose. It is          *\r
 * provided "as is" without express or implied warranty.                  *\r
 **************************************************************************/\r
\r
// call Config_XXX for each period\r
// author: Eulogio Serradilla <eulogio.serradilla@cern.ch>\r
\r
#include <Riostream.h>\r
#include <TROOT.h>\r
#include <TSystem.h>\r
#include <TString.h>\r
#include <TFileMerger.h>\r
\r
#include "AliLnDriver.h"\r
#include "Config.h"\r
\r
Int_t LHC10bcde(const TString& species    = "Deuteron",\r
                const TString& inputDir   = "~/alice/input",\r
                const TString& outputDir  = "~/alice/output",\r
                const TString& outputTag  = "lhc10bcde",\r
                const TString& multTag    = "",\r
                const TString& multCorTag = "",\r
                Bool_t normToInel         = 1,  // for mult\r
                Bool_t drawOutput         = 1,  // for batch\r
                Int_t option              = 0)\r
{\r
//\r
// call Config_XXX for each period, merge the corrected pt and then get the results\r
//\r
	using namespace std;\r
	\r
	const Int_t kNper = 4;\r
	const TString kPeriod[kNper]    = { "lhc10b", "lhc10c", "lhc10d", "lhc10e" };\r
	const TString kOutputTag[kNper] = { "lhc10b", "lhc10c", "lhc10d", "lhc10e" };\r
	\r
	Double_t ymin = (species=="Proton") ? 1.1e-6 : 1.1e-8;\r
	Double_t ymax = (species=="Proton") ? 4.e-1 : 4.e-4;\r
	\r
	TFileMerger m;\r
	\r
	for(Int_t i=0; i<kNper; ++i)\r
	{\r
		cout << endl << "Period: " << kPeriod[i] << endl;\r
		\r
		TString arg =           inputDir       + "\","\r
			      + "\""  + outputDir      + "\","\r
			      + "\""  + kPeriod[i]     + "\","\r
			      + "\""  + kOutputTag[i]  + "\","\r
			      + "\""  + multTag        + "\","\r
			      + "\""  + multCorTag;\r
			\r
		if(species=="Proton" && option==0)\r
		{\r
			cout << "Config_Proton_TPC_LHC10x.C" << endl << endl;\r
			gROOT->ProcessLine(Form(".x Config_Proton_TPC_LHC10x.C+g(\"%s\",%d,0)", arg.Data(), normToInel));\r
		}\r
		else if(species=="Proton" && option==1)\r
		{\r
			cout << "Config_Proton_TOF_LHC10x.C" << endl << endl;\r
			gROOT->ProcessLine(Form(".x Config_Proton_TOF_LHC10x.C+g(\"%s\",%d,0)", arg.Data(), normToInel));\r
		}\r
		else if(species=="Proton" && option==2)\r
		{\r
			cout << "Config_Proton_TPCTOF_LHC10x.C" << endl << endl;\r
			gROOT->ProcessLine(Form(".x Config_Proton_TPCTOF_LHC10x.C+g(\"%s\",%d,0)", arg.Data(), normToInel));\r
		}\r
		else if(species=="Deuteron")\r
		{\r
			cout << "Config_Deuteron_TOF_LHC10x.C" << endl << endl;\r
			gROOT->ProcessLine(Form(".x Config_Deuteron_TOF_LHC10x.C+g(\"%s\",%d,0)", arg.Data(), normToInel));\r
		}\r
		else\r
		{\r
			cerr << "unknown species/option: " << species << "/" << option << endl;\r
			cerr << "usage: Proton/0, Proton/1, Proton/2 or Deuteron" << endl;\r
			exit(1);\r
		}\r
		\r
		TString ptfile = outputDir + "/" + species + "-" + kOutputTag[i] + "-Pt.root";\r
		m.AddFile(ptfile,0);\r
	}\r
	\r
	TString outputPt      = outputDir + "/" + species + "-" + outputTag + "-Pt.root";\r
	TString outputRatio   = outputDir + "/" + species + "-" + outputTag + "-Ratio.root";\r
	TString outputSpectra = outputDir + "/" + species + "-" + outputTag + "-Spectra.root";\r
	\r
	// pt\r
	\r
	m.OutputFile(outputPt.Data());\r
	m.Merge();\r
	\r
	// spectra\r
	\r
	AliLnDriver driver;\r
	\r
	driver.SetSpecies(species);\r
	\r
	driver.SetOutputFilenames(outputPt, outputRatio, outputSpectra);\r
	driver.SetOutputTag(outputTag);\r
	\r
	driver.SetMakeCorrections(0);\r
	driver.SetMakePt(0);\r
	driver.SetMakeRatio(1);\r
	driver.SetMakeSpectra(1);\r
	\r
	driver.Run();\r
	\r
	// merge all results for comparison\r
	\r
	TFileMerger m2, m3;\r
	\r
	for(Int_t i=0; i<kNper; ++i)\r
	{\r
		m2.AddFile((outputDir + "/" + species + "-" + kPeriod[i] + "-Ratio.root").Data(),0);\r
		m3.AddFile((outputDir + "/" + species + "-" + kPeriod[i] + "-Spectra.root").Data(),0);\r
	}\r
	\r
	m2.AddFile(outputRatio.Data(),0);\r
	m3.AddFile(outputSpectra.Data(),0);\r
	\r
	TString allRatios = outputDir + "/" + species + "-lhc10bcde1-Ratio.root";\r
	TString allSpectra = outputDir + "/" + species + "-lhc10bcde1-Spectra.root";\r
	\r
	m2.OutputFile(allRatios.Data());\r
	m3.OutputFile(allSpectra.Data());\r
	\r
	m2.Merge();\r
	m3.Merge();\r
	\r
	// compare periods\r
	\r
	if(!drawOutput) return 0;\r
	\r
	gROOT->ProcessLine(Form(".x DrawDir.C+g(\"%s\",\"Anti%s%s_Ratio_Pt\",\"lhc10bcde\",0,4.5, 0., 1.8, \"p_{T} (GeV/c)\", \"neg/pos\", 1, \"cRatio\",\"Particle ratio\")", allRatios.Data(), species.Data(), species.Data()));\r
	\r
	DrawOutputSpectraMult(allSpectra, species, ymin, ymax, 1, "lhc10bcde");\r
	\r
	return 0;\r
}\r