forked from AmbaPant/mantid
-
Notifications
You must be signed in to change notification settings - Fork 1
/
EventsListsShmemManager.cpp
64 lines (52 loc) · 2.59 KB
/
EventsListsShmemManager.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
// NScD Oak Ridge National Laboratory, European Spallation Source,
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidParallel/IO/EventsListsShmemManager.h"
#include <random>
#include <sstream>
#include <utility>
#include "MantidParallel/IO/EventsListsShmemManager.h"
#include "MantidTypes/Event/TofEvent.h"
namespace Mantid::Parallel::IO {
EventsListsShmemManager::EventsListsShmemManager(std::string segmentName, std::string elName)
: m_segmentName(std::move(segmentName)), m_chunksName(std::move(elName)), m_chunks(nullptr) {
m_segment = std::make_unique<ip::managed_shared_memory>(ip::open_only, m_segmentName.c_str());
m_allocatorInstance = std::make_unique<VoidAllocator>(m_segment->get_segment_manager());
m_chunks = m_segment->find<Chunks>(m_chunksName.c_str()).first;
if (!m_chunks)
throw std::invalid_argument("No event lists found.");
}
EventsListsShmemManager::EventsListsShmemManager(std::string segmentName, std::string elName, int /*unused*/)
: m_segmentName(std::move(segmentName)), m_chunksName(std::move(elName)), m_chunks(nullptr) {}
/// Appends ToF event to given pixel in given chunk of shared storage
void EventsListsShmemManager::appendEvent(std::size_t chunkN, std::size_t listN, const Types::Event::TofEvent &event) {
assert(m_chunks);
if (chunkN >= m_chunks->size())
throw std::invalid_argument(std::string("Number of chunks is ") + std::to_string(m_chunks->size()) +
", asked for index " + std::to_string(chunkN));
if (listN >= m_chunks->at(chunkN).size())
throw std::invalid_argument(std::string("Number of pixels is ") + std::to_string(m_chunks->at(chunkN).size()) +
", asked for index " + std::to_string(listN));
auto &list = m_chunks->at(chunkN).at(listN);
list.emplace_back(event);
}
std::ostream &operator<<(std::ostream &os, const EventsListsShmemManager &manager) {
os << "m_segmentName: " << manager.m_segmentName << " m_eventListsName: " << manager.m_chunksName << "\n";
for (auto &chunk : *manager.m_chunks) {
std::stringstream ss;
for (auto &list : chunk) {
ss << "[ ";
for (auto &event : list)
ss << event.tof() << ", ";
ss << "]\n";
}
os << ss.str() << "\v";
}
os << "\n\n";
return os;
}
const VoidAllocator &EventsListsShmemManager::alloc() const { return *m_allocatorInstance.get(); }
} // namespace Mantid::Parallel::IO