Files
cv_networking_pipeline/my_ndi_source.cpp
PodmogilnyjIvan 988ab528af v0.1
2021-12-03 03:54:58 -08:00

106 lines
3.9 KiB
C++

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <chrono>
#include <iostream>
#include <csignal>
#include <sstream>
#include <Processing.NDI.Lib.h>
using namespace std::chrono;
#ifdef _WIN32
#ifdef _WIN64
#pragma comment(lib, "Processing.NDI.Lib.x64.lib")
#else // _WIN64
#pragma comment(lib, "Processing.NDI.Lib.x86.lib")
#endif // _WIN64
#endif // _WIN32
static bool is_terminated = false;
void signalHandler(int signum) {
// Ctrl+C -> SIGINT
// SystemD -> SIGTERM
std::cout << "Interrupt signal (" << signum << ") received.";
is_terminated = true;
};
int main(int argc, char* argv[]) {
signal(SIGINT, signalHandler);
signal(SIGTERM, signalHandler);
// Not required, but "correct" (see the SDK documentation.
if (!NDIlib_initialize()) {
return 0;
}
// We create the NDI sender
NDIlib_send_instance_t pNDI_send = NDIlib_send_create();
if (!pNDI_send) return 0;
int64_t total_frames_counter = 0;
while (!is_terminated) {
// Get the current time
const auto start_send = high_resolution_clock::now();
// Send 200 frames
for (int idx = 200; idx; idx--) {
// Fill in the buffer. It is likely that you would do something much smarter than this.
// We are going to create a 1920x1080 interlaced frame at 29.97Hz.
int xres = 1920;
int yres = 1920;
std::string metadata_string = "<metadata_string " + std::to_string(total_frames_counter) + ">";
NDIlib_video_frame_v2_t ndi_video_frame(
xres, yres, NDIlib_FourCC_type_BGRX,
30000, 1001, 16.0 / 9.0,
NDIlib_frame_format_type_progressive,
0,
(uint8_t*)malloc(xres * yres * 4),
0,
metadata_string.c_str()
);
// std::cout << "hi there" << std::endl;
// std::cout << "xres = " << ndi_video_frame.xres << std::endl;
// std::cout << "yres = " << ndi_video_frame.yres << std::endl;
// std::cout << "frame_rate_N = " << ndi_video_frame.frame_rate_N << std::endl;
// std::cout << "frame_rate_D = " << ndi_video_frame.frame_rate_D << std::endl;
// std::cout << "picture_aspect_ratio = " << ndi_video_frame.picture_aspect_ratio << std::endl;
// std::cout << "frame_format_type = " << ndi_video_frame.frame_format_type << std::endl;
// std::cout << "timecode = " << ndi_video_frame.timecode << std::endl;
// std::cout << "line_stride_in_bytes = " << ndi_video_frame.line_stride_in_bytes << std::endl;
// std::cout << "p_metadata = " << ndi_video_frame.p_metadata << std::endl;
// std::cout << "timestamp = " << ndi_video_frame.timestamp << std::endl;
memset((void*)ndi_video_frame.p_data, 0x33, ndi_video_frame.xres * ndi_video_frame.yres * 4);
ndi_video_frame.timestamp = total_frames_counter;
ndi_video_frame.timecode = total_frames_counter;
// memset((void*)ndi_video_frame.p_data, (idx & 1) ? 255 : 0, ndi_video_frame.xres*ndi_video_frame.yres * 4);
// We now submit the frame. Note that this call will be clocked so that we end up submitting at exactly 29.97fps.
NDIlib_send_send_video_v2(pNDI_send, &ndi_video_frame);
total_frames_counter += 1;
free(ndi_video_frame.p_data);
// Free the video frame
// NDIlib_recv_free_video(pNDI_recv, &video_frame);
// free(ndi_video_frame.p_metadata);
}
// Just display something helpful
printf("200 frames sent, at %1.2ffps\n", 200.0f / duration_cast<duration<float>>(high_resolution_clock::now() - start_send).count());
}
// Destroy the NDI sender
NDIlib_send_destroy(pNDI_send);
// Not required, but nice
NDIlib_destroy();
// Success
return 0;
}