void treegraph(TString filename) { gROOT->SetStyle("Plain"); gStyle->SetOptDate(); Double_t x, y; Int_t nlines = 0; TFile *f = new TFile("graph.root","RECREATE"); TCanvas *canvas_graph = new TCanvas("canvas_graph", "y vs x",467,89,400,700); TTree t; t.ReadFile(filename,"x:y"); t.Draw("x:y","","goff"); TGraph *g = new TGraph(t.GetSelectedRows(),t.GetV1(),t.GetV2()); g->SetTitle(filename+": Y vs X"); g->GetXaxis()->SetTitle("x[a.u.]"); g->GetYaxis()->SetTitle("y[a.u.]"); g->SetMarkerStyle(21); g->Draw("AP"); f->Write(); }
/* * The Ferromagnet_Scan function reads in * a data file of magentic field scans * from within the ferromagnet and returns * a graph of the magnetic permeability * of the ferromagnet vs the external field * provided by the Helmholtz coil. */ TGraphErrors* plot_uvB( const TString scan_file, TF1* calib_fit, double R, double R_sig ) { /*Read in Data File to ROOT Tree*/ cout<< "processing file " << scan_file <<endl; TTree *TData = new TTree(); TData->ReadFile(scan_file, "t/D:I:B"); /*Use TTree Draw command to write branches to usable arrays*/ int n = TData->Draw("I:TMath::Abs(B):0.0:0.005", "", "goff"); vector<double> B_ext, B_in, Bratio, u; for(int i = 0; i < n; i++) { B_ext.push_back( calib_fit->Eval(TData->GetV1()[i]) ); B_in.push_back( TData->GetV2()[i] ); Bratio.push_back( B_in[i] / B_ext[i]); u.push_back( (Bratio[i]*(R**2) + Bratio[i] - 2 - 2*sqrt((Bratio[i]**2)*(R**2) - Bratio[i]*(R**2) - Bratio[i] + 1 ) ) / (Bratio[i]*(R**2) - Bratio[i]) ); } TGraphErrors *g_uvB = new TGraphErrors(n, &B_ext[0], &u[0], TData->GetV3(), TData->GetV4()); /* g_uvB->Fit("pol1", "", "", 10, 60); cout << "Permeability at 50: " << g_uvB->GetFunction("pol1")->Eval(50) << endl; cout << "Permeability at 40: " << g_uvB->GetFunction("pol1")->Eval(40) << endl; cout << "Permeability at 30: " << g_uvB->GetFunction("pol1")->Eval(30) << endl; */ return g_uvB; }
makePlot_check_all() { gStyle->SetOptStat(0); vector< TString > v_infile; // v_infile.push_back( TString("output/fitResults_run001_1layer_sheath.root") ); // v_infile.push_back( TString("output/fitResults_run002_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run003_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run004_2layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run005_2layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run006_3layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run007_4layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run008_2layer_cylinder.root") ); v_infile.push_back( TString("output/fitResults_run009_2layer_cylinder.root") ); // v_infile.push_back( TString("output/fitResults_run010_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run011_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run012_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run013_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run014_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run015_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run016_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run017_2layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run018_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run019_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run020_1layer_sheath.root") ); v_infile.push_back( TString("output/fitResults_run021_1layer.root") ); for ( unsigned f = 0; f < v_infile.size(); f++ ) { cout << "Plot " << v_infile.at(f) << endl; TCanvas *c0 = new TCanvas(); TFile *fin = new TFile( v_infile.at(f) ); TTree *tin = (TTree*)fin->Get("fitResults"); tin->Draw("bshield:bext:bshield_err:bext_err"); TGraphErrors *gshield = new TGraphErrors( tin->GetEntries(), tin->GetV2(), tin->GetV1(), tin->GetV4(), tin->GetV3() ); gshield->SetTitle(v_infile.at(f)); gshield->GetXaxis()->SetTitle("B_{ext} [mT]"); gshield->GetYaxis()->SetTitle("B_{shield} [mT]"); gshield->Draw("AP"); fin->Close(); if ( f == 0 ) c0->Print("plot_check_all.ps("); else if ( f == v_infile.size() - 1 ) c0->Print("plot_check_all.ps)"); else c0->Print("plot_check_all.ps"); } }
/* * The Calibration function reads in the * calibration file and returns a graph * that is used to find the relation * between current and magentic field * for the Helmholtz coil. */ TF1* Calibrate( const char* f_calib ) { /*Read in Calibration File*/ cout<< "processing file " << f_calib <<endl; TTree *TCalib = new TTree(); TCalib->ReadFile(f_calib, "t/D:I:B"); int n = TCalib->Draw("I:TMath::Abs(B)", "", "goff"); TGraph *g_calib = new TGraph(n, TCalib->GetV1(), TCalib->GetV2() ); g_calib->SetTitle(""); // g_calib->Draw("AP"); g_calib->Fit("pol1", "q"); TF1* calib_fit = g_calib->GetFunction("pol1"); return calib_fit; }
TGraphErrors* plot_ramp( const TString scan_file, double R, double R_sig ) { /*Read in Data File to ROOT Tree*/ cout<< "processing file " << scan_file <<endl; TTree *TData = new TTree(); TData->ReadFile(scan_file, "t/D:x:y:z:B1:B1Range:B2:B2Range:B3:B3Range:V:T1:T2:T3:T4:T5:T6:T7:T8"); /*Use TTree Draw command to write branches to usable vector*/ int n = TData->Draw("TMath::Abs(B2):TMath::Abs(B3):0.0:0.005", "", "goff"); vector<double> B_ext, B_in, Bratio, u, u_err; for(int i = 0; i < n; i++) { // if(i == 0) // { // B_ext.push_back(100.00); // } // else{ // B_ext.push_back( TData->GetV1()[i]); // } B_ext.push_back( TData->GetV1()[i] ); B_in.push_back( TData->GetV2()[i]); // B_in.push_back( 90.53); // if(TMath::Abs(B_in[i]) <= TMath::Abs(B_ext[i])) // { Bratio.push_back( B_in[i] / B_ext[i]); u.push_back( (Bratio[i]*(R**2) + Bratio[i] - 2 - 2*sqrt((Bratio[i]**2)*(R**2) - Bratio[i]*(R**2) - Bratio[i] + 1 ) ) / (Bratio[i]*(R**2) - Bratio[i]) ); // u_err.push_back( u[i]*( 5.5*((0.005/Bratio[i])**2.) + 10*((R_sig/R)**2.) )**0.5 ); u_err.push_back( 0.01 ); // u_err.push_back( u[i]*( 5.5*((0.01/Bratio[i])**2.)) )**0.5 ); // } } TGraphErrors *g_uvB = new TGraphErrors(n, &B_ext[0], &u[0], TData->GetV3(), &u_err[0]); return g_uvB; }
int evaluate( std::string filelist, std::string outfile ) { gStyle->SetOptStat(0); TCanvas *ctemp = new TCanvas(); TCanvas *cres = new TCanvas("TimeDependence"); TH1F* hres = new TH1F("hres","",100,0,650); hres->GetYaxis()->SetRangeUser(0,50); hres->SetTitle(""); hres->GetXaxis()->SetTitle("time (s)"); hres->GetYaxis()->SetTitle("B_{int} (mT)"); hres->Draw(); leg = new TLegend(0.2,0.6,0.9,0.9); // leg->SetHeader("The Legend Title"); // option "C" allows to center the header leg->SetNColumns(5); vector< double > v_Bint; vector< double > v_BintErr; vector< double > v_Bext; vector< double > v_BextErr; /* Loop over all lines in input file */ std::ifstream infilelist(filelist); std::string line; unsigned colorcounter=38; while (std::getline(infilelist, line)) { // skip lines with '#' and empty lines if ( line.find("#") != string::npos ) { cout << "Skip line " << line << endl; continue; } if ( line == "" ) continue; //cout << "Processing file " << line << endl; TString infilename("data_calib/"); infilename.Append(line); TFile *fin = new TFile( infilename ); TTree *tin = (TTree*)fin->Get("t"); ctemp->cd(); tin->Draw("Bi:time"); TGraph *gtime = new TGraph(tin->GetEntries(), &(tin->GetV2()[0]), &(tin->GetV1()[0])); gtime->SetLineColor(colorcounter); colorcounter++; TH1F* hBext = new TH1F("hBext","",100,0,1000); tin->Draw("Bo >> hBext"); cres->cd(); gtime->Draw("lsame"); double Bext_i = hBext->GetMean(); double BextErr_i = hBext->GetRMS(); double Bint_i = gtime->Eval(590); double BintErr_i = 0; /* add legend entry */ TString legname("B_ext ~ "); legname += (int)Bext_i; leg->AddEntry(gtime,legname,"l"); cout << "B_ext: " << Bext_i << " \t B_int: " << Bint_i << endl; v_Bint.push_back(Bint_i); v_BintErr.push_back(BintErr_i); v_Bext.push_back(Bext_i); v_BextErr.push_back(BextErr_i); } cres->cd(); leg->Draw(); TGraphErrors *gfinal = new TGraphErrors(v_Bext.size(), &(v_Bext[0]), &(v_Bint[0]), &(v_BextErr[0]), &(v_BintErr[0])); gfinal->Sort(); gfinal->SetName("Bint_Vs_Bext"); gfinal->SetTitle(""); gfinal->GetXaxis()->SetTitle("B_{ext} (mT)"); gfinal->GetYaxis()->SetTitle("B_{int} (mT)"); TCanvas *cfinal = new TCanvas(); gfinal->Draw("APL"); /* Save output graph */ TString outfilename("output/"); outfilename.Append(outfile); TFile *fout = new TFile(outfilename,"RECREATE"); cres->Write(); gfinal->Write(); fout->Close(); /* Write result to txt output file */ TString outfilenametxt = outfilename; outfilenametxt.ReplaceAll(".root",".txt"); ofstream foutxt; foutxt.open( outfilenametxt ); foutxt << "# Bo sig_Bo Bi sig_Bi shield sig_shield sf sig_sf time_dependent" << endl; for ( int i = 0; i < gfinal->GetN(); i++ ) { double Bo = gfinal->GetX()[i]; double sig_Bo = gfinal->GetEX()[i]; double Bi = gfinal->GetY()[i]; double sig_Bi = gfinal->GetEY()[i]; double shield = 0; double sig_shield = 0; double sf = 0; double sig_sf = 0; double time_dependent = 0; foutxt << Bo << " " << sig_Bo << " " << Bi << " " << sig_Bi << " " << shield << " " << sig_shield << " " << sf << " " << sig_sf << " " << time_dependent << endl; } return 0; }
/*#include "TTree.h" #include "TCanvas.h" #include "TGraph.h" #include "TMultiGraph.h" #include "TRint.h" #include <time.h> #include <stdio.h> const short MaxNN = 6; */ void make_scatterplot(TString data_file, TString plot_name, TString header) { //TApplication program = new TRInt(); //pull in data TTree *t = new TTree(); t->ReadFile(data_file); t->SetName("t"); //TString name = "run2.png"; TCanvas *BG = new TCanvas("c1", "Read Velocity on Local Disk for CMS3 Files", 1920, 1080); BG->cd(); TPad *c = new TPad("MainPad", "My main pad", 0, 0, 1, 0.9); c->Draw(); c->Divide(2,2); //In first slot have Time vs. Buffer for 1 Concurrent Read c->cd(1); int n1 = t->Draw("VelocityMBps:BufferSize", "ConcurrentReads==1", "goff"); TGraph *ghist1 = new TGraph(n1, t->GetV2(), t->GetV1()); ghist1->SetName("ghist1"); ghist1->SetMarkerStyle(3); ghist1->SetMarkerColor(1); ghist1->SetTitle("Single File Read"); ghist1->GetXaxis()->SetTitle("Buffer Size (Bytes)"); ghist1->GetYaxis()->SetTitle("Read Velocity (MB/s)"); ghist1->Draw("ap"); //In second slot have Time vs. Buffer for 3 Concurrent Reads c->cd(2); int n2 = t->Draw("VelocityMBps:BufferSize", "ConcurrentReads==3", "goff"); TGraph *ghist2 = new TGraph(n2, t->GetV2(), t->GetV1()); Double_t DP2x[n2], DP2y[n2]; ghist2->SetName("ghist2"); ghist2->SetMarkerStyle(3); ghist2->SetMarkerColor(1); ghist2->SetTitle("3 Concurrent Reads"); ghist2->GetXaxis()->SetTitle("Buffer Size (Bytes)"); ghist2->GetYaxis()->SetTitle("Read Velocity (MB/s)"); ghist2->Draw("ap"); //In third slot have Time vs. Buffer for 3 Concurrent Reads c->cd(3); int n3 = t->Draw("VelocityMBps:BufferSize", "ConcurrentReads==6", "goff"); TGraph *ghist3 = new TGraph(n3, t->GetV2(), t->GetV1()); Double_t DP3x[n3], DP3y[n3]; ghist3->SetName("ghist3"); ghist3->SetMarkerStyle(3); ghist3->SetMarkerColor(1); ghist3->SetTitle("6 Concurrent Reads"); ghist3->GetXaxis()->SetTitle("Buffer Size (Bytes)"); ghist3->GetYaxis()->SetTitle("Read Velocity (MB/s)"); ghist3->Draw("ap"); //In fourth slot have Time vs. Buffer for 10 Concurrent Reads c->cd(4); int n4 = t->Draw("VelocityMBps:BufferSize", "ConcurrentReads==10", "goff"); TGraph *ghist4 = new TGraph(n4, t->GetV2(), t->GetV1()); ghist4->SetMarkerStyle(3); ghist4->SetName("ghist4"); ghist4->SetMarkerColor(1); ghist4->SetTitle("10 Concurrent Reads"); ghist4->GetXaxis()->SetTitle("Buffer Size (Bytes)"); ghist4->GetYaxis()->SetTitle("Read Velocity (MB/s)"); ghist4->Draw("ap"); //Draw to screen //Initialize Canvas c->cd(0); c->Draw(); BG->cd(); TText *title = new TText(.5,.95, header); title->SetTextAlign(22); title->Draw(); gDirectory->Add(ghist1); gDirectory->Add(ghist2); gDirectory->Add(ghist3); gDirectory->Add(ghist4); gDirectory->Add(t); gPad->SaveAs(plot_name); }
void fillGraphFromTreeVar(std::string& treedrawspec,int index,wGraph_t *&pwg) { // Sample treedrawspec: // mytree:"TDCwinstart[%d]:runnum","evtnum==1","P" // vector<string> v_tokens; string tid; TString drawspec; Tokenize(treedrawspec,v_tokens,":",true); if( (v_tokens.size() < 2) || (!v_tokens[0].size()) || (!v_tokens[2].size()) ) { cerr << "malformed root tree draw spec treeid:\"varexp\",\"selection\",option: " << treedrawspec << endl; return; } tid = v_tokens[0]; for (size_t i=2; i<v_tokens.size(); i++) { drawspec += v_tokens[i]; } int fmtcnt = drawspec.CountChar('%'); if (fmtcnt) { // use index for tree array var switch(fmtcnt) { case 1: drawspec = Form(drawspec,index); break; case 2: drawspec = Form(drawspec,index,index); break; case 3: drawspec = Form(drawspec,index,index,index); break; case 4: drawspec = Form(drawspec,index,index,index,index); break; case 5: drawspec = Form(drawspec,index,index,index,index,index); break; case 6: drawspec = Form(drawspec,index,index,index,index,index,index); break; default: cerr << "More than six fmt specifiers in drawspec found, fix me! " << drawspec <<endl; exit(-1); } } if( gl_verbose) cout<<"drawspec="<<drawspec<<endl; TTree *tree = findTree(tid); assert (tree); // can't use comma as delimiter since histo with binning spec may be supplied TObjArray *tokens = drawspec.Tokenize("\""); TString hname; TString varexp = ((TObjString *)(*tokens)[0])->GetString(); if( gl_verbose) cout<<"varexp="<<varexp<<endl; switch(tokens->GetEntriesFast()) { case 1: tree->Draw(varexp,"","goff"); break; case 3: { TString cut = ((TObjString *)(*tokens)[2])->GetString(); tree->Draw(varexp,cut,"goff"); } break; case 4: // assume the cut string is blank { TString gopt = ((TObjString *)(*tokens)[3])->GetString(); gopt = gopt + " goff"; tree->Draw(varexp,"",gopt); } break; case 5: { TString cut = ((TObjString *)(*tokens)[2])->GetString(); TString gopt = ((TObjString *)(*tokens)[4])->GetString(); gopt = gopt + " goff"; tree->Draw(varexp,cut,gopt); } break; default: cerr << "malformed root tree draw spec treeid:varexp,selection,option"; for (int i=0; i<tokens->GetEntriesFast(); i++) cerr << i<<": "<< ((TObjString *)(*tokens)[i])->GetString() << " "; cerr << endl; break; } assert(tree->GetSelectedRows()); if (!pwg) pwg = new wGraph_t(); assert(pwg); pwg->gr = new TGraph(tree->GetSelectedRows(), tree->GetV2(), tree->GetV1()); } // fillGraphFromTreeVar
int plot_compare_fun4all_eicroot() { /**** Chose input files ****/ /* File with IR mangets configuration*/ TString fname_irmag("example/proton-magnets-250GeV-opt2.dat"); /* File with Fun4All output*/ TString fname_fun4all("example/eRHIC_proton-magnets-250GeV-opt2_250GeV_0mrad.root"); /* File with EICROOT output*/ TString fname_eicroot("example/eicroot-track_proton-magnets-250GeV-opt2_250GeV_0mrad.txt"); /* Open iput file with trajectories from GEANT4 */ TFile *fin = new TFile(fname_fun4all); /* Get tree from file */ TTree *tin = (TTree*)fin->Get("T"); int nhits = 0; tin->SetBranchAddress("n_G4HIT_FWDDISC",&nhits); /* create graph of particle trajectory */ /* Use only first event (for now) */ tin->GetEntry(0); cout << "hits: " << nhits << endl; tin->Draw("G4HIT_FWDDISC.x:G4HIT_FWDDISC.z","Entry$==0",""); TGraph* g1 = new TGraph(nhits*2, &(tin->GetV2()[0]), &(tin->GetV1()[0])); g1->SetMarkerStyle(7); g1->SetMarkerSize(1); g1->SetMarkerColor(kRed); /* Get tree from file */ TTree *tin2 = new TTree(); tin2->ReadFile(fname_eicroot,"x/F:y:z"); int nhits = tin2->GetEntries(); tin2->Draw("x:z","",""); TGraph* g2 = new TGraph(nhits, &(tin2->GetV2()[0]), &(tin2->GetV1()[0])); g2->SetMarkerStyle(7); g2->SetMarkerSize(1); g2->SetMarkerColor(kGreen+1); /* Create frame histogram for plot */ TH1F *h1 = new TH1F("h1","",10,0,15000); h1->GetXaxis()->SetRangeUser(0,5000); h1->GetYaxis()->SetRangeUser(-50,70); h1->GetXaxis()->SetTitle("Z(cm)"); h1->GetYaxis()->SetTitle("X(cm)"); /* Plot frame histogram */ TCanvas *c1 = new TCanvas(); h1->Draw("AXIS"); /* Read IR configuration file- this needs to go somewhere else using parameters and a .root file to store them */ ifstream irstream(fname_irmag); if(!irstream.is_open()) { cout << "ERROR: Could not open IR configuration file " << fname_irmag << endl; return -1; } while(!irstream.eof()){ string str; getline(irstream, str); if(str[0] == '#') continue; //for comments stringstream ss(str); string name; double center_z, center_x, center_y, aperture_radius, length, angle, B, gradient; ss >> name >> center_z >> center_x >> center_y >> aperture_radius >> length >> angle >> B >> gradient; if ( name == "" ) continue; //for empty lines // convert units from m to cm center_x *= 100; center_y *= 100; center_z *= 100; aperture_radius *= 100; length *= 100; // define magnet outer radius float outer_radius = 30.0; // cm //flip sign of dipole field component- positive y axis in Geant4 is defined as 'up', // positive z axis as the hadron-going direction // in a right-handed coordinate system x,y,z B *= -1; // convert angle from millirad to rad angle = (angle / 1000.); // Place IR component cout << "New IR component: " << name << " at z = " << center_z << endl; string volname = "IRMAGNET_"; volname.append(name); /* Draw box for magnet position on canvas */ TPolyLine *b1 = TraceBox( angle, center_z, center_x, length, aperture_radius, outer_radius ); //upper box TPolyLine *b2 = TraceBox( angle, center_z, center_x, length, -1 * aperture_radius, -1 * outer_radius ); //lower box if(B != 0 && gradient == 0.0){ //dipole magnet b1->SetFillColor(kOrange+1); b2->SetFillColor(kOrange+1); } else if( B == 0 && gradient != 0.0){ //quad magnet b1->SetFillColor(kBlue+1); b2->SetFillColor(kBlue+1); } else{ //placeholder magnet b1->SetFillColor(kGray+1); b2->SetFillColor(kGray+1); } b1->Draw("Fsame"); b2->Draw("Fsame"); } /* draw particle trajectory */ g2->Draw("LPsame"); g1->Draw("LPsame"); return 0; }
plot_stability_mean(){ gStyle->SetOptStat(0); TTree *tfit = new TTree(); tfit->ReadFile("pi0peak_fit_Run13pp510ERT_reformat.txt","run_index/F:run_number:nevents:sector:mean:dmean:sigma:dsigma:chisquare"); TTree *tfit_raw = new TTree(); tfit_raw->ReadFile("pi0peak_fit_raw_Run13pp510ERT_reformat.txt","run_index/F:run_number:nevents:sector:mean:dmean:sigma:dsigma:chisquare"); TLine *lpi0 = new TLine( 0, 0.137, tfit->GetEntries("sector==0"), 0.137 ); lpi0->SetLineColor(kRed); // frame TH1F* hframe = new TH1F("hframe","",875,0,875); hframe->GetYaxis()->SetRangeUser(0.12,0.16); hframe->GetXaxis()->SetTitle("run"); hframe->GetYaxis()->SetTitle("mean [GeV]"); hframe->SetLineColor(kWhite); // sectors PbSc-West tfit->Draw("mean:run_index:dmean","sector==0"); TGraphErrors *g_mean_PbScW = new TGraphErrors( tfit->GetEntries("sector==0"), tfit->GetV2(), tfit->GetV1(), 0, tfit->GetV3()); tfit_raw->Draw("mean:run_index:dmean","sector==0"); TGraphErrors *g_mean_PbScW_raw = new TGraphErrors( tfit_raw->GetEntries("sector==0"), tfit_raw->GetV2(), tfit_raw->GetV1(), 0, tfit_raw->GetV3()); // sectors PbSc-East tfit->Draw("mean:run_index:dmean","sector==1"); TGraphErrors *g_mean_PbScE = new TGraphErrors( tfit->GetEntries("sector==1"), tfit->GetV2(), tfit->GetV1(), 0, tfit->GetV3()); tfit_raw->Draw("mean:run_index:dmean","sector==1"); TGraphErrors *g_mean_PbScE_raw = new TGraphErrors( tfit_raw->GetEntries("sector==1"), tfit_raw->GetV2(), tfit_raw->GetV1(), 0, tfit_raw->GetV3()); // sectors PbGl-East tfit->Draw("mean:run_index:dmean","sector==2"); TGraphErrors *g_mean_PbGlE = new TGraphErrors( tfit->GetEntries("sector==2"), tfit->GetV2(), tfit->GetV1(), 0, tfit->GetV3()); tfit_raw->Draw("mean:run_index:dmean","sector==2"); TGraphErrors *g_mean_PbGlE_raw = new TGraphErrors( tfit_raw->GetEntries("sector==2"), tfit_raw->GetV2(), tfit_raw->GetV1(), 0, tfit_raw->GetV3()); // Plotting TCanvas *c_PbScW = new TCanvas(); hframe->Draw(); g_mean_PbScW->Draw("Psame"); lpi0->Draw("same"); c_PbScW->Print("plots-escale-check/mpi0_mean_PbScW.png"); TCanvas *c_PbScW_raw = new TCanvas(); hframe->Draw(); g_mean_PbScW_raw->Draw("Psame"); lpi0->Draw("same"); c_PbScW_raw->Print("plots-escale-check/mpi0_mean_PbScW_raw.png"); TCanvas *c_PbScE = new TCanvas(); hframe->Draw(); g_mean_PbScE->Draw("Psame"); lpi0->Draw("same"); c_PbScE->Print("plots-escale-check/mpi0_mean_PbScE.png"); TCanvas *c_PbScE_raw = new TCanvas(); hframe->Draw(); g_mean_PbScE_raw->Draw("Psame"); lpi0->Draw("same"); c_PbScE_raw->Print("plots-escale-check/mpi0_mean_PbScE_raw.png"); TCanvas *c_PbGlE = new TCanvas(); hframe->Draw(); g_mean_PbGlE->Draw("Psame"); lpi0->Draw("same"); c_PbGlE->Print("plots-escale-check/mpi0_mean_PbGlE.png"); TCanvas *c_PbGlE_raw = new TCanvas(); hframe->Draw(); g_mean_PbGlE_raw->Draw("Psame"); lpi0->Draw("same"); c_PbGlE_raw->Print("plots-escale-check/mpi0_mean_PbGlE_raw.png"); }
int plot_Fun4All_All_DeltaEta_DeltaPhi() { const std::string inFile = "LeptoAna_1000events_All.root"; const std::string inDirectory = "/gpfs/mnt/gpfs02/phenix/scratch/jlab/Leptoquark/"; std::string inputFile = inDirectory+inFile; TFile *f = TFile::Open(inputFile.c_str()); TTree *t = (TTree*)f->Get("ntp_leptoquark"); // const int Nevent = t->GetMaximum("event"); const int Nevent = 100; cout << "Running " << Nevent << " events" << endl; const int Nentries1 = t->Draw("isMaxEnergyJet","(isMaxEnergyJet<10)*(calorimeterid<10)","goff"); Double_t *arr_jet = t->GetV1(); vector<int> v_jet(Nentries1); for(int i = 0; i < Nentries1; i++) { v_jet[i] = (int)arr_jet[i]; } const int Nentries = t->Draw("towereta:towerphi:towerenergy:event","(isMaxEnergyJet<10)*(calorimeterid<10)","goff"); Double_t *arr_eta = t->GetV1(); Double_t *arr_phi = t->GetV2(); Double_t *arr_e = t->GetV3(); Double_t *arr_event = t->GetV4(); if(Nentries1 != Nentries) { cerr << "ERROR: Draw commands to not return the same dimensions. Check that your logical expressions are the same." << endl; return -1; } vector<double> v_DeltaEta, v_DeltaPhi, v_DeltaTheta, v_Energy; vector<double> v_DeltaEta_j2, v_DeltaPhi_j2, v_DeltaTheta_j2, v_Energy_j2; for(int i = 0; i < Nevent; i++) { double Emax = 0; int Emax_i = 0; double Emax_j2 = 0; int Emax_i_j2 = 0; for(int j = 0; j < Nentries; j++) { if((t->GetV4()[j]-1 == i) && (v_jet[j] == 1)) { if(t->GetV3()[j] > Emax) { Emax = t->GetV3()[j]; Emax_i = j; } } if((t->GetV4()[j]-1 == i) && (v_jet[j] == 2)) { if(t->GetV3()[j] > Emax_j2) { Emax_j2 = t->GetV3()[j]; Emax_i_j2 = j; } } } for(int j = 0; j < Nentries; j++) { if((t->GetV4()[j]-1 == i) && (v_jet[j] == 1)) { v_DeltaEta.push_back(t->GetV1()[j] - t->GetV1()[Emax_i]); v_DeltaTheta.push_back(2*TMath::ATan(TMath::Power(TMath::E(),-1*t->GetV1()[j])) - 2*TMath::ATan(TMath::Power(TMath::E(),-1*t->GetV1()[Emax_i]))); v_DeltaPhi.push_back(t->GetV2()[j] - t->GetV2()[Emax_i]); v_Energy.push_back(t->GetV3()[j]); } if((t->GetV4()[j]-1 == i) && (v_jet[j] == 2)) { v_DeltaEta_j2.push_back(t->GetV1()[j] - t->GetV1()[Emax_i_j2]); v_DeltaTheta_j2.push_back(2*TMath::ATan(TMath::Power(TMath::E(),-1*t->GetV1()[j])) - 2*TMath::ATan(TMath::Power(TMath::E(),-1*t->GetV1()[Emax_i_j2]))); v_DeltaPhi_j2.push_back(t->GetV2()[j] - t->GetV2()[Emax_i_j2]); v_Energy_j2.push_back(t->GetV3()[j]); } } } //----------------------------------------------------------------------------------------------------------- gStyle->SetOptStat(0); double xmin = -1; double xmax = 1; double ymin = -0.5; double ymax = 0.5; std::string title = "isMaxJetEnergy = 1"; TCanvas *c1 = new TCanvas(); TH2D *h1 = new TH2D("h1",title.c_str(),40,xmin,xmax,40,ymin,ymax); for(int i = 0; (unsigned)i < v_DeltaEta.size(); i++) { h1->Fill(v_DeltaEta[i],v_DeltaPhi[i], v_Energy[i]/Nevent); } c1->SetLogz(); h1->Draw("colz"); // h1->SetMinimum(1); // h1->SetMaximum(1000); h1->GetXaxis()->SetTitle("#Delta#eta"); h1->GetYaxis()->SetTitle("#Delta#phi"); c1->Update(); //----------------------------------------------------------------------------------------------------------- std::string title2 = "isMaxJetEnergy = 2"; TCanvas *c2 = new TCanvas(); TH2D *h2 = new TH2D("h2",title2.c_str(),40,xmin,xmax,40,ymin,ymax); for(int i = 0; (unsigned)i < v_DeltaEta_j2.size(); i++) { h2->Fill(v_DeltaEta_j2[i],v_DeltaPhi_j2[i], v_Energy_j2[i]/Nevent); } c2->SetLogz(); h2->Draw("colz"); // h2->SetMinimum(1); // h2->SetMaximum(1000); h2->GetXaxis()->SetTitle("#Delta#eta"); h2->GetYaxis()->SetTitle("#Delta#phi"); c2->Update(); //----------------------------------------------------------------------------------------------------------- TCanvas *c3 = new TCanvas(); TH1D *h3 = h1->ProjectionY(); h3->SetLineColor(kGreen+3); TH1D *h4 = h2->ProjectionY(); h4->SetLineColor(kBlue); TF1 *f3 = new TF1("f3", "gaus", ymin, ymax); f3->SetLineColor(kGreen+3); h3->Fit("f3","Q R"); TF1 *f4 = new TF1("f4", "gaus", ymin, ymax); f4->SetLineColor(kBlue+2); h4->Fit("f4","Q R"); h3->Draw(); h4->Draw("SAME"); TLegend *leg = new TLegend(0.2,0.9,0.7,0.75); leg->SetBorderSize(1); leg->AddEntry(h3,title.c_str(),"l"); leg->AddEntry(h4,title2.c_str(),"l"); leg->Draw("SAME"); c3->Update(); cout << endl; cout << "*******************************************************" << endl; cout << "Ratio of Gaussian Width to Max Energy:" << endl; cout << " for: " << title << " -> " << f3->GetParameter(2) << " / " << h3->GetBinContent(h3->GetMaximumBin()) << " = " << f3->GetParameter(2) / h3->GetBinContent(h3->GetMaximumBin()) << endl; cout << " for: " << title2 << " -> " << f4->GetParameter(2) << " / " << h4->GetBinContent(h4->GetMaximumBin()) << " = " << f4->GetParameter(2) / h4->GetBinContent(h4->GetMaximumBin()) << endl; cout << " Ratio of ratios: " << (f3->GetParameter(2) / h3->GetBinContent(h3->GetMaximumBin())) / (f4->GetParameter(2) / h4->GetBinContent(h4->GetMaximumBin())) << endl; cout << "*******************************************************" << endl; cout << endl; c1->Close(); c2->Close(); c3->Close(); //----------------------------------------------------------------------------------------------------------- vector<double> v_Ratios; for(int i = 0; i < Nevent; i++) { TH1D *h5 = new TH1D("h5","",20,ymin*2,ymax*2); TH1D *h6 = new TH1D("h6","",20,ymin*2,ymax*2); for(int j = 0; j < Nentries/100; j++) { if((t->GetV4()[j]-1 == i) && (v_jet[j] == 1)) { h5->Fill(v_DeltaPhi[j], v_Energy[j]); } if((t->GetV4()[j]-1 == i) && (v_jet[j] == 2)) { h6->Fill(v_DeltaPhi_j2[j], v_Energy_j2[j]); } } TF1 *f5 = new TF1("f5", "gaus", ymin, ymax); f3->SetLineColor(kGreen+3); h5->Fit("f5","Q"); TF1 *f6 = new TF1("f6", "gaus", ymin, ymax); f6->SetLineColor(kBlue+2); h6->Fit("f6","Q"); TCanvas *c5 = new TCanvas(); h5->Draw(); TCanvas *c6 = new TCanvas(); h6->Draw(); // cout << endl; // cout << "Ratio of Gaussian Width to Max Energy:" << endl; // cout << " for: " << title << " -> " << f5->GetParameter(2) << " / " << h5->GetBinContent(h5->GetMaximumBin()) // << " = " << f5->GetParameter(2) / h5->GetBinContent(h5->GetMaximumBin()) << endl; // cout << " for: " << title2 << " -> " << f6->GetParameter(2) << " / " << h6->GetBinContent(h6->GetMaximumBin()) // << " = " << f6->GetParameter(2) / h6->GetBinContent(h6->GetMaximumBin()) << endl; cout << " Ratio of ratios for event " << i+1 << " : " << (f5->GetParameter(2) / h5->GetBinContent(h5->GetMaximumBin())) / (f6->GetParameter(2) / h6->GetBinContent(h6->GetMaximumBin())) << endl; if((f5->GetParameter(2) / h5->GetBinContent(h5->GetMaximumBin())) / (f6->GetParameter(2) / h6->GetBinContent(h6->GetMaximumBin())) != NAN) v_Ratios.push_back((f5->GetParameter(2) / h5->GetBinContent(h5->GetMaximumBin())) / (f6->GetParameter(2) / h6->GetBinContent(h6->GetMaximumBin()))); delete h5; delete h6; c5->Close(); c6->Close(); } TCanvas *c7 = new TCanvas(); TH1D *h7 = new TH1D("h7","",400,-200,200); for(int i = 0; (unsigned)i < v_Ratios.size(); i++) { h7->Fill(v_Ratios[i]); } h7->Draw(); return 0; }
int plot_track_multi() { /* Select input files (output from Fun4All) for plotting */ vector<string> v_filenames; v_filenames.push_back( "data/eRHIC_updated-magnets-2017_proton_275GeV_22mrad.root" ); v_filenames.push_back( "data/eRHIC_updated-magnets-2017_proton_275GeV_27mrad.root" ); v_filenames.push_back( "data/eRHIC_updated-magnets-2017_proton_275GeV_17mrad.root" ); TObjArray* graphs = new TObjArray(); for ( i = 0; i < v_filenames.size(); i++ ) { /* Open iput file with trajectories from GEANT4 */ TFile *fin = new TFile( v_filenames.at(i).c_str(), "OPEN" ); /* Get tree from file */ TTree *tin = (TTree*)fin->Get("T"); int nhits = 0; tin->SetBranchAddress("n_G4HIT_FWDDISC",&nhits); /* create graph of particle trajectory */ /* Use only first event (for now) */ tin->GetEntry(0); cout << "hits: " << nhits << endl; tin->Draw("G4HIT_FWDDISC.x:G4HIT_FWDDISC.z","Entry$==0",""); TGraph* g1 = new TGraph(nhits*2, &(tin->GetV2()[0]), &(tin->GetV1()[0])); g1->SetMarkerStyle(7); g1->SetMarkerSize(1); graphs->Add( g1 ); } /* Create frame histogram for plot */ TH1F *h1 = new TH1F("h1","",10,0,15000); h1->GetXaxis()->SetRangeUser(0,12000); h1->GetYaxis()->SetRangeUser(-50,200); h1->GetXaxis()->SetTitle("Z(cm)"); h1->GetYaxis()->SetTitle("X(cm)"); /* Plot frame histogram */ TCanvas *c1 = new TCanvas(); h1->Draw("AXIS"); /* Read IR configuration file- this needs to go somewhere else using parameters and a .root file to store them */ string irfile = "data/updated-magnets-2017.dat"; ifstream irstream(irfile.c_str()); while(!irstream.eof()){ string str; getline(irstream, str); if(str[0] == '#') continue; //for comments stringstream ss(str); string name; double center_z, center_x, center_y, aperture_radius, length, angle, B, gradient; ss >> name >> center_z >> center_x >> center_y >> aperture_radius >> length >> angle >> B >> gradient; if ( name == "" ) continue; //for empty lines // convert units from m to cm center_x *= 100; center_y *= 100; center_z *= 100; aperture_radius *= 100; length *= 100; // define magnet outer radius float outer_radius = 30.0; // cm //flip sign of dipole field component- positive y axis in Geant4 is defined as 'up', // positive z axis as the hadron-going direction // in a right-handed coordinate system x,y,z B *= -1; // convert angle from millirad to rad angle = (angle / 1000.); // Place IR component cout << "New IR component: " << name << " at z = " << center_z << endl; string volname = "IRMAGNET_"; volname.append(name); /* Draw box for magnet position on canvas */ TPolyLine *b1 = TraceBox( angle, center_z, center_x, length, aperture_radius, outer_radius ); //upper box TPolyLine *b2 = TraceBox( angle, center_z, center_x, length, -1 * aperture_radius, -1 * outer_radius ); //lower box if(B != 0 && gradient == 0.0){ //dipole magnet b1->SetFillColor(kOrange+1); b2->SetFillColor(kOrange+1); } else if( B == 0 && gradient != 0.0){ //quad magnet b1->SetFillColor(kBlue+1); b2->SetFillColor(kBlue+1); } else{ //placeholder magnet b1->SetFillColor(kGray+1); b2->SetFillColor(kGray+1); } b1->Draw("Fsame"); b2->Draw("Fsame"); } /* draw particle trajectory */ for ( int i = 0; i < graphs->GetEntries(); i++ ) { graphs->At(i)->Draw("LPsame"); } c1->Print("multitrack_new.eps"); return 0; }
int TrackParametrization( TString csvfile="fitslices_out.csv" ) { /* Read data from input file */ TTree *tres = new TTree(); tres->ReadFile( csvfile, "ptrue:etatrue:psig:psig_err:pmean:pmean_err:norm", ',' ); /* Print read-in tree */ tres->Print(); /* colors array */ unsigned colors[8] = {1,2,3,4,6,7,14,16}; /* Create vector of theta values to include for visualization*/ vector< double > etas_vis; etas_vis.push_back(-2.75); etas_vis.push_back(-2.25); etas_vis.push_back(-1.75); etas_vis.push_back(-0.25); etas_vis.push_back( 0.25); etas_vis.push_back( 1.75); etas_vis.push_back( 2.25); // etas_vis.push_back(-3.25); // etas_vis.push_back(-2.25); // etas_vis.push_back(-1.25); // etas_vis.push_back(-0.25); // etas_vis.push_back( 0.25); // etas_vis.push_back( 1.25); // etas_vis.push_back( 2.25); // etas_vis.push_back( 3.25); /* Create vector of theta values to include for fitting*/ vector< double > etas_fit; for ( double eta = -4.45; eta < 4.5; eta += 0.1 ) etas_fit.push_back( eta ); /* Create fit function */ TF1* f_momres = new TF1("f_momres", "sqrt( [0]*[0] + [1]*[1]*x*x )" ); cout << "\nFit function: " << f_momres->GetTitle() << "\n" << endl; /* Create scratch canvas */ TCanvas *cscratch = new TCanvas("cscratch"); /* Create framehistogram */ TH1F* hframe = new TH1F("hframe","",100,0,40); hframe->GetYaxis()->SetRangeUser(0,0.15); hframe->GetYaxis()->SetNdivisions(505); hframe->GetXaxis()->SetTitle("Momentum (GeV/c)"); hframe->GetYaxis()->SetTitle("#sigma_{p}/p"); /* create combined canvas plot */ TCanvas *c1 = new TCanvas(); hframe->Draw(); /* Create legend */ TLegend* leg_eta = new TLegend( 0.2, 0.6, 0.5, 0.9); leg_eta->SetNColumns(2); /* Create ofstream to write fit parameter results */ ofstream ofsfit("track_momres_new.csv"); ofsfit<<"eta,par1,par1err,par2,par2err"<<endl; /* Create resolution-vs-momentum plot with fits for each selected theta value */ for ( int i = 0; i < etas_fit.size(); i++ ) { /* Switch to scratch canvas */ cscratch->cd(); double eta = etas_fit.at(i); /* No tracking outside -4 < eta < 4 */ if ( eta < -4 || eta > 4 ) continue; cout << "\n***Eta = " << eta << endl; /* Define range of theta because float comparison with fixed value doesn't work too well for cuts in ROOT trees */ double eta_min = eta * 0.999; double eta_max = eta * 1.001; /* Cut for tree */ TCut cutx( Form("ptrue > 1 && ( (etatrue > 0 && (etatrue > %f && etatrue < %f)) || (etatrue < 0 && (etatrue < %f && etatrue > %f)) )", eta_min, eta_max, eta_min, eta_max) ); /* "Draw" tree on scratch canvas to fill V1...V4 arrays */ tres->Draw("psig:ptrue:psig_err:0", cutx ); /* Create TGraphErrors with selected data from tree */ TGraphErrors *gres = new TGraphErrors( tres->GetEntries(cutx), &(tres->GetV2())[0], &(tres->GetV1())[0], &(tres->GetV4())[0], &(tres->GetV3())[0] ); /* reset function parameters before fit */ f_momres->SetParameter(0,0.1); f_momres->SetParameter(1,0.1); /* Only plot pseudorapidities listed on etas_vis; if not plotting, still do the fit */ bool vis = false; int vi = 0; for ( vi = 0; vi < etas_vis.size(); vi++ ) { if ( abs( etas_vis.at(vi) - eta ) < 0.001 ) { vis = true; break; } } if ( vis ) { /* Add graph to legend */ leg_eta->AddEntry(gres, Form("#eta = %.1f", eta), "P"); /* Add graph to plot */ c1->cd(); gres->SetMarkerColor(colors[vi]); gres->Draw("Psame"); f_momres->SetLineColor(colors[vi]); gres->Fit(f_momres); } else { gres->Fit(f_momres); } /* Write fir results to file */ double par1 = f_momres->GetParameter(0); double par1err = f_momres->GetParError(0); double par2 = f_momres->GetParameter(1); double par2err = f_momres->GetParError(1); ofsfit << eta << "," << par1 << "," << par1err << "," << par2 << "," << par2err << endl; } /* Draw legend */ c1->cd(); //TCanvas *c2 = new TCanvas(); //hframe->Draw(); leg_eta->Draw(); /* Print plots */ c1->Print("track_momres_vareta.eps"); //c2->Print("track_momres_vareta_legend.eps"); /* Close output stream */ ofsfit.close(); return 0; }