rdmaTestServer.main.cpp

//
// Created by jonas on 08.04.21.
//

#include "fmt/format.h"
#include <thread>
#include <opencv2/opencv.hpp>

#include "rdmaLib.hpp"
#include "RDMAServer.hpp"
#include <csignal>


volatile bool done = false;

std::unique_ptr<RDMAServer> s = nullptr;

void intHandler(int sig) {
    fmt::print("Signal received! {}\n", sig);
    s.reset();
    done = true;
}


void receiveCB(Buffer<char> &&b) {
    auto start = std::chrono::high_resolution_clock::now();

    cv::Mat inputMatrix(HEIGHT, WIDTH, CV_8UC1, (void *) b.data());
    Buffer<char> sendB = s->getSendBuffer();
    cv::Mat outputMatrix(HEIGHT, WIDTH, CV_8UC1, (void *) sendB.data());
    cv::GaussianBlur(inputMatrix, outputMatrix, cv::Size(0, 0), 5);

    auto time = std::chrono::duration<double, std::milli>(std::chrono::high_resolution_clock::now() - start);
    fmt::print("Calculation took {}ms\n", time.count());

    s->send(std::move(sendB));
}


int main(int argc, char *argv[]) {

    s = std::make_unique<RDMAServer>(42069, BUFFER_SIZE, BUFFER_SIZE, []() { fmt::print("Client connected\n"); },
                                     &receiveCB);
    signal(SIGINT, intHandler);
    signal(SIGTERM, intHandler);

    while (not done) { std::this_thread::sleep_for(std::chrono::milliseconds(500)); }


    return 0;
}