mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 08:41:16 +00:00
73 lines
2.7 KiB
C++
73 lines
2.7 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH *
|
|
* *
|
|
* This software is distributed under the terms of the *
|
|
* GNU Lesser General Public Licence (LGPL) version 3, *
|
|
* copied verbatim in the file "LICENSE" *
|
|
********************************************************************************/
|
|
#ifndef FAIRMQMESSAGESHM_H_
|
|
#define FAIRMQMESSAGESHM_H_
|
|
|
|
#include <fairmq/shmem/Manager.h>
|
|
|
|
#include "FairMQMessage.h"
|
|
#include "FairMQUnmanagedRegion.h"
|
|
|
|
#include <zmq.h>
|
|
|
|
#include <boost/interprocess/mapped_region.hpp>
|
|
|
|
#include <cstddef> // size_t
|
|
#include <atomic>
|
|
|
|
class FairMQSocketSHM;
|
|
|
|
class FairMQMessageSHM final : public FairMQMessage
|
|
{
|
|
friend class FairMQSocketSHM;
|
|
|
|
public:
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, FairMQTransportFactory* factory = nullptr);
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, const size_t size, FairMQTransportFactory* factory = nullptr);
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr, FairMQTransportFactory* factory = nullptr);
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, FairMQUnmanagedRegionPtr& region, void* data, const size_t size, void* hint = 0, FairMQTransportFactory* factory = nullptr);
|
|
|
|
FairMQMessageSHM(const FairMQMessageSHM&) = delete;
|
|
FairMQMessageSHM operator=(const FairMQMessageSHM&) = delete;
|
|
|
|
void Rebuild() override;
|
|
void Rebuild(const size_t size) override;
|
|
void Rebuild(void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr) override;
|
|
|
|
void* GetData() const override;
|
|
size_t GetSize() const override { return fSize; }
|
|
|
|
bool SetUsedSize(const size_t size) override;
|
|
|
|
fair::mq::Transport GetType() const override { return fTransportType; }
|
|
|
|
void Copy(const FairMQMessage& msg) override;
|
|
|
|
~FairMQMessageSHM() override { CloseMessage(); }
|
|
|
|
private:
|
|
fair::mq::shmem::Manager& fManager;
|
|
zmq_msg_t fMessage;
|
|
bool fQueued;
|
|
bool fMetaCreated;
|
|
static std::atomic<bool> fInterrupted;
|
|
static fair::mq::Transport fTransportType;
|
|
size_t fRegionId;
|
|
mutable fair::mq::shmem::Region* fRegionPtr;
|
|
boost::interprocess::managed_shared_memory::handle_t fHandle;
|
|
size_t fSize;
|
|
size_t fHint;
|
|
mutable char* fLocalPtr;
|
|
|
|
bool InitializeChunk(const size_t size);
|
|
zmq_msg_t* GetMessage() { return &fMessage; }
|
|
void CloseMessage();
|
|
};
|
|
|
|
#endif /* FAIRMQMESSAGESHM_H_ */
|