105 lines
3.2 KiB
C++
105 lines
3.2 KiB
C++
// Copyright (2019-2023) Paul Scherrer Institute
|
|
|
|
#ifndef JUNGFRAUJOCH_ACQUISITIONDEVICE_H
|
|
#define JUNGFRAUJOCH_ACQUISITIONDEVICE_H
|
|
|
|
#include <map>
|
|
#include <mutex>
|
|
#include <future>
|
|
#include <unistd.h>
|
|
|
|
#include "jfjoch.pb.h"
|
|
|
|
#include "../common/Definitions.h"
|
|
#include "../common/DiffractionExperiment.h"
|
|
#include "../common/Logger.h"
|
|
|
|
#include "../common/ThreadSafeFIFO.h"
|
|
#include "../jungfrau/JFCalibration.h"
|
|
|
|
#include "AcquisitionCounters.h"
|
|
#include "Completion.h"
|
|
|
|
struct IntegrationResult {
|
|
int64_t count;
|
|
int64_t sum;
|
|
int64_t sum2;
|
|
int64_t reserved[5];
|
|
};
|
|
|
|
class AcquisitionDevice {
|
|
std::vector<int16_t> buffer_err;
|
|
|
|
std::chrono::time_point<std::chrono::system_clock> start_time;
|
|
std::chrono::time_point<std::chrono::system_clock> end_time;
|
|
|
|
int64_t expected_frames;
|
|
|
|
virtual void Start(const DiffractionExperiment& experiment) = 0;
|
|
virtual void Finalize() {}; // do clean-up after action is done
|
|
virtual void StartSendingWorkRequests() {};
|
|
void SendWorkRequest(uint32_t handle);
|
|
protected:
|
|
AcquisitionCounters counters;
|
|
|
|
ThreadSafeFIFO<Completion> work_completion_queue;
|
|
ThreadSafeFIFO<WorkRequest> work_request_queue;
|
|
|
|
std::vector<uint16_t *> buffer_device;
|
|
Logger *logger;
|
|
|
|
uint16_t data_stream;
|
|
uint32_t max_modules = 1;
|
|
uint64_t mac_addr;
|
|
uint32_t ipv4_addr;
|
|
std::atomic<uint32_t> work_completion_count;
|
|
|
|
explicit AcquisitionDevice(uint16_t data_stream);
|
|
|
|
void UnmapBuffers();
|
|
void MapBuffersStandard(size_t c2h_buffer_count, size_t h2c_buffer_count, int16_t numa_node);
|
|
public:
|
|
static constexpr const uint64_t HandleNotValid = UINT64_MAX;
|
|
|
|
virtual ~AcquisitionDevice() { UnmapBuffers(); };
|
|
|
|
void StartAction(const DiffractionExperiment &experiment);
|
|
void PrepareAction(const DiffractionExperiment &experiment);
|
|
|
|
void WaitForActionComplete();
|
|
virtual void Cancel() = 0;
|
|
|
|
void EnableLogging(Logger *logger);
|
|
|
|
// Post measurement statistics - only guaranteed valid after WaitForActionComplete ends
|
|
uint64_t GetBytesReceived() const;
|
|
|
|
void SaveStatistics(const DiffractionExperiment &experiment, JFJochProtoBuf::AcquisitionDeviceStatistics &statistics) const;
|
|
virtual JFJochProtoBuf::FPGAStatus GetStatus() const { return {}; };
|
|
|
|
const int16_t *GetFrameBuffer(size_t frame_number, uint16_t module_number) const;
|
|
const IntegrationResult *GetIntegrationResult(size_t frame_number, uint16_t module_number) const;
|
|
|
|
void FrameBufferRelease(size_t frame_number, uint16_t module_number);
|
|
const int16_t *GetErrorFrameBuffer() const;
|
|
|
|
int16_t *GetDeviceBuffer(size_t handle);
|
|
|
|
// Calibration
|
|
virtual void InitializeCalibration(const DiffractionExperiment &experiment, const JFCalibration &calib);
|
|
virtual void InitializeIntegrationMap(const DiffractionExperiment &experiment, const std::vector<uint16_t> &v);
|
|
const AcquisitionCounters& Counters() const;
|
|
|
|
virtual std::string GetIPv4Address() const;
|
|
virtual std::string GetMACAddress() const;
|
|
virtual uint16_t GetUDPPort() const;
|
|
virtual int32_t GetNUMANode() const;
|
|
virtual uint32_t GetCompletedDescriptors() const = 0;
|
|
virtual std::vector<uint16_t> GetInternalGeneratorFrame() const {
|
|
return {};
|
|
}
|
|
};
|
|
|
|
|
|
#endif //JUNGFRAUJOCH_ACQUISITIONDEVICE_H
|