C++ 将gstreamer YUV 4:2:0（I420）原始帧转换为OpenCV:：Mat
我是新来的，仍在学习C++ 将gstreamer YUV 4:2:0（I420）原始帧转换为OpenCV:：Mat,c++,c,opencv,gstreamer,yuv,C++,C,Opencv,Gstreamer,Yuv,我是新来的，仍在学习gstreamer及其概念，因此在gstreamer教程中提到的后续实践就是这样做的一部分 很容易完成本教程末尾提到的练习，即将video/x-rawsrc pad动态链接到videoconvert和autovideosink 为了提高学习曲线，我决定制作一个C++ ggTalver虚拟插件，链接到 opencv并将原始帧转换成CV::Mat ，并通过 CV:从现在起，C++gstreamer插件（元素）将被称为myfilter 但首先我需要一个capsFilter链接vid
gstreamervideo/x-rawvideoconvertautovideosink为了提高学习曲线，我决定制作一个C++ ggTalver虚拟插件，链接到myfilter
capsFiltervideoconvertmyfiltervideoconvertBGRI420autovideosinkmyfilterBGRfakesinkBGRcv:：Matcv:：imshowmyfilter多亏了斯文·尼尔森，答案很简单
cv:：Mat frame（cv:：Size（宽度、高度）、cv_8UC3，（char*）map.data（宽度*3）+2）
我不得不将opencv的cv:：Mat:：AUTO_STEP
更改为cols*elemSize（）
为（宽度*3）+2
，因为854不能被4整除。这里是胡乱猜测，但可能是某种填充，因为854不能被4整除。
#include <gst/gst.h>
/* Functions below print the Capabilities in a human-friendly format */
static gboolean print_field(GQuark field, const GValue *value, gpointer pfx) {
gchar *str = gst_value_serialize(value);
g_print("%s %15s: %s\n", (gchar *)pfx, g_quark_to_string(field), str);
g_free(str);
return TRUE;
}
static void print_caps(const GstCaps *caps, const gchar *pfx) {
guint i;
g_return_if_fail(caps != NULL);
if (gst_caps_is_any(caps)) {
g_print("%sANY\n", pfx);
return;
}
if (gst_caps_is_empty(caps)) {
g_print("%sEMPTY\n", pfx);
return;
}
for (i = 0; i < gst_caps_get_size(caps); i++) {
GstStructure *structure = gst_caps_get_structure(caps, i);
g_print("%s%s\n", pfx, gst_structure_get_name(structure));
gst_structure_foreach(structure, print_field, (gpointer)pfx);
}
}
static void print_pad_capabilities(GstElement *element, gchar *pad_name) {
GstPad *pad = NULL;
GstCaps *caps = NULL;
/* Retrieve pad */
pad = gst_element_get_static_pad(element, pad_name);
if (!pad) {
g_printerr("Could not retrieve pad '%s'\n", pad_name);
return;
}
/* Retrieve negotiated caps (or acceptable caps if negotiation is not finished yet) */
caps = gst_pad_get_current_caps(pad);
if (!caps) caps = gst_pad_query_caps(pad, NULL);
/* Print and free */
g_print("Caps for the %s pad:\n", pad_name);
print_caps(caps, " ");
gst_caps_unref(caps);
gst_object_unref(pad);
}
/* Structure to contain all our information, so we can pass it to callbacks */
typedef struct _CustomData {
GstElement *pipeline;
GstElement *source;
GstElement *cap_filter;
GstElement *convert_video;
GstElement *dummy_plugin;
GstElement *sink_video;
GstElement *convert_audio;
GstElement *resample_audio;
GstElement *sink_audio;
} CustomData;
/* Handler for the pad-added signal */
static void pad_added_handler(GstElement *src, GstPad *pad, CustomData *data);
int main(int argc, char *argv[]) {
CustomData data;
GstBus *bus;
GstMessage *msg;
GstStateChangeReturn ret;
gboolean terminate = FALSE;
/* Initialize GStreamer */
gst_init(&argc, &argv);
/* Create the elements */
data.source = gst_element_factory_make("uridecodebin", "source");
data.cap_filter = gst_element_factory_make("capsfilter", "capsfilter");
data.convert_video = gst_element_factory_make("videoconvert", "videoconvert");
data.dummy_plugin = gst_element_factory_make("myfilter", "myfilter");
// data.sink_video = gst_element_factory_make("autovideosink", "videosink");
data.sink_video = gst_element_factory_make("fakesink", "videosink");
data.convert_audio = gst_element_factory_make("audioconvert", "audioconvert");
data.resample_audio = gst_element_factory_make("audioresample", "audioresample");
data.sink_audio = gst_element_factory_make("autoaudiosink", "audiosink");
/* Create the empty pipeline */
data.pipeline = gst_pipeline_new("test-pipeline");
if (!data.pipeline || !data.source || !data.convert_audio || !data.resample_audio || !data.sink_audio ||
!data.convert_video || !data.sink_video || !data.dummy_plugin || !data.cap_filter) {
g_printerr("Not all elements could be created.\n");
return -1;
}
/* Build the pipeline. Note that we are NOT linking the source at this
* point. We will do it later. */
gst_bin_add_many(GST_BIN(data.pipeline), data.source, data.convert_audio, data.resample_audio, data.sink_audio,
data.convert_video, data.cap_filter, data.dummy_plugin, data.sink_video, NULL);
if (!gst_element_link_many(data.convert_audio, data.resample_audio, data.sink_audio, NULL)) {
g_printerr("Elements could not be linked.\n");
gst_object_unref(data.pipeline);
return -1;
}
/*prepare caps filter*/
GstCaps *filtercaps = gst_caps_new_simple("video/x-raw", "format", G_TYPE_STRING, "BGR", NULL);
g_object_set(G_OBJECT(data.cap_filter), "caps", filtercaps, NULL);
gst_caps_unref(filtercaps);
g_object_set(G_OBJECT(data.sink_video), "sync", 1, NULL);
if (!gst_element_link_many(data.convert_video, data.cap_filter, data.dummy_plugin, data.sink_video, NULL)) {
g_printerr("Elements could not be linked.\n");
gst_object_unref(data.pipeline);
return -1;
}
// g_print("In NULL state:\n");
// print_pad_capabilities(data.convert_video, "sink");
/* Set the URI to play */
g_object_set(data.source, "uri",
"https://www.freedesktop.org/software/gstreamer-sdk/data/media/sintel_trailer-480p.webm", NULL);
/* Connect to the pad-added signal */
g_signal_connect(data.source, "pad-added", G_CALLBACK(pad_added_handler), &data);
/* Start playing */
ret = gst_element_set_state(data.pipeline, GST_STATE_PLAYING);
if (ret == GST_STATE_CHANGE_FAILURE) {
g_printerr("Unable to set the pipeline to the playing state.\n");
gst_object_unref(data.pipeline);
return -1;
}
/* Listen to the bus */
bus = gst_element_get_bus(data.pipeline);
do {
msg = gst_bus_timed_pop_filtered(bus, GST_CLOCK_TIME_NONE,
GST_MESSAGE_STATE_CHANGED | GST_MESSAGE_ERROR | GST_MESSAGE_EOS);
/* Parse message */
if (msg != NULL) {
GError *err;
gchar *debug_info;
switch (GST_MESSAGE_TYPE(msg)) {
case GST_MESSAGE_ERROR:
gst_message_parse_error(msg, &err, &debug_info);
g_printerr("Error received from element %s: %s\n", GST_OBJECT_NAME(msg->src), err->message);
g_printerr("Debugging information: %s\n", debug_info ? debug_info : "none");
g_clear_error(&err);
g_free(debug_info);
terminate = TRUE;
break;
case GST_MESSAGE_EOS:
g_print("End-Of-Stream reached.\n");
terminate = TRUE;
break;
case GST_MESSAGE_STATE_CHANGED:
/* We are only interested in state-changed messages from the pipeline */
if (GST_MESSAGE_SRC(msg) == GST_OBJECT(data.pipeline)) {
// g_print("In Playing state:\n");
// print_pad_capabilities(data.convert_video, "sink");
}
break;
default:
/* We should not reach here */
g_printerr("Unexpected message received.\n");
break;
}
gst_message_unref(msg);
}
} while (!terminate);
/* Free resources */
gst_object_unref(bus);
gst_element_set_state(data.pipeline, GST_STATE_NULL);
gst_object_unref(data.pipeline);
return 0;
}
/* This function will be called by the pad-added signal */
static void pad_added_handler(GstElement *src, GstPad *new_pad, CustomData *data) {
g_print("Received new pad '%s' from '%s':\n", GST_PAD_NAME(new_pad), GST_ELEMENT_NAME(src));
GstPad *audio_sink_pad = gst_element_get_static_pad(data->convert_audio, "sink");
GstPad *video_sink_pad = gst_element_get_static_pad(data->convert_video, "sink");
GstPadLinkReturn ret;
GstStructure *new_pad_struct = NULL;
const gchar *new_pad_type = NULL;
if (!gst_pad_is_linked(video_sink_pad)) {
GstCaps *new_pad_caps = NULL;
new_pad_caps = gst_pad_get_current_caps(new_pad);
new_pad_struct = gst_caps_get_structure(new_pad_caps, 0);
new_pad_type = gst_structure_get_name(new_pad_struct);
if (g_str_has_prefix(new_pad_type, "video/x-raw")) {
// Check the frame format
const gchar *format = gst_structure_get_string(new_pad_struct, "format");
g_print("Frame Raw format is %s\n", format);
/* Attempt the link */
ret = gst_pad_link(new_pad, video_sink_pad);
if (GST_PAD_LINK_FAILED(ret)) {
g_print("Type is '%s' but link failed.\n", new_pad_type);
} else {
g_print("Link succeeded (type '%s').\n", new_pad_type);
}
} else
g_print("It has type '%s' which is not raw audio. Ignoring.\n", new_pad_type);
if (new_pad_caps != NULL) gst_caps_unref(new_pad_caps);
} else
g_print("video convert is already linked. Ignoring.\n");
if (!gst_pad_is_linked(audio_sink_pad)) {
GstCaps *new_pad_caps = NULL;
new_pad_caps = gst_pad_get_current_caps(new_pad);
new_pad_struct = gst_caps_get_structure(new_pad_caps, 0);
new_pad_type = gst_structure_get_name(new_pad_struct);
if (g_str_has_prefix(new_pad_type, "audio/x-raw")) {
/* Attempt the link */
ret = gst_pad_link(new_pad, audio_sink_pad);
if (GST_PAD_LINK_FAILED(ret)) {
g_print("Type is '%s' but link failed.\n", new_pad_type);
} else {
g_print("Link succeeded (type '%s').\n", new_pad_type);
}
} else
g_print("It has type '%s' which is not raw audio. Ignoring.\n", new_pad_type);
if (new_pad_caps != NULL) gst_caps_unref(new_pad_caps);
} else
g_print("audio convert is already linked. Ignoring.\n");
/* Unreference the sink pad */
gst_object_unref(audio_sink_pad);
gst_object_unref(video_sink_pad);
}
static GstFlowReturn gst_my_filter_chain(GstPad *pad, GstObject *parent, GstBuffer *buf) {
GstMyFilter *filter;
filter = GST_MYFILTER(parent);
GstCaps *current_caps = gst_pad_get_current_caps(filter->sinkpad);
GstStructure *new_pad_struct = gst_caps_get_structure(current_caps, 0);
const gchar *new_pad_type = gst_structure_get_name(new_pad_struct);
if (g_str_has_prefix(new_pad_type, "video/x-raw")) {
const gchar *format = gst_structure_get_string(new_pad_struct, "format");
gint width = 0, height = 0;
if (!gst_structure_get_int(new_pad_struct, "width", &width) ||
!gst_structure_get_int(new_pad_struct, "height", &height)) {
g_print("No width/height available\n");
}
g_print("The video size of this set of capabilities is %s %dx%d\n", format, width, height);
if (width && height) {
GstMapInfo map;
gst_buffer_map(buf, &map, GST_MAP_READ);
g_print("frame size in bytes: %lu\n", map.size);
// Convert gstreamer data to OpenCV Mat
cv::Mat frame(cv::Size(width, height), CV_8UC3, (char *)map.data);
cv::imshow("OpenCV Frame", frame);
cv::waitKey(1);
gst_buffer_unmap(buf, &map);
}
}
if (!current_caps) gst_caps_unref(current_caps);
/* just push out the incoming buffer without touching it */
return gst_pad_push(filter->srcpad, buf);
}