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)
38  x = strchr(v, 'X');
39  if (x == NULL)
40  return 0;
41  *width = atoi(v);
42  *height = atoi(&x[1]);
43  if (*width <= 0 || *height <= 0)
44  return 0;
45  else
46  return 1;
47 }
48 
49 int main(int argc, char *argv[]) {
50  char *fin, *fout;
51  FILE *fpin, *fpout;
52  uint8_t *inbuf, *outbuf;
53  uint8_t *inbuf_u, *outbuf_u;
54  uint8_t *inbuf_v, *outbuf_v;
55  int f, frames;
56  int width, height, target_width, target_height;
57 
58  exec_name = argv[0];
59 
60  if (argc < 5) {
61  printf("Incorrect parameters:\n");
62  usage();
63  return 1;
64  }
65 
66  fin = argv[1];
67  fout = argv[4];
68  if (!parse_dim(argv[2], &width, &height)) {
69  printf("Incorrect parameters: %s\n", argv[2]);
70  usage();
71  return 1;
72  }
73  if (!parse_dim(argv[3], &target_width, &target_height)) {
74  printf("Incorrect parameters: %s\n", argv[3]);
75  usage();
76  return 1;
77  }
78 
79  fpin = fopen(fin, "rb");
80  if (fpin == NULL) {
81  printf("Can't open file %s to read\n", fin);
82  usage();
83  return 1;
84  }
85  fpout = fopen(fout, "wb");
86  if (fpout == NULL) {
87  printf("Can't open file %s to write\n", fout);
88  usage();
89  return 1;
90  }
91  if (argc >= 6)
92  frames = atoi(argv[5]);
93  else
94  frames = INT_MAX;
95 
96  printf("Input size: %dx%d\n",
97  width, height);
98  printf("Target size: %dx%d, Frames: ",
99  target_width, target_height);
100  if (frames == INT_MAX)
101  printf("All\n");
102  else
103  printf("%d\n", frames);
104 
105  inbuf = (uint8_t*)malloc(width * height * 3 / 2);
106  outbuf = (uint8_t*)malloc(target_width * target_height * 3 / 2);
107  inbuf_u = inbuf + width * height;
108  inbuf_v = inbuf_u + width * height / 4;
109  outbuf_u = outbuf + target_width * target_height;
110  outbuf_v = outbuf_u + target_width * target_height / 4;
111  f = 0;
112  while (f < frames) {
113  if (fread(inbuf, width * height * 3 / 2, 1, fpin) != 1)
114  break;
115  vp9_resize_frame420(inbuf, width, inbuf_u, inbuf_v, width / 2,
116  height, width,
117  outbuf, target_width, outbuf_u, outbuf_v,
118  target_width / 2,
119  target_height, target_width);
120  fwrite(outbuf, target_width * target_height * 3 / 2, 1, fpout);
121  f++;
122  }
123  printf("%d frames processed\n", f);
124  fclose(fpin);
125  fclose(fpout);
126 
127  free(inbuf);
128  free(outbuf);
129  return 0;
130 }