tools_common.h

   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 TOOLS_COMMON_H_
  11 #define TOOLS_COMMON_H_
  12
  13 #include <stdio.h>
  14
  15 #include "./vpx_config.h"
  16 #include "vpx/vpx_codec.h"
  17 #include "vpx/vpx_image.h"
  18 #include "vpx/vpx_integer.h"
  19 #include "vpx_ports/msvc.h"
  20
  21 #if CONFIG_ENCODERS
  22 #include "./y4minput.h"
  23 #endif
  24
  25 #if defined(_MSC_VER)
  26 /* MSVS uses _f{seek,tell}i64. */
  27 #define fseeko _fseeki64
  28 #define ftello _ftelli64
  29 #elif defined(_WIN32)
  30 /* MinGW uses f{seek,tell}o64 for large files. */
  31 #define fseeko fseeko64
  32 #define ftello ftello64
  33 #endif  /* _WIN32 */
  34
  35 #if CONFIG_OS_SUPPORT
  36 #if defined(_MSC_VER)
  37 #include <io.h>  /* NOLINT */
  38 #define isatty   _isatty
  39 #define fileno   _fileno
  40 #else
  41 #include <unistd.h>  /* NOLINT */
  42 #endif  /* _MSC_VER */
  43 #endif  /* CONFIG_OS_SUPPORT */
  44
  45 /* Use 32-bit file operations in WebM file format when building ARM
  46  * executables (.axf) with RVCT. */
  47 #if !CONFIG_OS_SUPPORT
  48 #define fseeko fseek
  49 #define ftello ftell
  50 #endif  /* CONFIG_OS_SUPPORT */
  51
  52 #define LITERALU64(hi, lo) ((((uint64_t)hi) << 32) | lo)
  53
  54 #ifndef PATH_MAX
  55 #define PATH_MAX 512
  56 #endif
  57
  58 #define IVF_FRAME_HDR_SZ (4 + 8)  /* 4 byte size + 8 byte timestamp */
  59 #define IVF_FILE_HDR_SZ 32
  60
  61 #define RAW_FRAME_HDR_SZ sizeof(uint32_t)
  62
  63 #define VP8_FOURCC 0x30385056
  64 #define VP9_FOURCC 0x30395056
  65 #define VP10_FOURCC 0x303a5056
  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  // TOOLS_COMMON_H_