WebM Codec SDK
resize_util
1 /*
2  * Copyright (c) 2014 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 
11 #include <assert.h>
12 #include <limits.h>
13 #include <math.h>
14 #include <stdio.h>
15 #include <stdlib.h>
16 #include <string.h>
17 
18 #include "../tools_common.h"
19 #include "../vp9/encoder/vp9_resize.h"
20 
21 static const char *exec_name = NULL;
22 
23 static void usage() {
24  printf("Usage:\n");
25  printf("%s <input_yuv> <width>x<height> <target_width>x<target_height> ",
26  exec_name);
27  printf("<output_yuv> [<frames>]\n");
28 }
29 
30 void usage_exit(void) {
31  usage();
32  exit(EXIT_FAILURE);
33 }
34 
35 static int parse_dim(char *v, int *width, int *height) {
36  char *x = strchr(v, 'x');
37  if (x == NULL) x = strchr(v, 'X');
38  if (x == NULL) return 0;
39  *width = atoi(v);
40  *height = atoi(&x[1]);
41  if (*width <= 0 || *height <= 0)
42  return 0;
43  else
44  return 1;
45 }
46 
47 int main(int argc, char *argv[]) {
48  char *fin, *fout;
49  FILE *fpin, *fpout;
50  uint8_t *inbuf, *outbuf;
51  uint8_t *inbuf_u, *outbuf_u;
52  uint8_t *inbuf_v, *outbuf_v;
53  int f, frames;
54  int width, height, target_width, target_height;
55 
56  exec_name = argv[0];
57 
58  if (argc < 5) {
59  printf("Incorrect parameters:\n");
60  usage();
61  return 1;
62  }
63 
64  fin = argv[1];
65  fout = argv[4];
66  if (!parse_dim(argv[2], &width, &height)) {
67  printf("Incorrect parameters: %s\n", argv[2]);
68  usage();
69  return 1;
70  }
71  if (!parse_dim(argv[3], &target_width, &target_height)) {
72  printf("Incorrect parameters: %s\n", argv[3]);
73  usage();
74  return 1;
75  }
76 
77  fpin = fopen(fin, "rb");
78  if (fpin == NULL) {
79  printf("Can't open file %s to read\n", fin);
80  usage();
81  return 1;
82  }
83  fpout = fopen(fout, "wb");
84  if (fpout == NULL) {
85  printf("Can't open file %s to write\n", fout);
86  usage();
87  return 1;
88  }
89  if (argc >= 6)
90  frames = atoi(argv[5]);
91  else
92  frames = INT_MAX;
93 
94  printf("Input size: %dx%d\n", width, height);
95  printf("Target size: %dx%d, Frames: ", target_width, target_height);
96  if (frames == INT_MAX)
97  printf("All\n");
98  else
99  printf("%d\n", frames);
100 
101  inbuf = (uint8_t *)malloc(width * height * 3 / 2);
102  outbuf = (uint8_t *)malloc(target_width * target_height * 3 / 2);
103  inbuf_u = inbuf + width * height;
104  inbuf_v = inbuf_u + width * height / 4;
105  outbuf_u = outbuf + target_width * target_height;
106  outbuf_v = outbuf_u + target_width * target_height / 4;
107  f = 0;
108  while (f < frames) {
109  if (fread(inbuf, width * height * 3 / 2, 1, fpin) != 1) break;
110  vp9_resize_frame420(inbuf, width, inbuf_u, inbuf_v, width / 2, height,
111  width, outbuf, target_width, outbuf_u, outbuf_v,
112  target_width / 2, target_height, target_width);
113  fwrite(outbuf, target_width * target_height * 3 / 2, 1, fpout);
114  f++;
115  }
116  printf("%d frames processed\n", f);
117  fclose(fpin);
118  fclose(fpout);
119 
120  free(inbuf);
121  free(outbuf);
122  return 0;
123 }