ZMQHistogramConnection.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/ZMQConnection.h>
#include <daq/hbasf2/utils/HistogramMapping.h>
 
#include <framework/pcore/zmq/connections/ZMQNullConnection.h>
#include <framework/pcore/zmq/connections/ZMQConfirmedConnection.h>
#include <framework/pcore/zmq/connections/ZMQRawConnection.h>
 
#include <framework/pcore/zmq/messages/ZMQIdMessage.h>
 
#include <framework/pcore/EvtMessage.h>
#include <daq/dqm/DqmSharedMem.h>
 
#include <string>
#include <memory>
#include <map>
#include <optional>
 
namespace Belle2 {
  class ZMQHistoServerToFileOutput : public ZMQConnection {
  public:
    ZMQHistoServerToFileOutput(unsigned int maximalUncompressedBufferSize,
                               const std::string& dqmFileName,
                               const std::string& rootFileName);
 
    void mergeAndSend(const std::map<std::string, HistogramMapping>& storedMessages, const std::optional<unsigned int>& experiment,
                      const std::optional<unsigned int>& run,
                      EMessageTypes messageType);
    void clear();
 
    std::vector<zmq::socket_t*> getSockets() const final;
    void handleIncomingData();
    std::string getEndPoint() const { return "file://" + m_rootFileName; }
  private:
    DqmSharedMem* m_sharedMemory = nullptr;
    std::string m_dqmMemFileName;
    std::string m_rootFileName;
  };
 
  class ZMQHistoServerToZMQOutput : public ZMQConnection {
  public:
    ZMQHistoServerToZMQOutput(const std::string& outputAddress, const std::shared_ptr<ZMQParent>& parent);
 
    void mergeAndSend(const std::map<std::string, HistogramMapping>& storedMessages, const std::optional<unsigned int>& experiment,
                      const std::optional<unsigned int>& run,
                      EMessageTypes messageType);
    void clear() {}
 
    std::string getMonitoringJSON() const final { return m_output.getMonitoringJSON(); }
    void handleIncomingData() { m_output.handleIncomingData(); }
    bool isReady() const override { return m_output.isReady(); }
    std::vector<zmq::socket_t*> getSockets() const final { return m_output.getSockets(); }
    std::string getEndPoint() const { return m_output.getEndPoint(); }
  private:
    ZMQConfirmedOutput m_output;
 
    std::vector<char> m_outputBuffer;
    unsigned int m_maximalCompressedSize = 100'000'000;
  };
 
  class ZMQHistoServerToRawOutput : public ZMQConnection {
  public:
    ZMQHistoServerToRawOutput(const std::string& outputAddress, const std::shared_ptr<ZMQParent>& parent);
 
    void mergeAndSend(const std::map<std::string, HistogramMapping>& storedMessages, const std::optional<unsigned int>& experiment,
                      const std::optional<unsigned int>& run,
                      EMessageTypes messageType);
    void clear() {}
 
    std::string getMonitoringJSON() const final { return m_output.getMonitoringJSON(); }
    void handleIncomingData() { m_output.handleIncomingData(); }
    bool isReady() const override { return m_output.isReady(); }
    std::vector<zmq::socket_t*> getSockets() const final { return m_output.getSockets(); }
    std::string getEndPoint() const { return m_output.getEndPoint(); }
  private:
    ZMQRawOutput m_output;
  };
}