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
12/* This is a simple program showing how to initialize the decoder in XMA mode */
13#include <stdio.h>
14#include <stdlib.h>
15#include <stdarg.h>
16#include <string.h>
17#define VPX_CODEC_DISABLE_COMPAT 1
18#include "vpx_config.h"
19#include "vpx/vpx_decoder.h"
20#include "vpx/vpx_integer.h"
21#if CONFIG_VP9_DECODER
22#include "vpx/vp8dx.h"
23#endif
24
25static char *exec_name;
26static int   verbose = 0;
27
28static const struct {
29  const char *name;
30  vpx_codec_iface_t *iface;
31} ifaces[] = {
32#if CONFIG_VP9_DECODER
33  {"vp9",  &vpx_codec_vp8_dx_algo},
34#endif
35};
36
37static void usage_exit(void) {
38  int i;
39
40  printf("Usage: %s <options>\n\n"
41         "Options:\n"
42         "\t--codec <name>\tCodec to use (default=%s)\n"
43         "\t-h <height>\tHeight of the simulated video frame, in pixels\n"
44         "\t-w <width> \tWidth of the simulated video frame, in pixels\n"
45         "\t-v         \tVerbose mode (show individual segment sizes)\n"
46         "\t--help     \tShow this message\n"
47         "\n"
48         "Included decoders:\n"
49         "\n",
50         exec_name,
51         ifaces[0].name);
52
53  for (i = 0; i < sizeof(ifaces) / sizeof(ifaces[0]); i++)
54    printf("    %-6s - %s\n",
55           ifaces[i].name,
56           vpx_codec_iface_name(ifaces[i].iface));
57
58  exit(EXIT_FAILURE);
59}
60
61static void usage_error(const char *fmt, ...) {
62  va_list ap;
63  va_start(ap, fmt);
64  vprintf(fmt, ap);
65  printf("\n");
66  usage_exit();
67}
68
69void my_mem_dtor(vpx_codec_mmap_t *mmap) {
70  if (verbose)
71    printf("freeing segment %d\n", mmap->id);
72
73  free(mmap->priv);
74}
75
76int main(int argc, char **argv) {
77  vpx_codec_ctx_t           decoder;
78  vpx_codec_iface_t        *iface = ifaces[0].iface;
79  vpx_codec_iter_t          iter;
80  vpx_codec_dec_cfg_t       cfg;
81  vpx_codec_err_t           res = VPX_CODEC_OK;
82  unsigned int            alloc_sz = 0;
83  unsigned int            w = 352;
84  unsigned int            h = 288;
85  int                     i;
86
87  exec_name = argv[0];
88
89  for (i = 1; i < argc; i++) {
90    if (!strcmp(argv[i], "--codec")) {
91      if (i + 1 < argc) {
92        int j, k = -1;
93
94        i++;
95
96        for (j = 0; j < sizeof(ifaces) / sizeof(ifaces[0]); j++)
97          if (!strcmp(ifaces[j].name, argv[i]))
98            k = j;
99
100        if (k >= 0)
101          iface = ifaces[k].iface;
102        else
103          usage_error("Error: Unrecognized argument (%s) to --codec\n",
104                      argv[i]);
105      } else
106        usage_error("Error: Option --codec requires argument.\n");
107    } else if (!strcmp(argv[i], "-v"))
108      verbose = 1;
109    else if (!strcmp(argv[i], "-h"))
110      if (i + 1 < argc) {
111        h = atoi(argv[++i]);
112      } else
113        usage_error("Error: Option -h requires argument.\n");
114    else if (!strcmp(argv[i], "-w"))
115      if (i + 1 < argc) {
116        w = atoi(argv[++i]);
117      } else
118        usage_error("Error: Option -w requires argument.\n");
119    else if (!strcmp(argv[i], "--help"))
120      usage_exit();
121    else
122      usage_error("Error: Unrecognized option %s\n\n", argv[i]);
123  }
124
125  if (argc == 1)
126    printf("Using built-in defaults. For options, rerun with --help\n\n");
127
128  /* XMA mode is not supported on all decoders! */
129  if (!(vpx_codec_get_caps(iface) & VPX_CODEC_CAP_XMA)) {
130    printf("%s does not support XMA mode!\n", vpx_codec_iface_name(iface));
131    return EXIT_FAILURE;
132  }
133
134  /* The codec knows how much memory to allocate based on the size of the
135   * encoded frames. This data can be parsed from the bitstream with
136   * vpx_codec_peek_stream_info() if a bitstream is available. Otherwise,
137   * a fixed size can be used that will be the upper limit on the frame
138   * size the decoder can decode.
139   */
140  cfg.w = w;
141  cfg.h = h;
142
143  /* Initialize the decoder in XMA mode. */
144  if (vpx_codec_dec_init(&decoder, iface, &cfg, VPX_CODEC_USE_XMA)) {
145    printf("Failed to initialize decoder in XMA mode: %s\n",
146           vpx_codec_error(&decoder));
147    return EXIT_FAILURE;
148  }
149
150  /* Iterate through the list of memory maps, allocating them with the
151   * requested alignment.
152   */
153  iter = NULL;
154
155  do {
156    vpx_codec_mmap_t  mmap;
157    unsigned int    align;
158
159    res = vpx_codec_get_mem_map(&decoder, &mmap, &iter);
160    align = mmap.align ? mmap.align - 1 : 0;
161
162    if (!res) {
163      if (verbose)
164        printf("Allocating segment %u, size %lu, align %u %s\n",
165               mmap.id, mmap.sz, mmap.align,
166               mmap.flags & VPX_CODEC_MEM_ZERO ? "(ZEROED)" : "");
167
168      if (mmap.flags & VPX_CODEC_MEM_ZERO)
169        mmap.priv = calloc(1, mmap.sz + align);
170      else
171        mmap.priv = malloc(mmap.sz + align);
172
173      mmap.base = (void *)((((uintptr_t)mmap.priv) + align) &
174                  ~(uintptr_t)align);
175      mmap.dtor = my_mem_dtor;
176      alloc_sz += mmap.sz + align;
177
178      if (vpx_codec_set_mem_map(&decoder, &mmap, 1)) {
179        printf("Failed to set mmap: %s\n", vpx_codec_error(&decoder));
180        return EXIT_FAILURE;
181      }
182    } else if (res != VPX_CODEC_LIST_END) {
183      printf("Failed to get mmap: %s\n", vpx_codec_error(&decoder));
184      return EXIT_FAILURE;
185    }
186  } while (res != VPX_CODEC_LIST_END);
187
188  printf("%s\n    %d bytes external memory required for %dx%d.\n",
189         decoder.name, alloc_sz, cfg.w, cfg.h);
190  vpx_codec_destroy(&decoder);
191  return EXIT_SUCCESS;
192
193}
194