itten
105 lines
3.7 KiB
C++
105 lines
3.7 KiB
C++
#include <string>
|
|
#include <mxl/flow.h>
|
|
#include <mxl/time.h>
|
|
#include "NodeBase.hpp"
|
|
#include "FlowDef.hpp"
|
|
#include "V210.hpp"
|
|
#include "NDIHelper.hpp"
|
|
#include <Processing.NDI.Lib.h>
|
|
#include <cstring>
|
|
|
|
class NDIInNode : public dmf::NodeBase {
|
|
void run() override {
|
|
dmf::NDIHelper ndi_helper;
|
|
try {
|
|
std::vector<std::string> ndi_sources;
|
|
ndi_helper.find_sources(&ndi_sources, 5000);
|
|
log("Available NDI sources:");
|
|
for (const auto& source_name : ndi_sources){
|
|
log("%s", source_name.c_str());
|
|
}
|
|
uint32_t source_num = 0;
|
|
ndi_helper.select_source(source_num);
|
|
ndi_helper.get_source_info(source_num);
|
|
} catch (const std::runtime_error& e) {
|
|
log("Error: %s", e.what());
|
|
return;
|
|
}
|
|
|
|
const auto flow_info = config().at("flow_id");
|
|
const auto flow_id = flow_info.at("id").get<std::string>();
|
|
const int width = flow_info.value("width", ndi_helper.xres);
|
|
const int height = flow_info.value("height", ndi_helper.yres);
|
|
const int fps_num = flow_info.value("fps_num", ndi_helper.frame_N);
|
|
const int fps_den = flow_info.value("fps_den", ndi_helper.frame_D);
|
|
|
|
log("flow=%s %dx%d @ %d/%d fps", flow_id.c_str(), width, height, fps_num, fps_den);
|
|
|
|
const std::string flow_def =
|
|
dmf::make_video_flow_def(flow_id, node_id(), width, height, fps_num, fps_den);
|
|
|
|
mxlFlowWriter writer{};
|
|
mxlFlowConfigInfo cfg_info{};
|
|
bool created = false;
|
|
|
|
mxlStatus st = mxlCreateFlowWriter(
|
|
instance(), flow_def.c_str(), nullptr, &writer, &cfg_info, &created);
|
|
if (st != MXL_STATUS_OK) {
|
|
log("mxlCreateFlowWriter failed (status=%d)", st);
|
|
return;
|
|
}
|
|
|
|
const uint32_t stride = cfg_info.discrete.sliceSizes[0];
|
|
log("stride=%u B/line grain=%u B ring=%u grains",
|
|
stride, stride * static_cast<uint32_t>(height), cfg_info.discrete.grainCount);
|
|
|
|
const mxlRational rate = {fps_num, fps_den};
|
|
uint64_t index = mxlGetCurrentIndex(&rate);
|
|
log("start index=%llu", index);
|
|
|
|
// because NDI can drop to 1 FPS for static frames, even if source is 29.97p
|
|
const size_t frame_bytes = stride * height;
|
|
uint8_t* latest_buffer = (uint8_t*)malloc(frame_bytes);
|
|
bool last_ndi_frame_valid = false;
|
|
|
|
while (dmf::g_running.load(std::memory_order_relaxed)) {
|
|
if (ndi_helper.getV210_video_frame(0, latest_buffer, stride)) {
|
|
last_ndi_frame_valid = true;
|
|
}
|
|
|
|
uint8_t* buf = nullptr;
|
|
mxlGrainInfo grain{};
|
|
|
|
st = mxlFlowWriterOpenGrain(writer, index, &grain, &buf);
|
|
if (st != MXL_STATUS_OK) {
|
|
log("OpenGrain failed (status=%d), skipping index=%llu", st, index);
|
|
index++;
|
|
continue;
|
|
}
|
|
|
|
if (last_ndi_frame_valid) {
|
|
std::memcpy(buf, latest_buffer, frame_bytes);
|
|
grain.flags = 0;
|
|
} else {
|
|
grain.flags = MXL_GRAIN_FLAG_INVALID;
|
|
}
|
|
|
|
grain.validSlices = grain.totalSlices; // mark grain complete so readers can consume it
|
|
mxlFlowWriterCommitGrain(writer, &grain);
|
|
|
|
const uint64_t ns = mxlGetNsUntilIndex(index + 1, &rate);
|
|
if (ns > 0 && ns < 2'000'000'000ULL) mxlSleepForNs(ns);
|
|
index++;
|
|
}
|
|
|
|
log("stopped at index=%llu", index);
|
|
free(latest_buffer);
|
|
mxlReleaseFlowWriter(instance(), writer);
|
|
}
|
|
};
|
|
|
|
int main()
|
|
{
|
|
NDIInNode node;
|
|
return node.execute();
|
|
} |