JSortEvt.cc

Go to the documentation of this file.

#include <unistd.h>
#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 "JDAQ/JDAQEventIO.hh"
#include "JDAQ/JDAQTimesliceIO.hh"
#include "JDAQ/JDAQSummarysliceIO.hh"
 
#include "JSupport/JMeta.hh"
#include "JSupport/JSupport.hh"
#include "JSupport/JFileRecorder.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JEvtWeightFileScannerSet.hh"
 
#include "JTrigger/JTriggerParameters.hh"
 
#include "JReconstruction/JEvt.hh"
 
#include "JAAnet/JHead.hh"
 
#include "TTree.h"
 
 
/**
 * \file
 * Example program for sorting and 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 JTYPELIST<JRemove<JAllTypes_t, JDAQTimesliceTypes_t>::typelist, JFIT::JEvt>::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);
  }
 
 
  TTree::SetMaxTreeSize(1000000000000LL); // Set maximum tree size to 1TB
 
  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
 
  const size_t Ndigits = to_string(scanners.size()).size();
 
  for (JEvtWeightFileScannerSet<>::iterator scanner = scanners.begin(); scanner != scanners.end(); ++scanner) {
 
    const size_t Nfile = distance(scanners.begin(), scanner);
 
    const string filename = string(outputFile).replace(pos, 1, MAKE_STRING(setfill('0') << setw(Ndigits) << Nfile));
 
    NOTICE("Writing " << filename << "..." << endl);
 
    if (scanner->getFilelist().size()                                      == 1 &&
        symlink(scanner->getFilelist().begin()->c_str(), filename.c_str()) == 0) {
      continue;
    }
    
    JFileRecorder<typelist> out(filename.c_str());
    
    out.open();
 
    const Head& head = getHeader(scanner->getFilelist());
    
    out.put(head);
    
    out.put(JMeta(argc, argv));
 
    for (JMultipleFileScanner<Evt> in(scanner->getFilelist()); in.hasNext(); ) {
      
      const Evt* event = in.next();
 
      if (event != NULL) {
      
        STATUS("Event:" << RIGHT(10) << in.getCounter() << '\r'); DEBUG(endl);
 
        out.put(*event);
        
      } else {
 
        WARNING("Event " << in.getCounter() << " is empty; skip.");
      }
    }
 
    JMultipleFileScanner<JRemove<typelist, JAAnetTypes_t>::typelist> io(scanner->getFilelist());
 
    io >> out;
 
    out.close();
  }
  
  return 0;
}