OLD | NEW |
| (Empty) |
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 | |
25 static char *exec_name; | |
26 static int verbose = 0; | |
27 | |
28 static 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 | |
37 static 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 | |
61 static 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 | |
69 void 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 | |
76 int 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 } | |
OLD | NEW |