ZMQCollector.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun                                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#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;
  };
}