2013-07-29 05:34:06 +00:00
|
|
|
/*
|
|
|
|
* gstvaapiencoder_priv.h - VA encoder abstraction (private definitions)
|
|
|
|
*
|
|
|
|
* Copyright (C) 2013 Intel Corporation
|
|
|
|
*
|
|
|
|
* 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_VAAPI_ENCODER_PRIV_H
|
|
|
|
#define GST_VAAPI_ENCODER_PRIV_H
|
|
|
|
|
|
|
|
#include <gst/vaapi/gstvaapiencoder.h>
|
|
|
|
#include <gst/vaapi/gstvaapiencoder_objects.h>
|
|
|
|
#include <gst/vaapi/gstvaapicontext.h>
|
2013-12-03 15:11:46 +00:00
|
|
|
#include <gst/vaapi/gstvaapivideopool.h>
|
2013-07-29 05:34:06 +00:00
|
|
|
#include <gst/video/gstvideoutils.h>
|
|
|
|
|
|
|
|
G_BEGIN_DECLS
|
|
|
|
|
2013-12-04 16:55:18 +00:00
|
|
|
#define GST_VAAPI_ENCODER_CAST(encoder) \
|
|
|
|
((GstVaapiEncoder *)(encoder))
|
|
|
|
|
2013-07-29 05:34:06 +00:00
|
|
|
#define GST_VAAPI_ENCODER_CLASS(klass) \
|
|
|
|
((GstVaapiEncoderClass *)(klass))
|
|
|
|
|
|
|
|
#define GST_VAAPI_ENCODER_GET_CLASS(obj) \
|
|
|
|
GST_VAAPI_ENCODER_CLASS(GST_VAAPI_MINI_OBJECT_GET_CLASS(obj))
|
|
|
|
|
|
|
|
/* Get GstVaapiDisplay* */
|
|
|
|
#define GST_VAAPI_ENCODER_DISPLAY(encoder) \
|
|
|
|
(GST_VAAPI_ENCODER_CAST(encoder)->display)
|
|
|
|
|
|
|
|
/* Get VADisplay */
|
|
|
|
#define GST_VAAPI_ENCODER_VA_DISPLAY(encoder) \
|
|
|
|
(GST_VAAPI_ENCODER_CAST(encoder)->va_display)
|
|
|
|
|
|
|
|
/* Get GstVaapiContext* */
|
|
|
|
#define GST_VAAPI_ENCODER_CONTEXT(encoder) \
|
|
|
|
(GST_VAAPI_ENCODER_CAST(encoder)->context)
|
|
|
|
|
|
|
|
/* Get VAContext */
|
|
|
|
#define GST_VAAPI_ENCODER_VA_CONTEXT(encoder) \
|
|
|
|
(GST_VAAPI_ENCODER_CAST(encoder)->va_context)
|
|
|
|
|
|
|
|
#define GST_VAAPI_ENCODER_VIDEO_INFO(encoder) (GST_VAAPI_ENCODER_CAST(encoder)->video_info)
|
|
|
|
#define GST_VAAPI_ENCODER_CAPS(encoder) (GST_VAAPI_ENCODER_CAST(encoder)->caps)
|
|
|
|
#define GST_VAAPI_ENCODER_WIDTH(encoder) (GST_VAAPI_ENCODER_CAST(encoder)->video_info.width)
|
|
|
|
#define GST_VAAPI_ENCODER_HEIGHT(encoder) (GST_VAAPI_ENCODER_CAST(encoder)->video_info.height)
|
|
|
|
#define GST_VAAPI_ENCODER_FPS_N(encoder) (GST_VAAPI_ENCODER_CAST(encoder)->video_info.fps_n)
|
|
|
|
#define GST_VAAPI_ENCODER_FPS_D(encoder) (GST_VAAPI_ENCODER_CAST(encoder)->video_info.fps_d)
|
2014-01-06 14:10:36 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* GST_VAAPI_ENCODER_RATE_CONTROL:
|
|
|
|
* @encoder: a #GstVaapiEncoder
|
|
|
|
*
|
|
|
|
* Macro that evaluates to the rate control.
|
|
|
|
* This is an internal macro that does not do any run-time type check.
|
|
|
|
*/
|
|
|
|
#undef GST_VAAPI_ENCODER_RATE_CONTROL
|
|
|
|
#define GST_VAAPI_ENCODER_RATE_CONTROL(encoder) \
|
|
|
|
(GST_VAAPI_ENCODER_CAST (encoder)->rate_control)
|
2013-07-29 05:34:06 +00:00
|
|
|
|
|
|
|
#define GST_VAAPI_ENCODER_CHECK_STATUS(exp, err_num, err_reason, ...) \
|
|
|
|
if (!(exp)) { \
|
|
|
|
ret = err_num; \
|
|
|
|
GST_VAAPI_ENCODER_LOG_ERROR(err_reason, ## __VA_ARGS__); \
|
|
|
|
goto end; \
|
|
|
|
}
|
|
|
|
|
|
|
|
typedef struct _GstVaapiEncoderClass GstVaapiEncoderClass;
|
2014-01-06 14:10:36 +00:00
|
|
|
typedef struct _GstVaapiEncoderClassData GstVaapiEncoderClassData;
|
2013-07-29 05:34:06 +00:00
|
|
|
|
|
|
|
struct _GstVaapiEncoder
|
|
|
|
{
|
2014-01-03 15:57:09 +00:00
|
|
|
/*< private >*/
|
2013-07-29 05:34:06 +00:00
|
|
|
GstVaapiMiniObject parent_instance;
|
|
|
|
|
|
|
|
GstVaapiDisplay *display;
|
|
|
|
GstVaapiContext *context;
|
2013-12-04 17:48:35 +00:00
|
|
|
GstVaapiContextInfo context_info;
|
2013-07-29 05:34:06 +00:00
|
|
|
GstCaps *caps;
|
|
|
|
|
|
|
|
VADisplay va_display;
|
|
|
|
VAContextID va_context;
|
|
|
|
GstVideoInfo video_info;
|
|
|
|
GstVaapiRateControl rate_control;
|
2014-01-06 14:10:36 +00:00
|
|
|
guint32 rate_control_mask;
|
2014-01-06 17:01:33 +00:00
|
|
|
guint bitrate; /* kbps */
|
2013-07-29 05:34:06 +00:00
|
|
|
|
2013-12-04 16:05:17 +00:00
|
|
|
GMutex mutex;
|
2013-07-29 05:34:06 +00:00
|
|
|
GCond surface_free;
|
2013-12-03 15:11:46 +00:00
|
|
|
GCond codedbuf_free;
|
|
|
|
guint codedbuf_size;
|
|
|
|
GstVaapiVideoPool *codedbuf_pool;
|
2013-12-04 16:05:17 +00:00
|
|
|
GAsyncQueue *codedbuf_queue;
|
2014-01-06 14:10:36 +00:00
|
|
|
guint32 num_codedbuf_queued;
|
2013-07-29 05:34:06 +00:00
|
|
|
};
|
|
|
|
|
2014-01-06 14:10:36 +00:00
|
|
|
struct _GstVaapiEncoderClassData
|
|
|
|
{
|
|
|
|
/*< private >*/
|
|
|
|
GstVaapiCodec codec;
|
|
|
|
|
|
|
|
GstVaapiRateControl default_rate_control;
|
|
|
|
guint32 rate_control_mask;
|
|
|
|
};
|
|
|
|
|
|
|
|
#define GST_VAAPI_ENCODER_DEFINE_CLASS_DATA(CODEC) \
|
|
|
|
static const GstVaapiEncoderClassData g_class_data = { \
|
|
|
|
.codec = G_PASTE (GST_VAAPI_CODEC_, CODEC), \
|
|
|
|
.default_rate_control = DEFAULT_RATECONTROL, \
|
|
|
|
.rate_control_mask = SUPPORTED_RATECONTROLS, \
|
|
|
|
}
|
|
|
|
|
2013-07-29 05:34:06 +00:00
|
|
|
struct _GstVaapiEncoderClass
|
|
|
|
{
|
2014-01-03 15:57:09 +00:00
|
|
|
/*< private >*/
|
2013-12-04 10:54:40 +00:00
|
|
|
GstVaapiMiniObjectClass parent_class;
|
2013-07-29 05:34:06 +00:00
|
|
|
|
2014-01-06 14:10:36 +00:00
|
|
|
const GstVaapiEncoderClassData *class_data;
|
|
|
|
|
2013-07-29 05:34:06 +00:00
|
|
|
gboolean (*init) (GstVaapiEncoder * encoder);
|
2013-12-04 10:54:40 +00:00
|
|
|
void (*finalize) (GstVaapiEncoder * encoder);
|
2013-07-29 05:34:06 +00:00
|
|
|
|
|
|
|
GstCaps * (*set_format) (GstVaapiEncoder * encoder,
|
|
|
|
GstVideoCodecState * in_state,
|
|
|
|
GstCaps * ref_caps);
|
|
|
|
|
2013-12-04 17:48:35 +00:00
|
|
|
void (*set_context_info) (GstVaapiEncoder * encoder);
|
2013-07-29 05:34:06 +00:00
|
|
|
|
|
|
|
GstVaapiEncoderStatus (*reordering) (GstVaapiEncoder * encoder,
|
|
|
|
GstVideoCodecFrame * in,
|
|
|
|
GstVaapiEncPicture ** out);
|
|
|
|
GstVaapiEncoderStatus (*encode) (GstVaapiEncoder * encoder,
|
|
|
|
GstVaapiEncPicture * picture,
|
|
|
|
GstVaapiCodedBufferProxy * codedbuf);
|
|
|
|
|
|
|
|
GstVaapiEncoderStatus (*flush) (GstVaapiEncoder * encoder);
|
|
|
|
|
|
|
|
/* get_codec_data can be NULL */
|
|
|
|
GstVaapiEncoderStatus (*get_codec_data) (GstVaapiEncoder * encoder,
|
|
|
|
GstBuffer ** codec_data);
|
|
|
|
};
|
|
|
|
|
2013-12-04 10:54:40 +00:00
|
|
|
#define GST_VAAPI_ENCODER_CLASS_HOOK(codec, func) \
|
|
|
|
.func = G_PASTE (G_PASTE (G_PASTE (gst_vaapi_encoder_,codec),_), func)
|
|
|
|
|
|
|
|
#define GST_VAAPI_ENCODER_CLASS_INIT_BASE(CODEC) \
|
|
|
|
.parent_class = { \
|
|
|
|
.size = sizeof (G_PASTE (GstVaapiEncoder, CODEC)), \
|
|
|
|
.finalize = (GDestroyNotify) gst_vaapi_encoder_finalize \
|
|
|
|
}
|
2013-07-29 05:34:06 +00:00
|
|
|
|
2013-12-04 10:54:40 +00:00
|
|
|
#define GST_VAAPI_ENCODER_CLASS_INIT(CODEC, codec) \
|
|
|
|
GST_VAAPI_ENCODER_CLASS_INIT_BASE (CODEC), \
|
2014-01-06 14:10:36 +00:00
|
|
|
.class_data = &g_class_data, \
|
2013-12-04 10:54:40 +00:00
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, init), \
|
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, finalize), \
|
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, set_format), \
|
2013-12-04 17:48:35 +00:00
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, set_context_info), \
|
2013-12-04 10:54:40 +00:00
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, reordering), \
|
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, encode), \
|
|
|
|
GST_VAAPI_ENCODER_CLASS_HOOK (codec, flush)
|
|
|
|
|
|
|
|
G_GNUC_INTERNAL
|
2013-07-29 05:34:06 +00:00
|
|
|
GstVaapiEncoder *
|
|
|
|
gst_vaapi_encoder_new (const GstVaapiEncoderClass * klass,
|
|
|
|
GstVaapiDisplay * display);
|
|
|
|
|
2013-12-04 10:54:40 +00:00
|
|
|
G_GNUC_INTERNAL
|
2013-07-29 05:34:06 +00:00
|
|
|
void
|
|
|
|
gst_vaapi_encoder_finalize (GstVaapiEncoder * encoder);
|
|
|
|
|
2013-12-04 16:55:18 +00:00
|
|
|
G_GNUC_INTERNAL
|
2013-07-29 05:34:06 +00:00
|
|
|
GstVaapiSurfaceProxy *
|
|
|
|
gst_vaapi_encoder_create_surface (GstVaapiEncoder *
|
|
|
|
encoder);
|
|
|
|
|
2013-12-04 16:55:18 +00:00
|
|
|
static inline void
|
2013-07-29 05:34:06 +00:00
|
|
|
gst_vaapi_encoder_release_surface (GstVaapiEncoder * encoder,
|
2013-12-04 16:55:18 +00:00
|
|
|
GstVaapiSurfaceProxy * proxy)
|
|
|
|
{
|
|
|
|
gst_vaapi_surface_proxy_unref (proxy);
|
|
|
|
}
|
2013-07-29 05:34:06 +00:00
|
|
|
|
|
|
|
G_END_DECLS
|
|
|
|
|
|
|
|
#endif /* GST_VAAPI_ENCODER_PRIV_H */
|