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 |