void drawEventRPhi(double vxlb, double vylb, double vxl0, double vyl0, double pmu1, double phimu1, double pmu2, double phimu2, double ppr, double phipr, double ppi, double phipi, int colors = 0)
{
Color_t colMu1(1), colMu2(1), colPr(1), colPi(1), colL0(1), colLb(1);
switch(colors)
{
case 0:
colLb=1; colL0=2; colMu1=2; colMu2=2; colPr=3; colPi=4; break;
case 1:
colLb=11; colL0=50; colMu1=50; colMu2=50; colPr=8; colPi=9; break;
}
// draw the vertices
TMarker *m;
const double xlb = vxlb;
const double ylb = vylb;
m = new TMarker(xlb,ylb,7);
m->SetMarkerColor(colLb);
m->Draw();
const double xl0 = vxl0;
const double yl0 = vyl0;
m = new TMarker(xl0,yl0,7);
m->SetMarkerColor(colL0);
m->Draw();
// draw the l0 flight line
TLine *l;
l = new TLine(vxlb,vylb,vxl0,vyl0);
l->SetLineColor(colL0);
l->SetLineStyle(2);
l->Draw();
// draw the muons
TArrow *a;
const double xmu1 = xlb + scalemu * pmu1 * TMath::Cos(phimu1);
const double ymu1 = ylb + scalemu * pmu1 * TMath::Sin(phimu1);
a = new TArrow(xlb,ylb,xmu1,ymu1,.01,">");
a->SetLineColor(colMu1);
a->Draw();
const double xmu2 = xlb + scalemu * pmu2 * TMath::Cos(phimu2);
const double ymu2 = ylb + scalemu * pmu2 * TMath::Sin(phimu2);
a = new TArrow(xlb,ylb,xmu2,ymu2,.01,">");
a->SetLineColor(colMu2);
a->Draw();
// draw the p and pi
const double xpr = xl0 + scalepr * ppr * TMath::Cos(phipr);
const double ypr = yl0 + scalepr * ppr * TMath::Sin(phipr);
a = new TArrow(xl0,yl0,xpr,ypr,.01,">");
a->SetLineColor(colPr);
a->Draw();
const double xpi = xl0 + scalepi * ppi * TMath::Cos(phipi);
const double ypi = yl0 + scalepi * ppi * TMath::Sin(phipi);
a = new TArrow(xl0,yl0,xpi,ypi,.01,">");
a->SetLineColor(colPi);
a->Draw();
}
void gtime2(Int_t nsteps = 200, Int_t np=5000) {
if (np > 5000) np = 5000;
Int_t color[5000];
Double_t cosphi[5000], sinphi[5000], speed[5000];
TRandom3 r;
Double_t xmin = 0, xmax = 10, ymin = -10, ymax = 10;
TGraphTime *g = new TGraphTime(nsteps,xmin,ymin,xmax,ymax);
g->SetTitle("TGraphTime demo 2;X;Y");
Int_t i,s;
Double_t phi,fact = xmax/Double_t(nsteps);
for (i=0;i<np;i++) { //calculate some object parameters
speed[i] = r.Uniform(0.5,1);
phi = r.Gaus(0,TMath::Pi()/6.);
cosphi[i] = fact*speed[i]*TMath::Cos(phi);
sinphi[i] = fact*speed[i]*TMath::Sin(phi);
Double_t rc = r.Rndm();
color[i] = kRed;
if (rc > 0.3) color[i] = kBlue;
if (rc > 0.7) color[i] = kYellow;
}
for (s=0;s<nsteps;s++) { //fill the TGraphTime step by step
for (i=0;i<np;i++) {
Double_t xx = s*cosphi[i];
if (xx < xmin) continue;
Double_t yy = s*sinphi[i];
TMarker *m = new TMarker(xx,yy,25);
m->SetMarkerColor(color[i]);
m->SetMarkerSize(1.5 -s/(speed[i]*nsteps));
g->Add(m,s);
}
g->Add(new TPaveLabel(.70,.92,.98,.99,Form("shower at %5.3f nsec",3.*s/nsteps),"brNDC"),s);
}
g->Draw();
}
void DrawLabel(TString txt, double x, double y, int mstyle, int col, double msize) {
TMarker *m = new TMarker(x,y,mstyle);
m->SetNDC();
m->SetMarkerSize(msize); m->SetMarkerColor(col);
TLine *l = new TLine();
l->SetLineWidth(2); l->SetLineColor(col);
m->Draw();
tex->SetTextSize(0.04);
tex->SetTextAlign(12); tex->SetTextColor(col);
tex->DrawLatex(x+0.025,y,txt);
tex->SetTextColor(kBlack);
}
void myMarkerText(Double_t x,Double_t y,Int_t color,Int_t mstyle, const char *text,Float_t msize)
{
Double_t tsize=0.06;
TMarker *marker = new TMarker(x-(0.4*tsize),y,8);
marker->SetMarkerColor(color); marker->SetNDC();
marker->SetMarkerStyle(mstyle);
marker->SetMarkerSize(msize);
marker->Draw();
TLatex l; l.SetTextAlign(12); //l.SetTextSize(tsize);
l.SetNDC();
l.DrawLatex(x,y,text);
}
void myMarkerText(Double_t x,Double_t y,Int_t color,Int_t mstyle,char *text)
{
// printf("**myMarker: text= %s\ m ",text);
Double_t tsize=0.06;
TMarker *marker = new TMarker(x-(0.4*tsize),y,8);
marker->SetMarkerColor(color); marker->SetNDC();
marker->SetMarkerStyle(mstyle);
marker->SetMarkerSize(2.0);
marker->Draw();
TLatex l; l.SetTextAlign(12); //l.SetTextSize(tsize);
l.SetNDC();
l.DrawLatex(x,y,text);
}
// Draw on a given canvas the full set of working points
void overlayWorkingPoints(TCanvas *c1,
TTree *signalTree, TTree *backgroundTree,
const TString *cutFileNames,
int markerColor, int markerStyle,
TLegend *leg, const TString legendText){
// Now loop over working points
for(int iwp = 0; iwp<nWP; iwp++){
// Load the working point from a ROOT file
TFile *cutFile = new TFile(cutFileNames[iwp]);
if( !cutFile )
assert(0);
VarCut *cutObject = (VarCut*)cutFile->Get("cuts");
if( !cutObject )
assert(0);
// Compute the efficiencies
float effSignal, effBackground;
findEfficiencies(signalTree, backgroundTree, effSignal, effBackground,
cutObject);
printf("Computed eff for cut from %s, effS= %.3f effB= %.3f\n",
cutFileNames[iwp].Data(), effSignal, effBackground);
// Make a marker and draw it.
TMarker *marker = new TMarker(effSignal, 1.0-effBackground, 20);
marker->SetMarkerSize(2);
marker->SetMarkerColor(markerColor);
marker->SetMarkerStyle(markerStyle);
marker->Draw("same");
// Add marker to the legend only once. Do not draw the legend here,
// it is drawn in the main function later
if( iwp == 0 ){
if( !leg )
assert(0);
leg->AddEntry(marker, legendText, "p");
}
c1->Update();
cutFile->Close();
}
}
void showHistogram1d(TH1* histogram,
const std::string& xAxisTitle,
Float_t* genX,
const std::string& outputFileName)
{
TCanvas* canvas = new TCanvas("canvas", "canvas", 800, 600);
canvas->SetFillColor(10);
canvas->SetBorderSize(2);
canvas->SetTopMargin(0.10);
canvas->SetLeftMargin(0.16);
canvas->SetRightMargin(0.14);
canvas->SetBottomMargin(0.12);
TAxis* xAxis = histogram->GetXaxis();
xAxis->SetTitle(xAxisTitle.data());
xAxis->SetTitleOffset(1.15);
TAxis* yAxis = histogram->GetYaxis();
yAxis->SetTitle("Sampling Points");
yAxis->SetTitleOffset(1.60);
histogram->SetLineColor(1);
histogram->SetLineWidth(2);
histogram->SetMarkerColor(1);
histogram->SetMarkerStyle(20);
histogram->Draw("e1p");
TMarker* genMarker = 0;
if ( genX ) {
genMarker = new TMarker(*genX, 0.10, 34);
genMarker->SetMarkerColor(1);
genMarker->SetMarkerSize(2);
genMarker->Draw();
}
canvas->Update();
size_t idx = outputFileName.find_last_of('.');
std::string outputFileName_plot = std::string(outputFileName, 0, idx);
if ( idx != std::string::npos ) canvas->Print(std::string(outputFileName_plot).append(std::string(outputFileName, idx)).data());
canvas->Print(std::string(outputFileName_plot).append(".png").data());
canvas->Print(std::string(outputFileName_plot).append(".pdf").data());
canvas->Print(std::string(outputFileName_plot).append(".root").data());
delete genMarker;
delete canvas;
}
vector<TObject *> * mark_overflows(TH1F * histogram) {
Double_t maximum_value = histogram->GetMaximum();
Double_t minimum_value = histogram->GetMinimum();
vector<TObject *> * overflow_markers = new vector<TObject *>();
Float_t * bins = histogram->GetArray();
Int_t number_of_bins = histogram->GetNbinsX();
// note that the bins array has the following structure:
// bins[0] is the underflow bin
// bins[1] through bins[number_of_bins] are the actual bins
// bins[number_of_bins + 1] is the overflow bin
for(int j = 1; j <= number_of_bins; j++) {
if (bins[j] > maximum_value || bins[j] < minimum_value) {
Double_t x1 = histogram->GetXaxis()->GetBinLowEdge(j);
Double_t x2 = histogram->GetXaxis()->GetBinUpEdge(j);
TMarker * m = new TMarker();
m->SetX((x2 + x1) / 2.0);
m->SetMarkerColor(kRed);
m->SetMarkerSize(MARKER_SIZE * 2);
if(bins[j] > maximum_value) {
m->SetY(maximum_value - MARKER_Y_SHIFT);
m->SetMarkerStyle(22); // up arrow
} else {
m->SetY(minimum_value + MARKER_Y_SHIFT);
m->SetMarkerStyle(23); // down arrow
}
overflow_markers->push_back(m);
}
}
return overflow_markers;
}
void /home/users/m/k/mkomm/Diff13/analysis/unfolding/result/nominal/multi_top_y_nol()
{
//=========Macro generated from canvas: cvMulti0.984295447508/
//========= (Mon May 15 15:02:17 2017) by ROOT version6.02/05
TCanvas *cvMulti0.984295447508 = new TCanvas("cvMulti0.984295447508", "",0,0,800,650);
gStyle->SetOptFit(1);
gStyle->SetOptStat(0);
gStyle->SetOptTitle(0);
cvMulti0.984295447508->SetHighLightColor(2);
cvMulti0.984295447508->Range(-0.5419355,-0.2962025,3.329032,1.982278);
cvMulti0.984295447508->SetFillColor(0);
cvMulti0.984295447508->SetBorderMode(0);
cvMulti0.984295447508->SetBorderSize(2);
cvMulti0.984295447508->SetTickx(1);
cvMulti0.984295447508->SetTicky(1);
cvMulti0.984295447508->SetLeftMargin(0.14);
cvMulti0.984295447508->SetRightMargin(0.24);
cvMulti0.984295447508->SetTopMargin(0.08);
cvMulti0.984295447508->SetBottomMargin(0.13);
cvMulti0.984295447508->SetFrameFillStyle(0);
cvMulti0.984295447508->SetFrameLineWidth(0);
cvMulti0.984295447508->SetFrameBorderMode(0);
cvMulti0.984295447508->SetFrameBorderSize(0);
cvMulti0.984295447508->SetFrameFillStyle(0);
cvMulti0.984295447508->SetFrameLineWidth(0);
cvMulti0.984295447508->SetFrameBorderMode(0);
cvMulti0.984295447508->SetFrameBorderSize(0);
TH2F *axis0.09336527505115 = new TH2F("axis0.09336527505115","",50,0,2.4,50,0,1.8);
axis0.09336527505115->SetLineStyle(0);
axis0.09336527505115->SetMarkerStyle(20);
axis0.09336527505115->SetMarkerSize(0.16);
axis0.09336527505115->GetXaxis()->SetTitle("top quark |y|");
axis0.09336527505115->GetXaxis()->SetNdivisions(1005);
axis0.09336527505115->GetXaxis()->SetLabelFont(43);
axis0.09336527505115->GetXaxis()->SetLabelOffset(0.0077);
axis0.09336527505115->GetXaxis()->SetLabelSize(32);
axis0.09336527505115->GetXaxis()->SetTitleSize(35);
axis0.09336527505115->GetXaxis()->SetTickLength(0.04032258);
axis0.09336527505115->GetXaxis()->SetTitleOffset(1.15);
axis0.09336527505115->GetXaxis()->SetTitleFont(43);
axis0.09336527505115->GetYaxis()->SetTitle(" Scale factor");
axis0.09336527505115->GetYaxis()->SetNdivisions(512);
axis0.09336527505115->GetYaxis()->SetLabelFont(43);
axis0.09336527505115->GetYaxis()->SetLabelOffset(0.0077);
axis0.09336527505115->GetYaxis()->SetLabelSize(32);
axis0.09336527505115->GetYaxis()->SetTitleSize(35);
axis0.09336527505115->GetYaxis()->SetTickLength(0.03164557);
axis0.09336527505115->GetYaxis()->SetTitleOffset(1.3);
axis0.09336527505115->GetYaxis()->SetTitleFont(43);
axis0.09336527505115->GetZaxis()->SetLabelFont(43);
axis0.09336527505115->GetZaxis()->SetLabelOffset(0.0077);
axis0.09336527505115->GetZaxis()->SetLabelSize(32);
axis0.09336527505115->GetZaxis()->SetTitleSize(34);
axis0.09336527505115->GetZaxis()->SetTitleOffset(1.3);
axis0.09336527505115->GetZaxis()->SetTitleFont(43);
axis0.09336527505115->Draw("AXIS");
TF1 *tf13 = new TF1("tf1","1",0,2.4);
tf13->SetFillColor(19);
tf13->SetFillStyle(0);
tf13->SetMarkerStyle(20);
tf13->SetMarkerSize(0.16);
tf13->SetLineWidth(1);
tf13->GetXaxis()->SetNdivisions(505);
tf13->GetXaxis()->SetLabelFont(43);
tf13->GetXaxis()->SetLabelOffset(0.0077);
tf13->GetXaxis()->SetLabelSize(32);
tf13->GetXaxis()->SetTitleSize(34);
tf13->GetXaxis()->SetTickLength(0.05);
tf13->GetXaxis()->SetTitleOffset(1.15);
tf13->GetXaxis()->SetTitleFont(43);
tf13->GetYaxis()->SetNdivisions(512);
tf13->GetYaxis()->SetLabelFont(43);
tf13->GetYaxis()->SetLabelOffset(0.0077);
tf13->GetYaxis()->SetLabelSize(32);
tf13->GetYaxis()->SetTitleSize(34);
tf13->GetYaxis()->SetTitleOffset(1.3);
tf13->GetYaxis()->SetTitleFont(43);
tf13->Draw("LSame");
TMarker *marker = new TMarker(0.0892,0.9606917,20);
Int_t ci; // for color index setting
TColor *color; // for color definition with alpha
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
TLine *line = new TLine(0.0892,0.9359605,0.0892,0.9854228);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.2,0,0.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.0892,0.9606917,0.3142,0.9554859);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.3142,0.9554859,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.3142,0.9323344,0.3142,0.9786374);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.45,0,0.45,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.3142,0.9554859,0.5642,0.9562376);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.5642,0.9562376,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.5642,0.9291835,0.5642,0.9832918);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.7,0,0.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.5642,0.9562376,0.8142,0.940729);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.8142,0.940729,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.8142,0.9117791,0.8142,0.9696789);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.95,0,0.95,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.8142,0.940729,1.0642,1.028565);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.0642,1.028565,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.0642,0.9987074,1.0642,1.058422);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.2,0,1.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.0642,1.028565,1.3392,1.024151);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.3392,1.024151,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.3392,0.9916412,1.3392,1.05666);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.5,0,1.5,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.3392,1.024151,1.5892,0.9990193);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.5892,0.9990193,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.5892,0.9494996,1.5892,1.048539);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.7,0,1.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.5892,0.9990193,2.0392,0.9212695);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(2.0392,0.9212695,20);
ci = TColor::GetColor("#f9910c");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(2.0392,0.8767255,2.0392,0.9658136);
ci = TColor::GetColor("#f9910c");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(2.4,0,2.4,1.8);
line->SetLineStyle(2);
line->Draw();
marker = new TMarker(0.1036,1.197582,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.1036,1.070044,0.1036,1.325121);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.2,0,0.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.1036,1.197582,0.3286,1.07941);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.3286,1.07941,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.3286,0.9690778,0.3286,1.189742);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.45,0,0.45,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.3286,1.07941,0.5786,1.238685);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.5786,1.238685,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.5786,1.103282,0.5786,1.374089);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.7,0,0.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.5786,1.238685,0.8286,1.070291);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.8286,1.070291,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.8286,0.9410973,0.8286,1.199485);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.95,0,0.95,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.8286,1.070291,1.0786,0.7963992);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.0786,0.7963992,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.0786,0.6813629,1.0786,0.9114354);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.2,0,1.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.0786,0.7963992,1.3536,0.9233731);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.3536,0.9233731,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.3536,0.7879375,1.3536,1.058809);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.5,0,1.5,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.3536,0.9233731,1.6036,1.066334);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.6036,1.066334,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.6036,0.8539173,1.6036,1.278751);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.7,0,1.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.6036,1.066334,2.0536,1.283384);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(2.0536,1.283384,20);
ci = TColor::GetColor("#0a9128");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(2.0536,1.067281,2.0536,1.499488);
ci = TColor::GetColor("#0a9128");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(2.4,0,2.4,1.8);
line->SetLineStyle(2);
line->Draw();
marker = new TMarker(0.118,0.1968883,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.118,0.1722209,0.118,0.2215558);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.2,0,0.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.118,0.1968883,0.343,0.2348717);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.343,0.2348717,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.343,0.2130683,0.343,0.2566752);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.45,0,0.45,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.343,0.2348717,0.593,0.2059685);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.593,0.2059685,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.593,0.1802414,0.593,0.2316957);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.7,0,0.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.593,0.2059685,0.843,0.2345659);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.843,0.2345659,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.843,0.208081,0.843,0.2610507);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.95,0,0.95,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.843,0.2345659,1.093,0.3094085);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.093,0.3094085,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.093,0.2799161,1.093,0.3389009);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.2,0,1.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.093,0.3094085,1.368,0.347837);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.368,0.347837,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.368,0.2999248,1.368,0.3957492);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.5,0,1.5,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.368,0.347837,1.618,0.5309299);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.618,0.5309299,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.618,0.4081874,1.618,0.6536723);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.7,0,1.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.618,0.5309299,2.068,0.2949066);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(2.068,0.2949066,20);
ci = TColor::GetColor("#999999");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(2.068,0,2.068,0.7323252);
ci = TColor::GetColor("#999999");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(2.4,0,2.4,1.8);
line->SetLineStyle(2);
line->Draw();
marker = new TMarker(0.1252,0.9784018,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.1252,0.8521009,0.1252,1.104703);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.2,0,0.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.1252,0.9784018,0.3502,0.9447267);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.3502,0.9447267,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.3502,0.8411522,0.3502,1.048301);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.45,0,0.45,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.3502,0.9447267,0.6002,0.7547158);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.6002,0.7547158,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.6002,0.6180429,0.6002,0.8913887);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.7,0,0.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.6002,0.7547158,0.8502,1.12315);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(0.8502,1.12315,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(0.8502,0.9840732,0.8502,1.262226);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(0.95,0,0.95,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(0.8502,1.12315,1.1002,1.107947);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.1002,1.107947,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.1002,0.9620515,1.1002,1.253843);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.2,0,1.2,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.1002,1.107947,1.3752,0.6245485);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.3752,0.6245485,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.3752,0.462954,1.3752,0.786143);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.5,0,1.5,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.3752,0.6245485,1.6252,0.8379717);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(1.6252,0.8379717,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(1.6252,0.5506004,1.6252,1.125343);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(1.7,0,1.7,1.8);
line->SetLineStyle(2);
line->Draw();
line = new TLine(1.6252,0.8379717,2.0752,0.3802364);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(3);
line->Draw();
marker = new TMarker(2.0752,0.3802364,20);
ci = TColor::GetColor("#ff0505");
marker->SetMarkerColor(ci);
marker->SetMarkerStyle(20);
marker->SetMarkerSize(1.2);
marker->Draw();
line = new TLine(2.0752,0.06910222,2.0752,0.6913706);
ci = TColor::GetColor("#ff0505");
line->SetLineColor(ci);
line->SetLineWidth(2);
line->Draw();
line = new TLine(2.4,0,2.4,1.8);
line->SetLineStyle(2);
line->Draw();
TH2F *axis0.0933652750511_copy6 = new TH2F("axis0.0933652750511_copy6","",50,0,2.4,50,0,1.8);
axis0.0933652750511_copy6->SetDirectory(0);
axis0.0933652750511_copy6->SetLineStyle(0);
axis0.0933652750511_copy6->SetMarkerStyle(20);
axis0.0933652750511_copy6->SetMarkerSize(0.16);
axis0.0933652750511_copy6->GetXaxis()->SetTitle("top quark |y|");
axis0.0933652750511_copy6->GetXaxis()->SetNdivisions(1005);
axis0.0933652750511_copy6->GetXaxis()->SetLabelFont(43);
axis0.0933652750511_copy6->GetXaxis()->SetLabelOffset(0.0077);
axis0.0933652750511_copy6->GetXaxis()->SetLabelSize(32);
axis0.0933652750511_copy6->GetXaxis()->SetTitleSize(35);
axis0.0933652750511_copy6->GetXaxis()->SetTickLength(0.04032258);
axis0.0933652750511_copy6->GetXaxis()->SetTitleOffset(1.15);
axis0.0933652750511_copy6->GetXaxis()->SetTitleFont(43);
axis0.0933652750511_copy6->GetYaxis()->SetTitle(" Scale factor");
axis0.0933652750511_copy6->GetYaxis()->SetNdivisions(512);
axis0.0933652750511_copy6->GetYaxis()->SetLabelFont(43);
axis0.0933652750511_copy6->GetYaxis()->SetLabelOffset(0.0077);
axis0.0933652750511_copy6->GetYaxis()->SetLabelSize(32);
axis0.0933652750511_copy6->GetYaxis()->SetTitleSize(35);
axis0.0933652750511_copy6->GetYaxis()->SetTickLength(0.03164557);
axis0.0933652750511_copy6->GetYaxis()->SetTitleOffset(1.3);
axis0.0933652750511_copy6->GetYaxis()->SetTitleFont(43);
axis0.0933652750511_copy6->GetZaxis()->SetLabelFont(43);
axis0.0933652750511_copy6->GetZaxis()->SetLabelOffset(0.0077);
axis0.0933652750511_copy6->GetZaxis()->SetLabelSize(32);
axis0.0933652750511_copy6->GetZaxis()->SetTitleSize(34);
axis0.0933652750511_copy6->GetZaxis()->SetTitleOffset(1.3);
axis0.0933652750511_copy6->GetZaxis()->SetTitleFont(43);
axis0.0933652750511_copy6->Draw("sameaxis");
TPaveText *pt = new TPaveText(0.14,0.94,0.14,0.94,"brNDC");
pt->SetBorderSize(0);
pt->SetFillColor(0);
pt->SetTextAlign(11);
pt->SetTextFont(63);
pt->SetTextSize(34);
TText *AText = pt->AddText("CMS");
pt->Draw();
pt = new TPaveText(0.24,0.94,0.24,0.94,"brNDC");
pt->SetBorderSize(0);
pt->SetFillColor(0);
pt->SetTextAlign(11);
pt->SetTextFont(53);
pt->SetTextSize(34);
AText = pt->AddText("Preliminary");
pt->Draw();
cvMulti0.984295447508->Modified();
cvMulti0.984295447508->cd();
cvMulti0.984295447508->SetSelected(cvMulti0.984295447508);
}
void plotAnitaEventMap(Adu5Pat *patPtr,double phi,double theta){
double sourceLon,sourceLat,headLon,headLat,phi10Lon,phi10Lat,phi6Lon,phi6Lat,phi14Lon,phi14Lat,actualLat,actualLon,actual2Lat,actual2Lon;
float xEvent,yEvent,xAnita,yAnita,anitaLat,anitaLon,anitaAlt,xHead,yHead,x10,y10,x6,y6,x14,y14,yActual,xActual,yActual2,xActual2;
anitaLat = patPtr->latitude;
anitaLon = patPtr->longitude;
anitaAlt = patPtr->altitude;
UsefulAdu5Pat usefulPat(patPtr);
std::cout << "source " << std::endl;
//int sourceLoc = usefulPat.getSourceLonAndLatAltZero((180-phi)/180.*PI,(theta)/180.*PI,sourceLon,sourceLat);
int sourceLoc = usefulPat.getSourceLonAndLatAltZero((phi)/180.*PI,(theta)/180.*PI,sourceLon,sourceLat);
std::cout << std::endl << "heading " << std::endl;
int headLoc = usefulPat.getSourceLonAndLatAltZero(0./180.*PI,10./180.*PI,headLon,headLat);
std::cout << std::endl << "phi 10 " << std::endl;
int headLoc10 = usefulPat.getSourceLonAndLatAltZero(180./180.*PI,10./180.*PI,phi10Lon,phi10Lat);
std::cout << std::endl << "phi 14 " << std::endl;
int headLoc14 = usefulPat.getSourceLonAndLatAltZero(270./180.*PI,10./180.*PI,phi14Lon,phi14Lat);
std::cout << std::endl << "phi 6 " << std::endl;
int headLoc6 = usefulPat.getSourceLonAndLatAltZero(90./180.*PI,10./180.*PI,phi6Lon,phi6Lat);
std::cout << std::endl << "actual 14.5 " << std::endl;
int actualLoc = usefulPat.getSourceLonAndLatAltZero(phi/180.*PI,12.5/180.*PI,actualLon,actualLat);
std::cout << std::endl << "actual 4.5 " << std::endl;
int actualLoc2 = usefulPat.getSourceLonAndLatAltZero(phi/180.*PI,7.5/180.*PI,actual2Lon,actual2Lat);
//int sourceLoc = usefulPat.getSourceLonAndLatAltZero((phi)/180.*PI,(TMath::PiOver2()-theta)/180.*PI,sourceLon,sourceLat);
TImage *map = TImage::Open("/home/mottram/work/eventCorrelator/macros/antarcticaIceMap.png");
std::cout << "sourceLoc " << sourceLoc << " phi " << phi << " theta " << theta << " lon " << sourceLon << " lat " << sourceLat << std::endl;
gStyle->SetMarkerColor(kBlack);
//gStyle->SetMarkerSize(2);
gStyle->SetTextSize(0.02);
TMarker *anitaPos = new TMarker(xAnita,yAnita,23);
getRelXYFromLatLong(anitaLat,anitaLon,xAnita,yAnita);
getRelXYFromLatLong(static_cast<float>(sourceLat),static_cast<float>(sourceLon),xEvent,yEvent);
getRelXYFromLatLong(static_cast<float>(headLat),static_cast<float>(headLon),xHead,yHead);
getRelXYFromLatLong(static_cast<float>(phi10Lat),static_cast<float>(phi10Lon),x10,y10);
getRelXYFromLatLong(static_cast<float>(phi14Lat),static_cast<float>(phi14Lon),x14,y14);
getRelXYFromLatLong(static_cast<float>(phi6Lat),static_cast<float>(phi6Lon),x6,y6);
getRelXYFromLatLong(static_cast<float>(actualLat),static_cast<float>(actualLon),xActual,yActual);
getRelXYFromLatLong(static_cast<float>(actual2Lat),static_cast<float>(actual2Lon),xActual2,yActual2);
TCanvas *canMap=(TCanvas*)gROOT->FindObject("canMap");
if(!canMap)
canMap = new TCanvas("canMap","canMap",(int)xSize,(int)ySize);
canMap->Clear();
canMap->SetLogz();
canMap->SetTopMargin(0);
canMap->SetBottomMargin(0);
canMap->SetLeftMargin(0);
canMap->SetRightMargin(0);
map->Draw("");
TMarker *headingPos = new TMarker(xHead,yHead,29);
TMarker *heading14Pos = new TMarker(x14,y14,29);
TMarker *heading10Pos = new TMarker(x10,y10,29);
TMarker *heading6Pos = new TMarker(x6,y6,29);
TMarker *actualPos = new TMarker(xActual,yActual,29);
TMarker *actual2Pos = new TMarker(xActual2,yActual2,29);
headingPos->SetMarkerColor(kRed);
heading14Pos->SetMarkerColor(kGray);
heading10Pos->SetMarkerColor(kGray+2);
heading6Pos->SetMarkerColor(kViolet);
actualPos->SetMarkerColor(kRed+2);//12.5 theta
actual2Pos->SetMarkerColor(kBlue+2);//7.5 theta
headingPos->Draw("");
heading14Pos->Draw("");
heading10Pos->Draw("");
heading6Pos->Draw("");
actualPos->Draw("");
actual2Pos->Draw("");
anitaPos->DrawMarker(xAnita,yAnita);
TLatex *positionLabel=0;
char label[FILENAME_MAX];
if(sourceLoc==0){
if(anitaAlt<0){
sprintf(label,"Could not get event position, ANITA below 0 altitude!");
}
else if(theta>0){
sprintf(label,"Pointing upwards! Cannot locate source at ground position");
}
else{
sprintf(label,"Unkown error, cannot position source at 0 altitude");
}
positionLabel = new TLatex();
positionLabel->DrawLatex(0.05,0.95,label);
return;
}
TMarker *eventPos = new TMarker(xEvent,yEvent,29);
eventPos->SetMarkerColor(kBlack);
eventPos->Draw("");
sprintf(label,"ANITA location: lat %f; long %f; alt %f, x %f, y %f",anitaLat,anitaLon,anitaAlt,xAnita,yAnita);
positionLabel = new TLatex();
positionLabel->DrawLatex(0.05,0.97,label);
sprintf(label,"Event location: lat %f; long %f, x %f, y %f",sourceLat,sourceLon,xEvent,yEvent);
positionLabel->DrawLatex(0.05,0.94,label);
}
void drawMeasurement(int i, double m[5], char label[2][100], int aux[5],
double vstep, TH2F* histo, TCanvas* canvas) {
double lowY = (i+1)*vstep;
double uppY = (i+2)*vstep;
//double lowX = histo->GetBinLowEdge(1);
//double uppX = histo->GetBinLowEdge(histo->GetNbinsX()) +
// histo->GetBinWidth(histo->GetNbinsX());
double lowX = 0.25;
double uppX = 2.80;
double widthX = uppX - lowX;
// y-range of the histogram is [0...1]
double startX = lowX + 0.04*widthX;
TPaveText* text = new TPaveText(startX, lowY,
startX, uppY, "BR");
text->SetTextAlign(12);
text->SetFillColor(aux[1]);
text->SetTextColor(aux[0]);
text->SetLineColor(1);
text->SetBorderSize(0);
TText* t0 = text->AddText(" ");
t0->SetTextSize(0.08);
t0->SetTextFont(aux[2]);
TText* t1 = text->AddText(label[0]);
t1->SetTextSize(0.08);
t1->SetTextFont(aux[2]);
TText* t2 = text->AddText(label[1]);
t2->SetTextSize(0.08);
t2->SetTextFont(aux[2]);
text->Draw();
double ypos = 0.5*(lowY+uppY);
double mean = m[0];
double nErr1 = m[1];
double pErr1 = m[2];
double nErr2 = sqrt(m[1]*m[1]+m[3]*m[3]);
double pErr2 = sqrt(m[2]*m[2]+m[4]*m[4]);
// draw TGraphAsymmErrors 1 (stat only) |---*---|
TMarker* measurement = new TMarker(mean, ypos, aux[4]);
measurement->SetMarkerColor(aux[0]);
measurement->SetMarkerStyle(aux[4]);
measurement->SetMarkerSize(1.5);
//measurement->SetMarkerSize(1.75);
measurement->Draw();
double vsizeErr1 = 0.09*vstep;
TLine* l1 = new TLine(mean, ypos, mean-nErr1, ypos);
l1->SetLineWidth(aux[3]);
l1->SetLineColor(aux[0]);
l1->Draw();
TLine* l2 = new TLine(mean, ypos, mean+pErr1, ypos);
l2->SetLineWidth(aux[3]);
l2->SetLineColor(aux[0]);
l2->Draw();
TLine* l3 = new TLine(mean-nErr1, ypos-vsizeErr1,
mean-nErr1, ypos+vsizeErr1);
l3->SetLineWidth(aux[3]);
l3->SetLineColor(aux[0]);
l3->Draw();
TLine* l4 = new TLine(mean+pErr1, ypos-vsizeErr1,
mean+pErr1, ypos+vsizeErr1);
l4->SetLineWidth(aux[3]);
l4->SetLineColor(aux[0]);
l4->Draw();
// overlay TGraphAsymmErrors 2 (stat+syst) |----*-----|
double vsizeErr2 = 0.12*vstep;
TLine* l5 = new TLine(mean, ypos, mean-nErr2, ypos);
l5->SetLineWidth(aux[3]);
l5->SetLineColor(aux[0]);
l5->Draw();
TLine* l6 = new TLine(mean, ypos, mean+pErr2, ypos);
l6->SetLineWidth(aux[3]);
l6->SetLineColor(aux[0]);
l6->Draw();
TLine* l7 = new TLine(mean-nErr2, ypos-vsizeErr2,
mean-nErr2, ypos+vsizeErr2);
l7->SetLineWidth(aux[3]);
l7->SetLineColor(aux[0]);
l7->Draw();
TLine* l8 = new TLine(mean+pErr2, ypos-vsizeErr2,
mean+pErr2, ypos+vsizeErr2);
l8->SetLineWidth(aux[3]);
l8->SetLineColor(aux[0]);
l8->Draw();
// draw measurement label "XXX+/-YY+/-ZZ"
TPaveText* num = new TPaveText(uppX-0.32*widthX,
lowY, uppX-0.02*widthX, uppY, "BR");
num->SetTextAlign(12);
num->SetFillColor(aux[1]);
num->SetTextColor(aux[0]);
num->SetLineColor(aux[0]);
num->SetBorderSize(0);
TString str;
char s[100];
sprintf(s, "%4.2f#color[%d]{X}", m[0], aux[1]); str +=s;
if (m[1]==m[2]) { // sym. stat. errors
sprintf(s, "#pm %4.2f", m[1]); str +=s;
} else {
sprintf(s, "^{+%4.2f}", m[2]); str +=s;
sprintf(s, "_{-#color[%d]{|}%4.2f}", aux[1], m[1]); str +=s;
}
str += " (stat)";
if (m[3]!=0.0 || m[4]!=0.0) {
if (m[3]==m[4]) { // sym. syst. errors
sprintf(s, "#color[%d]{X}#pm% 4.2f", aux[1], m[3]); str +=s;
} else {
sprintf(s, "#color[%d]{X}", aux[1]); str +=s;
sprintf(s, "^{+%4.2f}", m[4]); str +=s;
sprintf(s, "_{-#color[%d]{|}%4.2f}", aux[1], m[3]); str +=s;
}
}
str += " (syst)";
TText* n0 = num->AddText(str);
n0->SetTextFont(aux[2]);
num->Draw();
return;
}
void drawEventZR(double vrlb, double vzlb, double vrl0, double vzl0, double pmu1, double etamu1, double pmu2, double etamu2, double ppr, double etapr, double ppi, double etapi, int colors = 0)
{
Color_t colMu1(1), colMu2(1), colPr(1), colPi(1), colL0(1), colLb(1);
switch(colors)
{
case 0:
colLb=1; colL0=2; colMu1=2; colMu2=2; colPr=3; colPi=4; break;
case 1:
colLb=11; colL0=50; colMu1=50; colMu2=50; colPr=8; colPi=9; break;
}
cout << etamu1 << " " << etamu2 << " " << etapr << " " << etapi << endl;
const double thetamu1 = 2*TMath::ATan(TMath::Exp(-etamu1));
const double thetamu2 = 2*TMath::ATan(TMath::Exp(-etamu2));
const double thetapr = 2*TMath::ATan(TMath::Exp(-etapr));
const double thetapi = 2*TMath::ATan(TMath::Exp(-etapi));
//const double thetamu1 = 2*TMath::ATan(TMath::Exp(-TMath::Abs(etamu1)))*sign(etamu1);
//const double thetamu2 = 2*TMath::ATan(TMath::Exp(-TMath::Abs(etamu2)))*sign(etamu2);
//const double thetapr = 2*TMath::ATan(TMath::Exp(-TMath::Abs(etapr)))*sign(etapr);
//const double thetapi = 2*TMath::ATan(TMath::Exp(-TMath::Abs(etapi)))*sign(etapi);
// draw the vertices
TMarker *m;
const double xlb = vzlb;
const double ylb = vrlb;
m = new TMarker(xlb,ylb,7);
m->SetMarkerColor(colLb);
m->Draw();
const double xl0 = vzl0;
const double yl0 = vrl0;
m = new TMarker(xl0,yl0,7);
m->SetMarkerColor(colL0);
m->Draw();
// draw the l0 flight line
TLine *l;
l = new TLine(vzlb,vrlb,vzl0,vrl0);
l->SetLineColor(colL0);
l->SetLineStyle(2);
l->Draw();
// draw the muons
TArrow *a;
const double xmu1 = xlb + scalemu * pmu1 * TMath::Cos(thetamu1);
const double ymu1 = ylb + scalemu * pmu1 * TMath::Sin(thetamu1);
a = new TArrow(xlb,ylb,xmu1,ymu1,.01,">");
a->SetLineColor(colMu1);
a->Draw();
const double xmu2 = xlb + scalemu * pmu2 * TMath::Cos(thetamu2);
const double ymu2 = ylb + scalemu * pmu2 * TMath::Sin(thetamu2);
a = new TArrow(xlb,ylb,xmu2,ymu2,.01,">");
a->SetLineColor(colMu2);
a->Draw();
// draw the p and pi
const double xpr = xl0 + scalepr * ppr * TMath::Cos(thetapr);
const double ypr = yl0 + scalepr * ppr * TMath::Sin(thetapr);
a = new TArrow(xl0,yl0,xpr,ypr,.01,">");
a->SetLineColor(colPr);
a->Draw();
const double xpi = xl0 + scalepi * ppi * TMath::Cos(thetapi);
const double ypi = yl0 + scalepi * ppi * TMath::Sin(thetapi);
a = new TArrow(xl0,yl0,xpi,ypi,.01,">");
a->SetLineColor(colPi);
a->Draw();
}
void showHistogram2d(TH2* histogram,
const std::string& xAxisTitle, const std::string& yAxisTitle,
int zAxisNormOption, double zMin, double zMax,
Float_t* genX, Float_t* genY,
const std::string& outputFileName)
{
TCanvas* canvas = new TCanvas("canvas", "canvas", 900, 800);
canvas->SetFillColor(10);
canvas->SetBorderSize(2);
canvas->SetTopMargin(0.10);
canvas->SetLeftMargin(0.12);
canvas->SetRightMargin(0.14);
canvas->SetBottomMargin(0.12);
histogram->SetTitle("");
histogram->SetStats(false);
int numBinsX = histogram->GetNbinsX();
int numBinsY = histogram->GetNbinsY();
if ( zAxisNormOption == kNormByQuantiles ) {
std::vector<double> binContents;
for ( int iBinX = 1; iBinX <= numBinsX; ++iBinX ) {
for ( int iBinY = 1; iBinY <= numBinsY; ++iBinY ) {
binContents.push_back(histogram->GetBinContent(iBinX, iBinY));
}
}
std::sort(binContents.begin(), binContents.end());
histogram->SetMinimum(binContents[TMath::Nint(0.05*binContents.size())]);
histogram->SetMaximum(binContents[TMath::Nint(0.95*binContents.size())]);
} else if ( zAxisNormOption == kNormByNegLogMax ) {
double maxBinContent = 0.;
for ( int iBinX = 1; iBinX <= numBinsX; ++iBinX ) {
for ( int iBinY = 1; iBinY <= numBinsY; ++iBinY ) {
double binContent = histogram->GetBinContent(iBinX, iBinY);
if ( binContent > maxBinContent ) {
std::cout << "binX = " << iBinX << " (x = " << histogram->GetXaxis()->GetBinCenter(iBinX) << "),"
<< " binY = " << iBinY << " (y = " << histogram->GetYaxis()->GetBinCenter(iBinY) << "): maxBinContent = " << maxBinContent << std::endl;
maxBinContent = binContent;
}
}
}
double logMaxBinContent = TMath::Log(maxBinContent);
for ( int iBinX = 1; iBinX <= numBinsX; ++iBinX ) {
for ( int iBinY = 1; iBinY <= numBinsY; ++iBinY ) {
double binContent = histogram->GetBinContent(iBinX, iBinY);
if ( binContent > 0. ) {
histogram->SetBinContent(iBinX, iBinY, -TMath::Log(binContent) + logMaxBinContent);
} else {
histogram->SetBinContent(iBinX, iBinY, -1.);
}
}
}
histogram->SetMinimum(0.);
histogram->SetMaximum(zMax);
} else if ( zAxisNormOption == kNormByValue ) {
histogram->SetMinimum(zMin);
histogram->SetMaximum(zMax);
} else assert(0);
TAxis* xAxis = histogram->GetXaxis();
xAxis->SetTitle(xAxisTitle.data());
xAxis->SetTitleOffset(1.15);
TAxis* yAxis = histogram->GetYaxis();
yAxis->SetTitle(yAxisTitle.data());
yAxis->SetTitleOffset(1.30);
gStyle->SetPalette(1,0);
histogram->Draw("COLZ");
TMarker* genMarker = 0;
if ( genX && genY ) {
genMarker = new TMarker(*genX, *genY, 34);
genMarker->SetMarkerColor(1);
genMarker->SetMarkerSize(2);
genMarker->Draw();
}
canvas->Update();
size_t idx = outputFileName.find_last_of('.');
std::string outputFileName_plot = std::string(outputFileName, 0, idx);
if ( idx != std::string::npos ) canvas->Print(std::string(outputFileName_plot).append(std::string(outputFileName, idx)).data());
canvas->Print(std::string(outputFileName_plot).append(".png").data());
canvas->Print(std::string(outputFileName_plot).append(".pdf").data());
canvas->Print(std::string(outputFileName_plot).append(".root").data());
delete genMarker;
delete canvas;
}
// draw the same thing but after reco
void genPlots02(std::string fullPath, int nOverlay = 500, bool custBinning = false)
{
const int fVerbose(1);
setTDRStyle();
gStyle->SetOptStat(112211);
gStyle->SetPalette(1);
// Canvas
c = new TCanvas("c2","c2",1000,600);
const unsigned int nPadX = 1;
const unsigned int nPadY = 1;
c->Divide(nPadX,nPadY);
const unsigned int nPads=nPadX*nPadY;
for(unsigned int i=1; i<=nPads; i++)
{
TPad* pad= (TPad*)c->cd(i);
pad->SetTopMargin(0.10);
pad->SetRightMargin(0.20);
pad->SetLeftMargin(0.15);
}
// Open file
TFile *f = TFile::Open(fullPath.c_str());
if (f==0)
{
cout << "File " << fullPath << " not found -- exiting" << endl;
return;
}
if(fVerbose>0)
cout << "Succesfully opened file " << fullPath << endl;
// Get TTree
TTree* t = (TTree*) f->Get("events");
if(fVerbose>0) cout << "Got TTree with " << t->GetEntries() << " entries" << endl;
// Do a cut, if needed
//t->Draw(">>lst","chi2lb>.1&&mlb>5.61&&mlb<5.63");
//t->Draw(">>lst","chi2lb>.1&&isSig==1");
t->Draw(">>lst","(rid1m&4)==4&&(rid2m&4)==4&&mjp>2.895&&mjp<3.295&&prob1m>0.1&&prob2m>0.1&&ptjp>2&&probjp>0.005&&ml0>1.101&&ml0<1.129&&probpr>0.02&&probpi>0.02&&rptpr>rptpi&&ptl0>3&&rptpr>1&&rptpi>0.5&&probl0>0.02&&alphal0<0.3&&d3l0>1&&d3l0/d3El0>10&&problb>0.001&&alphalb<0.3");
TEventList *lst;
lst = (TEventList*)gDirectory->Get("lst");
t->SetEventList(lst);
if(fVerbose>0) cout << "Got TTree with " << t->GetEntries() << " entries" << endl;
// Do plots
c->cd(1);
//doPlot2d(t,"hrzL0vtx", "vrl0:TMath::Abs(vzl0)",30,0,300,30,0,120,"Tit","|z|","r","cm","cm");
if (custBinning)
{
double newbinsX[]={0,2,4,6,8,10,12,14,16,18,20,22,24,26,28,30,32,34,36,38,40,42,44,46,48,50};
const int newbinsX_size = sizeof(newbinsX)/sizeof(double);
std::vector<double> binvecX(newbinsX,newbinsX+newbinsX_size);
//double newbinsY[]={0,1,2,3,4,5,6,7,8,9,10};
double newbinsY[]={0,0.5,1,2,4,8,16,32};
const int newbinsY_size = sizeof(newbinsY)/sizeof(double);
std::vector<double> binvecY(newbinsY,newbinsY+newbinsY_size);
doPlot2d(t,"hrzL0vtxreco", "vrl0:TMath::Abs(vzl0)",binvecX, binvecY,"#Lambda vertices","|z|","r","cm","cm");
}
else
{
doPlot2d(t,"hrzL0vtxreco", "vrl0:TMath::Abs(vzl0)",30,0,50,30,0,30,"#Lambda vertices","|z|","r","cm","cm");
}
// add tracker
TPad* pad;
pad = (TPad*)c->cd(1);
pad->Modified();
pad->Update();
repositionPalette("hrzL0vtxreco");
pad->Update();
pad->SetLogz();
drawTracker(pad);
if (nOverlay<=0) return;
int maxN = nOverlay;
if (maxN > t->GetEntries()) maxN = t->GetEntries();
double vrl0,vzl0,ppr,ppi,etapr,etapi;
t->SetBranchAddress("vrl0",&vrl0);
t->SetBranchAddress("vzl0",&vzl0);
t->SetBranchAddress("ppr",&ppr);
t->SetBranchAddress("etapr",&etapr);
t->SetBranchAddress("ppi",&ppi);
t->SetBranchAddress("etapi",&etapi);
double scalepr = 4;
double scalepi = 8;
{ // reference indicator
const double x1pr = 0; const double y1pr = -3;
const double x2pr = scalepr; const double y2pr = y1pr;
const double versatz = 14;
const double x1pi = x1pr+versatz; const double y1pi = -3;
const double x2pi = x2pr+versatz+scalepi; const double y2pi = y1pi;
TArrow *a;
a = new TArrow(x1pr,y1pr,x2pr,y2pr,.01,">");
a->SetLineColor(24);
a->Draw();
TLatex tl;
tl.SetTextSize(20);
tl.SetTextFont(4);
tl.DrawLatex(x1pr,y2pr-1.2,"p(p) / 1 GeV");
a = new TArrow(x1pi,y1pi,x2pi,y2pi,.01,">");
a->SetLineColor(20);
a->Draw();
tl.SetTextSize(20);
tl.SetTextFont(4);
tl.DrawLatex(x1pi,y2pi-1.2,"p(#pi) / 1 GeV");
}
for (int i = 0; i!=maxN; i++)
{
t->GetEntry(i);
const double thetapr = 2*TMath::ATan(TMath::Exp(-TMath::Abs(etapr)));
const double thetapi = 2*TMath::ATan(TMath::Exp(-TMath::Abs(etapi)));
const double x1=TMath::Abs(vzl0);
const double y1=vrl0;
const double x2pr=x1+scalepr*ppr*TMath::Cos(thetapr);
const double y2pr=y1+scalepr*ppr*TMath::Sin(thetapr);
const double x2pi=x1+scalepi*ppi*TMath::Cos(thetapi);
const double y2pi=y1+scalepi*ppi*TMath::Sin(thetapi);
TArrow *a;
a = new TArrow(x1,y1,x2pr,y2pr,.01,">");
a->SetLineColor(24);
a->Draw();
a = new TArrow(x1,y1,x2pi,y2pi,.01,">");
a->SetLineColor(20);
a->Draw();
TMarker *m = new TMarker(x1,y1,7);
m->SetMarkerColor(28);
m->Draw();
}
}
void plotFit(TString filename = "fp-d", TString pltmd = "tph") {
// CHECK FOR RIGHT INPUT ////////////////////////////////////////////////
string strpltmd = pltmd;
if( strpltmd.compare("tph") != 0 &&
strpltmd.compare("s2b") != 0 &&
strpltmd.compare("mmp") != 0 ) {error(4);};
// OPEN THE ROOT FILE //////////////////////////////////////////////////
gROOT->Reset();
gROOT->SetStyle("Plain");
gStyle->SetTitleBorderSize(0);
gStyle->SetPalette(1);
TCanvas *MyC = new TCanvas("MyC","Plot of the GAPP fit",200,10,700,500);
// Still to do: Automate the frame boundaries.
string strfile = filename, rootname = strfile + ".root";
TFile *rootfile = TFile::Open(rootname.c_str());
if(rootfile == NULL) error(1);
TTree *tree = (TTree*)rootfile->Get(strfile.c_str());
if(tree == NULL) error(2);
TBranch *fits2bbranch = (TBranch*)tree->GetBranch("fits2b");
TBranch *fittphbranch = (TBranch*)tree->GetBranch("fittph");
TBranch *fitxbranch = (TBranch*)tree->GetBranch("fitx");
if( (fits2bbranch == NULL) || (fittphbranch == NULL) || (fitxbranch == NULL) ) error(3);
Float_t fits2b, fittph, fitx;
tree->SetBranchAddress("fits2b",&fits2b);
tree->SetBranchAddress("fittph",&fittph);
tree->SetBranchAddress("fitx", &fitx);
// GET GRID /////////////////////////////////////////////////////////////
Int_t Npoints = (Int_t)tree->GetEntries();
Int_t point, zSteps = 0, ySteps = 0;
Float_t fitxMin = 100000, fitxMax = -1.0, s2bMin = 100000, tphMin = 100000;
Float_t s2bMax = -1.0, tphMax = -1.0;
for(point=0; point<Npoints; point++) {
tree->GetEntry(point);
if( fits2b > s2bMax ) {zSteps++; s2bMax = fits2b;}
if( fittph > tphMax ) {ySteps++; tphMax = fittph;}
};
const int s2bSteps = zSteps, tphSteps = ySteps, mmpSteps = Npoints;
Float_t s2bValues[s2bSteps], tphValues[tphSteps];
s2bMax = -1.0, tphMax = -1.0;
int s2bStep = 0, tphStep = 0;
for(point=0; point<Npoints; point++) {
tree->GetEntry(point);
if( fits2b > s2bMax ) {s2bValues[s2bStep] = fits2b; s2bStep++; s2bMax = fits2b;}
if( fittph > tphMax ) {tphValues[tphStep] = fittph; tphStep++; tphMax = fittph;}
};
// PREPARE PLOT /////////////////////////////////////////////////////////
string plottitle = "Model: " + strfile + " | Plot: ";
if( strpltmd.compare("tph") == 0 ) {
plottitle += "tan^{2}(#phi) over x for fixed sin^{2}(2#beta)";
}
if( strpltmd.compare("s2b") == 0 ) {
plottitle += "sin^{2}(2#beta) over x for fixed tan^{2}(#phi)";
}
if( strpltmd.compare("mmp") == 0 ) {
plottitle += "Masses of the new heavy gauge bosons";
}
// PLOT DATA ///////////////////////////////////////////////////////////
if (strpltmd.compare("tph") == 0) {
tphMax = -1.0;
// TGraph *tphplots[s2bSteps];
TGraph *tphplots[s2bSteps-30];
// for(s2bStep=0; s2bStep<s2bSteps; s2bStep++) {
for(s2bStep=0; s2bStep<s2bSteps-30; s2bStep++) {
Float_t tphArray[tphSteps], fitxArray[tphSteps];
tphStep = 0;
for(point=0; point<Npoints; point++) {
tree->GetEntry(point);
if(fits2b == s2bValues[s2bStep]) {
// tphArray[tphStep] = fittph;
tphArray[tphStep] = sqrt(1.0/(1.0+fittph));
if (fittph < tphMin) tphMin = fittph;
if (fittph > tphMax) tphMax = fittph;
fitxArray[tphStep] = fitx;
if (fitx < fitxMin) fitxMin = fitx;
if (fitx > fitxMax) fitxMax = fitx;
tphStep++;
TMarker *m = new TMarker(fitxArray[tphStep],tphArray[tphStep],20);
m->SetMarkerSize(2);
m->SetMarkerColor(31+tphStep);
m->Draw();
}
}
if (s2bStep == 0) {
TH1F* frame = MyC->DrawFrame(0.0,0.0,1.1*fitxMax,1.1);
// TH1F* frame = MyC->DrawFrame(0.7*fitxMin,0.7*tphMin,1.1*fitxMax,1.1*tphMax);
TAxis *xaxis = frame->GetXaxis();
TAxis *yaxis = frame->GetYaxis();
xaxis->SetTitle("x = u^{2}/v^{2}");
xaxis->CenterTitle();
xaxis->SetTitleOffset(1.);
xaxis->SetDecimals();
xaxis->SetLabelSize(0.03);
xaxis->SetLabelOffset(0.01);
yaxis->SetTitle("tan^{2}(#phi)");
yaxis->CenterTitle();
yaxis->SetTitleOffset(1.);
yaxis->SetDecimals();
yaxis->SetLabelSize(0.03);
yaxis->SetLabelOffset(0.01);
frame->SetTitle(plottitle.c_str());
}
tphplots[s2bStep] = new TGraph(tphSteps,fitxArray,tphArray);
tphplots[s2bStep]->SetMarkerStyle(20);
tphplots[s2bStep]->SetMarkerSize(0.4);
tphplots[s2bStep]->Draw("CP");
}
}
else if (strpltmd.compare("s2b") == 0) {
s2bMax = -1.0;
TGraph *s2bplots[tphSteps-100];
for(tphStep=0; tphStep<tphSteps-100; tphStep++) {
Float_t s2bArray[s2bSteps], fitxArray[s2bSteps];
s2bStep = 0;
for(point=0; point<Npoints; point++) {
tree->GetEntry(point);
if(fittph == tphValues[tphStep+20]) {
s2bArray[s2bStep] = fits2b;
if (fits2b < s2bMin) s2bMin = fits2b;
if (fits2b > s2bMax) s2bMax = fits2b;
fitxArray[s2bStep] = fitx;
if (fitx < fitxMin) fitxMin = fitx;
if (fitx > fitxMax) fitxMax = fitx;
s2bStep++;
}
}
if (tphStep == 0) {
TH1F* frame = MyC->DrawFrame(0.8*fitxMin,0.95*s2bMin,1.2*fitxMax,1.05*s2bMax);
TAxis *xaxis = frame->GetXaxis();
TAxis *yaxis = frame->GetYaxis();
xaxis->SetTitle("x = u^{2}/v^{2}");
xaxis->CenterTitle();
xaxis->SetTitleOffset(1.);
xaxis->SetDecimals();
xaxis->SetLabelSize(0.03);
xaxis->SetLabelOffset(0.01);
yaxis->SetTitle("sin^{2}(2#beta)");
yaxis->CenterTitle();
yaxis->SetTitleOffset(1.25);
yaxis->SetDecimals();
yaxis->SetLabelSize(0.03);
yaxis->SetLabelOffset(0.01);
frame->SetTitle(plottitle.c_str());
}
s2bplots[tphStep] = new TGraph(s2bSteps,fitxArray,s2bArray);
s2bplots[tphStep]->SetMarkerStyle(20);
s2bplots[tphStep]->SetMarkerSize(0.4);
s2bplots[tphStep]->Draw("C");
}
}
else if (strpltmd.compare("mmp") == 0) {
Float_t mzpArray[mmpSteps], mwpArray[mmpSteps];
Float_t mzpMin = 100000, mzpMax = -1.0, mwpMin = 100000, mwpMax = -1.0;
Float_t fitsph, fitcph;
Float_t Cz1, Cz2, Cz3, Cw1, Cw2, Cw3, Cw4, C1, C2;
string mdl(strfile,0,2);
if ( (mdl.compare("lr") == 0) ||
(mdl.compare("lp") == 0) ||
(mdl.compare("hp") == 0) ||
(mdl.compare("fp") == 0) ) {
string Higgs(strfile,3,1);
if (Higgs.compare("d") == 0) {
Cz1 = 11.95349795785275;
Cz2 = 30.63269990028513;
Cz3 = 42.58619785813789;
Cw1 = 21.29309892906894;
Cw2 = 9.339600971216193;
Cw3 = 30.63269990028513;
Cw4 = 42.58619785813789;
}
else if (Higgs.compare("t") == 0) {
Cz1 = 5.976748978926375;
Cz2 = 30.63269990028513;
Cz3 = 85.17239571627579;
Cw1 = 15.05649464522066;
Cw2 = 3.302047590161717;
Cw3 = 21.66058982554409;
Cw4 = 60.22597858088265;
} else {error(6);}
for(point=0; point<Npoints; point++) {
tree->GetEntry(point);
fitsph = fittph / (1.0 + fittph);
fitcph = 1.0 - fitsph;
if (fitsph != 0.0) {
mzpArray[point] = (0.001/sqrt(fitsph*fitcph*fitx)) * (Cz1*fitcph*fitcph + Cz2*fits2b + Cz3*fitx);
if (mzpArray[point] < mzpMin) mzpMin = mzpArray[point];
if (mzpArray[point] > mzpMax) mzpMax = mzpArray[point];
mwpArray[point] = (0.001/sqrt(fitsph*fitx)) * (Cw1 - Cw2*fitcph*fitcph + Cw3*fits2b + Cw4*fitx);
if (mwpArray[point] < mwpMin) mwpMin = mwpArray[point];
if (mwpArray[point] > mwpMax) mwpMax = mwpArray[point];
} else {
mzpArray[point] = 0.0;
mwpArray[point] = 0.0;
}
}
}
else if ( (mdl.compare("uu") == 0) ||
(mdl.compare("nu") == 0) ) {
C1 = 94.0397928463607
C2 = 77.1253849720165
for(point=0; point<Npoints; point++) {
tree->GetEntry(point);
fitsph = fittph / (1.0+fittph);
fitcph = 1.0 - fitsph;
if (fitsph != 0.0) {
mzpArray[point] = (0.001/sqrt(fitsph*fitcph*fitx)) * (C1*fitsph*fitsph + C2*fitx);
if (mzpArray[point] < mzpMin) mzpMin = mzpArray[point];
if (mzpArray[point] > mzpMax) mzpMax = mzpArray[point];
mwpArray[point] = (0.001/sqrt(fitsph*fitcph*fitx)) * (C1*fitsph*fitsph + C2*fitx);
if (mwpArray[point] < mwpMin) mwpMin = mwpArray[point];
if (mwpArray[point] > mwpMax) mwpMax = mwpArray[point];
} else {
mzpArray[point] = 0.0;
mwpArray[point] = 0.0;
}
}
}
void EventDisplayForBaby(bool sig=false, bool truth=true, bool fatjet=true, int event=-1)
{
gInterpreter->ExecuteMacro("~/macros/rootlogon.C");
// chain
TChain *ch = new TChain("tree");
if(!sig) ch->Add("babies/small_quick_cfA_746p1_nleps1_trig0ON.root");
if(sig) ch->Add("~/scratch/plots/1d_2015d_13Jan2016/2d/48ipb/*.root");
InitBaby(ch);
Int_t nentries = (Int_t)ch->GetEntries();
for(int i = 0; i<nentries; i++)
{
ch->GetEntry(i);
// apply selections (event==-1 && mj_>xxx && ht_>yyy && ...)
// or
// select an event (event!=-1 && event_=xxxxxx)
// "event" is one of the arguments
if(event>0 && event_!=event) continue;
//
// Event Display
//
float start=0.62;
float nextline = start;
float increment = 0.04;
float offset=0.02;
int npert;
int LSPcol[2] = {kCyan+1,kOrange-4};
if(!sig) npert = 5;
else npert=3;
float xalign = 0.82;
vector<TMarker> constituents, genpart;
TEllipse *cone[fjets_pt_->size()];
TH2F *h_fatjets = new TH2F("h_fatjets","h_fatjets", 230, -5.0, 5.0, 144, -3.141592, 3.141592);
cout << event_ << endl;
for(int ifj = 0; ifj< (int)fjets_pt_->size(); ifj++)
{ if(ifj==0) cout << "... Fatjets info(pT, eta, phi, mj)" << endl;
h_fatjets->Fill(fjets_eta_->at(ifj),fjets_phi_->at(ifj), fjets_m_->at(ifj));
cout << fjets_pt_->at(ifj) << " " << fjets_eta_->at(ifj) << " "
<< fjets_phi_->at(ifj) << " " << fjets_m_->at(ifj) << endl;
}
TString cname = Form("Event_%lli_",event_);
TCanvas *c = new TCanvas(cname,cname,1640,760);
gPad->SetRightMargin(0.35);
h_fatjets->SetZTitle("m_{j} [GeV]");
h_fatjets->GetZaxis()->SetTitleSize(0.05);
h_fatjets->GetZaxis()->SetTitleOffset(0.55);
h_fatjets->GetYaxis()->SetTitleOffset(0.6);
h_fatjets->Draw("colz");
h_fatjets->GetZaxis()->SetRangeUser(0,1.1*h_fatjets->GetMaximum());
h_fatjets->Draw("colz");
gPad->Update();
TPaletteAxis *palette = (TPaletteAxis*)h_fatjets->GetListOfFunctions()->FindObject("palette");
for(int ifj = 0; ifj< (int)fjets_pt_->size(); ifj++)
{
cone[ifj] = new TEllipse(fjets_eta_->at(ifj),fjets_phi_->at(ifj), 1.2, 1.2);
// cone[ifj]->SetFillStyle(3003);
cone[ifj]->SetFillStyle(0);
Int_t binx,biny,binz;
h_fatjets->GetBinXYZ(h_fatjets->FindBin(fjets_eta_->at(ifj),fjets_phi_->at(ifj)),binx,biny,binz);
//cout<<"bin x bin y "<<binx<<" "<<biny<<endl;
//cout<<"content "<<h_fatjets->GetBinContent(binx,biny)<<endl;
//cout<<palette<<endl;
Int_t ci = palette->GetBinColor(binx,biny);
if(ifj==0) ci = palette->GetBinColor(binx,biny);
//cout<<"color"<<ci<<endl;
cone[ifj]->SetFillColor(kGray);
cone[ifj]->SetLineColor(ci);
}
float lepPt, lepEta, lepPhi;
if(nels_==1)
{
for(int i=0; i<(int)els_pt_->size(); i++)
{
if(els_miniso_->at(i)>0.1) continue;
if(els_sigid_->at(i)!=1) continue;
if(els_pt_->at(i)<20) continue;
lepPt =els_pt_->at(i);
lepEta =els_eta_->at(i);
lepPhi =els_phi_->at(i);
}
}
if(nmus_==1)
{
for(int i=0; i<(int)mus_pt_->size(); i++)
{
if(mus_miniso_->at(i)>0.2) continue;
if(mus_sigid_->at(i)!=1) continue;
if(mus_pt_->at(i)<20) continue;
lepPt =mus_pt_->at(i);
lepEta =mus_eta_->at(i);
lepPhi =mus_phi_->at(i);
}
}
myText(xalign+0.01,0.96,Form("H_{T} = %.0f GeV",ht_),1,0.04);
myText(xalign+0.01,0.91,Form("M_{J} = %.0f GeV",mj_),1,0.04);
myText(xalign+0.01,0.86,Form("#slash{E}_{T} = %.0f GeV",met_),1,0.04);
myText(xalign+0.01,0.81,Form("m_{T} = %.0f GeV",mt_),1,0.04);
myText(xalign+0.01,0.76,Form("reco %s p_{T} = %.0f GeV", nmus_==1?"#mu":"e", lepPt), kBlack, 0.04);
nextline=0.76;
TMarker recolep = TMarker(lepEta, lepPhi, 27);
recolep.SetMarkerSize(4);
recolep.SetMarkerColor(kRed);
genpart.push_back(recolep);
TMarker mumark = TMarker(xalign,nextline,27);
mumark.SetNDC();
mumark.SetX(xalign);
mumark.SetY(nextline+0.01);
mumark.SetMarkerSize(2);
mumark.SetMarkerColor(kRed);
genpart.push_back(mumark);
myText(xalign+0.01,0.71,Form("-- large-R jets --"),1,0.04);
myText(xalign+0.01,0.67,Form("(pT, eta, phi, mass)"),1,0.03);
nextline=0.635;
for(int ifj = 0; ifj< (int)fjets_pt_->size(); ifj++)
{
myText(xalign+0.01,nextline,Form("%3.0f, %2.1f, %2.1f, %3.0f",
fjets_pt_->at(ifj),fjets_eta_->at(ifj),fjets_phi_->at(ifj),fjets_m_->at(ifj)),1,0.03);
nextline=nextline-0.035;
}
nextline=nextline-0.015;
myText(xalign+0.01,nextline,Form("-- AK4 jets --"),1,0.04);
nextline=nextline-0.04;
myText(xalign+0.01,nextline,Form("(pT, eta, phi)"),1,0.03);
nextline=nextline-0.035;
for(int ij = 0; ij< (int)jets_pt_->size(); ij++)
{
if(jets_islep_->at(ij)==1) continue;
myText(xalign+0.01,nextline,Form("%3.0f, %2.1f, %2.1f",
jets_pt_->at(ij),jets_eta_->at(ij),jets_phi_->at(ij)),1,0.03);
nextline=nextline-0.035;
}
if(truth) myText(xalign,0.68,"Gen p_{T} [GeV]",1,0.04);
if(truth) myText(xalign,0.68,"Gen p_{T} [GeV]",1,0.04);
TArrow line1 = TArrow();
if(truth)line1.DrawLineNDC(xalign-0.02,start+0.035,xalign+0.14,start+0.035);
TMarker met = TMarker(0, met_phi_, 27);
met.SetMarkerSize(4);
met.SetMarkerColor(kMagenta);
genpart.push_back(met);
TMarker metmark = TMarker(0.81,0.86,27);
metmark.SetNDC();
metmark.SetX(xalign);
metmark.SetY(0.87);
metmark.SetMarkerSize(2);
metmark.SetMarkerColor(kMagenta);
genpart.push_back(metmark);
TMarker jetmark = TMarker(0.5,0.5,20);
jetmark.SetNDC();
jetmark.SetMarkerSize(1.2);
jetmark.SetX(0.12);
jetmark.SetY(0.2);
genpart.push_back(jetmark);
myText(0.13,0.19,"AK4 Jets",kBlack,0.04);
TMarker bjetmark = TMarker(0.5,0.5,20);
bjetmark.SetNDC();
bjetmark.SetMarkerSize(1.2);
bjetmark.SetMarkerColor(8);
bjetmark.SetX(0.12);
bjetmark.SetY(0.16);
genpart.push_back(bjetmark);
myText(0.13,0.15,"CSVM AK4",kBlack,0.04);
myText(0.08,0.94,"Ring color indicates FJ mass",kBlack,0.04);
// if(sig) nextline = start - (4*npert+2)*increment-0.02-offset;
//else nextline = start - 2*npert*increment-offset;
for(int ij=0;ij<(int)jets_pt_->size();ij++)
{
if(jets_islep_->at(ij)==1) continue;
if(ij==0) cout << "... skinny jet info (pT eta phi)" << endl;
cout << jets_pt_->at(ij) << " " << jets_eta_->at(ij) << " " << jets_phi_->at(ij) << endl;
TMarker jet = TMarker(jets_eta_->at(ij),jets_phi_->at(ij),20);
jet.SetMarkerSize(1.2);
if(jets_csv_->at(ij)>0.890) jet.SetMarkerColor(8);
constituents.push_back(jet);
}
/*
bool drawn=false;
for(int ifj = 0; ifj< (int)fjets.size(); ifj++){
vector<fastjet::PseudoJet> cons = fjets[ifj].constituents();
for(int ics = 0; ics<(int)cons.size();ics++){
for(int ifj2=0; ifj2<(int)fjets.size(); ifj2++){
if(ifj2==ifj) continue;
TArrow first = TArrow(cons[ics].eta(),cons[ics].phi_std(), fjets[ifj].eta(),fjets[ifj].phi_std(),0.04,">");
if(deltaR(cons[ics].eta(),cons[ics].phi_std(), fjets[ifj2].eta(),fjets[ifj2].phi_std()) < 1.25){
//cout<<"type 1 line: con eta phi FJ eta phi"<<cons[ics].eta()<<" "<<cons[ics].phi_std()<<" "<<fjets[ifj2].eta()<<" "<<fjets[ifj2].phi_std()<<endl;
first.DrawArrow(cons[ics].eta(),cons[ics].phi_std(), fjets[ifj].eta(),fjets[ifj].phi_std(),0.0045,"|>");
drawn=true;
break;
}
else if(deltaR(cons[ics].eta(),cons[ics].phi_std(), fjets[ifj].eta(),fjets[ifj].phi_std()) > 1.15){
//cout<<"type 2 line: con eta phi FJ eta phi"<<cons[ics].eta()<<" "<<cons[ics].phi_std()<<" "<<fjets[ifj].eta()<<" "<<fjets[ifj].phi_std()<<endl;
first.DrawArrow(cons[ics].eta(),cons[ics].phi_std(), fjets[ifj].eta(),fjets[ifj].phi_std(),0.0045,"|>");
drawn=true;
break;}
}
}
cons.clear();
}
if(drawn) myText(0.12,0.04,"Arrows indicate ambiguous clustering ownership",kBlack,0.03);
*/
if(truth)
{
int col[4] = {30,38,44,46};
if(!sig) col[1]=46;
int LSP,top,b,W,Wda,gl;
LSP=0;top=0;b=0;W=0;Wda=0;gl=0;
TLorentzVector vgl[2];
TLorentzVector vt[4];
TLorentzVector vLSP[2];
for(unsigned int imc = 0; imc < mc_id_->size(); imc++)
{
int id= (int)mc_id_->at(imc);
int absid = abs(id);
int mid = (int)mc_mom_->at(imc);
int absmid= abs(mid);
if(id!=mid)
{
int marknum=0;
TString partname;
int color=0;
bool flag=false;
if(id==1000021){
if(gl<2) vgl[gl].SetPtEtaPhiM(mc_pt_->at(imc), mc_eta_->at(imc), mc_phi_->at(imc), 1500);
gl++;
}
if(id==1000022)
{
color = LSPcol[LSP];
partname = "#tilde{#chi}^{0}_{1}";
if(LSP==0) nextline=start-2*npert*increment;
else nextline = start - (4*npert+1)*increment-offset;
LSP++;
flag=true;
marknum=25;//27;
}
if(absid == 6)
{
//top
marknum=22;
partname = "t";
flag =true;
nextline = start - npert*top*increment;
color = col[top];
if(!sig) {}
if(sig) {if(top>1){ nextline-=(increment+offset);}}
if(top<4)vt[top].SetPtEtaPhiM(mc_pt_->at(imc), mc_eta_->at(imc), mc_phi_->at(imc), 172.5);
top++;
}
if(absid==5 && (mid==6 || mid==(-6)))
{
//b from top
marknum=23;
partname="b";
flag=true;
color = col[b];
nextline = start - npert*b*increment - 2*increment;
// if(!sig){}
if(sig) {if(b>1){ nextline-=(increment+offset);}}
b++;
}
if(absid==24 && (mid==6 || mid==(-6)))
{
//W
marknum=34;
partname = "W";
flag=true;
color = col[W];
nextline = start - npert*W*increment - increment;
// if(!sig){}
if(sig){if(W>1){ nextline-=(increment+offset);}}
W++;
}
if((absid==13 || absid== 11 || absid==15) && absmid==24)
{
marknum=29;
if(absid==13) partname = "#mu";
else if(absid==11) partname = "e";
else partname = "#tau";
if(!sig) flag =true;
color = col[TMath::FloorNint(Wda/2)];
nextline = start - npert*increment*TMath::FloorNint(Wda/2)-3*increment;
if(TMath::FloorNint(Wda/2) != Wda/2) nextline-=increment;
Wda++;
}
if((absid==12 || absid== 14 || absid==16) && absmid==24)
{
marknum=30;
partname = "#nu";
if(!sig) flag =true;
color = col[TMath::FloorNint(Wda/2)];
nextline = start - npert*increment*TMath::FloorNint(Wda/2)-4*increment;
//if(TMath::FloorNint(Wda/2) != Wda/2) nextline-=increment;
Wda++;
}
if((absid>0 && absid<5) && absmid==24)
{
marknum=3;
if(absid==1) partname="d";
if(absid==2) partname="u";
if(absid==3) partname="s";
if(absid==4) partname="c";
if(!sig) flag =true;
color = col[TMath::FloorNint(Wda/2)];
nextline = start - npert*increment*TMath::FloorNint(Wda/2)-3*increment;
// if(TMath::FloorNint(Wda/2) != Wda/2) nextline-=increment;
if(absid % 2 ==1) nextline-=increment;
Wda++;
}
if(flag)
{
TMarker temp = TMarker(mc_eta_->at(imc),mc_phi_->at(imc), marknum);
TMarker legmark = TMarker(0.81,nextline,marknum);
legmark.SetNDC();
legmark.SetX(xalign+0.01);
legmark.SetY(nextline+0.011);
TString label = partname + " " + Form("%.0f",mc_pt_->at(imc));
myText(xalign+0.03,nextline,partname,color,0.04);
myText(xalign+0.07,nextline,Form("%.0f",mc_pt_->at(imc)),color,0.04);
temp.SetMarkerColor(color);
legmark.SetMarkerColor(color);
temp.SetMarkerSize(2);
legmark.SetMarkerSize(2);
if(absid==6 || absid==13 || absid==1000022){ temp.SetMarkerSize(3);}
genpart.push_back(temp);
genpart.push_back(legmark);
flag=false;
}
}
}
if(sig && LSP==0){
int color = LSPcol[LSP];
TString partname = "#tilde{#chi}^{0}_{1}";
if(LSP==0) nextline=start-2*npert*increment;
else nextline = start - (4*npert+1)*increment-offset;
LSP++;
int marknum=25;//27;
// TMarker temp = TMarker(mc_doc_eta->at(imc),mc_doc_phi->at(imc), marknum);
vLSP[0]= vgl[0]-vt[0]-vt[1];
TMarker temp = TMarker(vLSP[0].Eta(),vLSP[0].Phi(), marknum);
TMarker legmark = TMarker(0.81,nextline,marknum);
legmark.SetNDC();
legmark.SetX(xalign+0.01);
legmark.SetY(nextline+0.011);
TString label = partname + " " + Form("%.0f",vLSP[0].Pt());
myText(xalign+0.03,nextline,partname,color,0.04);
myText(xalign+0.07,nextline,Form("%.0f",vLSP[0].Pt()),color,0.04);
temp.SetMarkerColor(color);
legmark.SetMarkerColor(color);
temp.SetMarkerSize(3);
legmark.SetMarkerSize(2);
genpart.push_back(temp);
genpart.push_back(legmark);
}
if(sig && LSP==1){
int color = LSPcol[LSP];
TString partname = "#tilde{#chi}^{0}_{1}";
if(LSP==0) nextline=start-2*npert*increment;
else nextline = start - (4*npert+1)*increment-offset;
LSP++;
int marknum=25;//27;
// TMarker temp = TMarker(mc_doc_eta->at(imc),mc_doc_phi->at(imc), marknum);
vLSP[1]= vgl[1]-vt[2]-vt[3];
TMarker temp = TMarker(vLSP[1].Eta(),vLSP[1].Phi(), marknum);
TMarker legmark = TMarker(0.81,nextline,marknum);
legmark.SetNDC();
legmark.SetX(xalign+0.01);
legmark.SetY(nextline+0.011);
TString label = partname + " " + Form("%.0f",vLSP[1].Pt());
myText(xalign+0.03,nextline,partname,color,0.04);
myText(xalign+0.07,nextline,Form("%.0f",vLSP[1].Pt()),color,0.04);
temp.SetMarkerColor(color);
legmark.SetMarkerColor(color);
temp.SetMarkerSize(3);
legmark.SetMarkerSize(2);
genpart.push_back(temp);
genpart.push_back(legmark);
}
}
for(int ix=0;ix<(int)genpart.size();ix++)
{
genpart[ix].Draw("same");
}
for(int iy=0;iy<(int)constituents.size();iy++)
{
constituents[iy].Draw("same");
}
for(int ifj = 0; ifj< (int)fjets_pt_->size(); ifj++)
{
if(fatjet) cone[ifj]->Draw();
}
h_fatjets->Draw("colz same");
h_fatjets->SetTitle(Form("Event %lli",event_));
h_fatjets->SetStats(0);
h_fatjets->SetXTitle("#eta");
h_fatjets->SetYTitle("#phi");
TString savename;
if(!sig) savename = "fig/EventDisplay_"+cname+Form("%i_FJ",nfjets_);
else savename = "fig/EventDisplay_"+cname+Form("%i_FJ",nfjets_);
if(truth) savename+="_truth";
else savename+="_notruth";
if(fatjet) savename+="_fatjet";
else savename+="_nofatjet";
c->Print(savename+".pdf");
//c->Print(savename+".C");
c->Close();
h_fatjets->Delete();
}
}
TMarker *buildMark(double x1,double x2, Color_t color = kBlack, Style_t style=20){
TMarker *out = new TMarker(x1,x2,style);
out->SetMarkerColor(color);
return out;
};
void Drawing()
{
frodo *fr = frodo::instance();
fr->Y1.clear();
fr->X1.clear();
fr->Y2.clear();
fr->X2.clear();
fr->Y3.clear();
fr->X3.clear();
fr->Y4.clear();
fr->X4.clear();
fr->AllX.clear();
fr->AllY.clear();
//----------------------------------------------------------------------------------------------------
TCanvas *HBD = new TCanvas("HBD","HBD Event Display",0,0,700,700);
HBD->Range(-50000,-50000,50000,50000);
double Qx1,Qx2,Qx3,Qx4,Qy1,Qy2,Qy3,Qy4;
Qx1=Qx2=Qx3=Qx4=Qy1=Qy2=Qy3=Qy4=0;
for(unsigned int i=0; i<60; i++)
{
Qx1 += fr->J1_XStrips[i].Q();
Qx2 += fr->J2_XStrips[i].Q();
Qx3 += fr->J3_XStrips[i].Q();
Qx4 += fr->J4_XStrips[i].Q();
}
for(unsigned int i=0; i<60; i++)
{
Qy1 += fr->J1_YStrips[i].Q();
Qy2 += fr->J2_YStrips[i].Q();
Qy3 += fr->J3_YStrips[i].Q();
Qy4 += fr->J4_YStrips[i].Q();
}
double Qtot1 = Qx1+Qy1;
double Qtot2 = Qx2+Qy2;
double Qtot3 = Qx3+Qy3;
double Qtot4 = Qx4+Qy4;
//J1 section:-------------------------------------------------------------------
double X1 = -49100; //left
double Y1 = -1375; //bottom
double X2 = -1000; //right
double Y2 = -1075; //top
for(unsigned int i=0; i<60; i++)
{
double T = fr->J1_YStrips[i].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot1*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J1_YStrips[i].Draw(X1,Y1,X2,Y2,1);
HBD->Update();
fr->Y1.push_back(Coordinates(fr->J1_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J1_YStrips[i].YCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int (Qfrac);
fr->J1_YStrips[i].Draw(X1,Y1,X2,Y2,index);
HBD->Update();
fr->Y1.push_back(Coordinates(fr->J1_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J1_YStrips[i].YCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int (Qfrac) - 49;
fr->J1_YStrips[i].Draw(X1,Y1,X2,Y2,2);
HBD->Update();
fr->Y1.push_back(Coordinates(fr->J1_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J1_YStrips[i].YCenter()));
}
Y2 = Y1 - 500;
Y1 = Y2 - 300;
}
double x1 = -1900;
double y1 = -48975;
double x2 = -1200;
double y2 = -48675;
for(unsigned int j=0; j<60; j++)
{
double T = fr->J1_XStrips[j].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot1*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J1_XStrips[j].Draw(x1,y1,x2,y2,1);
HBD->Update();
fr->X1.push_back(Coordinates(fr->J1_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J1_XStrips[j].XCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J1_XStrips[j].Draw(x1,y1,x2,y2,index);
HBD->Update();
fr->X1.push_back(Coordinates(fr->J1_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J1_XStrips[j].XCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac) - 49;
fr->J1_XStrips[j].Draw(x1,y1,x2,y2,2);
HBD->Update();
fr->X1.push_back(Coordinates(fr->J1_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J1_XStrips[j].XCenter()));
}
x2 = x1 - 100;
x1 = x2 - 700;
}
//J2 section:----------------------------------------------------------------
//cout<<"J2: "<<endl;
X1 = 900;
Y1 = -1375;
X2 = 49000;
Y2 = -1075;
for(unsigned int i=0; i<60; i++)
{
double T = fr->J2_YStrips[i].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot2*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J2_YStrips[i].Draw(X1,Y1,X2,Y2,1);
HBD->Update();
fr->Y2.push_back(Coordinates(fr->J2_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J2_YStrips[i].YCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J2_YStrips[i].Draw(X1,Y1,X2,Y2,index);
HBD->Update();
fr->Y2.push_back(Coordinates(fr->J2_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J2_YStrips[i].YCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac)-49;
fr->J2_YStrips[i].Draw(X1,Y1,X2,Y2,2);
HBD->Update();
fr->Y2.push_back(Coordinates(fr->J2_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J2_YStrips[i].YCenter()));
}
Y2 = Y1 - 500;
Y1 = Y2 - 300;
}
x1 = 48100;
y1 = -48975;
x2 = 48800;
y2 = -48675;
for(unsigned int j=0; j<60; j++)
{
double T = fr->J2_XStrips[j].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot2*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J2_XStrips[j].Draw(x1,y1,x2,y2,1);
HBD->Update();
fr->X2.push_back(Coordinates(fr->J2_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J2_XStrips[j].XCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J2_XStrips[j].Draw(x1,y1,x2,y2,index);
HBD->Update();
fr->X2.push_back(Coordinates(fr->J2_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J2_XStrips[j].XCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac)-49;
fr->J2_XStrips[j].Draw(x1,y1,x2,y2,2);
HBD->Update();
fr->X2.push_back(Coordinates(fr->J2_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J2_XStrips[j].XCenter()));
}
x2 = x1 - 100;
x1 = x2 - 700;
}
//J3 section:----------------------------------------------------------------
X1 = -49100;
Y1 = 1125;
X2 = -1000;
Y2 = 1425;
for(unsigned int i=0; i<60; i++)
{
double T = fr->J3_YStrips[i].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot3*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J3_YStrips[i].Draw(X1,Y1,X2,Y2,1);
HBD->Update();
fr->Y3.push_back(Coordinates(fr->J3_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J3_YStrips[i].YCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J3_YStrips[i].Draw(X1,Y1,X2,Y2,index);
HBD->Update();
fr->Y3.push_back(Coordinates(fr->J3_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J3_YStrips[i].YCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac)-49;
fr->J3_YStrips[i].Draw(X1,Y1,X2,Y2,2);
HBD->Update();
fr->Y3.push_back(Coordinates(fr->J3_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J3_YStrips[i].YCenter()));
}
Y1 = Y2 + 500;
Y2 = Y1 + 300;
}
x1 = -49100;
y1 = 1525;
x2 = -48400;
y2 = 1825;
for(unsigned int j=0; j<60; j++)
{
double T = fr->J3_XStrips[j].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot3*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J3_XStrips[j].Draw(x1,y1,x2,y2,1);
HBD->Update();
fr->X3.push_back(Coordinates(fr->J3_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J3_XStrips[j].XCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J3_XStrips[j].Draw(x1,y1,x2,y2,index);
HBD->Update();
fr->X3.push_back(Coordinates(fr->J3_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J3_XStrips[j].XCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac)-49;
fr->J3_XStrips[j].Draw(x1,y1,x2,y2,2);
HBD->Update();
fr->X3.push_back(Coordinates(fr->J3_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J3_XStrips[j].XCenter()));
}
x1 = x2 + 100;
x2 = x1 + 700;
}
//J4 section:-----------------------------------------------------------------
X1 = 950;
Y1 = 1125;
X2 = 49050;
Y2 = 1425;
for(unsigned int i=0; i<60; i++)
{
double T = fr->J4_YStrips[i].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot4*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J4_YStrips[i].Draw(X1,Y1,X2,Y2,1);
HBD->Update();
fr->Y4.push_back(Coordinates(fr->J4_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J4_YStrips[i].YCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J4_YStrips[i].Draw(X1,Y1,X2,Y2,index);
HBD->Update();
fr->Y4.push_back(Coordinates(fr->J4_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J4_YStrips[i].YCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac)-49;
fr->J4_YStrips[i].Draw(X1,Y1,X2,Y2,2);
HBD->Update();
fr->Y4.push_back(Coordinates(fr->J4_YStrips[i].YCenter()));
fr->AllY.push_back(Coordinates(fr->J4_YStrips[i].YCenter()));
}
Y1 = Y2 + 500;
Y2 = Y1 + 300;
}
x1 = 950;
y1 = 1525;
x2 = 1650;
y2 = 1825;
for(unsigned int j=0; j<60; j++)
{
double T = fr->J4_XStrips[j].Q();
//double Qfrac = exp(-T/20)*100;
double Qfrac = T/Qtot4*100;
if(Qfrac>0.04 && Qfrac<1)
{
fr->J4_XStrips[j].Draw(x1,y1,x2,y2,1);
HBD->Update();
fr->X4.push_back(Coordinates(fr->J4_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J4_XStrips[j].XCenter()));
}
else if(Qfrac>=1 && Qfrac<50)
{
int index = int(Qfrac);
fr->J4_XStrips[j].Draw(x1,y1,x2,y2,index);
HBD->Update();
fr->X4.push_back(Coordinates(fr->J4_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J4_XStrips[j].XCenter()));
}
else if(Qfrac>=50 && Qfrac<99.98)
{
//int index = int(Qfrac)-49;
fr->J4_XStrips[j].Draw(x1,y1,x2,y2,2);
HBD->Update();
fr->X4.push_back(Coordinates(fr->J4_XStrips[j].XCenter()));
fr->AllX.push_back(Coordinates(fr->J4_XStrips[j].XCenter()));
}
x1 = x2 + 100;
x2 = x1 + 700;
}
//Drawing the Central Cross:
X1 = -49100;
Y1 = -675;
X2 = 49050;
Y2 = 675;
TBox Hori (X1,Y1, X2,Y2);
Hori.SetFillColor(15);
Hori.DrawBox(X1,Y1, X2,Y2);
HBD->Update();
X1 = -500;
Y1 = -48975;
X2 = 500;
Y2 = 49025;
TBox Vert (X1,Y1, X2,Y2);
Vert.SetFillColor(15);
Vert.DrawBox(X1,Y1, X2,Y2);
HBD->Update();
//Drawing the real positions of the photons:
for(unsigned int a=0; a<fr->HBDParticles.size(); a++)
{
fr->HBDParticles[a].Draw();
HBD->Update();
}
//Finding the ring:----------------------------------------------------------------------------------
Hough = new TH3D ("Hough","Hough",500,-50000,50000,500,-50000,50000,500,0,50000);
Houghxy = new TH2D ("Houghxy","Houghxy",100,-50000,50000,100,-50000,50000);
Houghxr = new TH2D ("Houghxr","Houghxr",100,-50000,50000,100,0,50000);
Houghyr = new TH2D ("Houghyr","Houghyr",100,-50000,50000,100,0,50000);
//Forming J1234 PseudoPoints:
for(unsigned int s=0; s<fr->Y1.size(); s++)
{
for(unsigned int ps=0; ps<fr->X1.size(); ps++)
{
fr->J1_PseudoPoints.push_back(TVector3(fr->X1[ps].Cor(),fr->Y1[s].Cor(),0));
}
}
//cout<<"J1 size:"<<fr->J1_PseudoPoints.size()<<endl;
for(unsigned int s=0; s<fr->Y2.size(); s++)
{
for(unsigned int ps=0; ps<fr->X2.size(); ps++)
{
fr->J2_PseudoPoints.push_back(TVector3(fr->X2[ps].Cor(),fr->Y2[s].Cor(),0));
}
}
//cout<<"J2 size:"<<fr->J2_PseudoPoints.size()<<endl;
for(unsigned int s=0; s<fr->Y3.size(); s++)
{
for(unsigned int ps=0; ps<fr->X3.size(); ps++)
{
fr->J3_PseudoPoints.push_back(TVector3(fr->X3[ps].Cor(),fr->Y3[s].Cor(),0));
}
}
//cout<<"J3 size:"<<fr->J3_PseudoPoints.size()<<endl;
for(unsigned int s=0; s<fr->Y4.size(); s++)
{
for(unsigned int ps=0; ps<fr->X4.size(); ps++)
{
fr->J4_PseudoPoints.push_back(TVector3(fr->X4[ps].Cor(),fr->Y4[s].Cor(),0));
}
}
//Selecting 3 PseudoPoints in three different quadrants:
//J1,J2,J3:
if(fr->J1_PseudoPoints.size() >= 1 && fr->J2_PseudoPoints.size() >= 1 && fr->J3_PseudoPoints.size() >= 1)
{
for(unsigned int i=0; i<fr->J1_PseudoPoints.size(); i++)
{
for(unsigned int j=0; j<fr->J2_PseudoPoints.size(); j++)
{
for(unsigned int k=0; k<fr->J3_PseudoPoints.size(); k++)
{
double Rad = FindR(fr->J1_PseudoPoints[i],fr->J2_PseudoPoints[j],fr->J3_PseudoPoints[k]);
double Xcen = FindX(fr->J1_PseudoPoints[i].X(),fr->J1_PseudoPoints[i].Y(),fr->J2_PseudoPoints[j].X(),fr->J2_PseudoPoints[j].Y(),fr->J3_PseudoPoints[k].X(),fr->J3_PseudoPoints[k].Y());
double Ycen = FindY(fr->J1_PseudoPoints[i].X(),fr->J1_PseudoPoints[i].Y(),fr->J2_PseudoPoints[j].X(),fr->J2_PseudoPoints[j].Y(),fr->J3_PseudoPoints[k].X(),fr->J3_PseudoPoints[k].Y());
Hough->Fill(Xcen,Ycen,Rad);
Houghxy->Fill(Xcen,Ycen);
Houghyr->Fill(Ycen,Rad);
Houghxr->Fill(Xcen,Rad);
}
}
}
}
//J1,J2,J4:
if(fr->J1_PseudoPoints.size() >= 1 && fr->J2_PseudoPoints.size() >= 1 && fr->J4_PseudoPoints.size() >= 1)
{
for(unsigned int i=0; i<fr->J1_PseudoPoints.size(); i++)
{
for(unsigned int j=0; j<fr->J2_PseudoPoints.size(); j++)
{
for(unsigned int k=0; k<fr->J4_PseudoPoints.size(); k++)
{
double Rad = FindR(fr->J1_PseudoPoints[i],fr->J2_PseudoPoints[j],fr->J4_PseudoPoints[k]);
double Xcen = FindX(fr->J1_PseudoPoints[i].X(),fr->J1_PseudoPoints[i].Y(),fr->J2_PseudoPoints[j].X(),fr->J2_PseudoPoints[j].Y(),fr->J4_PseudoPoints[k].X(),fr->J4_PseudoPoints[k].Y());
double Ycen = FindY(fr->J1_PseudoPoints[i].X(),fr->J1_PseudoPoints[i].Y(),fr->J2_PseudoPoints[j].X(),fr->J2_PseudoPoints[j].Y(),fr->J4_PseudoPoints[k].X(),fr->J4_PseudoPoints[k].Y());
Hough->Fill(Xcen,Ycen,Rad);
Houghxy->Fill(Xcen,Ycen);
Houghyr->Fill(Ycen,Rad);
Houghxr->Fill(Xcen,Rad);
}
}
}
}
//J1,J3,J4:
if(fr->J1_PseudoPoints.size() >= 1 && fr->J3_PseudoPoints.size() >= 1 && fr->J4_PseudoPoints.size() >= 1)
{
for(unsigned int i=0; i<fr->J1_PseudoPoints.size(); i++)
{
for(unsigned int j=0; j<fr->J3_PseudoPoints.size(); j++)
{
for(unsigned int k=0; k<fr->J4_PseudoPoints.size(); k++)
{
double Rad = FindR(fr->J1_PseudoPoints[i],fr->J3_PseudoPoints[j],fr->J4_PseudoPoints[k]);
double Xcen = FindX(fr->J1_PseudoPoints[i].X(),fr->J1_PseudoPoints[i].Y(),fr->J3_PseudoPoints[j].X(),fr->J3_PseudoPoints[j].Y(),fr->J4_PseudoPoints[k].X(),fr->J4_PseudoPoints[k].Y());
double Ycen = FindY(fr->J1_PseudoPoints[i].X(),fr->J1_PseudoPoints[i].Y(),fr->J3_PseudoPoints[j].X(),fr->J3_PseudoPoints[j].Y(),fr->J4_PseudoPoints[k].X(),fr->J4_PseudoPoints[k].Y());
Hough->Fill(Xcen,Ycen,Rad);
Houghxy->Fill(Xcen,Ycen);
Houghyr->Fill(Ycen,Rad);
Houghxr->Fill(Xcen,Rad);
}
}
}
}
//J2,J3,J4;
if(fr->J2_PseudoPoints.size() >= 1 && fr->J3_PseudoPoints.size() >= 1 && fr->J4_PseudoPoints.size() >= 1)
{
for(unsigned int i=0; i<fr->J2_PseudoPoints.size(); i++)
{
for(unsigned int j=0; j<fr->J3_PseudoPoints.size(); j++)
{
for(unsigned int k=0; k<fr->J4_PseudoPoints.size(); k++)
{
double Rad = FindR(fr->J2_PseudoPoints[i],fr->J3_PseudoPoints[j],fr->J4_PseudoPoints[k]);
double Xcen = FindX(fr->J2_PseudoPoints[i].X(),fr->J2_PseudoPoints[i].Y(),fr->J3_PseudoPoints[j].X(),fr->J3_PseudoPoints[j].Y(),fr->J4_PseudoPoints[k].X(),fr->J4_PseudoPoints[k].Y());
double Ycen = FindY(fr->J2_PseudoPoints[i].X(),fr->J2_PseudoPoints[i].Y(),fr->J3_PseudoPoints[j].X(),fr->J3_PseudoPoints[j].Y(),fr->J4_PseudoPoints[k].X(),fr->J4_PseudoPoints[k].Y());
Hough->Fill(Xcen,Ycen,Rad);
Houghxy->Fill(Xcen,Ycen);
Houghyr->Fill(Ycen,Rad);
Houghxr->Fill(Xcen,Rad);
}
}
}
}
/*
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!DO NOT ERASE!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
//Forming PseudoPoints:
for(unsigned int ps=0; ps<fr->AllX.size(); ps++)
{
for(unsigned int s=0; s<fr->AllY.size(); s++)
{
fr->PseudoPoints.push_back(TVector3(fr->AllX[ps].Cor(),fr->AllY[s].Cor(),0));
}
}
//Full Combinatorial:
for(unsigned int i=0; i<fr->PseudoPoints.size()-2; i++)
{
for(unsigned int j=i+1; j<fr->PseudoPoints.size()-1; j++)
{
for(unsigned int k=j+1; k<fr->PseudoPoints.size(); k++)
{
double Rad = FindR(fr->PseudoPoints[i],fr->PseudoPoints[j],fr->PseudoPoints[k]);
double Xcen = FindX(fr->PseudoPoints[i].X(),fr->PseudoPoints[i].Y(),fr->PseudoPoints[j].X(),fr->PseudoPoints[j].Y(),fr->PseudoPoints[k].X(),fr->PseudoPoints[k].Y());
double Ycen = FindY(fr->PseudoPoints[i].X(),fr->PseudoPoints[i].Y(),fr->PseudoPoints[j].X(),fr->PseudoPoints[j].Y(),fr->PseudoPoints[k].X(),fr->PseudoPoints[k].Y());
Hough->Fill(Xcen,Ycen,Rad);
Houghxy->Fill(Xcen,Ycen);
Houghyr->Fill(Ycen,Rad);
Houghxr->Fill(Xcen,Rad);
}
}
}
//!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
*/
int xbin, ybin, zbin;
xbin=0; ybin=0; zbin=0;
Hough->GetMaximumBin(xbin, ybin, zbin);
//cout << "xbin= " << xbin << " ybin= " << ybin << " zbin= " << zbin << endl;
TAxis *xaxis = Hough->GetXaxis();
double Xrec=xaxis->GetBinCenter(xbin);
cout<<"Xrec= "<< Xrec <<endl;
TAxis *yaxis = Hough->GetYaxis();
double Yrec=yaxis->GetBinCenter(ybin);
cout<<"Yrec= "<< Yrec <<endl;
TAxis *zaxis = Hough->GetZaxis();
double Rrec=zaxis->GetBinCenter(zbin);
cout<<"Rrec= "<< Rrec <<endl;
//Drawing the founded ring and its center:
TEllipse ring (Xrec,Yrec,Rrec,Rrec,1,360,0);
ring.SetLineColor(2);
ring.SetFillStyle(0);
ring.DrawEllipse(Xrec,Yrec,Rrec,Rrec,0,360,0);
HBD->Update();
TMarker circumcenter (Xrec,Yrec,8);
circumcenter.SetMarkerColor(2);
circumcenter.SetMarkerSize(1);
circumcenter.DrawMarker(Xrec,Yrec);
HBD->Update();
}
void Neutrals (/*TH1 *h1*/)//const char *part="e")//,int nevent = 1000)
{
cout << "TEST VII" << endl;
char infile[100];
int pz[9] = {1,2,4,8,12,16,32,40,50};
double x[9];
double means[9] = {0};
double y[9];
double mval = 0;
int lastbincont = 0;
bool lastbin = false;
int var = 11;
double intnum[1000] = {0};
double varnum[1000] = {0};
char *part[] = {"gamma","neutron","anti_neutron"};
int i = 0;
int w =4;
//TCanvas *c3 = new TCanvas("TresGraphitos","stupid graph",1);
// TH1F *h4 = new TH1F("histo","histo",1000,0,.5);
//TH2F *h3 = new TH2F("HistoGraph","HistoGraph",1000,0,.55,1000,0,1.2);
// TMarker *grmark = new TMarker();
// TCanvas *c19 = new TCanvas();
// h3->Draw();
for (int q=0; q<3; q++)
{
for (int w=0; w<9;w++)
{
cout << "TEST VI" << endl;
char ffile[100];
sprintf(ffile,"/phenix/u/jpinkenburg/sPHENIX/analysis/AntiSigma/macros/HistoBaseFiles/%s_ThinBins.root", part[q]);
TFile *f = TFile::Open(ffile);
char hname[100];
sprintf(hname,"%s%dd",part[q],pz[w]);
TH1F *h1 = (TH1F *)f->Get(hname);
var = 11;
i = 0;
while (var <= 8001)
{
if (i<1000)
{
varnum[i] = .005*(i);
intnum[i] = (h1->Integral(var-10,var))/10000.;//(h1->Integral());
i++;
}
var+=10;
}
// varnum[0] = 0;
// intnum[0] = 0;
char fname [100];
sprintf(fname, "Gamma_Neutron_Hijing_Energy_Graphs.root");
TFile *fout = TFile::Open(fname,"UPDATE");
// h1->Write();
//fout->Write();
f->Close();
//char hname[100];
//sprintf(hname,"%s%dGeV",part[q],pz[w]);
// cout << h1->Integral() << " " << hname << " " << pz[i] << endl;
//TH2 *h2 = new TH2F("hname","hname",1,0,5,1,0,1.2);
gStyle->SetOptStat(0);
TMarker *mean = new TMarker();
mean->SetMarkerStyle(20);
mean->SetMarkerColor(3);
char canvname[100];
sprintf(canvname,"%s%d",part[q],pz[w]);
//TCanvas *c3 = new TCanvas(canvname,"stupid graph",1);
//double dtot[9] = {0.1,0.2,0.3,0.4,0.5,0.6,0.7,0.8,0.9};
cout << "TEST V" << endl;
TGraph *gr = new TGraph((i-2),varnum,intnum);
char gtitle[100];
sprintf(gtitle,"%s%dGeV;ConeSize;Percentage of Energy Deposited",part[q],pz[w]);
char gname[100];
sprintf(gname,"%s%dGeV",part[q],pz[w]);
cout << intnum[50] << " " << varnum[50] << endl;
gr->SetTitle(gtitle);
gr->SetName(gname);
gr->SetMarkerStyle(20);
if (part[q] == "anti_neutron")
{
gr->SetMarkerColor(4);
}
else if (part[q] == "neutron")
{
gr->SetMarkerColor(2);
}
else if (part[q] == "gamma")
{
gr->SetMarkerColor(3);
}
else
{
cout << "idiot" << endl;
}
// gr->SetMaximum(1.2);
//h2->Draw();
//gr->Draw("A*");
gr->Draw("ALP");
//c3->DrawFrame(0,0,0.5,1.2);
// gr->PaintGrapHist((i-1),varnum,intnum,"chopt");
// gr->SetHistogram(h4);
//h4->Write();
gr->Write();
cout << "TEST 1" << endl;
// TH2F *h3 = (TH2F *) "HistoGraph";
cout << "TEST II" << endl;
cout << "TEST III" << endl;
//h3->Write();
cout << "TEST IV" << endl;
//double x;
//double y;
//gr()->GetPoint(
fout->Write();
fout->Close();
// gr->Close();
for (int a=0;a<1000;a++)
{
intnum[a] = 0;
varnum[a] = 0;
}
}
}
}