it's alive

This commit is contained in:
itten
2026-07-03 14:01:13 +03:00
parent a422cd832a
commit 2c43f356d1
4 changed files with 238 additions and 0 deletions
+21
View File
@@ -73,18 +73,39 @@ FetchContent_Declare(json
URL https://github.com/nlohmann/json/releases/download/v3.11.3/json.tar.xz) URL https://github.com/nlohmann/json/releases/download/v3.11.3/json.tar.xz)
FetchContent_MakeAvailable(json) FetchContent_MakeAvailable(json)
# ── FFmpeg ──────────────────────────────────────────────────────────────────
find_package(PkgConfig REQUIRED)
# Check for FFmpeg components
pkg_check_modules(FFMPEG REQUIRED
libavformat
libavcodec
libswscale
libavutil
)
# Create an interface library for FFmpeg
add_library(ffmpeg INTERFACE)
target_include_directories(ffmpeg INTERFACE ${FFMPEG_INCLUDE_DIRS})
target_link_libraries(ffmpeg INTERFACE ${FFMPEG_LIBRARIES})
# ── Shared utilities (Signal.hpp, NodeBase.hpp, FlowDef.hpp, V210.hpp) ─────── # ── Shared utilities (Signal.hpp, NodeBase.hpp, FlowDef.hpp, V210.hpp) ───────
add_library(dmf-shared INTERFACE) add_library(dmf-shared INTERFACE)
target_include_directories(dmf-shared INTERFACE ${CMAKE_CURRENT_SOURCE_DIR}/shared) target_include_directories(dmf-shared INTERFACE ${CMAKE_CURRENT_SOURCE_DIR}/shared)
target_link_libraries(dmf-shared INTERFACE mxl nlohmann_json::nlohmann_json) target_link_libraries(dmf-shared INTERFACE mxl nlohmann_json::nlohmann_json)
# ── Basic nodes ──────────────────────────────────────────────────────────────
add_subdirectory(nodes/testpattern) add_subdirectory(nodes/testpattern)
add_subdirectory(nodes/fakesink) add_subdirectory(nodes/fakesink)
# ── NDI nodes ────────────────────────────────────────────────────────────────
set(NDI_SDK_DIR "" CACHE PATH "Path to NDI SDK root") set(NDI_SDK_DIR "" CACHE PATH "Path to NDI SDK root")
if(NDI_SDK_DIR) if(NDI_SDK_DIR)
add_subdirectory(nodes/ndiout) add_subdirectory(nodes/ndiout)
add_subdirectory(nodes/ndiin) add_subdirectory(nodes/ndiin)
endif() endif()
add_subdirectory(nodes/videoin)
# ── Core server ──────────────────────────────────────────────────────────────
add_subdirectory(studio-manager) add_subdirectory(studio-manager)
+3
View File
@@ -145,6 +145,9 @@ class NDIInNode : public dmf::NodeBase {
} }
grain.validSlices = grain.totalSlices; grain.validSlices = grain.totalSlices;
mxlFlowWriterCommitGrain(video_writer, &grain); mxlFlowWriterCommitGrain(video_writer, &grain);
const uint64_t ns = mxlGetNsUntilIndex(video_index + 1, &video_rate);
if (ns > 0 && ns < 2'000'000'000ULL) mxlSleepForNs(ns);
video_index++;
} else { } else {
log("video OpenGrain failed (%s) at index=%llu", dmf::mxl_status_str(st), video_index); log("video OpenGrain failed (%s) at index=%llu", dmf::mxl_status_str(st), video_index);
} }
+8
View File
@@ -0,0 +1,8 @@
add_executable(dmf-node-videoin main.cpp)
target_compile_features(dmf-node-videoin PRIVATE cxx_std_20)
target_link_libraries(dmf-node-videoin
PRIVATE
dmf-shared
ffmpeg
)
install(TARGETS dmf-node-videoin RUNTIME DESTINATION bin)
+206
View File
@@ -0,0 +1,206 @@
extern "C" {
#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
#include <libavutil/avutil.h>
#include <libswscale/swscale.h>
#include <libavutil/imgutils.h>
#include <libavutil/pixdesc.h>
}
#include <iostream>
#include <string>
#include <mxl/mxl.h>
#include <mxl/flow.h>
#include <mxl/time.h>
#include "V210.hpp"
#include "FlowDef.hpp"
#include "NodeBase.hpp"
int main(int argc, char* argv[]) {
// Open video file
AVFormatContext* formatContext = nullptr;
const char* filename = (argc > 1) ? argv[1] : "/home/itten/test-vid/0.ts";
if (avformat_open_input(&formatContext, filename, nullptr, nullptr) != 0) {
std::cerr << "Could not open file: " << filename << std::endl;
return 1;
}
// Find stream info
if (avformat_find_stream_info(formatContext, nullptr) < 0) {
std::cerr << "Could not find stream info" << std::endl;
avformat_close_input(&formatContext);
return 1;
}
// Find video stream
int videoStreamIndex = -1;
for (unsigned int i = 0; i < formatContext->nb_streams; i++) {
if (formatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
videoStreamIndex = i;
break;
}
}
if (videoStreamIndex == -1) {
std::cerr << "No video stream found" << std::endl;
avformat_close_input(&formatContext);
return 1;
}
// Get codec parameters
AVCodecParameters* codecParams = formatContext->streams[videoStreamIndex]->codecpar;
const AVCodec* codec = avcodec_find_decoder(codecParams->codec_id);
if (!codec) {
std::cerr << "Unsupported codec" << std::endl;
avformat_close_input(&formatContext);
return 1;
}
// Open codec
AVCodecContext* codecContext = avcodec_alloc_context3(codec);
avcodec_parameters_to_context(codecContext, codecParams);
if (avcodec_open2(codecContext, codec, nullptr) < 0) {
std::cerr << "Could not open codec" << std::endl;
avcodec_free_context(&codecContext);
avformat_close_input(&formatContext);
return 1;
}
std::cout << "Video info:" << std::endl;
std::cout << " Width: " << codecContext->width << std::endl;
std::cout << " Height: " << codecContext->height << std::endl;
std::cout << " Pixel format: " << av_get_pix_fmt_name(codecContext->pix_fmt) << std::endl;
std::cout << " Frame rate: " << codecContext->framerate.num << "/" << codecContext->framerate.den << std::endl;
// Create Sws context to convert to UYVY422
struct SwsContext *sws_ctx = sws_getContext(
codecContext->width, codecContext->height, codecContext->pix_fmt,
codecContext->width, codecContext->height, AV_PIX_FMT_UYVY422,
NULL, NULL, NULL, NULL
);
if (!sws_ctx) {
std::cerr << "Failed to create SwsContext" << std::endl;
}
// Allocate UYVY buffer
int uyvy_buf_size = av_image_get_buffer_size(
AV_PIX_FMT_UYVY422,
codecContext->width,
codecContext->height,
32
);
if (uyvy_buf_size < 0) {
std::cerr << "Failed to calculate UYVY buffer size" << std::endl;
}
uint8_t* uyvy_buffer = (uint8_t*)av_malloc(uyvy_buf_size);
if (!uyvy_buffer) {
std::cerr << "Failed to allocate UYVY buffer" << std::endl;
}
int uyvy_line_size = av_image_get_linesize(AV_PIX_FMT_UYVY422, codecContext->width, 0);
if (uyvy_line_size < 0) {
std::cerr << "Failed to get UYVY line size" << std::endl;
}
uint8_t* uyvy_data[4] = {uyvy_buffer, nullptr, nullptr, nullptr};
int uyvy_line_sizes[4] = {uyvy_line_size, 0, 0, 0};
// Allocate v210 buffer
int blocks_per_row = (codecContext->width + 5) / 6;
int v210_bytes_per_row = blocks_per_row * 16;
int v210_stride = ((v210_bytes_per_row + 63) / 64) * 64;
int v210_buffer_size = v210_stride * codecContext->height;
uint8_t* v210_buffer = (uint8_t*)av_malloc(v210_buffer_size);
if (!v210_buffer) {
std::cerr << "Failed to allocate V210 buffer" << std::endl;
}
// MXL prep
int fps_num = 25, fps_den = 1;
mxlInstance mxl_instance = mxlCreateInstance("/tmp/videotest-domain", nullptr);
std::string video_flow_def = "";
mxlFlowWriter video_writer = nullptr;
mxlFlowConfigInfo video_config = {};
std::string flow_uuid = "5fbec3b1-1b0f-417d-9059-8b94a47197ed";
video_flow_def = dmf::make_video_flow_def(
flow_uuid,
"libav video flow",
codecContext->width,
codecContext->height,
fps_num,
fps_den
);
bool is_flow_created = false;
mxlStatus vst =
mxlCreateFlowWriter(mxl_instance, video_flow_def.c_str(), "", &video_writer, &video_config, &is_flow_created);
if (vst != MXL_STATUS_OK) {
std::cerr << "MXL flow writer is not created. Reason: " << dmf::mxl_status_str(vst) << std::endl;
}
mxlRational video_rate = {fps_num, fps_den};
uint64_t video_index = mxlGetCurrentIndex(&video_rate);
// Allocate packets and frames
AVPacket* packet = av_packet_alloc();
AVFrame* frame = av_frame_alloc();
// Read and decode frames (example - just count them)
int frameCount = 0;
while (av_read_frame(formatContext, packet) >= 0) {
if (packet->stream_index == videoStreamIndex) {
if (avcodec_send_packet(codecContext, packet) == 0) {
while (avcodec_receive_frame(codecContext, frame) == 0) {
frameCount++;
// mxl part
mxlGrainInfo grain{};
uint8_t* buf = nullptr;
vst = mxlFlowWriterOpenGrain(video_writer, video_index, &grain, &buf);
if (vst == MXL_STATUS_OK) {
// source pix_fmt -> UYVY422 pix_fmt
sws_scale(sws_ctx, frame->data, frame->linesize, 0, codecContext->height, uyvy_data, uyvy_line_sizes);
// UYVY -> V210
dmf::v210::UYVYtoV210(
uyvy_buffer,
buf,
codecContext->width,
codecContext->height,
uyvy_line_size,
v210_stride
);
grain.flags = 0;
grain.validSlices = grain.totalSlices;
mxlFlowWriterCommitGrain(video_writer, &grain);
}
const uint64_t ns = mxlGetNsUntilIndex(video_index + 1, &video_rate);
if (ns > 0 && ns < 2'000'000'000ULL) mxlSleepForNs(ns);
video_index++;
av_frame_unref(frame);
}
}
}
av_packet_unref(packet);
}
std::cout << "Total frames: " << frameCount << std::endl;
// Cleanup MXL
mxlReleaseFlowWriter(mxl_instance, video_writer);
// Cleanup context
sws_freeContext(sws_ctx);
// Cleanup common
av_frame_free(&frame);
av_packet_free(&packet);
avcodec_free_context(&codecContext);
avformat_close_input(&formatContext);
return 0;
}