ZMQROIConnection.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 <framework/pcore/zmq/connections/ZMQRawConnection.h>
#include <framework/pcore/zmq/utils/ZMQParent.h>
#include <framework/pcore/zmq/messages/ZMQNoIdMessage.h>
#include <daq/roisend/h2m.h>
 
#include <zmq.hpp>
 
namespace Belle2 {
  class ZMQROIOutput : public ZMQConnection {
  public:
    ZMQROIOutput(const std::string& outputAddress, const std::shared_ptr<ZMQParent>& parent);
 
    void handleEvent(zmq::message_t message);
    std::string getMonitoringJSON() const final;
    void handleIncomingData();
    bool isReady() const final;
    std::vector<zmq::socket_t*> getSockets() const final;
    std::string getEndPoint() const { return m_rawOutput.getEndPoint(); }
  private:
    static constexpr const unsigned int HEADER_SIZE = sizeof(struct h2m_header_t);
    static constexpr const unsigned int TRAILER_SIZE = sizeof(struct h2m_footer_t);
 
    ZMQRawOutput m_rawOutput;
 
    void addHeader(char* buffer, unsigned int size);
    void addTrailer(char* buffer);
  };
 
  class ZMQDataAndROIOutput : public ZMQConnection {
  public:
    ZMQDataAndROIOutput(const std::string& dataAddress, const std::string& roiAddress, bool addEventSize,
                        const std::shared_ptr<ZMQParent>& parent);
 
    std::string getMonitoringJSON() const final;
    std::string getROIMonitoringJSON() const;
 
    void handleEvent(std::unique_ptr<ZMQNoIdMessage> message);
    void handleIncomingData();
    bool isReady() const final;
    std::vector<zmq::socket_t*> getSockets() const final;
    std::string getEndPoint() const { return m_dataOutput.getEndPoint() + ";" + m_roiOutput.getEndPoint(); }
  private:
    ZMQRawOutput m_dataOutput;
    ZMQRawOutput m_roiOutput;
  };
}