dat2root.cc

#include <TFile.h>
#include <TTree.h>
#include <TROOT.h>
#include <TRandom.h>
#include <TApplication.h>
#include <TCanvas.h>
#include <TPostScript.h>
#include <TAxis.h>
#include <TH1F.h>
#include <TH2F.h>
#include <TGraph.h>
#include <TGraphErrors.h>
#include <TLegend.h>
#include <TF1.h>
#include <TLine.h>
#include <TStyle.h>
#include <TProfile.h>
#include <TMapFile.h>
#include <TPaveStats.h>


#include <fstream>
#include <iomanip>
#include <string>
#include <iostream>

#include <stdio.h>
#include <stdlib.h>
#include <limits.h>  
#include <string.h>
#include <unistd.h>
#include <sys/mman.h>
#include <errno.h>
#include <ctype.h>
#include <math.h>
#include <time.h>

//LOCAL INCLUDES
#include "Aux.hh"

using namespace std;

std::string ParseCommandLine( int argc, char* argv[], std::string opt )
{
  for (int i = 1; i < argc; i++ )
    {
      std::string tmp( argv[i] );
      if ( tmp.find( opt ) != std::string::npos )
        {
          if ( tmp.find( "=" )  != std::string::npos ) return tmp.substr( tmp.find_last_of("=") + 1 );
	  if ( tmp.find( "--" ) != std::string::npos ) return "yes";
	}
    }
  
  return "";
};


int graphic_init();

TStyle* style;


