io_online.hh

Go to the documentation of this file.

#ifndef IOONLINEINCLUDED
#define IOONLINEINCLUDED

#include "Evt.hh"
#include "Hit.hh"
#include "Trk.hh"
#include "Exception.hh"

#include "../online/JDAQEvent.hh"
#include "../online/JDAQKeyHit.hh"
#include "../online/JDAQTimeslice.hh"
#include "../online/JDAQSummaryslice.hh"

#include "TStreamerInfo.h"
#include "TFile.h"
#include "TTree.h"

#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.t          = daqhit.getT();
}

/**
 * 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.t          = daqhit.getT();
}

/**
 * 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 < double, 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 = false; 
    evt.hits.push_back( h );
    M[ h.dom_id ][ h.channel_id ][ h.t ] = &(*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.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.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);
   }
 }
}

//inline void read(Det& det, const JDAQSummaryslice& ss )
//{
//  for(auto& sf: ss )  // JDAQSummarysclice derives from vector<JDAQSummaryFrame>
//    {
//      Dom& dom = det.doms[ sf.getModuleID() ];
//      for(auto& pmt: dom.pmts ) pmt.rate = sf.getRate( pmt.channel_id );
//    }
//}

/**
 * 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.

    JDAQSummaryFrame::ROOT_IO_VERSION = ((TStreamerInfo*)_f -> GetStreamerInfoList()->FindObject("KM3NETDAQ::JDAQSummaryslice"))->GetClassVersion(); 

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

    S = (TTree*) _f->Get("KM3NET_SUMMARYSLICE");

    if (!S) 
    {
      THROW(Exception, "Failed to get KM3NET_SUMMARYSLICE TTree.");
    }
    BS = S->GetBranch("KM3NET_SUMMARYSLICE");
    if (!BS) 
    {
     THROW(Exception, "Failed to get KM3NET_SUMMARYSLICE TBranch.");
    }

    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