ZMQCollector.h

/**************************************************************************
 * 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.                  *
 **************************************************************************/
#pragma once
 
#include <daq/hbasf2/apps/ZMQApp.h>
#include <framework/pcore/zmq/connections/ZMQConfirmedConnection.h>
#include <framework/pcore/zmq/connections/ZMQLoadBalancedConnection.h>
#include <framework/pcore/zmq/connections/ZMQRawConnection.h>
#include <daq/hbasf2/connections/ZMQROIConnection.h>
 
#include <boost/program_options.hpp>
 
namespace po = boost::program_options;
 
namespace Belle2 {
  class ZMQCollector : public ZMQStandardApp<ZMQConfirmedInput, ZMQLoadBalancedOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
    void handleTimeout() final;
 
  private:
    std::string m_inputAddress;
    std::string m_outputAddress;
    bool m_lax = false;
    unsigned int m_stopWaitingTime = 2;
  };
 
  class ZMQOutputAdapter : public ZMQStandardApp<ZMQLoadBalancedInput, ZMQRawOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
 
  private:
    std::string m_inputAddress;
    std::string m_outputAddress;
  };
 
  class ZMQProxyCollector : public ZMQStandardApp<ZMQConfirmedInput, ZMQConfirmedOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
    void handleTimeout() final;
 
  private:
    std::string m_inputAddress;
    std::string m_outputAddress;
    unsigned int m_stopWaitingTime = 2;
  };
 
  class ZMQFinalCollector : public ZMQStandardApp<ZMQConfirmedInput, ZMQRawOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
    void handleTimeout() final;
 
  private:
    std::string m_inputAddress;
    std::string m_outputAddress;
    bool m_addEventSize;
    unsigned int m_stopWaitingTime = 2;
  };
 
  class ZMQFinalCollectorWithROI : public ZMQStandardApp<ZMQConfirmedInput, ZMQDataAndROIOutput> {
  protected:
    void initialize() final;
    void addOptions(po::options_description& desc) final;
    void handleExternalSignal(EMessageTypes type) final;
    void handleInput() final;
    void fillMonitoringJSON(std::stringstream& buffer) const final;
    void handleTimeout() final;
 
  private:
    std::string m_inputAddress;
    std::string m_dataOutputAddress;
    std::string m_roiOutputAddress;
    bool m_addEventSize;
    unsigned int m_stopWaitingTime = 2;
  };
}