granicus.if.org Git - libvpx/blob - examples/set_maps.c

   1 /*
   2  *  Copyright (c) 2010 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
  12 // VP8 Set Active and ROI Maps
  13 // ===========================
  14 //
  15 // This is an example demonstrating how to control the VP8 encoder's
  16 // ROI and Active maps.
  17 //
  18 // ROI (Reigon of Interest) maps are a way for the application to assign
  19 // each macroblock in the image to a region, and then set quantizer and
  20 // filtering parameters on that image.
  21 //
  22 // Active maps are a way for the application to specify on a
  23 // macroblock-by-macroblock basis whether there is any activity in that
  24 // macroblock.
  25 //
  26 //
  27 // Configuration
  28 // -------------
  29 // An ROI map is set on frame 22. If the width of the image in macroblocks
  30 // is evenly divisble by 4, then the output will appear to have distinct
  31 // columns, where the quantizer, loopfilter, and static threshold differ
  32 // from column to column.
  33 //
  34 // An active map is set on frame 33. If the width of the image in macroblocks
  35 // is evenly divisble by 4, then the output will appear to have distinct
  36 // columns, where one column will have motion and the next will not.
  37 //
  38 // The active map is cleared on frame 44.
  39 //
  40 // Observing The Effects
  41 // ---------------------
  42 // Use the `simple_decoder` example to decode this sample, and observe
  43 // the change in the image at frames 22, 33, and 44.
  44
  45 #include <assert.h>
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <string.h>
  49
  50 #define VPX_CODEC_DISABLE_COMPAT 1
  51 #include "vpx/vp8cx.h"
  52 #include "vpx/vpx_encoder.h"
  53
  54 #include "./tools_common.h"
  55 #include "./video_writer.h"
  56
  57 static const char *exec_name;
  58
  59 void usage_exit() {
  60   fprintf(stderr, "Usage: %s <codec> <width> <height> <infile> <outfile>\n",
  61           exec_name);
  62   exit(EXIT_FAILURE);
  63 }
  64
  65 static void set_roi_map(const vpx_codec_enc_cfg_t *cfg,
  66                         vpx_codec_ctx_t *codec) {
  67   unsigned int i;
  68   vpx_roi_map_t roi;
  69   memset(&roi, 0, sizeof(roi));
  70
  71   roi.rows = (cfg->g_h + 15) / 16;
  72   roi.cols = (cfg->g_w + 15) / 16;
  73
  74   roi.delta_q[0] = 0;
  75   roi.delta_q[1] = -2;
  76   roi.delta_q[2] = -4;
  77   roi.delta_q[3] = -6;
  78
  79   roi.delta_lf[0] = 0;
  80   roi.delta_lf[1] = 1;
  81   roi.delta_lf[2] = 2;
  82   roi.delta_lf[3] = 3;
  83
  84   roi.static_threshold[0] = 1500;
  85   roi.static_threshold[1] = 1000;
  86   roi.static_threshold[2] = 500;
  87   roi.static_threshold[3] = 0;
  88
  89   roi.roi_map = (uint8_t *)malloc(roi.rows * roi.cols);
  90   for (i = 0; i < roi.rows * roi.cols; ++i)
  91     roi.roi_map[i] = i % 4;
  92
  93   if (vpx_codec_control(codec, VP8E_SET_ROI_MAP, &roi))
  94     die_codec(codec, "Failed to set ROI map");
  95
  96   free(roi.roi_map);
  97 }
  98
  99 static void set_active_map(const vpx_codec_enc_cfg_t *cfg,
 100                            vpx_codec_ctx_t *codec) {
 101   unsigned int i;
 102   vpx_active_map_t map = {0, 0, 0};
 103
 104   map.rows = (cfg->g_h + 15) / 16;
 105   map.cols = (cfg->g_w + 15) / 16;
 106
 107   map.active_map = (uint8_t *)malloc(map.rows * map.cols);
 108   for (i = 0; i < map.rows * map.cols; ++i)
 109     map.active_map[i] = i % 2;
 110
 111   if (vpx_codec_control(codec, VP8E_SET_ACTIVEMAP, &map))
 112     die_codec(codec, "Failed to set active map");
 113
 114   free(map.active_map);
 115 }
 116
 117 static void unset_active_map(const vpx_codec_enc_cfg_t *cfg,
 118                              vpx_codec_ctx_t *codec) {
 119   vpx_active_map_t map = {0, 0, 0};
 120
 121   map.rows = (cfg->g_h + 15) / 16;
 122   map.cols = (cfg->g_w + 15) / 16;
 123   map.active_map = NULL;
 124
 125   if (vpx_codec_control(codec, VP8E_SET_ACTIVEMAP, &map))
 126     die_codec(codec, "Failed to set active map");
 127 }
 128
 129 static int encode_frame(vpx_codec_ctx_t *codec,
 130                         vpx_image_t *img,
 131                         int frame_index,
 132                         VpxVideoWriter *writer) {
 133   int got_pkts = 0;
 134   vpx_codec_iter_t iter = NULL;
 135   const vpx_codec_cx_pkt_t *pkt = NULL;
 136   const vpx_codec_err_t res = vpx_codec_encode(codec, img, frame_index, 1, 0,
 137                                                VPX_DL_GOOD_QUALITY);
 138   if (res != VPX_CODEC_OK)
 139     die_codec(codec, "Failed to encode frame");
 140
 141   while ((pkt = vpx_codec_get_cx_data(codec, &iter)) != NULL) {
 142     got_pkts = 1;
 143
 144     if (pkt->kind == VPX_CODEC_CX_FRAME_PKT) {
 145       const int keyframe = (pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0;
 146       if (!vpx_video_writer_write_frame(writer,
 147                                         pkt->data.frame.buf,
 148                                         pkt->data.frame.sz,
 149                                         pkt->data.frame.pts)) {
 150         die_codec(codec, "Failed to write compressed frame");
 151       }
 152
 153       printf(keyframe ? "K" : ".");
 154       fflush(stdout);
 155     }
 156   }
 157
 158   return got_pkts;
 159 }
 160
 161 int main(int argc, char **argv) {
 162   FILE *infile = NULL;
 163   vpx_codec_ctx_t codec;
 164   vpx_codec_enc_cfg_t cfg;
 165   int frame_count = 0;
 166   vpx_image_t raw;
 167   vpx_codec_err_t res;
 168   VpxVideoInfo info;
 169   VpxVideoWriter *writer = NULL;
 170   const VpxInterface *encoder = NULL;
 171   const int fps = 2;        // TODO(dkovalev) add command line argument
 172   const double bits_per_pixel_per_frame = 0.067;
 173
 174   exec_name = argv[0];
 175   if (argc != 6)
 176     die("Invalid number of arguments");
 177
 178   memset(&info, 0, sizeof(info));
 179
 180   encoder = get_vpx_encoder_by_name(argv[1]);
 181   if (encoder == NULL) {
 182     die("Unsupported codec.");
 183   }
 184   assert(encoder != NULL);
 185   info.codec_fourcc = encoder->fourcc;
 186   info.frame_width = strtol(argv[2], NULL, 0);
 187   info.frame_height = strtol(argv[3], NULL, 0);
 188   info.time_base.numerator = 1;
 189   info.time_base.denominator = fps;
 190
 191   if (info.frame_width <= 0 ||
 192       info.frame_height <= 0 ||
 193       (info.frame_width % 2) != 0 ||
 194       (info.frame_height % 2) != 0) {
 195     die("Invalid frame size: %dx%d", info.frame_width, info.frame_height);
 196   }
 197
 198   if (!vpx_img_alloc(&raw, VPX_IMG_FMT_I420, info.frame_width,
 199                                              info.frame_height, 1)) {
 200     die("Failed to allocate image.");
 201   }
 202
 203   printf("Using %s\n", vpx_codec_iface_name(encoder->codec_interface()));
 204
 205   res = vpx_codec_enc_config_default(encoder->codec_interface(), &cfg, 0);
 206   if (res)
 207     die_codec(&codec, "Failed to get default codec config.");
 208
 209   cfg.g_w = info.frame_width;
 210   cfg.g_h = info.frame_height;
 211   cfg.g_timebase.num = info.time_base.numerator;
 212   cfg.g_timebase.den = info.time_base.denominator;
 213   cfg.rc_target_bitrate = (unsigned int)(bits_per_pixel_per_frame * cfg.g_w *
 214                                          cfg.g_h * fps / 1000);
 215   cfg.g_lag_in_frames = 0;
 216
 217   writer = vpx_video_writer_open(argv[5], kContainerIVF, &info);
 218   if (!writer)
 219     die("Failed to open %s for writing.", argv[5]);
 220
 221   if (!(infile = fopen(argv[4], "rb")))
 222     die("Failed to open %s for reading.", argv[4]);
 223
 224   if (vpx_codec_enc_init(&codec, encoder->codec_interface(), &cfg, 0))
 225     die_codec(&codec, "Failed to initialize encoder");
 226
 227   // Encode frames.
 228   while (vpx_img_read(&raw, infile)) {
 229     ++frame_count;
 230
 231     if (frame_count == 22 && encoder->fourcc == VP8_FOURCC) {
 232       set_roi_map(&cfg, &codec);
 233     } else if (frame_count == 33) {
 234       set_active_map(&cfg, &codec);
 235     } else if (frame_count == 44) {
 236       unset_active_map(&cfg, &codec);
 237     }
 238
 239     encode_frame(&codec, &raw, frame_count, writer);
 240   }
 241
 242   // Flush encoder.
 243   while (encode_frame(&codec, NULL, -1, writer)) {}
 244
 245   printf("\n");
 246   fclose(infile);
 247   printf("Processed %d frames.\n", frame_count);
 248
 249   vpx_img_free(&raw);
 250   if (vpx_codec_destroy(&codec))
 251     die_codec(&codec, "Failed to destroy codec.");
 252
 253   vpx_video_writer_close(writer);
 254
 255   return EXIT_SUCCESS;
 256 }