int main(int argc, char **argv){

  FILE* fp1;
  char stitle[200];
  int dummy;

  std::cout << "===Beginning program===" << std::endl;
  //**************************************
  //Arguments
  //**************************************
  std::string inputFilename = argv[1];
  std::string outputFilename = (inputFilename + "-full.root").c_str();

  int nEvents = atoi(argv[2]);

  std::string boardNumber = "1";
  std::string bNumber = ParseCommandLine( argc, argv, "--boardNumber" );
  if ( bNumber == ""  )
    {
      std::cerr << "[ERROR]: WRONG BOARD NUMBER--> " << bNumber << std::endl;
      boardNumber = "1";
    }
  else if ( bNumber == "1" || bNumber == "2" )
    {
       boardNumber = bNumber;
    }
  else
    {
      std::cerr << "[ERROR]: WRONG BOARD NUMBER--> " << bNumber << std::endl;
    }
  
 
  std::cout << "Using Calibration files for board number " << boardNumber << "\n";

  bool saveRaw = false;
  std::string _saveRaw = ParseCommandLine( argc, argv, "--saveRaw" );
  if ( _saveRaw == "yes" ) saveRaw = true;
  if (saveRaw) std::cout << "Saving Raw Pulses\n";

  std::string _drawDebugPulses = ParseCommandLine( argc, argv, "--debug" );
  bool drawDebugPulses = false;
  if ( _drawDebugPulses == "yes" ) {
    drawDebugPulses = true;
    std::cout << "draw: " << drawDebugPulses << std::endl;
  }

  bool doFilter = false;
  std::string _doFilter = ParseCommandLine( argc, argv, "--doFilter" );
  if ( _doFilter == "yes" ) saveRaw = true;
  if (doFilter) std::cout << "Using Algorithmic Frequency Filtering\n";

  bool doAmplificationCorrection = true;
  std::string _doAmplificationCorrection = ParseCommandLine( argc, argv, "--doAmplificationCorrection" );
  if ( _doAmplificationCorrection == "yes" ) doAmplificationCorrection = true;
  if (doAmplificationCorrection) std::cout << "Do Amplification Correction for Silicon Sensor Channel: 21\n";


  //**************************************
  //Load Voltage Calibration
  //**************************************
  std::cout << "===Loading Voltage Calibration===" << std::endl;
  double off_mean[4][9][1024];
  for( int i = 0; i < 4; i++){
    sprintf( stitle, "v1740_bd%s_group_%d_offset.txt", boardNumber.c_str(), i);

    fp1 = fopen( stitle, "r");
    printf("offset data : %s\n", stitle);

    for( int k = 0; k < 1024; k++)      
      for( int j = 0; j < 9; j++){      
	dummy = fscanf( fp1, "%lf ", &off_mean[i][j][k]);       
	if( k < 2 && 0)
	  printf("%5d  %8.4f\n", j, off_mean[i][j][k]);
      }
  
    fclose(fp1);
  }


  //**************************************
  //Load Time Calibration
  //**************************************
  double fdummy;
  double tcal_dV[4][1024];
  for( int i = 0; i < 4; i++){
    sprintf( stitle, "v1740_bd%s_group_%d_dV.txt", boardNumber.c_str(), i);

    fp1 = fopen( stitle, "r");
    printf("dV data : %s\n", stitle);

    for( int k = 0; k < 1024; k++)      
	dummy = fscanf( fp1, "%lf %lf %lf %lf %lf ", 
		        &fdummy, &fdummy, &fdummy, &fdummy, &tcal_dV[i][k]);       
    fclose(fp1);
  }
  double dV_sum[4] = {0, 0, 0, 0};
  for( int i = 0; i < 4; i++)
    for( int j = 0; j < 1024; j++)
    dV_sum[i] += tcal_dV[i][j];


  double tcal[4][1024];
  for( int i = 0; i < 4; i++)
    for( int j = 0; j < 1024; j++)
      {
	tcal[i][j] = tcal_dV[i][j]/dV_sum[i]*200.0;
      }

  
  //**************************************
  //Open output file, Define output Tree
  //**************************************
  TFile* file = new TFile( outputFilename.c_str(), "RECREATE", "CAEN V1742");
  TTree* tree = new TTree("pulse", "Wave Form");

  int event;
  short b_c[4][9][1024], tc[4]; 
  float time[4][1024];
  short raw[36][1024];
  short channel[36][1024];
  float channelCorrected[36][1024];
  float xmin[36];
  float base[36];
  float amp[36];
  float integral[36];
  float integralFull[36];
  float gauspeak[36];
  float linearTime0[36];
  float linearTime15[36];
  float linearTime30[36];
  float linearTime45[36];
  float linearTime60[36];
  int t[36864];
  int t0[1024];
 
  tree->Branch("event", &event, "event/I");
  tree->Branch("tc",   tc, "tc[4]/s");
  if (saveRaw) {
    tree->Branch("b_c",  b_c, "b_c[36864]/s"); //this is for 9 channels per group
    tree->Branch("raw", raw, "raw[36][1024]/S");   
    tree->Branch("t",  t, "t[36864]/I");    
  }
  tree->Branch("channel", channel, "channel[36][1024]/S");
  tree->Branch("t0",  t0, "t0[1024]/I");
  tree->Branch("time", time, "time[4][1024]/F");
  tree->Branch("xmin", xmin, "xmin[36]/F");
  tree->Branch("amp", amp, "amp[36]/F");
  tree->Branch("base", base, "base[36]/F");
  tree->Branch("int", integral, "int[36]/F");
  tree->Branch("intfull", integralFull, "intfull[36]/F");
  tree->Branch("gauspeak", gauspeak, "gauspeak[36]/F");
  tree->Branch("linearTime0", linearTime0, "linearTime0[36]/F");
  tree->Branch("linearTime15", linearTime15, "linearTime15[36]/F");
  tree->Branch("linearTime30", linearTime30, "linearTime30[36]/F");
  tree->Branch("linearTime45", linearTime45, "linearTime45[36]/F");
  tree->Branch("linearTime60", linearTime60, "linearTime60[36]/F");

  uint   event_header;
  uint   temp[3];
  ushort samples[9][1024];

  //define time bins
  for( int i  = 0; i < 36864; i++ ) t[i] = i;
  for( int i  = 0; i < 1024; i++ ) t0[i] = i;


  //*************************
  // Open Input File
  //*************************
  FILE* fpin = fopen( inputFilename.c_str(), "r");

  ushort _initVal = 666.0;
  int goodEvents = 0;

  std::cout << "open file" << std::endl;
  //*************************
  //Event Loop
  //*************************
  for( int eventn = 0; eventn < nEvents; eventn++){ 

    //if reached end of file, then quit
    if (feof(fpin)) break;
      
    if ( eventn%100 == 0 ) std::cout << "event: " << eventn << std::endl;
    // printf("---- loop  %5d\n", loop);
    event = goodEvents;

    //Reading First Header Word
    dummy = fread( &event_header, sizeof(uint), 1, fpin);
    uint evtSize =  event_header & 0x0fffffff;
    //Reading Second Header Word
    dummy = fread( &event_header, sizeof(uint), 1, fpin);  
    uint grM     = event_header & 0x0f;
    uint pattern = (event_header >> 8) & 0x2fff;
    uint bID     = (event_header >> 27) & 0x1f;
    //Reading Third Header Word
    dummy = fread( &event_header, sizeof(uint), 1, fpin);  
    //Reading Fourth Header Word
    dummy = fread( &event_header, sizeof(uint), 1, fpin);  

    
    //--------------------------------
    //Parsing group Mask into channels
    //--------------------------------
    bool _isGR_On[4];
    _isGR_On[0] = (grM & 0x01);
    _isGR_On[1] = (grM & 0x02);
    _isGR_On[2] = (grM & 0x04);
    _isGR_On[3] = (grM & 0x08);
    
    int ActiveGroupsN = 0;
    int realGroup[4] = {-1, -1, -1, -1};
    for ( int l = 0; l < 4; l++ )
      {
	if ( _isGR_On[l] ) 
	  {
	    realGroup[ActiveGroupsN] = l; 
	    ActiveGroupsN++;
	  }
      }
    
    if ( ActiveGroupsN < 4 )
      {
	std::cout << "----------------WARNING--------------" << std::endl;
	std::cout << "evtSize: " << evtSize << " grM: " << grM << " pattern: " << pattern 
		  << " bID: " << bID  << " number of Active groups: " << ActiveGroupsN << std::endl;
	std::cout << "------------------------------" << std::endl;
      }

    //************************************
    //Loop Over Channel Groups
    //************************************
    for( int group = 0; group < ActiveGroupsN; group++){
      //Reading Group Header
      dummy = fread( &event_header, sizeof(uint), 1, fpin);  
      
      ushort tcn = (event_header >> 20) & 0xfff;
      tc[realGroup[group]] = tcn;
      
      //Checking if all channels were active ( if 8 channels active return 3072)
      int nsample = (event_header & 0xfff)/3;
      //std::cout << "realGroup[group] #"<< realGroup[group] << "; nsample: " << nsample << std::endl;
      
      //Define Time coordinate
      time[realGroup[group]][0] = 0.0;
      for( int i = 1; i < 1024; i++){
	time[realGroup[group]][i] = float(i);
	//std::cout << "realGroup " << realGroup[group] << " " << i << " tcal --> " << tcal[0][i] << " " << time[realGroup[group]][i]  << " :::" << (i-1+tcn)%1024 << "\n";
	time[realGroup[group]][i] = float(tcal[realGroup[group]][(i-1+tcn)%1024] + time[realGroup[group]][i-1]);
	
      }      

      //************************************
      //Read Sample Info
      //************************************      
      for(int i = 0; i < nsample; i++){
	dummy = fread( &temp, sizeof(uint), 3, fpin);  
	samples[0][i] =  temp[0] & 0xfff;
	samples[1][i] = (temp[0] >> 12) & 0xfff;
	samples[2][i] = (temp[0] >> 24) | ((temp[1] & 0xf) << 8);
	samples[3][i] = (temp[1] >>  4) & 0xfff;
	samples[4][i] = (temp[1] >> 16) & 0xfff;
	samples[5][i] = (temp[1] >> 28) | ((temp[2] & 0xff) << 4);
	samples[6][i] = (temp[2] >>  8) & 0xfff;
	samples[7][i] =  temp[2] >> 20;	
      }

      for(int j = 0; j < nsample/8; j++){
	fread( &temp, sizeof(uint), 3, fpin);  
	samples[8][j*8+0] =  temp[0] & 0xfff;
	samples[8][j*8+1] = (temp[0] >> 12) & 0xfff;
	samples[8][j*8+2] = (temp[0] >> 24) | ((temp[1] & 0xf) << 8);
	samples[8][j*8+3] = (temp[1] >>  4) & 0xfff;
	samples[8][j*8+4] = (temp[1] >> 16) & 0xfff;
	samples[8][j*8+5] = (temp[1] >> 28) | ((temp[2] & 0xff) << 4);
	samples[8][j*8+6] = (temp[2] >>  8) & 0xfff;
	samples[8][j*8+7] =  temp[2] >> 20;
      }

      //std::cout << "====Event: " << event << std::endl;
      //************************************
      //Loop Over Channels 0 - 8
      //************************************      
      for(int i = 0; i < 9; i++) {

	int totalIndex = realGroup[group]*9 + i;
	
	//Fill pulses
	for(int j = 0; j < 1024; j++) {
	  b_c[realGroup[group]][i][j] = (short)(samples[i][j]);
	  raw[realGroup[group]*9 + i][j] = (short)(samples[i][j]);
	  channel[realGroup[group]*9 + i][j] = (short)((double)(samples[i][j]) - (double)(off_mean[realGroup[group]][i][(j+tcn)%1024]));
	}

	//Find the absolute minimum. This is only used as a rough determination to decide if we'll use the early time samples
	//or the late time samples to do the baseline fit
	int index_min = FindMinAbsolute(1024, channel[realGroup[group]*9 + i]); // return index of the minc

	//Make Pulse shape Graph
	TString pulseName = Form("pulse_event%d_group%d_ch%d", eventn, realGroup[group], i);
	TGraphErrors* pulse = new TGraphErrors( GetTGraph( channel[realGroup[group]*9 + i], time[realGroup[group]] ) );

	//estimate baseline
	float baseline;
        if ( index_min < 105 ) { baseline = GetBaseline( pulse, 850, 1020, pulseName);}
        else { baseline = GetBaseline( pulse, 5 ,150, pulseName);}
        base[realGroup[group]*9 + i] = baseline;

	//Correct pulse shape for baseline offset
	for(int j = 0; j < 1024; j++) {
	  channel[realGroup[group]*9 + i][j] = (short)((double)(channel[realGroup[group]*9 + i][j]) + baseline);
	  channelCorrected[realGroup[group]*9 + i][j] = channel[realGroup[group]*9 + i][j];
	}

	// DRS-glitch finder: zero out bins which have large difference
	// with respect to neighbors in only one or two bins
	for(int j = 0; j < 1024; j++) {
	  short a0 = abs(channel[realGroup[group]*9 + i][j-1]);
	  short a1 = abs(channel[realGroup[group]*9 + i][j]);
	  short a2 = abs(channel[realGroup[group]*9 + i][j+1]);
	  short a3 = abs(channel[realGroup[group]*9 + i][j+2]);
	  
	  if ( ( a1>3*a0 && a2>3*a0 && a2>3*a3 && a1>30) )
	    {
	      channel[realGroup[group]*9 + i][j] = 0;
	      channel[realGroup[group]*9 + i][j+1] = 0;
	    }
	  
	  if ( ( a1>3*a0 && a1>3*a2 && a1>30) )
	    channel[realGroup[group]*9 + i][j] = 0;
	}
	
	delete pulse;

	// Find Peak Location using the improved algorithm
	pulse = new TGraphErrors( GetTGraph( channel[realGroup[group]*9 + i], time[realGroup[group]] ) );
	//	pulse = GetTGraph( channel[realGroup[group]*9 + i], time[realGroup[group]] );
	index_min = FindRealMin (1024, channel[realGroup[group]*9 + i]); // return index of the min
	//if ( index_min > 0 ) std::cout << "ch: " << totalIndex << std::endl;
	xmin[realGroup[group]*9 + i] = index_min;
	

	if (doFilter) {
	  pulse = GetTGraphFilter( channel[realGroup[group]*9 + i], time[realGroup[group]], pulseName , false);
	}
	
	//Compute Amplitude : use units V
	Double_t tmpAmp = 0.0;
	Double_t tmpMin = 0.0;
	pulse->GetPoint(index_min, tmpMin, tmpAmp);
	amp[realGroup[group]*9 + i] = tmpAmp* (1.0 / 4096.0); 

	//Get Pulse Integral
	if ( xmin[realGroup[group]*9 + i] != 0 )
	  {
	    integral[realGroup[group]*9 + i] = GetPulseIntegral( index_min , channel[realGroup[group]*9 + i]);
	    integralFull[realGroup[group]*9 + i] = GetPulseIntegral( index_min , channel[realGroup[group]*9 + i], "full");
	  }
	else
	  {
	    integral[realGroup[group]*9 + i] = 0.0;
	    integralFull[realGroup[group]*9 + i] = 0.0;
	  }

	//Gauss Time-Stamping 
	Double_t min = 0.; Double_t low_edge =0.; Double_t high_edge =0.; Double_t y = 0.; 
	pulse->GetPoint(index_min, min, y);	
	pulse->GetPoint(index_min-3, low_edge, y); // get the time of the low edge of the fit range
	pulse->GetPoint(index_min+3, high_edge, y);  // get the time of the upper edge of the fit range	

	float timepeak = 0;
	float timecf0   = 0;
	float timecf15   = 0;
	float timecf30   = 0;
	float timecf45   = 0;
	float timecf60   = 0;
	if( drawDebugPulses)
	  {
	    if ( !(totalIndex == 8 || totalIndex == 17 || totalIndex == 26 || totalIndex == 35) ) timepeak =  GausFit_MeanTime(pulse, low_edge, high_edge, pulseName); // get the time stamp
	    float fs[5];
	    if ( xmin[realGroup[group]*9 + i] != 0.0 )
	      {
		if ( !(totalIndex == 8 || totalIndex == 17 || totalIndex == 26 || totalIndex == 35) ) RisingEdgeFitTime( pulse, index_min, fs, event, "linearFit_" + pulseName, true);
	      }
	    else
	      {
		for ( int kk = 0; kk < 5; kk++ ) fs[kk] = -999;
	      }
	    timecf0  = fs[0];
	    timecf15 = fs[1];
	    timecf30 = fs[2];
	    timecf45 = fs[3];
	    timecf60 = fs[4];	 
	  }
	else
	  {
	    timepeak =  GausFit_MeanTime(pulse, low_edge, high_edge); // get the time stamp
	    float fs[5];
	    if ( xmin[realGroup[group]*9 + i] != 0.0 )
	      {
		RisingEdgeFitTime( pulse, index_min, fs, event, "");
	      }
	   else
	     {
	       for ( int kk = 0; kk < 5; kk++ ) fs[kk] = -999;
	     }
	    timecf0  = fs[0];
	    timecf15 = fs[1];
	    timecf30 = fs[2];
	    timecf45 = fs[3];
	    timecf60 = fs[4];
	  }
	gauspeak[realGroup[group]*9 + i]   = timepeak;
	linearTime0[realGroup[group]*9 + i] = timecf0;
	linearTime15[realGroup[group]*9 + i] = timecf15;
	linearTime30[realGroup[group]*9 + i] = timecf30;
	linearTime45[realGroup[group]*9 + i] = timecf45;
	linearTime60[realGroup[group]*9 + i] = timecf60;

	delete pulse;
      }
      
      dummy = fread( &event_header, sizeof(uint), 1, fpin);
    }
    
    if ( ActiveGroupsN < 4 ) continue;
    tree->Fill();
    goodEvents++;
    
  }

  fclose(fpin);
  cout << "Processed total of " << goodEvents << " events\n";

  file->Write();
  file->Close();

  return 0;
  printf("dummy = %d\n", dummy);
}



