|
|
#include <string.h>
#include <string>
#include <TFile.h>
#include <TTree.h>
#include <map>
#include <fstream>
#include <TSystemDirectory.h>
#include <iostream>
using namespace std;
void csvToRoot(TString filename){}Int_t main(Int_t argc, const char *argv[]){ double timeoffset; int mwoffset; double timewindow; double timeoffset2; double timewindow2; if (argc==7){ timeoffset = atof(argv[2]); mwoffset = atoi(argv[3]); timewindow = atof(argv[4]); timeoffset2 = atof(argv[5]); timewindow2 = atof(argv[6]); printf("Time offsets: %f %f\n", timeoffset, timeoffset2); } else if (argc==2){ timeoffset = -0.50; //offset between ic and bpm readouts
mwoffset = 0; // offset for timestamped event. MW chamber position correlation seems to be better in other windows
timewindow = -0.099; //should be a negative number. window size in time to average over.
timeoffset2 = -0.50; //offset between ic and bpm readouts
timewindow2 = -0.05; //should be a negative number. window size in time to average over.
printf("Default timeoffsets:%f\n", timeoffset); printf("Default mwoffset:%i\n", mwoffset); } else { printf("./convert_batch PiN_run32.csv \n"); printf("or\n"); printf("./convert_batch PiN_run32.csv -0.25 2 -0.312 -0.21 0.05\n"); return 1; } const char *dbasename = "/work/leverington/beamprofilemonitor/hitdata/HIT_01-12-2016/database.csv"; const char *dirname = "/work/leverington/beamprofilemonitor/hitdata/HIT_01-12-2016/with_timestamp/csv/"; const char *outputdirname = "/work/leverington/beamprofilemonitor/hitdata/HIT_01-12-2016/with_timestamp/pin/"; const char *dirCsv = "/work/leverington/beamprofilemonitor/hitdata/HIT_01-12-2016/hit_csv/with_timestamp/"; const char *ext = ".csv";
//read database
string key, value; ifstream infile(dbasename, ios::in); map<string, string> database; while (infile >> key >> value) { database[key] = value; // cout << value << endl;
}
TSystemDirectory dir(dirname, dirname); if (true) { TSystemFile *file; TString fname = argv[1]; if (fname.EndsWith(ext)) {
TTree *tree = new TTree("t", "t"); cout << fname << endl; tree->ReadFile(TString(TString(dirname) + fname.Data()), "time/D:event_id:sync_out:sync_in:data_ok:led_voltage:current:ch00:ch01:ch02:ch03:ch04:ch05:ch06:ch07:ch08:ch09:ch10:ch11:ch12:ch13:ch14:ch15:ch16:ch17:ch18:ch19:ch20:ch21:ch22:ch23:ch24:ch25:ch26:ch27:ch28:ch29:ch30:ch31:ch32:ch33:ch34:ch35:ch36:ch37:ch38:ch39:ch40:ch41:ch42:ch43:ch44:ch45:ch46:ch47:ch48:ch49:ch50:ch51:ch52:ch53:ch54:ch55:ch56:ch57:ch58:ch59:ch60:ch61:ch62:ch63:ch64:ch65:ch66:ch67:ch68:ch69:ch70:ch71:ch72:ch73:ch74:ch75:ch76:ch77:ch78:ch79:ch80:ch81:ch82:ch83:ch84:ch85:ch86:ch87:ch88:ch89:ch90:ch91:ch92:ch93:ch94:ch95:ch96:ch97:ch98:ch99:ch100:ch101:ch102:ch103:ch104:ch105:ch106:ch107:ch108:ch109:ch110:ch111:ch112:ch113:ch114:ch115:ch116:ch117:ch118:ch119:ch120:ch121:ch122:ch123:ch124:ch125:ch126:ch127", '\t'); // tree->ReadFile(TString(TString(dirname) + fname.Data()), "time/D:event_id:sync_out:sync_in:data_ok:led_voltage:current:ch00:ch01:ch02:ch03:ch04:ch05:ch06:ch07:ch08:ch09:ch10:ch11:ch12:ch13:ch14:ch15:ch16:ch17:ch18:ch19:ch20:ch21:ch22:ch23:ch24:ch25:ch26:ch27:ch28:ch29:ch30:ch31:ch32:ch33:ch34:ch35:ch36:ch37:ch38:ch39:ch40:ch41:ch42:ch43:ch44:ch45:ch46:ch47:ch48:ch49:ch50:ch51:ch52:ch53:ch54:ch55:ch56:ch57:ch58:ch59:ch60:ch61:ch62:ch63", '\t');
TString outname = TString(outputdirname) + fname ; outname.Replace(outname.Length() - 4, 4, TString(TString(argv[2])+".root")); TFile *outfile = new TFile(outname.Data(), "recreate"); outfile->cd(); //outfile = tree->GetCurrentFile();
TTree *tree2 = new TTree("t2", "t2");
fname.Replace(fname.Length() - 3, 3, TString("root")); string key = string(fname.Data()); key.erase(key.end() - 5, key.end()); if(database[key] == "nothing"){ outfile->Close(); return 1; } printf("%s\n", database[key].c_str()); string value = string(dirCsv) + database[key] + ".csv"; printf("name1: %s , name2: %s\n", key.data(), value.data());
tree2->ReadFile(value.data(), "TIME2/D:IC1:IC2:MW1_FOCUSX:MW1_FOCUSY:MW2_FOCUSX:MW2_FOCUSY:MW1_POSX:MW1_POSY:MW2_POSX:MW2_POSY:TCS-SPAM:TCS-KO:BNEXTPOINT:ANALOG_IN1:ANALOG_IN2:ENERGY:INTENSITY:ION-SORT", '\t');
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 time; double energy; double intensity; double ionsort;
double time2;
tree->SetBranchAddress("time", &time);
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", &energy); tree2->SetBranchAddress("INTENSITY", &intensity); tree2->SetBranchAddress("ION-SORT", &ionsort);
TBranch *bic1 = tree->Branch("ic1", &ic1_avg, "ic1/D"); TBranch *bic2 = tree->Branch("ic2", &ic2_avg, "ic2/D"); TBranch *bmw1_focusx = tree->Branch("mw1_focusx", &mw1_focusx_avg, "mw1_focusx/D"); TBranch *bmw1_focusy = tree->Branch("mw1_focusy", &mw1_focusy_avg, "mw1_focusy/D"); TBranch *bmw2_focusx = tree->Branch("mw2_focusx", &mw2_focusx_avg, "mw2_focusx/D"); TBranch *bmw2_focusy = tree->Branch("mw2_focusy", &mw2_focusy_avg, "mw2_focusy/D"); TBranch *bmw1_posx = tree->Branch("mw1_posx", &mw1_posx_avg, "mw1_posx/D"); TBranch *bmw1_posy = tree->Branch("mw1_posy", &mw1_posy_avg, "mw1_posy/D"); TBranch *bmw2_posx = tree->Branch("mw2_posx", &mw2_posx_avg, "mw2_posx/D"); TBranch *bmw2_posy = tree->Branch("mw2_posy", &mw2_posy_avg, "mw2_posy/D"); TBranch *benergy = tree->Branch("energy", &energy, "energy/D"); TBranch *bintensity = tree->Branch("intensity", &intensity, "intensity/D"); TBranch *bionsort = tree->Branch("ionsort", &ionsort, "ionsort/D");
int currentEntryTree2 = 1;
int nevents = tree->GetEntries(); int nevents2 = tree2->GetEntries(); ic1_avg = 0; ic2_avg = 0; mw1_focusx_avg = 0; mw1_focusy_avg = 0; mw2_focusx_avg = 0; mw2_focusy_avg = 0; mw1_posx_avg = 0; mw1_posy_avg = 0; mw2_posx_avg = 0; mw2_posy_avg = 0; time2 = 0; tree2->GetEntry(currentEntryTree2);
for (int i = 0; i < nevents; i++) {
tree->GetEntry(i);
if (i % 10000 == 0) {
printf("merging event %d ,", i); printf("Time %f \n", time); printf("Entry hit %d ,", currentEntryTree2); printf("Time hit %f \n", time2); }
int count = 0; int count2 = 0;
while ( time2 < time + timeoffset && currentEntryTree2 < nevents2) { if (time2 - time - timeoffset > timewindow) { tree2->GetEntry(currentEntryTree2); if (ic1>0) ic1_avg += ic1; if (ic2>0) ic2_avg += ic2; if (ic1>0) count++; // if (i % 2000 == 0) printf("%i %2.3f %2.3f %2.3f %2.3f %2.3f %i \n", count, ic1, time2, time, timeoffset, time2 - time - timeoffset, mwoffset);
energy = energy; intensity = intensity; ionsort = ionsort; } tree2->GetEntry(currentEntryTree2); if ( time2 - time - timeoffset2 > timewindow2) { tree2->GetEntry(currentEntryTree2 + mwoffset); //why currentEtryTree2-4?
mw1_focusx_avg += mw1_focusx; mw1_focusy_avg += mw1_focusy; mw2_focusx_avg += mw2_focusx; mw2_focusy_avg += mw2_focusy; mw1_posx_avg += mw1_posx; mw1_posy_avg += mw1_posy; mw2_posx_avg += mw2_posx; mw2_posy_avg += mw2_posy;
if (i % 2000 == 0) printf("%i %2.3f %2.3f %2.3f %2.3f %2.3f %2.3f %i \n", count2, mw1_posx, mw2_posx, time2, time, timeoffset, time2 - time - timeoffset, mwoffset); 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_focusx_avg /= double(count2); mw1_focusy_avg /= double(count2); mw2_focusx_avg /= double(count2); mw2_focusy_avg /= double(count2); mw1_posx_avg /= double(count2); //the positions are weighted by the charge
mw1_posy_avg /= double(count2); mw2_posx_avg /= double(count2); mw2_posy_avg /= double(count2); } if(count>0){ ic1_avg /= double(count); ic2_avg /= double(count); } // if (i % 2000 == 0) cout << ic1_avg << " " << ic2_avg << endl;
if (i % 2000 == 0) cout << mw1_posx_avg << " " << mw2_posx_avg << endl;
bic1->Fill(); bic2->Fill(); bmw1_focusx->Fill(); bmw1_focusy->Fill(); bmw2_focusx->Fill(); bmw2_focusy->Fill(); bmw1_posx->Fill(); bmw1_posy->Fill(); bmw2_posx->Fill(); bmw2_posy->Fill(); bintensity->Fill(); benergy->Fill(); bionsort->Fill();
ic1_avg = 0.; ic2_avg = 0.; mw1_focusx_avg = 0.; mw1_focusy_avg = 0.; mw2_focusx_avg = 0.; mw2_focusy_avg = 0.; mw1_posx_avg = 0.; mw1_posy_avg = 0.; mw2_posx_avg = 0.; mw2_posy_avg = 0.; } tree->Write(); outfile->Close(); } } return 0;}
|
