JEvtSupportkit.hh

Go to the documentation of this file.

#ifndef __JRECONSTRUCTION__JEVTSUPPORTKIT__
#define __JRECONSTRUCTION__JEVTSUPPORTKIT__
 
#include <set>
 
#include "km3net-dataformat/offline/Evt.hh"
#include "km3net-dataformat/offline/Trk.hh"
#include "km3net-dataformat/definitions/reconstruction.hh"
 
#include "JLang/JUUID.hh"
 
#include "JReconstruction/JEvt.hh"
#include "JReconstruction/JFitStatus.hh"
#include "JReconstruction/JHistory.hh"
 
 
/**
 * \author mdejong
 */
namespace JRECONSTRUCTION {}
namespace JPP { using namespace JRECONSTRUCTION; }
 
/**
 * Support.
 */
namespace JRECONSTRUCTION {
 
  /**
   * Copy tracks.
   *
   * \param  __begin       begin of input tracks
   * \param  __end         end   of input tracks
   * \param  out           output tracks
   */
  inline void copy(const JFIT::JEvt::const_iterator __begin,
                   const JFIT::JEvt::const_iterator __end,
                   Evt& out)
  {
    using namespace std;
    using namespace JPP;
 
    struct : public set<JUUID> {
 
      inline int get_index(const JUUID& element) const
      {
        const_iterator i = this->find(element);
 
        if (i != this->end())
          return std::distance(this->begin(), i);
        else
          return -1;
      }
    } uuid;
 
    for (JFIT::JEvt::const_iterator fit = __begin; fit != __end; ++fit) {
      uuid.insert(fit->getUUID());
    }
 
    for (JFIT::JEvt::const_iterator fit = __begin; fit != __end; ++fit) {     // import reconstruction data
 
      Trk trk;
 
      trk.id        = uuid.get_index(fit->getUUID());
      trk.pos       = Vec(fit->getX(),  fit->getY(),  fit->getZ());
      trk.dir       = Vec(fit->getDX(), fit->getDY(), fit->getDZ());
      trk.t         = fit->getT();
      trk.E         = fit->getE();
      trk.lik       = fit->getQ();
      trk.rec_type  = JPP_RECONSTRUCTION_TYPE;
      trk.status    = (fit->getStatus() == COMPLETE_CHAIN ? TRK_ST_FINALSTATE : TRK_ST_UNDEFINED);
 
      if (fit->hasParentUUID()) {
        trk.mother_id = uuid.get_index(fit->getParentUUID());
      }
 
      for (JHistory::const_iterator i = fit->getHistory().begin(); i != fit->getHistory().end(); ++i) {
        trk.rec_stages.push_back(i->type);
      }
 
      for (int i = 0; i != fit->getN(); ++i) {
        trk.fitinf.push_back(fit->getW(i));
      }
 
      trk.error_matrix = fit->getV();
 
      out.trks.push_back(trk);
    }
  }
}
 
#endif