mirror of
https://github.com/paulscherrerinstitute/sf_daq_buffer.git
synced 2026-05-01 16:42:23 +02:00
First implementation of UDP receiver core
This commit is contained in:
@@ -0,0 +1,32 @@
|
||||
#ifndef UDPRECVMODULE_HPP
|
||||
#define UDPRECVMODULE_HPP
|
||||
|
||||
#include "RingBuffer.hpp"
|
||||
#include <thread>
|
||||
|
||||
class UdpRecvModule {
|
||||
|
||||
RingBuffer<UdpFrameMetadata>& ring_buffer_;
|
||||
|
||||
std::atomic_bool is_receiving_;
|
||||
std::thread receiving_thread_;
|
||||
|
||||
protected:
|
||||
void receive_thread(
|
||||
const uint16_t udp_port,
|
||||
const size_t udp_buffer_n_bytes);
|
||||
|
||||
public:
|
||||
UdpRecvModule(RingBuffer<UdpFrameMetadata>& ring_buffer);
|
||||
|
||||
virtual ~UdpRecvModule() = default;
|
||||
|
||||
void start_recv(
|
||||
const uint16_t udp_port,
|
||||
const size_t udp_buffer_n_bytes);
|
||||
void stop_recv();
|
||||
bool is_receiving();
|
||||
};
|
||||
|
||||
|
||||
#endif // UDPRECVMODULE_HPP
|
||||
@@ -0,0 +1,124 @@
|
||||
#include "UdpRecvModule.hpp"
|
||||
#include "jungfrau.hpp"
|
||||
#include <iostream>
|
||||
#include <UdpReceiver.hpp>
|
||||
|
||||
using namespace std;
|
||||
|
||||
UdpRecvModule::UdpRecvModule(RingBuffer<UdpFrameMetadata>& ring_buffer) :
|
||||
ring_buffer_(ring_buffer),
|
||||
is_receiving_(false)
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
void UdpRecvModule::start_recv(
|
||||
const uint16_t udp_port,
|
||||
const size_t udp_buffer_n_bytes)
|
||||
{
|
||||
if (is_receiving_ == true) {
|
||||
std::stringstream err_msg;
|
||||
|
||||
using namespace date;
|
||||
using namespace chrono;
|
||||
err_msg << "[" << system_clock::now() << "]";
|
||||
err_msg << "[UdpRecvModule::start_recv]";
|
||||
err_msg << " Receivers already running." << endl;
|
||||
|
||||
throw runtime_error(err_msg.str());
|
||||
}
|
||||
|
||||
#ifdef DEBUG_OUTPUT
|
||||
using namespace date;
|
||||
using namespace chrono;
|
||||
cout << "[" << system_clock::now() << "]";
|
||||
cout << "[UdpRecvModule::start_recv]";
|
||||
cout << " Starting with ";
|
||||
cout << "udp_port " << udp_port << endl;
|
||||
#endif
|
||||
|
||||
is_receiving_ = true;
|
||||
|
||||
if (receiving_thread_.joinable()) {
|
||||
receiving_thread_.join();
|
||||
}
|
||||
|
||||
receiving_thread_ = thread(
|
||||
&UdpRecvModule::receive_thread, this,
|
||||
udp_port,
|
||||
udp_buffer_n_bytes);
|
||||
}
|
||||
|
||||
void UdpRecvModule::receive_thread(
|
||||
const uint16_t udp_port,
|
||||
const size_t udp_buffer_n_bytes)
|
||||
{
|
||||
try {
|
||||
ring_buffer_.initialize(udp_buffer_n_bytes);
|
||||
|
||||
UdpReceiver udp_receiver;
|
||||
udp_receiver.bind(udp_port);
|
||||
|
||||
jungfrau_packet packet_buffer;
|
||||
char* frame_buffer = nullptr;
|
||||
shared_ptr<UdpFrameMetadata> metadata = nullptr;
|
||||
|
||||
while (is_receiving_.load(memory_order_relaxed)) {
|
||||
|
||||
if (!udp_receiver.receive(
|
||||
&packet_buffer,
|
||||
JUNGFRAU_BYTES_PER_PACKET)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
auto* current_metadata = metadata.get();
|
||||
|
||||
if (packet_buffer.framenum != current_metadata->frame_index) {
|
||||
if (frame_buffer != nullptr) {
|
||||
ring_buffer_.commit(metadata);
|
||||
}
|
||||
|
||||
metadata = make_shared<UdpFrameMetadata>();
|
||||
current_metadata = metadata.get();
|
||||
|
||||
current_metadata->frame_index = packet_buffer.framenum;
|
||||
current_metadata->pulse_id = packet_buffer.bunchid;
|
||||
current_metadata->frame_bytes_size = JUNGFRAU_DATA_BYTES_PER_FRAME;
|
||||
current_metadata->recv_packets_1 = 0;
|
||||
current_metadata->recv_packets_2 = 0;
|
||||
|
||||
frame_buffer = ring_buffer_.reserve(metadata);
|
||||
memset(frame_buffer, 0, JUNGFRAU_DATA_BYTES_PER_FRAME);
|
||||
}
|
||||
|
||||
size_t frame_buffer_offset =
|
||||
JUNGFRAU_DATA_BYTES_PER_PACKET * packet_buffer.packetnum;
|
||||
|
||||
memcpy(
|
||||
(void*) (frame_buffer + frame_buffer_offset),
|
||||
packet_buffer.data,
|
||||
JUNGFRAU_DATA_BYTES_PER_PACKET);
|
||||
|
||||
if (packet_buffer.packetnum < 64) {
|
||||
current_metadata->recv_packets_1 ^=
|
||||
(uint64_t)1 << packet_buffer.packetnum;
|
||||
} else {
|
||||
current_metadata->recv_packets_2 ^=
|
||||
(uint64_t)1 << (packet_buffer.packetnum - 64);
|
||||
}
|
||||
}
|
||||
|
||||
} catch (const std::exception& e) {
|
||||
is_receiving_ = false;
|
||||
|
||||
using namespace date;
|
||||
using namespace chrono;
|
||||
|
||||
cout << "[" << system_clock::now() << "]";
|
||||
cout << "[UdpRecvModule::receive_thread]";
|
||||
cout << " Stopped because of exception: " << endl;
|
||||
cout << e.what() << endl;
|
||||
|
||||
throw;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user