io_online.hh

Go to the documentation of this file.

#ifndef IOONLINEINCLUDED
#define IOONLINEINCLUDED

#include "km3net-dataformat/offline/Evt.hh"
#include "km3net-dataformat/offline/Hit.hh"
#include "km3net-dataformat/offline/Trk.hh"
#include "km3net-dataformat/offline/Exception.hh"

#include "km3net-dataformat/online/JDAQEvent.hh"
#include "km3net-dataformat/online/JDAQKeyHit.hh"
#include "km3net-dataformat/online/JDAQTimeslice.hh"
#include "km3net-dataformat/online/JDAQSummaryslice.hh"

#include "TStreamerInfo.h"
#include "TFile.h"
#include "TTree.h"
#include "km3net-dataformat/definitions/root.hh"

#include <map>
#include <vector>


using namespace KM3NETDAQ;

/**
 * Read a hit from a DAQ hit.
 *
 * \param  hit           hit
 * \param  daqhit        DAQ hit
 */
inline void read(Hit& hit, const JDAQHit& daqhit )
{
  hit.channel_id = daqhit.getPMT();
  hit.tot        = daqhit.getToT();
  hit.tdc        = daqhit.getT(); // GetT() just return the bare TDC
}

/**
 * Read a hit from a DAQ key hit.
 *
 * \param  hit           hit
 * \param  daqhit        DAQ key hit
 */
inline void read(Hit& hit, const JDAQKeyHit& daqhit )
{
  hit.id = hit.pmt_id = 0;

  hit.dom_id     = daqhit.getModuleID();
  hit.channel_id = daqhit.getPMT();
  hit.tot        = daqhit.getToT();
  hit.tdc        = daqhit.getT(); // GetT() just return the bare TDC
}

/**
 * Read an event from a DAQ event.
 *
 * \param  evt           evt
 * \param  de            DAQ event
 */
inline void read(Evt& evt, const JDAQEvent& de)
{
  evt.run_id          = de.getRunNumber();
  evt.det_id          = de.getDetectorID();
  evt.frame_index     = de.getFrameIndex();
  evt.trigger_counter = de.getCounter();
  evt.overlays        = de.getOverlays();
  evt.trigger_mask    = de.getTriggerMask();
  evt.t.SetSec(     de.getTimesliceStart().getUTCseconds() );
  evt.t.SetNanoSec( de.getTimesliceStart().getUTC16nanosecondcycles() * 16 );

  // The only way to the hits that are also in the triggeredhits collection
  // is by dom and channel id and time.

  Hit h;
  std::map<int, std::map< int, std::map < unsigned int, Hit*> > > M;

  const std::vector<JDAQSnapshotHit>&   snapshotHits = de.getHits<JDAQSnapshotHit>();
  const std::vector<JDAQTriggeredHit>& triggeredHits = de.getHits<JDAQTriggeredHit>();

  // http://stackoverflow.com/questions/10735135/reallocation-in-stdvector-after-stdvector-reserve
  evt.hits.clear();
  evt.hits.reserve(snapshotHits.size());

  for (auto& daqhit : snapshotHits ) // JDAQSnapshotHit
  {
    read( h, daqhit );
    h.trig = 0;
    evt.hits.push_back( h );
    M[ h.dom_id ][ h.channel_id ][ h.tdc ] = &(*evt.hits.rbegin());
  }

  for (auto& daqtrighit : triggeredHits)
  {
    Hit* g = M[daqtrighit.getModuleID()][daqtrighit.getPMT()][daqtrighit.getT()];

    if (g)
      g->trig = daqtrighit.getTriggerMask ();
    else
      THROW(Exception, "Failed to flag snaphot hit " << (int) daqtrighit.getModuleID() << "." << (int) daqtrighit.getPMT() << " " << daqtrighit.getT());
  }
}

/**
 * Read an event from a DAQ time slice.
 *
 * \param  evt           evt
 * \param  ts            DAQ time slice
 */

inline void read( Evt& evt, const JDAQTimeslice& ts )
{
  evt.run_id          = ts.getRunNumber();
  evt.det_id          = ts.getDetectorID();
  evt.frame_index     = ts.getFrameIndex();
  evt.t.SetSec( ts.getTimesliceStart().getUTCseconds() );
  evt.t.SetNanoSec( ts.getTimesliceStart().getUTC16nanosecondcycles() * 16 );
  evt.hits.clear();

  // a timeslice is a vector of JDAQSuperFrame's, which is a JDAQFrame, which
  // is an stl::vector-like object (supporting stl-like iteration.)

  Hit h; h.id = 0; h.pmt_id = 0;

  for (auto& sf : ts )
  {
    for (auto& daqhit : sf )
    {
      read( h, daqhit);
      h.dom_id = sf.getModuleID();
      evt.hits.push_back(h);
    }
  }
}


/**
 * Get summary slice from given file with given frame index.
 *
 * \param  f             pointer to ROOT file
 * \param  frame_index   frame index
 * \return               pointer to summary slice
 */
inline JDAQSummaryslice* get_summary_slice( TFile* f , int frame_index )
{
  static TFile* _f = 0;
  static TTree*  S = 0;
  static TBranch* BS = 0;
  JDAQSummaryslice *r = 0 ;

  if (_f != f) // setup for tree reading and build tree index
  {
    _f = f;

    // first we have to deal with the following....
    // The streamer of JDAQSummaryFrame needs to know what to do since
    // this information is not written to root file.

    const char* name = JDAQSummaryslice::Class()->GetName();

    JDAQSummaryFrame::ROOT_IO_VERSION = ((TStreamerInfo*)_f -> GetStreamerInfoList()->FindObject(name))->GetClassVersion();

    //print ("JDAQSummaryFrame::ROOT_IO_VERSION set to", JDAQSummaryFrame::ROOT_IO_VERSION );

    S = (TTree*) _f->Get( TTREE_ONLINE_TIMESLICE );

    if (!S)
    {
      THROW(Exception, "Failed to get timeslice TTree : " <<  TBRANCH_ONLINE_TIMESLICE );
    }
    BS = S->GetBranch( TBRANCH_ONLINE_TIMESLICE );
    if (!BS)
    {
      THROW(Exception, "Failed to get brach :" << TBRANCH_ONLINE_TIMESLICE );
    }

    std::cout << "building index to lookup timeslices..." << std::endl;
    int n = S->BuildIndex("frame_index");
    (void) n;
    BS->SetAddress( &r );
  }

  int nbytes = S->GetEntryWithIndex( frame_index );
  if ( nbytes == 0 )
  {
    THROW(Exception, "Failed to find entry with frame_index " << frame_index);
  }
  return r;

}

#endif