tools_common.h 4.16 KB
Newer Older
1 2 3 4 5 6 7 8 9
/*
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
 *
 *  Use of this source code is governed by a BSD-style license
 *  that can be found in the LICENSE file in the root of the source
 *  tree. An additional intellectual property rights grant can be found
 *  in the file PATENTS.  All contributing project authors may
 *  be found in the AUTHORS file in the root of the source tree.
 */
10 11 12 13 14
#ifndef TOOLS_COMMON_H_
#define TOOLS_COMMON_H_

#include <stdio.h>

15
#include "./vpx_config.h"
16
#include "vpx/vpx_codec.h"
17 18
#include "vpx/vpx_image.h"
#include "vpx/vpx_integer.h"
19
#include "vpx_ports/msvc.h"
20 21 22 23

#if CONFIG_ENCODERS
#include "./y4minput.h"
#endif
24 25

#if defined(_MSC_VER)
26
/* MSVS uses _f{seek,tell}i64. */
27 28
#define fseeko _fseeki64
#define ftello _ftelli64
29
typedef int64_t FileOffset;
30
#elif defined(_WIN32)
31
/* MinGW uses f{seek,tell}o64 for large files. */
32 33
#define fseeko fseeko64
#define ftello ftello64
34 35 36 37 38 39 40 41
typedef off64_t FileOffset;
#elif CONFIG_OS_SUPPORT
typedef off_t FileOffset;
/* Use 32-bit file operations in WebM file format when building ARM
 * executables (.axf) with RVCT. */
#else
#define fseeko fseek
#define ftello ftell
42
typedef long FileOffset; /* NOLINT */
43
#endif /* CONFIG_OS_SUPPORT */
44 45 46

#if CONFIG_OS_SUPPORT
#if defined(_MSC_VER)
47 48 49
#include <io.h> /* NOLINT */
#define isatty _isatty
#define fileno _fileno
50
#else
51 52 53
#include <unistd.h> /* NOLINT */
#endif              /* _MSC_VER */
#endif              /* CONFIG_OS_SUPPORT */
54 55 56 57 58 59 60

#define LITERALU64(hi, lo) ((((uint64_t)hi) << 32) | lo)

#ifndef PATH_MAX
#define PATH_MAX 512
#endif

61
#define IVF_FRAME_HDR_SZ (4 + 8) /* 4 byte size + 8 byte timestamp */
62 63 64 65
#define IVF_FILE_HDR_SZ 32

#define RAW_FRAME_HDR_SZ sizeof(uint32_t)

66 67
#define VP8_FOURCC 0x30385056
#define VP9_FOURCC 0x30395056
68

69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
enum VideoFileType {
  FILE_TYPE_RAW,
  FILE_TYPE_IVF,
  FILE_TYPE_Y4M,
  FILE_TYPE_WEBM
};

struct FileTypeDetectionBuffer {
  char buf[4];
  size_t buf_read;
  size_t position;
};

struct VpxRational {
  int numerator;
  int denominator;
};

struct VpxInputContext {
  const char *filename;
  FILE *file;
90
  int64_t length;
91 92
  struct FileTypeDetectionBuffer detect;
  enum VideoFileType file_type;
93 94
  uint32_t width;
  uint32_t height;
95
  struct VpxRational pixel_aspect_ratio;
96
  vpx_img_fmt_t fmt;
97
  vpx_bit_depth_t bit_depth;
98
  int only_i420;
99
  uint32_t fourcc;
100 101 102 103 104 105 106 107 108 109
  struct VpxRational framerate;
#if CONFIG_ENCODERS
  y4m_input y4m;
#endif
};

#ifdef __cplusplus
extern "C" {
#endif

110 111 112 113 114 115
#if defined(__GNUC__)
#define VPX_NO_RETURN __attribute__((noreturn))
#else
#define VPX_NO_RETURN
#endif

116
/* Sets a stdio stream into binary mode */
John Koleszar's avatar
John Koleszar committed
117
FILE *set_binary_mode(FILE *stream);
118

119 120
void die(const char *fmt, ...) VPX_NO_RETURN;
void fatal(const char *fmt, ...) VPX_NO_RETURN;
121 122
void warn(const char *fmt, ...);

123
void die_codec(vpx_codec_ctx_t *ctx, const char *s) VPX_NO_RETURN;
124

125
/* The tool including this file must define usage_exit() */
126
void usage_exit(void) VPX_NO_RETURN;
127 128

#undef VPX_NO_RETURN
129

130 131
int read_yuv_frame(struct VpxInputContext *input_ctx, vpx_image_t *yuv_frame);

132 133 134
typedef struct VpxInterface {
  const char *const name;
  const uint32_t fourcc;
135
  vpx_codec_iface_t *(*const codec_interface)();
136 137
} VpxInterface;

138
int get_vpx_encoder_count(void);
139 140 141
const VpxInterface *get_vpx_encoder_by_index(int i);
const VpxInterface *get_vpx_encoder_by_name(const char *name);

142
int get_vpx_decoder_count(void);
143 144 145
const VpxInterface *get_vpx_decoder_by_index(int i);
const VpxInterface *get_vpx_decoder_by_name(const char *name);
const VpxInterface *get_vpx_decoder_by_fourcc(uint32_t fourcc);
146 147 148

// TODO(dkovalev): move this function to vpx_image.{c, h}, so it will be part
// of vpx_image_t support
149 150
int vpx_img_plane_width(const vpx_image_t *img, int plane);
int vpx_img_plane_height(const vpx_image_t *img, int plane);
151
void vpx_img_write(const vpx_image_t *img, FILE *file);
152 153
int vpx_img_read(vpx_image_t *img, FILE *file);

154 155
double sse_to_psnr(double samples, double peak, double mse);

156
#if CONFIG_VP9_HIGHBITDEPTH
157 158 159 160 161
void vpx_img_upshift(vpx_image_t *dst, vpx_image_t *src, int input_shift);
void vpx_img_downshift(vpx_image_t *dst, vpx_image_t *src, int down_shift);
void vpx_img_truncate_16_to_8(vpx_image_t *dst, vpx_image_t *src);
#endif

162
#ifdef __cplusplus
163
} /* extern "C" */
164 165
#endif

166
#endif  // TOOLS_COMMON_H_