266 lines
9.1 KiB
C++
266 lines
9.1 KiB
C++
/**********************************************************/
|
|
/* */
|
|
/* Modified by Ryan From */
|
|
/* */
|
|
/* PXI SCAN CODE -- J.M. Allmond (ORNL) -- July 2016 */
|
|
/* */
|
|
/**********************************************************/
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <math.h>
|
|
#include <stdbool.h>
|
|
|
|
#include "TFile.h"
|
|
#include "TTree.h"
|
|
#include "TMath.h"
|
|
#include "TBenchmark.h"
|
|
|
|
#define RAND ((float) rand() / ((unsigned int) RAND_MAX + 1)) // random number in interval (0,1)
|
|
|
|
#define MAX_CRATES 2
|
|
#define MAX_BOARDS_PER_CRATE 13
|
|
#define MAX_CHANNELS_PER_BOARD 16
|
|
#define BOARD_START 2
|
|
|
|
#define MAX_ID MAX_CRATES*MAX_BOARDS_PER_CRATE*MAX_CHANNELS_PER_BOARD
|
|
|
|
#include "../mapping.h"
|
|
|
|
#include "../armory/evtReader.h"
|
|
|
|
unsigned long long int dataCount=0;
|
|
unsigned long long int pileUpCount=0;
|
|
unsigned long long int evtCount=0;
|
|
int tick2ns = 4;
|
|
|
|
///////////////////////////////////
|
|
// START OF MAIN FUNCTION //
|
|
///////////////////////////////////
|
|
int main(int argc, char **argv) {
|
|
|
|
printf("=====================================\n");
|
|
printf("=== evt --> root ===\n");
|
|
printf("=====================================\n");
|
|
|
|
// Check that the corrent number of arguments were provided.
|
|
if (argc <= 3) {
|
|
printf("Incorrect number of arguments:\n");
|
|
printf("%s [outFile] [timeWindow] [to File1] [to File2] .... \n", argv[0]);
|
|
printf(" outFile : output root file name\n");
|
|
printf(" timeWindow : number of tick, 1 tick = %d ns. default = 100 \n", tick2ns);
|
|
return 1;
|
|
}
|
|
|
|
TString outFileName = argv[1];
|
|
int timeWindow = atoi(argv[2]);
|
|
int nFile = argc - 3;
|
|
TString inFileName[nFile];
|
|
for( int i = 0 ; i < nFile ; i++){
|
|
inFileName[i] = argv[i+3];
|
|
}
|
|
|
|
printf("====================================\n");
|
|
|
|
evtReader * evt = new evtReader();
|
|
DataBlock * data = evt->data;
|
|
|
|
printf(" Number of input file : %d \n", nFile);
|
|
printf(" out file : \033[1;31m%s\033[m\n", outFileName.Data());
|
|
printf(" Event building time window : %d tics = %d nsec \n", timeWindow, timeWindow*tick2ns);
|
|
|
|
|
|
TFile * outRootFile = new TFile(outFileName, "recreate");
|
|
outRootFile->cd();
|
|
TTree * tree = new TTree("tree", outFileName);
|
|
|
|
unsigned long long evID = 0;
|
|
int multi = 0;
|
|
int id[MAX_ID] = {0};
|
|
double e[MAX_ID] = {TMath::QuietNaN()};
|
|
unsigned long long e_t[MAX_ID] = {0};
|
|
int cfd[MAX_ID] = {0};
|
|
bool pileup[MAX_ID] = {0};
|
|
int qdc[MAX_ID][8] = {0};
|
|
int multiClover = 0 ; /// this is total multiplicity for all crystal
|
|
int runID = 0; // date-run-fileID
|
|
int multiBeam = 0;
|
|
//unsigned short pileup[MAXMULTI];
|
|
|
|
tree->Branch("evID", &evID, "event_ID/l");
|
|
tree->Branch("multi", &multi, "multi/I");
|
|
tree->Branch("detID", id, "detID[multi]/I");
|
|
tree->Branch("e", e, "e[multi]/D");
|
|
tree->Branch("e_t", e_t, "e_timestamp[multi]/l");
|
|
//tree->Branch("pileup", pileup, "pileup[multi]/O");
|
|
tree->Branch("cfd", cfd, "cfd[multi]/I");
|
|
tree->Branch("qdc", qdc, "qdc[multi][8]/I");
|
|
tree->Branch("multiClover", &multiClover, "multiplicity_Clover/I");
|
|
tree->Branch("multiBeam", &multiBeam, "multiplicity_Beam/I");
|
|
tree->Branch("runID", &runID, "runID/I");
|
|
|
|
int countGP = 0; //gamma-particle coincident
|
|
double totalDataSize = 0;
|
|
int outFileCount = 0;
|
|
|
|
for( int i = 0; i < nFile; i++){
|
|
|
|
evt->OpenFile(inFileName[i], false);
|
|
if( evt->IsOpen() == false ) continue;
|
|
|
|
printf("==================================================== %d / %d\n", i+1, nFile);
|
|
printf("\033[1;31m%s \033[m\n", inFileName[i].Data());
|
|
|
|
int pos = inFileName[i].Last('/');
|
|
TString temp = inFileName[i];
|
|
temp.Remove(0, pos+1);
|
|
pos = temp.Last('-');
|
|
temp.Remove(0, pos+1);
|
|
pos = temp.Last('.');
|
|
temp.Remove(pos);
|
|
runID = atoi(temp.Data());
|
|
|
|
long long int etime = -1;
|
|
long long int tdif = -1;
|
|
|
|
while ( evt->IsEndOfFile() == false ) { //main while loop
|
|
|
|
if ( evt->ReadBlock() == -1) break;
|
|
|
|
if ( data->crate == 0 ) continue;
|
|
|
|
//Set reference time for event building
|
|
if (etime == -1) {
|
|
etime = data->time;
|
|
tdif = 0;
|
|
multi = 0;
|
|
multiClover = 0;
|
|
multiBeam = 0;
|
|
}else {
|
|
tdif = data->time - etime;
|
|
}
|
|
|
|
//Check for end of event, rewind, and break out of while loop
|
|
if (tdif > timeWindow) {
|
|
|
|
//Gate
|
|
//if( multiClover > 2 && multiBeam ==0 ) {
|
|
//
|
|
outRootFile->cd();
|
|
tree->Fill();
|
|
//printf("------------------\n");
|
|
//
|
|
// countGP++;
|
|
//}
|
|
|
|
evID ++;
|
|
|
|
//clear data
|
|
etime = data->time;
|
|
tdif = 0;
|
|
multi = 0;
|
|
multiClover = 0;
|
|
multiBeam = 0;
|
|
|
|
int haha = data->crate*MAX_BOARDS_PER_CRATE*MAX_CHANNELS_PER_BOARD + (data->slot-BOARD_START)*MAX_CHANNELS_PER_BOARD + data->ch;
|
|
id[multi] = mapping[haha];
|
|
e[multi] = data->energy;
|
|
e_t[multi] = data->time;
|
|
cfd[multi] = data->cfd_forced == 0 ? (data->cfd - 16384*(data->cfd_source == true ? 1 : 0))/2 : 0;
|
|
pileup[multi] = data->pileup;
|
|
for( int i = 0; i < 8; i++) qdc[multi][i] = data->QDCsum[i];
|
|
|
|
if( 0 <= id[multi] && id[multi] < 100 ) multiClover += 1;
|
|
if( id[multi] == 300 || id[multi] == 301 || id[multi] == 307 || id[multi] == 308 ) multiBeam += 1;
|
|
//printf("id: %d, multiBeam: %d, %llu, tdif %llu\n", id[multi], multiBeam, e_t[multi], tdif);
|
|
multi++ ;
|
|
}else{
|
|
//if within time window, fill array;
|
|
int haha = data->crate*MAX_BOARDS_PER_CRATE*MAX_CHANNELS_PER_BOARD + (data->slot-BOARD_START)*MAX_CHANNELS_PER_BOARD + data->ch;
|
|
id[multi] = mapping[haha];
|
|
e[multi] = data->energy;
|
|
e_t[multi] = data->time;
|
|
cfd[multi] = data->cfd_forced == 0 ? (data->cfd - 16384*(data->cfd_source == true ? 1 : 0))/2 : 0;
|
|
//printf(" %d , %d , %d, %d \n", data->cfd_forced, data->cfd_source, data->cfd, cfd[multi]);
|
|
pileup[multi] = data->pileup;
|
|
for( int i = 0; i < 8; i++) qdc[multi][i] = data->QDCsum[i];
|
|
if( 0 <= id[multi] && id[multi] < 100 ) multiClover += 1;
|
|
if( id[multi] == 300 || id[multi] == 301 || id[multi] == 307 || id[multi] == 308 ) multiBeam += 1;
|
|
//printf("id: %d, multiBeam: %d, %llu, tdif %llu\n", id[multi], multiBeam, e_t[multi], tdif);
|
|
multi++ ;
|
|
|
|
}
|
|
|
|
// total pileups
|
|
if (data->pileup == 1) {
|
|
pileUpCount++;
|
|
}
|
|
|
|
evt->PrintStatus(10000);
|
|
|
|
} // end main while loop
|
|
|
|
evt->PrintStatus(1);
|
|
printf("\n\n\n");
|
|
printf(" total number of event built : %llu\n", evID);
|
|
//printf(" total number of Gamma - GAGG coincdient : %d (%.3f %%)\n", countGP, countGP*1.0/evID);
|
|
|
|
outRootFile->cd();
|
|
tree->Write();
|
|
|
|
totalDataSize += (evt->GetFileSize())/1024./1024./1024.;
|
|
|
|
double rootFileSize = outRootFile->GetSize()/1024./1024./1024. ; // in GB
|
|
printf(" ----------- root file size : %.3f GB\n", rootFileSize);
|
|
printf(" ---------- total read size : %.3f GB\n", totalDataSize);
|
|
printf(" ----------- reduction rate : %.3f %%\n", rootFileSize*100./totalDataSize);
|
|
|
|
evt->CloseFile();
|
|
|
|
/*
|
|
if( rootFileSize > 3.0 ) {
|
|
break;
|
|
}*/
|
|
|
|
///try to open a new root file when file size > 2 GB
|
|
/*if( rootFileSize > 2.0 ) {
|
|
|
|
outRootFile->Close();
|
|
delete outRootFile;
|
|
delete tree;
|
|
outFileCount += 1;
|
|
|
|
if( outFileCount > 5 ) break;
|
|
|
|
TString outFileName2 = outFileName;
|
|
outFileName2.Insert(outFileName.Sizeof() - 6, Form("_%03d",outFileCount));
|
|
outRootFile = new TFile( outFileName2, "recreate");
|
|
|
|
tree = new TTree("tree", "tree");
|
|
tree->Branch("evID", &evID, "event_ID/l");
|
|
tree->Branch("multi", &multi, "multi/I");
|
|
tree->Branch("detID", id, "detID[multi]/I");
|
|
tree->Branch("e", e, "e[multi]/D");
|
|
tree->Branch("e_t", e_t, "e_timestamp[multi]/l");
|
|
tree->Branch("qdc", qdc, "qdc[multi][8]/I");
|
|
tree->Branch("multiClover", &multiClover, "multiplicity_crystal/I");
|
|
tree->Branch("multiBeam", &multiBeam, "multiplicity_GAGG/I");
|
|
tree->Branch("runID", &runID, "runID/I");
|
|
|
|
}*/
|
|
|
|
}
|
|
|
|
|
|
outRootFile->Close();
|
|
|
|
printf("\n\n\n==================== finished.\r\n");
|
|
|
|
//printf(" number of Gamma - GAGG coincdient : %d\n", countGP);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|