int
graphic_init( void){

  style = new TStyle("style", "style");
  
  style->SetLabelFont(132,"X");
  style->SetLabelFont(132,"Y");
  style->SetTitleFont(132,"X");
  style->SetTitleFont(132,"Y");
  style->SetTitleFont(132,"");
  style->SetTitleFontSize( 0.07);
  style->SetStatFont(132);
  style->GetAttDate()->SetTextFont(132);

  style->SetStatW(0.20);
  style->SetStatH(0.23);

  style->SetFuncColor(2);
  style->SetFuncWidth(2);
  style->SetLineWidth(2);
  
  style->SetOptFile(0);
  style->SetOptTitle(1);

  style->SetFrameBorderMode(0);
  style->SetCanvasBorderMode(0);
  style->SetPadBorderMode(0);
  style->SetTitleStyle(4000);
  style->SetPadColor(0);
  style->SetCanvasColor(0);

  style->SetTitleFillColor(0);

  style->SetTitleBorderSize(0);
  //  style->SetTitleX(0.3);
  //  style->SetTitleY(0.06);
  style->SetStatColor(0);
  style->SetStatBorderSize(1);

  style->SetOptStat("emri");
  // style->SetOptStat(1);
  style->SetOptFit(1);
  style->SetTitleOffset( 1.0,"Y");

  style->SetMarkerStyle(20);
  style->SetMarkerSize( 0.3);
  style->SetMarkerColor(4);

  // style->SetOptDate(21);

  //  style->SetPadGridX(1);
  //  style->SetPadGridY(1);


  style->cd();

  return 0;
}