ZMQHistogramConnection.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 <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/DqmMemFile.h>
#include <daq/rfarm/manager/SharedMem.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();
 
  private:
    SharedMem* 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 { return m_output.isReady(); }
    std::vector<zmq::socket_t*> getSockets() const final { return m_output.getSockets(); }
 
  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 { return m_output.isReady(); }
    std::vector<zmq::socket_t*> getSockets() const final { return m_output.getSockets(); }
 
  private:
    ZMQRawOutput m_output;
  };
}