gstvaapidecoder_priv.h 8.58 KB
Newer Older
1 2 3
/*
 *  gstvaapidecoder_priv.h - VA decoder abstraction (private definitions)
 *
4
 *  Copyright (C) 2010-2011 Splitted-Desktop Systems
5
 *    Author: Gwenole Beauchesne <gwenole.beauchesne@splitted-desktop.com>
6
 *  Copyright (C) 2011-2013 Intel Corporation
7
 *    Author: Gwenole Beauchesne <gwenole.beauchesne@intel.com>
8
 *
gb's avatar
gb committed
9 10 11 12
 *  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.
13
 *
gb's avatar
gb committed
14
 *  This library is distributed in the hope that it will be useful,
15
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
gb's avatar
gb committed
16 17
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *  Lesser General Public License for more details.
18
 *
gb's avatar
gb committed
19 20 21 22
 *  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
23 24 25 26 27 28 29
 */

#ifndef GST_VAAPI_DECODER_PRIV_H
#define GST_VAAPI_DECODER_PRIV_H

#include <glib.h>
#include <gst/vaapi/gstvaapidecoder.h>
30
#include <gst/vaapi/gstvaapidecoder_unit.h>
31
#include <gst/vaapi/gstvaapicontext.h>
32
#include "gstvaapiminiobject.h"
33 34 35

G_BEGIN_DECLS

36 37 38 39 40 41 42 43 44 45 46 47 48 49
#define GST_VAAPI_DECODER_CAST(decoder) \
    ((GstVaapiDecoder *)(decoder))

#define GST_VAAPI_DECODER_CLASS(klass) \
    ((GstVaapiDecoderClass *)(klass))

#define GST_VAAPI_IS_DECODER_CLASS(klass) \
    ((klass) != NULL))

#define GST_VAAPI_DECODER_GET_CLASS(obj) \
    GST_VAAPI_DECODER_CLASS(GST_VAAPI_MINI_OBJECT_GET_CLASS(obj))

typedef struct _GstVaapiDecoderClass            GstVaapiDecoderClass;
        struct _GstVaapiDecoderUnit;
50

51 52 53 54 55 56 57 58 59
/**
 * GST_VAAPI_PARSER_STATE:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstVaapiParserState of @decoder.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_PARSER_STATE
#define GST_VAAPI_PARSER_STATE(decoder) \
60
    (&GST_VAAPI_DECODER_CAST(decoder)->parser_state)
61

62 63 64 65 66 67 68 69 70
/**
 * GST_VAAPI_DECODER_DISPLAY:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstVaapiDisplay of @decoder.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_DISPLAY
#define GST_VAAPI_DECODER_DISPLAY(decoder) \
71
    GST_VAAPI_DECODER_CAST(decoder)->display
72 73 74 75 76 77 78 79 80 81

/**
 * GST_VAAPI_DECODER_CONTEXT:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstVaapiContext of @decoder.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_CONTEXT
#define GST_VAAPI_DECODER_CONTEXT(decoder) \
82
    GST_VAAPI_DECODER_CAST(decoder)->context
83 84 85 86 87 88 89 90 91 92

/**
 * GST_VAAPI_DECODER_CODEC:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstVaapiCodec of @decoder.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_CODEC
#define GST_VAAPI_DECODER_CODEC(decoder) \
93
    GST_VAAPI_DECODER_CAST(decoder)->codec
94

95 96 97 98 99 100 101 102 103 104
/**
 * GST_VAAPI_DECODER_CODEC_STATE:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstVideoCodecState holding codec state
 * for @decoder.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_CODEC_STATE
#define GST_VAAPI_DECODER_CODEC_STATE(decoder) \
105
    GST_VAAPI_DECODER_CAST(decoder)->codec_state
106

107 108 109 110 111 112 113 114 115 116
/**
 * GST_VAAPI_DECODER_CODEC_DATA:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstBuffer holding optional codec data
 * for @decoder.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_CODEC_DATA
#define GST_VAAPI_DECODER_CODEC_DATA(decoder) \
117
    GST_VAAPI_DECODER_CODEC_STATE(decoder)->codec_data
118

119 120 121 122 123 124 125 126 127 128 129 130
/**
 * GST_VAAPI_DECODER_CODEC_FRAME:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the #GstVideoCodecFrame holding decoder
 * units for the current frame.
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_CODEC_FRAME
#define GST_VAAPI_DECODER_CODEC_FRAME(decoder) \
    GST_VAAPI_PARSER_STATE(decoder)->current_frame

131 132 133 134 135 136 137 138 139
/**
 * GST_VAAPI_DECODER_WIDTH:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the coded width of the picture
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_WIDTH
#define GST_VAAPI_DECODER_WIDTH(decoder) \
140
    GST_VAAPI_DECODER_CODEC_STATE(decoder)->info.width
141 142 143 144 145 146 147 148 149 150

/**
 * GST_VAAPI_DECODER_HEIGHT:
 * @decoder: a #GstVaapiDecoder
 *
 * Macro that evaluates to the coded height of the picture
 * This is an internal macro that does not do any run-time type check.
 */
#undef  GST_VAAPI_DECODER_HEIGHT
#define GST_VAAPI_DECODER_HEIGHT(decoder) \
151
    GST_VAAPI_DECODER_CODEC_STATE(decoder)->info.height
152

gb's avatar
gb committed
153 154 155 156 157
/* End-of-Stream buffer */
#define GST_BUFFER_FLAG_EOS (GST_BUFFER_FLAG_LAST + 0)

#define GST_BUFFER_IS_EOS(buffer) \
    GST_BUFFER_FLAG_IS_SET(buffer, GST_BUFFER_FLAG_EOS)
