void drawtorroidmollers_rings(TString infilename="0", TString outfilenamestem="0")
{
if (infilename=="0" || outfilenamestem=="0") {
printf("Usage:\n\t.x drawtorroidmollers_rings_stamps.C(infilename, outfilename)\n\n");
printf("Where infilename is a .root file \n");
printf(" and outfilenamestem is the base name of an image file.\n");
return;
}
gROOT->Reset();
char inifilename[200];//, outfilenamestem[200], parttype[100];
Bool_t debug=0;
// Bool_t doproton=0;
// Bool_t dopostscript=0;
Double_t deltaZwindowsize=20; //in mm
sprintf(inifilename,"%s",infilename.Data());
const Int_t totplots=6;
Double_t distance[totplots]={6000,9000,14500,20000,24000,28250};
TH2F* histos[totplots];
// const Int_t totplots=1;
// Double_t distance[totplots]={28479};
char pstitle[200];
printf("Using data file %s\n",inifilename);
gROOT->SetStyle("Plain");
gStyle->SetPalette(1);
gStyle->SetOptStat(kFALSE);
TFile *file0 = TFile::Open(inifilename);
TCanvas *canvas = new TCanvas("canvas","Moller plots",800,800);
char title[100];
char weight[200];
TGraph* scat[totplots];
TLine line;
TArc arc;
arc.SetFillStyle(4000);
arc.SetNoEdges(kFALSE);
Double_t pi=3.14159265358979323846;
Double_t length=600, openingangle=pi/14.;
Double_t ymin, ymax, xmin, xmax;
TTree *geant = (TTree*)file0->Get("geant");
geant->SetMarkerStyle(20);
geant->SetMarkerSize(0.15);
for (int i=0; i<totplots; i++) {
char drawstring[200];
sprintf(drawstring,"y/1000:x/1000:1000*acos(pz2/sqrt(px2*px2+py2*py2+pz2*pz2))>>htemp%i",i);
sprintf(weight,"(z>(%.0f)&&z<(%.0f)&&type==0)",distance[i]-deltaZwindowsize,distance[i]+deltaZwindowsize);
if (debug) printf("%s\n%s\n",drawstring,weight);
canvas->cd(i+1);
Int_t num = geant->Draw(drawstring,weight,"colz");
histos[i]=(TH2F*)gPad->GetPrimitive(Form("htemp%i",i));
if (num<=0) {
printf("\n%.0f doesn't seem to be a good choice!\n\n",distance[i]);
} else {
if (debug) printf("%i successful events\n",num);
gPad->SetRightMargin(0.12);
gPad->SetLeftMargin(0.12);
gPad->SetTopMargin(0.12);
gPad->SetBottomMargin(0.12);
canvas->Update();
ymin=gPad->GetUymin();
ymax=gPad->GetUymax();
xmin=gPad->GetUxmin();
xmax=gPad->GetUxmax();
printf("xmin %f, xmax %f, ymin %f, ymax %f, deltax %f, deltay %f\n",xmin,xmax,ymin,ymax,xmax-xmin,ymax-ymin);
}
sprintf(title,"z=%.2f m;x (m);y (m)", distance[i]/1000.);
histos[i]->SetTitle(title);
canvas->Update();
canvas->Print(Form("%s_%.2fm.png",outfilenamestem.Data(),distance[i]/1000.));
}
}
void drawtorroidmollers_rings_compare(TString infilename1="0", TString infilename2="0",
TString outfilenamestem="0", TString usercut="1")
{
if (infilename1=="0" || infilename2=="0" || outfilenamestem=="0") {
printf("Usage:\n\t.x drawtorroidmollers_rings_stamps.C(infilename1, infilename2, outfilename, [cut])\n\n");
printf("Where infilenames is a .root file \n");
printf(" outfilenamestem is the base name of an image file, and\n");
printf(" cut is a cut string applied to the tree.\n");
return;
}
gROOT->Reset();
Bool_t debug=1;
Double_t deltaZwindowsize=25; //in mm
const Int_t totplots=4;
Double_t distance[totplots]={14500,20000,24000,28250};
// const Int_t totplots=1;
// Double_t distance[totplots]={28479};
char pstitle[200];
printf("Using data files %s\n and %s \n\n",infilename1.Data(), infilename2.Data());
gROOT->SetStyle("Plain");
gStyle->SetPalette(1);
gStyle->SetOptStat(kFALSE);
TFile *file1 = TFile::Open(infilename1.Data());
TTree *tree1 = (TTree*)file1->Get("geant");
TFile *file2 = TFile::Open(infilename2.Data());
TTree *tree2 = (TTree*)file2->Get("geant");
TH2F* file1histos[totplots];
TH2F* file2histos[totplots];
Int_t Palette1[100],Palette2[100];
UInt_t Number = 2;
Int_t nb=20;
Double_t Red[] = { 1.00, 1.00};
Double_t Green[] = { 0.00, 0.00};
Double_t Blue[] = { 1.00, 0.00};
Double_t Length[] = { 0.00, 1.00};
Int_t FI = TColor::CreateGradientColorTable(Number,Length,Red,Green,Blue,nb);
for (int i=0;i<nb;i++) Palette1[i] = FI+i;
Double_t Red[] = { 0.00, 0.00};
Double_t Green[] = { 1.00, 0.00};
Double_t Blue[] = { 1.00, 1.00};
TColor::CreateGradientColorTable(Number,Length,Red,Green,Blue,nb);
Int_t FI = TColor::CreateGradientColorTable(Number,Length,Red,Green,Blue,nb);
for (int i=0;i<nb;i++) Palette2[i] = FI+i;
TCanvas *canvas = new TCanvas("canvas","Moller plots",800,800);
char title[100];
char weight[200];
TGraph* scat[totplots];
TLine line;
TArc arc;
arc.SetFillStyle(4000);
arc.SetNoEdges(kFALSE);
Double_t pi=3.14159265358979323846;
Double_t ymin, ymax, xmin, xmax;
tree1->SetMarkerStyle(20);
tree1->SetMarkerSize(0.15);
tree2->SetMarkerStyle(20);
tree2->SetMarkerSize(0.15);
for (int i=0; i<totplots; i++) {
char drawstring[400];
gStyle->SetPalette(nb, Palette1);
sprintf(drawstring,"y/1000:x/1000:1000*acos(pz2/sqrt(px2*px2+py2*py2+pz2*pz2))>>file1hist%i",i);
// sprintf(drawstring,"y/1000:x/1000:5>>file1hist%i",i);
sprintf(weight,"(z>(%.0f)&&z<(%.0f)&&type==0)&&(%s)",distance[i]-deltaZwindowsize,distance[i]+deltaZwindowsize,usercut.Data());
if (debug) printf("%s\n%s\n",drawstring,weight);
Int_t num1 = tree2->Draw(drawstring,weight,"col");
file1histos[i]=(TH2F*)gPad->GetPrimitive(Form("file1hist%i",i));
gStyle->SetPalette(nb, Palette2);
sprintf(drawstring,"y/1000:x/1000:1000*acos(pz2/sqrt(px2*px2+py2*py2+pz2*pz2))>>file2hist%i",i);
// sprintf(drawstring,"y/1000:x/1000:3>>file2hist%i",i);
if (debug) printf("%s\n%s\n",drawstring,weight);
Int_t num2 = tree1->Draw(drawstring,weight,"col");
file2histos[i]=(TH2F*)gPad->GetPrimitive(Form("file2hist%i",i));
if (num1<=0 || num2<=0) {
printf("%i and %i successful events\n",num1,num2);
printf("\n%.0f doesn't seem to be a good choice!\n\n",distance[i]);
} else {
if (debug) printf("%i and %i successful events\n",num1,num2);
gPad->SetRightMargin(0.1);
gPad->SetLeftMargin(0.1);
gPad->SetTopMargin(0.1);
gPad->SetBottomMargin(0.1);
canvas->Update();
ymin=gPad->GetUymin();
ymax=gPad->GetUymax();
xmin=gPad->GetUxmin();
xmax=gPad->GetUxmax();
printf("xmin %f, xmax %f, ymin %f, ymax %f, deltax %f, deltay %f\n",xmin,xmax,ymin,ymax,xmax-xmin,ymax-ymin);
// if (xmin>0) {
// // file1histos[i]->SetAxisRange(0,xmax,"X");
// // file2histos[i]->SetAxisRange(0,xmax,"X");
// file1histos[i]->GetXaxis()->SetRangeUser(0,xmax);
// file2histos[i]->GetXaxis()->SetRangeUser(0,xmax);
// }
sprintf(title,"z=%.2f m;x (m);y (m)", distance[i]/1000.);
file1histos[i]->SetTitle(title);
file2histos[i]->SetTitle(title);
canvas->Clear();
file1histos[i]->Draw("colz");
// if (xmin>0) //file1histos[i]->GetXaxis()->SetRangeUser(0,xmax);
// gPad->SetUxmin(0);//file1histos[i]->SetAxisRange(0,xmax,"X");
canvas->Update();
canvas->Print(Form("%s_%.2fm_1.png",outfilenamestem.Data(),distance[i]/1000.));
file2histos[i]->Draw("col,same");
canvas->Update();
canvas->Print(Form("%s_%.2fm.png",outfilenamestem.Data(),distance[i]/1000.));
canvas->Clear();
file2histos[i]->Draw("colz");
// if (xmin>0) //file2histos[i]->GetXaxis()->SetRangeUser(0,xmax);
// gPad->SetUxmin(0);//file1histos[i]->SetAxisRange(0,xmax,"X");
canvas->Update();
canvas->Print(Form("%s_%.2fm_2.png",outfilenamestem.Data(),distance[i]/1000.));
}
}
}
