JPrintAAnet.cc

Go to the documentation of this file.

 
#include <string>
#include <iostream>
#include <iomanip>
#include <vector>
#include <cmath>
 
#include "km3net-dataformat/offline/Head.hh"
#include "km3net-dataformat/offline/Evt.hh"
 
#include "JAAnet/JAAnetToolkit.hh"
 
#include "JReconstruction/JEvtToolkit.hh"
 
#include "JSupport/JMultipleFileScanner.hh"
#include "JSupport/JSupport.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
namespace {
 
  /**
   * Print track.
   *
   * \param  out      output stream
   * \param  prefix   prefix
   * \param  trk      track
   */
  inline void print(std::ostream& out, const std::string& prefix, const Trk& trk)
  {
    using namespace std;
    
    out << prefix                      << ' '
        << setw(4)     << trk.id       << ' '
        << FIXED(7,2)  << trk.pos.x    << ' '
        << FIXED(7,2)  << trk.pos.y    << ' '
        << FIXED(7,2)  << trk.pos.z    << ' '
        << FIXED(7,4)  << trk.dir.x    << ' '
        << FIXED(7,4)  << trk.dir.y    << ' '
        << FIXED(7,4)  << trk.dir.z    << ' '
        << FIXED(11,1) << trk.t        << ' '
        << setw(3)     << trk.rec_type << ' '
        << setw(3)     << trk.status   << ':';
 
    for (const auto i : trk.rec_stages) {
      out << ' ' << i;
    }
  }
 
 
  /**
   * Recursively print track with given mother identifier.
   *
   * \param  out      output stream
   * \param  prefix   prefix
   * \param  evt      event
   * \param  id       identifier
   */
  inline void print(std::ostream& out, const std::string& prefix, const Evt& evt, const int id)
  {
    using namespace std;
    
    if (id != -1) {
      for (const Trk& i : evt.trks) {
        if (i.id == id) {
          out << endl;
          print(out, prefix, i);
          print(out, "  " + prefix, evt, i.mother_id);
          break;
        }
      }
    }
  }
}
 
/**
 * \file
 *
 * Example program to print track fit results from Evt formatted data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
 
  JMultipleFileScanner<Evt> inputFile;
  JLimit_t&      numberOfEvents = inputFile.getLimit();
  vector<string> keys;
  int            debug;
 
  try {
 
    JParser<> zap("Example program to print track fit results from Evt formatted data.");
 
    zap['f'] = make_field(inputFile);
    zap['n'] = make_field(numberOfEvents)      = JLimit::max();
    zap['k'] = make_field(keys,   "print optional weights: " << get_keys(getWeight))         = JPARSER::initialised();
    zap['d'] = make_field(debug)               = 2;
 
    zap(argc, argv);
  }
  catch(const exception& error) {
    FATAL(error.what() << endl);
  }
 
    
  while (inputFile.hasNext()) {
 
    cout << "event " << setw(10) << inputFile.getCounter() << endl;
    
    const Evt* evt = inputFile.next();
 
    if (has_reconstructed_jppmuon(*evt)) {
 
      const Trk trk = get_best_reconstructed_jppmuon(*evt);
 
      print(cout, "", trk);
      print(cout, "->", *evt, trk.mother_id);
 
      for (const auto& key : keys) {
        cout << ' ' << SCIENTIFIC(12,3) << getWeight(trk, key, 0.0);
      }
 
      cout << endl;
    }
 
    if (debug >= debug_t) {
      for (const Trk& i : evt->trks) {
        print(cout, "", i);
        cout << endl;
      }
    }
  }
}