<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> <html xmlns="http://www.w3.org/1999/xhtml"> <head> <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> <title>WebM VP8 Codec SDK: vpxenc</title> <link href="tabs.css" rel="stylesheet" type="text/css"/> <link href="doxygen.css" rel="stylesheet" type="text/css"/> </head> <body> <!-- Generated by Doxygen 1.7.3 --> <div id="top"> <div id="titlearea"> <table cellspacing="0" cellpadding="0"> <tbody> <tr style="height: 56px;"> <td style="padding-left: 0.5em;"> <div id="projectname">WebM VP8 Codec SDK</div> </td> </tr> </tbody> </table> </div> <div id="navrow1" class="tabs"> <ul class="tablist"> <li><a href="index.html"><span>Main Page</span></a></li> <li class="current"><a href="pages.html"><span>Related Pages</span></a></li> <li><a href="modules.html"><span>Modules</span></a></li> <li><a href="annotated.html"><span>Data Structures</span></a></li> <li><a href="files.html"><span>Files</span></a></li> </ul> </div> <div id="nav-path" class="navpath"> <ul> <li class="navelem"><a class="el" href="samples.html">Sample Code</a> </li> </ul> </div> </div> <div class="header"> <div class="headertitle"> <h1>vpxenc </h1> </div> </div> <div class="contents"> <div class="textblock"><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/*</span> <a name="l00002"></a>00002 <span class="comment"> * Copyright (c) 2010 The WebM project authors. All Rights Reserved.</span> <a name="l00003"></a>00003 <span class="comment"> *</span> <a name="l00004"></a>00004 <span class="comment"> * Use of this source code is governed by a BSD-style license</span> <a name="l00005"></a>00005 <span class="comment"> * that can be found in the LICENSE file in the root of the source</span> <a name="l00006"></a>00006 <span class="comment"> * tree. An additional intellectual property rights grant can be found</span> <a name="l00007"></a>00007 <span class="comment"> * in the file PATENTS. All contributing project authors may</span> <a name="l00008"></a>00008 <span class="comment"> * be found in the AUTHORS file in the root of the source tree.</span> <a name="l00009"></a>00009 <span class="comment"> */</span> <a name="l00010"></a>00010 <a name="l00011"></a>00011 <a name="l00012"></a>00012 <span class="comment">/* This is a simple program that encodes YV12 files and generates ivf</span> <a name="l00013"></a>00013 <span class="comment"> * files using the new interface.</span> <a name="l00014"></a>00014 <span class="comment"> */</span> <a name="l00015"></a>00015 <span class="preprocessor">#if defined(_WIN32) || !CONFIG_OS_SUPPORT</span> <a name="l00016"></a>00016 <span class="preprocessor"></span><span class="preprocessor">#define USE_POSIX_MMAP 0</span> <a name="l00017"></a>00017 <span class="preprocessor"></span><span class="preprocessor">#else</span> <a name="l00018"></a>00018 <span class="preprocessor"></span><span class="preprocessor">#define USE_POSIX_MMAP 1</span> <a name="l00019"></a>00019 <span class="preprocessor"></span><span class="preprocessor">#endif</span> <a name="l00020"></a>00020 <span class="preprocessor"></span> <a name="l00021"></a>00021 <span class="preprocessor">#include <stdio.h></span> <a name="l00022"></a>00022 <span class="preprocessor">#include <stdlib.h></span> <a name="l00023"></a>00023 <span class="preprocessor">#include <stdarg.h></span> <a name="l00024"></a>00024 <span class="preprocessor">#include <string.h></span> <a name="l00025"></a>00025 <span class="preprocessor">#include <limits.h></span> <a name="l00026"></a>00026 <span class="preprocessor">#include "<a class="code" href="vpx__encoder_8h.html" title="Describes the encoder algorithm interface to applications.">vpx/vpx_encoder.h</a>"</span> <a name="l00027"></a>00027 <span class="preprocessor">#if USE_POSIX_MMAP</span> <a name="l00028"></a>00028 <span class="preprocessor"></span><span class="preprocessor">#include <sys/types.h></span> <a name="l00029"></a>00029 <span class="preprocessor">#include <sys/stat.h></span> <a name="l00030"></a>00030 <span class="preprocessor">#include <sys/mman.h></span> <a name="l00031"></a>00031 <span class="preprocessor">#include <fcntl.h></span> <a name="l00032"></a>00032 <span class="preprocessor">#include <unistd.h></span> <a name="l00033"></a>00033 <span class="preprocessor">#endif</span> <a name="l00034"></a>00034 <span class="preprocessor"></span><span class="preprocessor">#include "vpx_version.h"</span> <a name="l00035"></a>00035 <span class="preprocessor">#include "<a class="code" href="vp8cx_8h.html" title="Provides definitions for using the VP8 encoder algorithm within the vpx Codec Interface.">vpx/vp8cx.h</a>"</span> <a name="l00036"></a>00036 <span class="preprocessor">#include "vpx_ports/mem_ops.h"</span> <a name="l00037"></a>00037 <span class="preprocessor">#include "vpx_ports/vpx_timer.h"</span> <a name="l00038"></a>00038 <span class="preprocessor">#include "tools_common.h"</span> <a name="l00039"></a>00039 <span class="preprocessor">#include "y4minput.h"</span> <a name="l00040"></a>00040 <span class="preprocessor">#include "libmkv/EbmlWriter.h"</span> <a name="l00041"></a>00041 <span class="preprocessor">#include "libmkv/EbmlIDs.h"</span> <a name="l00042"></a>00042 <a name="l00043"></a>00043 <span class="comment">/* Need special handling of these functions on Windows */</span> <a name="l00044"></a>00044 <span class="preprocessor">#if defined(_MSC_VER)</span> <a name="l00045"></a>00045 <span class="preprocessor"></span><span class="comment">/* MSVS doesn't define off_t, and uses _f{seek,tell}i64 */</span> <a name="l00046"></a>00046 <span class="keyword">typedef</span> __int64 off_t; <a name="l00047"></a>00047 <span class="preprocessor">#define fseeko _fseeki64</span> <a name="l00048"></a>00048 <span class="preprocessor"></span><span class="preprocessor">#define ftello _ftelli64</span> <a name="l00049"></a>00049 <span class="preprocessor"></span><span class="preprocessor">#elif defined(_WIN32)</span> <a name="l00050"></a>00050 <span class="preprocessor"></span><span class="comment">/* MinGW defines off_t, and uses f{seek,tell}o64 */</span> <a name="l00051"></a>00051 <span class="preprocessor">#define fseeko fseeko64</span> <a name="l00052"></a>00052 <span class="preprocessor"></span><span class="preprocessor">#define ftello ftello64</span> <a name="l00053"></a>00053 <span class="preprocessor"></span><span class="preprocessor">#endif</span> <a name="l00054"></a>00054 <span class="preprocessor"></span> <a name="l00055"></a>00055 <span class="preprocessor">#if defined(_MSC_VER)</span> <a name="l00056"></a>00056 <span class="preprocessor"></span><span class="preprocessor">#define LITERALU64(n) n</span> <a name="l00057"></a>00057 <span class="preprocessor"></span><span class="preprocessor">#else</span> <a name="l00058"></a>00058 <span class="preprocessor"></span><span class="preprocessor">#define LITERALU64(n) n##LLU</span> <a name="l00059"></a>00059 <span class="preprocessor"></span><span class="preprocessor">#endif</span> <a name="l00060"></a>00060 <span class="preprocessor"></span> <a name="l00061"></a>00061 <span class="comment">/* We should use 32-bit file operations in WebM file format</span> <a name="l00062"></a>00062 <span class="comment"> * when building ARM executable file (.axf) with RVCT */</span> <a name="l00063"></a>00063 <span class="preprocessor">#if !CONFIG_OS_SUPPORT</span> <a name="l00064"></a>00064 <span class="preprocessor"></span><span class="keyword">typedef</span> <span class="keywordtype">long</span> off_t; <a name="l00065"></a>00065 <span class="preprocessor">#define fseeko fseek</span> <a name="l00066"></a>00066 <span class="preprocessor"></span><span class="preprocessor">#define ftello ftell</span> <a name="l00067"></a>00067 <span class="preprocessor"></span><span class="preprocessor">#endif</span> <a name="l00068"></a>00068 <span class="preprocessor"></span> <a name="l00069"></a>00069 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">char</span> *exec_name; <a name="l00070"></a>00070 <a name="l00071"></a>00071 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keyword">struct </span>codec_item <a name="l00072"></a>00072 { <a name="l00073"></a>00073 <span class="keywordtype">char</span> <span class="keyword">const</span> *name; <a name="l00074"></a>00074 <span class="keyword">const</span> <a class="code" href="group__codec.html#gad654f3da60151f5dfef70aca00ef1e9e" title="Codec interface structure.">vpx_codec_iface_t</a> *iface; <a name="l00075"></a>00075 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> fourcc; <a name="l00076"></a>00076 } codecs[] = <a name="l00077"></a>00077 { <a name="l00078"></a>00078 <span class="preprocessor">#if CONFIG_VP8_ENCODER</span> <a name="l00079"></a>00079 <span class="preprocessor"></span> {<span class="stringliteral">"vp8"</span>, &vpx_codec_vp8_cx_algo, 0x30385056}, <a name="l00080"></a>00080 <span class="preprocessor">#endif</span> <a name="l00081"></a>00081 <span class="preprocessor"></span>}; <a name="l00082"></a>00082 <a name="l00083"></a>00083 <span class="keyword">static</span> <span class="keywordtype">void</span> usage_exit(); <a name="l00084"></a>00084 <a name="l00085"></a>00085 <span class="keywordtype">void</span> die(<span class="keyword">const</span> <span class="keywordtype">char</span> *fmt, ...) <a name="l00086"></a>00086 { <a name="l00087"></a>00087 va_list ap; <a name="l00088"></a>00088 va_start(ap, fmt); <a name="l00089"></a>00089 vfprintf(stderr, fmt, ap); <a name="l00090"></a>00090 fprintf(stderr, <span class="stringliteral">"\n"</span>); <a name="l00091"></a>00091 usage_exit(); <a name="l00092"></a>00092 } <a name="l00093"></a>00093 <a name="l00094"></a>00094 <span class="keyword">static</span> <span class="keywordtype">void</span> ctx_exit_on_error(<a class="code" href="structvpx__codec__ctx.html" title="Codec context structure.">vpx_codec_ctx_t</a> *ctx, <span class="keyword">const</span> <span class="keywordtype">char</span> *s) <a name="l00095"></a>00095 { <a name="l00096"></a>00096 <span class="keywordflow">if</span> (ctx-><a class="code" href="structvpx__codec__ctx.html#a6f448802b0675013fd8c5179675c30de">err</a>) <a name="l00097"></a>00097 { <a name="l00098"></a>00098 <span class="keyword">const</span> <span class="keywordtype">char</span> *detail = <a class="code" href="group__codec.html#ga29273cb552ed1a437fe263c4a0a54300" title="Retrieve detailed error information for codec context.">vpx_codec_error_detail</a>(ctx); <a name="l00099"></a>00099 <a name="l00100"></a>00100 fprintf(stderr, <span class="stringliteral">"%s: %s\n"</span>, s, <a class="code" href="group__codec.html#ga4d265df00d42b36a4f0e3eb83fc22c5e" title="Retrieve error synopsis for codec context.">vpx_codec_error</a>(ctx)); <a name="l00101"></a>00101 <a name="l00102"></a>00102 <span class="keywordflow">if</span> (detail) <a name="l00103"></a>00103 fprintf(stderr, <span class="stringliteral">" %s\n"</span>, detail); <a name="l00104"></a>00104 <a name="l00105"></a>00105 exit(EXIT_FAILURE); <a name="l00106"></a>00106 } <a name="l00107"></a>00107 } <a name="l00108"></a>00108 <a name="l00109"></a>00109 <span class="comment">/* This structure is used to abstract the different ways of handling</span> <a name="l00110"></a>00110 <span class="comment"> * first pass statistics.</span> <a name="l00111"></a>00111 <span class="comment"> */</span> <a name="l00112"></a>00112 <span class="keyword">typedef</span> <span class="keyword">struct</span> <a name="l00113"></a>00113 { <a name="l00114"></a>00114 <a class="code" href="structvpx__fixed__buf.html" title="Generic fixed size buffer structure.">vpx_fixed_buf_t</a> buf; <a name="l00115"></a>00115 <span class="keywordtype">int</span> pass; <a name="l00116"></a>00116 FILE *file; <a name="l00117"></a>00117 <span class="keywordtype">char</span> *buf_ptr; <a name="l00118"></a>00118 <span class="keywordtype">size_t</span> buf_alloc_sz; <a name="l00119"></a>00119 } stats_io_t; <a name="l00120"></a>00120 <a name="l00121"></a>00121 <span class="keywordtype">int</span> stats_open_file(stats_io_t *stats, <span class="keyword">const</span> <span class="keywordtype">char</span> *fpf, <span class="keywordtype">int</span> pass) <a name="l00122"></a>00122 { <a name="l00123"></a>00123 <span class="keywordtype">int</span> res; <a name="l00124"></a>00124 <a name="l00125"></a>00125 stats->pass = pass; <a name="l00126"></a>00126 <a name="l00127"></a>00127 <span class="keywordflow">if</span> (pass == 0) <a name="l00128"></a>00128 { <a name="l00129"></a>00129 stats->file = fopen(fpf, <span class="stringliteral">"wb"</span>); <a name="l00130"></a>00130 stats->buf.sz = 0; <a name="l00131"></a>00131 stats->buf.buf = NULL, <a name="l00132"></a>00132 res = (stats->file != NULL); <a name="l00133"></a>00133 } <a name="l00134"></a>00134 <span class="keywordflow">else</span> <a name="l00135"></a>00135 { <a name="l00136"></a>00136 <span class="preprocessor">#if 0</span> <a name="l00137"></a>00137 <span class="preprocessor"></span><span class="preprocessor">#elif USE_POSIX_MMAP</span> <a name="l00138"></a>00138 <span class="preprocessor"></span> <span class="keyword">struct </span>stat stat_buf; <a name="l00139"></a>00139 <span class="keywordtype">int</span> fd; <a name="l00140"></a>00140 <a name="l00141"></a>00141 fd = open(fpf, O_RDONLY); <a name="l00142"></a>00142 stats->file = fdopen(fd, <span class="stringliteral">"rb"</span>); <a name="l00143"></a>00143 fstat(fd, &stat_buf); <a name="l00144"></a>00144 stats->buf.sz = stat_buf.st_size; <a name="l00145"></a>00145 stats->buf.buf = mmap(NULL, stats->buf.sz, PROT_READ, MAP_PRIVATE, <a name="l00146"></a>00146 fd, 0); <a name="l00147"></a>00147 res = (stats->buf.buf != NULL); <a name="l00148"></a>00148 <span class="preprocessor">#else</span> <a name="l00149"></a>00149 <span class="preprocessor"></span> <span class="keywordtype">size_t</span> nbytes; <a name="l00150"></a>00150 <a name="l00151"></a>00151 stats->file = fopen(fpf, <span class="stringliteral">"rb"</span>); <a name="l00152"></a>00152 <a name="l00153"></a>00153 <span class="keywordflow">if</span> (fseek(stats->file, 0, SEEK_END)) <a name="l00154"></a>00154 { <a name="l00155"></a>00155 fprintf(stderr, <span class="stringliteral">"First-pass stats file must be seekable!\n"</span>); <a name="l00156"></a>00156 exit(EXIT_FAILURE); <a name="l00157"></a>00157 } <a name="l00158"></a>00158 <a name="l00159"></a>00159 stats->buf.sz = stats->buf_alloc_sz = ftell(stats->file); <a name="l00160"></a>00160 rewind(stats->file); <a name="l00161"></a>00161 <a name="l00162"></a>00162 stats->buf.buf = malloc(stats->buf_alloc_sz); <a name="l00163"></a>00163 <a name="l00164"></a>00164 <span class="keywordflow">if</span> (!stats->buf.buf) <a name="l00165"></a>00165 { <a name="l00166"></a>00166 fprintf(stderr, <span class="stringliteral">"Failed to allocate first-pass stats buffer (%d bytes)\n"</span>, <a name="l00167"></a>00167 stats->buf_alloc_sz); <a name="l00168"></a>00168 exit(EXIT_FAILURE); <a name="l00169"></a>00169 } <a name="l00170"></a>00170 <a name="l00171"></a>00171 nbytes = fread(stats->buf.buf, 1, stats->buf.sz, stats->file); <a name="l00172"></a>00172 res = (nbytes == stats->buf.sz); <a name="l00173"></a>00173 <span class="preprocessor">#endif</span> <a name="l00174"></a>00174 <span class="preprocessor"></span> } <a name="l00175"></a>00175 <a name="l00176"></a>00176 <span class="keywordflow">return</span> res; <a name="l00177"></a>00177 } <a name="l00178"></a>00178 <a name="l00179"></a>00179 <span class="keywordtype">int</span> stats_open_mem(stats_io_t *stats, <span class="keywordtype">int</span> pass) <a name="l00180"></a>00180 { <a name="l00181"></a>00181 <span class="keywordtype">int</span> res; <a name="l00182"></a>00182 stats->pass = pass; <a name="l00183"></a>00183 <a name="l00184"></a>00184 <span class="keywordflow">if</span> (!pass) <a name="l00185"></a>00185 { <a name="l00186"></a>00186 stats->buf.sz = 0; <a name="l00187"></a>00187 stats->buf_alloc_sz = 64 * 1024; <a name="l00188"></a>00188 stats->buf.buf = malloc(stats->buf_alloc_sz); <a name="l00189"></a>00189 } <a name="l00190"></a>00190 <a name="l00191"></a>00191 stats->buf_ptr = stats->buf.buf; <a name="l00192"></a>00192 res = (stats->buf.buf != NULL); <a name="l00193"></a>00193 <span class="keywordflow">return</span> res; <a name="l00194"></a>00194 } <a name="l00195"></a>00195 <a name="l00196"></a>00196 <a name="l00197"></a>00197 <span class="keywordtype">void</span> stats_close(stats_io_t *stats, <span class="keywordtype">int</span> last_pass) <a name="l00198"></a>00198 { <a name="l00199"></a>00199 <span class="keywordflow">if</span> (stats->file) <a name="l00200"></a>00200 { <a name="l00201"></a>00201 <span class="keywordflow">if</span> (stats->pass == last_pass) <a name="l00202"></a>00202 { <a name="l00203"></a>00203 <span class="preprocessor">#if 0</span> <a name="l00204"></a>00204 <span class="preprocessor"></span><span class="preprocessor">#elif USE_POSIX_MMAP</span> <a name="l00205"></a>00205 <span class="preprocessor"></span> munmap(stats->buf.buf, stats->buf.sz); <a name="l00206"></a>00206 <span class="preprocessor">#else</span> <a name="l00207"></a>00207 <span class="preprocessor"></span> free(stats->buf.buf); <a name="l00208"></a>00208 <span class="preprocessor">#endif</span> <a name="l00209"></a>00209 <span class="preprocessor"></span> } <a name="l00210"></a>00210 <a name="l00211"></a>00211 fclose(stats->file); <a name="l00212"></a>00212 stats->file = NULL; <a name="l00213"></a>00213 } <a name="l00214"></a>00214 <span class="keywordflow">else</span> <a name="l00215"></a>00215 { <a name="l00216"></a>00216 <span class="keywordflow">if</span> (stats->pass == last_pass) <a name="l00217"></a>00217 free(stats->buf.buf); <a name="l00218"></a>00218 } <a name="l00219"></a>00219 } <a name="l00220"></a>00220 <a name="l00221"></a>00221 <span class="keywordtype">void</span> stats_write(stats_io_t *stats, <span class="keyword">const</span> <span class="keywordtype">void</span> *pkt, <span class="keywordtype">size_t</span> len) <a name="l00222"></a>00222 { <a name="l00223"></a>00223 <span class="keywordflow">if</span> (stats->file) <a name="l00224"></a>00224 { <a name="l00225"></a>00225 <span class="keywordflow">if</span>(fwrite(pkt, 1, len, stats->file)); <a name="l00226"></a>00226 } <a name="l00227"></a>00227 <span class="keywordflow">else</span> <a name="l00228"></a>00228 { <a name="l00229"></a>00229 <span class="keywordflow">if</span> (stats->buf.sz + len > stats->buf_alloc_sz) <a name="l00230"></a>00230 { <a name="l00231"></a>00231 <span class="keywordtype">size_t</span> new_sz = stats->buf_alloc_sz + 64 * 1024; <a name="l00232"></a>00232 <span class="keywordtype">char</span> *new_ptr = realloc(stats->buf.buf, new_sz); <a name="l00233"></a>00233 <a name="l00234"></a>00234 <span class="keywordflow">if</span> (new_ptr) <a name="l00235"></a>00235 { <a name="l00236"></a>00236 stats->buf_ptr = new_ptr + (stats->buf_ptr - (<span class="keywordtype">char</span> *)stats->buf.buf); <a name="l00237"></a>00237 stats->buf.buf = new_ptr; <a name="l00238"></a>00238 stats->buf_alloc_sz = new_sz; <a name="l00239"></a>00239 } <span class="comment">/* else ... */</span> <a name="l00240"></a>00240 } <a name="l00241"></a>00241 <a name="l00242"></a>00242 memcpy(stats->buf_ptr, pkt, len); <a name="l00243"></a>00243 stats->buf.sz += len; <a name="l00244"></a>00244 stats->buf_ptr += len; <a name="l00245"></a>00245 } <a name="l00246"></a>00246 } <a name="l00247"></a>00247 <a name="l00248"></a>00248 <a class="code" href="structvpx__fixed__buf.html" title="Generic fixed size buffer structure.">vpx_fixed_buf_t</a> stats_get(stats_io_t *stats) <a name="l00249"></a>00249 { <a name="l00250"></a>00250 <span class="keywordflow">return</span> stats-><a class="code" href="structvpx__fixed__buf.html#ac93e43ae3fcc1023dca86d37016ae3f0">buf</a>; <a name="l00251"></a>00251 } <a name="l00252"></a>00252 <a name="l00253"></a>00253 <span class="keyword">enum</span> video_file_type <a name="l00254"></a>00254 { <a name="l00255"></a>00255 FILE_TYPE_RAW, <a name="l00256"></a>00256 FILE_TYPE_IVF, <a name="l00257"></a>00257 FILE_TYPE_Y4M <a name="l00258"></a>00258 }; <a name="l00259"></a>00259 <a name="l00260"></a>00260 <span class="keyword">struct </span>detect_buffer { <a name="l00261"></a>00261 <span class="keywordtype">char</span> buf[4]; <a name="l00262"></a>00262 <span class="keywordtype">size_t</span> buf_read; <a name="l00263"></a>00263 <span class="keywordtype">size_t</span> position; <a name="l00264"></a>00264 }; <a name="l00265"></a>00265 <a name="l00266"></a>00266 <a name="l00267"></a>00267 <span class="preprocessor">#define IVF_FRAME_HDR_SZ (4+8) </span><span class="comment">/* 4 byte size + 8 byte timestamp */</span> <a name="l00268"></a>00268 <span class="keyword">static</span> <span class="keywordtype">int</span> read_frame(FILE *f, <a class="code" href="structvpx__image.html" title="Image Descriptor.">vpx_image_t</a> *img, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> file_type, <a name="l00269"></a>00269 y4m_input *y4m, <span class="keyword">struct</span> detect_buffer *detect) <a name="l00270"></a>00270 { <a name="l00271"></a>00271 <span class="keywordtype">int</span> plane = 0; <a name="l00272"></a>00272 <span class="keywordtype">int</span> shortread = 0; <a name="l00273"></a>00273 <a name="l00274"></a>00274 <span class="keywordflow">if</span> (file_type == FILE_TYPE_Y4M) <a name="l00275"></a>00275 { <a name="l00276"></a>00276 <span class="keywordflow">if</span> (y4m_input_fetch_frame(y4m, f, img) < 1) <a name="l00277"></a>00277 <span class="keywordflow">return</span> 0; <a name="l00278"></a>00278 } <a name="l00279"></a>00279 <span class="keywordflow">else</span> <a name="l00280"></a>00280 { <a name="l00281"></a>00281 <span class="keywordflow">if</span> (file_type == FILE_TYPE_IVF) <a name="l00282"></a>00282 { <a name="l00283"></a>00283 <span class="keywordtype">char</span> junk[IVF_FRAME_HDR_SZ]; <a name="l00284"></a>00284 <a name="l00285"></a>00285 <span class="comment">/* Skip the frame header. We know how big the frame should be. See</span> <a name="l00286"></a>00286 <span class="comment"> * write_ivf_frame_header() for documentation on the frame header</span> <a name="l00287"></a>00287 <span class="comment"> * layout.</span> <a name="l00288"></a>00288 <span class="comment"> */</span> <a name="l00289"></a>00289 <span class="keywordflow">if</span>(fread(junk, 1, IVF_FRAME_HDR_SZ, f)); <a name="l00290"></a>00290 } <a name="l00291"></a>00291 <a name="l00292"></a>00292 <span class="keywordflow">for</span> (plane = 0; plane < 3; plane++) <a name="l00293"></a>00293 { <a name="l00294"></a>00294 <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> *ptr; <a name="l00295"></a>00295 <span class="keywordtype">int</span> w = (plane ? (1 + img-><a class="code" href="structvpx__image.html#a806bf23143bf00a0b3fdbd6ba030c483">d_w</a>) / 2 : img-><a class="code" href="structvpx__image.html#a806bf23143bf00a0b3fdbd6ba030c483">d_w</a>); <a name="l00296"></a>00296 <span class="keywordtype">int</span> h = (plane ? (1 + img-><a class="code" href="structvpx__image.html#a31bc5f045d4f3c2b6bb0f57bb53078e7">d_h</a>) / 2 : img-><a class="code" href="structvpx__image.html#a31bc5f045d4f3c2b6bb0f57bb53078e7">d_h</a>); <a name="l00297"></a>00297 <span class="keywordtype">int</span> r; <a name="l00298"></a>00298 <a name="l00299"></a>00299 <span class="comment">/* Determine the correct plane based on the image format. The for-loop</span> <a name="l00300"></a>00300 <span class="comment"> * always counts in Y,U,V order, but this may not match the order of</span> <a name="l00301"></a>00301 <span class="comment"> * the data on disk.</span> <a name="l00302"></a>00302 <span class="comment"> */</span> <a name="l00303"></a>00303 <span class="keywordflow">switch</span> (plane) <a name="l00304"></a>00304 { <a name="l00305"></a>00305 <span class="keywordflow">case</span> 1: <a name="l00306"></a>00306 ptr = img-><a class="code" href="structvpx__image.html#ab6258308ba7a5f4a113348120e20e2ce">planes</a>[img-><a class="code" href="structvpx__image.html#a1d734f8afa9200a21c2d9f6bcf8c04d8">fmt</a>==<a class="code" href="vpx__image_8h.html#a7a30a7bff7400fb83ad45fede5077193a6e4e8961bbcf96478c300dc4c3e1dc18">VPX_IMG_FMT_YV12</a>? <a class="code" href="vpx__image_8h.html#aca9436ec761457cc6d2e356e0ac2fd23">VPX_PLANE_V</a> : <a class="code" href="vpx__image_8h.html#a4770fc8fa60021a2229f25553152cf81">VPX_PLANE_U</a>]; <a name="l00307"></a>00307 <span class="keywordflow">break</span>; <a name="l00308"></a>00308 <span class="keywordflow">case</span> 2: <a name="l00309"></a>00309 ptr = img-><a class="code" href="structvpx__image.html#ab6258308ba7a5f4a113348120e20e2ce">planes</a>[img-><a class="code" href="structvpx__image.html#a1d734f8afa9200a21c2d9f6bcf8c04d8">fmt</a>==<a class="code" href="vpx__image_8h.html#a7a30a7bff7400fb83ad45fede5077193a6e4e8961bbcf96478c300dc4c3e1dc18">VPX_IMG_FMT_YV12</a>?<a class="code" href="vpx__image_8h.html#a4770fc8fa60021a2229f25553152cf81">VPX_PLANE_U</a> : <a class="code" href="vpx__image_8h.html#aca9436ec761457cc6d2e356e0ac2fd23">VPX_PLANE_V</a>]; <a name="l00310"></a>00310 <span class="keywordflow">break</span>; <a name="l00311"></a>00311 <span class="keywordflow">default</span>: <a name="l00312"></a>00312 ptr = img-><a class="code" href="structvpx__image.html#ab6258308ba7a5f4a113348120e20e2ce">planes</a>[plane]; <a name="l00313"></a>00313 } <a name="l00314"></a>00314 <a name="l00315"></a>00315 <span class="keywordflow">for</span> (r = 0; r < h; r++) <a name="l00316"></a>00316 { <a name="l00317"></a>00317 <span class="keywordtype">size_t</span> needed = w; <a name="l00318"></a>00318 <span class="keywordtype">size_t</span> buf_position = 0; <a name="l00319"></a>00319 <span class="keyword">const</span> <span class="keywordtype">size_t</span> left = detect->buf_read - detect->position; <a name="l00320"></a>00320 <span class="keywordflow">if</span> (left > 0) <a name="l00321"></a>00321 { <a name="l00322"></a>00322 <span class="keyword">const</span> <span class="keywordtype">size_t</span> more = (left < needed) ? left : needed; <a name="l00323"></a>00323 memcpy(ptr, detect->buf + detect->position, more); <a name="l00324"></a>00324 buf_position = more; <a name="l00325"></a>00325 needed -= more; <a name="l00326"></a>00326 detect->position += more; <a name="l00327"></a>00327 } <a name="l00328"></a>00328 <span class="keywordflow">if</span> (needed > 0) <a name="l00329"></a>00329 { <a name="l00330"></a>00330 shortread |= (fread(ptr + buf_position, 1, needed, f) < needed); <a name="l00331"></a>00331 } <a name="l00332"></a>00332 <a name="l00333"></a>00333 ptr += img-><a class="code" href="structvpx__image.html#ac9c7b83e3eea44cb680956f90dc789cf">stride</a>[plane]; <a name="l00334"></a>00334 } <a name="l00335"></a>00335 } <a name="l00336"></a>00336 } <a name="l00337"></a>00337 <a name="l00338"></a>00338 <span class="keywordflow">return</span> !shortread; <a name="l00339"></a>00339 } <a name="l00340"></a>00340 <a name="l00341"></a>00341 <a name="l00342"></a>00342 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> file_is_y4m(FILE *infile, <a name="l00343"></a>00343 y4m_input *y4m, <a name="l00344"></a>00344 <span class="keywordtype">char</span> detect[4]) <a name="l00345"></a>00345 { <a name="l00346"></a>00346 <span class="keywordflow">if</span>(memcmp(detect, <span class="stringliteral">"YUV4"</span>, 4) == 0) <a name="l00347"></a>00347 { <a name="l00348"></a>00348 <span class="keywordflow">return</span> 1; <a name="l00349"></a>00349 } <a name="l00350"></a>00350 <span class="keywordflow">return</span> 0; <a name="l00351"></a>00351 } <a name="l00352"></a>00352 <a name="l00353"></a>00353 <span class="preprocessor">#define IVF_FILE_HDR_SZ (32)</span> <a name="l00354"></a>00354 <span class="preprocessor"></span><span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> file_is_ivf(FILE *infile, <a name="l00355"></a>00355 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> *fourcc, <a name="l00356"></a>00356 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> *width, <a name="l00357"></a>00357 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> *height, <a name="l00358"></a>00358 <span class="keyword">struct</span> detect_buffer *detect) <a name="l00359"></a>00359 { <a name="l00360"></a>00360 <span class="keywordtype">char</span> raw_hdr[IVF_FILE_HDR_SZ]; <a name="l00361"></a>00361 <span class="keywordtype">int</span> is_ivf = 0; <a name="l00362"></a>00362 <a name="l00363"></a>00363 <span class="keywordflow">if</span>(memcmp(detect->buf, <span class="stringliteral">"DKIF"</span>, 4) != 0) <a name="l00364"></a>00364 <span class="keywordflow">return</span> 0; <a name="l00365"></a>00365 <a name="l00366"></a>00366 <span class="comment">/* See write_ivf_file_header() for more documentation on the file header</span> <a name="l00367"></a>00367 <span class="comment"> * layout.</span> <a name="l00368"></a>00368 <span class="comment"> */</span> <a name="l00369"></a>00369 <span class="keywordflow">if</span> (fread(raw_hdr + 4, 1, IVF_FILE_HDR_SZ - 4, infile) <a name="l00370"></a>00370 == IVF_FILE_HDR_SZ - 4) <a name="l00371"></a>00371 { <a name="l00372"></a>00372 { <a name="l00373"></a>00373 is_ivf = 1; <a name="l00374"></a>00374 <a name="l00375"></a>00375 <span class="keywordflow">if</span> (mem_get_le16(raw_hdr + 4) != 0) <a name="l00376"></a>00376 fprintf(stderr, <span class="stringliteral">"Error: Unrecognized IVF version! This file may not"</span> <a name="l00377"></a>00377 <span class="stringliteral">" decode properly."</span>); <a name="l00378"></a>00378 <a name="l00379"></a>00379 *fourcc = mem_get_le32(raw_hdr + 8); <a name="l00380"></a>00380 } <a name="l00381"></a>00381 } <a name="l00382"></a>00382 <a name="l00383"></a>00383 <span class="keywordflow">if</span> (is_ivf) <a name="l00384"></a>00384 { <a name="l00385"></a>00385 *width = mem_get_le16(raw_hdr + 12); <a name="l00386"></a>00386 *height = mem_get_le16(raw_hdr + 14); <a name="l00387"></a>00387 detect->position = 4; <a name="l00388"></a>00388 } <a name="l00389"></a>00389 <a name="l00390"></a>00390 <span class="keywordflow">return</span> is_ivf; <a name="l00391"></a>00391 } <a name="l00392"></a>00392 <a name="l00393"></a>00393 <a name="l00394"></a>00394 <span class="keyword">static</span> <span class="keywordtype">void</span> write_ivf_file_header(FILE *outfile, <a name="l00395"></a>00395 <span class="keyword">const</span> <a class="code" href="structvpx__codec__enc__cfg.html" title="Encoder configuration structure.">vpx_codec_enc_cfg_t</a> *cfg, <a name="l00396"></a>00396 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> fourcc, <a name="l00397"></a>00397 <span class="keywordtype">int</span> frame_cnt) <a name="l00398"></a>00398 { <a name="l00399"></a>00399 <span class="keywordtype">char</span> header[32]; <a name="l00400"></a>00400 <a name="l00401"></a>00401 <span class="keywordflow">if</span> (cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a70d62d87aae7d1168746577f14a6dccf" title="Multi-pass Encoding Mode.">g_pass</a> != <a class="code" href="group__encoder.html#gga476c5417f9c15a1dc5d3f68fa44c493fa7b6943a41868e8e26a77e9500f139ca1">VPX_RC_ONE_PASS</a> && cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a70d62d87aae7d1168746577f14a6dccf" title="Multi-pass Encoding Mode.">g_pass</a> != <a class="code" href="group__encoder.html#gga476c5417f9c15a1dc5d3f68fa44c493fa65da543b956d6a9b1f301ab8cc90d3a7">VPX_RC_LAST_PASS</a>) <a name="l00402"></a>00402 <span class="keywordflow">return</span>; <a name="l00403"></a>00403 <a name="l00404"></a>00404 header[0] = <span class="charliteral">'D'</span>; <a name="l00405"></a>00405 header[1] = <span class="charliteral">'K'</span>; <a name="l00406"></a>00406 header[2] = <span class="charliteral">'I'</span>; <a name="l00407"></a>00407 header[3] = <span class="charliteral">'F'</span>; <a name="l00408"></a>00408 mem_put_le16(header + 4, 0); <span class="comment">/* version */</span> <a name="l00409"></a>00409 mem_put_le16(header + 6, 32); <span class="comment">/* headersize */</span> <a name="l00410"></a>00410 mem_put_le32(header + 8, fourcc); <span class="comment">/* headersize */</span> <a name="l00411"></a>00411 mem_put_le16(header + 12, cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a>); <span class="comment">/* width */</span> <a name="l00412"></a>00412 mem_put_le16(header + 14, cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a>); <span class="comment">/* height */</span> <a name="l00413"></a>00413 mem_put_le32(header + 16, cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a>); <span class="comment">/* rate */</span> <a name="l00414"></a>00414 mem_put_le32(header + 20, cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a>); <span class="comment">/* scale */</span> <a name="l00415"></a>00415 mem_put_le32(header + 24, frame_cnt); <span class="comment">/* length */</span> <a name="l00416"></a>00416 mem_put_le32(header + 28, 0); <span class="comment">/* unused */</span> <a name="l00417"></a>00417 <a name="l00418"></a>00418 <span class="keywordflow">if</span>(fwrite(header, 1, 32, outfile)); <a name="l00419"></a>00419 } <a name="l00420"></a>00420 <a name="l00421"></a>00421 <a name="l00422"></a>00422 <span class="keyword">static</span> <span class="keywordtype">void</span> write_ivf_frame_header(FILE *outfile, <a name="l00423"></a>00423 <span class="keyword">const</span> <a class="code" href="structvpx__codec__cx__pkt.html" title="Encoder output packet.">vpx_codec_cx_pkt_t</a> *pkt) <a name="l00424"></a>00424 { <a name="l00425"></a>00425 <span class="keywordtype">char</span> header[12]; <a name="l00426"></a>00426 <a class="code" href="group__encoder.html#ga7e711b0a71c65aef8f0faea8bd57b05f" title="Time Stamp Type.">vpx_codec_pts_t</a> pts; <a name="l00427"></a>00427 <a name="l00428"></a>00428 <span class="keywordflow">if</span> (pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a41f395b39516343c1329a4a85a0084f2">kind</a> != <a class="code" href="group__encoder.html#gga28a79375279536526552af3a83d2ed72a2261aae5594289400e812fb1e6b6b0cc">VPX_CODEC_CX_FRAME_PKT</a>) <a name="l00429"></a>00429 <span class="keywordflow">return</span>; <a name="l00430"></a>00430 <a name="l00431"></a>00431 pts = pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.pts; <a name="l00432"></a>00432 mem_put_le32(header, pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.sz); <a name="l00433"></a>00433 mem_put_le32(header + 4, pts & 0xFFFFFFFF); <a name="l00434"></a>00434 mem_put_le32(header + 8, pts >> 32); <a name="l00435"></a>00435 <a name="l00436"></a>00436 <span class="keywordflow">if</span>(fwrite(header, 1, 12, outfile)); <a name="l00437"></a>00437 } <a name="l00438"></a>00438 <a name="l00439"></a>00439 <a name="l00440"></a>00440 <span class="keyword">typedef</span> off_t EbmlLoc; <a name="l00441"></a>00441 <a name="l00442"></a>00442 <a name="l00443"></a>00443 <span class="keyword">struct </span>cue_entry <a name="l00444"></a>00444 { <a name="l00445"></a>00445 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> time; <a name="l00446"></a>00446 uint64_t loc; <a name="l00447"></a>00447 }; <a name="l00448"></a>00448 <a name="l00449"></a>00449 <a name="l00450"></a>00450 <span class="keyword">struct </span>EbmlGlobal <a name="l00451"></a>00451 { <a name="l00452"></a>00452 <span class="keywordtype">int</span> debug; <a name="l00453"></a>00453 <a name="l00454"></a>00454 FILE *stream; <a name="l00455"></a>00455 int64_t last_pts_ms; <a name="l00456"></a>00456 <a class="code" href="structvpx__rational.html" title="Rational Number.">vpx_rational_t</a> framerate; <a name="l00457"></a>00457 <a name="l00458"></a>00458 <span class="comment">/* These pointers are to the start of an element */</span> <a name="l00459"></a>00459 off_t position_reference; <a name="l00460"></a>00460 off_t seek_info_pos; <a name="l00461"></a>00461 off_t segment_info_pos; <a name="l00462"></a>00462 off_t track_pos; <a name="l00463"></a>00463 off_t cue_pos; <a name="l00464"></a>00464 off_t cluster_pos; <a name="l00465"></a>00465 <a name="l00466"></a>00466 <span class="comment">/* This pointer is to a specific element to be serialized */</span> <a name="l00467"></a>00467 off_t track_id_pos; <a name="l00468"></a>00468 <a name="l00469"></a>00469 <span class="comment">/* These pointers are to the size field of the element */</span> <a name="l00470"></a>00470 EbmlLoc startSegment; <a name="l00471"></a>00471 EbmlLoc startCluster; <a name="l00472"></a>00472 <a name="l00473"></a>00473 uint32_t cluster_timecode; <a name="l00474"></a>00474 <span class="keywordtype">int</span> cluster_open; <a name="l00475"></a>00475 <a name="l00476"></a>00476 <span class="keyword">struct </span>cue_entry *cue_list; <a name="l00477"></a>00477 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> cues; <a name="l00478"></a>00478 <a name="l00479"></a>00479 }; <a name="l00480"></a>00480 <a name="l00481"></a>00481 <a name="l00482"></a>00482 <span class="keywordtype">void</span> Ebml_Write(EbmlGlobal *glob, <span class="keyword">const</span> <span class="keywordtype">void</span> *buffer_in, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> len) <a name="l00483"></a>00483 { <a name="l00484"></a>00484 <span class="keywordflow">if</span>(fwrite(buffer_in, 1, len, glob->stream)); <a name="l00485"></a>00485 } <a name="l00486"></a>00486 <a name="l00487"></a>00487 <a name="l00488"></a>00488 <span class="keywordtype">void</span> Ebml_Serialize(EbmlGlobal *glob, <span class="keyword">const</span> <span class="keywordtype">void</span> *buffer_in, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> len) <a name="l00489"></a>00489 { <a name="l00490"></a>00490 <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> *q = (<span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> *)buffer_in + len - 1; <a name="l00491"></a>00491 <a name="l00492"></a>00492 <span class="keywordflow">for</span>(; len; len--) <a name="l00493"></a>00493 Ebml_Write(glob, q--, 1); <a name="l00494"></a>00494 } <a name="l00495"></a>00495 <a name="l00496"></a>00496 <a name="l00497"></a>00497 <span class="comment">/* Need a fixed size serializer for the track ID. libmkv provdes a 64 bit</span> <a name="l00498"></a>00498 <span class="comment"> * one, but not a 32 bit one.</span> <a name="l00499"></a>00499 <span class="comment"> */</span> <a name="l00500"></a>00500 <span class="keyword">static</span> <span class="keywordtype">void</span> Ebml_SerializeUnsigned32(EbmlGlobal *glob, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> class_id, uint64_t ui) <a name="l00501"></a>00501 { <a name="l00502"></a>00502 <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> sizeSerialized = 4 | 0x80; <a name="l00503"></a>00503 Ebml_WriteID(glob, class_id); <a name="l00504"></a>00504 Ebml_Serialize(glob, &sizeSerialized, 1); <a name="l00505"></a>00505 Ebml_Serialize(glob, &ui, 4); <a name="l00506"></a>00506 } <a name="l00507"></a>00507 <a name="l00508"></a>00508 <a name="l00509"></a>00509 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00510"></a>00510 Ebml_StartSubElement(EbmlGlobal *glob, EbmlLoc *ebmlLoc, <a name="l00511"></a>00511 <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> class_id) <a name="l00512"></a>00512 { <a name="l00513"></a>00513 <span class="comment">//todo this is always taking 8 bytes, this may need later optimization</span> <a name="l00514"></a>00514 <span class="comment">//this is a key that says lenght unknown</span> <a name="l00515"></a>00515 <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> <span class="keywordtype">long</span> unknownLen = LITERALU64(0x01FFFFFFFFFFFFFF); <a name="l00516"></a>00516 <a name="l00517"></a>00517 Ebml_WriteID(glob, class_id); <a name="l00518"></a>00518 *ebmlLoc = ftello(glob->stream); <a name="l00519"></a>00519 Ebml_Serialize(glob, &unknownLen, 8); <a name="l00520"></a>00520 } <a name="l00521"></a>00521 <a name="l00522"></a>00522 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00523"></a>00523 Ebml_EndSubElement(EbmlGlobal *glob, EbmlLoc *ebmlLoc) <a name="l00524"></a>00524 { <a name="l00525"></a>00525 off_t pos; <a name="l00526"></a>00526 uint64_t size; <a name="l00527"></a>00527 <a name="l00528"></a>00528 <span class="comment">/* Save the current stream pointer */</span> <a name="l00529"></a>00529 pos = ftello(glob->stream); <a name="l00530"></a>00530 <a name="l00531"></a>00531 <span class="comment">/* Calculate the size of this element */</span> <a name="l00532"></a>00532 size = pos - *ebmlLoc - 8; <a name="l00533"></a>00533 size |= LITERALU64(0x0100000000000000); <a name="l00534"></a>00534 <a name="l00535"></a>00535 <span class="comment">/* Seek back to the beginning of the element and write the new size */</span> <a name="l00536"></a>00536 fseeko(glob->stream, *ebmlLoc, SEEK_SET); <a name="l00537"></a>00537 Ebml_Serialize(glob, &size, 8); <a name="l00538"></a>00538 <a name="l00539"></a>00539 <span class="comment">/* Reset the stream pointer */</span> <a name="l00540"></a>00540 fseeko(glob->stream, pos, SEEK_SET); <a name="l00541"></a>00541 } <a name="l00542"></a>00542 <a name="l00543"></a>00543 <a name="l00544"></a>00544 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00545"></a>00545 write_webm_seek_element(EbmlGlobal *ebml, <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> <span class="keywordtype">id</span>, off_t pos) <a name="l00546"></a>00546 { <a name="l00547"></a>00547 uint64_t offset = pos - ebml->position_reference; <a name="l00548"></a>00548 EbmlLoc start; <a name="l00549"></a>00549 Ebml_StartSubElement(ebml, &start, Seek); <a name="l00550"></a>00550 Ebml_SerializeBinary(ebml, SeekID, <span class="keywordtype">id</span>); <a name="l00551"></a>00551 Ebml_SerializeUnsigned64(ebml, SeekPosition, offset); <a name="l00552"></a>00552 Ebml_EndSubElement(ebml, &start); <a name="l00553"></a>00553 } <a name="l00554"></a>00554 <a name="l00555"></a>00555 <a name="l00556"></a>00556 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00557"></a>00557 write_webm_seek_info(EbmlGlobal *ebml) <a name="l00558"></a>00558 { <a name="l00559"></a>00559 <a name="l00560"></a>00560 off_t pos; <a name="l00561"></a>00561 <a name="l00562"></a>00562 <span class="comment">/* Save the current stream pointer */</span> <a name="l00563"></a>00563 pos = ftello(ebml->stream); <a name="l00564"></a>00564 <a name="l00565"></a>00565 <span class="keywordflow">if</span>(ebml->seek_info_pos) <a name="l00566"></a>00566 fseeko(ebml->stream, ebml->seek_info_pos, SEEK_SET); <a name="l00567"></a>00567 <span class="keywordflow">else</span> <a name="l00568"></a>00568 ebml->seek_info_pos = pos; <a name="l00569"></a>00569 <a name="l00570"></a>00570 { <a name="l00571"></a>00571 EbmlLoc start; <a name="l00572"></a>00572 <a name="l00573"></a>00573 Ebml_StartSubElement(ebml, &start, SeekHead); <a name="l00574"></a>00574 write_webm_seek_element(ebml, Tracks, ebml->track_pos); <a name="l00575"></a>00575 write_webm_seek_element(ebml, Cues, ebml->cue_pos); <a name="l00576"></a>00576 write_webm_seek_element(ebml, Info, ebml->segment_info_pos); <a name="l00577"></a>00577 Ebml_EndSubElement(ebml, &start); <a name="l00578"></a>00578 } <a name="l00579"></a>00579 { <a name="l00580"></a>00580 <span class="comment">//segment info</span> <a name="l00581"></a>00581 EbmlLoc startInfo; <a name="l00582"></a>00582 uint64_t frame_time; <a name="l00583"></a>00583 <a name="l00584"></a>00584 frame_time = (uint64_t)1000 * ebml->framerate.den <a name="l00585"></a>00585 / ebml->framerate.num; <a name="l00586"></a>00586 ebml->segment_info_pos = ftello(ebml->stream); <a name="l00587"></a>00587 Ebml_StartSubElement(ebml, &startInfo, Info); <a name="l00588"></a>00588 Ebml_SerializeUnsigned(ebml, TimecodeScale, 1000000); <a name="l00589"></a>00589 Ebml_SerializeFloat(ebml, Segment_Duration, <a name="l00590"></a>00590 ebml->last_pts_ms + frame_time); <a name="l00591"></a>00591 Ebml_SerializeString(ebml, 0x4D80, <a name="l00592"></a>00592 ebml->debug ? <span class="stringliteral">"vpxenc"</span> : <span class="stringliteral">"vpxenc"</span> VERSION_STRING); <a name="l00593"></a>00593 Ebml_SerializeString(ebml, 0x5741, <a name="l00594"></a>00594 ebml->debug ? <span class="stringliteral">"vpxenc"</span> : <span class="stringliteral">"vpxenc"</span> VERSION_STRING); <a name="l00595"></a>00595 Ebml_EndSubElement(ebml, &startInfo); <a name="l00596"></a>00596 } <a name="l00597"></a>00597 } <a name="l00598"></a>00598 <a name="l00599"></a>00599 <a name="l00600"></a>00600 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00601"></a>00601 write_webm_file_header(EbmlGlobal *glob, <a name="l00602"></a>00602 <span class="keyword">const</span> <a class="code" href="structvpx__codec__enc__cfg.html" title="Encoder configuration structure.">vpx_codec_enc_cfg_t</a> *cfg, <a name="l00603"></a>00603 <span class="keyword">const</span> <span class="keyword">struct</span> <a class="code" href="structvpx__rational.html" title="Rational Number.">vpx_rational</a> *fps) <a name="l00604"></a>00604 { <a name="l00605"></a>00605 { <a name="l00606"></a>00606 EbmlLoc start; <a name="l00607"></a>00607 Ebml_StartSubElement(glob, &start, EBML); <a name="l00608"></a>00608 Ebml_SerializeUnsigned(glob, EBMLVersion, 1); <a name="l00609"></a>00609 Ebml_SerializeUnsigned(glob, EBMLReadVersion, 1); <span class="comment">//EBML Read Version</span> <a name="l00610"></a>00610 Ebml_SerializeUnsigned(glob, EBMLMaxIDLength, 4); <span class="comment">//EBML Max ID Length</span> <a name="l00611"></a>00611 Ebml_SerializeUnsigned(glob, EBMLMaxSizeLength, 8); <span class="comment">//EBML Max Size Length</span> <a name="l00612"></a>00612 Ebml_SerializeString(glob, DocType, <span class="stringliteral">"webm"</span>); <span class="comment">//Doc Type</span> <a name="l00613"></a>00613 Ebml_SerializeUnsigned(glob, DocTypeVersion, 2); <span class="comment">//Doc Type Version</span> <a name="l00614"></a>00614 Ebml_SerializeUnsigned(glob, DocTypeReadVersion, 2); <span class="comment">//Doc Type Read Version</span> <a name="l00615"></a>00615 Ebml_EndSubElement(glob, &start); <a name="l00616"></a>00616 } <a name="l00617"></a>00617 { <a name="l00618"></a>00618 Ebml_StartSubElement(glob, &glob->startSegment, Segment); <span class="comment">//segment</span> <a name="l00619"></a>00619 glob->position_reference = ftello(glob->stream); <a name="l00620"></a>00620 glob->framerate = *fps; <a name="l00621"></a>00621 write_webm_seek_info(glob); <a name="l00622"></a>00622 <a name="l00623"></a>00623 { <a name="l00624"></a>00624 EbmlLoc trackStart; <a name="l00625"></a>00625 glob->track_pos = ftello(glob->stream); <a name="l00626"></a>00626 Ebml_StartSubElement(glob, &trackStart, Tracks); <a name="l00627"></a>00627 { <a name="l00628"></a>00628 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> trackNumber = 1; <a name="l00629"></a>00629 uint64_t trackID = 0; <a name="l00630"></a>00630 <a name="l00631"></a>00631 EbmlLoc start; <a name="l00632"></a>00632 Ebml_StartSubElement(glob, &start, TrackEntry); <a name="l00633"></a>00633 Ebml_SerializeUnsigned(glob, TrackNumber, trackNumber); <a name="l00634"></a>00634 glob->track_id_pos = ftello(glob->stream); <a name="l00635"></a>00635 Ebml_SerializeUnsigned32(glob, TrackUID, trackID); <a name="l00636"></a>00636 Ebml_SerializeUnsigned(glob, TrackType, 1); <span class="comment">//video is always 1</span> <a name="l00637"></a>00637 Ebml_SerializeString(glob, CodecID, <span class="stringliteral">"V_VP8"</span>); <a name="l00638"></a>00638 { <a name="l00639"></a>00639 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> pixelWidth = cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a>; <a name="l00640"></a>00640 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> pixelHeight = cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a>; <a name="l00641"></a>00641 <span class="keywordtype">float</span> frameRate = (float)fps-><a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a>/(<span class="keywordtype">float</span>)fps-><a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a>; <a name="l00642"></a>00642 <a name="l00643"></a>00643 EbmlLoc videoStart; <a name="l00644"></a>00644 Ebml_StartSubElement(glob, &videoStart, Video); <a name="l00645"></a>00645 Ebml_SerializeUnsigned(glob, PixelWidth, pixelWidth); <a name="l00646"></a>00646 Ebml_SerializeUnsigned(glob, PixelHeight, pixelHeight); <a name="l00647"></a>00647 Ebml_SerializeFloat(glob, FrameRate, frameRate); <a name="l00648"></a>00648 Ebml_EndSubElement(glob, &videoStart); <span class="comment">//Video</span> <a name="l00649"></a>00649 } <a name="l00650"></a>00650 Ebml_EndSubElement(glob, &start); <span class="comment">//Track Entry</span> <a name="l00651"></a>00651 } <a name="l00652"></a>00652 Ebml_EndSubElement(glob, &trackStart); <a name="l00653"></a>00653 } <a name="l00654"></a>00654 <span class="comment">// segment element is open</span> <a name="l00655"></a>00655 } <a name="l00656"></a>00656 } <a name="l00657"></a>00657 <a name="l00658"></a>00658 <a name="l00659"></a>00659 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00660"></a>00660 write_webm_block(EbmlGlobal *glob, <a name="l00661"></a>00661 <span class="keyword">const</span> <a class="code" href="structvpx__codec__enc__cfg.html" title="Encoder configuration structure.">vpx_codec_enc_cfg_t</a> *cfg, <a name="l00662"></a>00662 <span class="keyword">const</span> <a class="code" href="structvpx__codec__cx__pkt.html" title="Encoder output packet.">vpx_codec_cx_pkt_t</a> *pkt) <a name="l00663"></a>00663 { <a name="l00664"></a>00664 <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> block_length; <a name="l00665"></a>00665 <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> track_number; <a name="l00666"></a>00666 <span class="keywordtype">unsigned</span> <span class="keywordtype">short</span> block_timecode = 0; <a name="l00667"></a>00667 <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> flags; <a name="l00668"></a>00668 int64_t pts_ms; <a name="l00669"></a>00669 <span class="keywordtype">int</span> start_cluster = 0, is_keyframe; <a name="l00670"></a>00670 <a name="l00671"></a>00671 <span class="comment">/* Calculate the PTS of this frame in milliseconds */</span> <a name="l00672"></a>00672 pts_ms = pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.pts * 1000 <a name="l00673"></a>00673 * (uint64_t)cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a> / (uint64_t)cfg-><a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a>; <a name="l00674"></a>00674 <span class="keywordflow">if</span>(pts_ms <= glob->last_pts_ms) <a name="l00675"></a>00675 pts_ms = glob->last_pts_ms + 1; <a name="l00676"></a>00676 glob->last_pts_ms = pts_ms; <a name="l00677"></a>00677 <a name="l00678"></a>00678 <span class="comment">/* Calculate the relative time of this block */</span> <a name="l00679"></a>00679 <span class="keywordflow">if</span>(pts_ms - glob->cluster_timecode > SHRT_MAX) <a name="l00680"></a>00680 start_cluster = 1; <a name="l00681"></a>00681 <span class="keywordflow">else</span> <a name="l00682"></a>00682 block_timecode = pts_ms - glob->cluster_timecode; <a name="l00683"></a>00683 <a name="l00684"></a>00684 is_keyframe = (pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.flags & <a class="code" href="group__encoder.html#ga7a896eb3bc74f33419e6a9812c60e4c9">VPX_FRAME_IS_KEY</a>); <a name="l00685"></a>00685 <span class="keywordflow">if</span>(start_cluster || is_keyframe) <a name="l00686"></a>00686 { <a name="l00687"></a>00687 <span class="keywordflow">if</span>(glob->cluster_open) <a name="l00688"></a>00688 Ebml_EndSubElement(glob, &glob->startCluster); <a name="l00689"></a>00689 <a name="l00690"></a>00690 <span class="comment">/* Open the new cluster */</span> <a name="l00691"></a>00691 block_timecode = 0; <a name="l00692"></a>00692 glob->cluster_open = 1; <a name="l00693"></a>00693 glob->cluster_timecode = pts_ms; <a name="l00694"></a>00694 glob->cluster_pos = ftello(glob->stream); <a name="l00695"></a>00695 Ebml_StartSubElement(glob, &glob->startCluster, Cluster); <span class="comment">//cluster</span> <a name="l00696"></a>00696 Ebml_SerializeUnsigned(glob, Timecode, glob->cluster_timecode); <a name="l00697"></a>00697 <a name="l00698"></a>00698 <span class="comment">/* Save a cue point if this is a keyframe. */</span> <a name="l00699"></a>00699 <span class="keywordflow">if</span>(is_keyframe) <a name="l00700"></a>00700 { <a name="l00701"></a>00701 <span class="keyword">struct </span>cue_entry *cue; <a name="l00702"></a>00702 <a name="l00703"></a>00703 glob->cue_list = realloc(glob->cue_list, <a name="l00704"></a>00704 (glob->cues+1) * <span class="keyword">sizeof</span>(<span class="keyword">struct</span> cue_entry)); <a name="l00705"></a>00705 cue = &glob->cue_list[glob->cues]; <a name="l00706"></a>00706 cue->time = glob->cluster_timecode; <a name="l00707"></a>00707 cue->loc = glob->cluster_pos; <a name="l00708"></a>00708 glob->cues++; <a name="l00709"></a>00709 } <a name="l00710"></a>00710 } <a name="l00711"></a>00711 <a name="l00712"></a>00712 <span class="comment">/* Write the Simple Block */</span> <a name="l00713"></a>00713 Ebml_WriteID(glob, SimpleBlock); <a name="l00714"></a>00714 <a name="l00715"></a>00715 block_length = pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.sz + 4; <a name="l00716"></a>00716 block_length |= 0x10000000; <a name="l00717"></a>00717 Ebml_Serialize(glob, &block_length, 4); <a name="l00718"></a>00718 <a name="l00719"></a>00719 track_number = 1; <a name="l00720"></a>00720 track_number |= 0x80; <a name="l00721"></a>00721 Ebml_Write(glob, &track_number, 1); <a name="l00722"></a>00722 <a name="l00723"></a>00723 Ebml_Serialize(glob, &block_timecode, 2); <a name="l00724"></a>00724 <a name="l00725"></a>00725 flags = 0; <a name="l00726"></a>00726 <span class="keywordflow">if</span>(is_keyframe) <a name="l00727"></a>00727 flags |= 0x80; <a name="l00728"></a>00728 <span class="keywordflow">if</span>(pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.flags & <a class="code" href="group__encoder.html#ga060d7e881d249bdb6e1a29b42e559685">VPX_FRAME_IS_INVISIBLE</a>) <a name="l00729"></a>00729 flags |= 0x08; <a name="l00730"></a>00730 Ebml_Write(glob, &flags, 1); <a name="l00731"></a>00731 <a name="l00732"></a>00732 Ebml_Write(glob, pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.buf, pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.sz); <a name="l00733"></a>00733 } <a name="l00734"></a>00734 <a name="l00735"></a>00735 <a name="l00736"></a>00736 <span class="keyword">static</span> <span class="keywordtype">void</span> <a name="l00737"></a>00737 write_webm_file_footer(EbmlGlobal *glob, <span class="keywordtype">long</span> hash) <a name="l00738"></a>00738 { <a name="l00739"></a>00739 <a name="l00740"></a>00740 <span class="keywordflow">if</span>(glob->cluster_open) <a name="l00741"></a>00741 Ebml_EndSubElement(glob, &glob->startCluster); <a name="l00742"></a>00742 <a name="l00743"></a>00743 { <a name="l00744"></a>00744 EbmlLoc start; <a name="l00745"></a>00745 <span class="keywordtype">int</span> i; <a name="l00746"></a>00746 <a name="l00747"></a>00747 glob->cue_pos = ftello(glob->stream); <a name="l00748"></a>00748 Ebml_StartSubElement(glob, &start, Cues); <a name="l00749"></a>00749 <span class="keywordflow">for</span>(i=0; i<glob->cues; i++) <a name="l00750"></a>00750 { <a name="l00751"></a>00751 <span class="keyword">struct </span>cue_entry *cue = &glob->cue_list[i]; <a name="l00752"></a>00752 EbmlLoc start; <a name="l00753"></a>00753 <a name="l00754"></a>00754 Ebml_StartSubElement(glob, &start, CuePoint); <a name="l00755"></a>00755 { <a name="l00756"></a>00756 EbmlLoc start; <a name="l00757"></a>00757 <a name="l00758"></a>00758 Ebml_SerializeUnsigned(glob, CueTime, cue->time); <a name="l00759"></a>00759 <a name="l00760"></a>00760 Ebml_StartSubElement(glob, &start, CueTrackPositions); <a name="l00761"></a>00761 Ebml_SerializeUnsigned(glob, CueTrack, 1); <a name="l00762"></a>00762 Ebml_SerializeUnsigned64(glob, CueClusterPosition, <a name="l00763"></a>00763 cue->loc - glob->position_reference); <a name="l00764"></a>00764 <span class="comment">//Ebml_SerializeUnsigned(glob, CueBlockNumber, cue->blockNumber);</span> <a name="l00765"></a>00765 Ebml_EndSubElement(glob, &start); <a name="l00766"></a>00766 } <a name="l00767"></a>00767 Ebml_EndSubElement(glob, &start); <a name="l00768"></a>00768 } <a name="l00769"></a>00769 Ebml_EndSubElement(glob, &start); <a name="l00770"></a>00770 } <a name="l00771"></a>00771 <a name="l00772"></a>00772 Ebml_EndSubElement(glob, &glob->startSegment); <a name="l00773"></a>00773 <a name="l00774"></a>00774 <span class="comment">/* Patch up the seek info block */</span> <a name="l00775"></a>00775 write_webm_seek_info(glob); <a name="l00776"></a>00776 <a name="l00777"></a>00777 <span class="comment">/* Patch up the track id */</span> <a name="l00778"></a>00778 fseeko(glob->stream, glob->track_id_pos, SEEK_SET); <a name="l00779"></a>00779 Ebml_SerializeUnsigned32(glob, TrackUID, glob->debug ? 0xDEADBEEF : hash); <a name="l00780"></a>00780 <a name="l00781"></a>00781 fseeko(glob->stream, 0, SEEK_END); <a name="l00782"></a>00782 } <a name="l00783"></a>00783 <a name="l00784"></a>00784 <a name="l00785"></a>00785 <span class="comment">/* Murmur hash derived from public domain reference implementation at</span> <a name="l00786"></a>00786 <span class="comment"> * http://sites.google.com/site/murmurhash/</span> <a name="l00787"></a>00787 <span class="comment"> */</span> <a name="l00788"></a>00788 <span class="keyword">static</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> murmur ( <span class="keyword">const</span> <span class="keywordtype">void</span> * key, <span class="keywordtype">int</span> len, <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> seed ) <a name="l00789"></a>00789 { <a name="l00790"></a>00790 <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> m = 0x5bd1e995; <a name="l00791"></a>00791 <span class="keyword">const</span> <span class="keywordtype">int</span> r = 24; <a name="l00792"></a>00792 <a name="l00793"></a>00793 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> h = seed ^ len; <a name="l00794"></a>00794 <a name="l00795"></a>00795 <span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> * data = (<span class="keyword">const</span> <span class="keywordtype">unsigned</span> <span class="keywordtype">char</span> *)key; <a name="l00796"></a>00796 <a name="l00797"></a>00797 <span class="keywordflow">while</span>(len >= 4) <a name="l00798"></a>00798 { <a name="l00799"></a>00799 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> k; <a name="l00800"></a>00800 <a name="l00801"></a>00801 k = data[0]; <a name="l00802"></a>00802 k |= data[1] << 8; <a name="l00803"></a>00803 k |= data[2] << 16; <a name="l00804"></a>00804 k |= data[3] << 24; <a name="l00805"></a>00805 <a name="l00806"></a>00806 k *= m; <a name="l00807"></a>00807 k ^= k >> r; <a name="l00808"></a>00808 k *= m; <a name="l00809"></a>00809 <a name="l00810"></a>00810 h *= m; <a name="l00811"></a>00811 h ^= k; <a name="l00812"></a>00812 <a name="l00813"></a>00813 data += 4; <a name="l00814"></a>00814 len -= 4; <a name="l00815"></a>00815 } <a name="l00816"></a>00816 <a name="l00817"></a>00817 <span class="keywordflow">switch</span>(len) <a name="l00818"></a>00818 { <a name="l00819"></a>00819 <span class="keywordflow">case</span> 3: h ^= data[2] << 16; <a name="l00820"></a>00820 <span class="keywordflow">case</span> 2: h ^= data[1] << 8; <a name="l00821"></a>00821 <span class="keywordflow">case</span> 1: h ^= data[0]; <a name="l00822"></a>00822 h *= m; <a name="l00823"></a>00823 }; <a name="l00824"></a>00824 <a name="l00825"></a>00825 h ^= h >> 13; <a name="l00826"></a>00826 h *= m; <a name="l00827"></a>00827 h ^= h >> 15; <a name="l00828"></a>00828 <a name="l00829"></a>00829 <span class="keywordflow">return</span> h; <a name="l00830"></a>00830 } <a name="l00831"></a>00831 <a name="l00832"></a>00832 <span class="preprocessor">#include "math.h"</span> <a name="l00833"></a>00833 <a name="l00834"></a>00834 <span class="keyword">static</span> <span class="keywordtype">double</span> vp8_mse2psnr(<span class="keywordtype">double</span> Samples, <span class="keywordtype">double</span> Peak, <span class="keywordtype">double</span> Mse) <a name="l00835"></a>00835 { <a name="l00836"></a>00836 <span class="keywordtype">double</span> psnr; <a name="l00837"></a>00837 <a name="l00838"></a>00838 <span class="keywordflow">if</span> ((<span class="keywordtype">double</span>)Mse > 0.0) <a name="l00839"></a>00839 psnr = 10.0 * log10(Peak * Peak * Samples / Mse); <a name="l00840"></a>00840 <span class="keywordflow">else</span> <a name="l00841"></a>00841 psnr = 60; <span class="comment">// Limit to prevent / 0</span> <a name="l00842"></a>00842 <a name="l00843"></a>00843 <span class="keywordflow">if</span> (psnr > 60) <a name="l00844"></a>00844 psnr = 60; <a name="l00845"></a>00845 <a name="l00846"></a>00846 <span class="keywordflow">return</span> psnr; <a name="l00847"></a>00847 } <a name="l00848"></a>00848 <a name="l00849"></a>00849 <a name="l00850"></a>00850 <span class="preprocessor">#include "args.h"</span> <a name="l00851"></a>00851 <a name="l00852"></a>00852 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t debugmode = ARG_DEF(<span class="stringliteral">"D"</span>, <span class="stringliteral">"debug"</span>, 0, <a name="l00853"></a>00853 <span class="stringliteral">"Debug mode (makes output deterministic)"</span>); <a name="l00854"></a>00854 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t outputfile = ARG_DEF(<span class="stringliteral">"o"</span>, <span class="stringliteral">"output"</span>, 1, <a name="l00855"></a>00855 <span class="stringliteral">"Output filename"</span>); <a name="l00856"></a>00856 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t use_yv12 = ARG_DEF(NULL, <span class="stringliteral">"yv12"</span>, 0, <a name="l00857"></a>00857 <span class="stringliteral">"Input file is YV12 "</span>); <a name="l00858"></a>00858 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t use_i420 = ARG_DEF(NULL, <span class="stringliteral">"i420"</span>, 0, <a name="l00859"></a>00859 <span class="stringliteral">"Input file is I420 (default)"</span>); <a name="l00860"></a>00860 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t codecarg = ARG_DEF(NULL, <span class="stringliteral">"codec"</span>, 1, <a name="l00861"></a>00861 <span class="stringliteral">"Codec to use"</span>); <a name="l00862"></a>00862 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t passes = ARG_DEF(<span class="stringliteral">"p"</span>, <span class="stringliteral">"passes"</span>, 1, <a name="l00863"></a>00863 <span class="stringliteral">"Number of passes (1/2)"</span>); <a name="l00864"></a>00864 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t pass_arg = ARG_DEF(NULL, <span class="stringliteral">"pass"</span>, 1, <a name="l00865"></a>00865 <span class="stringliteral">"Pass to execute (1/2)"</span>); <a name="l00866"></a>00866 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t fpf_name = ARG_DEF(NULL, <span class="stringliteral">"fpf"</span>, 1, <a name="l00867"></a>00867 <span class="stringliteral">"First pass statistics file name"</span>); <a name="l00868"></a>00868 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t limit = ARG_DEF(NULL, <span class="stringliteral">"limit"</span>, 1, <a name="l00869"></a>00869 <span class="stringliteral">"Stop encoding after n input frames"</span>); <a name="l00870"></a>00870 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t deadline = ARG_DEF(<span class="stringliteral">"d"</span>, <span class="stringliteral">"deadline"</span>, 1, <a name="l00871"></a>00871 <span class="stringliteral">"Deadline per frame (usec)"</span>); <a name="l00872"></a>00872 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t best_dl = ARG_DEF(NULL, <span class="stringliteral">"best"</span>, 0, <a name="l00873"></a>00873 <span class="stringliteral">"Use Best Quality Deadline"</span>); <a name="l00874"></a>00874 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t good_dl = ARG_DEF(NULL, <span class="stringliteral">"good"</span>, 0, <a name="l00875"></a>00875 <span class="stringliteral">"Use Good Quality Deadline"</span>); <a name="l00876"></a>00876 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t rt_dl = ARG_DEF(NULL, <span class="stringliteral">"rt"</span>, 0, <a name="l00877"></a>00877 <span class="stringliteral">"Use Realtime Quality Deadline"</span>); <a name="l00878"></a>00878 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t verbosearg = ARG_DEF(<span class="stringliteral">"v"</span>, <span class="stringliteral">"verbose"</span>, 0, <a name="l00879"></a>00879 <span class="stringliteral">"Show encoder parameters"</span>); <a name="l00880"></a>00880 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t psnrarg = ARG_DEF(NULL, <span class="stringliteral">"psnr"</span>, 0, <a name="l00881"></a>00881 <span class="stringliteral">"Show PSNR in status line"</span>); <a name="l00882"></a>00882 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t framerate = ARG_DEF(NULL, <span class="stringliteral">"fps"</span>, 1, <a name="l00883"></a>00883 <span class="stringliteral">"Stream frame rate (rate/scale)"</span>); <a name="l00884"></a>00884 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t use_ivf = ARG_DEF(NULL, <span class="stringliteral">"ivf"</span>, 0, <a name="l00885"></a>00885 <span class="stringliteral">"Output IVF (default is WebM)"</span>); <a name="l00886"></a>00886 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *main_args[] = <a name="l00887"></a>00887 { <a name="l00888"></a>00888 &debugmode, <a name="l00889"></a>00889 &outputfile, &codecarg, &passes, &pass_arg, &fpf_name, &limit, &deadline, <a name="l00890"></a>00890 &best_dl, &good_dl, &rt_dl, <a name="l00891"></a>00891 &verbosearg, &psnrarg, &use_ivf, &framerate, <a name="l00892"></a>00892 NULL <a name="l00893"></a>00893 }; <a name="l00894"></a>00894 <a name="l00895"></a>00895 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t usage = ARG_DEF(<span class="stringliteral">"u"</span>, <span class="stringliteral">"usage"</span>, 1, <a name="l00896"></a>00896 <span class="stringliteral">"Usage profile number to use"</span>); <a name="l00897"></a>00897 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t threads = ARG_DEF(<span class="stringliteral">"t"</span>, <span class="stringliteral">"threads"</span>, 1, <a name="l00898"></a>00898 <span class="stringliteral">"Max number of threads to use"</span>); <a name="l00899"></a>00899 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t profile = ARG_DEF(NULL, <span class="stringliteral">"profile"</span>, 1, <a name="l00900"></a>00900 <span class="stringliteral">"Bitstream profile number to use"</span>); <a name="l00901"></a>00901 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t width = ARG_DEF(<span class="stringliteral">"w"</span>, <span class="stringliteral">"width"</span>, 1, <a name="l00902"></a>00902 <span class="stringliteral">"Frame width"</span>); <a name="l00903"></a>00903 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t height = ARG_DEF(<span class="stringliteral">"h"</span>, <span class="stringliteral">"height"</span>, 1, <a name="l00904"></a>00904 <span class="stringliteral">"Frame height"</span>); <a name="l00905"></a>00905 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t timebase = ARG_DEF(NULL, <span class="stringliteral">"timebase"</span>, 1, <a name="l00906"></a>00906 <span class="stringliteral">"Stream timebase (frame duration)"</span>); <a name="l00907"></a>00907 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t error_resilient = ARG_DEF(NULL, <span class="stringliteral">"error-resilient"</span>, 1, <a name="l00908"></a>00908 <span class="stringliteral">"Enable error resiliency features"</span>); <a name="l00909"></a>00909 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t lag_in_frames = ARG_DEF(NULL, <span class="stringliteral">"lag-in-frames"</span>, 1, <a name="l00910"></a>00910 <span class="stringliteral">"Max number of frames to lag"</span>); <a name="l00911"></a>00911 <a name="l00912"></a>00912 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *global_args[] = <a name="l00913"></a>00913 { <a name="l00914"></a>00914 &use_yv12, &use_i420, &usage, &threads, &profile, <a name="l00915"></a>00915 &width, &height, &timebase, &framerate, &error_resilient, <a name="l00916"></a>00916 &lag_in_frames, NULL <a name="l00917"></a>00917 }; <a name="l00918"></a>00918 <a name="l00919"></a>00919 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t dropframe_thresh = ARG_DEF(NULL, <span class="stringliteral">"drop-frame"</span>, 1, <a name="l00920"></a>00920 <span class="stringliteral">"Temporal resampling threshold (buf %)"</span>); <a name="l00921"></a>00921 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t resize_allowed = ARG_DEF(NULL, <span class="stringliteral">"resize-allowed"</span>, 1, <a name="l00922"></a>00922 <span class="stringliteral">"Spatial resampling enabled (bool)"</span>); <a name="l00923"></a>00923 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t resize_up_thresh = ARG_DEF(NULL, <span class="stringliteral">"resize-up"</span>, 1, <a name="l00924"></a>00924 <span class="stringliteral">"Upscale threshold (buf %)"</span>); <a name="l00925"></a>00925 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t resize_down_thresh = ARG_DEF(NULL, <span class="stringliteral">"resize-down"</span>, 1, <a name="l00926"></a>00926 <span class="stringliteral">"Downscale threshold (buf %)"</span>); <a name="l00927"></a>00927 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t end_usage = ARG_DEF(NULL, <span class="stringliteral">"end-usage"</span>, 1, <a name="l00928"></a>00928 <span class="stringliteral">"VBR=0 | CBR=1 | CQ=2"</span>); <a name="l00929"></a>00929 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t target_bitrate = ARG_DEF(NULL, <span class="stringliteral">"target-bitrate"</span>, 1, <a name="l00930"></a>00930 <span class="stringliteral">"Bitrate (kbps)"</span>); <a name="l00931"></a>00931 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t min_quantizer = ARG_DEF(NULL, <span class="stringliteral">"min-q"</span>, 1, <a name="l00932"></a>00932 <span class="stringliteral">"Minimum (best) quantizer"</span>); <a name="l00933"></a>00933 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t max_quantizer = ARG_DEF(NULL, <span class="stringliteral">"max-q"</span>, 1, <a name="l00934"></a>00934 <span class="stringliteral">"Maximum (worst) quantizer"</span>); <a name="l00935"></a>00935 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t undershoot_pct = ARG_DEF(NULL, <span class="stringliteral">"undershoot-pct"</span>, 1, <a name="l00936"></a>00936 <span class="stringliteral">"Datarate undershoot (min) target (%)"</span>); <a name="l00937"></a>00937 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t overshoot_pct = ARG_DEF(NULL, <span class="stringliteral">"overshoot-pct"</span>, 1, <a name="l00938"></a>00938 <span class="stringliteral">"Datarate overshoot (max) target (%)"</span>); <a name="l00939"></a>00939 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t buf_sz = ARG_DEF(NULL, <span class="stringliteral">"buf-sz"</span>, 1, <a name="l00940"></a>00940 <span class="stringliteral">"Client buffer size (ms)"</span>); <a name="l00941"></a>00941 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t buf_initial_sz = ARG_DEF(NULL, <span class="stringliteral">"buf-initial-sz"</span>, 1, <a name="l00942"></a>00942 <span class="stringliteral">"Client initial buffer size (ms)"</span>); <a name="l00943"></a>00943 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t buf_optimal_sz = ARG_DEF(NULL, <span class="stringliteral">"buf-optimal-sz"</span>, 1, <a name="l00944"></a>00944 <span class="stringliteral">"Client optimal buffer size (ms)"</span>); <a name="l00945"></a>00945 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *rc_args[] = <a name="l00946"></a>00946 { <a name="l00947"></a>00947 &dropframe_thresh, &resize_allowed, &resize_up_thresh, &resize_down_thresh, <a name="l00948"></a>00948 &end_usage, &target_bitrate, &min_quantizer, &max_quantizer, <a name="l00949"></a>00949 &undershoot_pct, &overshoot_pct, &buf_sz, &buf_initial_sz, &buf_optimal_sz, <a name="l00950"></a>00950 NULL <a name="l00951"></a>00951 }; <a name="l00952"></a>00952 <a name="l00953"></a>00953 <a name="l00954"></a>00954 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t bias_pct = ARG_DEF(NULL, <span class="stringliteral">"bias-pct"</span>, 1, <a name="l00955"></a>00955 <span class="stringliteral">"CBR/VBR bias (0=CBR, 100=VBR)"</span>); <a name="l00956"></a>00956 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t minsection_pct = ARG_DEF(NULL, <span class="stringliteral">"minsection-pct"</span>, 1, <a name="l00957"></a>00957 <span class="stringliteral">"GOP min bitrate (% of target)"</span>); <a name="l00958"></a>00958 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t maxsection_pct = ARG_DEF(NULL, <span class="stringliteral">"maxsection-pct"</span>, 1, <a name="l00959"></a>00959 <span class="stringliteral">"GOP max bitrate (% of target)"</span>); <a name="l00960"></a>00960 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *rc_twopass_args[] = <a name="l00961"></a>00961 { <a name="l00962"></a>00962 &bias_pct, &minsection_pct, &maxsection_pct, NULL <a name="l00963"></a>00963 }; <a name="l00964"></a>00964 <a name="l00965"></a>00965 <a name="l00966"></a>00966 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t kf_min_dist = ARG_DEF(NULL, <span class="stringliteral">"kf-min-dist"</span>, 1, <a name="l00967"></a>00967 <span class="stringliteral">"Minimum keyframe interval (frames)"</span>); <a name="l00968"></a>00968 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t kf_max_dist = ARG_DEF(NULL, <span class="stringliteral">"kf-max-dist"</span>, 1, <a name="l00969"></a>00969 <span class="stringliteral">"Maximum keyframe interval (frames)"</span>); <a name="l00970"></a>00970 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t kf_disabled = ARG_DEF(NULL, <span class="stringliteral">"disable-kf"</span>, 0, <a name="l00971"></a>00971 <span class="stringliteral">"Disable keyframe placement"</span>); <a name="l00972"></a>00972 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *kf_args[] = <a name="l00973"></a>00973 { <a name="l00974"></a>00974 &kf_min_dist, &kf_max_dist, &kf_disabled, NULL <a name="l00975"></a>00975 }; <a name="l00976"></a>00976 <a name="l00977"></a>00977 <a name="l00978"></a>00978 <span class="preprocessor">#if CONFIG_VP8_ENCODER</span> <a name="l00979"></a>00979 <span class="preprocessor"></span><span class="keyword">static</span> <span class="keyword">const</span> arg_def_t noise_sens = ARG_DEF(NULL, <span class="stringliteral">"noise-sensitivity"</span>, 1, <a name="l00980"></a>00980 <span class="stringliteral">"Noise sensitivity (frames to blur)"</span>); <a name="l00981"></a>00981 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t sharpness = ARG_DEF(NULL, <span class="stringliteral">"sharpness"</span>, 1, <a name="l00982"></a>00982 <span class="stringliteral">"Filter sharpness (0-7)"</span>); <a name="l00983"></a>00983 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t static_thresh = ARG_DEF(NULL, <span class="stringliteral">"static-thresh"</span>, 1, <a name="l00984"></a>00984 <span class="stringliteral">"Motion detection threshold"</span>); <a name="l00985"></a>00985 <span class="preprocessor">#endif</span> <a name="l00986"></a>00986 <span class="preprocessor"></span> <a name="l00987"></a>00987 <span class="preprocessor">#if CONFIG_VP8_ENCODER</span> <a name="l00988"></a>00988 <span class="preprocessor"></span><span class="keyword">static</span> <span class="keyword">const</span> arg_def_t cpu_used = ARG_DEF(NULL, <span class="stringliteral">"cpu-used"</span>, 1, <a name="l00989"></a>00989 <span class="stringliteral">"CPU Used (-16..16)"</span>); <a name="l00990"></a>00990 <span class="preprocessor">#endif</span> <a name="l00991"></a>00991 <span class="preprocessor"></span> <a name="l00992"></a>00992 <a name="l00993"></a>00993 <span class="preprocessor">#if CONFIG_VP8_ENCODER</span> <a name="l00994"></a>00994 <span class="preprocessor"></span><span class="keyword">static</span> <span class="keyword">const</span> arg_def_t token_parts = ARG_DEF(NULL, <span class="stringliteral">"token-parts"</span>, 1, <a name="l00995"></a>00995 <span class="stringliteral">"Number of token partitions to use, log2"</span>); <a name="l00996"></a>00996 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t auto_altref = ARG_DEF(NULL, <span class="stringliteral">"auto-alt-ref"</span>, 1, <a name="l00997"></a>00997 <span class="stringliteral">"Enable automatic alt reference frames"</span>); <a name="l00998"></a>00998 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t arnr_maxframes = ARG_DEF(NULL, <span class="stringliteral">"arnr-maxframes"</span>, 1, <a name="l00999"></a>00999 <span class="stringliteral">"AltRef Max Frames"</span>); <a name="l01000"></a>01000 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t arnr_strength = ARG_DEF(NULL, <span class="stringliteral">"arnr-strength"</span>, 1, <a name="l01001"></a>01001 <span class="stringliteral">"AltRef Strength"</span>); <a name="l01002"></a>01002 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t arnr_type = ARG_DEF(NULL, <span class="stringliteral">"arnr-type"</span>, 1, <a name="l01003"></a>01003 <span class="stringliteral">"AltRef Type"</span>); <a name="l01004"></a>01004 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keyword">struct </span>arg_enum_list tuning_enum[] = { <a name="l01005"></a>01005 {<span class="stringliteral">"psnr"</span>, VP8_TUNE_PSNR}, <a name="l01006"></a>01006 {<span class="stringliteral">"ssim"</span>, VP8_TUNE_SSIM}, <a name="l01007"></a>01007 {NULL, 0} <a name="l01008"></a>01008 }; <a name="l01009"></a>01009 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t tune_ssim = ARG_DEF_ENUM(NULL, <span class="stringliteral">"tune"</span>, 1, <a name="l01010"></a>01010 <span class="stringliteral">"Material to favor"</span>, tuning_enum); <a name="l01011"></a>01011 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t cq_level = ARG_DEF(NULL, <span class="stringliteral">"cq-level"</span>, 1, <a name="l01012"></a>01012 <span class="stringliteral">"Constrained Quality Level"</span>); <a name="l01013"></a>01013 <a name="l01014"></a>01014 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *vp8_args[] = <a name="l01015"></a>01015 { <a name="l01016"></a>01016 &cpu_used, &auto_altref, &noise_sens, &sharpness, &static_thresh, <a name="l01017"></a>01017 &token_parts, &arnr_maxframes, &arnr_strength, &arnr_type, <a name="l01018"></a>01018 &tune_ssim, &cq_level, NULL <a name="l01019"></a>01019 }; <a name="l01020"></a>01020 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> vp8_arg_ctrl_map[] = <a name="l01021"></a>01021 { <a name="l01022"></a>01022 <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca953ff0eaa2fcdc2ecd627e449b026853" title="control function to set vp8 encoder cpuused">VP8E_SET_CPUUSED</a>, <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca24a5530efabb8b4949d01622b1d72cb0">VP8E_SET_ENABLEAUTOALTREF</a>, <a name="l01023"></a>01023 <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca3fa90cb70bade72af3c2d8d91471a36c">VP8E_SET_NOISE_SENSITIVITY</a>, <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca815930923fff388825ee3cc4ef69bb28">VP8E_SET_SHARPNESS</a>, <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322ecab34559df04d3d662616300ed0682dcfd">VP8E_SET_STATIC_THRESHOLD</a>, <a name="l01024"></a>01024 <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca462a4f14f39a6eb5c4a4004fc8d0ad75">VP8E_SET_TOKEN_PARTITIONS</a>, <a name="l01025"></a>01025 <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca628d73f1a69fadd8aa57e0dfef578da8">VP8E_SET_ARNR_MAXFRAMES</a>, <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca0a8c40a768af935ae73faa53355c3244">VP8E_SET_ARNR_STRENGTH</a> , <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca4e0a0af58670012026c213083cd48f3a">VP8E_SET_ARNR_TYPE</a>, <a name="l01026"></a>01026 <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322eca126ccbe7bde63b59cec5ea68a82dd498">VP8E_SET_TUNING</a>, <a class="code" href="group__vp8__encoder.html#gga6deae3d561c838952552c3d3756322ecacd4f5b730bddb9788c3d0b82218ec503" title="control function to set constrained quality level">VP8E_SET_CQ_LEVEL</a>, 0 <a name="l01027"></a>01027 }; <a name="l01028"></a>01028 <span class="preprocessor">#endif</span> <a name="l01029"></a>01029 <span class="preprocessor"></span> <a name="l01030"></a>01030 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t *no_args[] = { NULL }; <a name="l01031"></a>01031 <a name="l01032"></a>01032 <span class="keyword">static</span> <span class="keywordtype">void</span> usage_exit() <a name="l01033"></a>01033 { <a name="l01034"></a>01034 <span class="keywordtype">int</span> i; <a name="l01035"></a>01035 <a name="l01036"></a>01036 fprintf(stderr, <span class="stringliteral">"Usage: %s <options> -o dst_filename src_filename \n"</span>, <a name="l01037"></a>01037 exec_name); <a name="l01038"></a>01038 <a name="l01039"></a>01039 fprintf(stderr, <span class="stringliteral">"\nOptions:\n"</span>); <a name="l01040"></a>01040 arg_show_usage(stdout, main_args); <a name="l01041"></a>01041 fprintf(stderr, <span class="stringliteral">"\nEncoder Global Options:\n"</span>); <a name="l01042"></a>01042 arg_show_usage(stdout, global_args); <a name="l01043"></a>01043 fprintf(stderr, <span class="stringliteral">"\nRate Control Options:\n"</span>); <a name="l01044"></a>01044 arg_show_usage(stdout, rc_args); <a name="l01045"></a>01045 fprintf(stderr, <span class="stringliteral">"\nTwopass Rate Control Options:\n"</span>); <a name="l01046"></a>01046 arg_show_usage(stdout, rc_twopass_args); <a name="l01047"></a>01047 fprintf(stderr, <span class="stringliteral">"\nKeyframe Placement Options:\n"</span>); <a name="l01048"></a>01048 arg_show_usage(stdout, kf_args); <a name="l01049"></a>01049 <span class="preprocessor">#if CONFIG_VP8_ENCODER</span> <a name="l01050"></a>01050 <span class="preprocessor"></span> fprintf(stderr, <span class="stringliteral">"\nVP8 Specific Options:\n"</span>); <a name="l01051"></a>01051 arg_show_usage(stdout, vp8_args); <a name="l01052"></a>01052 <span class="preprocessor">#endif</span> <a name="l01053"></a>01053 <span class="preprocessor"></span> fprintf(stderr, <span class="stringliteral">"\n"</span> <a name="l01054"></a>01054 <span class="stringliteral">"Included encoders:\n"</span> <a name="l01055"></a>01055 <span class="stringliteral">"\n"</span>); <a name="l01056"></a>01056 <a name="l01057"></a>01057 <span class="keywordflow">for</span> (i = 0; i < <span class="keyword">sizeof</span>(codecs) / <span class="keyword">sizeof</span>(codecs[0]); i++) <a name="l01058"></a>01058 fprintf(stderr, <span class="stringliteral">" %-6s - %s\n"</span>, <a name="l01059"></a>01059 codecs[i].name, <a name="l01060"></a>01060 <a class="code" href="group__codec.html#ga9eee39f450d8cdcf5aba0e2bda6e5aa8" title="Return the name for a given interface.">vpx_codec_iface_name</a>(codecs[i].iface)); <a name="l01061"></a>01061 <a name="l01062"></a>01062 exit(EXIT_FAILURE); <a name="l01063"></a>01063 } <a name="l01064"></a>01064 <a name="l01065"></a>01065 <span class="preprocessor">#define ARG_CTRL_CNT_MAX 10</span> <a name="l01066"></a>01066 <span class="preprocessor"></span> <a name="l01067"></a>01067 <a name="l01068"></a>01068 <span class="keywordtype">int</span> main(<span class="keywordtype">int</span> argc, <span class="keyword">const</span> <span class="keywordtype">char</span> **argv_) <a name="l01069"></a>01069 { <a name="l01070"></a>01070 <a class="code" href="structvpx__codec__ctx.html" title="Codec context structure.">vpx_codec_ctx_t</a> encoder; <a name="l01071"></a>01071 <span class="keyword">const</span> <span class="keywordtype">char</span> *in_fn = NULL, *out_fn = NULL, *stats_fn = NULL; <a name="l01072"></a>01072 <span class="keywordtype">int</span> i; <a name="l01073"></a>01073 FILE *infile, *outfile; <a name="l01074"></a>01074 <a class="code" href="structvpx__codec__enc__cfg.html" title="Encoder configuration structure.">vpx_codec_enc_cfg_t</a> cfg; <a name="l01075"></a>01075 <a class="code" href="group__codec.html#gada1084710837ad363b92f2379dd2b8d2" title="Algorithm return codes.">vpx_codec_err_t</a> res; <a name="l01076"></a>01076 <span class="keywordtype">int</span> pass, one_pass_only = 0; <a name="l01077"></a>01077 stats_io_t stats; <a name="l01078"></a>01078 <a class="code" href="structvpx__image.html" title="Image Descriptor.">vpx_image_t</a> raw; <a name="l01079"></a>01079 <span class="keyword">const</span> <span class="keyword">struct </span>codec_item *codec = codecs; <a name="l01080"></a>01080 <span class="keywordtype">int</span> frame_avail, got_data; <a name="l01081"></a>01081 <a name="l01082"></a>01082 <span class="keyword">struct </span>arg arg; <a name="l01083"></a>01083 <span class="keywordtype">char</span> **argv, **argi, **argj; <a name="l01084"></a>01084 <span class="keywordtype">int</span> arg_usage = 0, arg_passes = 1, arg_deadline = 0; <a name="l01085"></a>01085 <span class="keywordtype">int</span> arg_ctrls[ARG_CTRL_CNT_MAX][2], arg_ctrl_cnt = 0; <a name="l01086"></a>01086 <span class="keywordtype">int</span> arg_limit = 0; <a name="l01087"></a>01087 <span class="keyword">static</span> <span class="keyword">const</span> arg_def_t **ctrl_args = no_args; <a name="l01088"></a>01088 <span class="keyword">static</span> <span class="keyword">const</span> <span class="keywordtype">int</span> *ctrl_args_map = NULL; <a name="l01089"></a>01089 <span class="keywordtype">int</span> verbose = 0, show_psnr = 0; <a name="l01090"></a>01090 <span class="keywordtype">int</span> arg_use_i420 = 1; <a name="l01091"></a>01091 <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> cx_time = 0; <a name="l01092"></a>01092 <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> file_type, fourcc; <a name="l01093"></a>01093 y4m_input y4m; <a name="l01094"></a>01094 <span class="keyword">struct </span><a class="code" href="structvpx__rational.html" title="Rational Number.">vpx_rational</a> arg_framerate = {30, 1}; <a name="l01095"></a>01095 <span class="keywordtype">int</span> arg_have_framerate = 0; <a name="l01096"></a>01096 <span class="keywordtype">int</span> write_webm = 1; <a name="l01097"></a>01097 EbmlGlobal ebml = {0}; <a name="l01098"></a>01098 uint32_t hash = 0; <a name="l01099"></a>01099 uint64_t psnr_sse_total = 0; <a name="l01100"></a>01100 uint64_t psnr_samples_total = 0; <a name="l01101"></a>01101 <span class="keywordtype">double</span> psnr_totals[4] = {0, 0, 0, 0}; <a name="l01102"></a>01102 <span class="keywordtype">int</span> psnr_count = 0; <a name="l01103"></a>01103 <a name="l01104"></a>01104 exec_name = argv_[0]; <a name="l01105"></a>01105 ebml.last_pts_ms = -1; <a name="l01106"></a>01106 <a name="l01107"></a>01107 <span class="keywordflow">if</span> (argc < 3) <a name="l01108"></a>01108 usage_exit(); <a name="l01109"></a>01109 <a name="l01110"></a>01110 <a name="l01111"></a>01111 <span class="comment">/* First parse the codec and usage values, because we want to apply other</span> <a name="l01112"></a>01112 <span class="comment"> * parameters on top of the default configuration provided by the codec.</span> <a name="l01113"></a>01113 <span class="comment"> */</span> <a name="l01114"></a>01114 argv = argv_dup(argc - 1, argv_ + 1); <a name="l01115"></a>01115 <a name="l01116"></a>01116 <span class="keywordflow">for</span> (argi = argj = argv; (*argj = *argi); argi += arg.argv_step) <a name="l01117"></a>01117 { <a name="l01118"></a>01118 arg.argv_step = 1; <a name="l01119"></a>01119 <a name="l01120"></a>01120 <span class="keywordflow">if</span> (arg_match(&arg, &codecarg, argi)) <a name="l01121"></a>01121 { <a name="l01122"></a>01122 <span class="keywordtype">int</span> j, k = -1; <a name="l01123"></a>01123 <a name="l01124"></a>01124 <span class="keywordflow">for</span> (j = 0; j < <span class="keyword">sizeof</span>(codecs) / <span class="keyword">sizeof</span>(codecs[0]); j++) <a name="l01125"></a>01125 <span class="keywordflow">if</span> (!strcmp(codecs[j].name, arg.val)) <a name="l01126"></a>01126 k = j; <a name="l01127"></a>01127 <a name="l01128"></a>01128 <span class="keywordflow">if</span> (k >= 0) <a name="l01129"></a>01129 codec = codecs + k; <a name="l01130"></a>01130 <span class="keywordflow">else</span> <a name="l01131"></a>01131 die(<span class="stringliteral">"Error: Unrecognized argument (%s) to --codec\n"</span>, <a name="l01132"></a>01132 arg.val); <a name="l01133"></a>01133 <a name="l01134"></a>01134 } <a name="l01135"></a>01135 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &passes, argi)) <a name="l01136"></a>01136 { <a name="l01137"></a>01137 arg_passes = arg_parse_uint(&arg); <a name="l01138"></a>01138 <a name="l01139"></a>01139 <span class="keywordflow">if</span> (arg_passes < 1 || arg_passes > 2) <a name="l01140"></a>01140 die(<span class="stringliteral">"Error: Invalid number of passes (%d)\n"</span>, arg_passes); <a name="l01141"></a>01141 } <a name="l01142"></a>01142 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &pass_arg, argi)) <a name="l01143"></a>01143 { <a name="l01144"></a>01144 one_pass_only = arg_parse_uint(&arg); <a name="l01145"></a>01145 <a name="l01146"></a>01146 <span class="keywordflow">if</span> (one_pass_only < 1 || one_pass_only > 2) <a name="l01147"></a>01147 die(<span class="stringliteral">"Error: Invalid pass selected (%d)\n"</span>, one_pass_only); <a name="l01148"></a>01148 } <a name="l01149"></a>01149 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &fpf_name, argi)) <a name="l01150"></a>01150 stats_fn = arg.val; <a name="l01151"></a>01151 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &usage, argi)) <a name="l01152"></a>01152 arg_usage = arg_parse_uint(&arg); <a name="l01153"></a>01153 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &deadline, argi)) <a name="l01154"></a>01154 arg_deadline = arg_parse_uint(&arg); <a name="l01155"></a>01155 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &best_dl, argi)) <a name="l01156"></a>01156 arg_deadline = <a class="code" href="group__encoder.html#gab350573bea112f2fdf8e5677db3ac0da">VPX_DL_BEST_QUALITY</a>; <a name="l01157"></a>01157 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &good_dl, argi)) <a name="l01158"></a>01158 arg_deadline = <a class="code" href="group__encoder.html#ga5d00a9f9e10b9f49ca91e72c1f01c9fd">VPX_DL_GOOD_QUALITY</a>; <a name="l01159"></a>01159 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &rt_dl, argi)) <a name="l01160"></a>01160 arg_deadline = <a class="code" href="group__encoder.html#ga04253cc9ec1146d72fa8bb86bcf32144">VPX_DL_REALTIME</a>; <a name="l01161"></a>01161 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &use_yv12, argi)) <a name="l01162"></a>01162 { <a name="l01163"></a>01163 arg_use_i420 = 0; <a name="l01164"></a>01164 } <a name="l01165"></a>01165 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &use_i420, argi)) <a name="l01166"></a>01166 { <a name="l01167"></a>01167 arg_use_i420 = 1; <a name="l01168"></a>01168 } <a name="l01169"></a>01169 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &verbosearg, argi)) <a name="l01170"></a>01170 verbose = 1; <a name="l01171"></a>01171 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &limit, argi)) <a name="l01172"></a>01172 arg_limit = arg_parse_uint(&arg); <a name="l01173"></a>01173 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &psnrarg, argi)) <a name="l01174"></a>01174 show_psnr = 1; <a name="l01175"></a>01175 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &framerate, argi)) <a name="l01176"></a>01176 { <a name="l01177"></a>01177 arg_framerate = arg_parse_rational(&arg); <a name="l01178"></a>01178 arg_have_framerate = 1; <a name="l01179"></a>01179 } <a name="l01180"></a>01180 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &use_ivf, argi)) <a name="l01181"></a>01181 write_webm = 0; <a name="l01182"></a>01182 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &outputfile, argi)) <a name="l01183"></a>01183 out_fn = arg.val; <a name="l01184"></a>01184 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &debugmode, argi)) <a name="l01185"></a>01185 ebml.debug = 1; <a name="l01186"></a>01186 <span class="keywordflow">else</span> <a name="l01187"></a>01187 argj++; <a name="l01188"></a>01188 } <a name="l01189"></a>01189 <a name="l01190"></a>01190 <span class="comment">/* Ensure that --passes and --pass are consistent. If --pass is set and --passes=2,</span> <a name="l01191"></a>01191 <span class="comment"> * ensure --fpf was set.</span> <a name="l01192"></a>01192 <span class="comment"> */</span> <a name="l01193"></a>01193 <span class="keywordflow">if</span> (one_pass_only) <a name="l01194"></a>01194 { <a name="l01195"></a>01195 <span class="comment">/* DWIM: Assume the user meant passes=2 if pass=2 is specified */</span> <a name="l01196"></a>01196 <span class="keywordflow">if</span> (one_pass_only > arg_passes) <a name="l01197"></a>01197 { <a name="l01198"></a>01198 fprintf(stderr, <span class="stringliteral">"Warning: Assuming --pass=%d implies --passes=%d\n"</span>, <a name="l01199"></a>01199 one_pass_only, one_pass_only); <a name="l01200"></a>01200 arg_passes = one_pass_only; <a name="l01201"></a>01201 } <a name="l01202"></a>01202 <a name="l01203"></a>01203 <span class="keywordflow">if</span> (arg_passes == 2 && !stats_fn) <a name="l01204"></a>01204 die(<span class="stringliteral">"Must specify --fpf when --pass=%d and --passes=2\n"</span>, one_pass_only); <a name="l01205"></a>01205 } <a name="l01206"></a>01206 <a name="l01207"></a>01207 <span class="comment">/* Populate encoder configuration */</span> <a name="l01208"></a>01208 res = <a class="code" href="group__encoder.html#ga9ab8c79623312e9c5d5405d66a788f59" title="Get a default configuration.">vpx_codec_enc_config_default</a>(codec->iface, &cfg, arg_usage); <a name="l01209"></a>01209 <a name="l01210"></a>01210 <span class="keywordflow">if</span> (res) <a name="l01211"></a>01211 { <a name="l01212"></a>01212 fprintf(stderr, <span class="stringliteral">"Failed to get config: %s\n"</span>, <a name="l01213"></a>01213 <a class="code" href="group__codec.html#gaaddf5c1f609ef18c7c8800d102fcefa6" title="Convert error number to printable string.">vpx_codec_err_to_string</a>(res)); <a name="l01214"></a>01214 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01215"></a>01215 } <a name="l01216"></a>01216 <a name="l01217"></a>01217 <span class="comment">/* Change the default timebase to a high enough value so that the encoder</span> <a name="l01218"></a>01218 <span class="comment"> * will always create strictly increasing timestamps.</span> <a name="l01219"></a>01219 <span class="comment"> */</span> <a name="l01220"></a>01220 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a> = 1000; <a name="l01221"></a>01221 <a name="l01222"></a>01222 <span class="comment">/* Never use the library's default resolution, require it be parsed</span> <a name="l01223"></a>01223 <span class="comment"> * from the file or set on the command line.</span> <a name="l01224"></a>01224 <span class="comment"> */</span> <a name="l01225"></a>01225 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a> = 0; <a name="l01226"></a>01226 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a> = 0; <a name="l01227"></a>01227 <a name="l01228"></a>01228 <span class="comment">/* Now parse the remainder of the parameters. */</span> <a name="l01229"></a>01229 <span class="keywordflow">for</span> (argi = argj = argv; (*argj = *argi); argi += arg.argv_step) <a name="l01230"></a>01230 { <a name="l01231"></a>01231 arg.argv_step = 1; <a name="l01232"></a>01232 <a name="l01233"></a>01233 <span class="keywordflow">if</span> (0); <a name="l01234"></a>01234 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &threads, argi)) <a name="l01235"></a>01235 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#aff9d8d458c45c95dd542603210ba2b8b" title="Maximum number of threads to use.">g_threads</a> = arg_parse_uint(&arg); <a name="l01236"></a>01236 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &profile, argi)) <a name="l01237"></a>01237 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a3d7fe3a8b237338c094d5a956a41804d" title="Bitstream profile to use.">g_profile</a> = arg_parse_uint(&arg); <a name="l01238"></a>01238 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &width, argi)) <a name="l01239"></a>01239 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a> = arg_parse_uint(&arg); <a name="l01240"></a>01240 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &height, argi)) <a name="l01241"></a>01241 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a> = arg_parse_uint(&arg); <a name="l01242"></a>01242 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &timebase, argi)) <a name="l01243"></a>01243 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a> = arg_parse_rational(&arg); <a name="l01244"></a>01244 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &error_resilient, argi)) <a name="l01245"></a>01245 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a9abffd5b85a0babbe3073b763f3311e1" title="Enable error resilient mode.">g_error_resilient</a> = arg_parse_uint(&arg); <a name="l01246"></a>01246 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &lag_in_frames, argi)) <a name="l01247"></a>01247 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a992668d9e30305f3f7ab2672ea31a890" title="Allow lagged encoding.">g_lag_in_frames</a> = arg_parse_uint(&arg); <a name="l01248"></a>01248 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &dropframe_thresh, argi)) <a name="l01249"></a>01249 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a619269f9a6904de58c4790e6806a3905" title="Temporal resampling configuration, if supported by the codec.">rc_dropframe_thresh</a> = arg_parse_uint(&arg); <a name="l01250"></a>01250 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &resize_allowed, argi)) <a name="l01251"></a>01251 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a02a4e2f18fb0fdfff44df8b0d9a99d6c" title="Enable/disable spatial resampling, if supported by the codec.">rc_resize_allowed</a> = arg_parse_uint(&arg); <a name="l01252"></a>01252 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &resize_up_thresh, argi)) <a name="l01253"></a>01253 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a855599c0660f31dfcab4a64996b4f6ad" title="Spatial resampling up watermark.">rc_resize_up_thresh</a> = arg_parse_uint(&arg); <a name="l01254"></a>01254 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &resize_down_thresh, argi)) <a name="l01255"></a>01255 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a1be7bd7b0eaf1e99b5e4028c37c65488" title="Spatial resampling down watermark.">rc_resize_down_thresh</a> = arg_parse_uint(&arg); <a name="l01256"></a>01256 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &resize_down_thresh, argi)) <a name="l01257"></a>01257 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a1be7bd7b0eaf1e99b5e4028c37c65488" title="Spatial resampling down watermark.">rc_resize_down_thresh</a> = arg_parse_uint(&arg); <a name="l01258"></a>01258 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &end_usage, argi)) <a name="l01259"></a>01259 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a99c415edb6c6f909a095a57c3430a116" title="Rate control algorithm to use.">rc_end_usage</a> = arg_parse_uint(&arg); <a name="l01260"></a>01260 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &target_bitrate, argi)) <a name="l01261"></a>01261 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#ab8339685175d66710f482706cc9f0aed" title="Target data rate.">rc_target_bitrate</a> = arg_parse_uint(&arg); <a name="l01262"></a>01262 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &min_quantizer, argi)) <a name="l01263"></a>01263 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a1324600e3c63faaa7717e85cf6509dd9" title="Minimum (Best Quality) Quantizer.">rc_min_quantizer</a> = arg_parse_uint(&arg); <a name="l01264"></a>01264 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &max_quantizer, argi)) <a name="l01265"></a>01265 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a8abdd985d317b720e142335887d737df" title="Maximum (Worst Quality) Quantizer.">rc_max_quantizer</a> = arg_parse_uint(&arg); <a name="l01266"></a>01266 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &undershoot_pct, argi)) <a name="l01267"></a>01267 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#ad279fbb0a9fe9395340913161e28651a" title="Rate control undershoot tolerance.">rc_undershoot_pct</a> = arg_parse_uint(&arg); <a name="l01268"></a>01268 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &overshoot_pct, argi)) <a name="l01269"></a>01269 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4f2fbd015c1b1f395cc4a4b03d4ead5f" title="Rate control overshoot tolerance.">rc_overshoot_pct</a> = arg_parse_uint(&arg); <a name="l01270"></a>01270 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &buf_sz, argi)) <a name="l01271"></a>01271 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a61866272bb588cd86d28834f420430f0" title="Decoder Buffer Size.">rc_buf_sz</a> = arg_parse_uint(&arg); <a name="l01272"></a>01272 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &buf_initial_sz, argi)) <a name="l01273"></a>01273 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#aa9e4d6405994ef42d61d478cf6e0d5e0" title="Decoder Buffer Initial Size.">rc_buf_initial_sz</a> = arg_parse_uint(&arg); <a name="l01274"></a>01274 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &buf_optimal_sz, argi)) <a name="l01275"></a>01275 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#aafde485867e040a58504ad796e79e47f" title="Decoder Buffer Optimal Size.">rc_buf_optimal_sz</a> = arg_parse_uint(&arg); <a name="l01276"></a>01276 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &bias_pct, argi)) <a name="l01277"></a>01277 { <a name="l01278"></a>01278 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a21c21ff097890dc3a450731c9b504cf4" title="Two-pass mode CBR/VBR bias.">rc_2pass_vbr_bias_pct</a> = arg_parse_uint(&arg); <a name="l01279"></a>01279 <a name="l01280"></a>01280 <span class="keywordflow">if</span> (arg_passes < 2) <a name="l01281"></a>01281 fprintf(stderr, <a name="l01282"></a>01282 <span class="stringliteral">"Warning: option %s ignored in one-pass mode.\n"</span>, <a name="l01283"></a>01283 arg.name); <a name="l01284"></a>01284 } <a name="l01285"></a>01285 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &minsection_pct, argi)) <a name="l01286"></a>01286 { <a name="l01287"></a>01287 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a21eb9fce0844ae07b617bf3f0a25f5a5" title="Two-pass mode per-GOP minimum bitrate.">rc_2pass_vbr_minsection_pct</a> = arg_parse_uint(&arg); <a name="l01288"></a>01288 <a name="l01289"></a>01289 <span class="keywordflow">if</span> (arg_passes < 2) <a name="l01290"></a>01290 fprintf(stderr, <a name="l01291"></a>01291 <span class="stringliteral">"Warning: option %s ignored in one-pass mode.\n"</span>, <a name="l01292"></a>01292 arg.name); <a name="l01293"></a>01293 } <a name="l01294"></a>01294 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &maxsection_pct, argi)) <a name="l01295"></a>01295 { <a name="l01296"></a>01296 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#ab5212050b71b2d9f4dc663caa496949e" title="Two-pass mode per-GOP maximum bitrate.">rc_2pass_vbr_maxsection_pct</a> = arg_parse_uint(&arg); <a name="l01297"></a>01297 <a name="l01298"></a>01298 <span class="keywordflow">if</span> (arg_passes < 2) <a name="l01299"></a>01299 fprintf(stderr, <a name="l01300"></a>01300 <span class="stringliteral">"Warning: option %s ignored in one-pass mode.\n"</span>, <a name="l01301"></a>01301 arg.name); <a name="l01302"></a>01302 } <a name="l01303"></a>01303 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &kf_min_dist, argi)) <a name="l01304"></a>01304 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a0a7b5444ecb09745cbe8d5af17553846" title="Keyframe minimum interval.">kf_min_dist</a> = arg_parse_uint(&arg); <a name="l01305"></a>01305 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &kf_max_dist, argi)) <a name="l01306"></a>01306 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#ae018440136e271743376730413d25a9b" title="Keyframe maximum interval.">kf_max_dist</a> = arg_parse_uint(&arg); <a name="l01307"></a>01307 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (arg_match(&arg, &kf_disabled, argi)) <a name="l01308"></a>01308 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a491d67f061dcdb13f60c017563e9d788" title="Keyframe placement mode.">kf_mode</a> = <a class="code" href="group__encoder.html#gga9f461802aa4db35c04a8b23837987f40a56513f60cbf3434ca0092173005d0ab2">VPX_KF_DISABLED</a>; <a name="l01309"></a>01309 <span class="keywordflow">else</span> <a name="l01310"></a>01310 argj++; <a name="l01311"></a>01311 } <a name="l01312"></a>01312 <a name="l01313"></a>01313 <span class="comment">/* Handle codec specific options */</span> <a name="l01314"></a>01314 <span class="preprocessor">#if CONFIG_VP8_ENCODER</span> <a name="l01315"></a>01315 <span class="preprocessor"></span> <a name="l01316"></a>01316 <span class="keywordflow">if</span> (codec->iface == &vpx_codec_vp8_cx_algo) <a name="l01317"></a>01317 { <a name="l01318"></a>01318 ctrl_args = vp8_args; <a name="l01319"></a>01319 ctrl_args_map = vp8_arg_ctrl_map; <a name="l01320"></a>01320 } <a name="l01321"></a>01321 <a name="l01322"></a>01322 <span class="preprocessor">#endif</span> <a name="l01323"></a>01323 <span class="preprocessor"></span> <a name="l01324"></a>01324 <span class="keywordflow">for</span> (argi = argj = argv; (*argj = *argi); argi += arg.argv_step) <a name="l01325"></a>01325 { <a name="l01326"></a>01326 <span class="keywordtype">int</span> match = 0; <a name="l01327"></a>01327 <a name="l01328"></a>01328 arg.argv_step = 1; <a name="l01329"></a>01329 <a name="l01330"></a>01330 <span class="keywordflow">for</span> (i = 0; ctrl_args[i]; i++) <a name="l01331"></a>01331 { <a name="l01332"></a>01332 <span class="keywordflow">if</span> (arg_match(&arg, ctrl_args[i], argi)) <a name="l01333"></a>01333 { <a name="l01334"></a>01334 match = 1; <a name="l01335"></a>01335 <a name="l01336"></a>01336 <span class="keywordflow">if</span> (arg_ctrl_cnt < ARG_CTRL_CNT_MAX) <a name="l01337"></a>01337 { <a name="l01338"></a>01338 arg_ctrls[arg_ctrl_cnt][0] = ctrl_args_map[i]; <a name="l01339"></a>01339 arg_ctrls[arg_ctrl_cnt][1] = arg_parse_enum_or_int(&arg); <a name="l01340"></a>01340 arg_ctrl_cnt++; <a name="l01341"></a>01341 } <a name="l01342"></a>01342 } <a name="l01343"></a>01343 } <a name="l01344"></a>01344 <a name="l01345"></a>01345 <span class="keywordflow">if</span> (!match) <a name="l01346"></a>01346 argj++; <a name="l01347"></a>01347 } <a name="l01348"></a>01348 <a name="l01349"></a>01349 <span class="comment">/* Check for unrecognized options */</span> <a name="l01350"></a>01350 <span class="keywordflow">for</span> (argi = argv; *argi; argi++) <a name="l01351"></a>01351 <span class="keywordflow">if</span> (argi[0][0] == <span class="charliteral">'-'</span> && argi[0][1]) <a name="l01352"></a>01352 die(<span class="stringliteral">"Error: Unrecognized option %s\n"</span>, *argi); <a name="l01353"></a>01353 <a name="l01354"></a>01354 <span class="comment">/* Handle non-option arguments */</span> <a name="l01355"></a>01355 in_fn = argv[0]; <a name="l01356"></a>01356 <a name="l01357"></a>01357 <span class="keywordflow">if</span> (!in_fn) <a name="l01358"></a>01358 usage_exit(); <a name="l01359"></a>01359 <a name="l01360"></a>01360 <span class="keywordflow">if</span>(!out_fn) <a name="l01361"></a>01361 die(<span class="stringliteral">"Error: Output file is required (specify with -o)\n"</span>); <a name="l01362"></a>01362 <a name="l01363"></a>01363 memset(&stats, 0, <span class="keyword">sizeof</span>(stats)); <a name="l01364"></a>01364 <a name="l01365"></a>01365 <span class="keywordflow">for</span> (pass = one_pass_only ? one_pass_only - 1 : 0; pass < arg_passes; pass++) <a name="l01366"></a>01366 { <a name="l01367"></a>01367 <span class="keywordtype">int</span> frames_in = 0, frames_out = 0; <a name="l01368"></a>01368 <span class="keywordtype">unsigned</span> <span class="keywordtype">long</span> nbytes = 0; <a name="l01369"></a>01369 <span class="keyword">struct </span>detect_buffer detect; <a name="l01370"></a>01370 <a name="l01371"></a>01371 <span class="comment">/* Parse certain options from the input file, if possible */</span> <a name="l01372"></a>01372 infile = strcmp(in_fn, <span class="stringliteral">"-"</span>) ? fopen(in_fn, <span class="stringliteral">"rb"</span>) <a name="l01373"></a>01373 : set_binary_mode(stdin); <a name="l01374"></a>01374 <a name="l01375"></a>01375 <span class="keywordflow">if</span> (!infile) <a name="l01376"></a>01376 { <a name="l01377"></a>01377 fprintf(stderr, <span class="stringliteral">"Failed to open input file\n"</span>); <a name="l01378"></a>01378 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01379"></a>01379 } <a name="l01380"></a>01380 <a name="l01381"></a>01381 <span class="comment">/* For RAW input sources, these bytes will applied on the first frame</span> <a name="l01382"></a>01382 <span class="comment"> * in read_frame().</span> <a name="l01383"></a>01383 <span class="comment"> */</span> <a name="l01384"></a>01384 detect.buf_read = fread(detect.buf, 1, 4, infile); <a name="l01385"></a>01385 detect.position = 0; <a name="l01386"></a>01386 <a name="l01387"></a>01387 <span class="keywordflow">if</span> (detect.buf_read == 4 && file_is_y4m(infile, &y4m, detect.buf)) <a name="l01388"></a>01388 { <a name="l01389"></a>01389 <span class="keywordflow">if</span> (y4m_input_open(&y4m, infile, detect.buf, 4) >= 0) <a name="l01390"></a>01390 { <a name="l01391"></a>01391 file_type = FILE_TYPE_Y4M; <a name="l01392"></a>01392 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a> = y4m.pic_w; <a name="l01393"></a>01393 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a> = y4m.pic_h; <a name="l01394"></a>01394 <a name="l01395"></a>01395 <span class="comment">/* Use the frame rate from the file only if none was specified</span> <a name="l01396"></a>01396 <span class="comment"> * on the command-line.</span> <a name="l01397"></a>01397 <span class="comment"> */</span> <a name="l01398"></a>01398 <span class="keywordflow">if</span> (!arg_have_framerate) <a name="l01399"></a>01399 { <a name="l01400"></a>01400 arg_framerate.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a> = y4m.fps_n; <a name="l01401"></a>01401 arg_framerate.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a> = y4m.fps_d; <a name="l01402"></a>01402 } <a name="l01403"></a>01403 <a name="l01404"></a>01404 arg_use_i420 = 0; <a name="l01405"></a>01405 } <a name="l01406"></a>01406 <span class="keywordflow">else</span> <a name="l01407"></a>01407 { <a name="l01408"></a>01408 fprintf(stderr, <span class="stringliteral">"Unsupported Y4M stream.\n"</span>); <a name="l01409"></a>01409 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01410"></a>01410 } <a name="l01411"></a>01411 } <a name="l01412"></a>01412 <span class="keywordflow">else</span> <span class="keywordflow">if</span> (detect.buf_read == 4 && <a name="l01413"></a>01413 file_is_ivf(infile, &fourcc, &cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a>, &cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a>, &detect)) <a name="l01414"></a>01414 { <a name="l01415"></a>01415 file_type = FILE_TYPE_IVF; <a name="l01416"></a>01416 <span class="keywordflow">switch</span> (fourcc) <a name="l01417"></a>01417 { <a name="l01418"></a>01418 <span class="keywordflow">case</span> 0x32315659: <a name="l01419"></a>01419 arg_use_i420 = 0; <a name="l01420"></a>01420 <span class="keywordflow">break</span>; <a name="l01421"></a>01421 <span class="keywordflow">case</span> 0x30323449: <a name="l01422"></a>01422 arg_use_i420 = 1; <a name="l01423"></a>01423 <span class="keywordflow">break</span>; <a name="l01424"></a>01424 <span class="keywordflow">default</span>: <a name="l01425"></a>01425 fprintf(stderr, <span class="stringliteral">"Unsupported fourcc (%08x) in IVF\n"</span>, fourcc); <a name="l01426"></a>01426 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01427"></a>01427 } <a name="l01428"></a>01428 } <a name="l01429"></a>01429 <span class="keywordflow">else</span> <a name="l01430"></a>01430 { <a name="l01431"></a>01431 file_type = FILE_TYPE_RAW; <a name="l01432"></a>01432 } <a name="l01433"></a>01433 <a name="l01434"></a>01434 <span class="keywordflow">if</span>(!cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a> || !cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a>) <a name="l01435"></a>01435 { <a name="l01436"></a>01436 fprintf(stderr, <span class="stringliteral">"Specify stream dimensions with --width (-w) "</span> <a name="l01437"></a>01437 <span class="stringliteral">" and --height (-h).\n"</span>); <a name="l01438"></a>01438 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01439"></a>01439 } <a name="l01440"></a>01440 <a name="l01441"></a>01441 <span class="preprocessor">#define SHOW(field) fprintf(stderr, " %-28s = %d\n", #field, cfg.field)</span> <a name="l01442"></a>01442 <span class="preprocessor"></span> <a name="l01443"></a>01443 <span class="keywordflow">if</span> (verbose && pass == 0) <a name="l01444"></a>01444 { <a name="l01445"></a>01445 fprintf(stderr, <span class="stringliteral">"Codec: %s\n"</span>, <a class="code" href="group__codec.html#ga9eee39f450d8cdcf5aba0e2bda6e5aa8" title="Return the name for a given interface.">vpx_codec_iface_name</a>(codec->iface)); <a name="l01446"></a>01446 fprintf(stderr, <span class="stringliteral">"Source file: %s Format: %s\n"</span>, in_fn, <a name="l01447"></a>01447 arg_use_i420 ? <span class="stringliteral">"I420"</span> : <span class="stringliteral">"YV12"</span>); <a name="l01448"></a>01448 fprintf(stderr, <span class="stringliteral">"Destination file: %s\n"</span>, out_fn); <a name="l01449"></a>01449 fprintf(stderr, <span class="stringliteral">"Encoder parameters:\n"</span>); <a name="l01450"></a>01450 <a name="l01451"></a>01451 SHOW(g_usage); <a name="l01452"></a>01452 SHOW(g_threads); <a name="l01453"></a>01453 SHOW(g_profile); <a name="l01454"></a>01454 SHOW(g_w); <a name="l01455"></a>01455 SHOW(g_h); <a name="l01456"></a>01456 SHOW(g_timebase.num); <a name="l01457"></a>01457 SHOW(g_timebase.den); <a name="l01458"></a>01458 SHOW(g_error_resilient); <a name="l01459"></a>01459 SHOW(g_pass); <a name="l01460"></a>01460 SHOW(g_lag_in_frames); <a name="l01461"></a>01461 SHOW(rc_dropframe_thresh); <a name="l01462"></a>01462 SHOW(rc_resize_allowed); <a name="l01463"></a>01463 SHOW(rc_resize_up_thresh); <a name="l01464"></a>01464 SHOW(rc_resize_down_thresh); <a name="l01465"></a>01465 SHOW(rc_end_usage); <a name="l01466"></a>01466 SHOW(rc_target_bitrate); <a name="l01467"></a>01467 SHOW(rc_min_quantizer); <a name="l01468"></a>01468 SHOW(rc_max_quantizer); <a name="l01469"></a>01469 SHOW(rc_undershoot_pct); <a name="l01470"></a>01470 SHOW(rc_overshoot_pct); <a name="l01471"></a>01471 SHOW(rc_buf_sz); <a name="l01472"></a>01472 SHOW(rc_buf_initial_sz); <a name="l01473"></a>01473 SHOW(rc_buf_optimal_sz); <a name="l01474"></a>01474 SHOW(rc_2pass_vbr_bias_pct); <a name="l01475"></a>01475 SHOW(rc_2pass_vbr_minsection_pct); <a name="l01476"></a>01476 SHOW(rc_2pass_vbr_maxsection_pct); <a name="l01477"></a>01477 SHOW(kf_mode); <a name="l01478"></a>01478 SHOW(kf_min_dist); <a name="l01479"></a>01479 SHOW(kf_max_dist); <a name="l01480"></a>01480 } <a name="l01481"></a>01481 <a name="l01482"></a>01482 <span class="keywordflow">if</span>(pass == (one_pass_only ? one_pass_only - 1 : 0)) { <a name="l01483"></a>01483 <span class="keywordflow">if</span> (file_type == FILE_TYPE_Y4M) <a name="l01484"></a>01484 <span class="comment">/*The Y4M reader does its own allocation.</span> <a name="l01485"></a>01485 <span class="comment"> Just initialize this here to avoid problems if we never read any</span> <a name="l01486"></a>01486 <span class="comment"> frames.*/</span> <a name="l01487"></a>01487 memset(&raw, 0, <span class="keyword">sizeof</span>(raw)); <a name="l01488"></a>01488 <span class="keywordflow">else</span> <a name="l01489"></a>01489 <a class="code" href="vpx__image_8h.html#ab80398f28ca880368a922fa979eb66c1" title="Open a descriptor, allocating storage for the underlying image.">vpx_img_alloc</a>(&raw, arg_use_i420 ? <a class="code" href="vpx__image_8h.html#a7a30a7bff7400fb83ad45fede5077193af17ce49c9a6efd7c8c8ea8928d4ce26c">VPX_IMG_FMT_I420</a> : <a class="code" href="vpx__image_8h.html#a7a30a7bff7400fb83ad45fede5077193a6e4e8961bbcf96478c300dc4c3e1dc18">VPX_IMG_FMT_YV12</a>, <a name="l01490"></a>01490 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a5c165f5b41ca1158f2883983a2b7709c" title="Width of the frame.">g_w</a>, cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a4132bd89ce85bce7c08f2cc3b6f2b82e" title="Height of the frame.">g_h</a>, 1); <a name="l01491"></a>01491 } <a name="l01492"></a>01492 <a name="l01493"></a>01493 outfile = strcmp(out_fn, <span class="stringliteral">"-"</span>) ? fopen(out_fn, <span class="stringliteral">"wb"</span>) <a name="l01494"></a>01494 : set_binary_mode(stdout); <a name="l01495"></a>01495 <a name="l01496"></a>01496 <span class="keywordflow">if</span> (!outfile) <a name="l01497"></a>01497 { <a name="l01498"></a>01498 fprintf(stderr, <span class="stringliteral">"Failed to open output file\n"</span>); <a name="l01499"></a>01499 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01500"></a>01500 } <a name="l01501"></a>01501 <a name="l01502"></a>01502 <span class="keywordflow">if</span>(write_webm && fseek(outfile, 0, SEEK_CUR)) <a name="l01503"></a>01503 { <a name="l01504"></a>01504 fprintf(stderr, <span class="stringliteral">"WebM output to pipes not supported.\n"</span>); <a name="l01505"></a>01505 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01506"></a>01506 } <a name="l01507"></a>01507 <a name="l01508"></a>01508 <span class="keywordflow">if</span> (stats_fn) <a name="l01509"></a>01509 { <a name="l01510"></a>01510 <span class="keywordflow">if</span> (!stats_open_file(&stats, stats_fn, pass)) <a name="l01511"></a>01511 { <a name="l01512"></a>01512 fprintf(stderr, <span class="stringliteral">"Failed to open statistics store\n"</span>); <a name="l01513"></a>01513 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01514"></a>01514 } <a name="l01515"></a>01515 } <a name="l01516"></a>01516 <span class="keywordflow">else</span> <a name="l01517"></a>01517 { <a name="l01518"></a>01518 <span class="keywordflow">if</span> (!stats_open_mem(&stats, pass)) <a name="l01519"></a>01519 { <a name="l01520"></a>01520 fprintf(stderr, <span class="stringliteral">"Failed to open statistics store\n"</span>); <a name="l01521"></a>01521 <span class="keywordflow">return</span> EXIT_FAILURE; <a name="l01522"></a>01522 } <a name="l01523"></a>01523 } <a name="l01524"></a>01524 <a name="l01525"></a>01525 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a70d62d87aae7d1168746577f14a6dccf" title="Multi-pass Encoding Mode.">g_pass</a> = arg_passes == 2 <a name="l01526"></a>01526 ? pass ? <a class="code" href="group__encoder.html#gga476c5417f9c15a1dc5d3f68fa44c493fa65da543b956d6a9b1f301ab8cc90d3a7">VPX_RC_LAST_PASS</a> : <a class="code" href="group__encoder.html#gga476c5417f9c15a1dc5d3f68fa44c493faa0765945345b160905f3b762986dae3b">VPX_RC_FIRST_PASS</a> <a name="l01527"></a>01527 : <a class="code" href="group__encoder.html#gga476c5417f9c15a1dc5d3f68fa44c493fa7b6943a41868e8e26a77e9500f139ca1">VPX_RC_ONE_PASS</a>; <a name="l01528"></a>01528 <span class="preprocessor">#if VPX_ENCODER_ABI_VERSION > (1 + VPX_CODEC_ABI_VERSION)</span> <a name="l01529"></a>01529 <span class="preprocessor"></span> <a name="l01530"></a>01530 <span class="keywordflow">if</span> (pass) <a name="l01531"></a>01531 { <a name="l01532"></a>01532 cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a618616a07d9d8883c544731957b56f3c" title="Two-pass stats buffer.">rc_twopass_stats_in</a> = stats_get(&stats); <a name="l01533"></a>01533 } <a name="l01534"></a>01534 <a name="l01535"></a>01535 <span class="preprocessor">#endif</span> <a name="l01536"></a>01536 <span class="preprocessor"></span> <a name="l01537"></a>01537 <span class="keywordflow">if</span>(write_webm) <a name="l01538"></a>01538 { <a name="l01539"></a>01539 ebml.stream = outfile; <a name="l01540"></a>01540 write_webm_file_header(&ebml, &cfg, &arg_framerate); <a name="l01541"></a>01541 } <a name="l01542"></a>01542 <span class="keywordflow">else</span> <a name="l01543"></a>01543 write_ivf_file_header(outfile, &cfg, codec->fourcc, 0); <a name="l01544"></a>01544 <a name="l01545"></a>01545 <a name="l01546"></a>01546 <span class="comment">/* Construct Encoder Context */</span> <a name="l01547"></a>01547 <a class="code" href="group__encoder.html#ga3d490a2a9a6acd7c9ef82a603155f3cf" title="Convenience macro for vpx_codec_enc_init_ver()">vpx_codec_enc_init</a>(&encoder, codec->iface, &cfg, <a name="l01548"></a>01548 show_psnr ? <a class="code" href="group__encoder.html#ga57bb9eb1881c7e4bf86580660a5e40a0" title="Initialization-time Feature Enabling.">VPX_CODEC_USE_PSNR</a> : 0); <a name="l01549"></a>01549 ctx_exit_on_error(&encoder, <span class="stringliteral">"Failed to initialize encoder"</span>); <a name="l01550"></a>01550 <a name="l01551"></a>01551 <span class="comment">/* Note that we bypass the vpx_codec_control wrapper macro because</span> <a name="l01552"></a>01552 <span class="comment"> * we're being clever to store the control IDs in an array. Real</span> <a name="l01553"></a>01553 <span class="comment"> * applications will want to make use of the enumerations directly</span> <a name="l01554"></a>01554 <span class="comment"> */</span> <a name="l01555"></a>01555 <span class="keywordflow">for</span> (i = 0; i < arg_ctrl_cnt; i++) <a name="l01556"></a>01556 { <a name="l01557"></a>01557 <span class="keywordflow">if</span> (<a class="code" href="group__codec.html#gaf7849b69cc390b21d889c8c71f69ec14" title="Control algorithm.">vpx_codec_control_</a>(&encoder, arg_ctrls[i][0], arg_ctrls[i][1])) <a name="l01558"></a>01558 fprintf(stderr, <span class="stringliteral">"Error: Tried to set control %d = %d\n"</span>, <a name="l01559"></a>01559 arg_ctrls[i][0], arg_ctrls[i][1]); <a name="l01560"></a>01560 <a name="l01561"></a>01561 ctx_exit_on_error(&encoder, <span class="stringliteral">"Failed to control codec"</span>); <a name="l01562"></a>01562 } <a name="l01563"></a>01563 <a name="l01564"></a>01564 frame_avail = 1; <a name="l01565"></a>01565 got_data = 0; <a name="l01566"></a>01566 <a name="l01567"></a>01567 <span class="keywordflow">while</span> (frame_avail || got_data) <a name="l01568"></a>01568 { <a name="l01569"></a>01569 <a class="code" href="group__codec.html#ga6ea348f76b1f8a1fe50e14db684146c6" title="Iterator.">vpx_codec_iter_t</a> iter = NULL; <a name="l01570"></a>01570 <span class="keyword">const</span> <a class="code" href="structvpx__codec__cx__pkt.html" title="Encoder output packet.">vpx_codec_cx_pkt_t</a> *pkt; <a name="l01571"></a>01571 <span class="keyword">struct </span>vpx_usec_timer timer; <a name="l01572"></a>01572 int64_t frame_start, next_frame_start; <a name="l01573"></a>01573 <a name="l01574"></a>01574 <span class="keywordflow">if</span> (!arg_limit || frames_in < arg_limit) <a name="l01575"></a>01575 { <a name="l01576"></a>01576 frame_avail = read_frame(infile, &raw, file_type, &y4m, <a name="l01577"></a>01577 &detect); <a name="l01578"></a>01578 <a name="l01579"></a>01579 <span class="keywordflow">if</span> (frame_avail) <a name="l01580"></a>01580 frames_in++; <a name="l01581"></a>01581 <a name="l01582"></a>01582 fprintf(stderr, <a name="l01583"></a>01583 <span class="stringliteral">"\rPass %d/%d frame %4d/%-4d %7ldB \033[K"</span>, pass + 1, <a name="l01584"></a>01584 arg_passes, frames_in, frames_out, nbytes); <a name="l01585"></a>01585 } <a name="l01586"></a>01586 <span class="keywordflow">else</span> <a name="l01587"></a>01587 frame_avail = 0; <a name="l01588"></a>01588 <a name="l01589"></a>01589 vpx_usec_timer_start(&timer); <a name="l01590"></a>01590 <a name="l01591"></a>01591 frame_start = (cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a> * (int64_t)(frames_in - 1) <a name="l01592"></a>01592 * arg_framerate.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a>) / cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a> / arg_framerate.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a>; <a name="l01593"></a>01593 next_frame_start = (cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a> * (int64_t)(frames_in) <a name="l01594"></a>01594 * arg_framerate.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a>) <a name="l01595"></a>01595 / cfg.<a class="code" href="structvpx__codec__enc__cfg.html#a6498d378e4c29ef3e22258289e481087" title="Stream timebase units.">g_timebase</a>.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a> / arg_framerate.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a>; <a name="l01596"></a>01596 <a class="code" href="group__encoder.html#gaf990542e2aeb389f05fae3e9c7803639" title="Encode a frame.">vpx_codec_encode</a>(&encoder, frame_avail ? &raw : NULL, frame_start, <a name="l01597"></a>01597 next_frame_start - frame_start, <a name="l01598"></a>01598 0, arg_deadline); <a name="l01599"></a>01599 vpx_usec_timer_mark(&timer); <a name="l01600"></a>01600 cx_time += vpx_usec_timer_elapsed(&timer); <a name="l01601"></a>01601 ctx_exit_on_error(&encoder, <span class="stringliteral">"Failed to encode frame"</span>); <a name="l01602"></a>01602 got_data = 0; <a name="l01603"></a>01603 <a name="l01604"></a>01604 <span class="keywordflow">while</span> ((pkt = <a class="code" href="group__encoder.html#gae81cab25d66cf3bc59f1f75f8a5af720" title="Encoded data iterator.">vpx_codec_get_cx_data</a>(&encoder, &iter))) <a name="l01605"></a>01605 { <a name="l01606"></a>01606 got_data = 1; <a name="l01607"></a>01607 <a name="l01608"></a>01608 <span class="keywordflow">switch</span> (pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a41f395b39516343c1329a4a85a0084f2">kind</a>) <a name="l01609"></a>01609 { <a name="l01610"></a>01610 <span class="keywordflow">case</span> <a class="code" href="group__encoder.html#gga28a79375279536526552af3a83d2ed72a2261aae5594289400e812fb1e6b6b0cc">VPX_CODEC_CX_FRAME_PKT</a>: <a name="l01611"></a>01611 frames_out++; <a name="l01612"></a>01612 fprintf(stderr, <span class="stringliteral">" %6luF"</span>, <a name="l01613"></a>01613 (<span class="keywordtype">unsigned</span> <span class="keywordtype">long</span>)pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.sz); <a name="l01614"></a>01614 <a name="l01615"></a>01615 <span class="keywordflow">if</span>(write_webm) <a name="l01616"></a>01616 { <a name="l01617"></a>01617 <span class="comment">/* Update the hash */</span> <a name="l01618"></a>01618 <span class="keywordflow">if</span>(!ebml.debug) <a name="l01619"></a>01619 hash = murmur(pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.buf, <a name="l01620"></a>01620 pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.sz, hash); <a name="l01621"></a>01621 <a name="l01622"></a>01622 write_webm_block(&ebml, &cfg, pkt); <a name="l01623"></a>01623 } <a name="l01624"></a>01624 <span class="keywordflow">else</span> <a name="l01625"></a>01625 { <a name="l01626"></a>01626 write_ivf_frame_header(outfile, pkt); <a name="l01627"></a>01627 <span class="keywordflow">if</span>(fwrite(pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.buf, 1, <a name="l01628"></a>01628 pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a81e33bf4408a3983abb16492fee359ff">frame</a>.sz, outfile)); <a name="l01629"></a>01629 } <a name="l01630"></a>01630 nbytes += pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#acc7ee91ea9c907aea7c3b953324f7c26">raw</a>.<a class="code" href="structvpx__fixed__buf.html#a43d7f8247ed9752f65f0a7639817ae84">sz</a>; <a name="l01631"></a>01631 <span class="keywordflow">break</span>; <a name="l01632"></a>01632 <span class="keywordflow">case</span> <a class="code" href="group__encoder.html#gga28a79375279536526552af3a83d2ed72a77cd8b95d9891bcce36117f5f31994fd">VPX_CODEC_STATS_PKT</a>: <a name="l01633"></a>01633 frames_out++; <a name="l01634"></a>01634 fprintf(stderr, <span class="stringliteral">" %6luS"</span>, <a name="l01635"></a>01635 (<span class="keywordtype">unsigned</span> <span class="keywordtype">long</span>)pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a53db8576979fa7335e478eb849b256f1">twopass_stats</a>.<a class="code" href="structvpx__fixed__buf.html#a43d7f8247ed9752f65f0a7639817ae84">sz</a>); <a name="l01636"></a>01636 stats_write(&stats, <a name="l01637"></a>01637 pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a53db8576979fa7335e478eb849b256f1">twopass_stats</a>.<a class="code" href="structvpx__fixed__buf.html#ac93e43ae3fcc1023dca86d37016ae3f0">buf</a>, <a name="l01638"></a>01638 pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a53db8576979fa7335e478eb849b256f1">twopass_stats</a>.<a class="code" href="structvpx__fixed__buf.html#a43d7f8247ed9752f65f0a7639817ae84">sz</a>); <a name="l01639"></a>01639 nbytes += pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#acc7ee91ea9c907aea7c3b953324f7c26">raw</a>.<a class="code" href="structvpx__fixed__buf.html#a43d7f8247ed9752f65f0a7639817ae84">sz</a>; <a name="l01640"></a>01640 <span class="keywordflow">break</span>; <a name="l01641"></a>01641 <span class="keywordflow">case</span> <a class="code" href="group__encoder.html#gga28a79375279536526552af3a83d2ed72a11eefa1328e6d22eb7c844645a260628">VPX_CODEC_PSNR_PKT</a>: <a name="l01642"></a>01642 <a name="l01643"></a>01643 <span class="keywordflow">if</span> (show_psnr) <a name="l01644"></a>01644 { <a name="l01645"></a>01645 <span class="keywordtype">int</span> i; <a name="l01646"></a>01646 <a name="l01647"></a>01647 psnr_sse_total += pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a63744aeceb50355d6402d55309c151b6">psnr</a>.sse[0]; <a name="l01648"></a>01648 psnr_samples_total += pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a63744aeceb50355d6402d55309c151b6">psnr</a>.samples[0]; <a name="l01649"></a>01649 <span class="keywordflow">for</span> (i = 0; i < 4; i++) <a name="l01650"></a>01650 { <a name="l01651"></a>01651 fprintf(stderr, <span class="stringliteral">"%.3lf "</span>, pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a63744aeceb50355d6402d55309c151b6">psnr</a>.psnr[i]); <a name="l01652"></a>01652 psnr_totals[i] += pkt-><a class="code" href="structvpx__codec__cx__pkt.html#a7f97b060a23b7e89fe5b885c0074f696">data</a>.<a class="code" href="structvpx__codec__cx__pkt.html#a63744aeceb50355d6402d55309c151b6">psnr</a>.psnr[i]; <a name="l01653"></a>01653 } <a name="l01654"></a>01654 psnr_count++; <a name="l01655"></a>01655 } <a name="l01656"></a>01656 <a name="l01657"></a>01657 <span class="keywordflow">break</span>; <a name="l01658"></a>01658 <span class="keywordflow">default</span>: <a name="l01659"></a>01659 <span class="keywordflow">break</span>; <a name="l01660"></a>01660 } <a name="l01661"></a>01661 } <a name="l01662"></a>01662 <a name="l01663"></a>01663 fflush(stdout); <a name="l01664"></a>01664 } <a name="l01665"></a>01665 <a name="l01666"></a>01666 fprintf(stderr, <a name="l01667"></a>01667 <span class="stringliteral">"\rPass %d/%d frame %4d/%-4d %7ldB %7ldb/f %7"</span>PRId64<span class="stringliteral">"b/s"</span> <a name="l01668"></a>01668 <span class="stringliteral">" %7lu %s (%.2f fps)\033[K"</span>, pass + 1, <a name="l01669"></a>01669 arg_passes, frames_in, frames_out, nbytes, nbytes * 8 / frames_in, <a name="l01670"></a>01670 nbytes * 8 *(int64_t)arg_framerate.<a class="code" href="structvpx__rational.html#ae7774f21a22c9bef3aa73156c79f4731">num</a> / arg_framerate.<a class="code" href="structvpx__rational.html#a29dd2ab4001377b3aa21885ef969759f">den</a> / frames_in, <a name="l01671"></a>01671 cx_time > 9999999 ? cx_time / 1000 : cx_time, <a name="l01672"></a>01672 cx_time > 9999999 ? <span class="stringliteral">"ms"</span> : <span class="stringliteral">"us"</span>, <a name="l01673"></a>01673 (<span class="keywordtype">float</span>)frames_in * 1000000.0 / (<span class="keywordtype">float</span>)cx_time); <a name="l01674"></a>01674 <a name="l01675"></a>01675 <span class="keywordflow">if</span> ( (show_psnr) && (psnr_count>0) ) <a name="l01676"></a>01676 { <a name="l01677"></a>01677 <span class="keywordtype">int</span> i; <a name="l01678"></a>01678 <span class="keywordtype">double</span> ovpsnr = vp8_mse2psnr(psnr_samples_total, 255.0, <a name="l01679"></a>01679 psnr_sse_total); <a name="l01680"></a>01680 <a name="l01681"></a>01681 fprintf(stderr, <span class="stringliteral">"\nPSNR (Overall/Avg/Y/U/V)"</span>); <a name="l01682"></a>01682 <a name="l01683"></a>01683 fprintf(stderr, <span class="stringliteral">" %.3lf"</span>, ovpsnr); <a name="l01684"></a>01684 <span class="keywordflow">for</span> (i = 0; i < 4; i++) <a name="l01685"></a>01685 { <a name="l01686"></a>01686 fprintf(stderr, <span class="stringliteral">" %.3lf"</span>, psnr_totals[i]/psnr_count); <a name="l01687"></a>01687 } <a name="l01688"></a>01688 } <a name="l01689"></a>01689 <a name="l01690"></a>01690 <a class="code" href="group__codec.html#ga6ae21d96909660d0ac978b59a863f53b" title="Destroy a codec instance.">vpx_codec_destroy</a>(&encoder); <a name="l01691"></a>01691 <a name="l01692"></a>01692 fclose(infile); <a name="l01693"></a>01693 <a name="l01694"></a>01694 <span class="keywordflow">if</span>(write_webm) <a name="l01695"></a>01695 { <a name="l01696"></a>01696 write_webm_file_footer(&ebml, hash); <a name="l01697"></a>01697 } <a name="l01698"></a>01698 <span class="keywordflow">else</span> <a name="l01699"></a>01699 { <a name="l01700"></a>01700 <span class="keywordflow">if</span> (!fseek(outfile, 0, SEEK_SET)) <a name="l01701"></a>01701 write_ivf_file_header(outfile, &cfg, codec->fourcc, frames_out); <a name="l01702"></a>01702 } <a name="l01703"></a>01703 <a name="l01704"></a>01704 fclose(outfile); <a name="l01705"></a>01705 stats_close(&stats, arg_passes-1); <a name="l01706"></a>01706 fprintf(stderr, <span class="stringliteral">"\n"</span>); <a name="l01707"></a>01707 <a name="l01708"></a>01708 <span class="keywordflow">if</span> (one_pass_only) <a name="l01709"></a>01709 <span class="keywordflow">break</span>; <a name="l01710"></a>01710 } <a name="l01711"></a>01711 <a name="l01712"></a>01712 <a class="code" href="vpx__image_8h.html#af47ebaf286812d037425d41990d48a17" title="Close an image descriptor.">vpx_img_free</a>(&raw); <a name="l01713"></a>01713 free(argv); <a name="l01714"></a>01714 <span class="keywordflow">return</span> EXIT_SUCCESS; <a name="l01715"></a>01715 } </pre></div> </div></div> <hr class="footer"/><address class="footer"><small>Generated on Thu Mar 10 2011 for WebM VP8 Codec SDK by  <a href="http://www.doxygen.org/index.html"> <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.7.3 </small></address> </body> </html>