JMuonGandalf.cc

Go to the documentation of this file.

#include <string>

#include "evt/Head.hh"
#include "evt/Evt.hh"

#include "JDAQ/JDAQEvent.hh"
#include "JDAQ/JDAQTimeslice.hh"

#include "JDetector/JModuleRouter.hh"
#include "JDetector/JDetectorToolkit.hh"

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

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

#include "JFit/JEvt.hh"
#include "JFit/JMuonGandalf.hh"
#include "JFit/JMuonParameters.hh"
#include "JFit/JMuonGandalfParameters_t.hh"

#include "JLang/JSharedPointer.hh"


/**
 * \file
 *
 * Program to perform JFIT::JRegressor<JLine3Z,JGandalf> fit with I/O of JFIT::JEvt data.
 * \author gmaggi
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  using namespace KM3NETDAQ;

  typedef JTYPELIST<JAllTypes_t, JEvt>::typelist              typelist;
  typedef JParallelFileScanner< JTypeList<JDAQEvent, JEvt> >  JParallelFileScanner_t;

  JParallelFileScanner_t  inputFile;
  JFileRecorder<typelist> outputFile;
  JLimit_t&       numberOfEvents = inputFile.getLimit();
  string          detectorFile;
  string          pdfFile;
  JMuonParameters muonParams;
  int             debug;


  try { 

    JParser<> zap("Program to perform PDF fit of muon trajectory to data.");
    
    zap['f'] = make_field(inputFile);
    zap['o'] = make_field(outputFile)          = "gandalf.root";
    zap['a'] = make_field(detectorFile);
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['P'] = make_field(pdfFile);
    zap['@'] = make_field(muonParams)          = JPARSER::initialised();
    zap['d'] = make_field(debug)               = 1;
    
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }

  cout.tie(&cerr);

  JDetector detector;

  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }

  const JSharedPointer<const JModuleRouter> moduleRouter(new JModuleRouter(detector));
  
  outputFile.open();
  outputFile.put(JMeta(argc, argv));

  const JMuonGandalf muonGandalf(moduleRouter, muonParams, pdfFile);

  while (inputFile.hasNext()) {

    STATUS("event: " << setw(10) << inputFile.getCounter() << '\r'); DEBUG(endl);

    JParallelFileScanner_t::multi_pointer_type ps = inputFile.next();

    const JDAQEvent* tev  = ps;
    JEvt*            in   = ps;

    const JDAQTimeslice timeSlice(*tev, true);    

    JEvt out;

    muonGandalf.getJEvt(timeSlice, *in, out);

    outputFile.put(out);
  }
  STATUS(endl);

  JSingleFileScanner<JRemove<typelist, JEvt>::typelist> io(inputFile);

  io >> outputFile;

  outputFile.close();
}