mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-09-24 04:50:16 +00:00
36fd4d5d8a
The current get_profile just return one possible profile for the encode, which is not enough. For example, if we want to support HEVC 4:4:4 profile, the input of encode should be VYUA rather than NV12 in HEVC main profile. So the command line: gst-launch-1.0 videotestsrc num-buffers=200 ! capsfilter \ caps=video/x-raw,format=VUYA,width=800,height=600 ! vaapih265enc \ tune=low-power init-qp=30 ! fakesink can not work because vaapih265enc just report NV12 in sink caps, we need to specify the profile obviously like: gst-launch-1.0 videotestsrc num-buffers=200 ! capsfilter \ caps=video/x-raw,format=VUYA,width=800,height=600 ! vaapih265enc \ tune=low-power init-qp=30 ! capsfilter caps=video/x-h265, \ profile=main-444 ! fakesink The encode should have the ability to choose the profile based on input format automatically. If the input video format is VUYA, the main-444 profile should be auto choosed. We modify to let get_allowed_profiles of each encode sub class to return an array of all supported profiles based on downstream's allowed caps, or return NULL if no valid profiles specified by downstream. If no allowed profiles found, all profiles which belong to the current encoder's codec will be the candidates. The function gst_vaapi_encoder_get_surface_attributes collects the surface's attributes for that profile list we just get. So for this case, both NV12 and VUYA should be returned. TODO: some codec like VP9, need to implement the get_profile() function.
120 lines
4 KiB
C
120 lines
4 KiB
C
/*
|
|
* gstvaapiencode.h - VA-API video encoder
|
|
*
|
|
* Copyright (C) 2013-2014 Intel Corporation
|
|
* Author: Wind Yuan <feng.yuan@intel.com>
|
|
* Author: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public License
|
|
* as published by the Free Software Foundation; either version 2.1
|
|
* of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free
|
|
* Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
* Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#ifndef GST_VAAPIENCODE_H
|
|
#define GST_VAAPIENCODE_H
|
|
|
|
#include "gstvaapipluginbase.h"
|
|
#include <gst/vaapi/gstvaapiencoder.h>
|
|
|
|
#if USE_H264_FEI_ENCODER
|
|
#include <gst/vaapi/gstvaapisurface.h>
|
|
#include <gst/vaapi/gstvaapicodedbufferproxy.h>
|
|
#include "gstvaapifeivideometa.h"
|
|
#endif
|
|
|
|
G_BEGIN_DECLS
|
|
|
|
#define GST_TYPE_VAAPIENCODE \
|
|
(gst_vaapiencode_get_type ())
|
|
#define GST_VAAPIENCODE_CAST(obj) \
|
|
((GstVaapiEncode *)(obj))
|
|
#define GST_VAAPIENCODE(obj) \
|
|
(G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_VAAPIENCODE, GstVaapiEncode))
|
|
#define GST_VAAPIENCODE_CLASS(klass) \
|
|
(G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_VAAPIENCODE, GstVaapiEncodeClass))
|
|
#define GST_VAAPIENCODE_GET_CLASS(obj) \
|
|
(G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_VAAPIENCODE, GstVaapiEncodeClass))
|
|
#define GST_IS_VAAPIENCODE(obj) \
|
|
(G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_VAAPIENCODE))
|
|
#define GST_IS_VAAPIENCODE_CLASS(klass) \
|
|
(G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_VAAPIENCODE))
|
|
|
|
typedef struct _GstVaapiEncode GstVaapiEncode;
|
|
typedef struct _GstVaapiEncodeClass GstVaapiEncodeClass;
|
|
|
|
struct _GstVaapiEncode
|
|
{
|
|
/*< private >*/
|
|
GstVaapiPluginBase parent_instance;
|
|
|
|
GstVaapiEncoder *encoder;
|
|
GstVideoCodecState *input_state;
|
|
gboolean input_state_changed;
|
|
/* needs to be set by the subclass implementation */
|
|
gboolean need_codec_data;
|
|
GstVideoCodecState *output_state;
|
|
GPtrArray *prop_values;
|
|
GstCaps *allowed_sinkpad_caps;
|
|
};
|
|
|
|
struct _GstVaapiEncodeClass
|
|
{
|
|
/*< private >*/
|
|
GstVaapiPluginBaseClass parent_class;
|
|
|
|
guint prop_num;
|
|
gboolean (*set_config) (GstVaapiEncode * encode);
|
|
GstCaps * (*get_caps) (GstVaapiEncode * encode);
|
|
GstVaapiEncoder * (*alloc_encoder) (GstVaapiEncode * encode,
|
|
GstVaapiDisplay * display);
|
|
GstFlowReturn (*alloc_buffer) (GstVaapiEncode * encode,
|
|
GstVaapiCodedBuffer * coded_buf,
|
|
GstBuffer ** outbuf_ptr);
|
|
/* Get all possible profiles based on allowed caps */
|
|
GArray * (*get_allowed_profiles) (GstVaapiEncode * encode,
|
|
GstCaps * allowed);
|
|
|
|
#if USE_H264_FEI_ENCODER
|
|
|
|
gboolean (*load_control_data) (GstVaapiEncode *encoder,
|
|
GstVaapiFeiVideoMeta *feimeta,
|
|
GstVaapiSurfaceProxy *proxy);
|
|
|
|
GstVaapiFeiVideoMeta* (*save_stats_to_meta) (GstVaapiEncode *base_encode,
|
|
GstVaapiCodedBufferProxy *proxy);
|
|
|
|
#endif
|
|
};
|
|
|
|
GType
|
|
gst_vaapiencode_get_type (void) G_GNUC_CONST;
|
|
|
|
G_GNUC_INTERNAL
|
|
void
|
|
gst_vaapiencode_set_property_subclass (GObject * object,
|
|
guint prop_id, const GValue * value, GParamSpec * pspec);
|
|
|
|
G_GNUC_INTERNAL
|
|
void
|
|
gst_vaapiencode_get_property_subclass (GObject * object, guint prop_id,
|
|
GValue * value, GParamSpec * pspec);
|
|
|
|
G_GNUC_INTERNAL
|
|
gboolean
|
|
gst_vaapiencode_class_install_properties (GstVaapiEncodeClass * klass,
|
|
GObjectClass * encoder_class);
|
|
|
|
G_END_DECLS
|
|
|
|
#endif /* GST_VAAPIENCODE_H */
|