HLTStreamHelper.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - 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/utils/StreamHelper.h>
#include <framework/pcore/zmq/messages/ZMQNoIdMessage.h>
#include <framework/datastore/StoreObjPtr.h>
#include <framework/datastore/StoreArray.h>
 
#include <rawdata/dataobjects/RawFTSW.h>
#include <rawdata/dataobjects/RawTLU.h>
#include <rawdata/dataobjects/RawCDC.h>
#include <rawdata/dataobjects/RawSVD.h>
#include <rawdata/dataobjects/RawECL.h>
#include <rawdata/dataobjects/RawARICH.h>
#include <rawdata/dataobjects/RawTOP.h>
#include <rawdata/dataobjects/RawKLM.h>
#include <rawdata/dataobjects/RawTRG.h>
#include <rawdata/dataobjects/RawCOPPER.h>
#include <rawdata/dataobjects/RawPXD.h>
#include <tracking/dataobjects/ROIpayload.h>
#include <tracking/dataobjects/ROIid.h>
#include <framework/dataobjects/EventMetaData.h>
#include <mdst/dataobjects/SoftwareTriggerResult.h>
#include <hlt/softwaretrigger/dataobjects/SoftwareTriggerVariables.h>
#include <mdst/dataobjects/TRGSummary.h>
 
#include <string>
#include <memory>
 
namespace Belle2 {
  class HLTStreamHelper {
  public:
    void initialize();
 
    std::unique_ptr<ZMQNoIdMessage> stream(bool addPersistentDurability, bool streamTransientObjects);
 
    std::unique_ptr<ZMQNoIdMessage> streamRaw();
 
    std::unique_ptr<ZMQNoIdMessage> streamHistograms(bool compressed = true);
 
    void read(std::unique_ptr<ZMQNoIdMessage> message);
 
    void registerStoreObjects(bool addExpressRecoObjects);
 
  private:
    StreamHelper m_streamHelper;
 
    StoreObjPtr<EventMetaData> m_eventMetaData;
    StoreArray<RawSVD> m_rawSVDs;
    StoreArray<RawCDC> m_rawCDCs;
    StoreArray<RawTOP> m_rawTOPs;
    StoreArray<RawARICH> m_rawARICHs;
    StoreArray<RawECL> m_rawECLs;
    StoreArray<RawKLM> m_rawKLMs;
    StoreArray<RawTRG> m_rawTRGs;
    StoreArray<RawFTSW> m_rawFTSWs;
    StoreObjPtr<ROIpayload> m_roiPayload;
 
    StoreObjPtr<RandomGenerator> m_randomGenerator;
    StoreObjPtr<SoftwareTriggerResult> m_softwareTriggerResult;
    StoreObjPtr<SoftwareTrigger::SoftwareTriggerVariables> m_softwareTriggerVariables;
    StoreObjPtr<TRGSummary> m_triggerSummary;
    StoreArray<RawPXD> m_rawPXDs;
    StoreArray<ROIid> m_rois;
 
    std::vector<char> m_outputBuffer;
    unsigned int m_maximalCompressedSize = 100'000'000;
 
    zmq::message_t getROIMessageIfViable() const;
  };
}