restructure zmq socker interface

zmq socket can now return vector of frames. it knows end of transmission with header.data == 0
it can also send vector of frames
This commit is contained in:
Bechir
2024-04-09 08:32:53 +02:00
committed by Bechir Braham
parent cd46f59b99
commit a2499c96d6
18 changed files with 255 additions and 87 deletions

View File

@ -1,27 +1,33 @@
#include "aare/ZmqSocketReceiver.hpp"
#include "aare/network_io/ZmqSocketReceiver.hpp"
#include "aare/network_io/defs.hpp"
#include <cassert>
#include <fmt/core.h>
#include <string>
using namespace aare;
int main() {
// aare::logger::set_verbosity(aare::logger::DEBUG);
std::string endpoint = "tcp://localhost:5555";
aare::ZmqSocketReceiver socket(endpoint);
socket.connect();
char *data = new char[1024 * 1024 * 10];
aare::ZmqHeader header;
while (true) {
int rc = socket.receive(header, reinterpret_cast<std::byte *>(data));
aare::logger::info("Received bytes", rc, "Received header: ", header.to_string());
auto *data_int = reinterpret_cast<uint32_t *>(data);
for (uint32_t i = 0; i < header.npixelsx; i++) {
for (uint32_t j = 0; j < header.npixelsy; j++) {
// verify that the sent data is correct
assert(data_int[i * header.npixelsy + j] == i + j);
std::vector<ZmqFrame> v = socket.receive_n();
aare::logger::info("Received ", v.size(), " frames");
aare::logger::info("acquisition:", v[0].header.acqIndex);
aare::logger::info("Header size:", v[0].header.to_string().size());
aare::logger::info("Frame size:", v[0].frame.size());
aare::logger::info("Header:", v[0].header.to_string());
for (ZmqFrame zmq_frame : v) {
auto &[header, frame] = zmq_frame;
for (int i = 0; i < 1024; i++) {
for (int j = 0; j < 1024; j++) {
assert(*(uint32_t *)frame.get(i, j) == (uint32_t)i + j);
}
}
aare::logger::info("Frame verified");
}
aare::logger::info("Frame verified");
}
delete[] data;
return 0;
}