JMergeEvt.cc

Go to the documentation of this file.

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

#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"

#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
#include "Jeep/JeepToolkit.hh"

#include "JSupport/JMeta.hh"
#include "JSupport/JSupport.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JEvtWeightFileScannerSet.hh"

#include "JAAnet/JHead.hh"


/**
 * \file
 * Example program for merging Monte-Carlo files with compatible headers.
 *
 * Note: Already existing merged files will be overwritten.
 *
 * \author bjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef typename JTYPELIST<Head, JMeta, Evt>::typelist             typelist;
  
  JMultipleFileScanner_t  inputFiles;
  string                  outputFile;

  char                    wildcard;

  int                     debug;

  try {
    
    JParser<> zap;
    
    zap['f'] = make_field(inputFiles);
    zap['o'] = make_field(outputFile)         = "%.merged.root";
    zap['w'] = make_field(wildcard)           = '%';
    zap['d'] = make_field(debug)              = 1;

    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }

  const size_t pos = outputFile.find(wildcard);
  
  if (pos == string::npos) {
    FATAL("Valid wildcard must be specified (<" << outputFile << "> does not contain \'" << wildcard << "\').");
  }
  
  
  // Create set of files ordered based on header-info
  
  JEvtWeightFileScannerSet<> scanners(inputFiles);
  
  
  // Create merged output for all compatible files

  for (JEvtWeightFileScannerSet<>::iterator scanner = scanners.begin(); scanner != scanners.end(); ++scanner) {

    const JHead& header     = scanner->getHeader();
    
    const string identifier = scanners.getUniqueIdentifier(scanner);
    const string buffer     = string(outputFile).replace(pos, 1, identifier.c_str());

    NOTICE("Writing " << buffer << "..." << endl);
    
    JFileRecorder<typelist> out(buffer.c_str());
    
    out.open();

    Head head;
    copy(header, head);
    out.put(head);
    
    out.put(JMeta(argc, argv));

    for (JMultipleFileScanner<JMeta> in(scanner->getFilelist()); in.hasNext(); ) {
      out.put(*in.next());
    }
    
    while (scanner->hasNext()) {
      
      const Evt* event = scanner->next();

      if (event != NULL) {
      
        STATUS("Event:" << RIGHT(10) << scanner->getCounter() << '\r'); DEBUG(endl);
        out.put(*event);
        
      } else {

        WARNING("Event " << scanner->getCounter() << " is empty; skip.");
      }
    }

    out.close();
  }
  
  return 0;
}