DqmHistoManagerModule.cc

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#include <daq/dqm/modules/DqmHistoManagerModule.h>
 
#include <framework/core/Environment.h>
#include <framework/pcore/ProcHandler.h>
#include <framework/pcore/RbTuple.h>
 
#include "TKey.h"
#include "TText.h"
 
using namespace Belle2;
using namespace std;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(DqmHistoManager)
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
// Implementations
DqmHistoManagerModule::DqmHistoManagerModule() : Module(), m_initmain(false), m_initialized(false)
{
  // Module description
  setDescription("Module to manage histograms/Ntuples/TTrees");
  setPropertyFlags(Module::c_HistogramManager);
 
  // Parameters
  addParam("histoFileName", m_histfile, "Name of histogram output file.", string("histofile.root"));
  addParam("workDirName", m_workdir, "Name of working directory", string("."));
  addParam("HostName", m_hostname, "Name of host to send histograms", string("localhost"));
  addParam("Port", m_port, "Socket port number to connect", DQM_SOCKET);
  addParam("DumpInterval", m_interval, "Interval to dump histos", 1000);
  addParam("WriteInterval", m_dumpinterval, "Interval to write file", 10000);
}
 
DqmHistoManagerModule::~DqmHistoManagerModule()
{
  //  if (m_initmain) {
  //  if (ProcHandler::EvtProcID() == -1) {   // should be called from main proc.
  //    cout << "DqmHistoManager:: destructor called from pid=" << ProcHandler::EvtProcID() << endl;
  //    if (Environment::Instance().getNumberProcesses() > 0 && ProcHandler::EvtProcID() == -1) {
  /*
  if (Environment::Instance().getNumberProcesses() > 0) {
    cout << "DqmHistoManager:: adding histogram files" << endl;
    RbTupleManager::Instance().hadd();
    cout << "DqmHistoManager:: adding histogram files done" << endl;
  }
  */
  //  }
}
 
void DqmHistoManagerModule::initialize()
{
  RbTupleManager::Instance().init(Environment::Instance().getNumberProcesses(), m_histfile.c_str(), m_workdir.c_str());
 
  m_initmain = true;
  //  cout << "DqmHistoManager::initialization done" << endl;
 
  // Connect to Histogram Server
  //  m_sock = new EvtSocketSend(m_hostname, m_port);
  //  printf("EvtSocketSend : fd = %d\n", (m_sock->sock())->sock());
 
  // Message Handler
  m_msg = new MsgHandler(0);    // Compression level = 0
 
  // Clear event counter
  m_nevent = 0;
}
 
void DqmHistoManagerModule::beginRun()
{
  if (!m_initialized) {
    //    cout << "DqmHistoManager:: first pass in beginRun() : proc="
    //   << ProcHandler::EvtProcID() << endl;
    RbTupleManager::Instance().begin(ProcHandler::EvtProcID());
    // Connect to Histogram Server
    m_sock = new EvtSocketSend(m_hostname, m_port);
    printf("EvtSocketSend (Proc %d) : fd = %d\n", ProcHandler::EvtProcID(),
           (m_sock->sock())->sock());
 
    m_pstep = (ProcHandler::numEventProcesses() != 0) ? m_interval / ProcHandler::numEventProcesses() : m_interval;
    m_dstep = (ProcHandler::numEventProcesses() != 0) ? m_dumpinterval / ProcHandler::numEventProcesses() : m_dumpinterval;
 
    m_ptime = time(NULL);
    m_dtime = m_ptime;
    if (ProcHandler:: EvtProcID() < 100) {
      m_ptime -= m_pstep * ProcHandler::EvtProcID();
      m_dtime += m_dstep * ProcHandler::EvtProcID();
    }
    m_initialized = true;
  }
}
 
void DqmHistoManagerModule::endRun()
{
  if (!m_initialized) {
    //    cout << "DqmHistoManager:: first pass in endRun(): proc="
    //   << ProcHandler::EvtProcID() << endl;
    RbTupleManager::Instance().begin(ProcHandler::EvtProcID());
    m_initialized = true;
  }
}
 
