mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 16:46:47 +00:00
74 lines
2.0 KiB
C++
74 lines
2.0 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2014-2021 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 FAIR_MQ_TRANSPORTS_H
|
|
#define FAIR_MQ_TRANSPORTS_H
|
|
|
|
#include <fairmq/tools/Strings.h>
|
|
#include <memory>
|
|
#include <ostream>
|
|
#include <stdexcept>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
|
|
namespace fair::mq {
|
|
|
|
enum class Transport
|
|
{
|
|
DEFAULT,
|
|
ZMQ,
|
|
SHM,
|
|
OFI
|
|
};
|
|
|
|
struct TransportError : std::runtime_error
|
|
{
|
|
using std::runtime_error::runtime_error;
|
|
};
|
|
|
|
static std::unordered_map<std::string, Transport> TransportTypes{
|
|
{"default", Transport::DEFAULT},
|
|
{"zeromq", Transport::ZMQ},
|
|
{"shmem", Transport::SHM},
|
|
{"ofi", Transport::OFI}
|
|
};
|
|
|
|
static std::unordered_map<Transport, std::string> TransportNames{
|
|
{Transport::DEFAULT, "default"},
|
|
{Transport::ZMQ, "zeromq"},
|
|
{Transport::SHM, "shmem"},
|
|
{Transport::OFI, "ofi"}
|
|
};
|
|
|
|
inline std::string TransportName(Transport transport) { return TransportNames[transport]; }
|
|
|
|
inline Transport TransportType(const std::string& transport)
|
|
try {
|
|
return TransportTypes.at(transport);
|
|
} catch (std::out_of_range&) {
|
|
throw TransportError(tools::ToString("Unknown transport provided: ", transport));
|
|
}
|
|
|
|
inline std::ostream& operator<<(std::ostream& os, const Transport& transport)
|
|
{
|
|
return os << TransportName(transport);
|
|
}
|
|
|
|
inline auto GetEnabledTransports() -> std::vector<Transport>
|
|
{
|
|
#ifdef BUILD_OFI_TRANSPORT
|
|
return {Transport::ZMQ, Transport::SHM, Transport::OFI};
|
|
#else
|
|
return {Transport::ZMQ, Transport::SHM};
|
|
#endif
|
|
}
|
|
|
|
} // namespace fair::mq
|
|
|
|
#endif /* FAIR_MQ_TRANSPORTS_H */
|