JMergeEvt.cc

Go to the documentation of this file.

#include <uuid/uuid.h>
#include <iostream>
#include <iomanip>
#include <string>

#include "km3net-dataformat/definitions/weightlist.hh"

#include "km3net-dataformat/offline/MultiHead.hh"
#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 "JAAnet/JHead.hh"

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


/**
 * \file
 * Application for merging Monte-Carlo files.
 *
 * \author bjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;

  typedef typename JTYPELIST<JAAnetTypes_t, JMetaTypes_t>::typelist         typelist;
  
  JMultipleFileScanner_t  inputFiles;
  JFileRecorder<typelist> outputFile;

  int                     debug;

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

    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }
  
  
  // Create set of files ordered based on header-info
  
  JEvtWeightFileScannerSet<> scanners(inputFiles);
  
  // Create merged output for all compatible files
    
  outputFile.open();

  outputFile.put(JMeta(argc, argv));
  
  JHead commonHeader = scanners.begin()->getHeader();

  MultiHead eventHeaders;

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

    // Add header legacy information
    
    JHead header = scanner->getHeader();
    header.createUUID(); // Ensure UUID is set

    Head head;
    copy(header, head);

    eventHeaders.push_back(head);

    commonHeader = commonHeader.getMatch(header);

    // Add event-weight normalisations
    
    while (scanner->hasNext()) {
      
      const Evt* event = scanner->next();
      
      if (event != NULL) {
        
        STATUS("Event:" << RIGHT(10) << scanner->getCounter() << '\r'); DEBUG(endl);

        Evt out = *event;

        uuid_copy(out.header_uuid, header.UUID.uuid); 

        if (out.w.size() <= WEIGHTLIST_NORMALISATION) {
          out.w.resize(WEIGHTLIST_NORMALISATION + 1, 0.0);
        }

        out.w[WEIGHTLIST_NORMALISATION] = scanner->getNormalisation(*event);
        
        outputFile.put(out);
        
      } else {

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

  Head newHead;
  copy(commonHeader, newHead);
  
  outputFile.put(newHead);

  for (JMultipleFileScanner<MultiHead> in(inputFiles); in.hasNext(); ) {

    const MultiHead& multiHead = *in.next();

    eventHeaders.join(multiHead);
  }

  outputFile.put(eventHeaders);

  JMultipleFileScanner<JMetaTypes_t> io(inputFiles);

  io >> outputFile;
  
  outputFile.close();
  
  return 0;
}