void DqmHistoManagerModule::event()
{
  if (!m_initialized) {
    //    cout << "DqmHistoManager:: first pass in event() : proc="
    //   << ProcHandler::EvtProcID() << endl;
    RbTupleManager::Instance().begin(ProcHandler::EvtProcID());
    m_initialized = true;
  }
 
  time_t ctime = time(NULL);
 
  // Transfer hitograms over network
  if ((ctime - m_ptime) > m_interval) {
    //    printf ( "DqmHistoManager: event = %d\n", m_nevent );
    //    printf ( "DqmHistoManger: dumping histos.....\n" );
    m_msg->clear();
    m_nobjs = 0;
    // Stream histograms with directory structure
    StreamHistograms(gDirectory, m_msg);
 
    EvtMessage* msg = m_msg->encode_msg(MSG_EVENT);
    //    printf ( "Message Size = %d\n", msg->size() );
 
    printf("DqmHistoManger(proc:%d): dumping histos.....%d histos\n",
           ProcHandler::EvtProcID(), m_nobjs);
    fflush(stdout);
 
    (msg->header())->reserved[0] = 0;
    (msg->header())->reserved[1] = m_nobjs;
    (msg->header())->reserved[2] = 0;
    if (m_nobjs > 0) {
      m_sock->send(msg);
    }
    delete(msg);
    m_ptime = ctime;
  }
  // Dump histograms to file
  if ((ctime - m_dtime) > m_dumpinterval) {
    // Dump histograms to file  -> Turned Off
    //    RbTupleManager::Instance().dump();
    m_dtime = ctime;
  }
 
  m_nevent++;
 
}
 
void DqmHistoManagerModule::terminate()
{
  //  if ( ProcHandler::EvtProcID() >= 10000 ) {
  //    return;
  //  }
  if (m_initialized) {
    //    cout << "DqmHistoManager::terminating event process : PID=" << ProcHandler::EvtProcID() << endl;
    m_msg->clear();
 
    m_nobjs = 0;
    StreamHistograms(gDirectory, m_msg);
 
    printf("terminate : m_nobjs = %d\n", m_nobjs);
    EvtMessage* msg = m_msg->encode_msg(MSG_EVENT);
    (msg->header())->reserved[0] = 0;
    (msg->header())->reserved[1] = m_nobjs;
    (msg->header())->reserved[2] = 0;
 
    //m_sock->send(msg);
 
    delete(msg);
 
    // Dump hitograms to file  -> Turned Off (
    //    RbTupleManager::Instance().dump();
 
    //    RbTupleManager::Instance().terminate(); <- not used
    //    delete m_sock;
    //    delete m_msg;
  }
}
 
int DqmHistoManagerModule::StreamHistograms(TDirectory* curdir, MsgHandler* msg)
{
  TList* keylist = curdir->GetList();
  //    keylist->ls();
 
  TIter nextkey(keylist);
  TKey* key = 0;
  int nkeys = 0;
  int nobjs = 0;
  while ((key = (TKey*)nextkey())) {
    nkeys++;
    TObject* obj = curdir->FindObject(key->GetName());
    if (obj->IsA()->InheritsFrom("TH1")) {
      TH1* h1 = (TH1*) obj;
      //      printf ( "Key = %s, entry = %f\n", key->GetName(), h1->GetEntries() );
      //      if (h1->GetEntries() > 0) {    // Do not send empty histograms
      m_msg->add(h1, h1->GetName());
      nobjs++;
      m_nobjs++;
      //      }
    } else if (obj->IsA()->InheritsFrom(TDirectory::Class())) {
      //      printf ( "New directory found  %s, Go into subdir\n", obj->GetName() );
      TDirectory* tdir = (TDirectory*) obj;
      //      m_msg->add(tdir, tdir->GetName());
      TText subdir(0, 0, tdir->GetName());
      m_msg->add(&subdir, "SUBDIR:" + string(obj->GetName())) ;
      nobjs++;
      m_nobjs++;
      tdir->cd();
      StreamHistograms(tdir , msg);
      TText command(0, 0, "COMMAND:EXIT");
      m_msg->add(&command, "SUBDIR:EXIT");
      nobjs++;
      m_nobjs++;
      curdir->cd();
    }
  }
  return 0;
}