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