mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 08:41:16 +00:00
72 lines
2.4 KiB
C++
72 lines
2.4 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 : public FairMQMessage
|
|
{
|
|
friend class FairMQSocketSHM;
|
|
|
|
public:
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager);
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, const size_t size);
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, void* data, const size_t size, fairmq_free_fn* ffn, void* hint = nullptr);
|
|
FairMQMessageSHM(fair::mq::shmem::Manager& manager, FairMQUnmanagedRegionPtr& region, void* data, const size_t size);
|
|
|
|
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() override;
|
|
size_t GetSize() const override;
|
|
|
|
bool SetUsedSize(const size_t size) override;
|
|
|
|
FairMQ::Transport GetType() const override;
|
|
|
|
void Copy(const FairMQMessagePtr& msg) override;
|
|
|
|
~FairMQMessageSHM() override;
|
|
|
|
private:
|
|
fair::mq::shmem::Manager& fManager;
|
|
zmq_msg_t fMessage;
|
|
bool fQueued;
|
|
bool fMetaCreated;
|
|
static std::atomic<bool> fInterrupted;
|
|
static FairMQ::Transport fTransportType;
|
|
uint64_t fRegionId;
|
|
fair::mq::shmem::Region* fRegionPtr;
|
|
boost::interprocess::managed_shared_memory::handle_t fHandle;
|
|
size_t fSize;
|
|
char* fLocalPtr;
|
|
|
|
bool InitializeChunk(const size_t size);
|
|
zmq_msg_t* GetMessage();
|
|
void CloseMessage();
|
|
};
|
|
|
|
#endif /* FAIRMQMESSAGESHM_H_ */
|