<!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: vp8cx_set_ref.c</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>vp8cx_set_ref.c </h1> </div>
</div>
<div class="contents">
<div class="textblock">
<h1>VP8 Set Reference Frame</h1>
<p>This is an example demonstrating how to overwrite the VP8 encoder’s
internal reference frame. In the sample we set the last frame to the
current frame. If this is done at a cut scene it will avoid a keyframe.
This technique could be used to bounce between two cameras.</p>
<p>Note that the decoder would also have to set the reference frame to the
same value on the same frame, or the video will become corrupt.</p>
<h2>Usage</h2>
<p>This example adds a single argument to the <code>simple_encoder</code> example,
which specifies the frame number to update the reference frame on.
The parameter is parsed as follows:</p>
<pre class="c"><span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>argc<span style="color: #339933;">!=</span><span style="color: #0000dd;">6</span><span style="color: #009900;">)</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Usage: %s <width> <height> <infile> <outfile> <frame><span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span>
argv<span style="color: #009900;">[</span><span style="color: #0000dd;">0</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
update_frame_num <span style="color: #339933;">=</span> atoi<span style="color: #009900;">(</span>argv<span style="color: #009900;">[</span><span style="color: #0000dd;">5</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span><span style="color: #339933;">!</span>update_frame_num<span style="color: #009900;">)</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Couldn't parse frame number '%s'<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span> argv<span style="color: #009900;">[</span><span style="color: #0000dd;">5</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
</pre>
<h2>Extra Variables</h2>
<p>This example maintains the frame number passed on the command line
in the <code>update_frame_num</code> variable:</p>
<pre class="c"><span style="color: #993333;">int</span> update_frame_num <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span></pre>
<h2>Configuration</h2>
<p>The reference frame is updated on the frame specified on the command
line.</p>
<pre class="c">frame_avail <span style="color: #339933;">=</span> read_frame<span style="color: #009900;">(</span>infile<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>raw<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>frame_cnt <span style="color: #339933;">+</span> <span style="color: #0000dd;">1</span> <span style="color: #339933;">==</span> update_frame_num<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
vpx_ref_frame_t ref<span style="color: #339933;">;</span>
ref.<span style="color: #202020;">frame_type</span> <span style="color: #339933;">=</span> VP8_LAST_FRAME<span style="color: #339933;">;</span>
ref.<span style="color: #202020;">img</span> <span style="color: #339933;">=</span> raw<span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>vpx_codec_control<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> VP8_SET_REFERENCE<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>ref<span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die_codec<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #ff0000;">"Failed to set reference frame"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>vpx_codec_encode<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> frame_avail<span style="color: #339933;">?</span> <span style="color: #339933;">&</span>raw <span style="color: #339933;">:</span> NULL<span style="color: #339933;">,</span> frame_cnt<span style="color: #339933;">,</span>
<span style="color: #0000dd;">1</span><span style="color: #339933;">,</span> flags<span style="color: #339933;">,</span> VPX_DL_REALTIME<span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die_codec<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #ff0000;">"Failed to encode frame"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span></pre>
<h2>Observing The Effects</h2>
<p>Use the <code>simple_encoder</code> example to encode a sample with a cut scene.
Determine the frame number of the cut scene by looking for a generated
key-frame (indicated by a ‘K’). Supply that frame number as an argument
to this example, and observe that no key-frame is generated.</p>
<h1>Putting It All Together</h1>
<pre class="c"><span style="color: #808080; font-style: italic;">/*
Copyright (c) 2010 The WebM project authors. All Rights Reserved.
Use of this source code is governed by a BSD-style license
that can be found in the LICENSE file in the root of the source
tree. An additional intellectual property rights grant can be found
in the file PATENTS. All contributing project authors may
be found in the AUTHORS file in the root of the source tree.
*/</span>
<span style="color: #808080; font-style: italic;">/*
This is an example demonstrating how to overwrite the VP8 encoder's
internal reference frame. In the sample we set the last frame to the
current frame. If this is done at a cut scene it will avoid a keyframe.
This technique could be used to bounce between two cameras.
Note that the decoder would also have to set the reference frame to the
same value on the same frame, or the video will become corrupt.
*/</span>
<span style="color: #339933;">#include <stdio.h></span>
<span style="color: #339933;">#include <stdlib.h></span>
<span style="color: #339933;">#include <stdarg.h></span>
<span style="color: #339933;">#include <string.h></span>
<span style="color: #339933;">#define VPX_CODEC_DISABLE_COMPAT 1</span>
<span style="color: #339933;">#include "vpx/vpx_encoder.h"</span>
<span style="color: #339933;">#include "vpx/vp8cx.h"</span>
<span style="color: #339933;">#define interface (vpx_codec_vp8_cx())</span>
<span style="color: #339933;">#define fourcc 0x30385056</span>
<span style="color: #339933;">#define IVF_FILE_HDR_SZ (32)</span>
<span style="color: #339933;">#define IVF_FRAME_HDR_SZ (12)</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">void</span> mem_put_le16<span style="color: #009900;">(</span><span style="color: #993333;">char</span> <span style="color: #339933;">*</span>mem<span style="color: #339933;">,</span> <span style="color: #993333;">unsigned</span> <span style="color: #993333;">int</span> val<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
mem<span style="color: #009900;">[</span><span style="color: #0000dd;">0</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> val<span style="color: #339933;">;</span>
mem<span style="color: #009900;">[</span><span style="color: #0000dd;">1</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> val<span style="color: #339933;">>></span><span style="color: #0000dd;">8</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">void</span> mem_put_le32<span style="color: #009900;">(</span><span style="color: #993333;">char</span> <span style="color: #339933;">*</span>mem<span style="color: #339933;">,</span> <span style="color: #993333;">unsigned</span> <span style="color: #993333;">int</span> val<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
mem<span style="color: #009900;">[</span><span style="color: #0000dd;">0</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> val<span style="color: #339933;">;</span>
mem<span style="color: #009900;">[</span><span style="color: #0000dd;">1</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> val<span style="color: #339933;">>></span><span style="color: #0000dd;">8</span><span style="color: #339933;">;</span>
mem<span style="color: #009900;">[</span><span style="color: #0000dd;">2</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> val<span style="color: #339933;">>></span><span style="color: #0000dd;">16</span><span style="color: #339933;">;</span>
mem<span style="color: #009900;">[</span><span style="color: #0000dd;">3</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> val<span style="color: #339933;">>></span><span style="color: #0000dd;">24</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">void</span> die<span style="color: #009900;">(</span><span style="color: #993333;">const</span> <span style="color: #993333;">char</span> <span style="color: #339933;">*</span>fmt<span style="color: #339933;">,</span> ...<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
va_list ap<span style="color: #339933;">;</span>
va_start<span style="color: #009900;">(</span>ap<span style="color: #339933;">,</span> fmt<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
vprintf<span style="color: #009900;">(</span>fmt<span style="color: #339933;">,</span> ap<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>fmt<span style="color: #009900;">[</span>strlen<span style="color: #009900;">(</span>fmt<span style="color: #009900;">)</span><span style="color: #339933;">-</span><span style="color: #0000dd;">1</span><span style="color: #009900;">]</span> <span style="color: #339933;">!=</span> <span style="color: #ff0000;">'<span style="color: #000099; font-weight: bold;">\n</span>'</span><span style="color: #009900;">)</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
exit<span style="color: #009900;">(</span>EXIT_FAILURE<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">void</span> die_codec<span style="color: #009900;">(</span>vpx_codec_ctx_t <span style="color: #339933;">*</span>ctx<span style="color: #339933;">,</span> <span style="color: #993333;">const</span> <span style="color: #993333;">char</span> <span style="color: #339933;">*</span>s<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
<span style="color: #993333;">const</span> <span style="color: #993333;">char</span> <span style="color: #339933;">*</span>detail <span style="color: #339933;">=</span> vpx_codec_error_detail<span style="color: #009900;">(</span>ctx<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"%s: %s<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span> s<span style="color: #339933;">,</span> vpx_codec_error<span style="color: #009900;">(</span>ctx<span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>detail<span style="color: #009900;">)</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">" %s<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span>detail<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
exit<span style="color: #009900;">(</span>EXIT_FAILURE<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">int</span> read_frame<span style="color: #009900;">(</span>FILE <span style="color: #339933;">*</span>f<span style="color: #339933;">,</span> vpx_image_t <span style="color: #339933;">*</span>img<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
size_t nbytes<span style="color: #339933;">,</span> to_read<span style="color: #339933;">;</span>
<span style="color: #993333;">int</span> res <span style="color: #339933;">=</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">;</span>
to_read <span style="color: #339933;">=</span> img<span style="color: #339933;">-></span>w<span style="color: #339933;">*</span>img<span style="color: #339933;">-></span>h<span style="color: #339933;">*</span><span style="color: #0000dd;">3</span><span style="color: #339933;">/</span><span style="color: #0000dd;">2</span><span style="color: #339933;">;</span>
nbytes <span style="color: #339933;">=</span> fread<span style="color: #009900;">(</span>img<span style="color: #339933;">-></span>planes<span style="color: #009900;">[</span><span style="color: #0000dd;">0</span><span style="color: #009900;">]</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">,</span> to_read<span style="color: #339933;">,</span> f<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>nbytes <span style="color: #339933;">!=</span> to_read<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
res <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>nbytes <span style="color: #339933;">></span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"Warning: Read partial frame. Check your width & height!<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #b1b100;">return</span> res<span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">void</span> write_ivf_file_header<span style="color: #009900;">(</span>FILE <span style="color: #339933;">*</span>outfile<span style="color: #339933;">,</span>
<span style="color: #993333;">const</span> vpx_codec_enc_cfg_t <span style="color: #339933;">*</span>cfg<span style="color: #339933;">,</span>
<span style="color: #993333;">int</span> frame_cnt<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
<span style="color: #993333;">char</span> header<span style="color: #009900;">[</span><span style="color: #0000dd;">32</span><span style="color: #009900;">]</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>cfg<span style="color: #339933;">-></span>g_pass <span style="color: #339933;">!=</span> VPX_RC_ONE_PASS <span style="color: #339933;">&&</span> cfg<span style="color: #339933;">-></span>g_pass <span style="color: #339933;">!=</span> VPX_RC_LAST_PASS<span style="color: #009900;">)</span>
<span style="color: #b1b100;">return</span><span style="color: #339933;">;</span>
header<span style="color: #009900;">[</span><span style="color: #0000dd;">0</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> <span style="color: #ff0000;">'D'</span><span style="color: #339933;">;</span>
header<span style="color: #009900;">[</span><span style="color: #0000dd;">1</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> <span style="color: #ff0000;">'K'</span><span style="color: #339933;">;</span>
header<span style="color: #009900;">[</span><span style="color: #0000dd;">2</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> <span style="color: #ff0000;">'I'</span><span style="color: #339933;">;</span>
header<span style="color: #009900;">[</span><span style="color: #0000dd;">3</span><span style="color: #009900;">]</span> <span style="color: #339933;">=</span> <span style="color: #ff0000;">'F'</span><span style="color: #339933;">;</span>
mem_put_le16<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">4</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* version */</span>
mem_put_le16<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">6</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">32</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* headersize */</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">8</span><span style="color: #339933;">,</span> fourcc<span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* headersize */</span>
mem_put_le16<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">12</span><span style="color: #339933;">,</span> cfg<span style="color: #339933;">-></span>g_w<span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* width */</span>
mem_put_le16<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">14</span><span style="color: #339933;">,</span> cfg<span style="color: #339933;">-></span>g_h<span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* height */</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">16</span><span style="color: #339933;">,</span> cfg<span style="color: #339933;">-></span>g_timebase.<span style="color: #202020;">den</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* rate */</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">20</span><span style="color: #339933;">,</span> cfg<span style="color: #339933;">-></span>g_timebase.<span style="color: #202020;">num</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* scale */</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">24</span><span style="color: #339933;">,</span> frame_cnt<span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* length */</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">28</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #808080; font-style: italic;">/* unused */</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>fwrite<span style="color: #009900;">(</span>header<span style="color: #339933;">,</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">32</span><span style="color: #339933;">,</span> outfile<span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">static</span> <span style="color: #993333;">void</span> write_ivf_frame_header<span style="color: #009900;">(</span>FILE <span style="color: #339933;">*</span>outfile<span style="color: #339933;">,</span>
<span style="color: #993333;">const</span> vpx_codec_cx_pkt_t <span style="color: #339933;">*</span>pkt<span style="color: #009900;">)</span>
<span style="color: #009900;">{</span>
<span style="color: #993333;">char</span> header<span style="color: #009900;">[</span><span style="color: #0000dd;">12</span><span style="color: #009900;">]</span><span style="color: #339933;">;</span>
vpx_codec_pts_t pts<span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>pkt<span style="color: #339933;">-></span>kind <span style="color: #339933;">!=</span> VPX_CODEC_CX_FRAME_PKT<span style="color: #009900;">)</span>
<span style="color: #b1b100;">return</span><span style="color: #339933;">;</span>
pts <span style="color: #339933;">=</span> pkt<span style="color: #339933;">-></span>data.<span style="color: #202020;">frame</span>.<span style="color: #202020;">pts</span><span style="color: #339933;">;</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">,</span> pkt<span style="color: #339933;">-></span>data.<span style="color: #202020;">frame</span>.<span style="color: #202020;">sz</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">4</span><span style="color: #339933;">,</span> pts<span style="color: #339933;">&</span><span style="color: #208080;">0xFFFFFFFF</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
mem_put_le32<span style="color: #009900;">(</span>header<span style="color: #339933;">+</span><span style="color: #0000dd;">8</span><span style="color: #339933;">,</span> pts <span style="color: #339933;">>></span> <span style="color: #0000dd;">32</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>fwrite<span style="color: #009900;">(</span>header<span style="color: #339933;">,</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">12</span><span style="color: #339933;">,</span> outfile<span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #993333;">int</span> main<span style="color: #009900;">(</span><span style="color: #993333;">int</span> argc<span style="color: #339933;">,</span> <span style="color: #993333;">char</span> <span style="color: #339933;">**</span>argv<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
FILE <span style="color: #339933;">*</span>infile<span style="color: #339933;">,</span> <span style="color: #339933;">*</span>outfile<span style="color: #339933;">;</span>
vpx_codec_ctx_t codec<span style="color: #339933;">;</span>
vpx_codec_enc_cfg_t cfg<span style="color: #339933;">;</span>
<span style="color: #993333;">int</span> frame_cnt <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span>
<span style="color: #993333;">unsigned</span> <span style="color: #993333;">char</span> file_hdr<span style="color: #009900;">[</span>IVF_FILE_HDR_SZ<span style="color: #009900;">]</span><span style="color: #339933;">;</span>
<span style="color: #993333;">unsigned</span> <span style="color: #993333;">char</span> frame_hdr<span style="color: #009900;">[</span>IVF_FRAME_HDR_SZ<span style="color: #009900;">]</span><span style="color: #339933;">;</span>
vpx_image_t raw<span style="color: #339933;">;</span>
vpx_codec_err_t res<span style="color: #339933;">;</span>
<span style="color: #993333;">long</span> width<span style="color: #339933;">;</span>
<span style="color: #993333;">long</span> height<span style="color: #339933;">;</span>
<span style="color: #993333;">int</span> frame_avail<span style="color: #339933;">;</span>
<span style="color: #993333;">int</span> got_data<span style="color: #339933;">;</span>
<span style="color: #993333;">int</span> flags <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span>
<span style="color: #993333;">int</span> update_frame_num <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #808080; font-style: italic;">/* Open files */</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>argc<span style="color: #339933;">!=</span><span style="color: #0000dd;">6</span><span style="color: #009900;">)</span> <span style="color: #666666; font-style: italic;">//</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Usage: %s <width> <height> <infile> <outfile> <frame><span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span> <span style="color: #666666; font-style: italic;">//</span>
argv<span style="color: #009900;">[</span><span style="color: #0000dd;">0</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #666666; font-style: italic;">//</span>
update_frame_num <span style="color: #339933;">=</span> atoi<span style="color: #009900;">(</span>argv<span style="color: #009900;">[</span><span style="color: #0000dd;">5</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span><span style="color: #339933;">!</span>update_frame_num<span style="color: #009900;">)</span> <span style="color: #666666; font-style: italic;">//</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Couldn't parse frame number '%s'<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span> argv<span style="color: #009900;">[</span><span style="color: #0000dd;">5</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #666666; font-style: italic;">//</span>
width <span style="color: #339933;">=</span> strtol<span style="color: #009900;">(</span>argv<span style="color: #009900;">[</span><span style="color: #0000dd;">1</span><span style="color: #009900;">]</span><span style="color: #339933;">,</span> NULL<span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
height <span style="color: #339933;">=</span> strtol<span style="color: #009900;">(</span>argv<span style="color: #009900;">[</span><span style="color: #0000dd;">2</span><span style="color: #009900;">]</span><span style="color: #339933;">,</span> NULL<span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>width <span style="color: #339933;"><</span> <span style="color: #0000dd;">16</span> <span style="color: #339933;">||</span> width<span style="color: #339933;">%</span><span style="color:#800080;">2</span> <span style="color: #339933;">||</span> height <span style="color: #339933;"><</span><span style="color: #0000dd;">16</span> <span style="color: #339933;">||</span> height<span style="color: #339933;">%</span><span style="color:#800080;">2</span><span style="color: #009900;">)</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Invalid resolution: %ldx%ld"</span><span style="color: #339933;">,</span> width<span style="color: #339933;">,</span> height<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span><span style="color: #339933;">!</span>vpx_img_alloc<span style="color: #009900;">(</span><span style="color: #339933;">&</span>raw<span style="color: #339933;">,</span> VPX_IMG_FMT_I420<span style="color: #339933;">,</span> width<span style="color: #339933;">,</span> height<span style="color: #339933;">,</span> <span style="color: #0000dd;">1</span><span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Faile to allocate image"</span><span style="color: #339933;">,</span> width<span style="color: #339933;">,</span> height<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span><span style="color: #339933;">!</span><span style="color: #009900;">(</span>outfile <span style="color: #339933;">=</span> fopen<span style="color: #009900;">(</span>argv<span style="color: #009900;">[</span><span style="color: #0000dd;">4</span><span style="color: #009900;">]</span><span style="color: #339933;">,</span> <span style="color: #ff0000;">"wb"</span><span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Failed to open %s for writing"</span><span style="color: #339933;">,</span> argv<span style="color: #009900;">[</span><span style="color: #0000dd;">4</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"Using %s<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span>vpx_codec_iface_name<span style="color: #009900;">(</span>interface<span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #808080; font-style: italic;">/* Populate encoder configuration */</span>
res <span style="color: #339933;">=</span> vpx_codec_enc_config_default<span style="color: #009900;">(</span>interface<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>cfg<span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>res<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"Failed to get config: %s<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span> vpx_codec_err_to_string<span style="color: #009900;">(</span>res<span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">return</span> EXIT_FAILURE<span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<span style="color: #808080; font-style: italic;">/* Update the default configuration with our settings */</span>
cfg.<span style="color: #202020;">rc_target_bitrate</span> <span style="color: #339933;">=</span> width <span style="color: #339933;">*</span> height <span style="color: #339933;">*</span> cfg.<span style="color: #202020;">rc_target_bitrate</span>
<span style="color: #339933;">/</span> cfg.<span style="color: #202020;">g_w</span> <span style="color: #339933;">/</span> cfg.<span style="color: #202020;">g_h</span><span style="color: #339933;">;</span>
cfg.<span style="color: #202020;">g_w</span> <span style="color: #339933;">=</span> width<span style="color: #339933;">;</span>
cfg.<span style="color: #202020;">g_h</span> <span style="color: #339933;">=</span> height<span style="color: #339933;">;</span>
write_ivf_file_header<span style="color: #009900;">(</span>outfile<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>cfg<span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #808080; font-style: italic;">/* Open input file for this encoding pass */</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span><span style="color: #339933;">!</span><span style="color: #009900;">(</span>infile <span style="color: #339933;">=</span> fopen<span style="color: #009900;">(</span>argv<span style="color: #009900;">[</span><span style="color: #0000dd;">3</span><span style="color: #009900;">]</span><span style="color: #339933;">,</span> <span style="color: #ff0000;">"rb"</span><span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die<span style="color: #009900;">(</span><span style="color: #ff0000;">"Failed to open %s for reading"</span><span style="color: #339933;">,</span> argv<span style="color: #009900;">[</span><span style="color: #0000dd;">3</span><span style="color: #009900;">]</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #808080; font-style: italic;">/* Initialize codec */</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>vpx_codec_enc_init<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> interface<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>cfg<span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die_codec<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #ff0000;">"Failed to initialize encoder"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
frame_avail <span style="color: #339933;">=</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">;</span>
got_data <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">while</span><span style="color: #009900;">(</span>frame_avail <span style="color: #339933;">||</span> got_data<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
vpx_codec_iter_t iter <span style="color: #339933;">=</span> NULL<span style="color: #339933;">;</span>
<span style="color: #993333;">const</span> vpx_codec_cx_pkt_t <span style="color: #339933;">*</span>pkt<span style="color: #339933;">;</span>
frame_avail <span style="color: #339933;">=</span> read_frame<span style="color: #009900;">(</span>infile<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>raw<span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #666666; font-style: italic;">//</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>frame_cnt <span style="color: #339933;">+</span> <span style="color: #0000dd;">1</span> <span style="color: #339933;">==</span> update_frame_num<span style="color: #009900;">)</span> <span style="color: #009900;">{</span> <span style="color: #666666; font-style: italic;">//</span>
vpx_ref_frame_t ref<span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #666666; font-style: italic;">//</span>
ref.<span style="color: #202020;">frame_type</span> <span style="color: #339933;">=</span> VP8_LAST_FRAME<span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
ref.<span style="color: #202020;">img</span> <span style="color: #339933;">=</span> raw<span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #666666; font-style: italic;">//</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>vpx_codec_control<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> VP8_SET_REFERENCE<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>ref<span style="color: #009900;">)</span><span style="color: #009900;">)</span> <span style="color: #666666; font-style: italic;">//</span>
die_codec<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #ff0000;">"Failed to set reference frame"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #009900;">}</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #666666; font-style: italic;">//</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>vpx_codec_encode<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> frame_avail<span style="color: #339933;">?</span> <span style="color: #339933;">&</span>raw <span style="color: #339933;">:</span> NULL<span style="color: #339933;">,</span> frame_cnt<span style="color: #339933;">,</span> <span style="color: #666666; font-style: italic;">//</span>
<span style="color: #0000dd;">1</span><span style="color: #339933;">,</span> flags<span style="color: #339933;">,</span> VPX_DL_REALTIME<span style="color: #009900;">)</span><span style="color: #009900;">)</span> <span style="color: #666666; font-style: italic;">//</span>
die_codec<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #ff0000;">"Failed to encode frame"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span> <span style="color: #666666; font-style: italic;">//</span>
got_data <span style="color: #339933;">=</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">while</span><span style="color: #009900;">(</span> <span style="color: #009900;">(</span>pkt <span style="color: #339933;">=</span> vpx_codec_get_cx_data<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>iter<span style="color: #009900;">)</span><span style="color: #009900;">)</span> <span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
got_data <span style="color: #339933;">=</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">switch</span><span style="color: #009900;">(</span>pkt<span style="color: #339933;">-></span>kind<span style="color: #009900;">)</span> <span style="color: #009900;">{</span>
<span style="color: #b1b100;">case</span> VPX_CODEC_CX_FRAME_PKT<span style="color: #339933;">:</span>
write_ivf_frame_header<span style="color: #009900;">(</span>outfile<span style="color: #339933;">,</span> pkt<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>fwrite<span style="color: #009900;">(</span>pkt<span style="color: #339933;">-></span>data.<span style="color: #202020;">frame</span>.<span style="color: #202020;">buf</span><span style="color: #339933;">,</span> <span style="color: #0000dd;">1</span><span style="color: #339933;">,</span> pkt<span style="color: #339933;">-></span>data.<span style="color: #202020;">frame</span>.<span style="color: #202020;">sz</span><span style="color: #339933;">,</span>
outfile<span style="color: #009900;">)</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #000000; font-weight: bold;">break</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">default</span><span style="color: #339933;">:</span>
<span style="color: #000000; font-weight: bold;">break</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span>pkt<span style="color: #339933;">-></span>kind <span style="color: #339933;">==</span> VPX_CODEC_CX_FRAME_PKT
<span style="color: #339933;">&&</span> <span style="color: #009900;">(</span>pkt<span style="color: #339933;">-></span>data.<span style="color: #202020;">frame</span>.<span style="color: #202020;">flags</span> <span style="color: #339933;">&</span> VPX_FRAME_IS_KEY<span style="color: #009900;">)</span><span style="color: #339933;">?</span> <span style="color: #ff0000;">"K"</span><span style="color: #339933;">:</span><span style="color: #ff0000;">"."</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
fflush<span style="color: #009900;">(</span>stdout<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #009900;">}</span>
frame_cnt<span style="color: #339933;">++;</span>
<span style="color: #009900;">}</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
fclose<span style="color: #009900;">(</span>infile<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<a href="http://www.opengroup.org/onlinepubs/009695399/functions/printf.html"><span style="color: #000066;">printf</span></a><span style="color: #009900;">(</span><span style="color: #ff0000;">"Processed %d frames.<span style="color: #000099; font-weight: bold;">\n</span>"</span><span style="color: #339933;">,</span>frame_cnt<span style="color: #339933;">-</span><span style="color: #0000dd;">1</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span>vpx_codec_destroy<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #009900;">)</span><span style="color: #009900;">)</span>
die_codec<span style="color: #009900;">(</span><span style="color: #339933;">&</span>codec<span style="color: #339933;">,</span> <span style="color: #ff0000;">"Failed to destroy codec"</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #808080; font-style: italic;">/* Try to rewrite the file header with the actual frame count */</span>
<span style="color: #b1b100;">if</span><span style="color: #009900;">(</span><span style="color: #339933;">!</span>fseek<span style="color: #009900;">(</span>outfile<span style="color: #339933;">,</span> <span style="color: #0000dd;">0</span><span style="color: #339933;">,</span> SEEK_SET<span style="color: #009900;">)</span><span style="color: #009900;">)</span>
write_ivf_file_header<span style="color: #009900;">(</span>outfile<span style="color: #339933;">,</span> <span style="color: #339933;">&</span>cfg<span style="color: #339933;">,</span> frame_cnt<span style="color: #339933;">-</span><span style="color: #0000dd;">1</span><span style="color: #009900;">)</span><span style="color: #339933;">;</span>
fclose<span style="color: #009900;">(</span>outfile<span style="color: #009900;">)</span><span style="color: #339933;">;</span>
<span style="color: #b1b100;">return</span> EXIT_SUCCESS<span style="color: #339933;">;</span>
<span style="color: #009900;">}</span></pre>
</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>
