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