2 * Copyright (c) 2010 The WebM project authors. All Rights Reserved.
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
11 #include "vpx_config.h"
13 #if defined(_WIN32) || defined(__OS2__) || !CONFIG_OS_SUPPORT
14 #define USE_POSIX_MMAP 0
16 #define USE_POSIX_MMAP 1
25 #include "vpx/vpx_encoder.h"
27 #include "vpx/vpx_decoder.h"
30 #include <sys/types.h>
37 #if CONFIG_VP8_ENCODER || CONFIG_VP9_ENCODER
38 #include "vpx/vp8cx.h"
40 #if CONFIG_VP8_DECODER || CONFIG_VP9_DECODER
41 #include "vpx/vp8dx.h"
44 #include "vpx_ports/mem_ops.h"
45 #include "vpx_ports/vpx_timer.h"
46 #include "tools_common.h"
48 #include "libmkv/EbmlWriter.h"
49 #include "libmkv/EbmlIDs.h"
50 #include "third_party/libyuv/include/libyuv/scale.h"
52 /* Need special handling of these functions on Windows */
54 /* MSVS doesn't define off_t, and uses _f{seek,tell}i64 */
55 typedef __int64 off_t
;
56 #define fseeko _fseeki64
57 #define ftello _ftelli64
59 /* MinGW defines off_t as long
60 and uses f{seek,tell}o64/off64_t for large files */
61 #define fseeko fseeko64
62 #define ftello ftello64
66 #define LITERALU64(hi,lo) ((((uint64_t)hi)<<32)|lo)
68 /* We should use 32-bit file operations in WebM file format
69 * when building ARM executable file (.axf) with RVCT */
70 #if !CONFIG_OS_SUPPORT
76 /* Swallow warnings about unused results of fread/fwrite */
77 static size_t wrap_fread(void *ptr
, size_t size
, size_t nmemb
,
79 return fread(ptr
, size
, nmemb
, stream
);
81 #define fread wrap_fread
83 static size_t wrap_fwrite(const void *ptr
, size_t size
, size_t nmemb
,
85 return fwrite(ptr
, size
, nmemb
, stream
);
87 #define fwrite wrap_fwrite
90 static const char *exec_name
;
92 #define VP8_FOURCC (0x30385056)
93 #define VP9_FOURCC (0x30395056)
94 static const struct codec_item
{
96 const vpx_codec_iface_t
*(*iface
)(void);
97 const vpx_codec_iface_t
*(*dx_iface
)(void);
100 #if CONFIG_VP8_ENCODER && CONFIG_VP8_DECODER
101 {"vp8", &vpx_codec_vp8_cx
, &vpx_codec_vp8_dx
, VP8_FOURCC
},
102 #elif CONFIG_VP8_ENCODER && !CONFIG_VP8_DECODER
103 {"vp8", &vpx_codec_vp8_cx
, NULL
, VP8_FOURCC
},
105 #if CONFIG_VP9_ENCODER && CONFIG_VP9_DECODER
106 {"vp9", &vpx_codec_vp9_cx
, &vpx_codec_vp9_dx
, VP9_FOURCC
},
107 #elif CONFIG_VP9_ENCODER && !CONFIG_VP9_DECODER
108 {"vp9", &vpx_codec_vp9_cx
, NULL
, VP9_FOURCC
},
112 static void usage_exit();
114 #define LOG_ERROR(label) do \
116 const char *l=label;\
120 fprintf(stderr, "%s: ", l);\
121 vfprintf(stderr, fmt, ap);\
122 fprintf(stderr, "\n");\
126 void die(const char *fmt
, ...) {
132 void fatal(const char *fmt
, ...) {
138 void warn(const char *fmt
, ...) {
139 LOG_ERROR("Warning");
143 static void warn_or_exit_on_errorv(vpx_codec_ctx_t
*ctx
, int fatal
,
144 const char *s
, va_list ap
) {
146 const char *detail
= vpx_codec_error_detail(ctx
);
148 vfprintf(stderr
, s
, ap
);
149 fprintf(stderr
, ": %s\n", vpx_codec_error(ctx
));
152 fprintf(stderr
, " %s\n", detail
);
159 static void ctx_exit_on_error(vpx_codec_ctx_t
*ctx
, const char *s
, ...) {
163 warn_or_exit_on_errorv(ctx
, 1, s
, ap
);
167 static void warn_or_exit_on_error(vpx_codec_ctx_t
*ctx
, int fatal
,
168 const char *s
, ...) {
172 warn_or_exit_on_errorv(ctx
, fatal
, s
, ap
);
176 /* This structure is used to abstract the different ways of handling
177 * first pass statistics.
187 int stats_open_file(stats_io_t
*stats
, const char *fpf
, int pass
) {
193 stats
->file
= fopen(fpf
, "wb");
195 stats
->buf
.buf
= NULL
,
196 res
= (stats
->file
!= NULL
);
200 struct stat stat_buf
;
203 fd
= open(fpf
, O_RDONLY
);
204 stats
->file
= fdopen(fd
, "rb");
205 fstat(fd
, &stat_buf
);
206 stats
->buf
.sz
= stat_buf
.st_size
;
207 stats
->buf
.buf
= mmap(NULL
, stats
->buf
.sz
, PROT_READ
, MAP_PRIVATE
,
209 res
= (stats
->buf
.buf
!= NULL
);
213 stats
->file
= fopen(fpf
, "rb");
215 if (fseek(stats
->file
, 0, SEEK_END
))
216 fatal("First-pass stats file must be seekable!");
218 stats
->buf
.sz
= stats
->buf_alloc_sz
= ftell(stats
->file
);
221 stats
->buf
.buf
= malloc(stats
->buf_alloc_sz
);
224 fatal("Failed to allocate first-pass stats buffer (%lu bytes)",
225 (unsigned long)stats
->buf_alloc_sz
);
227 nbytes
= fread(stats
->buf
.buf
, 1, stats
->buf
.sz
, stats
->file
);
228 res
= (nbytes
== stats
->buf
.sz
);
235 int stats_open_mem(stats_io_t
*stats
, int pass
) {
241 stats
->buf_alloc_sz
= 64 * 1024;
242 stats
->buf
.buf
= malloc(stats
->buf_alloc_sz
);
245 stats
->buf_ptr
= stats
->buf
.buf
;
246 res
= (stats
->buf
.buf
!= NULL
);
251 void stats_close(stats_io_t
*stats
, int last_pass
) {
253 if (stats
->pass
== last_pass
) {
256 munmap(stats
->buf
.buf
, stats
->buf
.sz
);
258 free(stats
->buf
.buf
);
265 if (stats
->pass
== last_pass
)
266 free(stats
->buf
.buf
);
270 void stats_write(stats_io_t
*stats
, const void *pkt
, size_t len
) {
272 (void) fwrite(pkt
, 1, len
, stats
->file
);
274 if (stats
->buf
.sz
+ len
> stats
->buf_alloc_sz
) {
275 size_t new_sz
= stats
->buf_alloc_sz
+ 64 * 1024;
276 char *new_ptr
= realloc(stats
->buf
.buf
, new_sz
);
279 stats
->buf_ptr
= new_ptr
+ (stats
->buf_ptr
- (char *)stats
->buf
.buf
);
280 stats
->buf
.buf
= new_ptr
;
281 stats
->buf_alloc_sz
= new_sz
;
283 fatal("Failed to realloc firstpass stats buffer.");
286 memcpy(stats
->buf_ptr
, pkt
, len
);
287 stats
->buf
.sz
+= len
;
288 stats
->buf_ptr
+= len
;
292 vpx_fixed_buf_t
stats_get(stats_io_t
*stats
) {
296 /* Stereo 3D packed frame format */
297 typedef enum stereo_format
{
298 STEREO_FORMAT_MONO
= 0,
299 STEREO_FORMAT_LEFT_RIGHT
= 1,
300 STEREO_FORMAT_BOTTOM_TOP
= 2,
301 STEREO_FORMAT_TOP_BOTTOM
= 3,
302 STEREO_FORMAT_RIGHT_LEFT
= 11
305 enum video_file_type
{
311 struct detect_buffer
{
323 struct detect_buffer detect
;
324 enum video_file_type file_type
;
327 struct vpx_rational framerate
;
333 #define IVF_FRAME_HDR_SZ (4+8) /* 4 byte size + 8 byte timestamp */
334 static int read_frame(struct input_state
*input
, vpx_image_t
*img
) {
335 FILE *f
= input
->file
;
336 enum video_file_type file_type
= input
->file_type
;
337 y4m_input
*y4m
= &input
->y4m
;
338 struct detect_buffer
*detect
= &input
->detect
;
342 if (file_type
== FILE_TYPE_Y4M
) {
343 if (y4m_input_fetch_frame(y4m
, f
, img
) < 1)
346 if (file_type
== FILE_TYPE_IVF
) {
347 char junk
[IVF_FRAME_HDR_SZ
];
349 /* Skip the frame header. We know how big the frame should be. See
350 * write_ivf_frame_header() for documentation on the frame header
353 (void) fread(junk
, 1, IVF_FRAME_HDR_SZ
, f
);
356 for (plane
= 0; plane
< 3; plane
++) {
358 int w
= (plane
? (1 + img
->d_w
) / 2 : img
->d_w
);
359 int h
= (plane
? (1 + img
->d_h
) / 2 : img
->d_h
);
362 /* Determine the correct plane based on the image format. The for-loop
363 * always counts in Y,U,V order, but this may not match the order of
368 ptr
= img
->planes
[img
->fmt
== VPX_IMG_FMT_YV12
? VPX_PLANE_V
: VPX_PLANE_U
];
371 ptr
= img
->planes
[img
->fmt
== VPX_IMG_FMT_YV12
? VPX_PLANE_U
: VPX_PLANE_V
];
374 ptr
= img
->planes
[plane
];
377 for (r
= 0; r
< h
; r
++) {
379 size_t buf_position
= 0;
380 const size_t left
= detect
->buf_read
- detect
->position
;
382 const size_t more
= (left
< needed
) ? left
: needed
;
383 memcpy(ptr
, detect
->buf
+ detect
->position
, more
);
386 detect
->position
+= more
;
389 shortread
|= (fread(ptr
+ buf_position
, 1, needed
, f
) < needed
);
392 ptr
+= img
->stride
[plane
];
401 unsigned int file_is_y4m(FILE *infile
,
404 if (memcmp(detect
, "YUV4", 4) == 0) {
410 #define IVF_FILE_HDR_SZ (32)
411 unsigned int file_is_ivf(struct input_state
*input
,
412 unsigned int *fourcc
) {
413 char raw_hdr
[IVF_FILE_HDR_SZ
];
415 FILE *infile
= input
->file
;
416 unsigned int *width
= &input
->w
;
417 unsigned int *height
= &input
->h
;
418 struct detect_buffer
*detect
= &input
->detect
;
420 if (memcmp(detect
->buf
, "DKIF", 4) != 0)
423 /* See write_ivf_file_header() for more documentation on the file header
426 if (fread(raw_hdr
+ 4, 1, IVF_FILE_HDR_SZ
- 4, infile
)
427 == IVF_FILE_HDR_SZ
- 4) {
431 if (mem_get_le16(raw_hdr
+ 4) != 0)
432 warn("Unrecognized IVF version! This file may not decode "
435 *fourcc
= mem_get_le32(raw_hdr
+ 8);
440 *width
= mem_get_le16(raw_hdr
+ 12);
441 *height
= mem_get_le16(raw_hdr
+ 14);
442 detect
->position
= 4;
449 static void write_ivf_file_header(FILE *outfile
,
450 const vpx_codec_enc_cfg_t
*cfg
,
455 if (cfg
->g_pass
!= VPX_RC_ONE_PASS
&& cfg
->g_pass
!= VPX_RC_LAST_PASS
)
462 mem_put_le16(header
+ 4, 0); /* version */
463 mem_put_le16(header
+ 6, 32); /* headersize */
464 mem_put_le32(header
+ 8, fourcc
); /* headersize */
465 mem_put_le16(header
+ 12, cfg
->g_w
); /* width */
466 mem_put_le16(header
+ 14, cfg
->g_h
); /* height */
467 mem_put_le32(header
+ 16, cfg
->g_timebase
.den
); /* rate */
468 mem_put_le32(header
+ 20, cfg
->g_timebase
.num
); /* scale */
469 mem_put_le32(header
+ 24, frame_cnt
); /* length */
470 mem_put_le32(header
+ 28, 0); /* unused */
472 (void) fwrite(header
, 1, 32, outfile
);
476 static void write_ivf_frame_header(FILE *outfile
,
477 const vpx_codec_cx_pkt_t
*pkt
) {
481 if (pkt
->kind
!= VPX_CODEC_CX_FRAME_PKT
)
484 pts
= pkt
->data
.frame
.pts
;
485 mem_put_le32(header
, (int)pkt
->data
.frame
.sz
);
486 mem_put_le32(header
+ 4, pts
& 0xFFFFFFFF);
487 mem_put_le32(header
+ 8, pts
>> 32);
489 (void) fwrite(header
, 1, 12, outfile
);
492 static void write_ivf_frame_size(FILE *outfile
, size_t size
) {
494 mem_put_le32(header
, (int)size
);
495 (void) fwrite(header
, 1, 4, outfile
);
499 typedef off_t EbmlLoc
;
513 vpx_rational_t framerate
;
515 /* These pointers are to the start of an element */
516 off_t position_reference
;
518 off_t segment_info_pos
;
523 /* This pointer is to a specific element to be serialized */
526 /* These pointers are to the size field of the element */
527 EbmlLoc startSegment
;
528 EbmlLoc startCluster
;
530 uint32_t cluster_timecode
;
533 struct cue_entry
*cue_list
;
539 void Ebml_Write(EbmlGlobal
*glob
, const void *buffer_in
, unsigned long len
) {
540 (void) fwrite(buffer_in
, 1, len
, glob
->stream
);
543 #define WRITE_BUFFER(s) \
544 for(i = len-1; i>=0; i--)\
546 x = (char)(*(const s *)buffer_in >> (i * CHAR_BIT)); \
547 Ebml_Write(glob, &x, 1); \
549 void Ebml_Serialize(EbmlGlobal
*glob
, const void *buffer_in
, int buffer_size
, unsigned long len
) {
559 switch (buffer_size
) {
564 WRITE_BUFFER(int16_t)
567 WRITE_BUFFER(int32_t)
570 WRITE_BUFFER(int64_t)
578 /* Need a fixed size serializer for the track ID. libmkv provides a 64 bit
579 * one, but not a 32 bit one.
581 static void Ebml_SerializeUnsigned32(EbmlGlobal
*glob
, unsigned long class_id
, uint64_t ui
) {
582 unsigned char sizeSerialized
= 4 | 0x80;
583 Ebml_WriteID(glob
, class_id
);
584 Ebml_Serialize(glob
, &sizeSerialized
, sizeof(sizeSerialized
), 1);
585 Ebml_Serialize(glob
, &ui
, sizeof(ui
), 4);
590 Ebml_StartSubElement(EbmlGlobal
*glob
, EbmlLoc
*ebmlLoc
,
591 unsigned long class_id
) {
592 /* todo this is always taking 8 bytes, this may need later optimization */
593 /* this is a key that says length unknown */
594 uint64_t unknownLen
= LITERALU64(0x01FFFFFF, 0xFFFFFFFF);
596 Ebml_WriteID(glob
, class_id
);
597 *ebmlLoc
= ftello(glob
->stream
);
598 Ebml_Serialize(glob
, &unknownLen
, sizeof(unknownLen
), 8);
602 Ebml_EndSubElement(EbmlGlobal
*glob
, EbmlLoc
*ebmlLoc
) {
606 /* Save the current stream pointer */
607 pos
= ftello(glob
->stream
);
609 /* Calculate the size of this element */
610 size
= pos
- *ebmlLoc
- 8;
611 size
|= LITERALU64(0x01000000, 0x00000000);
613 /* Seek back to the beginning of the element and write the new size */
614 fseeko(glob
->stream
, *ebmlLoc
, SEEK_SET
);
615 Ebml_Serialize(glob
, &size
, sizeof(size
), 8);
617 /* Reset the stream pointer */
618 fseeko(glob
->stream
, pos
, SEEK_SET
);
623 write_webm_seek_element(EbmlGlobal
*ebml
, unsigned long id
, off_t pos
) {
624 uint64_t offset
= pos
- ebml
->position_reference
;
626 Ebml_StartSubElement(ebml
, &start
, Seek
);
627 Ebml_SerializeBinary(ebml
, SeekID
, id
);
628 Ebml_SerializeUnsigned64(ebml
, SeekPosition
, offset
);
629 Ebml_EndSubElement(ebml
, &start
);
634 write_webm_seek_info(EbmlGlobal
*ebml
) {
638 /* Save the current stream pointer */
639 pos
= ftello(ebml
->stream
);
641 if (ebml
->seek_info_pos
)
642 fseeko(ebml
->stream
, ebml
->seek_info_pos
, SEEK_SET
);
644 ebml
->seek_info_pos
= pos
;
649 Ebml_StartSubElement(ebml
, &start
, SeekHead
);
650 write_webm_seek_element(ebml
, Tracks
, ebml
->track_pos
);
651 write_webm_seek_element(ebml
, Cues
, ebml
->cue_pos
);
652 write_webm_seek_element(ebml
, Info
, ebml
->segment_info_pos
);
653 Ebml_EndSubElement(ebml
, &start
);
659 char version_string
[64];
661 /* Assemble version string */
663 strcpy(version_string
, "vpxenc");
665 strcpy(version_string
, "vpxenc ");
666 strncat(version_string
,
667 vpx_codec_version_str(),
668 sizeof(version_string
) - 1 - strlen(version_string
));
671 frame_time
= (uint64_t)1000 * ebml
->framerate
.den
672 / ebml
->framerate
.num
;
673 ebml
->segment_info_pos
= ftello(ebml
->stream
);
674 Ebml_StartSubElement(ebml
, &startInfo
, Info
);
675 Ebml_SerializeUnsigned(ebml
, TimecodeScale
, 1000000);
676 Ebml_SerializeFloat(ebml
, Segment_Duration
,
677 (double)(ebml
->last_pts_ms
+ frame_time
));
678 Ebml_SerializeString(ebml
, 0x4D80, version_string
);
679 Ebml_SerializeString(ebml
, 0x5741, version_string
);
680 Ebml_EndSubElement(ebml
, &startInfo
);
686 write_webm_file_header(EbmlGlobal
*glob
,
687 const vpx_codec_enc_cfg_t
*cfg
,
688 const struct vpx_rational
*fps
,
689 stereo_format_t stereo_fmt
,
690 unsigned int fourcc
) {
693 Ebml_StartSubElement(glob
, &start
, EBML
);
694 Ebml_SerializeUnsigned(glob
, EBMLVersion
, 1);
695 Ebml_SerializeUnsigned(glob
, EBMLReadVersion
, 1);
696 Ebml_SerializeUnsigned(glob
, EBMLMaxIDLength
, 4);
697 Ebml_SerializeUnsigned(glob
, EBMLMaxSizeLength
, 8);
698 Ebml_SerializeString(glob
, DocType
, "webm");
699 Ebml_SerializeUnsigned(glob
, DocTypeVersion
, 2);
700 Ebml_SerializeUnsigned(glob
, DocTypeReadVersion
, 2);
701 Ebml_EndSubElement(glob
, &start
);
704 Ebml_StartSubElement(glob
, &glob
->startSegment
, Segment
);
705 glob
->position_reference
= ftello(glob
->stream
);
706 glob
->framerate
= *fps
;
707 write_webm_seek_info(glob
);
711 glob
->track_pos
= ftello(glob
->stream
);
712 Ebml_StartSubElement(glob
, &trackStart
, Tracks
);
714 unsigned int trackNumber
= 1;
715 uint64_t trackID
= 0;
718 Ebml_StartSubElement(glob
, &start
, TrackEntry
);
719 Ebml_SerializeUnsigned(glob
, TrackNumber
, trackNumber
);
720 glob
->track_id_pos
= ftello(glob
->stream
);
721 Ebml_SerializeUnsigned32(glob
, TrackUID
, trackID
);
722 Ebml_SerializeUnsigned(glob
, TrackType
, 1);
723 Ebml_SerializeString(glob
, CodecID
,
724 fourcc
== VP8_FOURCC
? "V_VP8" : "V_VP9");
726 unsigned int pixelWidth
= cfg
->g_w
;
727 unsigned int pixelHeight
= cfg
->g_h
;
730 Ebml_StartSubElement(glob
, &videoStart
, Video
);
731 Ebml_SerializeUnsigned(glob
, PixelWidth
, pixelWidth
);
732 Ebml_SerializeUnsigned(glob
, PixelHeight
, pixelHeight
);
733 Ebml_SerializeUnsigned(glob
, StereoMode
, stereo_fmt
);
734 Ebml_EndSubElement(glob
, &videoStart
);
736 Ebml_EndSubElement(glob
, &start
); /* Track Entry */
738 Ebml_EndSubElement(glob
, &trackStart
);
740 /* segment element is open */
746 write_webm_block(EbmlGlobal
*glob
,
747 const vpx_codec_enc_cfg_t
*cfg
,
748 const vpx_codec_cx_pkt_t
*pkt
) {
749 unsigned long block_length
;
750 unsigned char track_number
;
751 unsigned short block_timecode
= 0;
754 int start_cluster
= 0, is_keyframe
;
756 /* Calculate the PTS of this frame in milliseconds */
757 pts_ms
= pkt
->data
.frame
.pts
* 1000
758 * (uint64_t)cfg
->g_timebase
.num
/ (uint64_t)cfg
->g_timebase
.den
;
759 if (pts_ms
<= glob
->last_pts_ms
)
760 pts_ms
= glob
->last_pts_ms
+ 1;
761 glob
->last_pts_ms
= pts_ms
;
763 /* Calculate the relative time of this block */
764 if (pts_ms
- glob
->cluster_timecode
> SHRT_MAX
)
767 block_timecode
= (unsigned short)pts_ms
- glob
->cluster_timecode
;
769 is_keyframe
= (pkt
->data
.frame
.flags
& VPX_FRAME_IS_KEY
);
770 if (start_cluster
|| is_keyframe
) {
771 if (glob
->cluster_open
)
772 Ebml_EndSubElement(glob
, &glob
->startCluster
);
774 /* Open the new cluster */
776 glob
->cluster_open
= 1;
777 glob
->cluster_timecode
= (uint32_t)pts_ms
;
778 glob
->cluster_pos
= ftello(glob
->stream
);
779 Ebml_StartSubElement(glob
, &glob
->startCluster
, Cluster
); /* cluster */
780 Ebml_SerializeUnsigned(glob
, Timecode
, glob
->cluster_timecode
);
782 /* Save a cue point if this is a keyframe. */
784 struct cue_entry
*cue
, *new_cue_list
;
786 new_cue_list
= realloc(glob
->cue_list
,
787 (glob
->cues
+ 1) * sizeof(struct cue_entry
));
789 glob
->cue_list
= new_cue_list
;
791 fatal("Failed to realloc cue list.");
793 cue
= &glob
->cue_list
[glob
->cues
];
794 cue
->time
= glob
->cluster_timecode
;
795 cue
->loc
= glob
->cluster_pos
;
800 /* Write the Simple Block */
801 Ebml_WriteID(glob
, SimpleBlock
);
803 block_length
= (unsigned long)pkt
->data
.frame
.sz
+ 4;
804 block_length
|= 0x10000000;
805 Ebml_Serialize(glob
, &block_length
, sizeof(block_length
), 4);
808 track_number
|= 0x80;
809 Ebml_Write(glob
, &track_number
, 1);
811 Ebml_Serialize(glob
, &block_timecode
, sizeof(block_timecode
), 2);
816 if (pkt
->data
.frame
.flags
& VPX_FRAME_IS_INVISIBLE
)
818 Ebml_Write(glob
, &flags
, 1);
820 Ebml_Write(glob
, pkt
->data
.frame
.buf
, (unsigned long)pkt
->data
.frame
.sz
);
825 write_webm_file_footer(EbmlGlobal
*glob
, long hash
) {
827 if (glob
->cluster_open
)
828 Ebml_EndSubElement(glob
, &glob
->startCluster
);
834 glob
->cue_pos
= ftello(glob
->stream
);
835 Ebml_StartSubElement(glob
, &start
, Cues
);
836 for (i
= 0; i
< glob
->cues
; i
++) {
837 struct cue_entry
*cue
= &glob
->cue_list
[i
];
840 Ebml_StartSubElement(glob
, &start
, CuePoint
);
844 Ebml_SerializeUnsigned(glob
, CueTime
, cue
->time
);
846 Ebml_StartSubElement(glob
, &start
, CueTrackPositions
);
847 Ebml_SerializeUnsigned(glob
, CueTrack
, 1);
848 Ebml_SerializeUnsigned64(glob
, CueClusterPosition
,
849 cue
->loc
- glob
->position_reference
);
850 Ebml_EndSubElement(glob
, &start
);
852 Ebml_EndSubElement(glob
, &start
);
854 Ebml_EndSubElement(glob
, &start
);
857 Ebml_EndSubElement(glob
, &glob
->startSegment
);
859 /* Patch up the seek info block */
860 write_webm_seek_info(glob
);
862 /* Patch up the track id */
863 fseeko(glob
->stream
, glob
->track_id_pos
, SEEK_SET
);
864 Ebml_SerializeUnsigned32(glob
, TrackUID
, glob
->debug
? 0xDEADBEEF : hash
);
866 fseeko(glob
->stream
, 0, SEEK_END
);
870 /* Murmur hash derived from public domain reference implementation at
871 * http:// sites.google.com/site/murmurhash/
873 static unsigned int murmur(const void *key
, int len
, unsigned int seed
) {
874 const unsigned int m
= 0x5bd1e995;
877 unsigned int h
= seed
^ len
;
879 const unsigned char *data
= (const unsigned char *)key
;
884 k
= (unsigned int)data
[0];
885 k
|= (unsigned int)data
[1] << 8;
886 k
|= (unsigned int)data
[2] << 16;
887 k
|= (unsigned int)data
[3] << 24;
919 static double vp8_mse2psnr(double Samples
, double Peak
, double Mse
) {
922 if ((double)Mse
> 0.0)
923 psnr
= 10.0 * log10(Peak
* Peak
* Samples
/ Mse
);
925 psnr
= MAX_PSNR
; /* Limit to prevent / 0 */
935 static const arg_def_t debugmode
= ARG_DEF("D", "debug", 0,
936 "Debug mode (makes output deterministic)");
937 static const arg_def_t outputfile
= ARG_DEF("o", "output", 1,
939 static const arg_def_t use_yv12
= ARG_DEF(NULL
, "yv12", 0,
940 "Input file is YV12 ");
941 static const arg_def_t use_i420
= ARG_DEF(NULL
, "i420", 0,
942 "Input file is I420 (default)");
943 static const arg_def_t codecarg
= ARG_DEF(NULL
, "codec", 1,
945 static const arg_def_t passes
= ARG_DEF("p", "passes", 1,
946 "Number of passes (1/2)");
947 static const arg_def_t pass_arg
= ARG_DEF(NULL
, "pass", 1,
948 "Pass to execute (1/2)");
949 static const arg_def_t fpf_name
= ARG_DEF(NULL
, "fpf", 1,
950 "First pass statistics file name");
951 static const arg_def_t limit
= ARG_DEF(NULL
, "limit", 1,
952 "Stop encoding after n input frames");
953 static const arg_def_t skip
= ARG_DEF(NULL
, "skip", 1,
954 "Skip the first n input frames");
955 static const arg_def_t deadline
= ARG_DEF("d", "deadline", 1,
956 "Deadline per frame (usec)");
957 static const arg_def_t best_dl
= ARG_DEF(NULL
, "best", 0,
958 "Use Best Quality Deadline");
959 static const arg_def_t good_dl
= ARG_DEF(NULL
, "good", 0,
960 "Use Good Quality Deadline");
961 static const arg_def_t rt_dl
= ARG_DEF(NULL
, "rt", 0,
962 "Use Realtime Quality Deadline");
963 static const arg_def_t quietarg
= ARG_DEF("q", "quiet", 0,
964 "Do not print encode progress");
965 static const arg_def_t verbosearg
= ARG_DEF("v", "verbose", 0,
966 "Show encoder parameters");
967 static const arg_def_t psnrarg
= ARG_DEF(NULL
, "psnr", 0,
968 "Show PSNR in status line");
969 enum TestDecodeFatality
{
974 static const struct arg_enum_list test_decode_enum
[] = {
975 {"off", TEST_DECODE_OFF
},
976 {"fatal", TEST_DECODE_FATAL
},
977 {"warn", TEST_DECODE_WARN
},
980 static const arg_def_t recontest
= ARG_DEF_ENUM(NULL
, "test-decode", 1,
981 "Test encode/decode mismatch",
983 static const arg_def_t framerate
= ARG_DEF(NULL
, "fps", 1,
984 "Stream frame rate (rate/scale)");
985 static const arg_def_t use_ivf
= ARG_DEF(NULL
, "ivf", 0,
986 "Output IVF (default is WebM)");
987 static const arg_def_t out_part
= ARG_DEF("P", "output-partitions", 0,
988 "Makes encoder output partitions. Requires IVF output!");
989 static const arg_def_t q_hist_n
= ARG_DEF(NULL
, "q-hist", 1,
990 "Show quantizer histogram (n-buckets)");
991 static const arg_def_t rate_hist_n
= ARG_DEF(NULL
, "rate-hist", 1,
992 "Show rate histogram (n-buckets)");
993 static const arg_def_t
*main_args
[] = {
995 &outputfile
, &codecarg
, &passes
, &pass_arg
, &fpf_name
, &limit
, &skip
,
996 &deadline
, &best_dl
, &good_dl
, &rt_dl
,
997 &quietarg
, &verbosearg
, &psnrarg
, &use_ivf
, &out_part
, &q_hist_n
, &rate_hist_n
,
1001 static const arg_def_t usage
= ARG_DEF("u", "usage", 1,
1002 "Usage profile number to use");
1003 static const arg_def_t threads
= ARG_DEF("t", "threads", 1,
1004 "Max number of threads to use");
1005 static const arg_def_t profile
= ARG_DEF(NULL
, "profile", 1,
1006 "Bitstream profile number to use");
1007 static const arg_def_t width
= ARG_DEF("w", "width", 1,
1009 static const arg_def_t height
= ARG_DEF("h", "height", 1,
1011 static const struct arg_enum_list stereo_mode_enum
[] = {
1012 {"mono", STEREO_FORMAT_MONO
},
1013 {"left-right", STEREO_FORMAT_LEFT_RIGHT
},
1014 {"bottom-top", STEREO_FORMAT_BOTTOM_TOP
},
1015 {"top-bottom", STEREO_FORMAT_TOP_BOTTOM
},
1016 {"right-left", STEREO_FORMAT_RIGHT_LEFT
},
1019 static const arg_def_t stereo_mode
= ARG_DEF_ENUM(NULL
, "stereo-mode", 1,
1020 "Stereo 3D video format", stereo_mode_enum
);
1021 static const arg_def_t timebase
= ARG_DEF(NULL
, "timebase", 1,
1022 "Output timestamp precision (fractional seconds)");
1023 static const arg_def_t error_resilient
= ARG_DEF(NULL
, "error-resilient", 1,
1024 "Enable error resiliency features");
1025 static const arg_def_t lag_in_frames
= ARG_DEF(NULL
, "lag-in-frames", 1,
1026 "Max number of frames to lag");
1028 static const arg_def_t
*global_args
[] = {
1029 &use_yv12
, &use_i420
, &usage
, &threads
, &profile
,
1030 &width
, &height
, &stereo_mode
, &timebase
, &framerate
,
1032 &lag_in_frames
, NULL
1035 static const arg_def_t dropframe_thresh
= ARG_DEF(NULL
, "drop-frame", 1,
1036 "Temporal resampling threshold (buf %)");
1037 static const arg_def_t resize_allowed
= ARG_DEF(NULL
, "resize-allowed", 1,
1038 "Spatial resampling enabled (bool)");
1039 static const arg_def_t resize_up_thresh
= ARG_DEF(NULL
, "resize-up", 1,
1040 "Upscale threshold (buf %)");
1041 static const arg_def_t resize_down_thresh
= ARG_DEF(NULL
, "resize-down", 1,
1042 "Downscale threshold (buf %)");
1043 static const struct arg_enum_list end_usage_enum
[] = {
1050 static const arg_def_t end_usage
= ARG_DEF_ENUM(NULL
, "end-usage", 1,
1051 "Rate control mode", end_usage_enum
);
1052 static const arg_def_t target_bitrate
= ARG_DEF(NULL
, "target-bitrate", 1,
1054 static const arg_def_t min_quantizer
= ARG_DEF(NULL
, "min-q", 1,
1055 "Minimum (best) quantizer");
1056 static const arg_def_t max_quantizer
= ARG_DEF(NULL
, "max-q", 1,
1057 "Maximum (worst) quantizer");
1058 static const arg_def_t undershoot_pct
= ARG_DEF(NULL
, "undershoot-pct", 1,
1059 "Datarate undershoot (min) target (%)");
1060 static const arg_def_t overshoot_pct
= ARG_DEF(NULL
, "overshoot-pct", 1,
1061 "Datarate overshoot (max) target (%)");
1062 static const arg_def_t buf_sz
= ARG_DEF(NULL
, "buf-sz", 1,
1063 "Client buffer size (ms)");
1064 static const arg_def_t buf_initial_sz
= ARG_DEF(NULL
, "buf-initial-sz", 1,
1065 "Client initial buffer size (ms)");
1066 static const arg_def_t buf_optimal_sz
= ARG_DEF(NULL
, "buf-optimal-sz", 1,
1067 "Client optimal buffer size (ms)");
1068 static const arg_def_t
*rc_args
[] = {
1069 &dropframe_thresh
, &resize_allowed
, &resize_up_thresh
, &resize_down_thresh
,
1070 &end_usage
, &target_bitrate
, &min_quantizer
, &max_quantizer
,
1071 &undershoot_pct
, &overshoot_pct
, &buf_sz
, &buf_initial_sz
, &buf_optimal_sz
,
1076 static const arg_def_t bias_pct
= ARG_DEF(NULL
, "bias-pct", 1,
1077 "CBR/VBR bias (0=CBR, 100=VBR)");
1078 static const arg_def_t minsection_pct
= ARG_DEF(NULL
, "minsection-pct", 1,
1079 "GOP min bitrate (% of target)");
1080 static const arg_def_t maxsection_pct
= ARG_DEF(NULL
, "maxsection-pct", 1,
1081 "GOP max bitrate (% of target)");
1082 static const arg_def_t
*rc_twopass_args
[] = {
1083 &bias_pct
, &minsection_pct
, &maxsection_pct
, NULL
1087 static const arg_def_t kf_min_dist
= ARG_DEF(NULL
, "kf-min-dist", 1,
1088 "Minimum keyframe interval (frames)");
1089 static const arg_def_t kf_max_dist
= ARG_DEF(NULL
, "kf-max-dist", 1,
1090 "Maximum keyframe interval (frames)");
1091 static const arg_def_t kf_disabled
= ARG_DEF(NULL
, "disable-kf", 0,
1092 "Disable keyframe placement");
1093 static const arg_def_t
*kf_args
[] = {
1094 &kf_min_dist
, &kf_max_dist
, &kf_disabled
, NULL
1098 static const arg_def_t noise_sens
= ARG_DEF(NULL
, "noise-sensitivity", 1,
1099 "Noise sensitivity (frames to blur)");
1100 static const arg_def_t sharpness
= ARG_DEF(NULL
, "sharpness", 1,
1101 "Filter sharpness (0-7)");
1102 static const arg_def_t static_thresh
= ARG_DEF(NULL
, "static-thresh", 1,
1103 "Motion detection threshold");
1104 static const arg_def_t cpu_used
= ARG_DEF(NULL
, "cpu-used", 1,
1105 "CPU Used (-16..16)");
1106 static const arg_def_t token_parts
= ARG_DEF(NULL
, "token-parts", 1,
1107 "Number of token partitions to use, log2");
1108 static const arg_def_t tile_cols
= ARG_DEF(NULL
, "tile-columns", 1,
1109 "Number of tile columns to use, log2");
1110 static const arg_def_t tile_rows
= ARG_DEF(NULL
, "tile-rows", 1,
1111 "Number of tile rows to use, log2");
1112 static const arg_def_t auto_altref
= ARG_DEF(NULL
, "auto-alt-ref", 1,
1113 "Enable automatic alt reference frames");
1114 static const arg_def_t arnr_maxframes
= ARG_DEF(NULL
, "arnr-maxframes", 1,
1115 "AltRef Max Frames");
1116 static const arg_def_t arnr_strength
= ARG_DEF(NULL
, "arnr-strength", 1,
1118 static const arg_def_t arnr_type
= ARG_DEF(NULL
, "arnr-type", 1,
1120 static const struct arg_enum_list tuning_enum
[] = {
1121 {"psnr", VP8_TUNE_PSNR
},
1122 {"ssim", VP8_TUNE_SSIM
},
1125 static const arg_def_t tune_ssim
= ARG_DEF_ENUM(NULL
, "tune", 1,
1126 "Material to favor", tuning_enum
);
1127 static const arg_def_t cq_level
= ARG_DEF(NULL
, "cq-level", 1,
1128 "Constant/Constrained Quality level");
1129 static const arg_def_t max_intra_rate_pct
= ARG_DEF(NULL
, "max-intra-rate", 1,
1130 "Max I-frame bitrate (pct)");
1131 static const arg_def_t lossless
= ARG_DEF(NULL
, "lossless", 1, "Lossless mode");
1132 #if CONFIG_VP9_ENCODER
1133 static const arg_def_t frame_parallel_decoding
= ARG_DEF(
1134 NULL
, "frame-parallel", 1, "Enable frame parallel decodability features");
1137 #if CONFIG_VP8_ENCODER
1138 static const arg_def_t
*vp8_args
[] = {
1139 &cpu_used
, &auto_altref
, &noise_sens
, &sharpness
, &static_thresh
,
1140 &token_parts
, &arnr_maxframes
, &arnr_strength
, &arnr_type
,
1141 &tune_ssim
, &cq_level
, &max_intra_rate_pct
,
1144 static const int vp8_arg_ctrl_map
[] = {
1145 VP8E_SET_CPUUSED
, VP8E_SET_ENABLEAUTOALTREF
,
1146 VP8E_SET_NOISE_SENSITIVITY
, VP8E_SET_SHARPNESS
, VP8E_SET_STATIC_THRESHOLD
,
1147 VP8E_SET_TOKEN_PARTITIONS
,
1148 VP8E_SET_ARNR_MAXFRAMES
, VP8E_SET_ARNR_STRENGTH
, VP8E_SET_ARNR_TYPE
,
1149 VP8E_SET_TUNING
, VP8E_SET_CQ_LEVEL
, VP8E_SET_MAX_INTRA_BITRATE_PCT
,
1154 #if CONFIG_VP9_ENCODER
1155 static const arg_def_t
*vp9_args
[] = {
1156 &cpu_used
, &auto_altref
, &noise_sens
, &sharpness
, &static_thresh
,
1157 &tile_cols
, &tile_rows
, &arnr_maxframes
, &arnr_strength
, &arnr_type
,
1158 &tune_ssim
, &cq_level
, &max_intra_rate_pct
, &lossless
,
1159 &frame_parallel_decoding
,
1162 static const int vp9_arg_ctrl_map
[] = {
1163 VP8E_SET_CPUUSED
, VP8E_SET_ENABLEAUTOALTREF
,
1164 VP8E_SET_NOISE_SENSITIVITY
, VP8E_SET_SHARPNESS
, VP8E_SET_STATIC_THRESHOLD
,
1165 VP9E_SET_TILE_COLUMNS
, VP9E_SET_TILE_ROWS
,
1166 VP8E_SET_ARNR_MAXFRAMES
, VP8E_SET_ARNR_STRENGTH
, VP8E_SET_ARNR_TYPE
,
1167 VP8E_SET_TUNING
, VP8E_SET_CQ_LEVEL
, VP8E_SET_MAX_INTRA_BITRATE_PCT
,
1168 VP9E_SET_LOSSLESS
, VP9E_SET_FRAME_PARALLEL_DECODING
,
1173 static const arg_def_t
*no_args
[] = { NULL
};
1175 static void usage_exit() {
1178 fprintf(stderr
, "Usage: %s <options> -o dst_filename src_filename \n",
1181 fprintf(stderr
, "\nOptions:\n");
1182 arg_show_usage(stderr
, main_args
);
1183 fprintf(stderr
, "\nEncoder Global Options:\n");
1184 arg_show_usage(stderr
, global_args
);
1185 fprintf(stderr
, "\nRate Control Options:\n");
1186 arg_show_usage(stderr
, rc_args
);
1187 fprintf(stderr
, "\nTwopass Rate Control Options:\n");
1188 arg_show_usage(stderr
, rc_twopass_args
);
1189 fprintf(stderr
, "\nKeyframe Placement Options:\n");
1190 arg_show_usage(stderr
, kf_args
);
1191 #if CONFIG_VP8_ENCODER
1192 fprintf(stderr
, "\nVP8 Specific Options:\n");
1193 arg_show_usage(stderr
, vp8_args
);
1195 #if CONFIG_VP9_ENCODER
1196 fprintf(stderr
, "\nVP9 Specific Options:\n");
1197 arg_show_usage(stderr
, vp9_args
);
1199 fprintf(stderr
, "\nStream timebase (--timebase):\n"
1200 " The desired precision of timestamps in the output, expressed\n"
1201 " in fractional seconds. Default is 1/1000.\n");
1202 fprintf(stderr
, "\n"
1203 "Included encoders:\n"
1206 for (i
= 0; i
< sizeof(codecs
) / sizeof(codecs
[0]); i
++)
1207 fprintf(stderr
, " %-6s - %s\n",
1209 vpx_codec_iface_name(codecs
[i
].iface()));
1215 #define HIST_BAR_MAX 40
1216 struct hist_bucket
{
1217 int low
, high
, count
;
1221 static int merge_hist_buckets(struct hist_bucket
*bucket
,
1224 int small_bucket
= 0, merge_bucket
= INT_MAX
, big_bucket
= 0;
1225 int buckets
= *buckets_
;
1228 /* Find the extrema for this list of buckets */
1229 big_bucket
= small_bucket
= 0;
1230 for (i
= 0; i
< buckets
; i
++) {
1231 if (bucket
[i
].count
< bucket
[small_bucket
].count
)
1233 if (bucket
[i
].count
> bucket
[big_bucket
].count
)
1237 /* If we have too many buckets, merge the smallest with an adjacent
1240 while (buckets
> max_buckets
) {
1241 int last_bucket
= buckets
- 1;
1243 /* merge the small bucket with an adjacent one. */
1244 if (small_bucket
== 0)
1246 else if (small_bucket
== last_bucket
)
1247 merge_bucket
= last_bucket
- 1;
1248 else if (bucket
[small_bucket
- 1].count
< bucket
[small_bucket
+ 1].count
)
1249 merge_bucket
= small_bucket
- 1;
1251 merge_bucket
= small_bucket
+ 1;
1253 assert(abs(merge_bucket
- small_bucket
) <= 1);
1254 assert(small_bucket
< buckets
);
1255 assert(big_bucket
< buckets
);
1256 assert(merge_bucket
< buckets
);
1258 if (merge_bucket
< small_bucket
) {
1259 bucket
[merge_bucket
].high
= bucket
[small_bucket
].high
;
1260 bucket
[merge_bucket
].count
+= bucket
[small_bucket
].count
;
1262 bucket
[small_bucket
].high
= bucket
[merge_bucket
].high
;
1263 bucket
[small_bucket
].count
+= bucket
[merge_bucket
].count
;
1264 merge_bucket
= small_bucket
;
1267 assert(bucket
[merge_bucket
].low
!= bucket
[merge_bucket
].high
);
1271 /* Remove the merge_bucket from the list, and find the new small
1272 * and big buckets while we're at it
1274 big_bucket
= small_bucket
= 0;
1275 for (i
= 0; i
< buckets
; i
++) {
1276 if (i
> merge_bucket
)
1277 bucket
[i
] = bucket
[i
+ 1];
1279 if (bucket
[i
].count
< bucket
[small_bucket
].count
)
1281 if (bucket
[i
].count
> bucket
[big_bucket
].count
)
1287 *buckets_
= buckets
;
1288 return bucket
[big_bucket
].count
;
1292 static void show_histogram(const struct hist_bucket
*bucket
,
1296 const char *pat1
, *pat2
;
1299 switch ((int)(log(bucket
[buckets
- 1].high
) / log(10)) + 1) {
1326 pat1
= "%12d %10s: ";
1327 pat2
= "%12d-%10d: ";
1331 for (i
= 0; i
< buckets
; i
++) {
1336 pct
= (float)(100.0 * bucket
[i
].count
/ total
);
1337 len
= HIST_BAR_MAX
* bucket
[i
].count
/ scale
;
1340 assert(len
<= HIST_BAR_MAX
);
1342 if (bucket
[i
].low
== bucket
[i
].high
)
1343 fprintf(stderr
, pat1
, bucket
[i
].low
, "");
1345 fprintf(stderr
, pat2
, bucket
[i
].low
, bucket
[i
].high
);
1347 for (j
= 0; j
< HIST_BAR_MAX
; j
++)
1348 fprintf(stderr
, j
< len
? "=" : " ");
1349 fprintf(stderr
, "\t%5d (%6.2f%%)\n", bucket
[i
].count
, pct
);
1354 static void show_q_histogram(const int counts
[64], int max_buckets
) {
1355 struct hist_bucket bucket
[64];
1362 for (i
= 0; i
< 64; i
++) {
1364 bucket
[buckets
].low
= bucket
[buckets
].high
= i
;
1365 bucket
[buckets
].count
= counts
[i
];
1371 fprintf(stderr
, "\nQuantizer Selection:\n");
1372 scale
= merge_hist_buckets(bucket
, &buckets
, max_buckets
);
1373 show_histogram(bucket
, buckets
, total
, scale
);
1377 #define RATE_BINS (100)
1383 struct hist_bucket bucket
[RATE_BINS
];
1388 static void init_rate_histogram(struct rate_hist
*hist
,
1389 const vpx_codec_enc_cfg_t
*cfg
,
1390 const vpx_rational_t
*fps
) {
1393 /* Determine the number of samples in the buffer. Use the file's framerate
1394 * to determine the number of frames in rc_buf_sz milliseconds, with an
1395 * adjustment (5/4) to account for alt-refs
1397 hist
->samples
= cfg
->rc_buf_sz
* 5 / 4 * fps
->num
/ fps
->den
/ 1000;
1399 /* prevent division by zero */
1400 if (hist
->samples
== 0)
1403 hist
->pts
= calloc(hist
->samples
, sizeof(*hist
->pts
));
1404 hist
->sz
= calloc(hist
->samples
, sizeof(*hist
->sz
));
1405 for (i
= 0; i
< RATE_BINS
; i
++) {
1406 hist
->bucket
[i
].low
= INT_MAX
;
1407 hist
->bucket
[i
].high
= 0;
1408 hist
->bucket
[i
].count
= 0;
1413 static void destroy_rate_histogram(struct rate_hist
*hist
) {
1419 static void update_rate_histogram(struct rate_hist
*hist
,
1420 const vpx_codec_enc_cfg_t
*cfg
,
1421 const vpx_codec_cx_pkt_t
*pkt
) {
1423 int64_t now
, then
, sum_sz
= 0, avg_bitrate
;
1425 now
= pkt
->data
.frame
.pts
* 1000
1426 * (uint64_t)cfg
->g_timebase
.num
/ (uint64_t)cfg
->g_timebase
.den
;
1428 idx
= hist
->frames
++ % hist
->samples
;
1429 hist
->pts
[idx
] = now
;
1430 hist
->sz
[idx
] = (int)pkt
->data
.frame
.sz
;
1432 if (now
< cfg
->rc_buf_initial_sz
)
1437 /* Sum the size over the past rc_buf_sz ms */
1438 for (i
= hist
->frames
; i
> 0 && hist
->frames
- i
< hist
->samples
; i
--) {
1439 int i_idx
= (i
- 1) % hist
->samples
;
1441 then
= hist
->pts
[i_idx
];
1442 if (now
- then
> cfg
->rc_buf_sz
)
1444 sum_sz
+= hist
->sz
[i_idx
];
1450 avg_bitrate
= sum_sz
* 8 * 1000 / (now
- then
);
1451 idx
= (int)(avg_bitrate
* (RATE_BINS
/ 2) / (cfg
->rc_target_bitrate
* 1000));
1454 if (idx
> RATE_BINS
- 1)
1455 idx
= RATE_BINS
- 1;
1456 if (hist
->bucket
[idx
].low
> avg_bitrate
)
1457 hist
->bucket
[idx
].low
= (int)avg_bitrate
;
1458 if (hist
->bucket
[idx
].high
< avg_bitrate
)
1459 hist
->bucket
[idx
].high
= (int)avg_bitrate
;
1460 hist
->bucket
[idx
].count
++;
1465 static void show_rate_histogram(struct rate_hist
*hist
,
1466 const vpx_codec_enc_cfg_t
*cfg
,
1471 for (i
= 0; i
< RATE_BINS
; i
++) {
1472 if (hist
->bucket
[i
].low
== INT_MAX
)
1474 hist
->bucket
[buckets
++] = hist
->bucket
[i
];
1477 fprintf(stderr
, "\nRate (over %dms window):\n", cfg
->rc_buf_sz
);
1478 scale
= merge_hist_buckets(hist
->bucket
, &buckets
, max_buckets
);
1479 show_histogram(hist
->bucket
, buckets
, hist
->total
, scale
);
1482 #define mmin(a, b) ((a) < (b) ? (a) : (b))
1483 static void find_mismatch(vpx_image_t
*img1
, vpx_image_t
*img2
,
1484 int yloc
[4], int uloc
[4], int vloc
[4]) {
1485 const unsigned int bsize
= 64;
1486 const unsigned int bsizey
= bsize
>> img1
->y_chroma_shift
;
1487 const unsigned int bsizex
= bsize
>> img1
->x_chroma_shift
;
1488 const int c_w
= (img1
->d_w
+ img1
->x_chroma_shift
) >> img1
->x_chroma_shift
;
1489 const int c_h
= (img1
->d_h
+ img1
->y_chroma_shift
) >> img1
->y_chroma_shift
;
1490 unsigned int match
= 1;
1492 yloc
[0] = yloc
[1] = yloc
[2] = yloc
[3] = -1;
1493 for (i
= 0, match
= 1; match
&& i
< img1
->d_h
; i
+= bsize
) {
1494 for (j
= 0; match
&& j
< img1
->d_w
; j
+= bsize
) {
1496 int si
= mmin(i
+ bsize
, img1
->d_h
) - i
;
1497 int sj
= mmin(j
+ bsize
, img1
->d_w
) - j
;
1498 for (k
= 0; match
&& k
< si
; k
++)
1499 for (l
= 0; match
&& l
< sj
; l
++) {
1500 if (*(img1
->planes
[VPX_PLANE_Y
] +
1501 (i
+ k
) * img1
->stride
[VPX_PLANE_Y
] + j
+ l
) !=
1502 *(img2
->planes
[VPX_PLANE_Y
] +
1503 (i
+ k
) * img2
->stride
[VPX_PLANE_Y
] + j
+ l
)) {
1506 yloc
[2] = *(img1
->planes
[VPX_PLANE_Y
] +
1507 (i
+ k
) * img1
->stride
[VPX_PLANE_Y
] + j
+ l
);
1508 yloc
[3] = *(img2
->planes
[VPX_PLANE_Y
] +
1509 (i
+ k
) * img2
->stride
[VPX_PLANE_Y
] + j
+ l
);
1517 uloc
[0] = uloc
[1] = uloc
[2] = uloc
[3] = -1;
1518 for (i
= 0, match
= 1; match
&& i
< c_h
; i
+= bsizey
) {
1519 for (j
= 0; match
&& j
< c_w
; j
+= bsizex
) {
1521 int si
= mmin(i
+ bsizey
, c_h
- i
);
1522 int sj
= mmin(j
+ bsizex
, c_w
- j
);
1523 for (k
= 0; match
&& k
< si
; k
++)
1524 for (l
= 0; match
&& l
< sj
; l
++) {
1525 if (*(img1
->planes
[VPX_PLANE_U
] +
1526 (i
+ k
) * img1
->stride
[VPX_PLANE_U
] + j
+ l
) !=
1527 *(img2
->planes
[VPX_PLANE_U
] +
1528 (i
+ k
) * img2
->stride
[VPX_PLANE_U
] + j
+ l
)) {
1531 uloc
[2] = *(img1
->planes
[VPX_PLANE_U
] +
1532 (i
+ k
) * img1
->stride
[VPX_PLANE_U
] + j
+ l
);
1533 uloc
[3] = *(img2
->planes
[VPX_PLANE_U
] +
1534 (i
+ k
) * img2
->stride
[VPX_PLANE_V
] + j
+ l
);
1541 vloc
[0] = vloc
[1] = vloc
[2] = vloc
[3] = -1;
1542 for (i
= 0, match
= 1; match
&& i
< c_h
; i
+= bsizey
) {
1543 for (j
= 0; match
&& j
< c_w
; j
+= bsizex
) {
1545 int si
= mmin(i
+ bsizey
, c_h
- i
);
1546 int sj
= mmin(j
+ bsizex
, c_w
- j
);
1547 for (k
= 0; match
&& k
< si
; k
++)
1548 for (l
= 0; match
&& l
< sj
; l
++) {
1549 if (*(img1
->planes
[VPX_PLANE_V
] +
1550 (i
+ k
) * img1
->stride
[VPX_PLANE_V
] + j
+ l
) !=
1551 *(img2
->planes
[VPX_PLANE_V
] +
1552 (i
+ k
) * img2
->stride
[VPX_PLANE_V
] + j
+ l
)) {
1555 vloc
[2] = *(img1
->planes
[VPX_PLANE_V
] +
1556 (i
+ k
) * img1
->stride
[VPX_PLANE_V
] + j
+ l
);
1557 vloc
[3] = *(img2
->planes
[VPX_PLANE_V
] +
1558 (i
+ k
) * img2
->stride
[VPX_PLANE_V
] + j
+ l
);
1567 static int compare_img(vpx_image_t
*img1
, vpx_image_t
*img2
)
1569 const int c_w
= (img1
->d_w
+ img1
->x_chroma_shift
) >> img1
->x_chroma_shift
;
1570 const int c_h
= (img1
->d_h
+ img1
->y_chroma_shift
) >> img1
->y_chroma_shift
;
1574 match
&= (img1
->fmt
== img2
->fmt
);
1575 match
&= (img1
->w
== img2
->w
);
1576 match
&= (img1
->h
== img2
->h
);
1578 for (i
= 0; i
< img1
->d_h
; i
++)
1579 match
&= (memcmp(img1
->planes
[VPX_PLANE_Y
]+i
*img1
->stride
[VPX_PLANE_Y
],
1580 img2
->planes
[VPX_PLANE_Y
]+i
*img2
->stride
[VPX_PLANE_Y
],
1583 for (i
= 0; i
< c_h
; i
++)
1584 match
&= (memcmp(img1
->planes
[VPX_PLANE_U
]+i
*img1
->stride
[VPX_PLANE_U
],
1585 img2
->planes
[VPX_PLANE_U
]+i
*img2
->stride
[VPX_PLANE_U
],
1588 for (i
= 0; i
< c_h
; i
++)
1589 match
&= (memcmp(img1
->planes
[VPX_PLANE_V
]+i
*img1
->stride
[VPX_PLANE_U
],
1590 img2
->planes
[VPX_PLANE_V
]+i
*img2
->stride
[VPX_PLANE_U
],
1597 #define NELEMENTS(x) (sizeof(x)/sizeof(x[0]))
1598 #define MAX(x,y) ((x)>(y)?(x):(y))
1599 #if CONFIG_VP8_ENCODER && !CONFIG_VP9_ENCODER
1600 #define ARG_CTRL_CNT_MAX NELEMENTS(vp8_arg_ctrl_map)
1601 #elif !CONFIG_VP8_ENCODER && CONFIG_VP9_ENCODER
1602 #define ARG_CTRL_CNT_MAX NELEMENTS(vp9_arg_ctrl_map)
1604 #define ARG_CTRL_CNT_MAX MAX(NELEMENTS(vp8_arg_ctrl_map), \
1605 NELEMENTS(vp9_arg_ctrl_map))
1608 /* Configuration elements common to all streams */
1609 struct global_config
{
1610 const struct codec_item
*codec
;
1621 enum TestDecodeFatality test_decode
;
1623 struct vpx_rational framerate
;
1626 int show_q_hist_buckets
;
1627 int show_rate_hist_buckets
;
1631 /* Per-stream configuration */
1632 struct stream_config
{
1633 struct vpx_codec_enc_cfg cfg
;
1635 const char *stats_fn
;
1636 stereo_format_t stereo_fmt
;
1637 int arg_ctrls
[ARG_CTRL_CNT_MAX
][2];
1640 int have_kf_max_dist
;
1644 struct stream_state
{
1646 struct stream_state
*next
;
1647 struct stream_config config
;
1649 struct rate_hist rate_hist
;
1652 uint64_t psnr_sse_total
;
1653 uint64_t psnr_samples_total
;
1654 double psnr_totals
[4];
1657 vpx_codec_ctx_t encoder
;
1658 unsigned int frames_out
;
1662 struct vpx_image
*img
;
1663 vpx_codec_ctx_t decoder
;
1668 void validate_positive_rational(const char *msg
,
1669 struct vpx_rational
*rat
) {
1676 die("Error: %s must be positive\n", msg
);
1679 die("Error: %s has zero denominator\n", msg
);
1683 static void parse_global_config(struct global_config
*global
, char **argv
) {
1684 char **argi
, **argj
;
1687 /* Initialize default parameters */
1688 memset(global
, 0, sizeof(*global
));
1689 global
->codec
= codecs
;
1691 global
->use_i420
= 1;
1692 /* Assign default deadline to good quality */
1693 global
->deadline
= VPX_DL_GOOD_QUALITY
;
1695 for (argi
= argj
= argv
; (*argj
= *argi
); argi
+= arg
.argv_step
) {
1698 if (arg_match(&arg
, &codecarg
, argi
)) {
1701 for (j
= 0; j
< sizeof(codecs
) / sizeof(codecs
[0]); j
++)
1702 if (!strcmp(codecs
[j
].name
, arg
.val
))
1706 global
->codec
= codecs
+ k
;
1708 die("Error: Unrecognized argument (%s) to --codec\n",
1711 } else if (arg_match(&arg
, &passes
, argi
)) {
1712 global
->passes
= arg_parse_uint(&arg
);
1714 if (global
->passes
< 1 || global
->passes
> 2)
1715 die("Error: Invalid number of passes (%d)\n", global
->passes
);
1716 } else if (arg_match(&arg
, &pass_arg
, argi
)) {
1717 global
->pass
= arg_parse_uint(&arg
);
1719 if (global
->pass
< 1 || global
->pass
> 2)
1720 die("Error: Invalid pass selected (%d)\n",
1722 } else if (arg_match(&arg
, &usage
, argi
))
1723 global
->usage
= arg_parse_uint(&arg
);
1724 else if (arg_match(&arg
, &deadline
, argi
))
1725 global
->deadline
= arg_parse_uint(&arg
);
1726 else if (arg_match(&arg
, &best_dl
, argi
))
1727 global
->deadline
= VPX_DL_BEST_QUALITY
;
1728 else if (arg_match(&arg
, &good_dl
, argi
))
1729 global
->deadline
= VPX_DL_GOOD_QUALITY
;
1730 else if (arg_match(&arg
, &rt_dl
, argi
))
1731 global
->deadline
= VPX_DL_REALTIME
;
1732 else if (arg_match(&arg
, &use_yv12
, argi
))
1733 global
->use_i420
= 0;
1734 else if (arg_match(&arg
, &use_i420
, argi
))
1735 global
->use_i420
= 1;
1736 else if (arg_match(&arg
, &quietarg
, argi
))
1738 else if (arg_match(&arg
, &verbosearg
, argi
))
1739 global
->verbose
= 1;
1740 else if (arg_match(&arg
, &limit
, argi
))
1741 global
->limit
= arg_parse_uint(&arg
);
1742 else if (arg_match(&arg
, &skip
, argi
))
1743 global
->skip_frames
= arg_parse_uint(&arg
);
1744 else if (arg_match(&arg
, &psnrarg
, argi
))
1745 global
->show_psnr
= 1;
1746 else if (arg_match(&arg
, &recontest
, argi
))
1747 global
->test_decode
= arg_parse_enum_or_int(&arg
);
1748 else if (arg_match(&arg
, &framerate
, argi
)) {
1749 global
->framerate
= arg_parse_rational(&arg
);
1750 validate_positive_rational(arg
.name
, &global
->framerate
);
1751 global
->have_framerate
= 1;
1752 } else if (arg_match(&arg
, &out_part
, argi
))
1753 global
->out_part
= 1;
1754 else if (arg_match(&arg
, &debugmode
, argi
))
1756 else if (arg_match(&arg
, &q_hist_n
, argi
))
1757 global
->show_q_hist_buckets
= arg_parse_uint(&arg
);
1758 else if (arg_match(&arg
, &rate_hist_n
, argi
))
1759 global
->show_rate_hist_buckets
= arg_parse_uint(&arg
);
1764 /* Validate global config */
1765 if (global
->passes
== 0) {
1766 #if CONFIG_VP9_ENCODER
1767 // Make default VP9 passes = 2 until there is a better quality 1-pass
1769 global
->passes
= (global
->codec
->iface
== vpx_codec_vp9_cx
? 2 : 1);
1776 /* DWIM: Assume the user meant passes=2 if pass=2 is specified */
1777 if (global
->pass
> global
->passes
) {
1778 warn("Assuming --pass=%d implies --passes=%d\n",
1779 global
->pass
, global
->pass
);
1780 global
->passes
= global
->pass
;
1786 void open_input_file(struct input_state
*input
) {
1787 unsigned int fourcc
;
1789 /* Parse certain options from the input file, if possible */
1790 input
->file
= strcmp(input
->fn
, "-") ? fopen(input
->fn
, "rb")
1791 : set_binary_mode(stdin
);
1794 fatal("Failed to open input file");
1796 if (!fseeko(input
->file
, 0, SEEK_END
)) {
1797 /* Input file is seekable. Figure out how long it is, so we can get
1800 input
->length
= ftello(input
->file
);
1801 rewind(input
->file
);
1804 /* For RAW input sources, these bytes will applied on the first frame
1807 input
->detect
.buf_read
= fread(input
->detect
.buf
, 1, 4, input
->file
);
1808 input
->detect
.position
= 0;
1810 if (input
->detect
.buf_read
== 4
1811 && file_is_y4m(input
->file
, &input
->y4m
, input
->detect
.buf
)) {
1812 if (y4m_input_open(&input
->y4m
, input
->file
, input
->detect
.buf
, 4,
1813 input
->only_i420
) >= 0) {
1814 input
->file_type
= FILE_TYPE_Y4M
;
1815 input
->w
= input
->y4m
.pic_w
;
1816 input
->h
= input
->y4m
.pic_h
;
1817 input
->framerate
.num
= input
->y4m
.fps_n
;
1818 input
->framerate
.den
= input
->y4m
.fps_d
;
1819 input
->use_i420
= 0;
1821 fatal("Unsupported Y4M stream.");
1822 } else if (input
->detect
.buf_read
== 4 && file_is_ivf(input
, &fourcc
)) {
1823 input
->file_type
= FILE_TYPE_IVF
;
1826 input
->use_i420
= 0;
1829 input
->use_i420
= 1;
1832 fatal("Unsupported fourcc (%08x) in IVF", fourcc
);
1835 input
->file_type
= FILE_TYPE_RAW
;
1840 static void close_input_file(struct input_state
*input
) {
1841 fclose(input
->file
);
1842 if (input
->file_type
== FILE_TYPE_Y4M
)
1843 y4m_input_close(&input
->y4m
);
1846 static struct stream_state
*new_stream(struct global_config
*global
,
1847 struct stream_state
*prev
) {
1848 struct stream_state
*stream
;
1850 stream
= calloc(1, sizeof(*stream
));
1852 fatal("Failed to allocate new stream.");
1854 memcpy(stream
, prev
, sizeof(*stream
));
1856 prev
->next
= stream
;
1858 vpx_codec_err_t res
;
1860 /* Populate encoder configuration */
1861 res
= vpx_codec_enc_config_default(global
->codec
->iface(),
1862 &stream
->config
.cfg
,
1865 fatal("Failed to get config: %s\n", vpx_codec_err_to_string(res
));
1867 /* Change the default timebase to a high enough value so that the
1868 * encoder will always create strictly increasing timestamps.
1870 stream
->config
.cfg
.g_timebase
.den
= 1000;
1872 /* Never use the library's default resolution, require it be parsed
1873 * from the file or set on the command line.
1875 stream
->config
.cfg
.g_w
= 0;
1876 stream
->config
.cfg
.g_h
= 0;
1878 /* Initialize remaining stream parameters */
1879 stream
->config
.stereo_fmt
= STEREO_FORMAT_MONO
;
1880 stream
->config
.write_webm
= 1;
1881 stream
->ebml
.last_pts_ms
= -1;
1883 /* Allows removal of the application version from the EBML tags */
1884 stream
->ebml
.debug
= global
->debug
;
1887 /* Output files must be specified for each stream */
1888 stream
->config
.out_fn
= NULL
;
1890 stream
->next
= NULL
;
1895 static int parse_stream_params(struct global_config
*global
,
1896 struct stream_state
*stream
,
1898 char **argi
, **argj
;
1900 static const arg_def_t
**ctrl_args
= no_args
;
1901 static const int *ctrl_args_map
= NULL
;
1902 struct stream_config
*config
= &stream
->config
;
1903 int eos_mark_found
= 0;
1905 /* Handle codec specific options */
1907 #if CONFIG_VP8_ENCODER
1908 } else if (global
->codec
->iface
== vpx_codec_vp8_cx
) {
1909 ctrl_args
= vp8_args
;
1910 ctrl_args_map
= vp8_arg_ctrl_map
;
1912 #if CONFIG_VP9_ENCODER
1913 } else if (global
->codec
->iface
== vpx_codec_vp9_cx
) {
1914 ctrl_args
= vp9_args
;
1915 ctrl_args_map
= vp9_arg_ctrl_map
;
1919 for (argi
= argj
= argv
; (*argj
= *argi
); argi
+= arg
.argv_step
) {
1922 /* Once we've found an end-of-stream marker (--) we want to continue
1923 * shifting arguments but not consuming them.
1925 if (eos_mark_found
) {
1928 } else if (!strcmp(*argj
, "--")) {
1934 else if (arg_match(&arg
, &outputfile
, argi
))
1935 config
->out_fn
= arg
.val
;
1936 else if (arg_match(&arg
, &fpf_name
, argi
))
1937 config
->stats_fn
= arg
.val
;
1938 else if (arg_match(&arg
, &use_ivf
, argi
))
1939 config
->write_webm
= 0;
1940 else if (arg_match(&arg
, &threads
, argi
))
1941 config
->cfg
.g_threads
= arg_parse_uint(&arg
);
1942 else if (arg_match(&arg
, &profile
, argi
))
1943 config
->cfg
.g_profile
= arg_parse_uint(&arg
);
1944 else if (arg_match(&arg
, &width
, argi
))
1945 config
->cfg
.g_w
= arg_parse_uint(&arg
);
1946 else if (arg_match(&arg
, &height
, argi
))
1947 config
->cfg
.g_h
= arg_parse_uint(&arg
);
1948 else if (arg_match(&arg
, &stereo_mode
, argi
))
1949 config
->stereo_fmt
= arg_parse_enum_or_int(&arg
);
1950 else if (arg_match(&arg
, &timebase
, argi
)) {
1951 config
->cfg
.g_timebase
= arg_parse_rational(&arg
);
1952 validate_positive_rational(arg
.name
, &config
->cfg
.g_timebase
);
1953 } else if (arg_match(&arg
, &error_resilient
, argi
))
1954 config
->cfg
.g_error_resilient
= arg_parse_uint(&arg
);
1955 else if (arg_match(&arg
, &lag_in_frames
, argi
))
1956 config
->cfg
.g_lag_in_frames
= arg_parse_uint(&arg
);
1957 else if (arg_match(&arg
, &dropframe_thresh
, argi
))
1958 config
->cfg
.rc_dropframe_thresh
= arg_parse_uint(&arg
);
1959 else if (arg_match(&arg
, &resize_allowed
, argi
))
1960 config
->cfg
.rc_resize_allowed
= arg_parse_uint(&arg
);
1961 else if (arg_match(&arg
, &resize_up_thresh
, argi
))
1962 config
->cfg
.rc_resize_up_thresh
= arg_parse_uint(&arg
);
1963 else if (arg_match(&arg
, &resize_down_thresh
, argi
))
1964 config
->cfg
.rc_resize_down_thresh
= arg_parse_uint(&arg
);
1965 else if (arg_match(&arg
, &end_usage
, argi
))
1966 config
->cfg
.rc_end_usage
= arg_parse_enum_or_int(&arg
);
1967 else if (arg_match(&arg
, &target_bitrate
, argi
))
1968 config
->cfg
.rc_target_bitrate
= arg_parse_uint(&arg
);
1969 else if (arg_match(&arg
, &min_quantizer
, argi
))
1970 config
->cfg
.rc_min_quantizer
= arg_parse_uint(&arg
);
1971 else if (arg_match(&arg
, &max_quantizer
, argi
))
1972 config
->cfg
.rc_max_quantizer
= arg_parse_uint(&arg
);
1973 else if (arg_match(&arg
, &undershoot_pct
, argi
))
1974 config
->cfg
.rc_undershoot_pct
= arg_parse_uint(&arg
);
1975 else if (arg_match(&arg
, &overshoot_pct
, argi
))
1976 config
->cfg
.rc_overshoot_pct
= arg_parse_uint(&arg
);
1977 else if (arg_match(&arg
, &buf_sz
, argi
))
1978 config
->cfg
.rc_buf_sz
= arg_parse_uint(&arg
);
1979 else if (arg_match(&arg
, &buf_initial_sz
, argi
))
1980 config
->cfg
.rc_buf_initial_sz
= arg_parse_uint(&arg
);
1981 else if (arg_match(&arg
, &buf_optimal_sz
, argi
))
1982 config
->cfg
.rc_buf_optimal_sz
= arg_parse_uint(&arg
);
1983 else if (arg_match(&arg
, &bias_pct
, argi
)) {
1984 config
->cfg
.rc_2pass_vbr_bias_pct
= arg_parse_uint(&arg
);
1986 if (global
->passes
< 2)
1987 warn("option %s ignored in one-pass mode.\n", arg
.name
);
1988 } else if (arg_match(&arg
, &minsection_pct
, argi
)) {
1989 config
->cfg
.rc_2pass_vbr_minsection_pct
= arg_parse_uint(&arg
);
1991 if (global
->passes
< 2)
1992 warn("option %s ignored in one-pass mode.\n", arg
.name
);
1993 } else if (arg_match(&arg
, &maxsection_pct
, argi
)) {
1994 config
->cfg
.rc_2pass_vbr_maxsection_pct
= arg_parse_uint(&arg
);
1996 if (global
->passes
< 2)
1997 warn("option %s ignored in one-pass mode.\n", arg
.name
);
1998 } else if (arg_match(&arg
, &kf_min_dist
, argi
))
1999 config
->cfg
.kf_min_dist
= arg_parse_uint(&arg
);
2000 else if (arg_match(&arg
, &kf_max_dist
, argi
)) {
2001 config
->cfg
.kf_max_dist
= arg_parse_uint(&arg
);
2002 config
->have_kf_max_dist
= 1;
2003 } else if (arg_match(&arg
, &kf_disabled
, argi
))
2004 config
->cfg
.kf_mode
= VPX_KF_DISABLED
;
2008 for (i
= 0; ctrl_args
[i
]; i
++) {
2009 if (arg_match(&arg
, ctrl_args
[i
], argi
)) {
2013 /* Point either to the next free element or the first
2014 * instance of this control.
2016 for (j
= 0; j
< config
->arg_ctrl_cnt
; j
++)
2017 if (config
->arg_ctrls
[j
][0] == ctrl_args_map
[i
])
2021 assert(j
< ARG_CTRL_CNT_MAX
);
2022 if (j
< ARG_CTRL_CNT_MAX
) {
2023 config
->arg_ctrls
[j
][0] = ctrl_args_map
[i
];
2024 config
->arg_ctrls
[j
][1] = arg_parse_enum_or_int(&arg
);
2025 if (j
== config
->arg_ctrl_cnt
)
2026 config
->arg_ctrl_cnt
++;
2037 return eos_mark_found
;
2041 #define FOREACH_STREAM(func)\
2044 struct stream_state *stream;\
2046 for(stream = streams; stream; stream = stream->next)\
2051 static void validate_stream_config(struct stream_state
*stream
) {
2052 struct stream_state
*streami
;
2054 if (!stream
->config
.cfg
.g_w
|| !stream
->config
.cfg
.g_h
)
2055 fatal("Stream %d: Specify stream dimensions with --width (-w) "
2056 " and --height (-h)", stream
->index
);
2058 for (streami
= stream
; streami
; streami
= streami
->next
) {
2059 /* All streams require output files */
2060 if (!streami
->config
.out_fn
)
2061 fatal("Stream %d: Output file is required (specify with -o)",
2064 /* Check for two streams outputting to the same file */
2065 if (streami
!= stream
) {
2066 const char *a
= stream
->config
.out_fn
;
2067 const char *b
= streami
->config
.out_fn
;
2068 if (!strcmp(a
, b
) && strcmp(a
, "/dev/null") && strcmp(a
, ":nul"))
2069 fatal("Stream %d: duplicate output file (from stream %d)",
2070 streami
->index
, stream
->index
);
2073 /* Check for two streams sharing a stats file. */
2074 if (streami
!= stream
) {
2075 const char *a
= stream
->config
.stats_fn
;
2076 const char *b
= streami
->config
.stats_fn
;
2077 if (a
&& b
&& !strcmp(a
, b
))
2078 fatal("Stream %d: duplicate stats file (from stream %d)",
2079 streami
->index
, stream
->index
);
2085 static void set_stream_dimensions(struct stream_state
*stream
,
2088 if (!stream
->config
.cfg
.g_w
) {
2089 if (!stream
->config
.cfg
.g_h
)
2090 stream
->config
.cfg
.g_w
= w
;
2092 stream
->config
.cfg
.g_w
= w
* stream
->config
.cfg
.g_h
/ h
;
2094 if (!stream
->config
.cfg
.g_h
) {
2095 stream
->config
.cfg
.g_h
= h
* stream
->config
.cfg
.g_w
/ w
;
2100 static void set_default_kf_interval(struct stream_state
*stream
,
2101 struct global_config
*global
) {
2102 /* Use a max keyframe interval of 5 seconds, if none was
2103 * specified on the command line.
2105 if (!stream
->config
.have_kf_max_dist
) {
2106 double framerate
= (double)global
->framerate
.num
/ global
->framerate
.den
;
2107 if (framerate
> 0.0)
2108 stream
->config
.cfg
.kf_max_dist
= (unsigned int)(5.0 * framerate
);
2113 static void show_stream_config(struct stream_state
*stream
,
2114 struct global_config
*global
,
2115 struct input_state
*input
) {
2117 #define SHOW(field) \
2118 fprintf(stderr, " %-28s = %d\n", #field, stream->config.cfg.field)
2120 if (stream
->index
== 0) {
2121 fprintf(stderr
, "Codec: %s\n",
2122 vpx_codec_iface_name(global
->codec
->iface()));
2123 fprintf(stderr
, "Source file: %s Format: %s\n", input
->fn
,
2124 input
->use_i420
? "I420" : "YV12");
2126 if (stream
->next
|| stream
->index
)
2127 fprintf(stderr
, "\nStream Index: %d\n", stream
->index
);
2128 fprintf(stderr
, "Destination file: %s\n", stream
->config
.out_fn
);
2129 fprintf(stderr
, "Encoder parameters:\n");
2136 SHOW(g_timebase
.num
);
2137 SHOW(g_timebase
.den
);
2138 SHOW(g_error_resilient
);
2140 SHOW(g_lag_in_frames
);
2141 SHOW(rc_dropframe_thresh
);
2142 SHOW(rc_resize_allowed
);
2143 SHOW(rc_resize_up_thresh
);
2144 SHOW(rc_resize_down_thresh
);
2146 SHOW(rc_target_bitrate
);
2147 SHOW(rc_min_quantizer
);
2148 SHOW(rc_max_quantizer
);
2149 SHOW(rc_undershoot_pct
);
2150 SHOW(rc_overshoot_pct
);
2152 SHOW(rc_buf_initial_sz
);
2153 SHOW(rc_buf_optimal_sz
);
2154 SHOW(rc_2pass_vbr_bias_pct
);
2155 SHOW(rc_2pass_vbr_minsection_pct
);
2156 SHOW(rc_2pass_vbr_maxsection_pct
);
2163 static void open_output_file(struct stream_state
*stream
,
2164 struct global_config
*global
) {
2165 const char *fn
= stream
->config
.out_fn
;
2167 stream
->file
= strcmp(fn
, "-") ? fopen(fn
, "wb") : set_binary_mode(stdout
);
2170 fatal("Failed to open output file");
2172 if (stream
->config
.write_webm
&& fseek(stream
->file
, 0, SEEK_CUR
))
2173 fatal("WebM output to pipes not supported.");
2175 if (stream
->config
.write_webm
) {
2176 stream
->ebml
.stream
= stream
->file
;
2177 write_webm_file_header(&stream
->ebml
, &stream
->config
.cfg
,
2179 stream
->config
.stereo_fmt
,
2180 global
->codec
->fourcc
);
2182 write_ivf_file_header(stream
->file
, &stream
->config
.cfg
,
2183 global
->codec
->fourcc
, 0);
2187 static void close_output_file(struct stream_state
*stream
,
2188 unsigned int fourcc
) {
2189 if (stream
->config
.write_webm
) {
2190 write_webm_file_footer(&stream
->ebml
, stream
->hash
);
2191 free(stream
->ebml
.cue_list
);
2192 stream
->ebml
.cue_list
= NULL
;
2194 if (!fseek(stream
->file
, 0, SEEK_SET
))
2195 write_ivf_file_header(stream
->file
, &stream
->config
.cfg
,
2197 stream
->frames_out
);
2200 fclose(stream
->file
);
2204 static void setup_pass(struct stream_state
*stream
,
2205 struct global_config
*global
,
2207 if (stream
->config
.stats_fn
) {
2208 if (!stats_open_file(&stream
->stats
, stream
->config
.stats_fn
,
2210 fatal("Failed to open statistics store");
2212 if (!stats_open_mem(&stream
->stats
, pass
))
2213 fatal("Failed to open statistics store");
2216 stream
->config
.cfg
.g_pass
= global
->passes
== 2
2217 ? pass
? VPX_RC_LAST_PASS
: VPX_RC_FIRST_PASS
2220 stream
->config
.cfg
.rc_twopass_stats_in
= stats_get(&stream
->stats
);
2222 stream
->cx_time
= 0;
2224 stream
->frames_out
= 0;
2228 static void initialize_encoder(struct stream_state
*stream
,
2229 struct global_config
*global
) {
2233 flags
|= global
->show_psnr
? VPX_CODEC_USE_PSNR
: 0;
2234 flags
|= global
->out_part
? VPX_CODEC_USE_OUTPUT_PARTITION
: 0;
2236 /* Construct Encoder Context */
2237 vpx_codec_enc_init(&stream
->encoder
, global
->codec
->iface(),
2238 &stream
->config
.cfg
, flags
);
2239 ctx_exit_on_error(&stream
->encoder
, "Failed to initialize encoder");
2241 /* Note that we bypass the vpx_codec_control wrapper macro because
2242 * we're being clever to store the control IDs in an array. Real
2243 * applications will want to make use of the enumerations directly
2245 for (i
= 0; i
< stream
->config
.arg_ctrl_cnt
; i
++) {
2246 int ctrl
= stream
->config
.arg_ctrls
[i
][0];
2247 int value
= stream
->config
.arg_ctrls
[i
][1];
2248 if (vpx_codec_control_(&stream
->encoder
, ctrl
, value
))
2249 fprintf(stderr
, "Error: Tried to set control %d = %d\n",
2252 ctx_exit_on_error(&stream
->encoder
, "Failed to control codec");
2256 if (global
->test_decode
!= TEST_DECODE_OFF
) {
2257 vpx_codec_dec_init(&stream
->decoder
, global
->codec
->dx_iface(), NULL
, 0);
2263 static void encode_frame(struct stream_state
*stream
,
2264 struct global_config
*global
,
2265 struct vpx_image
*img
,
2266 unsigned int frames_in
) {
2267 vpx_codec_pts_t frame_start
, next_frame_start
;
2268 struct vpx_codec_enc_cfg
*cfg
= &stream
->config
.cfg
;
2269 struct vpx_usec_timer timer
;
2271 frame_start
= (cfg
->g_timebase
.den
* (int64_t)(frames_in
- 1)
2272 * global
->framerate
.den
)
2273 / cfg
->g_timebase
.num
/ global
->framerate
.num
;
2274 next_frame_start
= (cfg
->g_timebase
.den
* (int64_t)(frames_in
)
2275 * global
->framerate
.den
)
2276 / cfg
->g_timebase
.num
/ global
->framerate
.num
;
2278 /* Scale if necessary */
2279 if (img
&& (img
->d_w
!= cfg
->g_w
|| img
->d_h
!= cfg
->g_h
)) {
2281 stream
->img
= vpx_img_alloc(NULL
, VPX_IMG_FMT_I420
,
2282 cfg
->g_w
, cfg
->g_h
, 16);
2283 I420Scale(img
->planes
[VPX_PLANE_Y
], img
->stride
[VPX_PLANE_Y
],
2284 img
->planes
[VPX_PLANE_U
], img
->stride
[VPX_PLANE_U
],
2285 img
->planes
[VPX_PLANE_V
], img
->stride
[VPX_PLANE_V
],
2287 stream
->img
->planes
[VPX_PLANE_Y
],
2288 stream
->img
->stride
[VPX_PLANE_Y
],
2289 stream
->img
->planes
[VPX_PLANE_U
],
2290 stream
->img
->stride
[VPX_PLANE_U
],
2291 stream
->img
->planes
[VPX_PLANE_V
],
2292 stream
->img
->stride
[VPX_PLANE_V
],
2293 stream
->img
->d_w
, stream
->img
->d_h
,
2299 vpx_usec_timer_start(&timer
);
2300 vpx_codec_encode(&stream
->encoder
, img
, frame_start
,
2301 (unsigned long)(next_frame_start
- frame_start
),
2302 0, global
->deadline
);
2303 vpx_usec_timer_mark(&timer
);
2304 stream
->cx_time
+= vpx_usec_timer_elapsed(&timer
);
2305 ctx_exit_on_error(&stream
->encoder
, "Stream %d: Failed to encode frame",
2310 static void update_quantizer_histogram(struct stream_state
*stream
) {
2311 if (stream
->config
.cfg
.g_pass
!= VPX_RC_FIRST_PASS
) {
2314 vpx_codec_control(&stream
->encoder
, VP8E_GET_LAST_QUANTIZER_64
, &q
);
2315 ctx_exit_on_error(&stream
->encoder
, "Failed to read quantizer");
2316 stream
->counts
[q
]++;
2321 static void get_cx_data(struct stream_state
*stream
,
2322 struct global_config
*global
,
2324 const vpx_codec_cx_pkt_t
*pkt
;
2325 const struct vpx_codec_enc_cfg
*cfg
= &stream
->config
.cfg
;
2326 vpx_codec_iter_t iter
= NULL
;
2329 while ((pkt
= vpx_codec_get_cx_data(&stream
->encoder
, &iter
))) {
2330 static size_t fsize
= 0;
2331 static off_t ivf_header_pos
= 0;
2333 switch (pkt
->kind
) {
2334 case VPX_CODEC_CX_FRAME_PKT
:
2335 if (!(pkt
->data
.frame
.flags
& VPX_FRAME_IS_FRAGMENT
)) {
2336 stream
->frames_out
++;
2339 fprintf(stderr
, " %6luF", (unsigned long)pkt
->data
.frame
.sz
);
2341 update_rate_histogram(&stream
->rate_hist
, cfg
, pkt
);
2342 if (stream
->config
.write_webm
) {
2343 /* Update the hash */
2344 if (!stream
->ebml
.debug
)
2345 stream
->hash
= murmur(pkt
->data
.frame
.buf
,
2346 (int)pkt
->data
.frame
.sz
,
2349 write_webm_block(&stream
->ebml
, cfg
, pkt
);
2351 if (pkt
->data
.frame
.partition_id
<= 0) {
2352 ivf_header_pos
= ftello(stream
->file
);
2353 fsize
= pkt
->data
.frame
.sz
;
2355 write_ivf_frame_header(stream
->file
, pkt
);
2357 fsize
+= pkt
->data
.frame
.sz
;
2359 if (!(pkt
->data
.frame
.flags
& VPX_FRAME_IS_FRAGMENT
)) {
2360 off_t currpos
= ftello(stream
->file
);
2361 fseeko(stream
->file
, ivf_header_pos
, SEEK_SET
);
2362 write_ivf_frame_size(stream
->file
, fsize
);
2363 fseeko(stream
->file
, currpos
, SEEK_SET
);
2367 (void) fwrite(pkt
->data
.frame
.buf
, 1, pkt
->data
.frame
.sz
,
2370 stream
->nbytes
+= pkt
->data
.raw
.sz
;
2374 if (global
->test_decode
!= TEST_DECODE_OFF
&& !stream
->mismatch_seen
) {
2375 vpx_codec_decode(&stream
->decoder
, pkt
->data
.frame
.buf
,
2376 pkt
->data
.frame
.sz
, NULL
, 0);
2377 if (stream
->decoder
.err
) {
2378 warn_or_exit_on_error(&stream
->decoder
,
2379 global
->test_decode
== TEST_DECODE_FATAL
,
2380 "Failed to decode frame %d in stream %d",
2381 stream
->frames_out
+ 1, stream
->index
);
2382 stream
->mismatch_seen
= stream
->frames_out
+ 1;
2387 case VPX_CODEC_STATS_PKT
:
2388 stream
->frames_out
++;
2389 stats_write(&stream
->stats
,
2390 pkt
->data
.twopass_stats
.buf
,
2391 pkt
->data
.twopass_stats
.sz
);
2392 stream
->nbytes
+= pkt
->data
.raw
.sz
;
2394 case VPX_CODEC_PSNR_PKT
:
2396 if (global
->show_psnr
) {
2399 stream
->psnr_sse_total
+= pkt
->data
.psnr
.sse
[0];
2400 stream
->psnr_samples_total
+= pkt
->data
.psnr
.samples
[0];
2401 for (i
= 0; i
< 4; i
++) {
2403 fprintf(stderr
, "%.3f ", pkt
->data
.psnr
.psnr
[i
]);
2404 stream
->psnr_totals
[i
] += pkt
->data
.psnr
.psnr
[i
];
2406 stream
->psnr_count
++;
2417 static void show_psnr(struct stream_state
*stream
) {
2421 if (!stream
->psnr_count
)
2424 fprintf(stderr
, "Stream %d PSNR (Overall/Avg/Y/U/V)", stream
->index
);
2425 ovpsnr
= vp8_mse2psnr((double)stream
->psnr_samples_total
, 255.0,
2426 (double)stream
->psnr_sse_total
);
2427 fprintf(stderr
, " %.3f", ovpsnr
);
2429 for (i
= 0; i
< 4; i
++) {
2430 fprintf(stderr
, " %.3f", stream
->psnr_totals
[i
] / stream
->psnr_count
);
2432 fprintf(stderr
, "\n");
2436 static float usec_to_fps(uint64_t usec
, unsigned int frames
) {
2437 return (float)(usec
> 0 ? frames
* 1000000.0 / (float)usec
: 0);
2441 static void test_decode(struct stream_state
*stream
,
2442 enum TestDecodeFatality fatal
,
2443 const struct codec_item
*codec
) {
2444 vpx_image_t enc_img
, dec_img
;
2446 if (stream
->mismatch_seen
)
2449 /* Get the internal reference frame */
2450 if (codec
->fourcc
== VP8_FOURCC
) {
2451 struct vpx_ref_frame ref_enc
, ref_dec
;
2454 width
= (stream
->config
.cfg
.g_w
+ 15) & ~15;
2455 height
= (stream
->config
.cfg
.g_h
+ 15) & ~15;
2456 vpx_img_alloc(&ref_enc
.img
, VPX_IMG_FMT_I420
, width
, height
, 1);
2457 enc_img
= ref_enc
.img
;
2458 vpx_img_alloc(&ref_dec
.img
, VPX_IMG_FMT_I420
, width
, height
, 1);
2459 dec_img
= ref_dec
.img
;
2461 ref_enc
.frame_type
= VP8_LAST_FRAME
;
2462 ref_dec
.frame_type
= VP8_LAST_FRAME
;
2463 vpx_codec_control(&stream
->encoder
, VP8_COPY_REFERENCE
, &ref_enc
);
2464 vpx_codec_control(&stream
->decoder
, VP8_COPY_REFERENCE
, &ref_dec
);
2466 struct vp9_ref_frame ref
;
2469 vpx_codec_control(&stream
->encoder
, VP9_GET_REFERENCE
, &ref
);
2471 vpx_codec_control(&stream
->decoder
, VP9_GET_REFERENCE
, &ref
);
2474 ctx_exit_on_error(&stream
->encoder
, "Failed to get encoder reference frame");
2475 ctx_exit_on_error(&stream
->decoder
, "Failed to get decoder reference frame");
2477 if (!compare_img(&enc_img
, &dec_img
)) {
2478 int y
[4], u
[4], v
[4];
2479 find_mismatch(&enc_img
, &dec_img
, y
, u
, v
);
2480 stream
->decoder
.err
= 1;
2481 warn_or_exit_on_error(&stream
->decoder
, fatal
== TEST_DECODE_FATAL
,
2482 "Stream %d: Encode/decode mismatch on frame %d at"
2483 " Y[%d, %d] {%d/%d},"
2484 " U[%d, %d] {%d/%d},"
2485 " V[%d, %d] {%d/%d}",
2486 stream
->index
, stream
->frames_out
,
2487 y
[0], y
[1], y
[2], y
[3],
2488 u
[0], u
[1], u
[2], u
[3],
2489 v
[0], v
[1], v
[2], v
[3]);
2490 stream
->mismatch_seen
= stream
->frames_out
;
2493 vpx_img_free(&enc_img
);
2494 vpx_img_free(&dec_img
);
2498 static void print_time(const char *label
, int64_t etl
) {
2499 int hours
, mins
, secs
;
2503 etl
-= hours
* 3600;
2508 fprintf(stderr
, "[%3s %2d:%02d:%02d] ",
2509 label
, hours
, mins
, secs
);
2511 fprintf(stderr
, "[%3s unknown] ", label
);
2515 int main(int argc
, const char **argv_
) {
2518 int frame_avail
, got_data
;
2520 struct input_state input
= {0};
2521 struct global_config global
;
2522 struct stream_state
*streams
= NULL
;
2523 char **argv
, **argi
;
2524 uint64_t cx_time
= 0;
2528 exec_name
= argv_
[0];
2533 /* Setup default input stream settings */
2534 input
.framerate
.num
= 30;
2535 input
.framerate
.den
= 1;
2537 input
.only_i420
= 1;
2539 /* First parse the global configuration values, because we want to apply
2540 * other parameters on top of the default configuration provided by the
2543 argv
= argv_dup(argc
- 1, argv_
+ 1);
2544 parse_global_config(&global
, argv
);
2547 /* Now parse each stream's parameters. Using a local scope here
2548 * due to the use of 'stream' as loop variable in FOREACH_STREAM
2551 struct stream_state
*stream
= NULL
;
2554 stream
= new_stream(&global
, stream
);
2558 } while (parse_stream_params(&global
, stream
, argv
));
2561 /* Check for unrecognized options */
2562 for (argi
= argv
; *argi
; argi
++)
2563 if (argi
[0][0] == '-' && argi
[0][1])
2564 die("Error: Unrecognized option %s\n", *argi
);
2566 /* Handle non-option arguments */
2573 /* Decide if other chroma subsamplings than 4:2:0 are supported */
2574 if (global
.codec
->fourcc
== VP9_FOURCC
)
2575 input
.only_i420
= 0;
2578 for (pass
= global
.pass
? global
.pass
- 1 : 0; pass
< global
.passes
; pass
++) {
2579 int frames_in
= 0, seen_frames
= 0;
2580 int64_t estimated_time_left
= -1;
2581 int64_t average_rate
= -1;
2582 off_t lagged_count
= 0;
2584 open_input_file(&input
);
2586 /* If the input file doesn't specify its w/h (raw files), try to get
2587 * the data from the first stream's configuration.
2589 if (!input
.w
|| !input
.h
)
2591 if (stream
->config
.cfg
.g_w
&& stream
->config
.cfg
.g_h
) {
2592 input
.w
= stream
->config
.cfg
.g_w
;
2593 input
.h
= stream
->config
.cfg
.g_h
;
2598 /* Update stream configurations from the input file's parameters */
2599 if (!input
.w
|| !input
.h
)
2600 fatal("Specify stream dimensions with --width (-w) "
2601 " and --height (-h)");
2602 FOREACH_STREAM(set_stream_dimensions(stream
, input
.w
, input
.h
));
2603 FOREACH_STREAM(validate_stream_config(stream
));
2605 /* Ensure that --passes and --pass are consistent. If --pass is set and
2606 * --passes=2, ensure --fpf was set.
2608 if (global
.pass
&& global
.passes
== 2)
2610 if (!stream
->config
.stats_fn
)
2611 die("Stream %d: Must specify --fpf when --pass=%d"
2612 " and --passes=2\n", stream
->index
, global
.pass
);
2615 /* Use the frame rate from the file only if none was specified
2616 * on the command-line.
2618 if (!global
.have_framerate
)
2619 global
.framerate
= input
.framerate
;
2621 FOREACH_STREAM(set_default_kf_interval(stream
, &global
));
2623 /* Show configuration */
2624 if (global
.verbose
&& pass
== 0)
2625 FOREACH_STREAM(show_stream_config(stream
, &global
, &input
));
2627 if (pass
== (global
.pass
? global
.pass
- 1 : 0)) {
2628 if (input
.file_type
== FILE_TYPE_Y4M
)
2629 /*The Y4M reader does its own allocation.
2630 Just initialize this here to avoid problems if we never read any
2632 memset(&raw
, 0, sizeof(raw
));
2635 input
.use_i420
? VPX_IMG_FMT_I420
2637 input
.w
, input
.h
, 32);
2639 FOREACH_STREAM(init_rate_histogram(&stream
->rate_hist
,
2640 &stream
->config
.cfg
,
2641 &global
.framerate
));
2644 FOREACH_STREAM(setup_pass(stream
, &global
, pass
));
2645 FOREACH_STREAM(open_output_file(stream
, &global
));
2646 FOREACH_STREAM(initialize_encoder(stream
, &global
));
2651 while (frame_avail
|| got_data
) {
2652 struct vpx_usec_timer timer
;
2654 if (!global
.limit
|| frames_in
< global
.limit
) {
2655 frame_avail
= read_frame(&input
, &raw
);
2659 seen_frames
= frames_in
> global
.skip_frames
?
2660 frames_in
- global
.skip_frames
: 0;
2662 if (!global
.quiet
) {
2663 float fps
= usec_to_fps(cx_time
, seen_frames
);
2664 fprintf(stderr
, "\rPass %d/%d ", pass
+ 1, global
.passes
);
2666 if (stream_cnt
== 1)
2668 "frame %4d/%-4d %7"PRId64
"B ",
2669 frames_in
, streams
->frames_out
, (int64_t)streams
->nbytes
);
2671 fprintf(stderr
, "frame %4d ", frames_in
);
2673 fprintf(stderr
, "%7"PRId64
" %s %.2f %s ",
2674 cx_time
> 9999999 ? cx_time
/ 1000 : cx_time
,
2675 cx_time
> 9999999 ? "ms" : "us",
2676 fps
>= 1.0 ? fps
: fps
* 60,
2677 fps
>= 1.0 ? "fps" : "fpm");
2678 print_time("ETA", estimated_time_left
);
2679 fprintf(stderr
, "\033[K");
2685 if (frames_in
> global
.skip_frames
) {
2686 vpx_usec_timer_start(&timer
);
2687 FOREACH_STREAM(encode_frame(stream
, &global
,
2688 frame_avail
? &raw
: NULL
,
2690 vpx_usec_timer_mark(&timer
);
2691 cx_time
+= vpx_usec_timer_elapsed(&timer
);
2693 FOREACH_STREAM(update_quantizer_histogram(stream
));
2696 FOREACH_STREAM(get_cx_data(stream
, &global
, &got_data
));
2698 if (!got_data
&& input
.length
&& !streams
->frames_out
) {
2699 lagged_count
= global
.limit
? seen_frames
: ftello(input
.file
);
2700 } else if (input
.length
) {
2705 int frame_in_lagged
= (seen_frames
- lagged_count
) * 1000;
2707 rate
= cx_time
? frame_in_lagged
* (int64_t)1000000 / cx_time
: 0;
2708 remaining
= 1000 * (global
.limit
- global
.skip_frames
2709 - seen_frames
+ lagged_count
);
2711 off_t input_pos
= ftello(input
.file
);
2712 off_t input_pos_lagged
= input_pos
- lagged_count
;
2713 int64_t limit
= input
.length
;
2715 rate
= cx_time
? input_pos_lagged
* (int64_t)1000000 / cx_time
: 0;
2716 remaining
= limit
- input_pos
+ lagged_count
;
2719 average_rate
= (average_rate
<= 0)
2721 : (average_rate
* 7 + rate
) / 8;
2722 estimated_time_left
= average_rate
? remaining
/ average_rate
: -1;
2725 if (got_data
&& global
.test_decode
!= TEST_DECODE_OFF
)
2726 FOREACH_STREAM(test_decode(stream
, global
.test_decode
, global
.codec
));
2733 fprintf(stderr
, "\n");
2736 FOREACH_STREAM(fprintf(
2738 "\rPass %d/%d frame %4d/%-4d %7"PRId64
"B %7lub/f %7"PRId64
"b/s"
2739 " %7"PRId64
" %s (%.2f fps)\033[K\n", pass
+ 1,
2740 global
.passes
, frames_in
, stream
->frames_out
, (int64_t)stream
->nbytes
,
2741 seen_frames
? (unsigned long)(stream
->nbytes
* 8 / seen_frames
) : 0,
2742 seen_frames
? (int64_t)stream
->nbytes
* 8
2743 * (int64_t)global
.framerate
.num
/ global
.framerate
.den
2746 stream
->cx_time
> 9999999 ? stream
->cx_time
/ 1000 : stream
->cx_time
,
2747 stream
->cx_time
> 9999999 ? "ms" : "us",
2748 usec_to_fps(stream
->cx_time
, seen_frames
));
2751 if (global
.show_psnr
)
2752 FOREACH_STREAM(show_psnr(stream
));
2754 FOREACH_STREAM(vpx_codec_destroy(&stream
->encoder
));
2756 if (global
.test_decode
!= TEST_DECODE_OFF
) {
2757 FOREACH_STREAM(vpx_codec_destroy(&stream
->decoder
));
2760 close_input_file(&input
);
2762 if (global
.test_decode
== TEST_DECODE_FATAL
) {
2763 FOREACH_STREAM(res
|= stream
->mismatch_seen
);
2765 FOREACH_STREAM(close_output_file(stream
, global
.codec
->fourcc
));
2767 FOREACH_STREAM(stats_close(&stream
->stats
, global
.passes
- 1));
2773 if (global
.show_q_hist_buckets
)
2774 FOREACH_STREAM(show_q_histogram(stream
->counts
,
2775 global
.show_q_hist_buckets
));
2777 if (global
.show_rate_hist_buckets
)
2778 FOREACH_STREAM(show_rate_histogram(&stream
->rate_hist
,
2779 &stream
->config
.cfg
,
2780 global
.show_rate_hist_buckets
));
2781 FOREACH_STREAM(destroy_rate_histogram(&stream
->rate_hist
));
2783 #if CONFIG_INTERNAL_STATS
2784 /* TODO(jkoleszar): This doesn't belong in this executable. Do it for now,
2785 * to match some existing utilities.
2788 FILE *f
= fopen("opsnr.stt", "a");
2789 if (stream
->mismatch_seen
) {
2790 fprintf(f
, "First mismatch occurred in frame %d\n",
2791 stream
->mismatch_seen
);
2793 fprintf(f
, "No mismatch detected in recon buffers\n");
2802 return res
? EXIT_FAILURE
: EXIT_SUCCESS
;