int main( int argc, char* argv[] ) {
runName = "test_10";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
std::string fileName = "PosAn_" + runName + ".root";
TFile* file = TFile::Open( fileName.c_str() );
std::cout << "-> Opened file: " << fileName << std::endl;
std::string outputdir = "Plots_" + runName;
std::string mkdir_command = "mkdir -p " + outputdir;
system(mkdir_command.c_str());
TStyle* style = DrawTools::setStyle();
style->cd();
drawSinglePositionPlot( outputdir, file, runName, "" );
drawSinglePositionPlot( outputdir, file, runName, "_singleEle" );
//drawSinglePlot( outputdir, "cef3_spectrum" , file, "cef3" , "ADC Counts", 4, 0., 3500., 10, true );
//drawSinglePlot( outputdir, "cef3_corr_spectrum" , file, "cef3_corr", "ADC Counts", 4, 0., 3500., 10, true );
drawSinglePlot( outputdir, "cef3_spectrum_lin" , file, "cef3" , "ADC Counts", 4, 0., 3500., 10, false );
drawSinglePlot( outputdir, "cef3_corr_spectrum_lin" , file, "cef3_corr", "ADC Counts", 4, 0., 3500., 10, false );
return 0;
}
int main( int argc, char* argv[] ) {
std::string runName = "BTF_92_20140430-020137_beam_uncalib";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
TFile* file = TFile::Open(Form("PosAn_%s.root", runName.c_str()) );
TTree* tree = (TTree*)file->Get("tree_passedEvents");
std::string outputdir = "CeF3_Calibration_" + runName;
std::string mkdir_command = "mkdir -p " + outputdir;
system( mkdir_command.c_str() );
std::string ofsName = outputdir + "/constants.txt";
ofstream ofs(ofsName.c_str());
std::vector<float> calibConstants;
for( unsigned i=0; i<4; ++i ) {
TF1* f1 = fitSingleElectronPeak( outputdir, i, tree );
float mean = f1->GetParameter(1);
float sigma = f1->GetParameter(2);
calibConstants.push_back(sigma/mean);
std::cout << std::endl;
std::cout << "Channel " << i << std::endl;
std::cout << " Mean : " << mean << std::endl;
std::cout << " Sigma : " << sigma << std::endl;
std::cout << " Resolution : " << sigma/mean << std::endl;
ofs << i << "\t" << mean << std::endl;
}
ofs.close();
std::cout << "-> Saved constants in: " << ofsName << std::endl;
checkTotalResolution( outputdir, tree );
return 0;
}
int main( int argc, char* argv[] ) {
DrawTools::setStyle();
if( argc<2 ) {
std::cout << "ERROR. You need to specify the name of the run you want to process." << std::endl;
exit(1);
}
std::string runName = "";
std::string tag = "V00";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
if( argc>2 ) {
std::string tag_str(argv[2]);
tag = tag_str;
}
} else {
std::cout << "Usage:" << std::endl;
std::cout << "./makeAnalysisTree [runName] ([tag]) ([config])" << std::endl;
exit(12345);
}
std::string fileName = "./analysisTrees_"+tag+"/Reco_" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
if( file==0 ) {
std::cout << "ERROR! Din't find file " << fileName << std::endl;
std::cout << "Exiting." << std::endl;
exit(11);
}
TTree* tree = (TTree*)file->Get("recoTree");
TCanvas c1("dummy");
tree->Draw("(cef3_time_at_frac50[1]-mcp_time_frac50):cef3_maxAmpl[1]>>h","cef3_time_at_thresh[1]>0 && cef3_time_at_thresh[1]<50 && mcp_max_amplitude>300 && cef3_maxAmpl[1]>10 && TMath::Abs(cef3_time_at_frac50[1]-mcp_time_frac50)<20","colz");
tree->Draw("(cef3_time_at_frac50[1]-mcp_time_frac50):cef3_maxAmpl[1]>>hprof","cef3_time_at_thresh[1]>0 && cef3_time_at_thresh[1]<50 && mcp_max_amplitude>300 && cef3_maxAmpl[1]>10 && TMath::Abs(cef3_time_at_frac50[1]-mcp_time_frac50)<20","profXsamep");
c1.SaveAs("stocazzo.png");
}
int main( int argc, char* argv[] ) {
DrawTools::setStyle();
TFile *inputFile;
std::string runName = "";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
std::string filename_str = "dcap://t3se01.psi.ch:22125//pnfs/psi.ch/cms/trivcat/store/user/micheli/rawData/output_run" + runName + ".root";
char* hostName = getenv ("HOSTNAME");
TString hostString(hostName);
bool isLxplus = hostString.Contains("lxplus");
if(isLxplus) filename_str = "xroot://t3se01.psi.ch:1094//store/user/micheli/rawData/output_run" + runName + ".root";
inputFile=TFile::Open(filename_str.c_str());
if( inputFile==0 ) {
std::cout << "ERROR! Din't find file " << filename_str << std::endl;
std::cout << "Exiting." << std::endl;
exit(11);
}
}
TTree* inputTree=(TTree*)inputFile->Get("outputTree");
WaveformUtil t(inputTree);
t.Loop();
return 0;
}
int main( int argc, char* argv[] ) {
runName = "test_10";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
std::string tag = "V02";
if( argc>2 ) {
std::string tag_str(argv[2]);
tag = tag_str;
}
std::string fileName = "PosAnTrees_" + tag + "/PosAn_" + runName + ".root";
TFile* file = TFile::Open( fileName.c_str() );
std::cout << "-> Opened file: " << fileName << std::endl;
std::string outputdir = "Plots_" + runName + "_" + tag;
std::string mkdir_command = "mkdir -p " + outputdir;
system(mkdir_command.c_str());
TStyle* style = DrawTools::setStyle();
style->cd();
drawSinglePositionPlot( outputdir, file, runName, "" );
drawSinglePositionPlot( outputdir, file, runName, "_singleEle" );
drawPositionResolutionXY( outputdir, file, runName, "bgo", "Beam" );
return 0;
}
int main( int argc, char* argv[] ) {
DrawTools::setStyle();
std::string runName = "precalib_BGO_pedestal_noSource";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
NORDERS = 6;
if( argc>2 ) {
NORDERS = atoi(argv[2]);
std::cout << "-> NORDER is set to: " << NORDERS << std::endl;
}
std::string fileName = "../PositionAnalysis/data/run_" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
if( file==0 ) {
std::cout << "ERROR! Din't find file " << fileName << std::endl;
std::cout << "Exiting." << std::endl;
exit(11);
}
TTree* tree = (TTree*)file->Get("eventRawData");
UInt_t evtNumber;
tree->SetBranchAddress( "evtNumber", &evtNumber );
UInt_t adcData[40];
tree->SetBranchAddress( "adcData", adcData );
UInt_t adcBoard[40];
tree->SetBranchAddress( "adcBoard", adcBoard );
UInt_t adcChannel[40];
tree->SetBranchAddress( "adcChannel", adcChannel );
int nentries = tree->GetEntries();
std::string outfileName = "calibAn_" + runName + ".root";
TFile* outFile = TFile::Open( outfileName.c_str(), "RECREATE" );
TH1D* h1_cef3_0 = new TH1D("cef3_0", "", 400, 0., 400.);
TH1D* h1_cef3_1 = new TH1D("cef3_1", "", 400, 0., 400.);
TH1D* h1_cef3_2 = new TH1D("cef3_2", "", 400, 0., 400.);
TH1D* h1_cef3_3 = new TH1D("cef3_3", "", 400, 0., 400.);
TH1D* h1_cef3_tot = new TH1D("cef3_tot", "", 400, 0., 4.*400.);
TH1D* h1_cef3_corr_0 = new TH1D("cef3_corr_0", "", 400, 0., 400.);
TH1D* h1_cef3_corr_1 = new TH1D("cef3_corr_1", "", 385, 0., 400.4);
TH1D* h1_cef3_corr_2 = new TH1D("cef3_corr_2", "", 400, 0., 404.);
TH1D* h1_cef3_corr_3 = new TH1D("cef3_corr_3", "", 400, 0., 400.);
TH1D* h1_cef3_corr_tot = new TH1D("cef3_corr_tot", "", 400, 0., 4.*400.);
TH1D* h1_cef3_pedSubtracted_0 = new TH1D("cef3_pedSubtracted_0", "", 400, 0., 400.);
TH1D* h1_cef3_pedSubtracted_1 = new TH1D("cef3_pedSubtracted_1", "", 400, 0., 400.);
TH1D* h1_cef3_pedSubtracted_2 = new TH1D("cef3_pedSubtracted_2", "", 400, 0., 400.);
TH1D* h1_cef3_pedSubtracted_3 = new TH1D("cef3_pedSubtracted_3", "", 400, 0., 400.);
TH1D* h1_cef3_pedSubtracted_corr_0 = new TH1D("cef3_pedSubtracted_corr_0", "", 400, 0., 400.*1.11228);
TH1D* h1_cef3_pedSubtracted_corr_1 = new TH1D("cef3_pedSubtracted_corr_1", "", 400, 0., 400.*0.855333);
TH1D* h1_cef3_pedSubtracted_corr_2 = new TH1D("cef3_pedSubtracted_corr_2", "", 400, 0., 400.*0.97973);
TH1D* h1_cef3_pedSubtracted_corr_3 = new TH1D("cef3_pedSubtracted_corr_3", "", 400, 0., 400.*1.08781);
TH1D* h1_cef3_pedSubtracted_corr_sum = new TH1D("cef3_pedSubtracted_sum", "", 400, 0., 400.);
TH1D* h1_cef3_pedSubtracted_corr_sum_dummy = new TH1D("cef3_pedSubtracted_sum", "", 400, 0., 400.);
TH1D* h1_cef3_pedSubtracted_corr_rebin_0 = new TH1D("cef3_pedSubtracted_corr_rebin_0", "", 200, 0., 400.*1.11228);
TH1D* h1_cef3_pedSubtracted_corr_rebin_1 = new TH1D("cef3_pedSubtracted_corr_rebin_1", "", 200, 0., 400.*0.855333);
TH1D* h1_cef3_pedSubtracted_corr_rebin_2 = new TH1D("cef3_pedSubtracted_corr_rebin_2", "", 200, 0., 400.*0.97973);
TH1D* h1_cef3_pedSubtracted_corr_rebin_3 = new TH1D("cef3_pedSubtracted_corr_rebin_3", "", 200, 0., 400.*1.08781);
TH1D* h1_cef3_pedSubtracted_corr_muQ_0 = new TH1D("cef3_pedSubtracted_corr_muQ_0", "", 400, 0., 400.*1.19514);
TH1D* h1_cef3_pedSubtracted_corr_muQ_1 = new TH1D("cef3_pedSubtracted_corr_muQ_1", "", 400, 0., 400.*0.860177);
TH1D* h1_cef3_pedSubtracted_corr_muQ_2 = new TH1D("cef3_pedSubtracted_corr_muQ_2", "", 400, 0., 400.*0.952363);
TH1D* h1_cef3_pedSubtracted_corr_muQ_3 = new TH1D("cef3_pedSubtracted_corr_muQ_3", "", 400, 0., 400.*0.977169);
TH1D* h1_cef3_pedSubtracted_corr_muMean_0 = new TH1D("cef3_pedSubtracted_corr_muMean_0", "", 400, 0., 400.*1.09716);
TH1D* h1_cef3_pedSubtracted_corr_muMean_1 = new TH1D("cef3_pedSubtracted_corr_muMean_1", "", 400, 0., 400.*0.843707);
TH1D* h1_cef3_pedSubtracted_corr_muMean_2 = new TH1D("cef3_pedSubtracted_corr_muMean_2", "", 400, 0., 400.*0.966413);
TH1D* h1_cef3_pedSubtracted_corr_muMean_3 = new TH1D("cef3_pedSubtracted_corr_muMean_3", "", 400, 0., 400.*1.07241);
// there is only one cosmic run
unsigned int runNumber_;
runNumber_=91;
std::string pedestalFileName = "../PositionAnalysis/pedestalFile.root";
std::vector<std::pair<float, float> > pedestals = getPedestals( "cef3", pedestalFileName, runNumber_ );
std::cout << std::endl;
std::cout << "-> Got pedestals of CeF3: " << std::endl;
for( unsigned i=0; i<CEF3_CHANNELS; ++i )
std::cout << " CeF3 Channel " << i << ": " << pedestals[i].first << " (+- " << pedestals[i].second << ")" << std::endl;
std::cout << std::endl;
int nSigma=4;
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
tree->GetEntry(iEntry);
if( iEntry % 5000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
for( unsigned i=0; i<40; ++i ) {
int board = adcBoard[i];
int channel= adcChannel[i];
float cef3=0;
if( board==CEF3_ADC_BOARD ) {
if( channel==(CEF3_ADC_START_CHANNEL ) ){
h1_cef3_0->Fill(adcData[i]);
cef3+=adcData[i];
if(adcData[i]>(pedestals[0].first + nSigma*pedestals[0].second)) h1_cef3_pedSubtracted_0->Fill(adcData[i]-pedestals[0].first);
}
else if( channel==(CEF3_ADC_START_CHANNEL+1) ){
h1_cef3_1->Fill(adcData[i]);
cef3+=adcData[i];
if(adcData[i]>(pedestals[1].first + nSigma*pedestals[1].second))h1_cef3_pedSubtracted_1->Fill(adcData[i]-pedestals[1].first);
}
else if( channel==(CEF3_ADC_START_CHANNEL+2) ) {
h1_cef3_2->Fill(adcData[i]);
cef3+=adcData[i];
if(adcData[i]>(pedestals[2].first + nSigma*pedestals[2].second))h1_cef3_pedSubtracted_2->Fill(adcData[i]-pedestals[2].first);
}
else if( channel==(CEF3_ADC_START_CHANNEL+3) ) {
h1_cef3_3->Fill(adcData[i]);
cef3+=adcData[i];
if(adcData[i]>(pedestals[3].first + nSigma*pedestals[3].second))h1_cef3_pedSubtracted_3->Fill(adcData[i]-pedestals[3].first);
}
h1_cef3_tot->Fill(cef3);
}
}
}
h1_cef3_0->SetLineWidth(2);
h1_cef3_1->SetLineWidth(2);
h1_cef3_2->SetLineWidth(2);
h1_cef3_3->SetLineWidth(2);
h1_cef3_0->SetLineColor(kBlack);
h1_cef3_1->SetLineColor(kRed);
h1_cef3_2->SetLineColor(kBlue);
h1_cef3_3->SetLineColor(kMagenta);
h1_cef3_corr_0->SetLineWidth(2);
h1_cef3_corr_1->SetLineWidth(2);
h1_cef3_corr_2->SetLineWidth(2);
h1_cef3_corr_3->SetLineWidth(2);
h1_cef3_corr_0->SetLineColor(kBlack);
h1_cef3_corr_1->SetLineColor(kRed);
h1_cef3_corr_2->SetLineColor(kBlue);
h1_cef3_corr_3->SetLineColor(kMagenta);
h1_cef3_pedSubtracted_0->SetLineWidth(2);
h1_cef3_pedSubtracted_1->SetLineWidth(2);
h1_cef3_pedSubtracted_2->SetLineWidth(2);
h1_cef3_pedSubtracted_3->SetLineWidth(2);
h1_cef3_pedSubtracted_0->SetLineColor(kBlack);
h1_cef3_pedSubtracted_1->SetLineColor(kRed);
h1_cef3_pedSubtracted_2->SetLineColor(kBlue);
h1_cef3_pedSubtracted_3->SetLineColor(kMagenta);
h1_cef3_pedSubtracted_corr_0->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_1->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_2->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_3->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_0->SetLineColor(kBlack);
h1_cef3_pedSubtracted_corr_1->SetLineColor(kRed);
h1_cef3_pedSubtracted_corr_2->SetLineColor(kBlue);
h1_cef3_pedSubtracted_corr_3->SetLineColor(kMagenta);
h1_cef3_pedSubtracted_corr_muQ_0->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muQ_1->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muQ_2->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muQ_3->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muQ_0->SetLineColor(kBlack);
h1_cef3_pedSubtracted_corr_muQ_1->SetLineColor(kRed);
h1_cef3_pedSubtracted_corr_muQ_2->SetLineColor(kBlue);
h1_cef3_pedSubtracted_corr_muQ_3->SetLineColor(kMagenta);
h1_cef3_pedSubtracted_corr_muMean_0->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muMean_1->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muMean_2->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muMean_3->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_muMean_0->SetLineColor(kBlack);
h1_cef3_pedSubtracted_corr_muMean_1->SetLineColor(kRed);
h1_cef3_pedSubtracted_corr_muMean_2->SetLineColor(kBlue);
h1_cef3_pedSubtracted_corr_muMean_3->SetLineColor(kMagenta);
//plot uncorr energy (before intercalibration)
TCanvas* cuncorr = new TCanvas( "cuncorr", "", 600, 600 );
cuncorr->cd();
cuncorr->SetLogy();
TH1D* histo_axes = new TH1D("cef3_0", "", 400, 0., 350.);
histo_axes->GetXaxis()->SetRangeUser(80.,350.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_0->GetMaximum()+h1_cef3_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
TLegend* legend = new TLegend( 0.7, 0.7, 0.90, 0.9 );
legend->SetLineColor(0);
legend->SetFillColor(0);
legend->SetFillStyle(0);
legend->SetTextSize(0.038);
legend->AddEntry( h1_cef3_0, "channel 0", "L" );
legend->AddEntry( h1_cef3_1, "channel 1", "L" );
legend->AddEntry( h1_cef3_2, "channel 2", "L" );
legend->AddEntry( h1_cef3_3, "channel 3", "L" );
legend->Draw("same");
h1_cef3_0->Draw("same");
h1_cef3_1->Draw("same");
h1_cef3_2->Draw("same");
h1_cef3_3->Draw("same");
cuncorr->SaveAs("uncorrEnergyAllChannels.png");
cuncorr->SaveAs("uncorrEnergyAllChannels.eps");
TCanvas* cuncorr_2 = new TCanvas( "cuncorr_2", "", 600, 600 );
cuncorr_2->cd();
cuncorr_2->SetLogy();
histo_axes->GetXaxis()->SetRangeUser(120.,200.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_0->GetMaximum()+h1_cef3_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
h1_cef3_0->Draw("same");
h1_cef3_1->Draw("same");
h1_cef3_2->Draw("same");
h1_cef3_3->Draw("same");
legend->Draw("same");
cuncorr_2->SaveAs("uncorrEnergyAllChannels_zoom.png");
cuncorr_2->SaveAs("uncorrEnergyAllChannels_zoom.eps");
//plot uncorr energy (before intercalibration) for pedSubtracted
TCanvas* cuncorr_pedSubtracted = new TCanvas( "cuncorr_pedSubtracted", "", 600, 600 );
cuncorr_pedSubtracted->cd();
cuncorr_pedSubtracted->SetLogy();
histo_axes->GetXaxis()->SetRangeUser(0.,250.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
legend->Draw("same");
h1_cef3_pedSubtracted_0->Draw("same");
h1_cef3_pedSubtracted_1->Draw("same");
h1_cef3_pedSubtracted_2->Draw("same");
h1_cef3_pedSubtracted_3->Draw("same");
cuncorr_pedSubtracted->SaveAs("uncorrEnergyAllChannelspedSubtracted.png");
cuncorr_pedSubtracted->SaveAs("uncorrEnergyAllChannelspedSubtracted.eps");
TCanvas* cuncorr_pedSubtracted_2 = new TCanvas( "cuncorr_pedSubtracted_2", "", 600, 600 );
cuncorr_pedSubtracted_2->cd();
cuncorr_pedSubtracted_2->SetLogy();
histo_axes->GetXaxis()->SetRangeUser(20.,100.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
h1_cef3_pedSubtracted_0->Draw("same");
h1_cef3_pedSubtracted_1->Draw("same");
h1_cef3_pedSubtracted_2->Draw("same");
h1_cef3_pedSubtracted_3->Draw("same");
legend->Draw("same");
cuncorr_pedSubtracted_2->SaveAs("uncorrEnergyAllChannelspedSubtracted_zoom.png");
cuncorr_pedSubtracted_2->SaveAs("uncorrEnergyAllChannelspedSubtracted_zoom.eps");
//intercalibration of single fibers with photoelectrons
FitResults fr_0 = fitSingleHisto( h1_cef3_0, 110., 135., 138., 185. );
FitResults fr_1 = fitSingleHisto( h1_cef3_1, 100., 125., 125., 190. );
FitResults fr_2 = fitSingleHisto( h1_cef3_2, 100., 125., 128., 190. );
FitResults fr_3 = fitSingleHisto( h1_cef3_3, 100., 125., 137., 198. );
std::vector<float> lowerFitBoundary;
lowerFitBoundary.push_back(137.);
lowerFitBoundary.push_back(125.);
lowerFitBoundary.push_back(127.);
lowerFitBoundary.push_back(137.);
FitResults fr_pedSubtracted_0 = fitSingleHisto( h1_cef3_pedSubtracted_0, 0., 0., lowerFitBoundary[0]-pedestals[0].first, 185.-pedestals[0].first );
FitResults fr_pedSubtracted_1 = fitSingleHisto( h1_cef3_pedSubtracted_1, 0., 0., lowerFitBoundary[1]-pedestals[1].first, 190.-pedestals[1].first );
FitResults fr_pedSubtracted_2 = fitSingleHisto( h1_cef3_pedSubtracted_2, 0., 0., lowerFitBoundary[2]-pedestals[2].first, 190.-pedestals[2].first );
FitResults fr_pedSubtracted_3 = fitSingleHisto( h1_cef3_pedSubtracted_3, 0., 0., lowerFitBoundary[3]-pedestals[3].first, 198.-pedestals[3].first );
std::vector<float> correctionFactors = intercalibrateFibers(fr_0,fr_1,fr_2,fr_3,true);
std::vector<float> correctionFactors_pedSubtracted = intercalibrateFibers(fr_pedSubtracted_0,fr_pedSubtracted_1,fr_pedSubtracted_2,fr_pedSubtracted_3,true);
std::vector<float> correctionFactors_pedSubtracted_muQ = intercalibrateFibers(fr_pedSubtracted_0,fr_pedSubtracted_1,fr_pedSubtracted_2,fr_pedSubtracted_3,false);
std::vector<float> mu;
mu.push_back(fr_0.mu);
mu.push_back(fr_1.mu);
mu.push_back(fr_2.mu);
mu.push_back(fr_3.mu);
float mumean=0.;
for(int i=0;i<4;i++){
mumean+=mu[i];
}
mumean=mumean/4.;
std::vector<float> correctionFactors_pedSubtracted_muMean= intercalibrateFibers(fr_pedSubtracted_0,fr_pedSubtracted_1,fr_pedSubtracted_2,fr_pedSubtracted_3,false,mumean);
for(int i=0;i<4;i++){
std::cout<<"correctionFactors "<<correctionFactors[i]<<std::endl;
std::cout<<"correctionFactors pedSub "<<correctionFactors_pedSubtracted[i]<<std::endl;
std::cout<<"correctionFactors pedSub muQ "<<correctionFactors_pedSubtracted_muQ[i]<<std::endl;
std::cout<<"correctionFactors pedSub muMean "<<correctionFactors_pedSubtracted_muMean[i]<<std::endl;
}
TRandom a;
a.SetSeed(100);
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
tree->GetEntry(iEntry);
if( iEntry % 5000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
for( unsigned i=0; i<40; ++i ) {
int board = adcBoard[i];
int channel= adcChannel[i];
float cef3_corr=0;
if( board==CEF3_ADC_BOARD ) {
if( channel==(CEF3_ADC_START_CHANNEL ) ){
h1_cef3_corr_0->Fill(adcData[i]*correctionFactors[0]);
if(adcData[i]>(pedestals[0].first + nSigma*pedestals[0].second)){
h1_cef3_pedSubtracted_corr_0->Fill((adcData[i]-pedestals[0].first)*correctionFactors_pedSubtracted[0]);
h1_cef3_pedSubtracted_corr_rebin_0->Fill((adcData[i]-pedestals[0].first)*correctionFactors_pedSubtracted[0]);
h1_cef3_pedSubtracted_corr_muQ_0->Fill((adcData[i]-pedestals[0].first)*correctionFactors_pedSubtracted_muQ[0]);
h1_cef3_pedSubtracted_corr_muMean_0->Fill((adcData[i]-pedestals[0].first)*correctionFactors_pedSubtracted_muMean[0]);
h1_cef3_pedSubtracted_corr_sum->Fill(a.Uniform(adcData[i]-pedestals[0].first-0.5,adcData[i]-pedestals[0].first+0.5)*correctionFactors_pedSubtracted[0]);
}
cef3_corr+=adcData[i]*correctionFactors[0];
}
else if( channel==(CEF3_ADC_START_CHANNEL+1) ){
h1_cef3_corr_1->Fill(adcData[i]*correctionFactors[1]);
if(adcData[i]>(pedestals[1].first + nSigma*pedestals[1].second)) {
h1_cef3_pedSubtracted_corr_1->Fill((adcData[i]-pedestals[1].first)*correctionFactors_pedSubtracted[1]);
h1_cef3_pedSubtracted_corr_rebin_1->Fill((adcData[i]-pedestals[1].first)*correctionFactors_pedSubtracted[1]);
h1_cef3_pedSubtracted_corr_muQ_1->Fill((adcData[i]-pedestals[1].first)*correctionFactors_pedSubtracted_muQ[1]);
h1_cef3_pedSubtracted_corr_muMean_1->Fill((adcData[i]-pedestals[1].first)*correctionFactors_pedSubtracted_muMean[1]);
h1_cef3_pedSubtracted_corr_sum->Fill(a.Uniform(adcData[i]-pedestals[1].first-0.5,adcData[i]-pedestals[1].first+0.5)*correctionFactors_pedSubtracted[1]);
}
cef3_corr+=adcData[i]*correctionFactors[1];
}
else if( channel==(CEF3_ADC_START_CHANNEL+2) ) {
h1_cef3_corr_2->Fill(adcData[i]*correctionFactors[2]);
if(adcData[i]>(pedestals[2].first + nSigma*pedestals[2].second)) {
h1_cef3_pedSubtracted_corr_2->Fill((adcData[i]-pedestals[2].first)*correctionFactors_pedSubtracted[2]);
h1_cef3_pedSubtracted_corr_rebin_2->Fill((adcData[i]-pedestals[2].first)*correctionFactors_pedSubtracted[2]);
h1_cef3_pedSubtracted_corr_muQ_2->Fill((adcData[i]-pedestals[2].first)*correctionFactors_pedSubtracted_muQ[2]);
h1_cef3_pedSubtracted_corr_muMean_2->Fill((adcData[i]-pedestals[2].first)*correctionFactors_pedSubtracted_muMean[2]);
h1_cef3_pedSubtracted_corr_sum->Fill(a.Uniform(adcData[i]-pedestals[2].first-0.5,adcData[i]-pedestals[2].first+0.5)*correctionFactors_pedSubtracted[2]);
}
cef3_corr+=adcData[i]*correctionFactors[2];
}
else if( channel==(CEF3_ADC_START_CHANNEL+3) ) {
h1_cef3_corr_3->Fill(adcData[i]*correctionFactors[3]);
if(adcData[i]>(pedestals[3].first + nSigma*pedestals[3].second)){
h1_cef3_pedSubtracted_corr_3->Fill((adcData[i]-pedestals[3].first)*correctionFactors_pedSubtracted[3]);
h1_cef3_pedSubtracted_corr_rebin_3->Fill((adcData[i]-pedestals[3].first)*correctionFactors_pedSubtracted[3]);
h1_cef3_pedSubtracted_corr_muQ_3->Fill((adcData[i]-pedestals[3].first)*correctionFactors_pedSubtracted_muQ[3]);
h1_cef3_pedSubtracted_corr_muMean_3->Fill((adcData[i]-pedestals[3].first)*correctionFactors_pedSubtracted_muMean[3]);
h1_cef3_pedSubtracted_corr_sum->Fill(a.Uniform(adcData[i]-pedestals[3].first-0.5,adcData[i]-pedestals[3].first+0.5)*correctionFactors_pedSubtracted[3]);
}
cef3_corr+=adcData[i]*correctionFactors[3];
}
h1_cef3_tot->Fill(cef3_corr);
}
}
}
//plot corr energy (after intercalibration)
TCanvas* ccorr = new TCanvas( "ccorr", "", 600, 600 );
ccorr->cd();
ccorr->SetLogy();
TH2D* h2_axes_corr = new TH2D("axes_corr", "", 10, 0., 350., 10, 10., 1.1*h1_cef3_corr_0->GetMaximum() );
h2_axes_corr->GetXaxis()->SetRangeUser(80.,350);
h2_axes_corr->SetXTitle( "ADC Counts" );
h2_axes_corr->Draw();
h1_cef3_corr_0->Draw("same");
h1_cef3_corr_1->Draw("same");
h1_cef3_corr_2->Draw("same");
h1_cef3_corr_3->Draw("same");
legend->Draw("same");
ccorr->SaveAs("corrEnergyAllChannels.png");
ccorr->SaveAs("corrEnergyAllChannels.eps");
//pedsubtracted plot
ccorr->Clear();
histo_axes->GetXaxis()->SetRangeUser(0.,250.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
histo_axes->Draw();
h1_cef3_pedSubtracted_corr_0->Draw("same");
h1_cef3_pedSubtracted_corr_1->Draw("same");
h1_cef3_pedSubtracted_corr_2->Draw("same");
h1_cef3_pedSubtracted_corr_3->Draw("same");
legend->Draw("same");
ccorr->SaveAs("corrEnergyAllChannelspedSubtracted.png");
ccorr->SaveAs("corrEnergyAllChannelspedSubtracted.eps");
h1_cef3_pedSubtracted_corr_sum->Rebin(2);
FitResults fr_corr_sum = fitSingleHisto_sum( h1_cef3_pedSubtracted_corr_sum, 0., 0.,13.,68.,true );
//plot for the paper. all corr channels sum and fit
ccorr->Clear();
ccorr->SetLogy(0);
ccorr->cd();
h1_cef3_pedSubtracted_corr_sum->SetLineWidth(2);
h1_cef3_pedSubtracted_corr_sum->GetXaxis()->SetRangeUser(0.,150.);
h1_cef3_pedSubtracted_corr_sum->GetYaxis()->SetRangeUser(50.,h1_cef3_pedSubtracted_corr_sum->GetMaximum()+h1_cef3_pedSubtracted_corr_sum->GetMaximum()*0.10);
h1_cef3_pedSubtracted_corr_sum->SetYTitle( "Events / 2" );
h1_cef3_pedSubtracted_corr_sum->SetXTitle( "ADC Counts" );
h1_cef3_pedSubtracted_corr_sum->Draw();
// ccorr->SetLogy(1);
TPaveText* labelTop = DrawTools::getLabelTop("Cosmic Data");
labelTop->Draw("same");
gPad->RedrawAxis();
ccorr->SaveAs("sum_fitted.png");
ccorr->SaveAs("sum_fitted.eps");
ccorr->SaveAs("sum_fitted.C");
//ccorr->SaveAs("sum_fitted_log.png");
ccorr->Clear();
h1_cef3_pedSubtracted_corr_sum->Draw();
labelTop->Draw("same");
ccorr->SetLogy(1);
ccorr->SaveAs("sum_fitted_log.png");
ccorr->SaveAs("sum_fitted_log.eps");
ccorr->SaveAs("sum_fitted_log.C");
//unconstrained fit
FitResults fr_corr_unconstrained_sum = fitSingleHisto_sum( h1_cef3_pedSubtracted_corr_sum, 0., 0.,14.,69.,false );
h1_cef3_pedSubtracted_corr_sum->GetXaxis()->SetRangeUser(0.,150.);
h1_cef3_pedSubtracted_corr_sum->GetYaxis()->SetRangeUser(50.,h1_cef3_pedSubtracted_corr_sum->GetMaximum()+h1_cef3_pedSubtracted_corr_sum->GetMaximum()*0.10);
h1_cef3_pedSubtracted_corr_sum->SetXTitle( "ADC Counts" );
h1_cef3_pedSubtracted_corr_sum->Draw();
ccorr->SetLogy(1);
ccorr->SaveAs("sum_fitted_log_unc.png");
ccorr->Clear();
ccorr->cd();
h1_cef3_pedSubtracted_corr_sum_dummy->Add(h1_cef3_pedSubtracted_corr_0);
h1_cef3_pedSubtracted_corr_sum_dummy->Add(h1_cef3_pedSubtracted_corr_1);
h1_cef3_pedSubtracted_corr_sum_dummy->Add(h1_cef3_pedSubtracted_corr_2);
h1_cef3_pedSubtracted_corr_sum_dummy->Add(h1_cef3_pedSubtracted_corr_3);
h1_cef3_pedSubtracted_corr_sum_dummy->Rebin(2);
FitResults fr_corr_sum_dummy = fitSingleHisto_sum( h1_cef3_pedSubtracted_corr_sum_dummy, 0., 0.,13.,68.,true );
h1_cef3_pedSubtracted_corr_sum_dummy->GetXaxis()->SetRangeUser(0.,150.);
h1_cef3_pedSubtracted_corr_sum_dummy->GetYaxis()->SetRangeUser(50.,h1_cef3_pedSubtracted_corr_sum->GetMaximum()+h1_cef3_pedSubtracted_corr_sum->GetMaximum()*0.10);
h1_cef3_pedSubtracted_corr_sum_dummy->SetXTitle( "ADC Counts" );
ccorr->SetLogy(0);
h1_cef3_pedSubtracted_corr_sum_dummy->Draw();
ccorr->SaveAs("sum_fitted_log_dummy.png");
std::cout<<"######################## mean random:"<<h1_cef3_pedSubtracted_corr_sum->GetMean()<<"+-"<<h1_cef3_pedSubtracted_corr_sum->GetMeanError()<<std::endl;
std::cout<<"######################## mean not random:"<<h1_cef3_pedSubtracted_corr_sum_dummy->GetMean()<<"+-"<<h1_cef3_pedSubtracted_corr_sum_dummy->GetMeanError()<<std::endl;
cuncorr_pedSubtracted_2->Clear();
cuncorr_pedSubtracted_2->cd();
cuncorr_pedSubtracted_2->SetLogy();
histo_axes->GetXaxis()->SetRangeUser(20.,100.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
h1_cef3_pedSubtracted_corr_0->Draw("same");
h1_cef3_pedSubtracted_corr_1->Draw("same");
h1_cef3_pedSubtracted_corr_2->Draw("same");
h1_cef3_pedSubtracted_corr_3->Draw("same");
legend->Draw("same");
cuncorr_pedSubtracted_2->SaveAs("corrEnergyAllChannelspedSubtracted_zoom.png");
cuncorr_pedSubtracted_2->SaveAs("corrEnergyAllChannelspedSubtracted_zoom.eps");
//pedsubtracted plot
ccorr->Clear();
histo_axes->GetXaxis()->SetRangeUser(0.,250.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
histo_axes->Draw();
h1_cef3_pedSubtracted_corr_muQ_0->Draw("same");
h1_cef3_pedSubtracted_corr_muQ_1->Draw("same");
h1_cef3_pedSubtracted_corr_muQ_2->Draw("same");
h1_cef3_pedSubtracted_corr_muQ_3->Draw("same");
legend->Draw("same");
ccorr->SaveAs("corr_muQEnergyAllChannelspedSubtracted.png");
ccorr->SaveAs("corr_muQEnergyAllChannelspedSubtracted.eps");
cuncorr_pedSubtracted_2->Clear();
cuncorr_pedSubtracted_2->cd();
cuncorr_pedSubtracted_2->SetLogy();
histo_axes->GetXaxis()->SetRangeUser(20.,100.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
h1_cef3_pedSubtracted_corr_muQ_0->Draw("same");
h1_cef3_pedSubtracted_corr_muQ_1->Draw("same");
h1_cef3_pedSubtracted_corr_muQ_2->Draw("same");
h1_cef3_pedSubtracted_corr_muQ_3->Draw("same");
legend->Draw("same");
cuncorr_pedSubtracted_2->SaveAs("corr_muQEnergyAllChannelspedSubtracted_zoom.png");
cuncorr_pedSubtracted_2->SaveAs("corr_muQEnergyAllChannelspedSubtracted_zoom.eps");
//pedsubtracted muMean plot
ccorr->Clear();
histo_axes->GetXaxis()->SetRangeUser(0.,250.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
histo_axes->Draw();
h1_cef3_pedSubtracted_corr_muMean_0->Draw("same");
h1_cef3_pedSubtracted_corr_muMean_1->Draw("same");
h1_cef3_pedSubtracted_corr_muMean_2->Draw("same");
h1_cef3_pedSubtracted_corr_muMean_3->Draw("same");
legend->Draw("same");
ccorr->SaveAs("corr_muMeanEnergyAllChannelspedSubtracted.png");
ccorr->SaveAs("corr_muMeanEnergyAllChannelspedSubtracted.eps");
cuncorr_pedSubtracted_2->Clear();
cuncorr_pedSubtracted_2->cd();
cuncorr_pedSubtracted_2->SetLogy();
histo_axes->GetXaxis()->SetRangeUser(20.,100.);
histo_axes->GetYaxis()->SetRangeUser(10.,h1_cef3_pedSubtracted_0->GetMaximum()+h1_cef3_pedSubtracted_0->GetMaximum()*0.10);
histo_axes->SetXTitle( "ADC Counts" );
histo_axes->Draw();
h1_cef3_pedSubtracted_corr_muMean_0->Draw("same");
h1_cef3_pedSubtracted_corr_muMean_1->Draw("same");
h1_cef3_pedSubtracted_corr_muMean_2->Draw("same");
h1_cef3_pedSubtracted_corr_muMean_3->Draw("same");
legend->Draw("same");
cuncorr_pedSubtracted_2->SaveAs("corr_muMeanEnergyAllChannelspedSubtracted_zoom.png");
cuncorr_pedSubtracted_2->SaveAs("corr_muMeanEnergyAllChannelspedSubtracted_zoom.eps");
TCanvas* ccorr_2 = new TCanvas( "ccorr_2", "", 600, 600 );
ccorr_2->cd();
ccorr_2->SetLogy();
h2_axes_corr->GetXaxis()->SetRangeUser(120.,200.);
h2_axes_corr->Draw();
h1_cef3_corr_0->Draw("same");
h1_cef3_corr_1->Draw("same");
h1_cef3_corr_2->Draw("same");
h1_cef3_corr_3->Draw("same");
legend->Draw("same");
ccorr_2->SaveAs("corrEnergyAllChannels_zoom.png");
ccorr_2->SaveAs("corrEnergyAllChannels_zoom.eps");
//fit corrected histos
FitResults fr_corr_0 = fitSingleHisto( h1_cef3_corr_0, 0.966779*120., 0.966779*130., 0.966779*138., 0.966779*185. );
FitResults fr_corr_1 = fitSingleHisto( h1_cef3_corr_1, 1.04148*100., 1.04148*125., 1.04148*125., 1.04148*189. );
FitResults fr_corr_2 = fitSingleHisto( h1_cef3_corr_2, 1.01311*100., 1.01311*125., 1.01311*128., 1.01311*189. );
FitResults fr_corr_3 = fitSingleHisto( h1_cef3_corr_3, 0.984112*100., 0.984112*125., 0.984112*137., 0.984112*200. );
FitResults fr_pedSubtracted_corr_0 = fitSingleHisto( h1_cef3_pedSubtracted_corr_0, 0., 0., lowerFitBoundary[0]+2-pedestals[0].first, 188.-pedestals[0].first );
FitResults fr_pedSubtracted_corr_1 = fitSingleHisto( h1_cef3_pedSubtracted_corr_1, 0., 0., lowerFitBoundary[1]-3-pedestals[1].first, 182.-pedestals[1].first );
FitResults fr_pedSubtracted_corr_2 = fitSingleHisto( h1_cef3_pedSubtracted_corr_2, 0., 0., lowerFitBoundary[2]+1-pedestals[2].first, 188.-pedestals[2].first );
FitResults fr_pedSubtracted_corr_3 = fitSingleHisto( h1_cef3_pedSubtracted_corr_3, 0., 0., lowerFitBoundary[3]+1-pedestals[3].first, 198.-pedestals[3].first );
FitResults fr_pedSubtracted_corr_rebin_0 = fitSingleHisto( h1_cef3_pedSubtracted_corr_rebin_0, 0., 0., lowerFitBoundary[0]+2-pedestals[0].first, 189.-pedestals[0].first );
FitResults fr_pedSubtracted_corr_rebin_1 = fitSingleHisto( h1_cef3_pedSubtracted_corr_rebin_1, 0., 0., lowerFitBoundary[1]-3-pedestals[1].first, 177.-pedestals[1].first );
FitResults fr_pedSubtracted_corr_rebin_2 = fitSingleHisto( h1_cef3_pedSubtracted_corr_rebin_2, 0., 0., lowerFitBoundary[2]+1-pedestals[2].first, 181.-pedestals[2].first );
FitResults fr_pedSubtracted_corr_rebin_3 = fitSingleHisto( h1_cef3_pedSubtracted_corr_rebin_3, 0., 0., lowerFitBoundary[3]+1-pedestals[3].first, 196.-pedestals[3].first );
h1_cef3_0->Write();
h1_cef3_1->Write();
h1_cef3_2->Write();
h1_cef3_3->Write();
h1_cef3_tot->Write();
h1_cef3_corr_0->Write();
h1_cef3_corr_1->Write();
h1_cef3_corr_2->Write();
h1_cef3_corr_3->Write();
h1_cef3_corr_tot->Write();
outFile->Write();
doSummaryPlots(fr_pedSubtracted_0,fr_pedSubtracted_1,fr_pedSubtracted_2,fr_pedSubtracted_3,fr_pedSubtracted_corr_0,fr_pedSubtracted_corr_1,fr_pedSubtracted_corr_2,fr_pedSubtracted_corr_3);
return 0;
}
int main( int argc, char* argv[] ) {
std::string runName = "precalib_BGO_pedestal_noSource";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
std::string tag = "V00";
if( argc>2 ) {
std::string tag_str(argv[2]);
tag = tag_str;
}
TString runName_tstr(runName);
bool isOnlyRunNumber = !(runName_tstr.BeginsWith("BTF_"));
TChain* tree = new TChain("recoTree");
if( isOnlyRunNumber ) {
std::cout << "-> We believe you are passing the program only the run number!" << std::endl;
std::cout << "-> So for instance you are passing '246' for run 'BTF_246_20140501-212512_beam'" << std::endl;
std::cout << "(if this is not the case this means TROUBLE)" << std::endl;
std::cout << "-> Will look for runs matching run number: " << runName << std::endl;
tree->Add(Form("analysisTrees_%s/Reco_BTF_%s_*beam.root/recoTree", tag.c_str(), runName.c_str()) );
if( tree->GetEntries()==0 ) {
std::cout << "WARNING! Didn't find any events matching run: " << runName << std::endl;
std::cout << "Exiting" << std::endl;
exit(1913);
}
} else {
std::string fileName = "analysisTrees_"+tag+"/Reco_" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
if( file==0 ) {
std::cout << "ERROR! Din't find file " << fileName << std::endl;
std::cout << "Exiting." << std::endl;
exit(11);
}
tree = (TChain*)file->Get("recoTree");
}
UInt_t evtNumber;
tree->SetBranchAddress( "evtNumber", &evtNumber );
UInt_t adcData[40];
tree->SetBranchAddress( "adcData", adcData );
UInt_t adcBoard[40];
tree->SetBranchAddress( "adcBoard", adcBoard );
UInt_t adcChannel[40];
tree->SetBranchAddress( "adcChannel", adcChannel );
unsigned int runNumber;
int nHodoFibersX;
int nHodoFibersY;
int nHodoClustersX;
int nHodoClustersY;
float cef3_corr[CEF3_CHANNELS];
float bgo_corr[BGO_CHANNELS];
float scintFront;
float pos_hodoClustX[HODOX_CHANNELS];
float pos_hodoClustY[HODOY_CHANNELS];
int nFibres_hodoClustX[HODOX_CHANNELS];
int nFibres_hodoClustY[HODOY_CHANNELS];
float xBeam, yBeam;
bool isSingleEle_scintFront;
bool cef3_ok;
bool cef3_corr_ok;
bool bgo_ok;
bool bgo_corr_ok;
tree->SetBranchAddress( "runNumber", &runNumber );
tree->SetBranchAddress( "scintFront", &scintFront );
tree->SetBranchAddress( "cef3_corr", cef3_corr );
tree->SetBranchAddress( "bgo_corr", bgo_corr );
tree->SetBranchAddress( "nHodoFibersX", &nHodoFibersX );
tree->SetBranchAddress( "nHodoFibersY", &nHodoFibersY );
tree->SetBranchAddress( "nHodoClustersX", &nHodoClustersX );
tree->SetBranchAddress( "pos_hodoClustX", pos_hodoClustX );
tree->SetBranchAddress( "nFibres_hodoClustX", nFibres_hodoClustX );
tree->SetBranchAddress( "nHodoClustersY", &nHodoClustersY );
tree->SetBranchAddress( "pos_hodoClustY", pos_hodoClustY );
tree->SetBranchAddress( "nFibres_hodoClustY", nFibres_hodoClustY );
tree->SetBranchAddress( "scintFront", &scintFront );
tree->SetBranchAddress( "isSingleEle_scintFront", &isSingleEle_scintFront );
tree->SetBranchAddress( "xBeam", &xBeam );
tree->SetBranchAddress( "yBeam", &yBeam );
tree->SetBranchAddress( "cef3_ok", &cef3_ok );
tree->SetBranchAddress( "cef3_corr_ok", &cef3_corr_ok );
tree->SetBranchAddress( "bgo_ok", &bgo_ok );
tree->SetBranchAddress( "bgo_corr_ok", &bgo_corr_ok );
int nentries = tree->GetEntries();
if( isOnlyRunNumber ) {
// modify runname in such a way that it's useful for getBeamPosition and outfile:
runName = "BTF_" + runName + "_beam";
}
std::string outputdir = "SingleElectronSelectionTrees_"+tag;
system( Form("mkdir -p %s", outputdir.c_str()) );
std::string outfileName = outputdir + "/SingleEleSelAn_" + runName + ".root";
TFile* outfile = TFile::Open( outfileName.c_str(), "RECREATE" );
TTree* outTree = new TTree("singleEleSelTree","singleEleSelTree");
outTree->Branch( "run", &runNumber, "run/i" );
outTree->Branch( "scintFront", &scintFront, "scintFront/F" );
outTree->Branch( "isSingleEle_scintFront", &isSingleEle_scintFront, "isSingleEle_scintFront/O" );
outTree->Branch( "nHodoClustersX", &nHodoClustersX, "nHodoClustersX/I" );
outTree->Branch( "nHodoClustersY", &nHodoClustersY, "nHodoClustersY/I" );
outTree->Branch( "cef3_corr", cef3_corr, "cef3_corr[4]/F" );
outTree->Branch( "bgo_corr", bgo_corr, "bgo_corr[8]/F" );
outTree->Branch( "bgo_corr_ok", &bgo_corr_ok, "bgo_corr_ok/O");
outTree->Branch( "xBeam", &xBeam, "xBeam/F" );
outTree->Branch( "yBeam", &yBeam, "yBeam/F" );
TH1D* h1_cef3_corr_tot = new TH1D("cef3_corr_tot", "", 1500, 0., 15000);
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
tree->GetEntry(iEntry);
if( iEntry % 10000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
if( cef3_ok ) {
std::vector<float> v_cef3_corr;
for(unsigned i=0; i<CEF3_CHANNELS; ++i) v_cef3_corr.push_back(cef3_corr[i]);
float eTot_corr = sumVector(v_cef3_corr);
if( cef3_corr_ok ) {
h1_cef3_corr_tot->Fill(eTot_corr);
}
outTree->Fill();
}
}
FitResults fr_0 = fitSingleHisto( h1_cef3_corr_tot, 0.,0., 1000., 11500. );
outfile->cd();
h1_cef3_corr_tot->Write();
outTree->Write();
outfile->Close();
std::cout << "-> Histograms saved in: " << outfile->GetName() << std::endl;
return 0;
}
int main( int argc, char* argv[] ) {
DrawTools::setStyle();
std::string inputDir = "./analysisTrees";
std::string runName = "323";
std::string tag = "V01";
if( argc == 3 ) {
std::string runName_str(argv[1]);
runName = runName_str;
std::string tag_str(argv[2]);
tag = tag_str;
}else if (argc==4){
std::string inputDir_str(argv[2]);
inputDir =inputDir_str;
} else{
std::cout<<"Usage:"<<std::endl;
std::cout<<"./calibrateCef3 runNr tag [inputDir]"<<std::endl;
exit(12345);
}
if(tag!="default") inputDir = inputDir + "_"+tag;
TFile* file = TFile::Open(Form("%s/Reco_%s.root", inputDir.c_str(),runName.c_str()));
std::cout<<"opening file:"<<file->GetName();
TTree* tree = (TTree*)file->Get("recoTree");
std::string outputdir = "CeF3Calibration/";
std::string mkdir_command = "mkdir -p " + outputdir;
system( mkdir_command.c_str() );
std::string ofsName = outputdir +Form( "/constants_%s_%s.txt", runName.c_str(), tag.c_str() );
ofstream ofs(ofsName.c_str());
std::string ofsNameU = outputdir + Form( "/constants_uncert_%s_%s.txt", runName.c_str(), tag.c_str() );
ofstream ofsU(ofsNameU.c_str());
std::vector<float> cef3_calibration;
std::vector<float> cef3_calib_uncert;
for( unsigned i=0; i<4; ++i ) {
TF1* f1 = fitSingleElectronPeak( outputdir, i, tree );
float mean = f1->GetParameter(1);
float sigma = f1->GetParameter(2);
float mean_err = f1->GetParError(1);
std::cout << std::endl;
std::cout << "Channel " << i << std::endl;
std::cout << " Mean : " << mean << std::endl;
std::cout << " Sigma : " << sigma << std::endl;
std::cout << " Resolution : " << sigma/mean << std::endl;
cef3_calibration.push_back(mean);
cef3_calib_uncert.push_back(mean_err);
}
float cef3CalibrationAverage = sumVector(cef3_calibration)/cef3_calibration.size();
for(unsigned i=0; i<cef3_calibration.size(); ++i ){
cef3_calib_uncert[i] = abs(cef3CalibrationAverage-cef3_calibration[i]/4.)/(cef3_calibration[i]*cef3_calibration[i])*cef3_calib_uncert[i];
ofsU << cef3_calib_uncert[i] << std::endl;
cef3_calibration[i] = cef3CalibrationAverage/cef3_calibration[i];
ofs << cef3_calibration[i] << std::endl;
}
ofs.close();
ofsU.close();
if(checkIntercal == true){
checkIntercalibration(cef3_calibration, cef3_calib_uncert, outputdir, runName, tag);
}
std::cout << "-> Saved constants in: " << ofsName << std::endl;
checkTotalResolution( outputdir, tree );
return 0;
}
int main( int argc, char* argv[] ) {
DrawTools::setStyle();
TGaxis::SetMaxDigits(3);
std::string tag="V0";
std::string runName = "";
if( argc<3 ) {
std::cout << "USAGE: ./alignTracking [startTag] [run]" << std::endl;
exit(11);
} else if(argc>1) {
std::string tag_str(argv[1]);
tag = tag_str;
if(argc>2){
std::string runName_str(argv[2]);
runName = runName_str;
}
}
// this is the dir in which the offsets will be saved:
std::string constDirName = "Alignment";
system(Form("mkdir -p %s", constDirName.c_str()));
std::string constFileName = constDirName+"/offsets_"+tag+".txt";
AlignmentOfficer alignOfficer(constFileName);
// TFile* file = TFile::Open("data/run_487.root");
// TFile* file = TFile::Open("data/run_428.root"); //200 GeV
// TFile* file = TFile::Open("data/run_407.root"); //150 GeV
// TFile* file = TFile::Open("data/run_418.root"); //100 GeV
//TFile* file = TFile::Open("data/run_398.root"); //50 GeV
// TFile* file = TFile::Open("data/run_455.root"); //20 GeV
// TFile* file = TFile::Open("data/run_508.root"); //20 GeV seems to be better than run 455
// TFile* file = TFile::Open("data/run_457.root"); //15 GeV
//TFile* file = TFile::Open("data/run_431.root"); //10 GeV
// TFile* file = TFile::Open("data/run_273.root");
std::string fileName="dcap://t3se01.psi.ch:22125//pnfs/psi.ch/cms/trivcat/store/user/micheli/rawData/output_run"+runName+".root";
char* hostName = getenv ("HOSTNAME");
TString hostString(hostName);
bool isLxplus = hostString.Contains("lxplus");
if(isLxplus) fileName= "xroot://t3se01.psi.ch:1094//store/user/micheli/rawData/output_run" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
TTree* tree = (TTree*)file->Get("outputTree");
std::cout<<"Reading run:"<<runName<<std::endl;
// Declaration of leaf types
UInt_t runNumber;
UInt_t spillNumber;
UInt_t evtNumber;
std::vector<float> *BGOvalues;
std::vector<float> *SCINTvalues;
std::vector<float> *TDCreco;
std::vector<float> *digi_charge_integrated;
std::vector<float> *digi_max_amplitude;
std::vector<float> *digi_pedestal;
std::vector<float> *digi_pedestal_rms;
std::vector<float> *digi_time_at_frac30;
std::vector<float> *digi_time_at_frac50;
std::vector<float> *digi_time_at_max;
std::vector<bool> *HODOX1;
std::vector<bool> *HODOX2;
std::vector<bool> *HODOY1;
std::vector<bool> *HODOY2;
std::vector<float> *HODOSMALLvalues;
Float_t TableX;
Float_t TableY;
Float_t CeF3HV;
Float_t BGOHV;
Float_t BeamEnergy;
Float_t BeamTilt;
Int_t IsPhysics;
std::vector<int> *nTdcHits;
std::vector<float> *digi_charge_integrated_sub;
std::vector<float> *digi_max_amplitude_sub;
std::vector<float> *digi_pedestal_sub;
std::vector<float> *digi_pedestal_rms_sub;
std::vector<float> *digi_charge_integrated_corr2;
std::vector<float> *digi_max_amplitude_corr2;
// List of branches
TBranch *b_runNumber; //!
TBranch *b_spillNumber; //!
TBranch *b_evtNumber; //!
TBranch *b_BGOvalues; //!
TBranch *b_SCINTvalues; //!
TBranch *b_HODOSMALLvalues; //!
TBranch *b_TDCreco; //!
TBranch *b_digi_charge_integrated; //!
TBranch *b_digi_max_amplitude; //!
TBranch *b_digi_pedestal; //!
TBranch *b_digi_pedestal_rms; //!
TBranch *b_digi_time_at_frac30; //!
TBranch *b_digi_time_at_frac50; //!
TBranch *b_digi_time_at_max; //!
TBranch *b_HODOX1; //!
TBranch *b_HODOX2; //!
TBranch *b_HODOY1; //!
TBranch *b_HODOY2; //!
TBranch *b_TableX; //!
TBranch *b_TableY; //!
TBranch *b_CeF3HV; //!
TBranch *b_BGOHV; //!
TBranch *b_BeamEnergy; //!
TBranch *b_BeamTilt; //!
TBranch *b_IsPhysics; //!
TBranch *b_nTdcHits; //!
TBranch *b_digi_charge_integrated_sub; //!
TBranch *b_digi_max_amplitude_sub; //!
TBranch *b_digi_pedestal_sub; //!
TBranch *b_digi_pedestal_rms_sub; //!
TBranch *b_digi_charge_integrated_corr2; //!
TBranch *b_digi_max_amplitude_corr2; //!
// Set object pointer
BGOvalues = 0;
SCINTvalues = 0;
TDCreco = 0;
HODOSMALLvalues = 0;
digi_charge_integrated = 0;
digi_max_amplitude = 0;
digi_pedestal = 0;
digi_pedestal_rms = 0;
digi_time_at_frac30 = 0;
digi_time_at_frac50 = 0;
digi_time_at_max = 0;
HODOX1 = 0;
HODOX2 = 0;
HODOY1 = 0;
HODOY2 = 0;
nTdcHits = 0;
digi_charge_integrated_sub = 0;
digi_max_amplitude_sub = 0;
digi_pedestal_sub = 0;
digi_pedestal_rms_sub = 0;
digi_charge_integrated_corr2 = 0;
digi_max_amplitude_corr2 = 0;
//Set Branch Adresses and branch pointers
tree->SetBranchAddress("runNumber", &runNumber, &b_runNumber);
tree->SetBranchAddress("spillNumber", &spillNumber, &b_spillNumber);
tree->SetBranchAddress("evtNumber", &evtNumber, &b_evtNumber);
tree->SetBranchAddress("BGOvalues", &BGOvalues, &b_BGOvalues);
tree->SetBranchAddress("SCINTvalues", &SCINTvalues, &b_SCINTvalues);
tree->SetBranchAddress("HODOSMALLvalues", &HODOSMALLvalues, &b_HODOSMALLvalues);
tree->SetBranchAddress("TDCreco", &TDCreco, &b_TDCreco);
tree->SetBranchAddress("digi_charge_integrated", &digi_charge_integrated, &b_digi_charge_integrated);
tree->SetBranchAddress("digi_max_amplitude", &digi_max_amplitude, &b_digi_max_amplitude);
tree->SetBranchAddress("digi_pedestal", &digi_pedestal, &b_digi_pedestal);
tree->SetBranchAddress("digi_pedestal_rms", &digi_pedestal_rms, &b_digi_pedestal_rms);
tree->SetBranchAddress("digi_time_at_frac30", &digi_time_at_frac30, &b_digi_time_at_frac30);
tree->SetBranchAddress("digi_time_at_frac50", &digi_time_at_frac50, &b_digi_time_at_frac50);
tree->SetBranchAddress("digi_time_at_max", &digi_time_at_max, &b_digi_time_at_max);
tree->SetBranchAddress("HODOX1", &HODOX1, &b_HODOX1);
tree->SetBranchAddress("HODOX2", &HODOX2, &b_HODOX2);
tree->SetBranchAddress("HODOY1", &HODOY1, &b_HODOY1);
tree->SetBranchAddress("HODOY2", &HODOY2, &b_HODOY2);
tree->SetBranchAddress("TableX", &TableX, &b_TableX);
tree->SetBranchAddress("TableY", &TableY, &b_TableY);
tree->SetBranchAddress("CeF3HV", &CeF3HV, &b_CeF3HV);
tree->SetBranchAddress("BGOHV", &BGOHV, &b_BGOHV);
tree->SetBranchAddress("BeamEnergy", &BeamEnergy, &b_BeamEnergy);
tree->SetBranchAddress("BeamTilt", &BeamTilt, &b_BeamTilt);
tree->SetBranchAddress("IsPhysics", &IsPhysics, &b_IsPhysics);
tree->SetBranchAddress("nTdcHits", &nTdcHits, &b_nTdcHits);
tree->SetBranchAddress("digi_charge_integrated_sub", &digi_charge_integrated_sub, &b_digi_charge_integrated_sub);
tree->SetBranchAddress("digi_max_amplitude_sub", &digi_max_amplitude_sub, &b_digi_max_amplitude_sub);
tree->SetBranchAddress("digi_pedestal_sub", &digi_pedestal_sub, &b_digi_pedestal_sub);
tree->SetBranchAddress("digi_pedestal_rms_sub", &digi_pedestal_rms_sub, &b_digi_pedestal_rms_sub);
tree->SetBranchAddress("digi_charge_integrated_corr2", &digi_charge_integrated_corr2, &b_digi_charge_integrated_corr2);
tree->SetBranchAddress("digi_max_amplitude_corr2", &digi_max_amplitude_corr2, &b_digi_max_amplitude_corr2);
int nClusters_hodoX1;
int nFibres_hodoX1[HODOX1_CHANNELS];
float pos_hodoX1[HODOX1_CHANNELS];
int nClusters_hodoY1;
int nFibres_hodoY1[HODOY1_CHANNELS];
float pos_hodoY1[HODOY1_CHANNELS];
int nClusters_hodoX2;
int nFibres_hodoX2[HODOX2_CHANNELS];
float pos_hodoX2[HODOX2_CHANNELS];
int nClusters_hodoY2;
int nFibres_hodoY2[HODOY2_CHANNELS];
float pos_hodoY2[HODOY2_CHANNELS];
float wc_x;
float wc_y;
//int nBins = 80*2*2*2*2;
int nBins = 80;
// int nBinsWC = 40*5;
int nBinsWC = 80;
float xMin = -20.;
float xMax = 20.;
TH1F* h1_wc_y_low = new TH1F("wc_y_low", "", nBinsWC, xMin, xMax);
TH1F* h1_wc_y_hi = new TH1F("wc_y_hi" , "", nBinsWC, xMin, xMax);
TH1F* h1_wc_x_low = new TH1F("wc_x_low", "", nBinsWC, xMin, xMax);
TH1F* h1_wc_x_hi = new TH1F("wc_x_hi" , "", nBinsWC, xMin, xMax);
TH1F* h1_hodoY1_low = new TH1F("hodoY1_low", "", nBins, xMin, xMax);
TH1F* h1_hodoY1_hi = new TH1F("hodoY1_hi" , "", nBins, xMin, xMax);
TH1F* h1_hodoX1_low = new TH1F("hodoX1_low", "", nBins, xMin, xMax);
TH1F* h1_hodoX1_hi = new TH1F("hodoX1_hi" , "", nBins, xMin, xMax);
TH1F* h1_hodoY2_low = new TH1F("hodoY2_low", "", nBins, xMin, xMax);
TH1F* h1_hodoY2_hi = new TH1F("hodoY2_hi" , "", nBins, xMin, xMax);
TH1F* h1_hodoX2_low = new TH1F("hodoX2_low", "", nBins, xMin, xMax);
TH1F* h1_hodoX2_hi = new TH1F("hodoX2_hi" , "", nBins, xMin, xMax);
int nentries = tree->GetEntries();
if(nentries>10000) nentries=10000;
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
tree->GetEntry(iEntry);
if( iEntry % 10000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
std::vector<bool> hodoX1_values(HODOX1_CHANNELS, -1.);
std::vector<bool> hodoY1_values(HODOY1_CHANNELS, -1.);
assignValuesBool( hodoX1_values, *HODOX1, 0. );
assignValuesBool( hodoY1_values, *HODOY1, 0. );
std::vector<bool> hodoX2_values(HODOX2_CHANNELS, -1.);
std::vector<bool> hodoY2_values(HODOY2_CHANNELS, -1.);
assignValuesBool( hodoX2_values, *HODOX2, 0 );
assignValuesBool( hodoY2_values, *HODOY2, 0 );
std::vector<float> hodoSmallX_values(HODOSMALLX_CHANNELS, -1.);
std::vector<float> hodoSmallY_values(HODOSMALLY_CHANNELS, -1.);
assignValues( hodoSmallY_values, *HODOSMALLvalues, 0 );
assignValues( hodoSmallX_values, *HODOSMALLvalues, 4 );
// hodo cluster reconstruction
int clusterMaxFibres = 4;
doHodoReconstructionBool( hodoX1_values , nClusters_hodoX1 , nFibres_hodoX1 , pos_hodoX1 , 0.5, clusterMaxFibres, 0. );
doHodoReconstructionBool( hodoY1_values , nClusters_hodoY1 , nFibres_hodoY1 , pos_hodoY1 , 0.5, clusterMaxFibres, 0. );
doHodoReconstructionBool( hodoX2_values , nClusters_hodoX2 , nFibres_hodoX2 , pos_hodoX2 , 0.5, clusterMaxFibres, 0. );
doHodoReconstructionBool( hodoY2_values , nClusters_hodoY2 , nFibres_hodoY2 , pos_hodoY2 , 0.5, clusterMaxFibres, 0. );
alignOfficer.fix("hodoX1", nClusters_hodoX1, pos_hodoX1);
alignOfficer.fix("hodoY1", nClusters_hodoY1, pos_hodoY1);
alignOfficer.fix("hodoX2", nClusters_hodoX2, pos_hodoX2);
alignOfficer.fix("hodoY2", nClusters_hodoY2, pos_hodoY2);
wc_x = TDCreco->at(0);
wc_y = TDCreco->at(1);
if( runNumber>=170 ) wc_y = -wc_y; // temporary fix
wc_x += alignOfficer.getOffset("wc_x");
wc_y += alignOfficer.getOffset("wc_y");
// float hodoSmallY_low = digi_max_amplitude->at(6);
// float hodoSmallY_hi = digi_max_amplitude->at(7);
//// //float hodoSmallX_low = digi_max_amplitude->at(5);
//// //float hodoSmallX_hi = digi_max_amplitude->at(4);
// float hodoSmallX_low = digi_max_amplitude->at(4);
// float hodoSmallX_hi = digi_max_amplitude->at(5);
bool isOctober2015Run= (runNumber > 3900. && runNumber<4600);
float hodoSmallY_low = HODOSMALLvalues->at(1);
float hodoSmallY_hi = HODOSMALLvalues->at(2);
float hodoSmallX_low = HODOSMALLvalues->at(5);
float hodoSmallX_hi = HODOSMALLvalues->at(6);
if (isOctober2015Run){
hodoSmallY_low = HODOSMALLvalues->at(0);
hodoSmallY_hi = HODOSMALLvalues->at(1);
hodoSmallX_low = HODOSMALLvalues->at(2);
hodoSmallX_hi = HODOSMALLvalues->at(3);
}
// std::cout<< hodoSmallY_low<<" "<<hodoSmallY_hi<<" "<<hodoSmallX_low<<" "<<hodoSmallX_hi<<std::endl;
std::vector<float> pedMeanX,pedMeanY, pedSigmaX, pedSigmaY;
if(!isOctober2015Run){
pedMeanY.push_back(141.30);
pedMeanY.push_back(152.90);
pedMeanX.push_back(139.89);
pedMeanX.push_back(151.73);
pedSigmaY.push_back(1.25);
pedSigmaY.push_back(1.60);
pedSigmaX.push_back(1.32);
pedSigmaX.push_back(1.31);
}else{//ped value for october 2015 run
pedMeanY.push_back(212.5);
pedMeanY.push_back(202.0);
pedMeanX.push_back(218.6);//this channels are not working in these runs....
pedMeanX.push_back(208.0);
pedSigmaY.push_back(1.51);
pedSigmaY.push_back(1.67);
pedSigmaX.push_back(2.04);
pedSigmaX.push_back(1.83);
}
// y low
if( hodoSmallY_low>pedMeanY[0]+5*pedSigmaY[0] ) {
int posOf2FibClustY1Low = 0;
int posOf2FibClustY2Low=0;
int nrOf2FibreClustersY1Low = 0;
for( int i=0; i<nClusters_hodoY1; ++i ) {
if( nFibres_hodoY1[i]==2 ) {
++nrOf2FibreClustersY1Low;
posOf2FibClustY1Low= i ;}
}
//And for Y2 LOW:
int nrOf2FibreClustersY2Low = 0;
for( int i=0; i<nClusters_hodoY2; ++i ) {
if( nFibres_hodoY2[i]==2 ) {
++nrOf2FibreClustersY2Low;
posOf2FibClustY2Low=i; }
}
if((nrOf2FibreClustersY2Low==1||nClusters_hodoY2==1)&& (nrOf2FibreClustersY1Low==1 || nClusters_hodoY1==1) && abs(wc_y) <200 && nTdcHits->at(2)>0 && nTdcHits->at(3)>0 && nTdcHits->at(2)<3 && nTdcHits->at(3)<3 && ( nTdcHits->at(2) + nTdcHits->at(3))<4){
h1_wc_y_low->Fill( wc_y );
h1_hodoY1_low->Fill( pos_hodoY1[posOf2FibClustY1Low] );
h1_hodoY2_low->Fill( pos_hodoY2[posOf2FibClustY2Low] );
}
}
// y hi
if( hodoSmallY_hi>pedMeanY[1]+5*pedSigmaY[1] ) {
int posOf2FibClustY1Hi = 0;
int posOf2FibClustY2Hi=0;
int nrOf2FibreClustersY1Hi = 0;
for( int i=0; i<nClusters_hodoY1; ++i ) {
if( nFibres_hodoY1[i]==2 ) {
++nrOf2FibreClustersY1Hi;
posOf2FibClustY1Hi= i ;}
}
//And for Y2 HI:
int nrOf2FibreClustersY2Hi = 0;
for( int i=0; i<nClusters_hodoY2; ++i ) {
if( nFibres_hodoY2[i]==2 ) {
++nrOf2FibreClustersY2Hi;
posOf2FibClustY2Hi=i; }
}
if((nrOf2FibreClustersY2Hi==1||nClusters_hodoY2==1)&& (nrOf2FibreClustersY1Hi==1 || nClusters_hodoY1==1) && abs(wc_y) <200 && nTdcHits->at(2)>0 && nTdcHits->at(3)>0 && nTdcHits->at(2)<3 && nTdcHits->at(3)<3 && ( nTdcHits->at(2) + nTdcHits->at(3))<4){
h1_wc_y_hi->Fill( wc_y );
h1_hodoY1_hi->Fill( pos_hodoY1[ posOf2FibClustY1Hi] );
h1_hodoY2_hi->Fill( pos_hodoY2[posOf2FibClustY2Hi] );
}
}
// x low
if( hodoSmallX_low>pedMeanX[0]+5*pedSigmaX[0] ) {
int posOf2FibClustX1Low = 0;
int posOf2FibClustX2Low=0;
int nrOf2FibreClustersX1Low = 0;
for( int i=0; i<nClusters_hodoX1; ++i ) {
if( nFibres_hodoX1[i]==2 ) {
++nrOf2FibreClustersX1Low;
posOf2FibClustX1Low= i ;}
}
//And for X2 LOW:
int nrOf2FibreClustersX2Low = 0;
for( int i=0; i<nClusters_hodoX2; ++i ) {
if( nFibres_hodoX2[i]==2 ) {
++nrOf2FibreClustersX2Low;
posOf2FibClustX2Low=i; }
}
if((nrOf2FibreClustersX2Low==1||nClusters_hodoX2==1)&& (nrOf2FibreClustersX1Low==1 || nClusters_hodoX1==1) && abs(wc_x)<200 && nTdcHits->at(0)>0 && nTdcHits->at(1)>0 && nTdcHits->at(0)<3 && nTdcHits->at(1)<3 && ( nTdcHits->at(0) + nTdcHits->at(1) )<4){
h1_wc_x_low->Fill( wc_x );
h1_hodoX1_low->Fill( pos_hodoX1[ posOf2FibClustX1Low] );
h1_hodoX2_low->Fill( pos_hodoX2[posOf2FibClustX2Low] );
}
}
// x hi
if( hodoSmallX_hi>pedMeanX[1]+5*pedSigmaX[1] ) {
int posOf2FibClustX1Hi = 0;
int posOf2FibClustX2Hi=0;
int nrOf2FibreClustersX1Hi = 0;
for( int i=0; i<nClusters_hodoX1; ++i ) {
if( nFibres_hodoX1[i]==2 ) {
++nrOf2FibreClustersX1Hi;
posOf2FibClustX1Hi= i ;}
}
//And for X2 HI:
int nrOf2FibreClustersX2Hi = 0;
for( int i=0; i<nClusters_hodoX2; ++i ) {
if( nFibres_hodoX2[i]==2 ) {
++nrOf2FibreClustersX2Hi;
posOf2FibClustX2Hi=i; }
}
if((nrOf2FibreClustersX2Hi==1||nClusters_hodoX2==1)&& (nrOf2FibreClustersX1Hi==1 || nClusters_hodoX1==1) && abs(wc_x) <200 && nTdcHits->at(0)>0 && nTdcHits->at(1)>0 && nTdcHits->at(0)<3 && nTdcHits->at(1)<3 && ( nTdcHits->at(0) + nTdcHits->at(1) )<4){
h1_wc_x_hi->Fill( wc_x );
h1_hodoX1_hi->Fill( pos_hodoX1[ posOf2FibClustX1Hi] );
h1_hodoX2_hi->Fill( pos_hodoX2[posOf2FibClustX2Hi] );
}
}
} // for entries
std::string outputdir = "AlignmentPlots_"+tag;
system( Form("mkdir -p %s", outputdir.c_str()));
float offset_wc_y_low = fitAndDraw( outputdir, h1_wc_y_low );
float offset_wc_y_hi = fitAndDraw( outputdir, h1_wc_y_hi );
float offset_wc_x_low = fitAndDraw( outputdir, h1_wc_x_low );
float offset_wc_x_hi = fitAndDraw( outputdir, h1_wc_x_hi );
float offset_hodoY1_low = fitAndDraw( outputdir, h1_hodoY1_low );
float offset_hodoY1_hi = fitAndDraw( outputdir, h1_hodoY1_hi );
float offset_hodoX1_low = fitAndDraw( outputdir, h1_hodoX1_low );
float offset_hodoX1_hi = fitAndDraw( outputdir, h1_hodoX1_hi );
float offset_hodoY2_low = fitAndDraw( outputdir, h1_hodoY2_low );
float offset_hodoY2_hi = fitAndDraw( outputdir, h1_hodoY2_hi );
float offset_hodoX2_low = fitAndDraw( outputdir, h1_hodoX2_low );
float offset_hodoX2_hi = fitAndDraw( outputdir, h1_hodoX2_hi );
float offset_wc_y = 0.5*(offset_wc_y_low +offset_wc_y_hi);
float offset_wc_x = 0.5*(offset_wc_x_low +offset_wc_x_hi);
float offset_hodoY1 = 0.5*(offset_hodoY1_low+offset_hodoY1_hi);
float offset_hodoX1 = 0.5*(offset_hodoX1_low+offset_hodoX1_hi);
float offset_hodoY2 = 0.5*(offset_hodoY2_low+offset_hodoY2_hi);
float offset_hodoX2 = 0.5*(offset_hodoX2_low+offset_hodoX2_hi);
std::cout << "offset_wc_y: " << offset_wc_y << std::endl;
std::cout << "offset_wc_x: " << offset_wc_x << std::endl;
std::cout << "offset_hodoY1: " << offset_hodoY1 << std::endl;
std::cout << "offset_hodoX1: " << offset_hodoX1 << std::endl;
std::cout << "offset_hodoY2: " << offset_hodoY2 << std::endl;
std::cout << "offset_hodoX2: " << offset_hodoX2 << std::endl;
std::string ofsName = constDirName + "/offsets_new.txt";
ofstream ofs(ofsName.c_str());
ofs << "wc_y " << -offset_wc_y << std::endl;
ofs << "wc_x " << -offset_wc_x << std::endl;
ofs << "hodoY1 " << -offset_hodoY1 << std::endl;
ofs << "hodoX1 " << -offset_hodoX1 << std::endl;
ofs << "hodoY2 " << -offset_hodoY2 << std::endl;
ofs << "hodoX2 " << -offset_hodoX2 << std::endl;
ofs.close();
std::cout << "-> Saved offsets in: " << ofsName << std::endl;
return 0;
}
int main( int argc, char* argv[] ) {
setStyle();
std::string runName = "test_10";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
std::string fileName = "../PositionAnalysis/PosAn_" + runName + ".root";
TFile* file = TFile::Open( fileName.c_str() );
std::cout << "-> Opened file: " << fileName << std::endl;
TTree* tree = (TTree*)file->Get("tree_passedEvents");
float hodox_corr[HODOX_CHANNELS], hodoy_corr[HODOY_CHANNELS],cef3[CEF3_CHANNELS],cef3_corr[CEF3_CHANNELS],cef3_pedSubtracted[CEF3_CHANNELS],scintFront;
int nHodoClustersX,nHodoClustersY;
tree->SetBranchAddress( "nHodoClustersX", &nHodoClustersX );
tree->SetBranchAddress( "nHodoClustersY", &nHodoClustersY );
tree->SetBranchAddress( "hodox_corr", hodox_corr );
tree->SetBranchAddress( "hodoy_corr", hodoy_corr );
tree->SetBranchAddress( "cef3", cef3 );
tree->SetBranchAddress( "cef3_corr", cef3_corr );
tree->SetBranchAddress( "cef3_pedSubtracted", cef3_pedSubtracted );
tree->SetBranchAddress( "scintFront", &scintFront );
int nentries = tree->GetEntries();
std::string outfileName = "PhotoElectronAn_" + runName + ".root";
TFile* outfile = TFile::Open( outfileName.c_str(), "RECREATE" );
TH1D* h1_singleEleEnergyDistribution = new TH1D("singleEleEnergyDistribution", "",1000 , 0 , 10000);
TH1D* h1_singleEleEnergyDistributionCalib = new TH1D("singleEleEnergyDistribution", "",1000 , 0 , 10000);
TH1D* h1_singleEleEnergyDistribution_ch[4];
TH1D* h1_singleEleEnergyDistributionCalib_ch[4];
// Q1 calibration
// std::vector<float> correctionFactors;
// correctionFactors.push_back(1.11228);
// correctionFactors.push_back(0.855333);
// correctionFactors.push_back(0.9802);
// correctionFactors.push_back(1.08781);
// mu*Q1 calibration
// std::vector<float> correctionFactors;
// correctionFactors.push_back(1.19514);
// correctionFactors.push_back(0.860177);
// correctionFactors.push_back(0.952363);
// correctionFactors.push_back(0.977169);
//mumean*Q1 calibration
std::vector<float> correctionFactors;
correctionFactors.push_back(1.09716);
correctionFactors.push_back(0.843707);
correctionFactors.push_back(0.966413);
correctionFactors.push_back(1.07241);
for (int i=0;i<4;++i){
TString nameHisto="singleEnergyDistribution";
TString nameHistoCalib="singleEnergyDistributionCalib";
nameHisto+=i;
nameHistoCalib+=i;
h1_singleEleEnergyDistribution_ch[i] = new TH1D(nameHisto, "",100 , 0 , 4000.);
h1_singleEleEnergyDistributionCalib_ch[i] = new TH1D(nameHistoCalib, "",100 , 0 , 4000);
}
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
tree->GetEntry(iEntry);
if( iEntry % 1000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
bool isSingleEle=scintFront>500. && scintFront<2000. && nHodoClustersX==1 && nHodoClustersY==1;
float cef3Total=0.;
float cef3TotalCalib=0.;
for (int i=0;i<CEF3_CHANNELS;i++){
float energy=cef3_pedSubtracted[i];
cef3Total+=energy;
cef3TotalCalib+=energy*correctionFactors[i];
if(isSingleEle){
h1_singleEleEnergyDistribution_ch[i]->Fill(energy);
h1_singleEleEnergyDistributionCalib_ch[i]->Fill(energy*correctionFactors[i]);
}
}
if(isSingleEle) {
h1_singleEleEnergyDistribution->Fill(cef3Total);
h1_singleEleEnergyDistributionCalib->Fill(cef3TotalCalib);
}
}
TCanvas * c1 = new TCanvas("c1", "c1", 600, 600);
c1->cd();
TH2D* h2_axes = new TH2D("axes", "", 10, 100., 2000., 10, 0., 1.15*h1_singleEleEnergyDistributionCalib_ch[2]->GetMaximum() );
h2_axes->SetXTitle( "ADC Counts" );
h2_axes->Draw();
h1_singleEleEnergyDistributionCalib_ch[0]->SetLineColor(kBlack);
h1_singleEleEnergyDistributionCalib_ch[1]->SetLineColor(kRed);
h1_singleEleEnergyDistributionCalib_ch[2]->SetLineColor(kBlue);
h1_singleEleEnergyDistributionCalib_ch[3]->SetLineColor(kMagenta);
h1_singleEleEnergyDistributionCalib_ch[0]->SetLineWidth(2);
h1_singleEleEnergyDistributionCalib_ch[1]->SetLineWidth(2);
h1_singleEleEnergyDistributionCalib_ch[2]->SetLineWidth(2);
h1_singleEleEnergyDistributionCalib_ch[3]->SetLineWidth(2);
TLegend* legend = new TLegend( 0.7, 0.7, 0.90, 0.9 );
legend->SetLineColor(0);
legend->SetFillColor(0);
legend->SetFillStyle(0);
legend->SetTextSize(0.038);
legend->AddEntry( h1_singleEleEnergyDistributionCalib_ch[0], "channel 0", "L" );
legend->AddEntry( h1_singleEleEnergyDistributionCalib_ch[1], "channel 1", "L" );
legend->AddEntry( h1_singleEleEnergyDistributionCalib_ch[2], "channel 2", "L" );
legend->AddEntry( h1_singleEleEnergyDistributionCalib_ch[3], "channel 3", "L" );
legend->Draw("same");
h1_singleEleEnergyDistributionCalib_ch[2]->Draw("same");
h1_singleEleEnergyDistributionCalib_ch[0]->Draw("same");
h1_singleEleEnergyDistributionCalib_ch[1]->Draw("same");
h1_singleEleEnergyDistributionCalib_ch[3]->Draw("same");
c1->SaveAs("calibratedChannels.png");
c1->SaveAs("calibratedChannels.eps");
c1->Clear();
h2_axes->Draw();
h1_singleEleEnergyDistribution_ch[0]->SetLineColor(kBlack);
h1_singleEleEnergyDistribution_ch[1]->SetLineColor(kRed);
h1_singleEleEnergyDistribution_ch[2]->SetLineColor(kBlue);
h1_singleEleEnergyDistribution_ch[3]->SetLineColor(kMagenta);
h1_singleEleEnergyDistribution_ch[0]->SetLineWidth(2);
h1_singleEleEnergyDistribution_ch[1]->SetLineWidth(2);
h1_singleEleEnergyDistribution_ch[2]->SetLineWidth(2);
h1_singleEleEnergyDistribution_ch[3]->SetLineWidth(2);
h1_singleEleEnergyDistribution_ch[0]->Draw("same");
h1_singleEleEnergyDistribution_ch[1]->Draw("same");
h1_singleEleEnergyDistribution_ch[2]->Draw("same");
h1_singleEleEnergyDistribution_ch[3]->Draw("same");
legend->Draw("same");
c1->SaveAs("uncalibratedChannels.png");
c1->SaveAs("uncalibratedChannels.eps");
//plot mu,rms of uncalib distribution
TH1D* h1_mean = new TH1D("mean", "", 4, -0.5, 3.5);
TH1D* h1_rms = new TH1D("rms", "", 4, -0.5, 3.5);
TF1* f1_gaus = new TF1( "f1_gaus", "gaus",0,2000 );
f1_gaus->SetRange(h1_singleEleEnergyDistribution_ch[0]->GetMean()-3*h1_singleEleEnergyDistribution_ch[0]->GetRMS(),h1_singleEleEnergyDistribution_ch[0]->GetMean()+3*h1_singleEleEnergyDistribution_ch[0]->GetRMS() );
h1_singleEleEnergyDistribution_ch[0]->Fit(f1_gaus,"R");
h1_mean->SetBinContent( 1, f1_gaus->GetParameter(1) );
h1_mean->SetBinError( 1, f1_gaus->GetParError(1) );
h1_rms->SetBinContent( 1, f1_gaus->GetParameter(2) );
h1_rms->SetBinError( 1, f1_gaus->GetParError(2) );
f1_gaus->SetRange(h1_singleEleEnergyDistribution_ch[1]->GetMean()-3*h1_singleEleEnergyDistribution_ch[1]->GetRMS(),h1_singleEleEnergyDistribution_ch[1]->GetMean()+3*h1_singleEleEnergyDistribution_ch[1]->GetRMS() );
h1_singleEleEnergyDistribution_ch[1]->Fit(f1_gaus,"R");
h1_mean->SetBinContent( 2, f1_gaus->GetParameter(1) );
h1_mean->SetBinError( 2, f1_gaus->GetParError(1) );
h1_rms->SetBinContent( 2, f1_gaus->GetParameter(2) );
h1_rms->SetBinError( 2, f1_gaus->GetParError(2) );
f1_gaus->SetRange(h1_singleEleEnergyDistribution_ch[2]->GetMean()-3*h1_singleEleEnergyDistribution_ch[2]->GetRMS(),h1_singleEleEnergyDistribution_ch[2]->GetMean()+3*h1_singleEleEnergyDistribution_ch[2]->GetRMS() );
h1_singleEleEnergyDistribution_ch[2]->Fit(f1_gaus,"R");
h1_mean->SetBinContent( 3, f1_gaus->GetParameter(1) );
h1_mean->SetBinError( 3, f1_gaus->GetParError(2) );
h1_rms->SetBinContent( 3, f1_gaus->GetParameter(2) );
h1_rms->SetBinError( 3, f1_gaus->GetParError(2) );
f1_gaus->SetRange(h1_singleEleEnergyDistribution_ch[3]->GetMean()-3*h1_singleEleEnergyDistribution_ch[3]->GetRMS(),h1_singleEleEnergyDistribution_ch[3]->GetMean()+3*h1_singleEleEnergyDistribution_ch[3]->GetRMS() );
h1_singleEleEnergyDistribution_ch[3]->Fit(f1_gaus,"R");
h1_mean->SetBinContent( 4, f1_gaus->GetParameter(1) );
h1_mean->SetBinError( 4, f1_gaus->GetParError(1) );
h1_rms->SetBinContent( 4, f1_gaus->GetParameter(2) );
h1_rms->SetBinError( 4, f1_gaus->GetParError(2) );
c1->Clear();
h1_mean->SetXTitle( "Channel Number");
h1_mean->SetMarkerStyle(20);
h1_mean->SetMarkerSize(1.6);
h1_mean->GetYaxis()->SetRangeUser(720.,890.);
h1_mean->Draw();
c1->SaveAs("muUncalibratedChannels.png");
c1->SaveAs("muUncalibratedChannels.eps");
c1->Clear();
h1_rms->SetXTitle( "Channel Number");
h1_rms->SetMarkerStyle(20);
h1_rms->SetMarkerSize(1.6);
h1_rms->Draw();
c1->SaveAs("rmsUncalibratedChannels.png");
c1->SaveAs("rmsUncalibratedChannels.eps");
//plot mu,rms of uncalib distribution
TH1D* h1_meanCalib = new TH1D("meanCalib", "", 4, -0.5, 3.5);
TH1D* h1_rmsCalib = new TH1D("rmsCalib", "", 4, -0.5, 3.5);
f1_gaus->SetRange(h1_singleEleEnergyDistributionCalib_ch[0]->GetMean()-3*h1_singleEleEnergyDistributionCalib_ch[0]->GetRMS(),h1_singleEleEnergyDistributionCalib_ch[0]->GetMean()+3*h1_singleEleEnergyDistributionCalib_ch[0]->GetRMS() );
h1_singleEleEnergyDistributionCalib_ch[0]->Fit(f1_gaus,"R");
h1_meanCalib->SetBinContent( 1, f1_gaus->GetParameter(1) );
h1_meanCalib->SetBinError( 1, f1_gaus->GetParError(1) );
h1_rmsCalib->SetBinContent( 1, f1_gaus->GetParameter(2) );
h1_rmsCalib->SetBinError( 1, f1_gaus->GetParError(2) );
f1_gaus->SetRange(h1_singleEleEnergyDistributionCalib_ch[1]->GetMean()-3*h1_singleEleEnergyDistributionCalib_ch[1]->GetRMS(),h1_singleEleEnergyDistributionCalib_ch[1]->GetMean()+3*h1_singleEleEnergyDistributionCalib_ch[1]->GetRMS() );
h1_singleEleEnergyDistributionCalib_ch[1]->Fit(f1_gaus,"R");
h1_meanCalib->SetBinContent( 2, f1_gaus->GetParameter(1) );
h1_meanCalib->SetBinError( 2, f1_gaus->GetParError(1) );
h1_rmsCalib->SetBinContent( 2, f1_gaus->GetParameter(2) );
h1_rmsCalib->SetBinError( 2, f1_gaus->GetParError(2) );
f1_gaus->SetRange(h1_singleEleEnergyDistributionCalib_ch[2]->GetMean()-3*h1_singleEleEnergyDistributionCalib_ch[2]->GetRMS(),h1_singleEleEnergyDistributionCalib_ch[2]->GetMean()+3*h1_singleEleEnergyDistributionCalib_ch[2]->GetRMS() );
h1_singleEleEnergyDistributionCalib_ch[2]->Fit(f1_gaus,"R");
h1_meanCalib->SetBinContent( 3, f1_gaus->GetParameter(1) );
h1_meanCalib->SetBinError( 3, f1_gaus->GetParError(2) );
h1_rmsCalib->SetBinContent( 3, f1_gaus->GetParameter(2) );
h1_rmsCalib->SetBinError( 3, f1_gaus->GetParError(2) );
f1_gaus->SetRange(h1_singleEleEnergyDistributionCalib_ch[3]->GetMean()-3*h1_singleEleEnergyDistributionCalib_ch[3]->GetRMS(),h1_singleEleEnergyDistributionCalib_ch[3]->GetMean()+3*h1_singleEleEnergyDistributionCalib_ch[3]->GetRMS() );
h1_singleEleEnergyDistributionCalib_ch[3]->Fit(f1_gaus,"R");
h1_meanCalib->SetBinContent( 4, f1_gaus->GetParameter(1) );
h1_meanCalib->SetBinError( 4, f1_gaus->GetParError(1) );
h1_rmsCalib->SetBinContent( 4, f1_gaus->GetParameter(2) );
h1_rmsCalib->SetBinError( 4, f1_gaus->GetParError(2) );
c1->Clear();
h1_meanCalib->GetYaxis()->SetRangeUser(720.,890.);
h1_meanCalib->SetXTitle( "Channel Number");
h1_meanCalib->SetMarkerStyle(20);
h1_meanCalib->SetMarkerSize(1.6);
h1_meanCalib->Draw();
c1->SaveAs("muCalibratedChannels.png");
c1->SaveAs("muCalibratedChannels.eps");
c1->Clear();
h1_rmsCalib->SetXTitle( "Channel Number");
h1_rmsCalib->SetMarkerStyle(20);
h1_rmsCalib->SetMarkerSize(1.6);
h1_rmsCalib->Draw();
c1->SaveAs("rmsCalibratedChannels.png");
c1->SaveAs("rmsCalibratedChannels.eps");
float mean=0.,rms=0.;
float meanCalib=0.,rmscalib=0.;
for (int i=1;i<=4;i++){
mean+=h1_mean->GetBinContent(i);
meanCalib+=h1_meanCalib->GetBinContent(i);
}
mean=mean/4.;
meanCalib=meanCalib/4.;
for (int i=1;i<=4;i++){
rms+= (h1_mean->GetBinContent(i)-mean)*(h1_mean->GetBinContent(i)-mean);
rmscalib+= (h1_meanCalib->GetBinContent(i)-meanCalib)*(h1_meanCalib->GetBinContent(i)-meanCalib);
}
rms=sqrt(rms/4);
rmscalib=sqrt(rmscalib/4);
std::cout<<"rms: "<<rms<<" rms_calib: "<<rmscalib<<std::endl;
outfile->Write();
outfile->Close();
return 0;
}
int main( int argc, char* argv[] ) {
std::string runName = "precalib_BGO_pedestal_noSource";
if( argc>1 ) {
std::string runName_str(argv[1]);
runName = runName_str;
}
std::string tag = "V00";
if( argc>2 ) {
std::string tag_str(argv[2]);
tag = tag_str;
}
TString runName_tstr(runName);
bool isOnlyRunNumber = !(runName_tstr.BeginsWith("BTF_"));
TChain* tree = new TChain("recoTree");
if( isOnlyRunNumber ) {
std::cout << "-> We believe you are passing the program only the run number!" << std::endl;
std::cout << "-> So for instance you are passing '246' for run 'BTF_246_20140501-212512_beam'" << std::endl;
std::cout << "(if this is not the case this means TROUBLE)" << std::endl;
std::cout << "-> Will look for runs matching run number: " << runName << std::endl;
tree->Add(Form("analysisTrees_%s/Reco_BTF_%s_*beam.root/recoTree", tag.c_str(), runName.c_str()) );
if( tree->GetEntries()==0 ) {
std::cout << "WARNING! Didn't find any events matching run: " << runName << std::endl;
std::cout << "Exiting" << std::endl;
exit(1913);
}
} else {
std::string fileName = "analysisTrees_"+tag+"/Reco_" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
if( file==0 ) {
std::cout << "ERROR! Din't find file " << fileName << std::endl;
std::cout << "Exiting." << std::endl;
exit(11);
}
tree = (TChain*)file->Get("recoTree");
}
UInt_t evtNumber;
tree->SetBranchAddress( "evtNumber", &evtNumber );
UInt_t adcData[40];
tree->SetBranchAddress( "adcData", adcData );
UInt_t adcBoard[40];
tree->SetBranchAddress( "adcBoard", adcBoard );
UInt_t adcChannel[40];
tree->SetBranchAddress( "adcChannel", adcChannel );
unsigned int runNumber;
int nHodoFibersX;
int nHodoFibersY;
int nHodoClustersX;
int nHodoClustersY;
float cef3_corr[CEF3_CHANNELS];
float bgo_corr[BGO_CHANNELS];
float scintFront;
float pos_hodoClustX[HODOX_CHANNELS];
float pos_hodoClustY[HODOY_CHANNELS];
int nFibres_hodoClustX[HODOX_CHANNELS];
int nFibres_hodoClustY[HODOY_CHANNELS];
float xBeam, yBeam;
bool isSingleEle_scintFront;
bool cef3_ok;
bool cef3_corr_ok;
bool bgo_ok;
bool bgo_corr_ok;
tree->SetBranchAddress( "runNumber", &runNumber );
tree->SetBranchAddress( "scintFront", &scintFront );
tree->SetBranchAddress( "cef3_corr", cef3_corr );
tree->SetBranchAddress( "bgo_corr", bgo_corr );
tree->SetBranchAddress( "nHodoFibersX", &nHodoFibersX );
tree->SetBranchAddress( "nHodoFibersY", &nHodoFibersY );
tree->SetBranchAddress( "nHodoClustersX", &nHodoClustersX );
tree->SetBranchAddress( "pos_hodoClustX", pos_hodoClustX );
tree->SetBranchAddress( "nFibres_hodoClustX", nFibres_hodoClustX );
tree->SetBranchAddress( "nHodoClustersY", &nHodoClustersY );
tree->SetBranchAddress( "pos_hodoClustY", pos_hodoClustY );
tree->SetBranchAddress( "nFibres_hodoClustY", nFibres_hodoClustY );
tree->SetBranchAddress( "scintFront", &scintFront );
tree->SetBranchAddress( "isSingleEle_scintFront", &isSingleEle_scintFront );
tree->SetBranchAddress( "xBeam", &xBeam );
tree->SetBranchAddress( "yBeam", &yBeam );
tree->SetBranchAddress( "cef3_ok", &cef3_ok );
tree->SetBranchAddress( "cef3_corr_ok", &cef3_corr_ok );
tree->SetBranchAddress( "bgo_ok", &bgo_ok );
tree->SetBranchAddress( "bgo_corr_ok", &bgo_corr_ok );
int nBins = 500;
float xMax = 25.*3./2.;
TH1D* h1_xPos = new TH1D("xPos", "", nBins, -xMax, xMax);
TH1D* h1_yPos = new TH1D("yPos", "", nBins, -xMax, xMax);
TH2D* h2_xyPos = new TH2D("xyPos", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_singleEle = new TH1D("xPos_singleEle", "", nBins, -xMax, xMax);
TH1D* h1_yPos_singleEle = new TH1D("yPos_singleEle", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_singleEle = new TH2D("xyPos_singleEle", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_new = new TH1D("xPos_new", "", nBins, -xMax, xMax);
TH1D* h1_yPos_new = new TH1D("yPos_new", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_new = new TH2D("xyPos_new", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_new_singleEle = new TH1D("xPos_new_singleEle", "", nBins, -xMax, xMax);
TH1D* h1_yPos_new_singleEle = new TH1D("yPos_new_singleEle", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_new_singleEle = new TH2D("xyPos_new_singleEle", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_bgo = new TH1D("xPos_bgo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_bgo = new TH1D("yPos_bgo", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_bgo = new TH2D("xyPos_bgo", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_singleEle_bgo = new TH1D("xPos_singleEle_bgo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_singleEle_bgo = new TH1D("yPos_singleEle_bgo", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_singleEle_bgo = new TH2D("xyPos_singleEle_bgo", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_hodo = new TH1D("xPos_hodo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_hodo = new TH1D("yPos_hodo", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_hodo = new TH2D("xyPos_hodo", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_singleEle_hodo = new TH1D("xPos_singleEle_hodo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_singleEle_hodo = new TH1D("yPos_singleEle_hodo", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_singleEle_hodo = new TH2D("xyPos_singleEle_hodo", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_singleEle_hodoClust = new TH1D("xPos_singleEle_hodoClust", "", nBins, -xMax, xMax);
TH1D* h1_yPos_singleEle_hodoClust = new TH1D("yPos_singleEle_hodoClust", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_singleEle_hodoClust = new TH2D("xyPos_singleEle_hodoClust", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_calo = new TH1D("xPos_calo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo = new TH1D("yPos_calo", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_calo = new TH2D("xyPos_calo", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_singleEle_calo = new TH1D("xPos_singleEle_calo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_singleEle_calo = new TH1D("yPos_singleEle_calo", "", nBins, -xMax, xMax);
TH2D* h2_xyPos_singleEle_calo = new TH2D("xyPos_singleEle_calo", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH1D* h1_xPos_calo_vs_hodo = new TH1D("xPos_calo_vs_hodo", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo_vs_hodo = new TH1D("yPos_calo_vs_hodo", "", nBins, -xMax, xMax);
TH1D* h1_xPos_calo_vs_beam = new TH1D("xPos_calo_vs_beam", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo_vs_beam = new TH1D("yPos_calo_vs_beam", "", nBins, -xMax, xMax);
TH1D* h1_xPos_calo_vs_hodo_singleElectron = new TH1D("xPos_calo_vs_hodo_singleElectron", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo_vs_hodo_singleElectron = new TH1D("yPos_calo_vs_hodo_singleElectron", "", nBins, -xMax, xMax);
TH1D* h1_xPos_calo_vs_beam_singleElectron = new TH1D("xPos_calo_vs_beam_singleElectron", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo_vs_beam_singleElectron = new TH1D("yPos_calo_vs_beam_singleElectron", "", nBins, -xMax, xMax);
TH1D* h1_xPos_calo_vs_hodo_singleElectron_HR = new TH1D("xPos_calo_vs_hodo_singleElectron_HR", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo_vs_hodo_singleElectron_HR = new TH1D("yPos_calo_vs_hodo_singleElectron_HR", "", nBins, -xMax, xMax);
TH1D* h1_xPos_calo_vs_beam_singleElectron_HR = new TH1D("xPos_calo_vs_beam_singleElectron_HR", "", nBins, -xMax, xMax);
TH1D* h1_yPos_calo_vs_beam_singleElectron_HR = new TH1D("yPos_calo_vs_beam_singleElectron_HR", "", nBins, -xMax, xMax);
TH2D* h2_correlation_cef3_hodo_xPos = new TH2D("correlation_cef3_hodo_xPos", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_hodo_yPos = new TH2D("correlation_cef3_hodo_yPos", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_bgo_xPos = new TH2D("correlation_cef3_bgo_xPos", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_bgo_yPos = new TH2D("correlation_cef3_bgo_yPos", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_hodo_bgo_xPos = new TH2D("correlation_hodo_bgo_xPos", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_hodo_bgo_yPos = new TH2D("correlation_hodo_bgo_yPos", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_hodo_xPos_singleEle = new TH2D("correlation_cef3_hodo_xPos_singleEle", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_hodo_yPos_singleEle = new TH2D("correlation_cef3_hodo_yPos_singleEle", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_bgo_xPos_singleEle = new TH2D("correlation_cef3_bgo_xPos_singleEle", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_cef3_bgo_yPos_singleEle = new TH2D("correlation_cef3_bgo_yPos_singleEle", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_hodo_bgo_xPos_singleEle = new TH2D("correlation_hodo_bgo_xPos_singleEle", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
TH2D* h2_correlation_hodo_bgo_yPos_singleEle = new TH2D("correlation_hodo_bgo_yPos_singleEle", "", 100, -12.5, 12.5, 100, -12.5, 12.5);
int nentries = tree->GetEntries();
if( isOnlyRunNumber ) {
// modify runname in such a way that it's useful for getBeamPosition and outfile:
runName = "BTF_" + runName + "_beam";
}
std::string outputdir = "PosAnTrees_"+tag;
system( Form("mkdir -p %s", outputdir.c_str()) );
std::string outfileName = outputdir + "/PosAn_" + runName + ".root";
TFile* outfile = TFile::Open( outfileName.c_str(), "RECREATE" );
TTree* outTree = new TTree("posTree","posTree");
float xPos_calo_, yPos_calo_;
float xPos_bgo_, yPos_bgo_;
float xPos_new_, yPos_new_;
float xPos_regr2D_, yPos_regr2D_;
float r02_, r13_;
outTree->Branch( "isSingleEle_scintFront", &isSingleEle_scintFront, "isSingleEle_scintFront/O" );
outTree->Branch( "nHodoClustersX", &nHodoClustersX, "nHodoClustersX/I" );
outTree->Branch( "nHodoClustersY", &nHodoClustersY, "nHodoClustersY/I" );
outTree->Branch( "cef3_corr", cef3_corr, "cef3_corr[4]/F" );
outTree->Branch( "r02", &r02_, "r02_/F" );
outTree->Branch( "r13", &r13_, "r13_/F" );
outTree->Branch( "xBeam", &xBeam, "xBeam/F" );
outTree->Branch( "yBeam", &yBeam, "yBeam/F" );
outTree->Branch( "xPos_bgo", &xPos_bgo_, "xPos_bgo_/F" );
outTree->Branch( "yPos_bgo", &yPos_bgo_, "yPos_bgo_/F" );
outTree->Branch( "xPos_calo", &xPos_calo_, "xPos_calo_/F" );
outTree->Branch( "yPos_calo", &yPos_calo_, "yPos_calo_/F" );
outTree->Branch( "xPos_new", &xPos_new_, "xPos_new_/F" );
outTree->Branch( "yPos_new", &yPos_new_, "yPos_new_/F" );
outTree->Branch( "xPos_regr2D", &xPos_regr2D_, "xPos_regr2D_/F" );
outTree->Branch( "yPos_regr2D", &yPos_regr2D_, "yPos_regr2D_/F" );
float diag02_calo_;
float diag13_calo_;
outTree->Branch( "diag02_calo", &diag02_calo_, "diag02_calo_/F" );
outTree->Branch( "diag13_calo", &diag13_calo_, "diag13_calo_/F" );
float diag02_new_;
float diag13_new_;
outTree->Branch( "diag02_new", &diag02_new_, "diag02_new_/F" );
outTree->Branch( "diag13_new", &diag13_new_, "diag13_new_/F" );
float diag02_beam_;
float diag13_beam_;
outTree->Branch( "diag02_beam", &diag02_beam_, "diag02_beam_/F" );
outTree->Branch( "diag13_beam", &diag13_beam_, "diag13_beam_/F" );
std::vector<float> xbgo, ybgo;
for( unsigned i=0; i<BGO_CHANNELS; ++i ) {
float x,y;
RunHelper::getBGOCoordinates( i, x, y );
xbgo.push_back( x );
ybgo.push_back( y );
}
float cef3_regr[CEF3_CHANNELS];
//TMVA::Reader* readerRegrX = new TMVA::Reader( "!Color:!Silent" );
//readerRegrX->AddVariable("cef3_corr[0]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[0] );
//readerRegrX->AddVariable("cef3_corr[1]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[1] );
//readerRegrX->AddVariable("cef3_corr[2]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[2] );
//readerRegrX->AddVariable("cef3_corr[3]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[3] );
//TMVA::Reader* readerRegrY = new TMVA::Reader( "!Color:!Silent" );
//readerRegrY->AddVariable("cef3_corr[0]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[0] );
//readerRegrY->AddVariable("cef3_corr[1]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[1] );
//readerRegrY->AddVariable("cef3_corr[2]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[2] );
//readerRegrY->AddVariable("cef3_corr[3]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3])", &cef3_regr[3] ); // let try this trick
//TMVA::Reader* readerRegr2D = new TMVA::Reader( "!Color:!Silent" );
//readerRegr2D->AddVariable("cef3_corr[0]", &cef3_corr_[0] );
//readerRegr2D->AddVariable("cef3_corr[1]", &cef3_corr_[1] );
//readerRegr2D->AddVariable("cef3_corr[2]", &cef3_corr_[2] );
//readerRegr2D->AddVariable("cef3_corr[3]", &cef3_corr_[3] );
//readerRegr2D->BookMVA( "MLP", "TMVA/weights/TMVARegression_MLP.weights.xml" );
std::vector<std::string> methodNames;
//methodNames.push_back("BDTG");
////methodNames.push_back("FDA_MT");
//methodNames.push_back("LD");
//methodNames.push_back("MLP");
////methodNames.push_back("PDERS");
//
//std::cout << "-> Booking TMVA Reader" << std::endl;
//for( unsigned i=0; i<methodNames.size(); ++i ) {
// readerRegrX->BookMVA( methodNames[i], Form("TMVA/weights/TMVARegression_%s.weights.xml", methodNames[i].c_str()) );
// readerRegrY->BookMVA( methodNames[i], Form("TMVA/weights/TMVARegression_%s.weights.xml", methodNames[i].c_str()) );
//}
std::vector< TH1D* > h1_xPos_regr_vs_calo;
std::vector< TH1D* > h1_yPos_regr_vs_calo;
std::vector< TH2D* > h2_xyPos_regr;
std::vector< TH1D* > h1_xPos_regr_vs_calo_singleEle;
std::vector< TH1D* > h1_yPos_regr_vs_calo_singleEle;
std::vector< TH2D* > h2_xyPos_singleEle_regr;
for( unsigned i=0; i<methodNames.size(); ++i ) {
TH1D* newHistx = new TH1D( Form("xPos_regr%s_vs_calo", methodNames[i].c_str()), "", nBins, -xMax, xMax);
h1_xPos_regr_vs_calo.push_back( newHistx );
TH1D* newHisty = new TH1D( Form("yPos_regr%s_vs_calo", methodNames[i].c_str()), "", nBins, -xMax, xMax);
h1_yPos_regr_vs_calo.push_back( newHisty );
TH2D* newHistxy = new TH2D( Form("xyPos_regr%s", methodNames[i].c_str()), "", nBins, -xMax, xMax, nBins, -xMax, xMax);
h2_xyPos_regr.push_back( newHistxy );
TH1D* newHistx_singleEle = new TH1D( Form("xPos_regr%s_vs_calo_singleEle", methodNames[i].c_str()), "", nBins, -xMax, xMax);
h1_xPos_regr_vs_calo_singleEle.push_back( newHistx_singleEle );
TH1D* newHisty_singleEle = new TH1D( Form("yPos_regr%s_vs_calo_singleEle", methodNames[i].c_str()), "", nBins, -xMax, xMax);
h1_yPos_regr_vs_calo_singleEle.push_back( newHisty_singleEle );
TH2D* newHistxy_singleEle = new TH2D( Form("xyPos_singleEle_regr%s", methodNames[i].c_str()), "", nBins, -xMax, xMax, nBins, -xMax, xMax);
h2_xyPos_singleEle_regr.push_back( newHistxy_singleEle );
}
TH2D* h2_xyPos_regr2D = new TH2D("xyPos_regr2D", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
TH2D* h2_xyPos_singleEle_regr2D = new TH2D("xyPos_singleEle_regr2D", "", nBins, -xMax, xMax, nBins, -xMax, xMax);
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
xPos_bgo_ = -999.;
yPos_bgo_ = -999.;
xPos_calo_ = -999.;
yPos_calo_ = -999.;
tree->GetEntry(iEntry);
if( iEntry % 5000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
if( !bgo_corr_ok ) continue;
r02_ = cef3_corr[0]/cef3_corr[2];
r13_ = cef3_corr[1]/cef3_corr[3];
// FIRST GET POSITION FROM HODOSCOPE:
float xPos_hodo = getMeanposHodo(nHodoClustersX, pos_hodoClustX);
float yPos_hodo = getMeanposHodo(nHodoClustersY, pos_hodoClustY);
if( xPos_hodo>-100. )
h1_xPos_hodo->Fill(xPos_hodo);
if( yPos_hodo>-100. )
h1_yPos_hodo->Fill(yPos_hodo);
if( xPos_hodo>-100. && yPos_hodo>-100. )
h2_xyPos_hodo->Fill(xPos_hodo, yPos_hodo);
if( isSingleEle_scintFront ) {
if( xPos_hodo>-100. )
h1_xPos_singleEle_hodo->Fill(xPos_hodo);
if( yPos_hodo>-100. )
h1_yPos_singleEle_hodo->Fill(yPos_hodo);
if( xPos_hodo>-100. && yPos_hodo>-100. )
h2_xyPos_singleEle_hodo->Fill(xPos_hodo, yPos_hodo);
}
std::vector<float> xPosW_bgo;
std::vector<float> yPosW_bgo;
std::vector<float> v_bgo_corr;
for( unsigned i=0; i<BGO_CHANNELS; ++i ) v_bgo_corr.push_back(bgo_corr[i]);
float eTot_bgo_corr = sumVector(v_bgo_corr);
if( bgo_ok && bgo_corr_ok ) {
// 0 1 2
// 3 4
// 5 6 7
xPosW_bgo.push_back(bgo_corr[0]*xbgo[0]);
xPosW_bgo.push_back(bgo_corr[1]*xbgo[1]);
xPosW_bgo.push_back(bgo_corr[2]*xbgo[2]);
xPosW_bgo.push_back(bgo_corr[3]*xbgo[3]);
xPosW_bgo.push_back(bgo_corr[4]*xbgo[4]);
xPosW_bgo.push_back(bgo_corr[5]*xbgo[5]);
xPosW_bgo.push_back(bgo_corr[6]*xbgo[6]);
xPosW_bgo.push_back(bgo_corr[7]*xbgo[7]);
yPosW_bgo.push_back(bgo_corr[0]*ybgo[0]);
yPosW_bgo.push_back(bgo_corr[1]*ybgo[1]);
yPosW_bgo.push_back(bgo_corr[2]*ybgo[2]);
yPosW_bgo.push_back(bgo_corr[3]*ybgo[3]);
yPosW_bgo.push_back(bgo_corr[4]*ybgo[4]);
yPosW_bgo.push_back(bgo_corr[5]*ybgo[5]);
yPosW_bgo.push_back(bgo_corr[6]*ybgo[6]);
yPosW_bgo.push_back(bgo_corr[7]*ybgo[7]);
xPos_bgo_ = sumVector( xPosW_bgo )/eTot_bgo_corr;
yPos_bgo_ = sumVector( yPosW_bgo )/eTot_bgo_corr;
h1_xPos_bgo->Fill( xPos_bgo_ );
h1_yPos_bgo->Fill( yPos_bgo_ );
h2_xyPos_bgo->Fill( xPos_bgo_, yPos_bgo_ );
h2_correlation_hodo_bgo_xPos->Fill( xPos_hodo, xPos_bgo_ );
h2_correlation_hodo_bgo_yPos->Fill( yPos_hodo, yPos_bgo_ );
if( isSingleEle_scintFront ) {
h1_xPos_singleEle_bgo->Fill( xPos_bgo_ );
h1_yPos_singleEle_bgo->Fill( yPos_bgo_ );
h2_xyPos_singleEle_bgo->Fill( xPos_bgo_, yPos_bgo_ );
h2_correlation_hodo_bgo_xPos_singleEle->Fill( xPos_hodo, xPos_bgo_ );
h2_correlation_hodo_bgo_yPos_singleEle->Fill( yPos_hodo, yPos_bgo_ );
}
} // if bgo ok
// THEN USE CeF3 DATA:
if( cef3_ok ) {
std::vector<float> v_cef3_corr;
for(unsigned i=0; i<CEF3_CHANNELS; ++i) v_cef3_corr.push_back(cef3_corr[i]);
float eTot_corr = sumVector(v_cef3_corr);
if( cef3_corr_ok ) {
// 0 1
//
//
// 3 2
float position = 12. - 0.696; // using FN's infallible trigonometry
//std::vector
std::vector<float> xPosW;
xPosW.push_back(cef3_corr[0]*(-position));
xPosW.push_back(cef3_corr[1]*(+position));
xPosW.push_back(cef3_corr[2]*(+position));
xPosW.push_back(cef3_corr[3]*(-position));
std::vector<float> yPosW;
yPosW.push_back(cef3_corr[0]*(+position));
yPosW.push_back(cef3_corr[1]*(+position));
yPosW.push_back(cef3_corr[2]*(-position));
yPosW.push_back(cef3_corr[3]*(-position));
getCeF3Position( v_cef3_corr, xPos_new_, yPos_new_ );
//diag02_new_ = xPos_new_;
//diag13_new_ = yPos_new_;
float pi = 3.14159;
float theta = pi/4.; // 45 degrees
TVector2 vnew( xPos_new_, yPos_new_ );
TVector2 dnew = vnew.Rotate(-theta);
diag02_new_ = dnew.Y();
diag13_new_ = dnew.X();
TVector2 vBeam( xBeam, yBeam );
TVector2 dBeam = vBeam.Rotate(-theta);
diag02_beam_ = dBeam.Y();
diag13_beam_ = dBeam.X();
float xPos = sumVector(xPosW)/eTot_corr;
float yPos = sumVector(yPosW)/eTot_corr;
h1_xPos->Fill( xPos );
h1_yPos->Fill( yPos );
h2_xyPos->Fill( xPos, yPos );
h1_xPos_new->Fill( xPos_new_ );
h1_yPos_new->Fill( yPos_new_ );
h2_xyPos_new->Fill( xPos_new_, yPos_new_ );
// positioning with all 9 calorimeter channels:
//float xPos_calo = sumVector( xPosW_bgo )/(eTot_bgo_corr + eTot_corr*0.07); // cef3 is in 0,0
//float yPos_calo = sumVector( yPosW_bgo )/(eTot_bgo_corr + eTot_corr*0.08); // so counts only in denominator
xPos_calo_ = sumVector( xPosW_bgo )/(eTot_bgo_corr + eTot_corr*0.06); // cef3 is in 0,0
yPos_calo_ = sumVector( yPosW_bgo )/(eTot_bgo_corr + eTot_corr*0.10); // so counts only in denominator
//float xPos_calo = sumVector( xPosW_bgo )/(eTot_bgo_corr + eTot_corr*0.791577); // cef3 is in 0,0
//float yPos_calo = sumVector( yPosW_bgo )/(eTot_bgo_corr + eTot_corr*0.791577); // so counts only in denominator
TVector2 vcalo( xPos_calo_, yPos_calo_ );
TVector2 dcalo = vcalo.Rotate(-theta);
diag02_calo_ = dcalo.Y();
diag13_calo_ = dcalo.X();
//xPos_regr2D_ = readerRegr2D->EvaluateRegression( "MLP" )[0];
//yPos_regr2D_ = readerRegr2D->EvaluateRegression( "MLP" )[1];
cef3_regr[0] = cef3_corr[0]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3]);
cef3_regr[1] = cef3_corr[1]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3]);
cef3_regr[2] = cef3_corr[2]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3]);
cef3_regr[3] = cef3_corr[3]/(cef3_corr[0]+cef3_corr[1]+cef3_corr[2]+cef3_corr[3]);
if( bgo_ok && bgo_corr_ok ) {
h1_xPos_calo->Fill( xPos_calo_ );
h1_yPos_calo->Fill( yPos_calo_ );
h2_xyPos_calo->Fill( xPos_calo_, yPos_calo_ );
//for( unsigned i=0; i<methodNames.size(); ++i ) {
// Float_t xPos_regr = (readerRegrX->EvaluateRegression( methodNames[i] ))[0];
// Float_t yPos_regr = (readerRegrY->EvaluateRegression( methodNames[i] ))[0];
// h1_xPos_regr_vs_calo[i]->Fill( xPos_regr-xPos_calo_ );
// h1_yPos_regr_vs_calo[i]->Fill( yPos_regr-yPos_calo_ );
// h2_xyPos_regr[i]->Fill( xPos_regr, yPos_regr );
//}
//h2_xyPos_regr2D->Fill( xPos_regr2D_, yPos_regr2D_ );
h1_xPos_calo_vs_hodo->Fill( xPos_calo_-xPos_hodo );
h1_yPos_calo_vs_hodo->Fill( yPos_calo_-yPos_hodo );
h1_xPos_calo_vs_beam->Fill( xPos_calo_-xBeam );
h1_yPos_calo_vs_beam->Fill( yPos_calo_-yBeam );
// CORRELATIONS BETWEEN CALO AND HODO:
h2_correlation_cef3_bgo_xPos->Fill( xPos, xPos_bgo_ );
h2_correlation_cef3_bgo_yPos->Fill( yPos, yPos_bgo_ );
}
if( isSingleEle_scintFront ) {
h1_xPos_singleEle->Fill( xPos );
h1_yPos_singleEle->Fill( yPos );
h2_xyPos_singleEle->Fill( xPos, yPos );
h1_xPos_new_singleEle->Fill( xPos_new_ );
h1_yPos_new_singleEle->Fill( yPos_new_ );
h2_xyPos_new_singleEle->Fill( xPos_new_, yPos_new_ );
h1_xPos_calo_vs_hodo_singleElectron->Fill( xPos_calo_-xPos_hodo );
h1_yPos_calo_vs_hodo_singleElectron->Fill( yPos_calo_-yPos_hodo );
h1_xPos_calo_vs_beam_singleElectron->Fill( xPos_calo_-xBeam );
h1_yPos_calo_vs_beam_singleElectron->Fill( yPos_calo_-yBeam );
if( nHodoClustersX==1 && nHodoClustersY==1 && nFibres_hodoClustX[0]<=2 && nFibres_hodoClustY[0]<=2 ) {
h1_xPos_calo_vs_hodo_singleElectron_HR->Fill( xPos_calo_-xPos_hodo );
h1_yPos_calo_vs_hodo_singleElectron_HR->Fill( yPos_calo_-yPos_hodo );
h1_xPos_calo_vs_beam_singleElectron_HR->Fill( xPos_calo_-xBeam );
h1_yPos_calo_vs_beam_singleElectron_HR->Fill( yPos_calo_-yBeam );
}
h2_correlation_cef3_hodo_xPos_singleEle->Fill( xPos, xPos_hodo );
h2_correlation_cef3_hodo_yPos_singleEle->Fill( yPos, yPos_hodo );
if( bgo_ok && bgo_corr_ok ) {
h1_xPos_singleEle_calo->Fill( xPos_calo_ );
h1_yPos_singleEle_calo->Fill( yPos_calo_ );
h2_xyPos_singleEle_calo->Fill( xPos_calo_, yPos_calo_ );
//for( unsigned i=0; i<methodNames.size(); ++i ) {
// Float_t xPos_regr = (readerRegrX->EvaluateRegression( methodNames[i] ))[0];
// Float_t yPos_regr = (readerRegrY->EvaluateRegression( methodNames[i] ))[0];
// h1_xPos_regr_vs_calo_singleEle[i]->Fill( xPos_regr-xPos_calo_ );
// h1_yPos_regr_vs_calo_singleEle[i]->Fill( yPos_regr-yPos_calo_ );
// h2_xyPos_singleEle_regr[i]->Fill( xPos_regr, yPos_regr );
//}
//h2_xyPos_singleEle_regr2D->Fill( xPos_regr2D_, yPos_regr2D_ );
h2_correlation_cef3_bgo_xPos_singleEle->Fill( xPos, xPos_bgo_ );
h2_correlation_cef3_bgo_yPos_singleEle->Fill( yPos, yPos_bgo_ );
}
}
outTree->Fill();
} // if cef3_ok
}
}
outfile->cd();
outTree->Write();
for( unsigned i=0; i<h1_xPos_regr_vs_calo.size(); ++i ) {
h1_xPos_regr_vs_calo[i]->Write();
h1_yPos_regr_vs_calo[i]->Write();
h1_xPos_regr_vs_calo_singleEle[i]->Write();
h1_yPos_regr_vs_calo_singleEle[i]->Write();
h2_xyPos_regr[i]->Write();
h2_xyPos_singleEle_regr[i]->Write();
}
h1_xPos->Write();
h1_yPos->Write();
h2_xyPos->Write();
h1_xPos_singleEle->Write();
h1_yPos_singleEle->Write();
h2_xyPos_singleEle->Write();
h1_xPos_new->Write();
h1_yPos_new->Write();
h2_xyPos_new->Write();
h1_xPos_new_singleEle->Write();
h1_yPos_new_singleEle->Write();
h2_xyPos_new_singleEle->Write();
h1_xPos_bgo->Write();
h1_yPos_bgo->Write();
h2_xyPos_bgo->Write();
h1_xPos_hodo->Write();
h1_yPos_hodo->Write();
h2_xyPos_hodo->Write();
h1_xPos_calo_vs_hodo->Write();
h1_yPos_calo_vs_hodo->Write();
h1_xPos_calo_vs_beam->Write();
h1_yPos_calo_vs_beam->Write();
h1_xPos_calo_vs_hodo_singleElectron->Write();
h1_yPos_calo_vs_hodo_singleElectron->Write();
h1_xPos_calo_vs_beam_singleElectron->Write();
h1_yPos_calo_vs_beam_singleElectron->Write();
h1_xPos_calo_vs_hodo_singleElectron_HR->Write();
h1_yPos_calo_vs_hodo_singleElectron_HR->Write();
h1_xPos_calo_vs_beam_singleElectron_HR->Write();
h1_yPos_calo_vs_beam_singleElectron_HR->Write();
std::cout << std::endl;
h2_correlation_cef3_hodo_xPos->Write();
h2_correlation_cef3_hodo_yPos->Write();
h2_correlation_cef3_bgo_xPos->Write();
h2_correlation_cef3_bgo_yPos->Write();
h2_correlation_hodo_bgo_xPos->Write();
h2_correlation_hodo_bgo_yPos->Write();
h1_xPos_singleEle_bgo->Write();
h1_yPos_singleEle_bgo->Write();
h2_xyPos_singleEle_bgo->Write();
h1_xPos_singleEle_hodo->Write();
h1_yPos_singleEle_hodo->Write();
h2_xyPos_singleEle_hodo->Write();
h1_xPos_singleEle_hodoClust->Write();
h1_yPos_singleEle_hodoClust->Write();
h2_xyPos_singleEle_hodoClust->Write();
h1_xPos_calo->Write();
h1_yPos_calo->Write();
h2_xyPos_calo->Write();
h1_xPos_singleEle_calo->Write();
h1_yPos_singleEle_calo->Write();
h2_xyPos_singleEle_calo->Write();
h2_correlation_cef3_hodo_xPos_singleEle->Write();
h2_correlation_cef3_hodo_yPos_singleEle->Write();
h2_correlation_cef3_bgo_xPos_singleEle->Write();
h2_correlation_cef3_bgo_yPos_singleEle->Write();
h2_correlation_hodo_bgo_xPos_singleEle->Write();
h2_correlation_hodo_bgo_yPos_singleEle->Write();
outfile->Close();
std::cout << "-> Histograms saved in: " << outfile->GetName() << std::endl;
return 0;
}
int main( int argc, char* argv[] ) {
std::string runName = "precalib_BGO_pedestal_noSource";
std::string tag = "default";
if( argc>2 ) {
std::string runName_str(argv[1]);
runName = runName_str;
std::string tag_str(argv[2]);
tag = tag_str;
}else{
std::cout<<"Usage:"<<std::endl;
std::cout<<"./makeSimulationToAnalysisTree BeamEnergy(in MeV) tag"<<std::endl;
exit(12345);
}
TString runName_tstr(runName);
bool isIdeal = false;
//run name = energy for now, might have to change this for different beams...
std::string fileName = "OriginalSimulationData/EEShash" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
if( file==0 ) {
std::string fileName = "OriginalSimulationData/EEShashIdeal_" + runName + ".root";
TFile* file = TFile::Open(fileName.c_str());
isIdeal = true;
}
if( file==0 ) {
std::cout << "ERROR! Din't find file " << fileName << std::endl;
std::cout << "Exiting." << std::endl;
exit(11);
}
TChain* tree = new TChain("EEShash");
tree = (TChain*)file->Get("EEShash");
//std::string outdir = "analysisTrees_" + tag;
std::string outdir = "OriginalSimulationData";
system( Form("mkdir -p %s", outdir.c_str()) );
std::string outfileName;
if(isIdeal==0){ outfileName = outdir + "/Reco_Simulation" + runName + ".root";}
else { outfileName = outdir + "/Reco_Simulation_Ideal_" + runName + ".root";}
TFile* outfile = TFile::Open( outfileName.c_str(), "RECREATE" );
TTree* outTree = new TTree("recoTree","recoTree");
double Eact_0;
tree->SetBranchAddress("Eact_0", &Eact_0);
double Eact_1;
tree->SetBranchAddress("Eact_1", &Eact_1);
double Eact_2;
tree->SetBranchAddress("Eact_2", &Eact_2);
double Eact_3;
tree->SetBranchAddress("Eact_3", &Eact_3);
double Eact_4;
tree->SetBranchAddress("Eact_4", &Eact_4);
double Eact_5;
tree->SetBranchAddress("Eact_5", &Eact_5);
double Eact_6;
tree->SetBranchAddress("Eact_6", &Eact_6);
double Eact_7;
tree->SetBranchAddress("Eact_7", &Eact_7);
double Eact_8;
tree->SetBranchAddress("Eact_8", &Eact_8);
double Eact_9;
tree->SetBranchAddress("Eact_9", &Eact_9);
double Eact_10;
tree->SetBranchAddress("Eact_10", &Eact_10);
double Eact_11;
tree->SetBranchAddress("Eact_11", &Eact_11);
double Eact_12;
tree->SetBranchAddress("Eact_12", &Eact_12);
double Eact_13;
tree->SetBranchAddress("Eact_13", &Eact_13);
double Eact_14;
tree->SetBranchAddress("Eact_14", &Eact_14);
double Ebgo_0;
tree->SetBranchAddress( "Ebgo_0", &Ebgo_0);
double Ebgo_1;
tree->SetBranchAddress( "Ebgo_1", &Ebgo_1);
double Ebgo_2;
tree->SetBranchAddress( "Ebgo_2", &Ebgo_2);
double Ebgo_3;
tree->SetBranchAddress("Ebgo_3", &Ebgo_3);
double Ebgo_4;
tree->SetBranchAddress("Ebgo_4", &Ebgo_4);
double Ebgo_5;
tree->SetBranchAddress("Ebgo_5", &Ebgo_5);
double Ebgo_6;
tree->SetBranchAddress("Ebgo_6", &Ebgo_6);
double Ebgo_7;
tree->SetBranchAddress("Ebgo_7", &Ebgo_7);
double Ebgo;
tree->SetBranchAddress("Ebgo", &Ebgo);
double Eabs;
tree->SetBranchAddress("Eabs", &Eabs);
double Eact;
tree->SetBranchAddress("Eact", &Eact);
double Escint1;
tree->SetBranchAddress("Escint1", &Escint1);
double Ehodo11;
tree->SetBranchAddress("Ehodo11", &Ehodo11);
double Ehodo12;
tree->SetBranchAddress("Ehodo12", &Ehodo12);
double Escint;
tree->SetBranchAddress("Escint", &Escint);
double Ehodo;
tree->SetBranchAddress("Ehodo", &Ehodo);
double xPosition;
tree->SetBranchAddress("xPosition",&xPosition);
double yPosition;
tree->SetBranchAddress("yPosition",&yPosition);
float cef3_corr_[CEF3_CHANNELS],bgo_corr_[BGO_CHANNELS];
bool isSingleEle_scintFront_=1;
int nHodoClustersY=1; int nHodoClustersX=1;
int nHodoClusters1Y=1; int nHodoClusters1X=1;
float E_abs;
float xPos;
float yPos;
int bgo_chan=BGO_CHANNELS;
int cef3_chan=CEF3_CHANNELS;
outTree->Branch( "nHodoClustersX", &nHodoClustersX, "nHodoClustersX/I" );
outTree->Branch( "nHodoClustersY", &nHodoClustersY, "nHodoClustersY/I" );
outTree->Branch( "nHodoClusters1X", &nHodoClusters1X, "nHodoClusters1X/I" );
outTree->Branch( "nHodoClusters1Y", &nHodoClusters1Y, "nHodoClusters1Y/I" );
outTree->Branch( "cef3_chan", &cef3_chan, "cef3_chan/I" );
outTree->Branch( "bgo_chan", &bgo_chan, "bgo_chan/I" );
outTree->Branch( "bgo_corr", bgo_corr_, "bgo_corr_[bgo_chan]/F" );
outTree->Branch( "cef3_corr", cef3_corr_, "cef3_corr_[cef3_chan]/F" );
outTree->Branch( "isSingleEle_scintFront", &isSingleEle_scintFront_, "isSingleEle_scintFront_/O" );
outTree->Branch( "E_abs", &E_abs, "E_abs/F");
outTree->Branch( "xPos", &xPos, "xPos/F");
outTree->Branch( "yPos", &yPos, "yPos/F");
float LYSF[] = {0.85, 0.94, 0.95, 0.98, 1.00, 1.02, 1.05, 1.05, 1.05, 0.74};
// float LYSF[] = {0.85, 0.94, 0.95, 0.98, 1.00, 0.7, 1.05, 1.05, 1.05, 0.74};
// float LYSF[] = {0.87, 0.96, 0.97, 1.00, 1.02, 1.04, 1.07, 1.07, 1.08, 0.75};
//as a test with the best being the best one
// float LYSF[] = {0.81, 0.89, 0.90, 0.93, 0.95, 0.97, 1.00, 1.00, 1.00, 0.70};
//Test if 6.crystal fails
// float LYSFH[] = {0.87, 0.96, 0.97, 1.00, 1.02, 0.50, 1.07, 1.07, 1.08, 1.07, 1.08, 1.09, 1.09, 1.04, 0.75};
///
float LYSFH[] = {0.87, 0.96, 0.97, 1.00, 1.02, 1.04, 1.07, 1.07, 1.08, 1.07, 1.08, 1.09, 1.09, 1.04, 0.75};
/*
std::string fullVarName = "";
for( unsigned i=0; i<15; ++i ) {
std::string plusSign = (i==0) ? "" : " + ";
std::string thisPiece(Form("%s%f*Eact_%d", plusSign.c_str(), LYSFH[i], i));
fullVarName += thisPiece;
}
*/
unsigned nentries = tree->GetEntries();
double hodoEff;
double scintEff;
for( unsigned iEntry=0; iEntry<nentries; ++iEntry ) {
tree->GetEntry(iEntry);
if( iEntry % 5000 == 0 ) std::cout << "Entry: " << iEntry << " / " << nentries << std::endl;
xPos = xPosition;
yPos = yPosition;
if(Ehodo11>0.1){
nHodoClustersX=1;
hodoEff++;
} else{ nHodoClusters1X=0;}
if(Ehodo12>0.1){
nHodoClustersY=1;
hodoEff++;
} else{ nHodoClusters1Y=0; }
if(Escint1>0.25){
isSingleEle_scintFront_=1;
scintEff++;
} else{ isSingleEle_scintFront_=0;}
if(Ehodo>0.1){
nHodoClustersX=1; nHodoClustersY=1;
hodoEff++;
} else{ nHodoClustersX=0; nHodoClustersY=0;}
/*
if(Escint>0.25 && Escint< 1.5){
isSingleEle_scintFront_=1;
scintEff++;
} else{ isSingleEle_scintFront_=0;}
*/
E_abs= Eabs;
std::vector<float> cef3_corr;
for( int i=0; i<CEF3_CHANNELS; ++i ) cef3_corr.push_back(-1.);
std::vector<float> bgo_corr;
for( int i=0; i<BGO_CHANNELS; ++i ) bgo_corr.push_back(-1.);
/*
for(int i=0; i<10; ++i){
cef3_corr[0]= (LYSF[0]*Eact_0+LYSF[1]*Eact_1+LYSF[2]*Eact_2 +LYSF[3]*Eact_3 +LYSF[4]*Eact_4 +LYSF[5]*Eact_5 +LYSF[6]*Eact_6 +LYSF[7]*Eact_7 +LYSF[8]*Eact_8 +LYSF[9]*Eact_9)/4. ; //yes not very elegant, but it didn't want to work otherwise
// cef3_corr[0] = Eact /4.;
}
*/
for(int i=0; i<15; ++i){
cef3_corr[0]= (LYSFH[0]*Eact_0+LYSFH[1]*Eact_1+LYSFH[2]*Eact_2 +LYSFH[3]*Eact_3 +LYSFH[4]*Eact_4 +LYSFH[5]*Eact_5 +LYSFH[6]*Eact_6 +LYSFH[7]*Eact_7 +LYSFH[8]*Eact_8 +LYSFH[9]*Eact_9 +LYSFH[10]*Eact_10 +LYSFH[11]*Eact_11 +LYSFH[12]*Eact_12 +LYSFH[13]*Eact_13 +LYSFH[14]*Eact_14 )/4. ; //yes not very elegant, but it didn't want to work otherwise
// cef3_corr[0] = Eact /4.;
}
bgo_corr[0] = Ebgo/8.;
bgo_corr[1] = Ebgo/8.;
bgo_corr[2] = Ebgo/8.;
bgo_corr[3] = Ebgo/8.;
bgo_corr[4] = Ebgo/8.;
bgo_corr[5] = Ebgo/8.;
bgo_corr[6] = Ebgo/8.;
bgo_corr[7] = Ebgo/8.;
for(int k=0; k<CEF3_CHANNELS; ++k){
cef3_corr_[k] = cef3_corr[0];
}
for (int k=0; k<BGO_CHANNELS; ++k){
bgo_corr_[k] = bgo_corr[k];
}
outTree->Fill();
}
outfile->cd();
outTree->Write();
outfile->Close();
std::cout << "-> Simulated Analysis Tree saved in: " << outfile->GetName() << std::endl;
std::cout << "Scintillator efficiency = " << scintEff/nentries << std::endl;
std::cout << "Hodoscope efficiency = " << hodoEff/nentries << std::endl;
return 0;
}
