JPulseFinder_IO.hh

Go to the documentation of this file.

#ifndef __INPUT_OUTPUT__
#define __INPUT_OUTPUT__
 
//c++ standard library
#include <iostream>
 
// Jpp
#include "JDAQ/JDAQTimeslice.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JModuleLocation.hh"
 
//root
#include "TVector.h"
 
// namespaces
using namespace std;
using namespace KM3NETDAQ ; // for JDAQTimeSlice
using namespace JSUPPORT;   // for JFileScanner
using namespace JLANG;   // for JMultipleFileScanner
using namespace JDETECTOR; // for JModuleLocation
 
 
 
 
/**
 * \author rgruiz
 */
 
 
 
/**
 * Structure to store the different command line arguments for JPulseFinder. 
 */
struct IO{
 
  JMultipleFileScanner<> ifnames ;
 
  string ofname ;
 
  string detector_file ;
 
  int string_number ;
 
  unsigned int pulse_period_16ns ;
 
  int up_pmts ;
 
  int down_pmts ;
 
  int number_neighbors ;
 
  int overall_delay_16ns ;
 
  int stagger_16ns ;
 
  double voltage ;
 
};
 
 
/**
 * Parses the command line options and fills an IO structure with them. 
 *
 * \param options an option structure
 * \param argc the number of command line arguments
 * \param argv the command line arguments
 *
 * \return 1 if works 2 if it doesn't work
 */
inline int read_user_options(IO &options, int argc, char **argv){
 
  int a = 0;
 
  try {
 
    JParser<string> zap;
 
    zap["o"] = make_field(options.ofname) = "out.root" ;
 
    zap["f"] = make_field(options.ifnames) ; 
 
    zap["a"] = make_field(options.detector_file) ;
 
    zap["V"] = make_field(options.voltage) = 0;
 
    zap["s"] = make_field(options.string_number) ;
 
    zap["p"] = make_field(options.pulse_period_16ns) ;
 
    zap["n"] = make_field(options.number_neighbors) = 2 ;
 
    zap["t"] = make_field(options.up_pmts) = 1 ;
 
    zap["b"] = make_field(options.down_pmts) = 4 ;
 
    zap["d"] = make_field(options.overall_delay_16ns) = 0;
 
    zap["S"] = make_field(options.stagger_16ns) = 0;
 
    if (zap.read(argc, argv) != 0)
 
      a = 1;
 
  }
 
  catch(const exception &error) {
 
    ERROR(error.what() << endl);
 
    a = 2;
 
  }
 
  return a;
 
}
 
 
 
/**
 * Writes a .root file with the nanobeacon pulses found by JPulseFinder.
 *
 * \param options an IO structure
 * \param ToT_vs_time vector of TH2D with the nanobeacon pulses
 * \param ToT vector of TH1 with the ToT distribution of the nanobeacon pulses
 * \param time vector of TH1 with the time distribution of the nanobeacon pulses
 */
void write_output(IO options , vector < vector < TH2D* > > ToT_vs_time , vector < vector < TH1D* > > ToT , vector < vector < TH1D* > > time){
 
  JMultipleFileScanner<JDAQTimeslice> inputFile(options.ifnames) ;
 
  int run_number = 0 ;
  
  inputFile.rewind() ;
 
  if(inputFile.hasNext()){
 
    run_number = inputFile.next()->getRunNumber() ;
 
  }
 
  TFile* f = new TFile(options.ofname.c_str(),"RECREATE") ;
 
  TVectorD Run (2) ;
 
  Run[0] = run_number ;
 
  Run[1] = options.voltage ;
 
  Run.Write("Run_Info") ;
 
  for( int i = 0 ; i <  (int)ToT_vs_time.size() ; ++i ) {
 
    char dirname[200] ;
 
    sprintf(dirname, "S%iF%i", options.string_number, i + 1 ) ;
 
    f->mkdir(dirname)->cd() ;
 
    for( int j = 0 ; j < (int)ToT_vs_time[0].size()  ; ++j ) {
      
      if (ToT_vs_time[i][j]==NULL) continue ;
 
      if( ToT_vs_time[i][j]->Integral(1 , ToT_vs_time[i][j]->GetNbinsX() - 1 , 1 , ToT_vs_time[i][j]->GetNbinsY() - 1)>0 ) {
 
        ToT_vs_time[i][j]->Write() ;
        
        ToT[i][j]->Write() ;
        
        time[i][j]->Write() ;
        
      }     
      
    }
    
    f->cd() ;
 
  }
 
  f->Close() ;
 
  delete (f) ;
 
}
 
#endif