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["stagger"] = 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