examples/resize_util.c

   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 }