| 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 |