JohannesItten
bb7d1bb5a3
Pass MXL grain stride (from sliceSizes[0]) into get_next_frame so
YUV422P10toV210 writes with the correct line width instead of a
self-computed value that may not match the MXL buffer.
Fix audio stream detection: condition was inverted (!= -1 → == -1),
so the first audio stream was never picked up.
Add return false at end of get_next_frame to fix UB when g_running
goes false and the loop exits without returning.
Replace av_frame_unref/av_packet_unref with av_frame_free/av_packet_free
in destructor — unref only releases data, not the struct itself.
Add av_freep(&p10_data[0]) in destructor to free av_image_alloc memory.
Remove unused p10_buffer and v210_buffer allocations.
Read filename from config("file") instead of hardcoded path.
Add early return if mxlCreateFlowWriter fails.
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
84 lines
2.9 KiB
C++
84 lines
2.9 KiB
C++
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 <string>
|
|
#include <mxl/flow.h>
|
|
#include <mxl/time.h>
|
|
#include "V210.hpp"
|
|
#include "FlowDef.hpp"
|
|
#include "NodeBase.hpp"
|
|
#include "VideoReader.hpp"
|
|
|
|
class VideoInNode : public dmf::NodeBase {
|
|
void run() override {
|
|
const std::string filename = config().value("file", std::string{});
|
|
if (filename.empty()) { log("config missing 'file'"); return; }
|
|
log("VideoIn Node started with file: %s", filename.c_str());
|
|
dmf::VideoReader video_reader(filename);
|
|
|
|
const auto video_flow_info = config().at("video_flow_id");
|
|
const auto video_flow_id = video_flow_info.at("id").get<std::string>();
|
|
const int width = video_flow_info.value("width", video_reader.source_info.width);
|
|
const int height = video_flow_info.value("height", video_reader.source_info.height);
|
|
const int fps_num = video_flow_info.value("fps_num", video_reader.source_info.fps_num);
|
|
const int fps_den = video_flow_info.value("fps_den", video_reader.source_info.fps_den);
|
|
|
|
mxlFlowWriter video_writer = nullptr;
|
|
mxlFlowConfigInfo video_config = {};
|
|
std::string video_flow_def = dmf::make_video_flow_def(
|
|
video_flow_id,
|
|
node_id(),
|
|
width,
|
|
height,
|
|
fps_num,
|
|
fps_den
|
|
);
|
|
bool created = false;
|
|
mxlStatus vst = mxlCreateFlowWriter(
|
|
instance(),
|
|
video_flow_def.c_str(),
|
|
"",
|
|
&video_writer,
|
|
&video_config,
|
|
&created
|
|
);
|
|
if (vst != MXL_STATUS_OK) {
|
|
log("mxlCreateFlowWriter failed (%s)", dmf::mxl_status_str(vst));
|
|
return;
|
|
}
|
|
const uint32_t video_stride = video_config.discrete.sliceSizes[0];
|
|
|
|
mxlRational video_rate = {fps_num, fps_den};
|
|
uint64_t video_index = mxlGetCurrentIndex(&video_rate);
|
|
|
|
while (dmf::g_running.load(std::memory_order_relaxed)) {
|
|
uint8_t* buf = nullptr;
|
|
mxlGrainInfo grain{};
|
|
vst = mxlFlowWriterOpenGrain(video_writer, video_index, &grain, &buf);
|
|
if (vst == MXL_STATUS_OK) {
|
|
if (!video_reader.get_next_frame(buf, video_stride, nullptr)) continue;
|
|
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++;
|
|
}
|
|
|
|
mxlReleaseFlowWriter(instance(), video_writer);
|
|
}
|
|
};
|
|
|
|
|
|
int main() {
|
|
VideoInNode node;
|
|
return node.execute();
|
|
}
|