1 /*
2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 /*!\file
12  * \brief Provides the high level interface to wrap decoder algorithms.
13  *
14  */
15 #include <string.h>
16 #include "vpx/internal/vpx_codec_internal.h"
17 
18 #define SAVE_STATUS(ctx, var) (ctx ? (ctx->err = var) : var)
19 
get_alg_priv(vpx_codec_ctx_t * ctx)20 static vpx_codec_alg_priv_t *get_alg_priv(vpx_codec_ctx_t *ctx) {
21   return (vpx_codec_alg_priv_t *)ctx->priv;
22 }
23 
vpx_codec_dec_init_ver(vpx_codec_ctx_t * ctx,vpx_codec_iface_t * iface,const vpx_codec_dec_cfg_t * cfg,vpx_codec_flags_t flags,int ver)24 vpx_codec_err_t vpx_codec_dec_init_ver(vpx_codec_ctx_t *ctx,
25                                        vpx_codec_iface_t *iface,
26                                        const vpx_codec_dec_cfg_t *cfg,
27                                        vpx_codec_flags_t flags, int ver) {
28   vpx_codec_err_t res;
29 
30   if (ver != VPX_DECODER_ABI_VERSION)
31     res = VPX_CODEC_ABI_MISMATCH;
32   else if (!ctx || !iface)
33     res = VPX_CODEC_INVALID_PARAM;
34   else if (iface->abi_version != VPX_CODEC_INTERNAL_ABI_VERSION)
35     res = VPX_CODEC_ABI_MISMATCH;
36   else if ((flags & VPX_CODEC_USE_POSTPROC) &&
37            !(iface->caps & VPX_CODEC_CAP_POSTPROC))
38     res = VPX_CODEC_INCAPABLE;
39   else if ((flags & VPX_CODEC_USE_ERROR_CONCEALMENT) &&
40            !(iface->caps & VPX_CODEC_CAP_ERROR_CONCEALMENT))
41     res = VPX_CODEC_INCAPABLE;
42   else if ((flags & VPX_CODEC_USE_INPUT_FRAGMENTS) &&
43            !(iface->caps & VPX_CODEC_CAP_INPUT_FRAGMENTS))
44     res = VPX_CODEC_INCAPABLE;
45   else if (!(iface->caps & VPX_CODEC_CAP_DECODER))
46     res = VPX_CODEC_INCAPABLE;
47   else {
48     memset(ctx, 0, sizeof(*ctx));
49     ctx->iface = iface;
50     ctx->name = iface->name;
51     ctx->priv = NULL;
52     ctx->init_flags = flags;
53     ctx->config.dec = cfg;
54 
55     res = ctx->iface->init(ctx, NULL);
56     if (res) {
57       ctx->err_detail = ctx->priv ? ctx->priv->err_detail : NULL;
58       vpx_codec_destroy(ctx);
59     }
60   }
61 
62   return SAVE_STATUS(ctx, res);
63 }
64 
vpx_codec_peek_stream_info(vpx_codec_iface_t * iface,const uint8_t * data,unsigned int data_sz,vpx_codec_stream_info_t * si)65 vpx_codec_err_t vpx_codec_peek_stream_info(vpx_codec_iface_t *iface,
66                                            const uint8_t *data,
67                                            unsigned int data_sz,
68                                            vpx_codec_stream_info_t *si) {
69   vpx_codec_err_t res;
70 
71   if (!iface || !data || !data_sz || !si ||
72       si->sz < sizeof(vpx_codec_stream_info_t))
73     res = VPX_CODEC_INVALID_PARAM;
74   else {
75     /* Set default/unknown values */
76     si->w = 0;
77     si->h = 0;
78 
79     res = iface->dec.peek_si(data, data_sz, si);
80   }
81 
82   return res;
83 }
84 
vpx_codec_get_stream_info(vpx_codec_ctx_t * ctx,vpx_codec_stream_info_t * si)85 vpx_codec_err_t vpx_codec_get_stream_info(vpx_codec_ctx_t *ctx,
86                                           vpx_codec_stream_info_t *si) {
87   vpx_codec_err_t res;
88 
89   if (!ctx || !si || si->sz < sizeof(vpx_codec_stream_info_t))
90     res = VPX_CODEC_INVALID_PARAM;
91   else if (!ctx->iface || !ctx->priv)
92     res = VPX_CODEC_ERROR;
93   else {
94     /* Set default/unknown values */
95     si->w = 0;
96     si->h = 0;
97 
98     res = ctx->iface->dec.get_si(get_alg_priv(ctx), si);
99   }
100 
101   return SAVE_STATUS(ctx, res);
102 }
103 
vpx_codec_decode(vpx_codec_ctx_t * ctx,const uint8_t * data,unsigned int data_sz,void * user_priv,long deadline)104 vpx_codec_err_t vpx_codec_decode(vpx_codec_ctx_t *ctx, const uint8_t *data,
105                                  unsigned int data_sz, void *user_priv,
106                                  long deadline) {
107   vpx_codec_err_t res;
108 
109   /* Sanity checks */
110   /* NULL data ptr allowed if data_sz is 0 too */
111   if (!ctx || (!data && data_sz) || (data && !data_sz))
112     res = VPX_CODEC_INVALID_PARAM;
113   else if (!ctx->iface || !ctx->priv)
114     res = VPX_CODEC_ERROR;
115   else {
116     res = ctx->iface->dec.decode(get_alg_priv(ctx), data, data_sz, user_priv,
117                                  deadline);
118   }
119 
120   return SAVE_STATUS(ctx, res);
121 }
122 
vpx_codec_get_frame(vpx_codec_ctx_t * ctx,vpx_codec_iter_t * iter)123 vpx_image_t *vpx_codec_get_frame(vpx_codec_ctx_t *ctx, vpx_codec_iter_t *iter) {
124   vpx_image_t *img;
125 
126   if (!ctx || !iter || !ctx->iface || !ctx->priv)
127     img = NULL;
128   else
129     img = ctx->iface->dec.get_frame(get_alg_priv(ctx), iter);
130 
131   return img;
132 }
133 
vpx_codec_register_put_frame_cb(vpx_codec_ctx_t * ctx,vpx_codec_put_frame_cb_fn_t cb,void * user_priv)134 vpx_codec_err_t vpx_codec_register_put_frame_cb(vpx_codec_ctx_t *ctx,
135                                                 vpx_codec_put_frame_cb_fn_t cb,
136                                                 void *user_priv) {
137   vpx_codec_err_t res;
138 
139   if (!ctx || !cb)
140     res = VPX_CODEC_INVALID_PARAM;
141   else if (!ctx->iface || !ctx->priv ||
142            !(ctx->iface->caps & VPX_CODEC_CAP_PUT_FRAME))
143     res = VPX_CODEC_ERROR;
144   else {
145     ctx->priv->dec.put_frame_cb.u.put_frame = cb;
146     ctx->priv->dec.put_frame_cb.user_priv = user_priv;
147     res = VPX_CODEC_OK;
148   }
149 
150   return SAVE_STATUS(ctx, res);
151 }
152 
vpx_codec_register_put_slice_cb(vpx_codec_ctx_t * ctx,vpx_codec_put_slice_cb_fn_t cb,void * user_priv)153 vpx_codec_err_t vpx_codec_register_put_slice_cb(vpx_codec_ctx_t *ctx,
154                                                 vpx_codec_put_slice_cb_fn_t cb,
155                                                 void *user_priv) {
156   vpx_codec_err_t res;
157 
158   if (!ctx || !cb)
159     res = VPX_CODEC_INVALID_PARAM;
160   else if (!ctx->iface || !ctx->priv ||
161            !(ctx->iface->caps & VPX_CODEC_CAP_PUT_SLICE))
162     res = VPX_CODEC_ERROR;
163   else {
164     ctx->priv->dec.put_slice_cb.u.put_slice = cb;
165     ctx->priv->dec.put_slice_cb.user_priv = user_priv;
166     res = VPX_CODEC_OK;
167   }
168 
169   return SAVE_STATUS(ctx, res);
170 }
171 
vpx_codec_set_frame_buffer_functions(vpx_codec_ctx_t * ctx,vpx_get_frame_buffer_cb_fn_t cb_get,vpx_release_frame_buffer_cb_fn_t cb_release,void * cb_priv)172 vpx_codec_err_t vpx_codec_set_frame_buffer_functions(
173     vpx_codec_ctx_t *ctx, vpx_get_frame_buffer_cb_fn_t cb_get,
174     vpx_release_frame_buffer_cb_fn_t cb_release, void *cb_priv) {
175   vpx_codec_err_t res;
176 
177   if (!ctx || !cb_get || !cb_release) {
178     res = VPX_CODEC_INVALID_PARAM;
179   } else if (!ctx->iface || !ctx->priv ||
180              !(ctx->iface->caps & VPX_CODEC_CAP_EXTERNAL_FRAME_BUFFER)) {
181     res = VPX_CODEC_ERROR;
182   } else {
183     res = ctx->iface->dec.set_fb_fn(get_alg_priv(ctx), cb_get, cb_release,
184                                     cb_priv);
185   }
186 
187   return SAVE_STATUS(ctx, res);
188 }
189