Skip to content

Commit

Permalink
Merge pull request #25220 from cardinia/from-CMSSW_10_4_X_2018-11-12-…
Browse files Browse the repository at this point in the history 
…2300

Update of the Zmumu validation tool, pull request to master branch
  • Loading branch information
@cmsbuild
cmsbuild committed Nov 21, 2018
2 parents 3c4c772 + 8dbb6c0 commit 4d5eced
Show file tree
Hide file tree
Showing 5 changed files with 113 additions and 14 deletions.
3 changes: 3 additions & 0 deletions Alignment/OfflineValidation/python/TkAlAllInOneTool/plottingOptions.py
View file
Expand Up @@ -168,6 +168,9 @@ class PlottingOptionsZMuMu(BasePlottingOptions):
"rebinetadiff": "2",
"rebineta": "2",
"rebinpt": "8",
"AutoSetRange": "false",
"CustomMinY": "90.85",
"CustomMaxY": "91.5",
}
needpackages = {"MuonAnalysis/MomentumScaleCalibration"}
validationclass = ZMuMuValidation
Expand Down
2 changes: 1 addition & 1 deletion Alignment/OfflineValidation/python/TkAlAllInOneTool/zMuMuValidationTemplates.py
View file
Expand Up @@ -307,6 +307,6 @@
TkAlStyle::legendheader = ".oO[legendheader]Oo.";
TkAlStyle::set(.oO[publicationstatus]Oo., .oO[era]Oo., ".oO[customtitle]Oo.", ".oO[customrighttitle]Oo.");
MultiHistoOverlapAll_.oO[resonance]Oo.(separatebycommas(filenames), separatebycommas(titles), separatebycommas(colors), separatebycommas(linestyles_new), separatebycommas(markerstyles_new), ".oO[datadir]Oo./.oO[PlotsDirName]Oo.", .oO[switchONfit]Oo.);
MultiHistoOverlapAll_.oO[resonance]Oo.(separatebycommas(filenames), separatebycommas(titles), separatebycommas(colors), separatebycommas(linestyles_new), separatebycommas(markerstyles_new), ".oO[datadir]Oo./.oO[PlotsDirName]Oo.", .oO[switchONfit]Oo., .oO[AutoSetRange]Oo., .oO[CustomMinY]Oo., .oO[CustomMaxY]Oo.);
}
"""
114 changes: 105 additions & 9 deletions MuonAnalysis/MomentumScaleCalibration/test/Macros/RooFit/MultiHistoOverlapAll_Base.C
View file
Expand Up @@ -102,7 +102,7 @@ struct Plot_1D{
}
}

void plot(TString fitFormula=""){
void plot(TString fitFormula="", bool AutoSetRange=false){
// Determine the plot ranges
unsigned int nValidHistos = histogram.size();
const unsigned int nValidHistosLimit = 4;
Expand All @@ -126,16 +126,31 @@ struct Plot_1D{
}
overallAvg /= overallSigmaSqInv;

// Update August 2018, implemented autorange for histograms
float MaxY=-1000.;
float MinY=1000.;


for (unsigned int f=0; f<nValidHistos; f++){
for (int bin=1; bin<=histogram.at(f)->GetNbinsX(); bin++){
float bincontent = histogram.at(f)->GetBinContent(bin);
float binerror = histogram.at(f)->GetBinError(bin);
if (binerror==0 && bincontent==0) continue;
if ((bincontent + binerror)>deviationThreshold*overallAvg) rangeMaxReduction = 0;
if(bincontent>=MaxY)MaxY=bincontent;
if(bincontent<=MinY)MinY=bincontent;

}
}
if (nValidHistos>nValidHistosLimit && rangeMaxReduction!=0) dampingFactorEff = dampingFactorEff*0.7;

float Yrange = MaxY-MinY;
if(AutoSetRange){ // Overwtite fixed range for Y axis with the one computed for the current histogram
ymin=MinY-Yrange/10.;
ymax=Yrange*1.6+ymin;
}


if (ymin>=ymax){ // If range is not already set
ymin = absMin/rangeFactor[0];
ymax = absMax*(rangeFactor[1]+dampingFactorEff-rangeMaxReduction);
Expand Down Expand Up @@ -220,15 +235,15 @@ void splitOptionRecursive(string rawoption, vector<string>& splitoptions, char d
}

void getCustomRanges(TString type, TString resonance, int iP, float minmax_plot[2]);
void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, string linestyles, string markerstyles, TString directory, TString resonance, TString type, bool switchONfit);
void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, string linestyles, string markerstyles, TString directory, TString resonance, TString type, bool switchONfit, bool AutoSetRange=false, float CustomMinY=90.85, float CustomMaxY=91.4);

void MultiHistoOverlapAll_Base(string files, string labels, string colors, string linestyles, string markerstyles, TString directory, TString resonance, bool switchONfit=false){
void MultiHistoOverlapAll_Base(string files, string labels, string colors, string linestyles, string markerstyles, TString directory, TString resonance, bool switchONfit=false, bool AutoSetRange=false, float CustomMinY=90.85, float CustomMaxY=91.4){
gSystem->mkdir(directory, true);
MultiHistoOverlapAll_Base_one(files, labels, colors, linestyles, markerstyles, directory, resonance, "mean", switchONfit);
MultiHistoOverlapAll_Base_one(files, labels, colors, linestyles, markerstyles, directory, resonance, "sigma", switchONfit);
MultiHistoOverlapAll_Base_one(files, labels, colors, linestyles, markerstyles, directory, resonance, "mean", switchONfit, AutoSetRange, CustomMinY, CustomMaxY);
MultiHistoOverlapAll_Base_one(files, labels, colors, linestyles, markerstyles, directory, resonance, "sigma", switchONfit, AutoSetRange, CustomMinY, CustomMaxY);
}

void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, string linestyles, string markerstyles, TString directory, TString resonance, TString type, bool switchONfit){
void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, string linestyles, string markerstyles, TString directory, TString resonance, TString type, bool switchONfit, bool AutoSetRange, float CustomMinY, float CustomMaxY){
gROOT->Reset();
if (TkAlStyle::status() == NO_STATUS) TkAlStyle::set(INTERNAL);
gROOT->ForceStyle();
Expand Down Expand Up @@ -266,6 +281,7 @@ void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, s
}

const int nhistos=8;
const int n2Dhistos = 2;
TFile** file = new TFile*[nfiles];
for (int f=0; f<nfiles; f++) file[f] = TFile::Open((strValidation_file[f]).c_str(), "read");

Expand All @@ -279,6 +295,10 @@ void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, s
TString("MassVsCosThetaCS/allHistos/") + type + TString("Histo"),
TString("MassVsPhiCS/allHistos/") + type + TString("Histo")
};
TString histo2Dname[n2Dhistos] ={
TString("MassVsEtaPhiPlus/allHistos/") + type + TString("Histo"),
TString("MassVsEtaPhiMinus/allHistos/") + type + TString("Histo"),
};
TString xtitle[nhistos] ={
"p^{T}_{#mu} (GeV)",
"#phi_{#mu+}",
Expand All @@ -289,6 +309,14 @@ void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, s
"cos #theta_{CS}",
"#phi_{CS}"
};
TString x2Dtitle[n2Dhistos] ={
"#phi_{#mu+}",
"#phi_{#mu-}"
};
TString y2Dtitle[n2Dhistos] ={
"#eta_{#mu+}",
"#eta_{#mu-}"
};
TString ytitle;
if (type=="mean") ytitle = "M_{#mu#mu} (GeV)";
else if (type=="sigma") ytitle = "#sigma_{#mu#mu} (GeV)";
Expand All @@ -302,6 +330,10 @@ void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, s
type + "MassVsCosThetaCS_ALL",
type + "MassVsPhiCS_ALL"
};
TString plotname2D[n2Dhistos] ={
type + "MassVsEtaPhiPlus",
type + "MassVsEtaPhiMinus"
};
TString fitFormula[nhistos]={
"[0]+[1]*cos(x+[2])",
"[0]+[1]*cos(x+[2])",
Expand Down Expand Up @@ -333,13 +365,76 @@ void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, s
static_cast<float>(-TMath::Pi())
};


// Plot 2D hystograms
// TO BE FIXED: something strange appears in the 2D plots for the sigma, this should be further investigated
Double_t zMass(91.1876);
Double_t Y1SMass(9.46);
Double_t maxDist(0.);
Double_t Mass=zMass;
Double_t fixrange=1.5;
if (resonance=="Y1S"){
Mass=Y1SMass;
fixrange=0.15;
}
for (int f=0; f<nfiles; f++){

if (file[f]!=0 && !file[f]->IsZombie()){


TH2D* histo2D[2];
histo2D[0]= (TH2D*)file[f]->Get(histo2Dname[0]); //histoMassVsEtaPhiPlus
histo2D[1]= (TH2D*)file[f]->Get(histo2Dname[1]); //histoMassVsEtaPhiMinus
for(int s=0;s<2;s++){
TCanvas dummycanvas;
dummycanvas.SetFillColor(0);
dummycanvas.cd()->SetTopMargin(0.07);
dummycanvas.cd()->SetRightMargin(0.15);
dummycanvas.cd()->SetLeftMargin(0.12);
histo2D[s]->SetTitle("");
histo2D[s]->GetZaxis()->SetLabelFont(42);
histo2D[s]->GetXaxis()->SetTitle(x2Dtitle[s]);
histo2D[s]->GetYaxis()->SetTitle(y2Dtitle[s]);
histo2D[s]->GetYaxis()->SetTitleOffset(0.9);
if(type=="mean"){//This is made to ensure that the range options are not used for the sigma
Double_t zMin=Mass;
Double_t zMax=Mass;

for(int nbinX=1;nbinX<=histo2D[s]->GetNbinsX();nbinX++){
for(int nbinY=1;nbinY<=histo2D[s]->GetNbinsY();nbinY++){
Double_t value = histo2D[s]->GetBinContent(nbinX,nbinY);
if(value<zMin)zMin=value;
if(value>zMax)zMax=value;
}
}
maxDist=fabs(zMax-Mass);
if(fabs(Mass-zMin) >= maxDist) maxDist=fabs(Mass-zMin);
histo2D[s]->GetZaxis()->SetRangeUser(Mass-fixrange,Mass+fixrange);//Default range Zmass +- 1.5 GeV (Ymass +- 0.15 GeV), NOTE: this will create empty bins when the bin content is either lower or higher than the fixed range
if(AutoSetRange)histo2D[s]->GetZaxis()->SetRangeUser(zMin,zMax); //Set range automatically

}
histo2D[s]->Draw("COLZ");
TString alignment_label = strValidation_label.at(f);
alignment_label.ReplaceAll(" ","_");
dummycanvas.SaveAs(Form("%s/%sc%s%s%s", directory.Data(),alignment_label.Data(),"_", plotname2D[s].Data(), ".png"));
dummycanvas.SaveAs(Form("%s/%sc%s%s%s", directory.Data(),alignment_label.Data(),"_", plotname2D[s].Data(), ".pdf"));
}

}

};


for (int iP=0; iP<nhistos; iP++){
TString theFitFormula="";
if (switchONfit && type=="mean") theFitFormula = fitFormula[iP];

float minmax_plot[2]={ -1, -1 };
//getCustomRanges(type, resonance, iP, minmax_plot);

if (type=="mean") {
minmax_plot[0]=CustomMinY;
minmax_plot[1]=CustomMaxY;
}
Plot_1D thePlot(directory, histoname[iP], plotname[iP], xtitle[iP], ytitle, plot_xmin[iP], plot_xmax[iP], minmax_plot[0], minmax_plot[1]);
for (int f=0; f<nfiles; f++){
int color=0, linestyle=0, markerstyle=0;
Expand Down Expand Up @@ -369,7 +464,9 @@ void MultiHistoOverlapAll_Base_one(string files, string labels, string colors, s
}
thePlot.addHistogramFromFile(file[f], strValidation_label.at(f), color, linestyle, markerstyle);
}
thePlot.plot(theFitFormula);
if(type=="mean")thePlot.plot(theFitFormula,AutoSetRange);
else thePlot.plot(theFitFormula,true);

}

for (int f=nfiles-1; f>=0; f--){ if (file[f]!=0 && file[f]->IsOpen()) file[f]->Close(); }
Expand Down Expand Up @@ -523,4 +620,3 @@ void getCustomRanges(TString type, TString resonance, int iP, float minmax_plot[
} // End custom ranges

}

4 changes: 2 additions & 2 deletions MuonAnalysis/MomentumScaleCalibration/test/Macros/RooFit/MultiHistoOverlapAll_Y1S.C
View file
@@ -1,5 +1,5 @@
#include "MultiHistoOverlapAll_Base.C"

void MultiHistoOverlapAll_Y1S(string files, string labels, string colors = "", string linestyles = "", string markerstyles = "", TString directory = ".", bool switchONfit = false){
MultiHistoOverlapAll_Base(files, labels, colors, linestyles, markerstyles, directory, "Y1S", switchONfit);
void MultiHistoOverlapAll_Y1S(string files, string labels, string colors = "", string linestyles = "", string markerstyles = "", TString directory = ".", bool switchONfit = false, bool AutoSetRange=false, float CustomMinY=9.3, float CustomMaxY=9.6){//DEFAULT RANGE TO BE FIXED
MultiHistoOverlapAll_Base(files, labels, colors, linestyles, markerstyles, directory, "Y1S", switchONfit, AutoSetRange, CustomMinY, CustomMaxY);
}
4 changes: 2 additions & 2 deletions MuonAnalysis/MomentumScaleCalibration/test/Macros/RooFit/MultiHistoOverlapAll_Z.C
View file
@@ -1,5 +1,5 @@
#include "MultiHistoOverlapAll_Base.C"

void MultiHistoOverlapAll_Z(string files, string labels, string colors = "", string linestyles = "", string markerstyles = "", TString directory = ".", bool switchONfit = false){
MultiHistoOverlapAll_Base(files, labels, colors, linestyles, markerstyles, directory, "Z", switchONfit);
void MultiHistoOverlapAll_Z(string files, string labels, string colors = "", string linestyles = "", string markerstyles = "", TString directory = ".", bool switchONfit = false, bool AutoSetRange=false, float CustomMinY=90.85, float CustomMaxY=91.4){
MultiHistoOverlapAll_Base(files, labels, colors, linestyles, markerstyles, directory, "Z", switchONfit, AutoSetRange, CustomMinY, CustomMaxY);
}

0 comments on commit 4d5eced

Please sign in to comment.