158

159 160 161 162 163
#define GST_VAAPI_DECODER_GET_PRIVATE(obj)                      \
    (G_TYPE_INSTANCE_GET_PRIVATE((obj),                         \
                                 GST_VAAPI_TYPE_DECODER,        \
                                 GstVaapiDecoderPrivate))

164 165 166 167
typedef enum {
    GST_VAAPI_DECODER_STATUS_DROP_FRAME = -2
} GstVaapiDecoderStatusPrivate;

168 169
typedef struct _GstVaapiParserState GstVaapiParserState;
struct _GstVaapiParserState {
170 171 172
    GstVideoCodecFrame *current_frame;
    GstAdapter         *current_adapter;
    GstAdapter         *input_adapter;
173
    gint                input_offset1;
174 175 176
    gint                input_offset2;
    GstAdapter         *output_adapter;
    GstVaapiDecoderUnit next_unit;
177 178
    guint               next_unit_pending       : 1;
    guint               at_eos                  : 1;
179 180
};

181 182 183 184 185 186 187 188 189 190
/**
 * GstVaapiDecoder:
 *
 * A VA decoder base instance.
 */
struct _GstVaapiDecoder {
    /*< private >*/
    GstVaapiMiniObject  parent_instance;

    gpointer            user_data;
191
    GstVaapiDisplay    *display;
Gwenole Beauchesne's avatar
Gwenole Beauchesne committed
192
    VADisplay           va_display;
193
    GstVaapiContext    *context;
Gwenole Beauchesne's avatar
Gwenole Beauchesne committed
194
    VAContextID         va_context;
195
    GstVaapiCodec       codec;
196
    GstVideoCodecState *codec_state;
197 198
    GAsyncQueue        *buffers;
    GAsyncQueue        *frames;
199
    GstVaapiParserState parser_state;
200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225
    GstVaapiDecoderStateChangedFunc codec_state_changed_func;
    gpointer            codec_state_changed_data;
};

/**
 * GstVaapiDecoderClass:
 *
 * A VA decoder base class.
 */
struct _GstVaapiDecoderClass {
    /*< private >*/
    GstVaapiMiniObjectClass parent_class;

    gboolean              (*create)(GstVaapiDecoder *decoder);
    void                  (*destroy)(GstVaapiDecoder *decoder);
    GstVaapiDecoderStatus (*parse)(GstVaapiDecoder *decoder,
        GstAdapter *adapter, gboolean at_eos,
        struct _GstVaapiDecoderUnit *unit);
    GstVaapiDecoderStatus (*decode)(GstVaapiDecoder *decoder,
        struct _GstVaapiDecoderUnit *unit);
    GstVaapiDecoderStatus (*start_frame)(GstVaapiDecoder *decoder,
        struct _GstVaapiDecoderUnit *unit);
    GstVaapiDecoderStatus (*end_frame)(GstVaapiDecoder *decoder);
    GstVaapiDecoderStatus (*flush)(GstVaapiDecoder *decoder);
    GstVaapiDecoderStatus (*decode_codec_data)(GstVaapiDecoder *decoder,
        const guchar *buf, guint buf_size);
226 227
};

228 229 230 231 232 233 234 235 236
G_GNUC_INTERNAL
GstVaapiDecoder *
gst_vaapi_decoder_new(const GstVaapiDecoderClass *klass,
    GstVaapiDisplay *display, GstCaps *caps);

G_GNUC_INTERNAL
void
gst_vaapi_decoder_finalize(GstVaapiDecoder *decoder);

237
G_GNUC_INTERNAL
gb's avatar
gb committed
238 239 240 241 242
void
gst_vaapi_decoder_set_picture_size(
    GstVaapiDecoder    *decoder,
    guint               width,
    guint               height
243
);
gb's avatar
gb committed
244

245
G_GNUC_INTERNAL
gb's avatar
gb committed
246 247 248 249 250
void
gst_vaapi_decoder_set_framerate(
    GstVaapiDecoder    *decoder,
    guint               fps_n,
    guint               fps_d
251
);
gb's avatar
gb committed
252

253
G_GNUC_INTERNAL
gb's avatar
gb committed
254 255 256 257 258
void
gst_vaapi_decoder_set_pixel_aspect_ratio(
    GstVaapiDecoder    *decoder,
    guint               par_n,
    guint               par_d
259
);
gb's avatar
gb committed
260

261 262 263 264 265
G_GNUC_INTERNAL
void
gst_vaapi_decoder_set_interlace_mode(GstVaapiDecoder *decoder,
    GstVideoInterlaceMode mode);

266
G_GNUC_INTERNAL
267
void
268
gst_vaapi_decoder_set_interlaced(GstVaapiDecoder *decoder, gboolean interlaced);
269

270
G_GNUC_INTERNAL
271 272
gboolean
gst_vaapi_decoder_ensure_context(
273 274
    GstVaapiDecoder     *decoder,
    GstVaapiContextInfo *cip
275
);
276

277
G_GNUC_INTERNAL
278
void
279 280
gst_vaapi_decoder_push_frame(GstVaapiDecoder *decoder,
    GstVideoCodecFrame *frame);
281

282
G_GNUC_INTERNAL
283
GstVaapiDecoderStatus
284
gst_vaapi_decoder_check_status(GstVaapiDecoder *decoder);
285

286 287 288 289
G_GNUC_INTERNAL
GstVaapiDecoderStatus
gst_vaapi_decoder_decode_codec_data(GstVaapiDecoder *decoder);

290 291 292
G_END_DECLS

#endif /* GST_VAAPI_DECODER_PRIV_H */