StorageStreamHelper.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/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>
 
#include <framework/pcore/DataStoreStreamer.h>
#include <daq/storage/BinData.h>
 
#include <framework/dataobjects/EventT0.h>
#include <hlt/dataobjects/OnlineEventT0.h>
 
namespace Belle2 {
  class StorageStreamHelper {
  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;
    DataStoreStreamer* m_streamer;
 
    BinData m_data;
    BinData m_data_hlt;
    BinData m_data_pxd;
    int* m_buf;
    int m_eb2 = 1;
 
    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;
 
    StoreArray<OnlineEventT0> m_onlineEventT0;
 
    StoreObjPtr<EventT0> m_eventT0;
 
    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;
    */
  };
}