68 lines
2.2 KiB
C++
68 lines
2.2 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#ifndef JUNGFRAUJOCH_ACQUISITIONCOUNTERS_H
|
|
#define JUNGFRAUJOCH_ACQUISITIONCOUNTERS_H
|
|
|
|
#include <condition_variable>
|
|
#include <shared_mutex>
|
|
#include <vector>
|
|
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/Definitions.h"
|
|
#include "Completion.h"
|
|
|
|
// AcquisitionCounters are used for information that needs to be accessed during data collection,
|
|
// so uses mutex to ensure consistency
|
|
|
|
class AcquisitionCounters {
|
|
constexpr static const uint16_t expected_packets_per_module = 128;
|
|
constexpr static const uint64_t max_modules = 32;
|
|
|
|
mutable std::shared_mutex m;
|
|
mutable std::condition_variable_any data_updated;
|
|
|
|
std::vector<uint64_t> handle_for_frame;
|
|
std::vector<uint8_t> full_module_collected; // vector of bool might have weird concurrency behavior
|
|
|
|
std::vector<Completion> saved_completions;
|
|
|
|
uint64_t total_packets;
|
|
std::vector<uint64_t> packets_per_module;
|
|
|
|
uint64_t slowest_head;
|
|
std::vector<uint64_t> head;
|
|
bool acquisition_finished;
|
|
uint64_t expected_frames;
|
|
uint64_t nmodules = max_modules;
|
|
|
|
public:
|
|
static constexpr const uint64_t HandleNotFound = UINT64_MAX;
|
|
AcquisitionCounters();
|
|
|
|
void Reset(const DiffractionExperiment &experiment, uint16_t data_stream);
|
|
void UpdateCounters(const Completion *c);
|
|
void SetAcquisitionFinished();
|
|
|
|
uint64_t GetBufferHandleAndClear(size_t frame, uint16_t module_number);
|
|
|
|
uint64_t GetHead(uint16_t module_number) const;
|
|
uint64_t GetSlowestHead() const;
|
|
|
|
void WaitForFrame(size_t curr_frame, uint16_t module_number = UINT16_MAX) const;
|
|
int64_t CalculateDelay(size_t curr_frame, uint16_t module_number = UINT16_MAX) const; // mutex acquired indirectly
|
|
uint64_t GetBufferHandle(size_t frame, uint16_t module_number) const;
|
|
bool IsFullModuleCollected(size_t frame, uint16_t module_number) const;
|
|
bool IsAcquisitionFinished() const;
|
|
|
|
Completion GetCompletion(size_t curr_frame, uint16_t module_number) const;
|
|
|
|
uint64_t GetTotalPackets() const;
|
|
uint64_t GetTotalPackets(uint16_t module_number) const;
|
|
|
|
uint64_t GetExpectedPackets() const;
|
|
uint64_t GetExpectedPacketsPerModule() const;
|
|
};
|
|
|
|
|
|
#endif //JUNGFRAUJOCH_ACQUISITIONCOUNTERS_H
|