int main(int argc, char* argv[]) {
if (argc != 7 && argc != 8) {
std::cout << "USAGE: ./draw_vs_npv [data_dataset] [mc_SIGNAL_dataset] [mc_BG_dataset] [recoType] [jetAlgo] [norm ('LUMI' or 'SHAPE')] [flags=\"\"]" << std::endl;
exit(23);
}
gROOT->SetBatch();
std::string data_dataset(argv[1]);
std::string mc_photonjet(argv[2]);
std::string mc_QCD(argv[3]);
std::string recoType(argv[4]);
std::string jetAlgo(argv[5]);
std::string norm(argv[6]);
if (norm != "LUMI" && norm != "SHAPE") {
std::cout << "'" << norm << "' normalization not implemented yet." << std::endl;
std::cout << "Only 'LUMI' and 'SHAPE' currently supported." << std::endl;
std::cout << "Exiting." << std::endl;
exit(9811);
}
std::string flags = "";
if (argc == 8) {
std::string flags_str(argv[7]);
flags = flags_str;
}
std::string algoType;
if (recoType == "calo") {
algoType = jetAlgo;
} else {
algoType = recoType + jetAlgo;
}
if (recoType == "jpt" && jetAlgo == "akt4") {
algoType = "jptak4";
}
jetAlgo = (jetAlgo == "ak4") ? "AK4" : "AK8";
recoType = (recoType == "pf") ? "PFlow" : "Calo";
std::string postFix = recoType + jetAlgo;
postFix += "chs";
//Float_t etamax = 3.;
//bool sameEvents = false; //until njets histos have no overflows... or maybe use GetEntries instead of integral?
drawBase* db = new drawBase("PhotonJet", recoType, jetAlgo, OUTPUT_GRAPHS);
db->set_pdf_aussi((bool)false);
db->set_flags(flags);
db->set_isCMSArticle(false);
std::cout << "flags set." << std::endl;
TString dataFileName;
if (flags.length() > 0) {
dataFileName = TString::Format("PhotonJet_%s_%s_%s.root", data_dataset.c_str(), postFix.c_str(), flags.c_str());
} else {
dataFileName = TString::Format("PhotonJet_%s_%s.root", data_dataset.c_str(), postFix.c_str());
}
TFile* dataFile = TFile::Open(dataFileName);
if (dataFile) {
std::cout << "Opened data file '" << dataFileName << "'." << std::endl;
db->add_dataFile(dataFile, data_dataset);
}
TString mc1FileName;
if (flags.length() > 0) {
mc1FileName = TString::Format("PhotonJet_%s_%s_%s.root", mc_photonjet.c_str(), postFix.c_str(), flags.c_str());
} else {
mc1FileName = TString::Format("PhotonJet_%s_%s.root", mc_photonjet.c_str(), postFix.c_str());
}
TFile* mcPhotonJetFile = TFile::Open(mc1FileName);
std::cout << "Opened mc file '" << mc1FileName << "'." << std::endl;
if (mcPhotonJetFile) {
db->add_mcFile(mcPhotonJetFile, mc_photonjet, "#gamma + jets MC", BALANCING);
}
if (mc_QCD != "") {
TString mc2FileName;
if (flags.length() > 0) {
mc2FileName = TString::Format("PhotonJet_%s_%s_%s.root", mc_QCD.c_str(), postFix.c_str(), flags.c_str());
} else {
mc2FileName = TString::Format("PhotonJet_%s_%s.root", mc_QCD.c_str(), postFix.c_str());
}
TFile* mcQCDFile = TFile::Open(mc2FileName);
std::cout << "Opened mc file '" << mc2FileName << "'." << std::endl;
if (mcQCDFile && mc_QCD != mc_photonjet) {
db->add_mcFile(mcQCDFile, mc_QCD, "QCD MC", MPF);
}
}
// MC should already be normalized to a lumi of 1 pb-1
// Read luminosity
double dLumi = 1e6;
if (dataFile) {
TParameter<double>* lumi = static_cast<TParameter<double>*>(dataFile->Get("analysis/luminosity"));
dLumi = lumi->GetVal();
}
std::cout<< "Lumi "<< dLumi << std::endl;
// db->set_lumi(dLumi * 1e-6);
db->set_lumi(dLumi);
if (norm == "LUMI") {
db->set_lumiNormalization();
} else {
db->set_shapeNormalization();
}
db->setFolder("analysis");
std::string outputDir = "PhotonJetPlots_" + db->get_fullSuffix() + "/vs_npv";
db->set_outputdir(outputDir);
bool log = true;
gErrorIgnoreLevel = kWarning;
db->setOutputGraphs(OUTPUT_GRAPHS);
VertexBinning vertexBinning;
std::vector<std::pair<int, int> > vertexBins = vertexBinning.getBinning();
EtaBinning etaBinning;
size_t etaBinningSize = etaBinning.size();
db->set_rebin(2);
db->setFolder("analysis/vertex");
for (size_t i = 0; i < etaBinningSize; i++) {
db->set_legendTitle(etaBinning.getBinTitle(i));
TString responseName = TString::Format("resp_balancing_%s", etaBinning.getBinName(i).c_str());
db->drawHisto_vs_vertex(vertexBins, responseName.Data(), "Balancing Response", "", "Events", log);
// Raw jets
//responseName = TString::Format("resp_balancing_raw_%s", etaBinning.getBinName(i).c_str());
//db->drawHisto_vs_vertex(ptBins, responseName.Data(), "Balancing Response (raw jets)", "", "Events", log);
responseName = TString::Format("resp_mpf_%s", etaBinning.getBinName(i).c_str());
db->drawHisto_vs_vertex(vertexBins, responseName.Data(), "MPF Response", "", "Events", log);
// Raw jets
//responseName = TString::Format("resp_mpf_raw_%s", etaBinning.getBinName(i).c_str());
//db->drawHisto_vs_vertex(ptBins, responseName.Data(), "MPF Response (raw ME_{T})", "", "Events", log);
}
// Special case eta < 1.3
db->set_legendTitle("|#eta| < 1.3");
db->drawHisto_vs_vertex(vertexBins, "resp_balancing_eta0013", "Balancing Response", "", "Events", log);
//db->drawHisto_vs_pt(ptBins, "resp_balancing_raw_eta013", "Balancing Response (raw jets)", "", "Events", log);
db->drawHisto_vs_vertex(vertexBins, "resp_mpf_eta0013", "MPF Response", "", "Events", log);
//db->drawHisto_vs_vertex(ptBins, "resp_mpf_raw_eta013", "MPF Response (raw ME_{T})", "", "Events", log);
delete db;
db = NULL;
return 0;
}
int main(int argc, char* argv[]) {
if (argc != 7 && argc != 8) {
std::cout << "USAGE: ./drawPhotonJet [data_dataset] [mc_SIGNAL_dataset] [mc_BG_dataset] [recoType] [jetAlgo]" << std::endl;
exit(23);
}
setPrettyStyle();
std::string data_dataset(argv[1]);
std::string mc_photonjet(argv[2]);
std::string mc_QCD(argv[3]);
std::string recoType(argv[4]);
std::string jetAlgo(argv[5]);
std::string algoType;
if (recoType == "calo") {
algoType = jetAlgo;
} else {
algoType = recoType + jetAlgo;
}
if (recoType == "jpt" && jetAlgo == "akt5") {
algoType = "jptak5";
}
std::string flags = "";
jetAlgo = (jetAlgo == "ak5") ? "AK5" : "AK7";
recoType = (recoType == "pf") ? "PFlow" : "Calo";
std::string postFix = recoType + jetAlgo;
postFix += "chs";
TString dataFileName;
if (flags.length() > 0) {
dataFileName = TString::Format("PhotonJet_%s_%s_%s.root", data_dataset.c_str(), postFix.c_str(), flags.c_str());
} else {
dataFileName = TString::Format("PhotonJet_%s_%s.root", data_dataset.c_str(), postFix.c_str());
}
TFile* dataFile = TFile::Open(dataFileName);
std::cout << "Opened data file '" << dataFileName << "'." << std::endl;
//db->add_dataFile(dataFile, data_dataset);
TString mc1FileName;
if (flags.length() > 0) {
mc1FileName = TString::Format("PhotonJet_%s_%s_%s.root", mc_photonjet.c_str(), postFix.c_str(), flags.c_str());
} else {
mc1FileName = TString::Format("PhotonJet_%s_%s.root", mc_photonjet.c_str(), postFix.c_str());
}
TFile* mcPhotonJetFile = TFile::Open(mc1FileName);
std::cout << "Opened mc file '" << mc1FileName << "'." << std::endl;
if (mcPhotonJetFile) {
//db->add_mcFile(mcPhotonJetFile, mc_photonjet, "#gamma+jet MC", 46);
}
if (mc_QCD != "") {
TString mc2FileName;
if (flags.length() > 0) {
mc2FileName = TString::Format("PhotonJet_%s_%s_%s.root", mc_QCD.c_str(), postFix.c_str(), flags.c_str());
} else {
mc2FileName = TString::Format("PhotonJet_%s_%s.root", mc_QCD.c_str(), postFix.c_str());
}
TFile* mcQCDFile = TFile::Open(mc2FileName);
std::cout << "Opened mc file '" << mc2FileName << "'." << std::endl;
if (mcQCDFile && mc_QCD != mc_photonjet) {
//db->add_mcFile(mcQCDFile, mc_QCD, "QCD MC", 38);
}
}
// Create output directory
mkdir("plots", 0755);
TString directoryName = "";
if (mc_QCD.length() == 0)
directoryName = TString::Format("plots/%s_vs_%s_%s", data_dataset.c_str(), mc_photonjet.c_str(), postFix.c_str());
else
directoryName = TString::Format("plots/%s_vs_%s_plus_%s_%s", data_dataset.c_str(), mc_photonjet.c_str(), mc_QCD.c_str(), postFix.c_str());
mkdir(directoryName, 0755);
directoryName = TString::Format("%s/extrapolation", directoryName.Data());
mkdir(directoryName, 0755);
TParameter<double>* pLumi = static_cast<TParameter<double>*>(dataFile->Get("analysis/luminosity"));
double lumi = pLumi->GetVal() * 1e-9;
//bool log = true;
gErrorIgnoreLevel = kWarning;
EtaBinning etaBinning;
size_t etaBinningSize = etaBinning.size();
TString rootFolder = "analysis/new_extrapolation";
for (size_t i = 0; i < etaBinningSize; i++) {
const std::string& etaName = etaBinning.getBinName(i);
std::cout << "Processing " << etaName << std::endl;
TString responseName = TString::Format("%s/extrap_resp_balancing_%s_graph", rootFolder.Data(), etaName.c_str());
TString outputName = TString::Format("%s/extrap_resp_balancing_%s.pdf", directoryName.Data(), etaName.c_str());
TGraphErrors* data = (TGraphErrors*) dataFile->Get(responseName);
TGraphErrors* mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "Balancing", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response", etaBinning.getBinTitle(i), lumi, outputName.Data());
// Raw jets
responseName = TString::Format("%s/extrap_resp_balancing_raw_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_balancing_raw_%s.pdf", directoryName.Data(), etaName.c_str());
data = (TGraphErrors*) dataFile->Get(responseName);
mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "Balancing", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response (raw jets)", etaBinning.getBinTitle(i), lumi, outputName.Data());
responseName = TString::Format("%s/extrap_resp_mpf_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_mpf_%s.pdf", directoryName.Data(), etaName.c_str());
data = (TGraphErrors*) dataFile->Get(responseName);
mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "MPF", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response", etaBinning.getBinTitle(i), lumi, outputName.Data());
// Raw jets
responseName = TString::Format("%s/extrap_resp_mpf_raw_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_mpf_raw_%s.pdf", directoryName.Data(), etaName.c_str());
data = (TGraphErrors*) dataFile->Get(responseName);
mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "MPF", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response (raw #slashed{E_{t}})", etaBinning.getBinTitle(i), lumi, outputName.Data());
}
// Special case eta < 1.3
{
const std::string& etaName = "eta013";
std::cout << "Processing " << etaName << std::endl;
TString responseName = TString::Format("%s/extrap_resp_balancing_%s_graph", rootFolder.Data(), etaName.c_str());
TString outputName = TString::Format("%s/extrap_resp_balancing_%s.pdf", directoryName.Data(), etaName.c_str());
TGraphErrors* data = (TGraphErrors*) dataFile->Get(responseName);
TGraphErrors* mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "Balancing", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response", "#eta #leq 1.3", lumi, outputName.Data());
// Raw jets
responseName = TString::Format("%s/extrap_resp_balancing_raw_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_balancing_raw_%s.pdf", directoryName.Data(), etaName.c_str());
data = (TGraphErrors*) dataFile->Get(responseName);
mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "Balancing", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response (raw jets)", "#eta #leq 1.3", lumi, outputName.Data());
responseName = TString::Format("%s/extrap_resp_mpf_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_mpf_%s.pdf", directoryName.Data(), etaName.c_str());
data = (TGraphErrors*) dataFile->Get(responseName);
mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "MPF", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response", "#eta #leq 1.3", lumi, outputName.Data());
// Raw jets
responseName = TString::Format("%s/extrap_resp_mpf_raw_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_mpf_raw_%s.pdf", directoryName.Data(), etaName.c_str());
data = (TGraphErrors*) dataFile->Get(responseName);
mc = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawGraphs(data, mc, "MPF", "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response (raw #slashed{E_{t}})", "#eta #leq 1.3", lumi, outputName.Data());
}
//db->set_legendTitle("|#eta| < 1.3");
//db->drawHisto_vs_pt(ptBins, "resp_mpf_eta013", "MPF Response", "", "Events", log);
//db->drawHisto_vs_pt(ptBins, "resp_mpf_raw_eta013", "MPF Response (raw ME_{T})", "", "Events", log);
for (size_t i = 0; i < etaBinningSize; i++) {
const std::string& etaName = etaBinning.getBinName(i);
std::cout << "Processing " << etaName << std::endl;
TString responseName = TString::Format("%s/extrap_resp_balancing_%s_graph", rootFolder.Data(), etaName.c_str());
TString outputName = TString::Format("%s/extrap_resp_combined_%s", directoryName.Data(), etaName.c_str());
TGraphErrors* balancingData = (TGraphErrors*) dataFile->Get(responseName);
TGraphErrors* balancingMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
responseName = TString::Format("%s/extrap_resp_mpf_%s_graph", rootFolder.Data(), etaName.c_str());
TGraphErrors* mpfData = (TGraphErrors*) dataFile->Get(responseName);
TGraphErrors* mpfMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawCombinedGraphs(balancingData, balancingMC, mpfData, mpfMC, "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response", etaBinning.getBinTitle(i), lumi, outputName.Data());
responseName = TString::Format("%s/extrap_resp_balancing_raw_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_combined_raw_%s", directoryName.Data(), etaName.c_str());
balancingData = (TGraphErrors*) dataFile->Get(responseName);
balancingMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
responseName = TString::Format("%s/extrap_resp_mpf_raw_%s_graph", rootFolder.Data(), etaName.c_str());
mpfData = (TGraphErrors*) dataFile->Get(responseName);
mpfMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawCombinedGraphs(balancingData, balancingMC, mpfData, mpfMC, "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response (raw objects)", etaBinning.getBinTitle(i), lumi, outputName.Data());
}
// Eta < 1.3
{
const std::string& etaName = "eta013";
std::cout << "Processing " << etaName << std::endl;
TString responseName = TString::Format("%s/extrap_resp_balancing_%s_graph", rootFolder.Data(), etaName.c_str());
TString outputName = TString::Format("%s/extrap_resp_combined_%s", directoryName.Data(), etaName.c_str());
TGraphErrors* balancingData = (TGraphErrors*) dataFile->Get(responseName);
TGraphErrors* balancingMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
responseName = TString::Format("%s/extrap_resp_mpf_%s_graph", rootFolder.Data(), etaName.c_str());
TGraphErrors* mpfData = (TGraphErrors*) dataFile->Get(responseName);
TGraphErrors* mpfMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawCombinedGraphs(balancingData, balancingMC, mpfData, mpfMC, "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response", "#eta #leq 1.3", lumi, outputName.Data());
responseName = TString::Format("%s/extrap_resp_balancing_raw_%s_graph", rootFolder.Data(), etaName.c_str());
outputName = TString::Format("%s/extrap_resp_combined_raw_%s", directoryName.Data(), etaName.c_str());
balancingData = (TGraphErrors*) dataFile->Get(responseName);
balancingMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
responseName = TString::Format("%s/extrap_resp_mpf_raw_%s_graph", rootFolder.Data(), etaName.c_str());
mpfData = (TGraphErrors*) dataFile->Get(responseName);
mpfMC = (TGraphErrors*) mcPhotonJetFile->Get(responseName);
drawCombinedGraphs(balancingData, balancingMC, mpfData, mpfMC, "p_{t}^{2^{nd} jet} / p_{t}^{#gamma}", "Jet response (raw objects)", "#eta #leq 1.3", lumi, outputName.Data());
}
return 0;
}
