JRunAnalyzer.cc

Go to the documentation of this file.

// c++ standard library
#include <iostream>
#include <iomanip>

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

#include "JAAnet/JAAnetDictionary.hh"

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

#include "JTools/JRange.hh"
#include "JLang/JToken.hh"
#include "JGizmo/JGizmoToolkit.hh"

#include "JSupport/JSupport.hh"
#include "JSupport/JMeta.hh"
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JTreeScanner.hh"
#include "JSupport/JMonteCarloFileSupportkit.hh"
#include "JSupport/JFileRecorder.hh"

#include "JReconstruction/JEvt.hh"
#include "JReconstruction/JEvtToolkit.hh"

#include "Jeep/JParser.hh"

#include "km3net-dataformat/online/JDAQ.hh"
#include "km3net-dataformat/online/JDAQHeader.hh"

#include "JDAQ/JDAQEventIO.hh"
#include "JDAQ/JDAQTimesliceIO.hh"
#include "JDAQ/JDAQSummarysliceIO.hh"

#include "JLang/JObjectReader.hh"

#include "TROOT.h"
#include "JROOT/JRootFileReader.hh"
#include "JROOT/JROOTClassSelector.hh"
#include "JROOT/JRootToolkit.hh"
#include "JROOT/JRootFileWriter.hh"
#include "JROOT/JTreeParameters.hh"

#include "JSupport/JSupport.hh"
#include "JSupport/JSupportToolkit.hh"

#include "JRunAnalyzer/JRunInfo.hh"

//JRunAnalyzer
#include "JRunAnalyzer.hh"
#include "JRunHistograms.hh"

#include <TSystem.h>
#include "TTree.h"

/**                                                                                       
 * \author rgruiz, adomi                                                                   
 */

int main(int argc, char **argv) {

  using namespace KM3NETDAQ;
  using namespace JPP;

  typedef JToken<';'>      JToken_t;
  
  JSingleFileScanner<>     inputFile;
  JFileRecorder<JTYPELIST<run_info, JMeta, TH1D, TH2D>::typelist>  outputFile;
  string                   detectorFile;
  JLimit_t                 nTimeslices;
  JLimit_t                 nSummaryslices;
  JLimit_t                 nEvents;
  vector<JToken_t>         parameters;

  try {

    JParser<> zap;
    zap['f'] = make_field(inputFile      , "input file "    );
    zap['o'] = make_field(outputFile     , "output file"    ) = "out.root"; 
    zap['a'] = make_field(detectorFile   , "detector file"  );  
    zap['t'] = make_field(nTimeslices    , "number of timeslices"   ) = JLimit::max();  
    zap['s'] = make_field(nSummaryslices , "number of summaryslices") = JLimit::max(); 
    zap['n'] = make_field(nEvents        , "number of events"       ) = JLimit::max();
    zap['@'] = make_field(parameters, "parameter values, e.g: \"p0 = 1.0;\"");
    
    zap(argc,argv);
  }
  catch(const exception &error) {
    ERROR(error.what() << endl);
  }

  JDetector detector;

  try {
    load (detectorFile, detector);
  }
  catch(const JException & error) {
    FATAL( "FATAL ERROR. Could not open detector file '" << detectorFile << "'." << endl);
  }
  
  int analysis_level = 0;
  
  for (vector<JToken_t>::const_iterator i = parameters.begin(); i != parameters.end(); ++i) {

    if(getParameter(*i) == 0 && getValue(*i) == 1) analysis_level = 1;
    if(getParameter(*i) == 0 && getValue(*i) == 2) analysis_level = 2;

  }
 
  if(analysis_level == 1){
    cout << "ANALYSIS OF ALL AVAILABLE PMTs INFORMATION. It may take a while!" << endl;
  } else if (analysis_level == 2){
    cout << "BASIC ANALYSIS AT TRIGGER AND RECONSTRUCTION LEVEL." << endl;
  } else {
    cout << "JRUNANALYZER RUNNING WITH A REDUCED NUMBER OF PLOTS." << endl;
  }
  
  JSingleFileScanner<JHeaderTypes_t> in(inputFile);
  
  Head head;

  try{
    head = getHeader(in); 
  }catch(const exception&){}
  
  JHead buffer(head);

  buffer.DAQ.livetime_s = getLivetime(inputFile);

  buffer.push(& JAANET::JHead::DAQ);

  copy(buffer, head);

  cout << "START\n";
  
  outputFile.open();

  JRootFile* p = dynamic_cast<JRootFile*>(outputFile.get());

  if (p == NULL) {
    FATAL("WRONG FILE TYPE");
  }
  
  TFile* out = p->getFile();

  cout << out->GetName() << endl;

  outputFile.put(JMeta(argc, argv));

  JDAQHeader header;
    
  for(JTreeScanner<JDAQEvent> in(inputFile); in.hasNext();){

    const JDAQEvent* event = in.next();
    
    if (header.getDetectorID() != event->getDetectorID() ||
        header.getRunNumber () != event->getRunNumber ()) {
      
      header = event->getDAQHeader();
      
      putObject(out, header);

      break;
    }
  }

  const run_info runinfo(header.getDetectorID(), 
                         header.getRunNumber(), 
                         buffer.DAQ.livetime_s, 
                         buffer.livetime.numberOfSeconds);
    
  outputFile.put(runinfo);
    
  JRunAnalyzer Analyzer (inputFile, detector, out, nTimeslices, nSummaryslices, nEvents, analysis_level);

  cout << "read events\n";
  Analyzer.readEvents                        ();
  cout << "read summarydata\n";
  Analyzer.readSummaryData                   ();
  cout << "read timeslices\n";
  Analyzer.readTimesliceData<JDAQTimeslice>  ();
  Analyzer.readTimesliceData<JDAQTimesliceL0>();
  Analyzer.readTimesliceData<JDAQTimesliceL1>();
  Analyzer.readTimesliceData<JDAQTimesliceL2>();
  Analyzer.readTimesliceData<JDAQTimesliceSN>();
  
  if(analysis_level == 2){
    cout << "read reconstructed events\n";
    Analyzer.readRecoEvents                  ();
  }
  
  Analyzer.writeToFile(out, analysis_level);

  outputFile.close();
}