set_maps.c 7.34 KB
Newer Older
John Koleszar's avatar
John Koleszar committed
1
/*
2
 *  Copyright (c) 2010 The WebM project authors. All Rights Reserved.
John Koleszar's avatar
John Koleszar committed
3
 *
4
 *  Use of this source code is governed by a BSD-style license
5 6
 *  that can be found in the LICENSE file in the root of the source
 *  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.
John Koleszar's avatar
John Koleszar committed
9 10 11
 */


12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
// VP8 Set Active and ROI Maps
// ===========================
//
// This is an example demonstrating how to control the VP8 encoder's
// ROI and Active maps.
//
// ROI (Reigon of Interest) maps are a way for the application to assign
// each macroblock in the image to a region, and then set quantizer and
// filtering parameters on that image.
//
// Active maps are a way for the application to specify on a
// macroblock-by-macroblock basis whether there is any activity in that
// macroblock.
//
//
// Configuration
// -------------
// An ROI map is set on frame 22. If the width of the image in macroblocks
// is evenly divisble by 4, then the output will appear to have distinct
// columns, where the quantizer, loopfilter, and static threshold differ
// from column to column.
//
// An active map is set on frame 33. If the width of the image in macroblocks
// is evenly divisble by 4, then the output will appear to have distinct
// columns, where one column will have motion and the next will not.
//
// The active map is cleared on frame 44.
//
// Observing The Effects
// ---------------------
// Use the `simple_decoder` example to decode this sample, and observe
// the change in the image at frames 22, 33, and 44.

John Koleszar's avatar
John Koleszar committed
45 46 47
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
48

John Koleszar's avatar
John Koleszar committed
49
#define VPX_CODEC_DISABLE_COMPAT 1
50
#include "vpx/vp8cx.h"
51
#include "vpx/vpx_encoder.h"
John Koleszar's avatar
John Koleszar committed
52

53 54
#include "./tools_common.h"
#include "./video_writer.h"
John Koleszar's avatar
John Koleszar committed
55

56
static const char *exec_name;
John Koleszar's avatar
John Koleszar committed
57

58
void usage_exit() {
59 60
  fprintf(stderr, "Usage: %s <codec> <width> <height> <infile> <outfile>\n",
          exec_name);
61
  exit(EXIT_FAILURE);
John Koleszar's avatar
John Koleszar committed
62 63
}

64 65
static void set_roi_map(const vpx_codec_enc_cfg_t *cfg,
                        vpx_codec_ctx_t *codec) {
66
  unsigned int i;
Yaowu Xu's avatar
Yaowu Xu committed
67 68
  vpx_roi_map_t roi;
  memset(&roi, 0, sizeof(roi));
John Koleszar's avatar
John Koleszar committed
69

70 71
  roi.rows = (cfg->g_h + 15) / 16;
  roi.cols = (cfg->g_w + 15) / 16;
John Koleszar's avatar
John Koleszar committed
72

73 74 75 76
  roi.delta_q[0] = 0;
  roi.delta_q[1] = -2;
  roi.delta_q[2] = -4;
  roi.delta_q[3] = -6;
77

78 79 80 81
  roi.delta_lf[0] = 0;
  roi.delta_lf[1] = 1;
  roi.delta_lf[2] = 2;
  roi.delta_lf[3] = 3;
John Koleszar's avatar
John Koleszar committed
82

83 84 85 86
  roi.static_threshold[0] = 1500;
  roi.static_threshold[1] = 1000;
  roi.static_threshold[2] = 500;
  roi.static_threshold[3] = 0;
John Koleszar's avatar
John Koleszar committed
87

88 89 90
  roi.roi_map = (uint8_t *)malloc(roi.rows * roi.cols);
  for (i = 0; i < roi.rows * roi.cols; ++i)
    roi.roi_map[i] = i % 4;
John Koleszar's avatar
John Koleszar committed
91

92 93 94 95
  if (vpx_codec_control(codec, VP8E_SET_ROI_MAP, &roi))
    die_codec(codec, "Failed to set ROI map");

  free(roi.roi_map);
John Koleszar's avatar
John Koleszar committed
96 97
}

98 99
static void set_active_map(const vpx_codec_enc_cfg_t *cfg,
                           vpx_codec_ctx_t *codec) {
100
  unsigned int i;
Yaowu Xu's avatar
Yaowu Xu committed
101
  vpx_active_map_t map = {0, 0, 0};
John Koleszar's avatar
John Koleszar committed
102

103 104
  map.rows = (cfg->g_h + 15) / 16;
  map.cols = (cfg->g_w + 15) / 16;
John Koleszar's avatar
John Koleszar committed
105

106 107 108
  map.active_map = (uint8_t *)malloc(map.rows * map.cols);
  for (i = 0; i < map.rows * map.cols; ++i)
    map.active_map[i] = i % 2;
John Koleszar's avatar
John Koleszar committed
109

110 111
  if (vpx_codec_control(codec, VP8E_SET_ACTIVEMAP, &map))
    die_codec(codec, "Failed to set active map");
John Koleszar's avatar
John Koleszar committed
112

113 114 115 116 117
  free(map.active_map);
}

static void unset_active_map(const vpx_codec_enc_cfg_t *cfg,
                             vpx_codec_ctx_t *codec) {
Yaowu Xu's avatar
Yaowu Xu committed
118
  vpx_active_map_t map = {0, 0, 0};
119

120 121
  map.rows = (cfg->g_h + 15) / 16;
  map.cols = (cfg->g_w + 15) / 16;
122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
  map.active_map = NULL;

  if (vpx_codec_control(codec, VP8E_SET_ACTIVEMAP, &map))
    die_codec(codec, "Failed to set active map");
}

