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);
  }

  

  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();
}