2 * Copyright (c) 2013 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 "./vpxstats.h"
17 #include "./tools_common.h"
19 int stats_open_file(stats_io_t
*stats
, const char *fpf
, int pass
) {
24 stats
->file
= fopen(fpf
, "wb");
26 stats
->buf
.buf
= NULL
;
27 res
= (stats
->file
!= NULL
);
33 fd
= open(fpf
, O_RDONLY
);
34 stats
->file
= fdopen(fd
, "rb");
36 stats
->buf
.sz
= stat_buf
.st_size
;
37 stats
->buf
.buf
= mmap(NULL
, stats
->buf
.sz
, PROT_READ
, MAP_PRIVATE
, fd
, 0);
38 res
= (stats
->buf
.buf
!= NULL
);
42 stats
->file
= fopen(fpf
, "rb");
44 if (fseek(stats
->file
, 0, SEEK_END
))
45 fatal("First-pass stats file must be seekable!");
47 stats
->buf
.sz
= stats
->buf_alloc_sz
= ftell(stats
->file
);
50 stats
->buf
.buf
= malloc(stats
->buf_alloc_sz
);
53 fatal("Failed to allocate first-pass stats buffer (%lu bytes)",
54 (unsigned int)stats
->buf_alloc_sz
);
56 nbytes
= fread(stats
->buf
.buf
, 1, stats
->buf
.sz
, stats
->file
);
57 res
= (nbytes
== stats
->buf
.sz
);
58 #endif /* USE_POSIX_MMAP */
64 int stats_open_mem(stats_io_t
*stats
, int pass
) {
70 stats
->buf_alloc_sz
= 64 * 1024;
71 stats
->buf
.buf
= malloc(stats
->buf_alloc_sz
);
74 stats
->buf_ptr
= stats
->buf
.buf
;
75 res
= (stats
->buf
.buf
!= NULL
);
79 void stats_close(stats_io_t
*stats
, int last_pass
) {
81 if (stats
->pass
== last_pass
) {
83 munmap(stats
->buf
.buf
, stats
->buf
.sz
);
86 #endif /* USE_POSIX_MMAP */
92 if (stats
->pass
== last_pass
)
97 void stats_write(stats_io_t
*stats
, const void *pkt
, size_t len
) {
99 (void) fwrite(pkt
, 1, len
, stats
->file
);
101 if (stats
->buf
.sz
+ len
> stats
->buf_alloc_sz
) {
102 size_t new_sz
= stats
->buf_alloc_sz
+ 64 * 1024;
103 char *new_ptr
= realloc(stats
->buf
.buf
, new_sz
);
106 stats
->buf_ptr
= new_ptr
+ (stats
->buf_ptr
- (char *)stats
->buf
.buf
);
107 stats
->buf
.buf
= new_ptr
;
108 stats
->buf_alloc_sz
= new_sz
;
110 fatal("Failed to realloc firstpass stats buffer.");
114 memcpy(stats
->buf_ptr
, pkt
, len
);
115 stats
->buf
.sz
+= len
;
116 stats
->buf_ptr
+= len
;
120 vpx_fixed_buf_t
stats_get(stats_io_t
*stats
) {
124 double vp8_mse2psnr(double samples
, double peak
, double mse
) {
125 const int kMaxPSNR
= 100;
126 double psnr
= kMaxPSNR
;
129 psnr
= 10.0 * log10(peak
* peak
* samples
/ mse
);