static void encode_frame(vpx_codec_ctx_t *codec,
                         vpx_image_t *img,
                         int frame_index,
                         VpxVideoWriter *writer) {
  vpx_codec_iter_t iter = NULL;
  const vpx_codec_cx_pkt_t *pkt = NULL;
  const vpx_codec_err_t res = vpx_codec_encode(codec, img, frame_index, 1, 0,
                                               VPX_DL_GOOD_QUALITY);
  if (res != VPX_CODEC_OK)
    die_codec(codec, "Failed to encode frame");

  while ((pkt = vpx_codec_get_cx_data(codec, &iter)) != NULL) {
    if (pkt->kind == VPX_CODEC_CX_FRAME_PKT) {
      const int keyframe = (pkt->data.frame.flags & VPX_FRAME_IS_KEY) != 0;
      if (!vpx_video_writer_write_frame(writer,
                                        pkt->data.frame.buf,
                                        pkt->data.frame.sz,
                                        pkt->data.frame.pts)) {
        die_codec(codec, "Failed to write compressed frame");
      }

      printf(keyframe ? "K" : ".");
      fflush(stdout);
    }
  }
John Koleszar's avatar
John Koleszar committed
153 154 155
}

int main(int argc, char **argv) {
156
  FILE *infile = NULL;
Yaowu Xu's avatar
Yaowu Xu committed
157 158
  vpx_codec_ctx_t codec;
  vpx_codec_enc_cfg_t cfg;
159
  int frame_count = 0;
Yaowu Xu's avatar
Yaowu Xu committed
160
  vpx_image_t raw;
161
  vpx_codec_err_t res;
Yaowu Xu's avatar
Yaowu Xu committed
162
  VpxVideoInfo info;
163 164 165
  VpxVideoWriter *writer = NULL;
  const VpxInterface *encoder = NULL;
  const int fps = 2;        // TODO(dkovalev) add command line argument
166
  const double bits_per_pixel_per_frame = 0.067;
167 168

  exec_name = argv[0];
169
  if (argc != 6)
170 171
    die("Invalid number of arguments");

Yaowu Xu's avatar
Yaowu Xu committed
172 173
  memset(&info, 0, sizeof(info));

174
  encoder = get_vpx_encoder_by_name(argv[1]);
175 176 177 178
  if (!encoder)
    die("Unsupported codec.");

  info.codec_fourcc = encoder->fourcc;
179 180
  info.frame_width = strtol(argv[2], NULL, 0);
  info.frame_height = strtol(argv[3], NULL, 0);
181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
  info.time_base.numerator = 1;
  info.time_base.denominator = fps;

  if (info.frame_width <= 0 ||
      info.frame_height <= 0 ||
      (info.frame_width % 2) != 0 ||
      (info.frame_height % 2) != 0) {
    die("Invalid frame size: %dx%d", info.frame_width, info.frame_height);
  }

  if (!vpx_img_alloc(&raw, VPX_IMG_FMT_I420, info.frame_width,
                                             info.frame_height, 1)) {
    die("Failed to allocate image.");
  }

196
  printf("Using %s\n", vpx_codec_iface_name(encoder->codec_interface()));
197

198
  res = vpx_codec_enc_config_default(encoder->codec_interface(), &cfg, 0);
199 200 201 202 203 204 205
  if (res)
    die_codec(&codec, "Failed to get default codec config.");

  cfg.g_w = info.frame_width;
  cfg.g_h = info.frame_height;
  cfg.g_timebase.num = info.time_base.numerator;
  cfg.g_timebase.den = info.time_base.denominator;
206 207
  cfg.rc_target_bitrate = (unsigned int)(bits_per_pixel_per_frame * cfg.g_w *
                                         cfg.g_h * fps / 1000);
208
  cfg.g_lag_in_frames = 0;
209

210
  writer = vpx_video_writer_open(argv[5], kContainerIVF, &info);
211
  if (!writer)
212
    die("Failed to open %s for writing.", argv[5]);
213

214 215
  if (!(infile = fopen(argv[4], "rb")))
    die("Failed to open %s for reading.", argv[4]);
216

217
  if (vpx_codec_enc_init(&codec, encoder->codec_interface(), &cfg, 0))
218 219 220 221 222
    die_codec(&codec, "Failed to initialize encoder");

  while (vpx_img_read(&raw, infile)) {
    ++frame_count;

223
    if (frame_count == 22 && encoder->fourcc == VP8_FOURCC) {
224 225 226 227 228
      set_roi_map(&cfg, &codec);
    } else if (frame_count == 33) {
      set_active_map(&cfg, &codec);
    } else if (frame_count == 44) {
      unset_active_map(&cfg, &codec);
229
    }
John Koleszar's avatar
John Koleszar committed
230

231 232 233 234 235 236 237 238 239 240 241 242 243 244
    encode_frame(&codec, &raw, frame_count, writer);
  }
  encode_frame(&codec, NULL, -1, writer);
  printf("\n");
  fclose(infile);
  printf("Processed %d frames.\n", frame_count);

  vpx_img_free(&raw);
  if (vpx_codec_destroy(&codec))
    die_codec(&codec, "Failed to destroy codec.");

  vpx_video_writer_close(writer);

  return EXIT_SUCCESS;
John Koleszar's avatar
John Koleszar committed
245
}