1 /*
2  *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 #ifndef VPX_TOOLS_COMMON_H_
11 #define VPX_TOOLS_COMMON_H_
12 
13 #include <stdio.h>
14 
15 #define INLINE __inline
16 #include "vpx_codec.h"
17 #include <stdint.h>
18 
19 #if CONFIG_ENCODERS
20 #include "./y4minput.h"
21 #endif
22 
23 #if defined(_MSC_VER)
24 /* MSVS uses _f{seek,tell}i64. */
25 #define fseeko _fseeki64
26 #define ftello _ftelli64
27 typedef int64_t FileOffset;
28 #elif defined(_WIN32)
29 /* MinGW uses f{seek,tell}o64 for large files. */
30 #define fseeko fseeko64
31 #define ftello ftello64
32 typedef Off64 FileOffset;
33 #elif CONFIG_OS_SUPPORT
34 typedef off_t FileOffset;
35 /* Use 32-bit file operations in WebM file format when building ARM
36  * executables (.axf) with RVCT. */
37 #else
38 #define fseeko fseek
39 #define ftello ftell
40 typedef long FileOffset; /* NOLINT */
41 #endif /* CONFIG_OS_SUPPORT */
42 
43 #if CONFIG_OS_SUPPORT
44 #ifdef _WIN32
45 #include <io.h> /* NOLINT */
46 #define isatty _isatty
47 #define fileno _fileno
48 #else
49 #include <unistd.h> /* NOLINT */
50 #endif              /* _MSC_VER */
51 #endif              /* CONFIG_OS_SUPPORT */
52 
53 #define LITERALU64(hi, lo) ((((uint64_t)hi) << 32) | lo)
54 
55 #ifndef PATH_MAX
56 #define PATH_MAX 512
57 #endif
58 
59 #define IVF_FRAME_HDR_SZ (4 + 8) /* 4 byte size + 8 byte timestamp */
60 #define IVF_FILE_HDR_SZ 32
61 
62 #define RAW_FRAME_HDR_SZ sizeof(uint32_t)
63 
64 #define VP8_FOURCC 0x30385056
65 #define VP9_FOURCC 0x30395056
66 
67 enum VideoFileType {
68   FILE_TYPE_RAW,
69   FILE_TYPE_IVF,
70   FILE_TYPE_Y4M,
71   FILE_TYPE_WEBM
72 };
73 
74 struct FileTypeDetectionBuffer {
75   char buf[4];
76   size_t buf_read;
77   size_t position;
78 };
79 
80 struct VpxRational {
81   int numerator;
82   int denominator;
83 };
84 
85 struct VpxInputContext {
86   const char *filename;
87   FILE *file;
88   int64_t length;
89   struct FileTypeDetectionBuffer detect;
90   enum VideoFileType file_type;
91   uint32_t width;
92   uint32_t height;
93   struct VpxRational pixel_aspect_ratio;
94   vpx_img_fmt_t fmt;
95   vpx_bit_depth_t bit_depth;
96   int only_i420;
97   uint32_t fourcc;
98   struct VpxRational framerate;
99 #if CONFIG_ENCODERS
100   y4m_input y4m;
101 #endif
102 };
103 
104 #ifdef __cplusplus
105 extern "C" {
106 #endif
107 
108 #if defined(__GNUC__)
109 #define VPX_NO_RETURN __attribute__((noreturn))
110 #else
111 #define VPX_NO_RETURN
112 #endif
113 
114 /* Sets a stdio stream into binary mode */
115 FILE *set_binary_mode(FILE *stream);
116 
117 void die(const char *fmt, ...) VPX_NO_RETURN;
118 void fatal(const char *fmt, ...) VPX_NO_RETURN;
119 void warn(const char *fmt, ...);
120 
121 void die_codec(vpx_codec_ctx_t *ctx, const char *s) VPX_NO_RETURN;
122 
123 /* The tool including this file must define usage_exit() */
124 void usage_exit(void) VPX_NO_RETURN;
125 
126 #undef VPX_NO_RETURN
127 
128 int read_yuv_frame(struct VpxInputContext *input_ctx, vpx_image_t *yuv_frame);
129 
130 typedef struct VpxInterface {
131   const char *const name;
132   const uint32_t fourcc;
133   vpx_codec_iface_t *(*const codec_interface)();
134 } VpxInterface;
135 
136 int get_vpx_encoder_count(void);
137 const VpxInterface *get_vpx_encoder_by_index(int i);
138 const VpxInterface *get_vpx_encoder_by_name(const char *name);
139 
140 int get_vpx_decoder_count(void);
141 const VpxInterface *get_vpx_decoder_by_index(int i);
142 const VpxInterface *get_vpx_decoder_by_name(const char *name);
143 const VpxInterface *get_vpx_decoder_by_fourcc(uint32_t fourcc);
144 
145 // TODO(dkovalev): move this function to vpx_image.{c, h}, so it will be part
146 // of vpx_image_t support
147 int vpx_img_plane_width(const vpx_image_t *img, int plane);
148 int vpx_img_plane_height(const vpx_image_t *img, int plane);
149 void vpx_img_write(const vpx_image_t *img, FILE *file);
150 int vpx_img_read(vpx_image_t *img, FILE *file);
151 
152 double sse_to_psnr(double samples, double peak, double mse);
153 
154 #if CONFIG_VP9_HIGHBITDEPTH
155 void vpx_img_upshift(vpx_image_t *dst, vpx_image_t *src, int input_shift);
156 void vpx_img_downshift(vpx_image_t *dst, vpx_image_t *src, int down_shift);
157 void vpx_img_truncate_16_to_8(vpx_image_t *dst, vpx_image_t *src);
158 #endif
159 
160 #ifdef __cplusplus
161 } /* extern "C" */
162 #endif
163 
164 #endif // VPX_TOOLS_COMMON_H_
165