JRoot.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <fstream>
#include <iomanip>
#include <vector>
#include <limits>
#include <string.h>
 
#include "TROOT.h"
#include "TFile.h"
 
#include "JDAQ/JDAQEventIO.hh"
#include "JDAQ/JDAQTimesliceIO.hh"
#include "JDAQ/JDAQSummarysliceIO.hh"
 
#include "JROOT/JTreeReader.hh"
#include "JIO/JStreamIO.hh"
#include "JIO/JByteArrayIO.hh"
#include "JSupport/JSupport.hh"
#include "JSupport/JAutoTreeWriter.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
namespace JSUPPORT {
 
  /**
   * Get key for given DAQ data type.
   *
   * \param  type             data type
   * \return                  map element
   */
  template<>
  template<class T>
  int JAutoTreeWriter<int>::getKey(JLANG::JType<T> type)
  {
    return KM3NETDAQ::getDataType<T>();
  }
}
 
 
/**
 * \file
 * Auxiliary program to convert binary data to ROOT formatted data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
 
  vector<string> inputFile;
  string         outputFile;
  Long64_t       numberOfEvents;
  int            debug;
 
  try { 
 
    JParser<> zap("Auxiliary program to convert binary data to ROOT formatted data.");
    
    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile);
    zap['n'] = make_field(numberOfEvents)  = numeric_limits<Long64_t>::max();
    zap['d'] = make_field(debug)           = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
  
  cout.tie(&cerr);
 
  using namespace KM3NETDAQ;
  using namespace JPP;
 
 
  JAutoTreeWriter<int> writer;
 
  writer.open(outputFile.c_str());
 
  if (!writer.is_open()) {
    FATAL("Error opening file " << outputFile << endl);
  }
 
  writer.insert<JDAQTypes_t>();
 
 
  JDAQPreamble preamble;
  vector<char> buffer;
 
  Long64_t event_count = 0;
  
  for (vector<string>::const_iterator i = inputFile.begin(); i != inputFile.end(); ++i) {
    
    STATUS("File: " << *i << endl);
    
    ifstream in(i->c_str());
    
    for (JStreamReader is(in); event_count < numberOfEvents; ++event_count) {
 
      STATUS("event: " << setw(10) << event_count << '\r'); DEBUG(endl);
 
      if (!(is >> preamble)) {
        break;
      }
 
      DEBUG("length   " << preamble.getLength()   << endl);
      DEBUG("type     " << preamble.getDataType() << endl);
 
      if (preamble.getLength() < (int) getSizeof<JDAQPreamble>()) {
 
        ERROR("File " << *i << " invalid length " << preamble.getLength() << endl);
 
        break;
      }
 
      JAutoTreeWriter<int>::iterator i = writer.find(preamble.getDataType());
 
      if (i != writer.end()) {
 
        buffer.resize(preamble.getLength());
 
        memcpy(buffer.data(), static_cast<JDAQAbstractPreamble*>(&preamble), getSizeof<JDAQPreamble>());
        
        is.read(buffer.data()        + getSizeof<JDAQPreamble>(), 
                preamble.getLength() - getSizeof<JDAQPreamble>());
        
        JByteArrayReader bin(buffer.data(), buffer.size());
 
        i->second->copy(bin);
 
      } else {
 
        in.ignore((streamsize) (preamble.getLength() - getSizeof<JDAQPreamble>()));
 
        ERROR("Illegal data type, skip" << endl);
      }
    }
    STATUS(endl);
 
    in.close();
  }
 
  writer.close();
}