|
|
/// to do:
/// 1. correct for gap between arrays
#define convert_cxx
#include "hitutils.h"
#include <string>
#include <stdio.h>
#include <iostream>
#include <vector>
#include <utility>
#include <TH2.h>
#include <TF1.h>
#include <TStyle.h>
#include <TCanvas.h>
#include <TFile.h>
#include <TTree.h>
#include <TSystemDirectory.h>
#include <gsl/gsl_statistics.h>
#include <math.h>
#include <gsl/gsl_errno.h>
#include <gsl/gsl_fft_complex.h>
#include <TF1.h>
#include <TGraphErrors.h>
#include <gsl/gsl_sort.h>
#include <TVector.h>
#include "TStopwatch.h"
#include "Math/MinimizerOptions.h"
#include "TVirtualFitter.h"
#include "TMatrixD.h"
//using namespace std;
bool smooth_on = true;bool doFit = false;bool doLinInt = true;bool ethercat = true;
Double_t S[128], T[128], SumT, SumS, SumS2, SumST, SumT2, SumY, SumYS, SumYT, sigmaABC, muABC,p,c, b, b_den, b_num, SumYYP, SumYYM, MeanY;TMatrixD M1(3,3);TMatrixD M1inv(3,3);TVectorD ABC(3);TVectorD M2(3);Double_t M1elements_b0[9] = {0.};Double_t M1elements_b1[9] = {0.};Double_t M1elements_b2[9] = {0.};Double_t M1elements_b3[9] = {0.};
TMatrixD matrix(2,2);TMatrixD *M1pointer = &M1;TVectorD *M2pointer = &M2;TVectorD *ABCpointer = &ABC;
int eventID = 0;int framestart=0;int frameend = 0;double board_b0_ch[128];double board_b1_ch[128];double board_b2_ch[128];double board_b3_ch[128];double board_b0_ch_bkg[128];double board_b1_ch_bkg[128];double board_b2_ch_bkg[128];double board_b3_ch_bkg[128];double signal_b0 = 0.;double signal_b1 = 0.;double signal_b2 = 0.;double signal_b3 = 0.;
double signal_cor_b0 = 0.;double signal_cor_b1 = 0.;double signal_cor_b2 = 0.;double signal_cor_b3 = 0.;
double signal_b0_left = 0.;double signal_b1_left = 0.;double signal_b2_left = 0.;double signal_b3_left = 0.;
double signal_b0_right = 0.;double signal_b1_right = 0.;double signal_b2_right = 0.;double signal_b3_right = 0.;
double signal_gsl_b0[128];double signal_gsl_b1[128];double signal_gsl_b2[128];double signal_gsl_b3[128];double signal_reg[128];
double channellist_gsl_b0[128];double channellist_gsl_b1[128];double channellist_gsl_b2[128];double channellist_gsl_b3[128];double channellist[128];double channellist_reg[128];
double pos[128];
double maxchannelamp_b0 =0.;double maxchannelamp_b1 =0.;double maxchannelamp_b2 =0.;double maxchannelamp_b3 =0.;int maxchannel_b0 =0;int maxchannel_b1 =0;int maxchannel_b2 =0;int maxchannel_b3 =0;int numtocalc_b0 =0;int numtocalc_b1 =0;int numtocalc_b2 =0;int numtocalc_b3 =0;int numtocalc_reg =0;
int nfwhm_b0 = 0;int nfwhm_b1 = 0;int nfwhm_b2 = 0;int nfwhm_b3 = 0;
double beamPosX_b0,beamPosX_b1,beamPosX_b2,beamPosX_b3;double beamFocusX_b0,beamFocusX_b1,beamFocusX_b2,beamFocusX_b3;double beamPosX_fit_b0,beamPosX_fit_b1,beamPosX_fit_b2,beamPosX_fit_b3;double beamFocusX_fit_b0,beamFocusX_fit_b1,beamFocusX_fit_b2,beamFocusX_fit_b3; double beamSkewX_b0,beamSkewX_b1,beamSkewX_b2,beamSkewX_b3;double beamKurtX_b0,beamKurtX_b1,beamKurtX_b2,beamKurtX_b3; int beamNumX_b0,beamNumX_b1,beamNumX_b2,beamNumX_b3;double beamRsqr_b0,beamRsqr_b1,beamRsqr_b2,beamRsqr_b3;double beamChi2_fit_b0,beamChi2_fit_b1,beamChi2_fit_b2,beamChi2_fit_b3;
double beamPosX_reg_b0,beamPosX_reg_b1,beamPosX_reg_b2,beamPosX_reg_b3;double beamFocusX_reg_b0,beamFocusX_reg_b1,beamFocusX_reg_b2,beamFocusX_reg_b3; double beamPeakX_reg_b0,beamPeakX_reg_b1,beamPeakX_reg_b2,beamPeakX_reg_b3;
double beamSidebandNoise_b0, beamSidebandNoise_b1, beamSidebandNoise_b2, beamSidebandNoise_b3 ;int sidenumtocalc_b0,sidenumtocalc_b1,sidenumtocalc_b2,sidenumtocalc_b3;size_t size = 5;
const int length = 100; //length of the rolling average
double array_b0[length] = {0.};double array_b1[length] = {0.};double array_b2[length] = {0.};double array_b3[length] = {0.};
double arrayavg_b0 = 0.;double arrayavg_b1 = 0.;double arrayavg_b2 = 0.;double arrayavg_b3 = 0.;
bool graphsaved_b0 = false;bool graphsaved_b1 = false;bool graphsaved_b2 = false;bool graphsaved_b3 = false;
TVector beamontime(0,3,0.,0.,0.,0.,"END");TVector * beamontime_ptr = &beamontime;
double calibration_b0[128] = {0.};double calibration_b1[128] = {0.};double calibration_b2[128] = {0.};double calibration_b3[128] = {0.};
TF1 * gausfunc_b0 = new TF1("gausfunc_b0","gaus(0)+[3]");TF1 * gausfunc_b1 = new TF1("gausfunc_b1","gaus(0)+[3]");
TGraphErrors * gausgraph_b0;TGraphErrors * gausgraph_b1;Int_t lastfit_b0 = 0;Int_t lastfit_b1 = 0;double errorx[128];double errory[128];
bool samplenoise = false;bool beam_off = true;bool beam_wason = false;int beamoffcounter = 0;int beamoncounter = 0;
Double_t mybeta = 0.;Double_t lorentz_gamma = 0.;Double_t zsqr = 0.;Double_t sp_air = 0.;Double_t sp_ps = 0.;Double_t sp2_air = 0.;Double_t sp2_ps = 0.;Double_t intcorr = 1.;
Double_t x, q, median;
Double_t signaloffset_b0[30];Double_t signaloffset_b1[30];Double_t signaloffset_b2[30];Double_t signaloffset_b3[30];
TF1 * f_sp_air = new TF1("f_sp_air","[0]*pow(x,[1])+[2]", 50, 250); //stopping power of protons in Air [MeV cm2 /g]
TF1 * f_sp_ps = new TF1("f_sp_ps","[0]*pow(x,[1])+[2]", 50, 250); //stopping power of protons in polystyrene [MeV cm2 /g]
TF1 * f_h2sp_air = new TF1("f_h2sp_air","[0]*pow(x,[1])+[2]", 50, 250); //stopping power of helium in air [MeV cm2 /g]
TF1 * f_h2sp_ps = new TF1("f_h2sp_ps","[0]*pow(x,[1])+[2]", 50, 250); //stopping power of helium in polystyrene [MeV cm2 /g]
TF1 * f_c12sp_air = new TF1("f_c12sp_air","[0]*pow(x,[1])+[2]", 80, 480); //stopping power of carbon in air [MeV cm2 /g]
TF1 * f_c12sp_ps = new TF1("f_c12sp_ps","[0]*pow(x,[1])+[2]", 80, 480); //stopping power of carbon in polystyrene [MeV cm2 /g]
TF1 * f_o16sp_air = new TF1("f_o16sp_air","[0]*pow(x,[1])+[2]", 80, 480); //stopping power of oxygen in air [MeV cm2 /g]
TF1 * f_o16sp_ps = new TF1("f_o16sp_ps","[0]*pow(x,[1])+[2]", 80,480); //stopping power of oxygen in polystyrene [MeV cm2 /g]
///// ic1/SP intensity correction factor = 19.8E6+/-0.1E6 particles/s per nA/(Mevcm2/g)
/// compile with:
//// $ make clean; make
/// run with:
//// $ ./convert <path to data> <run>
//// $ ./convert /work/leverington/beamprofilemonitor/hitdata/HIT_17_12_2017/ run2
int analyse(int argc, char **argv){ f_sp_air->SetParameters(303.746, -0.873697,1.01335); //protons between 50 and 300 MeV
f_sp_ps->SetParameters(361.936, -0.892255, 1.19133); //protons between 50 and 220 MeV
f_h2sp_air->SetParameters(1236.51, -0.880225, 4.17041); //helium between 50 and 300 MeV
f_h2sp_ps->SetParameters(1387.08, -0.882686,4.60833); //heelium between 50 and 300 MeV
f_c12sp_air->SetParameters(13291.2, -0.925464,45.3876); //carbon between 80 and 480 MeV
f_c12sp_ps->SetParameters(14538.9, -0.922423,49.2859); //carbon between 80 and 480 MeV
f_o16sp_air->SetParameters(24624.9, -0.925425,80.6828); //oxygen between 80 and 480 MeV
f_o16sp_ps->SetParameters(26465.6, -0.928309,88.6728); //oxygen between 80 and 480 MeV
// TVirtualFitter::SetDefaultFitter("Minuit2");
// TVirtualFitter::SetPrecision(0.1);
// std::cout << "Default Fitter:" << TVirtualFitter::GetDefaultFitter() << std::endl;
//initialise some values;
for (int i = 0;i<128;i++){ board_b0_ch_bkg[i] = 0.; board_b1_ch_bkg[i] = 0.; board_b2_ch_bkg[i] = 0.; board_b3_ch_bkg[i] = 0.; channellist_gsl_b0[i] = 0; channellist_gsl_b1[i] = 0; channellist_gsl_b2[i] = 0; channellist_gsl_b3[i] = 0; calibration_b0[i] = 1.0; calibration_b1[i] = 1.0; calibration_b2[i] = 1.0; calibration_b3[i] = 1.0; errorx[i]=17.; errory[i]=17.;
if (i<64) {pos[i] = double(i);} else {pos[i] = double(i) + 0.2; } } if (argc > 3){ //open the file names specified in the command line
char * ethercatfile = argv[3]; // argv[1]= "/work/leverington/beamprofilemonitor/hitdata/HIT_17_12_2017/";
// argv[2]= "Run11"
char * filename = Form("%s%s.dat",argv[1],argv[2]); char * offsetfilename = Form("%s",argv[4]); char * timestampfilename = Form("%s%s_timestamp.csv",argv[1],argv[2]); char * rootfilename = Form("%s/root/%s.root",argv[1],argv[2]); ifstream file(filename, ifstream::in | ifstream::binary); ifstream timestampfile(timestampfilename, ifstream::in); ifstream offsetfile(offsetfilename, ifstream::in);
if (!file.is_open()){ printf(".dat did not open.\n"); return -1; //file could not be opened
} else { printf("Data file opened.\n"); } if (!timestampfile.is_open()){ printf("timestamp.csv did not open.\n"); ethercat = false; //return -1; //file could not be opened
} if (!offsetfile.is_open()){ printf("no offset.txt file found\n"); // return -1; //file could not be opened
for (int ii = 0; ii<30;ii++) { signaloffset_b0[ii] = 0.; signaloffset_b1[ii] = 0.; signaloffset_b2[ii] = 0.; signaloffset_b3[ii] = 0.; } } else { for (int ii = 0; ii<30;ii++) offsetfile >> signaloffset_b0[ii] >> signaloffset_b1[ii] >>signaloffset_b2[ii] >>signaloffset_b3[ii] ; } ///read in offsets produced by offsetcor.C script (needs to be run after convert_clean, then repeat convert_clean.)
///some variables to get the data
// int number_of_records_to_read = 4*10;
// BufferData* buffer = new BufferData[number_of_records_to_read];
//if(argc > 4) {
// framestart = atoi(argv[3]);
// frameend = atoi(argv[4]);
// }
int board_b = -1; long int fileframesize = getFileSize(filename) / ( 4*sizeof(BufferData) ); BufferData* dataptr = new BufferData(); if (fileframesize>0){ std::cout << "Number of frames:" << fileframesize << std::endl; //open ROOT file for saving histos and TTree.
TFile *rootFile = new TFile(rootfilename,"RECREATE"); if ( rootFile->IsOpen() ) printf("ROOT file opened successfully\n"); TH2D * th2_signal_vs_channel_b0 = new TH2D("th2_signal_vs_channel_b0","th2_signal_vs_channel_b0",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_b1 = new TH2D("th2_signal_vs_channel_b1","th2_signal_vs_channel_b1",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_b2 = new TH2D("th2_signal_vs_channel_b2","th2_signal_vs_channel_b2",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_b3 = new TH2D("th2_signal_vs_channel_b3","th2_signal_vs_channel_b3",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_sub_b0 = new TH2D("th2_signal_vs_channel_sub_b0","th2_signal_vs_channel_sub_b0",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_sub_b1 = new TH2D("th2_signal_vs_channel_sub_b1","th2_signal_vs_channel_sub_b1",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_sub_b2 = new TH2D("th2_signal_vs_channel_sub_b2","th2_signal_vs_channel_sub_b2",128,0,128,2200,-1000,10000); TH2D * th2_signal_vs_channel_sub_b3 = new TH2D("th2_signal_vs_channel_sub_b3","th2_signal_vs_channel_sub_b3",128,0,128,2200,-1000,10000); TH1D * th1_signal_b0 = new TH1D("th1_signal_b0","th1_signal_b0",2200,-10000,50000); TH1D * th1_signal_b1 = new TH1D("th1_signal_b1","th1_signal_b1",2200,-10000,50000); TH1D * th1_signal_b2 = new TH1D("th1_signal_b2","th1_signal_b2",2200,-10000,50000); TH1D * th1_signal_b3 = new TH1D("th1_signal_b3","th1_signal_b3",2200,-10000,50000); TH1D * h_regresidual_b0 = new TH1D("h_regresidual_b0","h_regresidual_b0", 200,-.20,.20); TH1D * h_regresidual_b1 = new TH1D("h_regresidual_b1","h_regresidual_b1", 200,-.20,.20); TH1D * h_regresidual_b2 = new TH1D("h_regresidual_b2","h_regresidual_b2", 200,-.20,.20); TH1D * h_regresidual_b3 = new TH1D("h_regresidual_b3","h_regresidual_b3", 200,-.20,.20);
TGraph * graph_bkg_b0 = new TGraph(); TGraph * graph_bkg_b1 = new TGraph(); TGraph * graph_bkg_b2 = new TGraph(); TGraph * graph_bkg_b3 = new TGraph(); TGraph * gr_b0; TGraph * gr_sm_b0; TGraph * gr_b1; TGraph * gr_sm_b1; TGraph * gr_b2; TGraph * gr_sm_b2; TGraph * gr_b3; TGraph * gr_sm_b3;
TH2D * th2_signal_vs_channel_bkg_b0 = new TH2D("th2_signal_vs_channel_bkg_b0","th2_signal_vs_channel_bkg_b0",128,0,128,1000,-500,500); TH2D * th2_signal_vs_channel_bkg_b1 = new TH2D("th2_signal_vs_channel_bkg_b1","th2_signal_vs_channel_bkg_b1",128,0,128,1000,-500,500); TH2D * th2_signal_vs_channel_bkg_b2 = new TH2D("th2_signal_vs_channel_bkg_b2","th2_signal_vs_channel_bkg_b2",128,0,128,1000,-500,500); TH2D * th2_signal_vs_channel_bkg_b3 = new TH2D("th2_signal_vs_channel_bkg_b3","th2_signal_vs_channel_bkg_b3",128,0,128,1000,-500,500); TH2D * th2_resid_vs_channel_bkg_b0 = new TH2D("th2_resid_vs_channel_bkg_b0","th2_resid_vs_channel_bkg_b0",128,0,128,300,-.15,.15); TH2D * th2_resid_vs_channel_bkg_b1 = new TH2D("th2_resid_vs_channel_bkg_b1","th2_resid_vs_channel_bkg_b1",128,0,128,300,-.15,.15); TH2D * th2_resid_vs_channel_bkg_b2 = new TH2D("th2_resid_vs_channel_bkg_b2","th2_resid_vs_channel_bkg_b2",128,0,128,300,-.15,.15); TH2D * th2_resid_vs_channel_bkg_b3 = new TH2D("th2_resid_vs_channel_bkg_b3","th2_resid_vs_channel_bkg_b3",128,0,128,300,-.15,.15);
TH1D* h_beamSignal_b0[30]; TH1D* h_beamSignal_b1[30]; TH1D* h_beamSignal_b2[30]; TH1D* h_beamSignal_b3[30]; int energy_indexbin = 0; char histname[30] = ""; TH1D* h_beamSignal_ic1[30];
TH1D* h_beamSignal_cor_b0[30]; TH1D* h_beamSignal_cor_b1[30]; TH1D* h_beamSignal_cor_b2[30]; TH1D* h_beamSignal_cor_b3[30]; Double_t beamSignal_cor_b0_mean[30]; Double_t beamSignal_cor_b1_mean[30]; Double_t beamSignal_cor_b2_mean[30]; Double_t beamSignal_cor_b3_mean[30]; Double_t mybeta_graph[30];
int goodevents_b0=0; int goodevents_b1=0; int goodevents_b2=0; int goodevents_b3=0; for (int iii=0; iii<=29; iii++){ sprintf(histname,"h_beamSignal_b0_%d",iii); h_beamSignal_b0[iii] = new TH1D(histname,histname,200,0,100000); sprintf(histname,"h_beamSignal_b1_%d",iii); h_beamSignal_b1[iii] = new TH1D(histname,histname,200,0,100000); sprintf(histname,"h_beamSignal_b2_%d",iii); h_beamSignal_b2[iii] = new TH1D(histname,histname,200,0,100000); sprintf(histname,"h_beamSignal_b3_%d",iii); h_beamSignal_b3[iii] = new TH1D(histname,histname,200,0,100000); } for (int iii=0; iii<=29; iii++){ sprintf(histname,"h_beamSignal_cor_b0_%d",iii); h_beamSignal_cor_b0[iii] = new TH1D(histname,histname,200,0,10); sprintf(histname,"h_beamSignal_cor_b1_%d",iii); h_beamSignal_cor_b1[iii] = new TH1D(histname,histname,200,0,10.); sprintf(histname,"h_beamSignal_cor_b2_%d",iii); h_beamSignal_cor_b2[iii] = new TH1D(histname,histname,200,0,10.); sprintf(histname,"h_beamSignal_cor_b3_%d",iii); h_beamSignal_cor_b3[iii] = new TH1D(histname,histname,200,0,10.); sprintf(histname,"h_beamSignal_ic1_%d",iii); h_beamSignal_ic1[iii] = new TH1D(histname,histname,200,0,1000.); } TTree *rootTree = new TTree("t","HIT Data Root Tree"); rootTree->Branch("beamPosX_reg_b0",&beamPosX_reg_b0,"beamPosX_reg_b0/D"); rootTree->Branch("beamPosX_reg_b1",&beamPosX_reg_b1,"beamPosX_reg_b1/D"); rootTree->Branch("beamPosX_reg_b2",&beamPosX_reg_b2,"beamPosX_reg_b2/D"); rootTree->Branch("beamPosX_reg_b3",&beamPosX_reg_b3,"beamPosX_reg_b3/D"); rootTree->Branch("beamFocusX_reg_b0",&beamFocusX_reg_b0,"beamFocusX_reg_b0/D"); rootTree->Branch("beamFocusX_reg_b1",&beamFocusX_reg_b1,"beamFocusX_reg_b1/D"); rootTree->Branch("beamFocusX_reg_b2",&beamFocusX_reg_b2,"beamFocusX_reg_b2/D"); rootTree->Branch("beamFocusX_reg_b3",&beamFocusX_reg_b3,"beamFocusX_reg_b3/D");
rootTree->Branch("beamPeakX_reg_b0",&beamPeakX_reg_b0,"beamPeakX_reg_b0/D"); rootTree->Branch("beamPeakX_reg_b1",&beamPeakX_reg_b1,"beamPeakX_reg_b1/D"); rootTree->Branch("beamPeakX_reg_b2",&beamPeakX_reg_b2,"beamPeakX_reg_b2/D"); rootTree->Branch("beamPeakX_reg_b3",&beamPeakX_reg_b3,"beamPeakX_reg_b3/D"); rootTree->Branch("beamRsqr_b0",&beamRsqr_b0,"beamRsqr_b0/D"); rootTree->Branch("beamRsqr_b1",&beamRsqr_b1,"beamRsqr_b1/D"); rootTree->Branch("beamRsqr_b2",&beamRsqr_b2,"beamRsqr_b2/D"); rootTree->Branch("beamRsqr_b3",&beamRsqr_b3,"beamRsqr_b3/D"); rootTree->Branch("sp2_ps",&sp2_ps,"sp2_ps/D"); rootTree->Branch("sp2_air",&sp2_air,"sp2_air/D"); rootTree->Branch("mybeta",&mybeta,"mybeta/D");
rootTree->Branch("beamSkewX_b0",&beamSkewX_b0,"beamSkewX_b0/D"); rootTree->Branch("beamSkewX_b1",&beamSkewX_b1,"beamSkewX_b1/D"); rootTree->Branch("beamSkewX_b2",&beamSkewX_b2,"beamSkewX_b2/D"); rootTree->Branch("beamSkewX_b3",&beamSkewX_b3,"beamSkewX_b3/D"); rootTree->Branch("beamKurtX_b0",&beamKurtX_b0,"beamKurtX_b0/D"); rootTree->Branch("beamKurtX_b1",&beamKurtX_b1,"beamKurtX_b1/D"); rootTree->Branch("beamKurtX_b2",&beamKurtX_b2,"beamKurtX_b2/D"); rootTree->Branch("beamKurtX_b3",&beamKurtX_b3,"beamKurtX_b3/D"); rootTree->Branch("beamNumX_b0",&numtocalc_b0,"beamNumX_b0/I"); rootTree->Branch("beamNumX_b1",&numtocalc_b1,"beamNumX_b1/I"); rootTree->Branch("beamNumX_b2",&numtocalc_b2,"beamNumX_b2/I"); rootTree->Branch("beamNumX_b3",&numtocalc_b3,"beamNumX_b3/I"); rootTree->Branch("beamSignal_b0",&signal_b0,"beamSignal_b0/D"); rootTree->Branch("beamSignal_b1",&signal_b1,"beamSignal_b1/D"); rootTree->Branch("beamSignal_b2",&signal_b2,"beamSignal_b2/D"); rootTree->Branch("beamSignal_b3",&signal_b3,"beamSignal_b3/D");
rootTree->Branch("beamSignal_cor_b0",&signal_cor_b0,"beamSignal_cor_b0/D"); rootTree->Branch("beamSignal_cor_b1",&signal_cor_b1,"beamSignal_cor_b1/D"); rootTree->Branch("beamSignal_cor_b2",&signal_cor_b2,"beamSignal_cor_b2/D"); rootTree->Branch("beamSignal_cor_b3",&signal_cor_b3,"beamSignal_cor_b3/D");
rootTree->Branch("eventID",&eventID,"eventID/I"); //rootTree->Branch("board_b0_ch",&board_b0_ch,"board_b0_ch[128]/D");
//rootTree->Branch("board_b1_ch",&board_b1_ch,"board_b1_ch[128]/D");
//rootTree->Branch("board_b2_ch",&board_b2_ch,"board_b2_ch[128]/D");
//rootTree->Branch("board_b3_ch",&board_b3_ch,"board_b3_ch[128]/D");
rootTree->Branch("arrayavg_b0",&arrayavg_b0,"arrayavg_b0/D"); rootTree->Branch("arrayavg_b1",&arrayavg_b1,"arrayavg_b1/D"); rootTree->Branch("arrayavg_b2",&arrayavg_b2,"arrayavg_b2/D"); rootTree->Branch("arrayavg_b3",&arrayavg_b3,"arrayavg_b3/D"); rootTree->Branch("M1elements_b0",&M1elements_b0,"M1elements_b0[9]/D"); rootTree->Branch("M1elements_b1",&M1elements_b1,"M1elements_b1[9]/D"); rootTree->Branch("M1elements_b2",&M1elements_b2,"M1elements_b2[9]/D"); rootTree->Branch("M1elements_b3",&M1elements_b3,"M1elements_b3[9]/D");
rootTree->Branch("goodevents_b0",&goodevents_b0,"goodevents_b0/I"); rootTree->Branch("goodevents_b1",&goodevents_b1,"goodevents_b1/I"); rootTree->Branch("goodevents_b2",&goodevents_b2,"goodevents_b2/I"); rootTree->Branch("goodevents_b3",&goodevents_b3,"goodevents_b3/I");
double time; // import and align matching ethercat data
TTree *tree2 = new TTree("t2", "t2"); if(ethercat){ std::cout << " Loading Ethercat data." << std::endl; tree2->ReadFile(ethercatfile, "RELTIME2/D:IC1/D:MW1_POSX/D:MW1_POSY/D:ANALOG_IN1/D:ENERGY_INDEX/D:INTENSITY_INDEX/D:ION-SORT/D:TIME2/D", '\t'); std::cout << "Ethercat data loaded." << std::endl; tree2->Print(); } int k = 0; // int count = 0;
double ic1, ic2, mw1_focusx, mw1_focusy, mw2_focusx, mw2_focusy, mw1_posx, mw1_posy, mw2_posx, mw2_posy; double ic1_avg, ic2_avg, mw1_focusx_avg, mw1_focusy_avg, mw2_focusx_avg, mw2_focusy_avg, mw1_posx_avg, mw1_posy_avg, mw2_posx_avg, mw2_posy_avg; double analog_in1; double energy_index; double energy; double energy_list_p[27] = {48.12, 59.82, 70.03, 79.17, 87.53, 95.3, 102.61, 109.56, 116.2, 122.57, 128.72, 134.65, 140.41, 146.01, 151.5, 156.89, 162.21, 167.46, 172.62, 177.7, 184.81, 191.87, 198.76, 205.48, 212.06, 218.51, 0.}; double energy_list_he[27] = {50.57, 61.86, 71.73, 80.64, 88.85, 96.52, 103.76, 110.64, 117.23, 123.55, 129.64, 135.55, 141.27, 146.84, 152.26, 157.56, 162.73, 167.8, 172.77, 177.64, 184.56, 191.54, 198.36, 205.03, 211.57, 217.98, 0.}; double energy_list_c[27] = {88.83,110.58,129.79,147.13,163.09,178.01,192.13,205.6,218.52,230.98,243.03,254.71,266.08,277.19,288.1,298.87,309.52,320.07,330.48,340.77,355.22,369.64,383.78,397.66,411.32,424.77,0.}; double energy_list_o[27] = {103.77, 129.65, 152.42, 172.98, 191.96, 209.63, 226.53, 242.58, 258.11, 273.04, 287.5, 301.55, 315.32, 328.61, 341.84, 354.88, 367.94, 380.48, 393.09, 405.65, 423.23, 440.82, 458.11, 475.28, 491.93, 508.38, 0.}; double intensity_index; double intensity; double intensity_list_p[16] = {0., 8.00E+07, 1.20E+08, 2.00E+08, 3.20E+08,4.00E+08, 6.00E+08, 8.00E+08, 1.20E+09, 2.00E+09, 3.20E+09, 4.00E+09, 6.00E+09, 8.00E+09, 1.20E+10, 2.00E+10}; double intensity_list_he[16] = { 0., 2.00E+07, 3.00E+07, 5.00E+07, 8.00E+07, 1.00E+08, 1.50E+08, 2.00E+08, 3.00E+08, 5.00E+08, 8.00E+08, 1.00E+09, 1.50E+09, 2.00E+09, 3.00E+09, 5.00E+09};
double intensity_list_c[16] = {0.,2.00E+06, 3.00E+06, 5.00E+06, 8.00E+06, 1.00E+07, 1.50E+07, 2.00E+07, 3.00E+07, 5.00E+07, 8.00E+07, 1.00E+08, 1.50E+08, 2.00E+08, 3.00E+08, 5.00E+08}; double intensity_list_o[16] = {0., 1.00E+06, 1.50E+06, 2.50E+06, 4.00E+06, 5.00E+06, 8.00E+06, 1.00E+07, 1.50E+07, 2.50E+07, 4.00E+07, 5.00E+07, 8.00E+07, 1.00E+08, 1.50E+08, 2.50E+08 }; double ionsort;
double rel_time2,time2;
double timeoffset; int mwoffset; double timewindow; double timeoffset2; double timewindow2;
if(ethercat){ rootTree->Branch("ic1_avg", &ic1_avg, "ic1_avg/D"); rootTree->Branch("mw1_posx", &mw1_posx_avg, "mw1_posx/D"); rootTree->Branch("mw1_posy", &mw1_posy_avg, "mw1_posy/D"); rootTree->Branch("energy_index", &energy_index, "energy_index/D"); rootTree->Branch("intensity_index", &intensity_index, "intensity_index/D"); rootTree->Branch("ionsort", &ionsort, "ionsort/D");
timeoffset = -0.050; //offset between ic and bpm readouts
mwoffset = 2; // offset for timestamped event. MW chamber position correlation seems to be better in other windows
timewindow = -0.0999; //should be a negative number. window size in time to average over.
timeoffset2 = -0.00; //offset between ic and bpm readouts
timewindow2 = -0.0999; //should be a negative number. window size in time to average over.
//////////////////////////////////////////////////////////////////
///// ETHERCAT DATA
//////////////////////////////////////////////////////////////////
// tree->SetBranchAddress("time", &time);
tree2->SetBranchAddress("RELTIME2", &rel_time2); tree2->SetBranchAddress("TIME2", &time2); tree2->SetBranchAddress("IC1", &ic1); // tree2->SetBranchAddress("IC2", &ic2);
//tree2->SetBranchAddress("MW1_FOCUSX", &mw1_focusx);
//tree2->SetBranchAddress("MW1_FOCUSY", &mw1_focusy);
///tree2->SetBranchAddress("MW2_FOCUSX", &mw2_focusx);
///tree2->SetBranchAddress("MW2_FOCUSY", &mw2_focusy);
tree2->SetBranchAddress("MW1_POSX", &mw1_posx); tree2->SetBranchAddress("MW1_POSY", &mw1_posy); //tree2->SetBranchAddress("MW2_POSX", &mw2_posx);
//tree2->SetBranchAddress("MW2_POSY", &mw2_posy);
tree2->SetBranchAddress("ENERGY_INDEX", &energy_index); tree2->SetBranchAddress("INTENSITY_INDEX", &intensity_index); tree2->SetBranchAddress("ION-SORT", &ionsort); tree2->SetBranchAddress("ANALOG_IN1", &analog_in1); }
int currentEntryTree2 = 1;
// int nevents = tree->GetEntries();
int nevents2 = tree2->GetEntries();
int icCounter = 0; int count = 0; int count2 = 0; //loop through recorded data frames
for (int frame = 0; frame<fileframesize; frame++){ // if (icCounter>10000) break;
// if(frame>2000&&frame<120000) continue;
eventID = frame; if (ethercat){ if (timestampfile) timestampfile >> time ; //printf("%i %f\n", eventID, time);
count= 0; count2 = 0; ic1_avg = 0.; mw1_posx_avg = 0.; mw1_posy_avg = 0.; tree2->GetEntry(currentEntryTree2); /* if (frame % 100 == 0)
{
printf("merging event %d ,", frame); printf("Time %f \n", time); printf("Entry hit %d ,", currentEntryTree2); printf("Time hit %f \n", time2); } */ while (time2 < time + timeoffset && currentEntryTree2 < nevents2 ) { if (time2 - time - timeoffset > timewindow) { tree2->GetEntry(currentEntryTree2); if (ic1>0.0) ic1_avg += ic1; if (ic1>0.0) count++; // if (frame % 200 == 0) printf("%i %2.3f %2.3f %2.3f %2.3f %2.3f %i \n", count, ic1, time2, time, timeoffset, time2 - time - timeoffset, mwoffset);
} tree2->GetEntry(currentEntryTree2); if ( time2 - time - timeoffset2 > timewindow2) { tree2->GetEntry(currentEntryTree2 + mwoffset); //why currentEtryTree2-4?
mw1_posx_avg += mw1_posx; mw1_posy_avg += mw1_posy; count2++; // if (i % 2000 == 0) printf("%i %2.3f %2.3f %2.3f %2.3f\n", count2, time, time2, ic1, mw1_posx);
//if (i % 2001 == 0) printf("%i %2.3f %2.3f %2.3f %2.3f\n", count2, time, time2, ic1, mw1_posx);
// if (i % 2002 == 0) printf("%i %2.3f %2.3f %2.3f %2.3f\n", count2, time, time2, ic1, mw1_posx);
// printf("%i %2.3f %2.3f %2.3f %2.3f\n", count2, time, time2, ic1, mw1_posx);
} // currentEntryTree2++;
tree2->GetEntry(currentEntryTree2); currentEntryTree2++; if (count2>0){ mw1_posx_avg /= double(count2); //the positions are weighted by the charge
mw1_posy_avg /= double(count2); } if(count>0){ ic1_avg /= double(count); if (ic1_avg>1.) icCounter++; // if (frame % 2000 == 0) printf("%i %f.2 %i \n", count, ic1_avg, icCounter);
} // std::cout << ic1_avg << " " << icCounter << std::endl;
} /////////end of ethercat matching
if (floor(energy_index/10.)>=0&&floor(energy_index/10)<=26){energy_indexbin = floor(energy_index/10.);} else {energy_indexbin=27;}
if (ionsort==3){ //protons
energy = energy_list_p[ energy_indexbin]; intensity = intensity_list_p[ int(intensity_index) ]; sp_air = f_sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_sp_air->Eval(energy); sp2_ps = f_sp_ps->Eval(energy); lorentz_gamma = energy/938.272 +1.; // lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 1.; } else if (ionsort==1){ //helium
energy = energy_list_he[ energy_indexbin ]; intensity = intensity_list_he[ int(intensity_index) ]; sp_air = f_h2sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_h2sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_h2sp_air->Eval(energy); sp2_ps = f_h2sp_ps->Eval(energy); lorentz_gamma = energy/932.1055 +1.;// lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 4.;
} else if (ionsort==4){ //carbon
energy = energy_list_c[ energy_indexbin]; intensity = intensity_list_c[ int(intensity_index) ];
sp_air = f_c12sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_c12sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_c12sp_air->Eval(energy); sp2_ps = f_c12sp_ps->Eval(energy); lorentz_gamma = energy/932.3539 +1.;// lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 36.;
} else if (ionsort==2){ //oxygen
energy = energy_list_o[ energy_indexbin]; intensity = intensity_list_o[ int(intensity_index) ]; sp_air = f_o16sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_o16sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_o16sp_air->Eval(energy); sp2_ps = f_o16sp_ps->Eval(energy); lorentz_gamma = energy/931.4418 +1.;// lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 64.; } else { sp_air = -1.; sp_ps = -1.; sp2_air = -1.; sp2_ps = -1.; lorentz_gamma = 0.; mybeta = 0.; zsqr = -1.; } // intcorr = (Intensity /19.8E6) / (totalcurrent/sp2_air/totaltime);
intcorr = (intensity / 19.8E6) / (ic1_avg/sp2_air)/ 3.11;
// if (frame>2000&&ic1_avg<0.5) continue;
}
///////////////////////////////////
//// HIT DATA
///////////////////////////////////
if (frame>2000 && beam_wason == true && beam_off == true) { beamoffcounter++; // cout << beamoffcounter<< endl;
} if (samplenoise==false && frame> 2000 && beam_wason == true && beam_off == true && beamoffcounter==5000) { samplenoise = true; }
//board_b 0
board_b=0; signal_b0 = 0.; maxchannelamp_b0 = 0.;
file.seekg(framestart*sizeof(BufferData)+board_b*sizeof(BufferData)+4*frame*sizeof(BufferData)); file.read ((char*)dataptr ,sizeof(BufferData)); if (dataptr->sync_frame.device_nr==board_b){ // std::cout << frame << " " << dataptr->sync_frame.device_nr << " " << dataptr->sync_frame.local_ctr << " " << dataptr->sync_frame.global_ctr << " " << dataptr->sync_frame.sma_state << std::endl;
if (frame%10000==0) std::cout << "Frame: " << frame << " (" <<double(frame)/double(fileframesize)*100.0 << "%)" << std::endl;
arrayavg_b0 = 0; //shift the events of the rolling average of the signal one left
for (int j = 1; j<length;j++){ array_b0[j-1] = array_b0[j]; arrayavg_b0 += array_b0[j-1]; // cout << array_b0[j-1] << " ";
} array_b0[length-1] = 0.; // cout << endl;
///loop over the channels
for (int i = 0;i<128;i++){ board_b0_ch[i] = dataptr->sensor_data[i]; if (frame<1000 || samplenoise==true){ if (samplenoise==true&& beamoffcounter==5000) { board_b0_ch_bkg[i]=0.; if(i==0) cout << frame << " Resampling noise " << beamoffcounter << endl; } board_b0_ch_bkg[i] += board_b0_ch[i]/1000.; //find baseline from the average of first 1000 events
if (samplenoise==true && beamoffcounter==6000) { beam_wason = false; samplenoise =false; cout << frame << " End sampling." << endl; } } else if (frame>=1000&& frame<2000){ board_b0_ch[i] -= board_b0_ch_bkg[i]; //histogram the subtracted baseline in the next 1000 events
th2_signal_vs_channel_bkg_b0->Fill(i,board_b0_ch[i]); if (i==1) graph_bkg_b0->SetPoint(graph_bkg_b0->GetN(), frame, board_b0_ch[i]); } else if (frame>=2000 && samplenoise==false) { board_b0_ch[i]-=board_b0_ch_bkg[i]; // the rest background subtracted
// board_b0_ch[i]*=calibration_b0[i]; //calibration factor
th2_signal_vs_channel_b0->Fill(i,board_b0_ch[i]);
//find the peak channel
if (board_b0_ch[i]> maxchannelamp_b0) { maxchannel_b0 = i; maxchannelamp_b0 = board_b0_ch[i]; // cout << maxchannel_b0 << " " <<maxchannelamp_b0 << endl;
}
///add new event to the rolling average.
if( board_b0_ch[i]>50.0){ array_b0[length-1] += board_b0_ch[i]; } ////////////////////////////////////////////
} }//end of 128
// th1_signal_b0->Fill(signal_b0);
if (frame>2000) { arrayavg_b0 += array_b0[length-1]/length; if (arrayavg_b0>= 20000) { beamoncounter++; } if (beamoncounter>100 && beam_wason == false && arrayavg_b0>= 20000) { beam_off = false; beam_wason = true; cout << frame << " Beam on " << arrayavg_b0 << endl; } if (arrayavg_b0< 10000) { if(beam_wason== true && beam_off==false ) { cout << frame << "Beam went OFF!" << arrayavg_b0 << endl; beamoffcounter = 0; } beam_off = true; goodevents_b0=0; goodevents_b1=0; goodevents_b2=0; goodevents_b3=0; } }
// cout << frame << " Rolling average: " << arrayavg_b0 << endl;// " ic: " << ic1_avg << endl;
} else { std::cout << "Error." << std::endl; }
//board_b 1
board_b=1; signal_b1 = 0.; maxchannelamp_b1 = 0.;
file.seekg(framestart*sizeof(BufferData)+board_b*sizeof(BufferData)+4*frame*sizeof(BufferData)); file.read ((char*)dataptr ,sizeof(BufferData)); if (dataptr->sync_frame.device_nr==board_b){ // std::cout << frame << " " << dataptr->sync_frame.device_nr << " " << dataptr->sync_frame.local_ctr << " " << dataptr->sync_frame.global_ctr << " " << dataptr->sync_frame.sma_state << std::endl;
arrayavg_b1 = 0; //shift the events of the rolling average of the signal one left
for (int j = 1; j<length;j++){ array_b1[j-1] = array_b1[j]; arrayavg_b1 += array_b1[j-1]; // cout << array_b1[j-1] << " ";
} array_b1[length-1] = 0.; // cout << endl;
///loop over the channels
for (int i = 0;i<128;i++){ board_b1_ch[i] = dataptr->sensor_data[i]; if (frame<1000){ board_b1_ch_bkg[i] += board_b1_ch[i]/1000.; //find baseline from the average of first 1000 events
} else if (frame>=1000&& frame<2000){ board_b1_ch[i] -= board_b1_ch_bkg[i]; //histogram the subtracted baseline in the next 1000 events
th2_signal_vs_channel_bkg_b1->Fill(i,board_b1_ch[i]); if (i==1) graph_bkg_b1->SetPoint(graph_bkg_b1->GetN(), frame, board_b1_ch[i]); } else if (frame>=2000) { board_b1_ch[i]-=board_b1_ch_bkg[i]; // the rest background subtracted
// board_b1_ch[i]*=calibration_b1[i]; //calibration factor
th2_signal_vs_channel_b1->Fill(i,board_b1_ch[i]);
//find the peak channel
if (board_b1_ch[i]> maxchannelamp_b1) { maxchannel_b1 = i; maxchannelamp_b1 = board_b1_ch[i]; // cout << maxchannel_b1 << " " <<maxchannelamp_b1 << endl;
}
///add new event to the rolling average.
if( board_b1_ch[i]>50.0){ array_b1[length-1] += board_b1_ch[i]; } ////////////////////////////////////////////
} }//end of 128
// th1_signal_b1->Fill(signal_b1);
arrayavg_b1 += array_b1[length-1]/length;
// cout << frame << " Rolling average: " << arrayavg_b1 << endl;// " ic: " << ic1_avg << endl;
} else { std::cout << "Error." << std::endl; }
//board_b 2
board_b=2; signal_b2 = 0.; maxchannelamp_b2 = 0.;
file.seekg(framestart*sizeof(BufferData)+board_b*sizeof(BufferData)+4*frame*sizeof(BufferData)); file.read ((char*)dataptr ,sizeof(BufferData)); if (dataptr->sync_frame.device_nr==board_b){ // std::cout << frame << " " << dataptr->sync_frame.device_nr << " " << dataptr->sync_frame.local_ctr << " " << dataptr->sync_frame.global_ctr << " " << dataptr->sync_frame.sma_state << std::endl;
arrayavg_b2 = 0; //shift the events of the rolling average of the signal one left
for (int j = 1; j<length;j++){ array_b2[j-1] = array_b2[j]; arrayavg_b2 += array_b2[j-1]; // cout << array_b2[j-1] << " ";
} array_b2[length-1] = 0.; // cout << endl;
///loop over the channels
for (int i = 0;i<128;i++){ board_b2_ch[i] = dataptr->sensor_data[i]; if (frame<1000){ board_b2_ch_bkg[i] += board_b2_ch[i]/1000.; //find baseline from the average of first 1000 events
} else if (frame>=1000&& frame<2000){ board_b2_ch[i] -= board_b2_ch_bkg[i]; //histogram the subtracted baseline in the next 1000 events
th2_signal_vs_channel_bkg_b2->Fill(i,board_b2_ch[i]); if (i==1) graph_bkg_b2->SetPoint(graph_bkg_b2->GetN(), frame, board_b2_ch[i]); } else if (frame>=2000) { board_b2_ch[i]-=board_b2_ch_bkg[i]; // the rest background subtracted
// board_b2_ch[i]*=calibration_b2[i]; //calibration factor
th2_signal_vs_channel_b2->Fill(i,board_b2_ch[i]);
//find the peak channel
if (board_b2_ch[i]> maxchannelamp_b2) { maxchannel_b2 = i; maxchannelamp_b2 = board_b2_ch[i]; // cout << maxchannel_b2 << " " <<maxchannelamp_b2 << endl;
}
///add new event to the rolling average.
if( board_b2_ch[i]>50.0){ array_b2[length-1] += board_b2_ch[i]; } ////////////////////////////////////////////
} }//end of 128
// th1_signal_b2->Fill(signal_b2);
arrayavg_b2 += array_b2[length-1]/length;
// cout << frame << " Rolling average: " << arrayavg_b2 << endl;// " ic: " << ic1_avg << endl;
} else { std::cout << "Error." << std::endl; }
//board_b 3
board_b=3; signal_b3 = 0.; maxchannelamp_b3 = 0.;
file.seekg(framestart*sizeof(BufferData)+board_b*sizeof(BufferData)+4*frame*sizeof(BufferData)); file.read ((char*)dataptr ,sizeof(BufferData)); if (dataptr->sync_frame.device_nr==board_b){ // std::cout << frame << " " << dataptr->sync_frame.device_nr << " " << dataptr->sync_frame.local_ctr << " " << dataptr->sync_frame.global_ctr << " " << dataptr->sync_frame.sma_state << std::endl;
arrayavg_b3 = 0; //shift the events of the rolling average of the signal one left
for (int j = 1; j<length;j++){ array_b3[j-1] = array_b3[j]; arrayavg_b3 += array_b3[j-1]; // cout << array_b3[j-1] << " ";
} array_b3[length-1] = 0.; // cout << endl;
///loop over the channels
for (int i = 0;i<128;i++){ board_b3_ch[i] = dataptr->sensor_data[i]; if (frame<1000){ board_b3_ch_bkg[i] += board_b3_ch[i]/1000.; //find baseline from the average of first 1000 events
} else if (frame>=1000&& frame<2000){ board_b3_ch[i] -= board_b3_ch_bkg[i]; //histogram the subtracted baseline in the next 1000 events
th2_signal_vs_channel_bkg_b3->Fill(i,board_b3_ch[i]); if (i==1) graph_bkg_b3->SetPoint(graph_bkg_b3->GetN(), frame, board_b3_ch[i]); } else if (frame>=2000) { board_b3_ch[i]-=board_b3_ch_bkg[i]; // the rest background subtracted
// board_b3_ch[i]*=calibration_b3[i]; //calibration factor
th2_signal_vs_channel_b3->Fill(i,board_b3_ch[i]);
//find the peak channel
if (board_b3_ch[i]> maxchannelamp_b3) { maxchannel_b3 = i; maxchannelamp_b3 = board_b3_ch[i]; // cout << maxchannel_b3 << " " <<maxchannelamp_b3 << endl;
}
///add new event to the rolling average.
if( board_b3_ch[i]>50.0){ array_b3[length-1] += board_b3_ch[i]; } ////////////////////////////////////////////
} }//end of 128
// th1_signal_b3->Fill(signal_b3);
arrayavg_b3 += array_b3[length-1]/length;
// cout << frame << " Rolling average: " << arrayavg_b3 << endl;// " ic: " << ic1_avg << endl;
} else { std::cout << "Error." << std::endl; }
////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////
//start the signal analysis
////////////////////////////////////////////////////////////
////////////////////////////////////////////////////////////
if (frame>=2000&&beam_off == false&&beam_wason == true){
////////////////////////////////////////////////////////////
//find the approx FWHM
////////////////////////////////////////////////////////////
nfwhm_b0 =0; nfwhm_b1 =0; nfwhm_b2 =0; nfwhm_b3 =0;
for (int i = 0;i<128;i++){ if (board_b0_ch[i] > maxchannelamp_b0/3.) nfwhm_b0++; //switched to a lower threshold 1/3 instead of 1/2
if (board_b1_ch[i] > maxchannelamp_b1/3.) nfwhm_b1++; if (board_b2_ch[i] > maxchannelamp_b2/3.) nfwhm_b2++; if (board_b3_ch[i] > maxchannelamp_b3/3.) nfwhm_b3++; signal_gsl_b0[i] = 0.; signal_gsl_b1[i] = 0.; signal_gsl_b2[i] = 0.; signal_gsl_b3[i] = 0.;
}
////////////////////////////////////////////////////////////
//integrate under the approximate peak
//build the channel list for statistical analysis
////////////////////////////////////////////////////////////
numtocalc_b0 = 0; numtocalc_b1 = 0; numtocalc_b2 = 0; numtocalc_b3 = 0; signal_b0 = 0. - signaloffset_b0[energy_indexbin];
// cout << energy_indexbin << endl;
for (int i = maxchannel_b0-nfwhm_b0 ; i <= maxchannel_b0 + nfwhm_b0; i++){ if (i>=0 && i<=127 && board_b0_ch[i]>0){ signal_b0 +=board_b0_ch[i]; signal_gsl_b0[numtocalc_b0]=board_b0_ch[i]; // channellist_gsl_b0[numtocalc_b0] = i;
channellist_gsl_b0[numtocalc_b0] = pos[i]; numtocalc_b0++; } } th1_signal_b0->Fill(signal_b0); //cout << signal_b0 << endl;
if (ic1_avg>0.1){ h_beamSignal_b0[energy_indexbin]->Fill(signal_b0); goodevents_b0++;
} signal_b1 = 0. - signaloffset_b1[energy_indexbin];; for (int i = maxchannel_b1-nfwhm_b1 ; i <= maxchannel_b1 + nfwhm_b1; i++){ if (i>=0 && i<=127&&board_b1_ch[i]>0){ signal_b1 +=board_b1_ch[i] ; signal_gsl_b1[numtocalc_b1]=board_b1_ch[i]; // channellist_gsl_b1[numtocalc_b1] = i;
channellist_gsl_b1[numtocalc_b1] = pos[i]; numtocalc_b1++; } } th1_signal_b1->Fill(signal_b1); if (ic1_avg>0.1){ h_beamSignal_b1[energy_indexbin]->Fill(signal_b1); goodevents_b1++; } signal_b2 = 0. - signaloffset_b2[energy_indexbin];; for (int i = maxchannel_b2-nfwhm_b2 ; i <= maxchannel_b2 + nfwhm_b2; i++){ if (i>=0 && i<=127&&board_b2_ch[i]>0){ signal_b2 +=board_b2_ch[i]; signal_gsl_b2[numtocalc_b2]=board_b2_ch[i]; // channellist_gsl_b2[numtocalc_b2] = i;
channellist_gsl_b2[numtocalc_b2] = pos[i]; numtocalc_b2++; } }
th1_signal_b2->Fill(signal_b2); if (ic1_avg>0.1) { h_beamSignal_b2[energy_indexbin]->Fill(signal_b2); goodevents_b2++; }
signal_b3 = 0. - signaloffset_b3[energy_indexbin]; for (int i = maxchannel_b3-nfwhm_b3 ; i <= maxchannel_b3 + nfwhm_b3; i++){ if (i>=0 && i<=127&&board_b3_ch[i]>0){ signal_b3 +=board_b3_ch[i]; signal_gsl_b3[numtocalc_b3]=board_b3_ch[i]; //channellist_gsl_b3[numtocalc_b3] = i;
channellist_gsl_b3[numtocalc_b3] = pos[i]; numtocalc_b3++; } } th1_signal_b3->Fill(signal_b3); if (ic1_avg>0.1) { h_beamSignal_b3[energy_indexbin]->Fill(signal_b3); h_beamSignal_ic1[energy_indexbin]->Fill(ic1_avg); goodevents_b3++; }
signal_cor_b0 = signal_b0*intcorr/(intensity/1.E6)/sp2_ps; signal_cor_b1 = signal_b1*intcorr/(intensity/1.E6)/sp2_ps; signal_cor_b2 = signal_b2*intcorr/(intensity/1.E6)/sp2_ps; signal_cor_b3 = signal_b3*intcorr/(intensity/1.E6)/sp2_ps; if (signal_cor_b0 >0.05 && signal_cor_b0 < 10.) {h_beamSignal_cor_b0[energy_indexbin]->Fill(signal_cor_b0);} if (signal_cor_b1 >0.05 && signal_cor_b1 < 10.) {h_beamSignal_cor_b1[energy_indexbin]->Fill(signal_cor_b1);} if (signal_cor_b2 >0.05 && signal_cor_b2 < 10.) {h_beamSignal_cor_b2[energy_indexbin]->Fill(signal_cor_b2);} if (signal_cor_b3 >0.05 && signal_cor_b3 < 10.) {h_beamSignal_cor_b3[energy_indexbin]->Fill(signal_cor_b3);}
///////////////// linear regression using Integration by parts of gaussian function.
if (doLinInt&&numtocalc_b0>=3&&numtocalc_b1>=3&&numtocalc_b2>=3&&numtocalc_b3>=3){ for (int j = 0; j<=4; j++){ if (j== 0) { copy(signal_gsl_b0,signal_gsl_b0+128, signal_reg); copy(channellist_gsl_b0,channellist_gsl_b0+128, channellist_reg); numtocalc_reg = numtocalc_b0; } else if (j== 1) { copy(signal_gsl_b1,signal_gsl_b1+128, signal_reg); copy(channellist_gsl_b1,channellist_gsl_b1+128, channellist_reg); numtocalc_reg = numtocalc_b1; } else if (j== 2) { copy(signal_gsl_b2,signal_gsl_b2+128, signal_reg); copy(channellist_gsl_b2,channellist_gsl_b2+128, channellist_reg); numtocalc_reg = numtocalc_b2; } else if (j== 3) { copy(signal_gsl_b3,signal_gsl_b3+128, signal_reg); copy(channellist_gsl_b3,channellist_gsl_b3+128, channellist_reg); numtocalc_reg = numtocalc_b3; } SumY = 0.; SumS = 0.; SumT = 0.; SumS2 = 0.; SumST = 0.; SumT2 = 0.; SumYS = 0.; SumYT = 0.; b_den = 0.; b_num = 0.; b = 0.; SumYYM = 0.; SumYYP = 0.; MeanY = 0.; for(int k=0; k<numtocalc_b0;k++){ if (k==0){ S[k]=0.; T[k]=0.; } else{ S[k] = S[k-1]+0.5*( signal_reg[k] + signal_reg[k-1] ) * ( channellist_reg[k] - channellist_reg[k-1] ); T[k] = T[k-1]+0.5*( channellist_reg[k] * signal_reg[k] + channellist_reg[k-1] * signal_reg[k-1] ) * ( channellist_reg[k] - channellist_reg[k-1] ); } // cout << S[k] << " " << T[k] << endl;
SumS += S[k]; SumT += T[k]; SumY += signal_reg[k]; SumS2 += S[k]*S[k]; SumST += S[k]*T[k]; SumT2 += T[k]*T[k]; SumYS += signal_reg[k]*S[k]; SumYT += signal_reg[k]*T[k]; MeanY+=signal_reg[k]; } MeanY/=numtocalc_reg;
M1(0,0) = SumT2; M1(0,1) = SumST; M1(0,2) = SumT; M1(1,0) = SumST; M1(1,1) = SumS2; M1(1,2) = SumS; M1(2,0) = SumT; M1(2,1) = SumS; M1(2,2) = numtocalc_reg; M2(0) = SumYT; M2(1) = SumYS; M2(2) = SumY; M1inv = M1.Invert(); ABC = M1inv * M2;
//calculate b,p,c ---> y = b*exp(-p*(x-c)*(x-c))
p = -ABC(0)/2.; c = -ABC(1)/ABC(0);
for(int k=0; k<numtocalc_reg;k++){ b_num += exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) * signal_reg[k]; b_den += exp(-2*p*(channellist_reg[k]-c)*(channellist_reg[k]-c)); } b = b_num/b_den;
//calculate R-squared
///residual plots (should be normally distributed). They are normalised to the standard error of the baseline (17 ADC units)
for(int k=0; k<numtocalc_reg;k++){ SumYYM+= (signal_reg[k]-MeanY)*(signal_reg[k]-MeanY); SumYYP+= (b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - MeanY )*(b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - MeanY ); if (j == 0) { h_regresidual_b0->Fill((b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); th2_resid_vs_channel_bkg_b0->Fill(channellist_reg[k],(b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); } if (j == 1) { h_regresidual_b1->Fill((b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); th2_resid_vs_channel_bkg_b1->Fill(channellist_reg[k],(b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); } if (j == 2) { h_regresidual_b2->Fill((b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); th2_resid_vs_channel_bkg_b2->Fill(channellist_reg[k],(b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); } if (j == 3) { h_regresidual_b3->Fill((b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); th2_resid_vs_channel_bkg_b3->Fill(channellist_reg[k],(b*exp(-p*(channellist_reg[k]-c)*(channellist_reg[k]-c)) - signal_reg[k])/signal_reg[k]); } } // cout << "R-squared = " << SumYYP/SumYYM << endl;
//final results
if (j == 0) { beamFocusX_reg_b0 = 2.3548/sqrt(2*p); beamPosX_reg_b0 = -ABC(1)/ ABC(0); beamPeakX_reg_b0 = b; beamRsqr_b0 = SumYYP/SumYYM; beamSkewX_b0 = gsl_stats_wskew_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b0,beamFocusX_reg_b0/2.3548); //skewness (symmetry)
beamKurtX_b0 = gsl_stats_wkurtosis_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b0,beamFocusX_reg_b0/2.3548); //excess kurtosis (well behaved tails)
M1elements_b0[0] = SumT2; M1elements_b0[1] = SumST; M1elements_b0[2] = SumT; M1elements_b0[3] = SumST; M1elements_b0[4] = SumS2; M1elements_b0[5] = SumS; M1elements_b0[6] = SumT; M1elements_b0[7] = SumS; M1elements_b0[8] = numtocalc_reg; } else if (j == 1) { beamFocusX_reg_b1 = 2.3548/sqrt(2*p); beamPosX_reg_b1 = -ABC(1)/ ABC(0); beamPeakX_reg_b1 = b; beamRsqr_b1 = SumYYP/SumYYM; beamSkewX_b1 = gsl_stats_wskew_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b1,beamFocusX_reg_b1/2.3548); //skewness (symmetry)
beamKurtX_b1 = gsl_stats_wkurtosis_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b1,beamFocusX_reg_b1/2.3548); //excess kurtosis (well behaved tails)
M1elements_b0[0] = SumT2; M1elements_b1[1] = SumST; M1elements_b1[2] = SumT; M1elements_b1[3] = SumST; M1elements_b1[4] = SumS2; M1elements_b1[5] = SumS; M1elements_b1[6] = SumT; M1elements_b1[7] = SumS; M1elements_b1[8] = numtocalc_reg;
} else if (j == 2) { beamFocusX_reg_b2 = 2.3548/sqrt(2*p); beamPosX_reg_b2 = -ABC(1)/ ABC(0); beamPeakX_reg_b2 = b;
beamRsqr_b2 = SumYYP/SumYYM; beamSkewX_b2 = gsl_stats_wskew_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b2,beamFocusX_reg_b2/2.3548); //skewness (symmetry)
beamKurtX_b2 = gsl_stats_wkurtosis_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b2,beamFocusX_reg_b2/2.3548); //excess kurtosis (well behaved tails)
M1elements_b2[0] = SumT2; M1elements_b2[1] = SumST; M1elements_b2[2] = SumT; M1elements_b2[3] = SumST; M1elements_b2[4] = SumS2; M1elements_b2[5] = SumS; M1elements_b2[6] = SumT; M1elements_b2[7] = SumS; M1elements_b2[8] = numtocalc_reg; } else if (j == 3) { beamFocusX_reg_b3 = 2.3548/sqrt(2*p); beamPosX_reg_b3 = -ABC(1)/ ABC(0); beamPeakX_reg_b3 = b; beamRsqr_b3 = SumYYP/SumYYM; beamSkewX_b3 = gsl_stats_wskew_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b3,beamFocusX_reg_b3/2.3548); //skewness (symmetry)
beamKurtX_b3 = gsl_stats_wkurtosis_m_sd(signal_reg,1,channellist_reg,1,numtocalc_reg,beamPosX_reg_b3,beamFocusX_reg_b3/2.3548); //excess kurtosis (well behaved tails)
M1elements_b0[0] = SumT2; M1elements_b3[1] = SumST; M1elements_b3[2] = SumT; M1elements_b3[3] = SumST; M1elements_b3[4] = SumS2; M1elements_b3[5] = SumS; M1elements_b3[6] = SumT; M1elements_b3[7] = SumS; M1elements_b3[8] = numtocalc_reg; }
} } else if (numtocalc_b0>0&&numtocalc_b1>0) {
beamPosX_reg_b0 = gsl_stats_wmean(signal_gsl_b0,1,channellist_gsl_b0,1,numtocalc_b0); //calculate the weighted mean
beamFocusX_reg_b0 = gsl_stats_wsd_with_fixed_mean(signal_gsl_b0,1,channellist_gsl_b0,1,numtocalc_b0,beamPosX_reg_b0); //Standard Deviation
beamFocusX_reg_b0 *=2.3548;//SD-->FWHM
beamPosX_reg_b1 = gsl_stats_wmean(signal_gsl_b1,1,channellist_gsl_b1,1,numtocalc_b1); //calculate the weighted mean
beamFocusX_reg_b1 = gsl_stats_wsd_with_fixed_mean(signal_gsl_b1,1,channellist_gsl_b1,1,numtocalc_b1,beamPosX_reg_b1); //Standard Deviation
beamNumX_b1 = numtocalc_b1; beamFocusX_reg_b1 *=2.3548;//SD-->FWHM
beamPosX_reg_b2 = gsl_stats_wmean(signal_gsl_b2,1,channellist_gsl_b2,1,numtocalc_b2); //calculate the weighted mean
beamFocusX_reg_b2 = gsl_stats_wsd_with_fixed_mean(signal_gsl_b2,1,channellist_gsl_b2,1,numtocalc_b2,beamPosX_reg_b2); //Standard Deviation
beamNumX_b2 = numtocalc_b2; beamFocusX_reg_b2 *=2.3548;//SD-->FWHM
beamPosX_reg_b3 = gsl_stats_wmean(signal_gsl_b3,1,channellist_gsl_b3,1,numtocalc_b3); //calculate the weighted mean
beamFocusX_reg_b3 = gsl_stats_wsd_with_fixed_mean(signal_gsl_b3,1,channellist_gsl_b3,1,numtocalc_b3,beamPosX_reg_b3); //Standard Deviation
beamNumX_b3 = numtocalc_b3; beamFocusX_reg_b3 *=2.3548;//SD-->FWHM
beamRsqr_b0 = .0; beamRsqr_b1 = .0; beamRsqr_b2 = .0; beamRsqr_b3 = .0;
} else { beamFocusX_reg_b0 = 0.; beamFocusX_reg_b1 = 0.; beamFocusX_reg_b1 = 0.; beamFocusX_reg_b1 = 0.; beamPosX_reg_b0 = 0.; beamPosX_reg_b1 = 0.; beamPosX_reg_b2 = 0.; beamPosX_reg_b3 = 0.; beamRsqr_b0 = .0; beamRsqr_b1 = .0; beamRsqr_b2 = .0; beamRsqr_b3 = .0;
}
rootTree->Fill();
} if (frameend>0 && frame+framestart>=frameend) break; }//end of loop over frames
graph_bkg_b0->SetName("graph_bkg_b0"); // graph_bkg_b0->Write();
graph_bkg_b1->SetName("graph_bkg_b1"); // graph_bkg_b1->Write();
graph_bkg_b2->SetName("graph_bkg_b2"); // graph_bkg_b2->Write();
graph_bkg_b3->SetName("graph_bkg_b3"); // graph_bkg_b3->Write();
// bic1->Fill();
// bmw1_posx->Fill();
//bmw1_posy->Fill();
///make reduced signal graph.
for (int jjj = 0; jjj<23;jjj++) { // beamSignal_cor_b0_mean[jjj] = h_beamSignal_cor_b0[jjj]->GetMean();
// beamSignal_cor_b1_mean[jjj] = h_beamSignal_cor_b1[jjj]->GetMean();
// beamSignal_cor_b2_mean[jjj] = h_beamSignal_cor_b2[jjj]->GetMean();
// beamSignal_cor_b3_mean[jjj] = h_beamSignal_cor_b3[jjj]->GetMean();
//// get the median (0.5 quantile)
q = 0.5; // 0.5 for "median"
if (ionsort==3){ //protons
energy = energy_list_p[ jjj]; intensity = intensity_list_p[ int(intensity_index) ]; sp_air = f_sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_sp_air->Eval(energy); sp2_ps = f_sp_ps->Eval(energy); lorentz_gamma = energy/938.272 +1.; // lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 1.; } else if (ionsort==1){ //helium
energy = energy_list_he[ jjj ]; intensity = intensity_list_he[ int(intensity_index) ]; sp_air = f_h2sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_h2sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_h2sp_air->Eval(energy); sp2_ps = f_h2sp_ps->Eval(energy); lorentz_gamma = energy/932.1055 +1.;// lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 4.;
} else if (ionsort==4){ //carbon
energy = energy_list_c[ jjj]; intensity = intensity_list_c[ int(intensity_index) ];
sp_air = f_c12sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_c12sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_c12sp_air->Eval(energy); sp2_ps = f_c12sp_ps->Eval(energy); lorentz_gamma = energy/932.3539 +1.;// lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 36.;
} else if (ionsort==2){ //oxygen
energy = energy_list_o[ jjj]; intensity = intensity_list_o[ int(intensity_index) ]; sp_air = f_o16sp_air->Eval(energy)*(intensity/1.0E6); sp_ps = f_o16sp_ps->Eval(energy)*(intensity/1.0E6); sp2_air = f_o16sp_air->Eval(energy); sp2_ps = f_o16sp_ps->Eval(energy); lorentz_gamma = energy/931.4418 +1.;// lorentz_gamma = E_u/M_u +1
mybeta = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma)); zsqr = 64.; } else { sp_air = -1.; sp_ps = -1.; sp2_air = -1.; sp2_ps = -1.; lorentz_gamma = 0.; mybeta = 0.; zsqr = -1.; } h_beamSignal_ic1[jjj]->ComputeIntegral(); //just a precaution
h_beamSignal_ic1[jjj]->GetQuantiles(1,&median, &q); //median = h_beamSignal_ic1[jjj]->GetMean();
// intcorr = (Intensity /19.8E6) / (totalcurrent/sp2_air/totaltime);
if (median>0&&energy>0) {intcorr = (intensity / 19.8E6) / (median/sp2_air)/ 3.11;} // else if (median>0&&energy>0) {intcorr = 0.; median = 0.;}
else {intcorr = 0.; median = 0.;} cout << "energy: " << energy << endl; cout << "intensity: " << intensity << endl; cout << "sp2_air: " << sp2_air << endl; cout << "median: " << median << endl; cout << "intcorr: " << intcorr << endl;
mybeta_graph[jjj] = sqrt(1. - 1./(lorentz_gamma*lorentz_gamma));
h_beamSignal_b0[jjj]->ComputeIntegral(); //just a precaution
h_beamSignal_b0[jjj]->GetQuantiles(1,&median, &q); beamSignal_cor_b0_mean[jjj] = median*intcorr /(intensity/1.E6)/sp2_ps; //beamSignal_cor_b0_mean[jjj] = h_beamSignal_b0[jjj]->GetMean()*intcorr /(intensity/1.E6)/sp2_ps;
cout << beamSignal_cor_b0_mean[jjj] << endl; h_beamSignal_b1[jjj]->ComputeIntegral(); //just a precaution
h_beamSignal_b1[jjj]->GetQuantiles(1,&median, &q); beamSignal_cor_b1_mean[jjj] = median*intcorr /(intensity/1.E6)/sp2_ps; //beamSignal_cor_b1_mean[jjj] = h_beamSignal_b1[jjj]->GetMean()*intcorr /(intensity/1.E6)/sp2_ps;
h_beamSignal_b2[jjj]->ComputeIntegral(); //just a precaution
h_beamSignal_b2[jjj]->GetQuantiles(1,&median, &q); beamSignal_cor_b2_mean[jjj] = median*intcorr /(intensity/1.E6)/sp2_ps; // beamSignal_cor_b2_mean[jjj] = h_beamSignal_b2[jjj]->GetMean()*intcorr /(intensity/1.E6)/sp2_ps;
h_beamSignal_b3[jjj]->ComputeIntegral(); //just a precaution
h_beamSignal_b3[jjj]->GetQuantiles(1,&median, &q); beamSignal_cor_b3_mean[jjj] = median*intcorr /(intensity/1.E6)/sp2_ps; //beamSignal_cor_b3_mean[jjj] = h_beamSignal_b3[jjj]->GetMean()*intcorr /(intensity/1.E6)/sp2_ps;
} TGraph * graph_b0 = new TGraph(23,mybeta_graph, beamSignal_cor_b0_mean); TGraph * graph_b1 = new TGraph(23,mybeta_graph, beamSignal_cor_b1_mean); TGraph * graph_b2 = new TGraph(23,mybeta_graph, beamSignal_cor_b2_mean); TGraph * graph_b3 = new TGraph(23,mybeta_graph, beamSignal_cor_b3_mean); graph_b0->SetName("graph_b0");graph_b0->Write(); graph_b1->SetName("graph_b1");graph_b1->Write(); graph_b2->SetName("graph_b2");graph_b2->Write(); graph_b3->SetName("graph_b3");graph_b3->Write();
rootFile->Write(); rootFile->Close(); } std::cout << eventID << " frames analysed." << std::endl; return 0; } else { std::cerr << "Error 1" << std::endl; return 1; } }
int main(int argc, char **argv){
analyse(argc, argv); }
|
