mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-13 08:41:16 +00:00
103 lines
3.1 KiB
C++
103 lines
3.1 KiB
C++
/********************************************************************************
|
|
* Copyright (C) 2019 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" *
|
|
********************************************************************************/
|
|
|
|
#include "DDSTopology.h"
|
|
|
|
#include <boost/range/iterator_range.hpp>
|
|
|
|
#include <fairmq/sdk/DDSEnvironment.h>
|
|
#include <fairmq/Tools.h>
|
|
|
|
#include <fairlogger/Logger.h>
|
|
|
|
#include <DDS/Topology.h>
|
|
|
|
#include <sstream>
|
|
#include <stdexcept>
|
|
#include <utility>
|
|
#include <memory>
|
|
|
|
namespace fair {
|
|
namespace mq {
|
|
namespace sdk {
|
|
|
|
struct DDSTopology::Impl
|
|
{
|
|
explicit Impl(Path topoFile, DDSEnvironment env)
|
|
: fEnv(std::move(env))
|
|
, fTopoFile(std::move(topoFile))
|
|
, fTopo(fTopoFile.string())
|
|
{}
|
|
|
|
explicit Impl(dds::topology_api::CTopology nativeTopology, DDSEnvironment env)
|
|
: fEnv(std::move(env))
|
|
, fTopo(std::move(nativeTopology))
|
|
{}
|
|
|
|
DDSEnvironment fEnv;
|
|
Path fTopoFile;
|
|
dds::topology_api::CTopology fTopo;
|
|
};
|
|
|
|
DDSTopology::DDSTopology(Path topoFile, DDSEnvironment env)
|
|
: fImpl(std::make_shared<Impl>(std::move(topoFile), std::move(env)))
|
|
{}
|
|
|
|
DDSTopology::DDSTopology(dds::topology_api::CTopology nativeTopology, DDSEnv env)
|
|
: fImpl(std::make_shared<Impl>(std::move(nativeTopology), std::move(env)))
|
|
{}
|
|
|
|
auto DDSTopology::GetEnv() const -> DDSEnvironment { return fImpl->fEnv; }
|
|
|
|
auto DDSTopology::GetTopoFile() const -> Path
|
|
{
|
|
auto file(fImpl->fTopoFile);
|
|
if (file.string().empty()) {
|
|
throw std::runtime_error("DDS topology xml spec file unknown");
|
|
}
|
|
return file;
|
|
}
|
|
|
|
int DDSTopology::GetNumRequiredAgents()
|
|
{
|
|
return fImpl->fTopo.getRequiredNofAgents();
|
|
}
|
|
|
|
std::vector<uint64_t> DDSTopology::GetDeviceList()
|
|
{
|
|
std::vector<uint64_t> taskIDs;
|
|
taskIDs.reserve(GetNumRequiredAgents());
|
|
|
|
// TODO make sure returned tasks are actually devices
|
|
auto itPair = fImpl->fTopo.getRuntimeTaskIterator(
|
|
[](const dds::topology_api::STopoRuntimeTask::FilterIterator_t::value_type& /*value*/) -> bool { return true; });
|
|
auto tasks = boost::make_iterator_range(itPair.first, itPair.second);
|
|
|
|
for (auto task : tasks) {
|
|
LOG(debug) << "Found task " << task.first << ": "
|
|
<< "Path: " << task.second.m_taskPath << ", "
|
|
<< "Name: " << task.second.m_task->getName() << "_" << task.second.m_taskIndex;
|
|
taskIDs.push_back(task.first);
|
|
}
|
|
|
|
return taskIDs;
|
|
}
|
|
|
|
auto DDSTopology::GetName() const -> std::string { return fImpl->fTopo.getName(); }
|
|
|
|
auto operator<<(std::ostream& os, const DDSTopology& t) -> std::ostream&
|
|
try {
|
|
return os << "DDS topology: " << t.GetName() << " (loaded from " << t.GetTopoFile() << ")";
|
|
} catch (std::runtime_error&) {
|
|
return os << "DDS topology: " << t.GetName();
|
|
}
|
|
|
|
} // namespace sdk
|
|
} // namespace mq
|
|
} // namespace fair
|