src/third_party/libvpx/source/libvpx/vpx/src/svc_encodeframe.c

   1 /*
   2  *  Copyright (c) 2013 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  * @file
  13  * VP9 SVC encoding support via libvpx
  14  */
  15
  16 #include <assert.h>
  17 #include <math.h>
  18 #include <limits.h>
  19 #include <stdarg.h>
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #define VPX_DISABLE_CTRL_TYPECHECKS 1
  24 #include "./vpx_config.h"
  25 #include "vpx/svc_context.h"
  26 #include "vpx/vp8cx.h"
  27 #include "vpx/vpx_encoder.h"
  28 #include "vpx_mem/vpx_mem.h"
  29 #include "vp9/common/vp9_onyxc_int.h"
  30
  31 #ifdef __MINGW32__
  32 #define strtok_r strtok_s
  33 #ifndef MINGW_HAS_SECURE_API
  34 // proto from /usr/x86_64-w64-mingw32/include/sec_api/string_s.h
  35 _CRTIMP char *__cdecl strtok_s(char *str, const char *delim, char **context);
  36 #endif  /* MINGW_HAS_SECURE_API */
  37 #endif  /* __MINGW32__ */
  38
  39 #ifdef _MSC_VER
  40 #define strdup _strdup
  41 #define strtok_r strtok_s
  42 #endif
  43
  44 #define SVC_REFERENCE_FRAMES 8
  45 #define SUPERFRAME_SLOTS (8)
  46 #define SUPERFRAME_BUFFER_SIZE (SUPERFRAME_SLOTS * sizeof(uint32_t) + 2)
  47 #define OPTION_BUFFER_SIZE 1024
  48 #define COMPONENTS 4  // psnr & sse statistics maintained for total, y, u, v
  49
  50 #define MAX_QUANTIZER 63
  51
  52 static const int DEFAULT_SCALE_FACTORS_NUM[VPX_SS_MAX_LAYERS] = {
  53   4, 5, 7, 11, 16
  54 };
  55
  56 static const int DEFAULT_SCALE_FACTORS_DEN[VPX_SS_MAX_LAYERS] = {
  57   16, 16, 16, 16, 16
  58 };
  59
  60 typedef enum {
  61   QUANTIZER = 0,
  62   BITRATE,
  63   SCALE_FACTOR,
  64   AUTO_ALT_REF,
  65   ALL_OPTION_TYPES
  66 } LAYER_OPTION_TYPE;
  67
  68 static const int option_max_values[ALL_OPTION_TYPES] = {
  69   63, INT_MAX, INT_MAX, 1
  70 };
  71
  72 static const int option_min_values[ALL_OPTION_TYPES] = {
  73   0, 0, 1, 0
  74 };
  75
  76 // One encoded frame
  77 typedef struct FrameData {
  78   void                     *buf;    // compressed data buffer
  79   size_t                    size;  // length of compressed data
  80   vpx_codec_frame_flags_t   flags;    /**< flags for this frame */
  81   struct FrameData         *next;
  82 } FrameData;
  83
  84 typedef struct SvcInternal {
  85   char options[OPTION_BUFFER_SIZE];        // set by vpx_svc_set_options
  86
  87   // values extracted from option, quantizers
  88   vpx_svc_extra_cfg_t svc_params;
  89   int enable_auto_alt_ref[VPX_SS_MAX_LAYERS];
  90   int bitrates[VPX_SS_MAX_LAYERS];
  91
  92   // accumulated statistics
  93   double psnr_sum[VPX_SS_MAX_LAYERS][COMPONENTS];   // total/Y/U/V
  94   uint64_t sse_sum[VPX_SS_MAX_LAYERS][COMPONENTS];
  95   uint32_t bytes_sum[VPX_SS_MAX_LAYERS];
  96
  97   // codec encoding values
  98   int width;    // width of highest layer
  99   int height;   // height of highest layer
 100   int kf_dist;  // distance between keyframes
 101
 102   // state variables
 103   int psnr_pkt_received;
 104   int layer;
 105   int use_multiple_frame_contexts;
 106
 107   char message_buffer[2048];
 108   vpx_codec_ctx_t *codec_ctx;
 109 } SvcInternal;
 110
 111 static SvcInternal *get_svc_internal(SvcContext *svc_ctx) {
 112   if (svc_ctx == NULL) return NULL;
 113   if (svc_ctx->internal == NULL) {
 114     SvcInternal *const si = (SvcInternal *)malloc(sizeof(*si));
 115     if (si != NULL) {
 116       memset(si, 0, sizeof(*si));
 117     }
 118     svc_ctx->internal = si;
 119   }
 120   return (SvcInternal *)svc_ctx->internal;
 121 }
 122
 123 static const SvcInternal *get_const_svc_internal(const SvcContext *svc_ctx) {
 124   if (svc_ctx == NULL) return NULL;
 125   return (const SvcInternal *)svc_ctx->internal;
 126 }
 127
 128 static void svc_log_reset(SvcContext *svc_ctx) {
 129   SvcInternal *const si = (SvcInternal *)svc_ctx->internal;
 130   si->message_buffer[0] = '\0';
 131 }
 132
 133 static int svc_log(SvcContext *svc_ctx, SVC_LOG_LEVEL level,
 134                    const char *fmt, ...) {
 135   char buf[512];
 136   int retval = 0;
 137   va_list ap;
 138   SvcInternal *const si = get_svc_internal(svc_ctx);
 139
 140   if (level > svc_ctx->log_level) {
 141     return retval;
 142   }
 143
 144   va_start(ap, fmt);
 145   retval = vsnprintf(buf, sizeof(buf), fmt, ap);
 146   va_end(ap);
 147
 148   if (svc_ctx->log_print) {
 149     printf("%s", buf);
 150   } else {
 151     strncat(si->message_buffer, buf,
 152             sizeof(si->message_buffer) - strlen(si->message_buffer) - 1);
 153   }
 154
 155   if (level == SVC_LOG_ERROR) {
 156     si->codec_ctx->err_detail = si->message_buffer;
 157   }
 158   return retval;
 159 }
 160
 161 static vpx_codec_err_t extract_option(LAYER_OPTION_TYPE type,
 162                                       char *input,
 163                                       int *value0,
 164                                       int *value1) {
 165   if (type == SCALE_FACTOR) {
 166     *value0 = strtol(input, &input, 10);
 167     if (*input++ != '/')
 168       return VPX_CODEC_INVALID_PARAM;
 169     *value1 = strtol(input, &input, 10);
 170
 171     if (*value0 < option_min_values[SCALE_FACTOR] ||
 172         *value1 < option_min_values[SCALE_FACTOR] ||
 173         *value0 > option_max_values[SCALE_FACTOR] ||
 174         *value1 > option_max_values[SCALE_FACTOR] ||
 175         *value0 > *value1)  // num shouldn't be greater than den
 176       return VPX_CODEC_INVALID_PARAM;
 177   } else {
 178     *value0 = atoi(input);
 179     if (*value0 < option_min_values[type] ||
 180         *value0 > option_max_values[type])
 181       return VPX_CODEC_INVALID_PARAM;
 182   }
 183   return VPX_CODEC_OK;
 184 }
 185
 186 static vpx_codec_err_t parse_layer_options_from_string(SvcContext *svc_ctx,
 187                                                        LAYER_OPTION_TYPE type,
 188                                                        const char *input,
 189                                                        int *option0,
 190                                                        int *option1) {
 191   int i;
 192   vpx_codec_err_t res = VPX_CODEC_OK;
 193   char *input_string;
 194   char *token;
 195   const char *delim = ",";
 196   char *save_ptr;
 197
 198   if (input == NULL || option0 == NULL ||
 199       (option1 == NULL && type == SCALE_FACTOR))
 200     return VPX_CODEC_INVALID_PARAM;
 201
 202   input_string = strdup(input);
 203   token = strtok_r(input_string, delim, &save_ptr);
 204   for (i = 0; i < svc_ctx->spatial_layers; ++i) {
 205     if (token != NULL) {
 206       res = extract_option(type, token, option0 + i, option1 + i);
 207       if (res != VPX_CODEC_OK)
 208         break;
 209       token = strtok_r(NULL, delim, &save_ptr);
 210     } else {
 211       break;
 212     }
 213   }
 214   if (res == VPX_CODEC_OK && i != svc_ctx->spatial_layers) {
 215     svc_log(svc_ctx, SVC_LOG_ERROR,
 216             "svc: layer params type: %d    %d values required, "
 217             "but only %d specified\n", type, svc_ctx->spatial_layers, i);
 218     res = VPX_CODEC_INVALID_PARAM;
 219   }
 220   free(input_string);
 221   return res;
 222 }
 223
 224 /**
 225  * Parse SVC encoding options
 226  * Format: encoding-mode=<svc_mode>,layers=<layer_count>
 227  *         scale-factors=<n1>/<d1>,<n2>/<d2>,...
 228  *         quantizers=<q1>,<q2>,...
 229  * svc_mode = [i|ip|alt_ip|gf]
 230  */
 231 static vpx_codec_err_t parse_options(SvcContext *svc_ctx, const char *options) {
 232   char *input_string;
 233   char *option_name;
 234   char *option_value;
 235   char *input_ptr;
 236   SvcInternal *const si = get_svc_internal(svc_ctx);
 237   vpx_codec_err_t res = VPX_CODEC_OK;
 238   int i, alt_ref_enabled = 0;
 239
 240   if (options == NULL) return VPX_CODEC_OK;
 241   input_string = strdup(options);
 242
 243   // parse option name
 244   option_name = strtok_r(input_string, "=", &input_ptr);
 245   while (option_name != NULL) {
 246     // parse option value
 247     option_value = strtok_r(NULL, " ", &input_ptr);
 248     if (option_value == NULL) {
 249       svc_log(svc_ctx, SVC_LOG_ERROR, "option missing value: %s\n",
 250               option_name);
 251       res = VPX_CODEC_INVALID_PARAM;
 252       break;
 253     }
 254     if (strcmp("spatial-layers", option_name) == 0) {
 255       svc_ctx->spatial_layers = atoi(option_value);
 256     } else if (strcmp("temporal-layers", option_name) == 0) {
 257       svc_ctx->temporal_layers = atoi(option_value);
 258     } else if (strcmp("scale-factors", option_name) == 0) {
 259       res = parse_layer_options_from_string(svc_ctx, SCALE_FACTOR, option_value,
 260                                             si->svc_params.scaling_factor_num,
 261                                             si->svc_params.scaling_factor_den);
 262       if (res != VPX_CODEC_OK) break;
 263     } else if (strcmp("max-quantizers", option_name) == 0) {
 264       res = parse_layer_options_from_string(svc_ctx, QUANTIZER, option_value,
 265                                             si->svc_params.max_quantizers,
 266                                             NULL);
 267       if (res != VPX_CODEC_OK) break;
 268     } else if (strcmp("min-quantizers", option_name) == 0) {
 269       res = parse_layer_options_from_string(svc_ctx, QUANTIZER, option_value,
 270                                             si->svc_params.min_quantizers,
 271                                             NULL);
 272       if (res != VPX_CODEC_OK) break;
 273     } else if (strcmp("auto-alt-refs", option_name) == 0) {
 274       res = parse_layer_options_from_string(svc_ctx, AUTO_ALT_REF, option_value,
 275                                             si->enable_auto_alt_ref, NULL);
 276       if (res != VPX_CODEC_OK) break;
 277     } else if (strcmp("bitrates", option_name) == 0) {
 278       res = parse_layer_options_from_string(svc_ctx, BITRATE, option_value,
 279                                             si->bitrates, NULL);
 280       if (res != VPX_CODEC_OK) break;
 281     } else if (strcmp("multi-frame-contexts", option_name) == 0) {
 282       si->use_multiple_frame_contexts = atoi(option_value);
 283     } else {
 284       svc_log(svc_ctx, SVC_LOG_ERROR, "invalid option: %s\n", option_name);
 285       res = VPX_CODEC_INVALID_PARAM;
 286       break;
 287     }
 288     option_name = strtok_r(NULL, "=", &input_ptr);
 289   }
 290   free(input_string);
 291
 292   for (i = 0; i < svc_ctx->spatial_layers; ++i) {
 293     if (si->svc_params.max_quantizers[i] > MAX_QUANTIZER ||
 294         si->svc_params.max_quantizers[i] < 0 ||
 295         si->svc_params.min_quantizers[i] > si->svc_params.max_quantizers[i] ||
 296         si->svc_params.min_quantizers[i] < 0)
 297       res = VPX_CODEC_INVALID_PARAM;
 298   }
 299
 300   if (si->use_multiple_frame_contexts &&
 301       (svc_ctx->spatial_layers > 3 ||
 302        svc_ctx->spatial_layers * svc_ctx->temporal_layers > 4))
 303     res = VPX_CODEC_INVALID_PARAM;
 304
 305   for (i = 0; i < svc_ctx->spatial_layers; ++i)
 306     alt_ref_enabled += si->enable_auto_alt_ref[i];
 307   if (alt_ref_enabled > REF_FRAMES - svc_ctx->spatial_layers) {
 308     svc_log(svc_ctx, SVC_LOG_ERROR,
 309             "svc: auto alt ref: Maxinum %d(REF_FRAMES - layers) layers could"
 310             "enabled auto alt reference frame, but % layers are enabled\n",
 311             REF_FRAMES - svc_ctx->spatial_layers, alt_ref_enabled);
 312     res = VPX_CODEC_INVALID_PARAM;
 313   }
 314
 315   return res;
 316 }
 317
 318 vpx_codec_err_t vpx_svc_set_options(SvcContext *svc_ctx, const char *options) {
 319   SvcInternal *const si = get_svc_internal(svc_ctx);
 320   if (svc_ctx == NULL || options == NULL || si == NULL) {
 321     return VPX_CODEC_INVALID_PARAM;
 322   }
 323   strncpy(si->options, options, sizeof(si->options));
 324   si->options[sizeof(si->options) - 1] = '\0';
 325   return VPX_CODEC_OK;
 326 }
 327
 328 void assign_layer_bitrates(const SvcContext *svc_ctx,
 329                            vpx_codec_enc_cfg_t *const enc_cfg) {
 330   int i;
 331   const SvcInternal *const si = get_const_svc_internal(svc_ctx);
 332
 333   if (si->bitrates[0] != 0) {
 334     enc_cfg->rc_target_bitrate = 0;
 335     for (i = 0; i < svc_ctx->spatial_layers; ++i) {
 336       enc_cfg->ss_target_bitrate[i] = (unsigned int)si->bitrates[i];
 337       enc_cfg->rc_target_bitrate += si->bitrates[i];
 338     }
 339   } else {
 340     float total = 0;
 341     float alloc_ratio[VPX_SS_MAX_LAYERS] = {0};
 342
 343     for (i = 0; i < svc_ctx->spatial_layers; ++i) {
 344       if (si->svc_params.scaling_factor_den[i] > 0) {
 345         alloc_ratio[i] = (float)(si->svc_params.scaling_factor_num[i] * 1.0 /
 346                                  si->svc_params.scaling_factor_den[i]);
 347
 348         alloc_ratio[i] *= alloc_ratio[i];
 349         total += alloc_ratio[i];
 350       }
 351     }
 352
 353     for (i = 0; i < VPX_SS_MAX_LAYERS; ++i) {
 354       if (total > 0) {
 355         enc_cfg->ss_target_bitrate[i] = (unsigned int)
 356             (enc_cfg->rc_target_bitrate * alloc_ratio[i] / total);
 357       }
 358     }
 359   }
 360 }
 361
 362 vpx_codec_err_t vpx_svc_init(SvcContext *svc_ctx, vpx_codec_ctx_t *codec_ctx,
 363                              vpx_codec_iface_t *iface,
 364                              vpx_codec_enc_cfg_t *enc_cfg) {
 365   vpx_codec_err_t res;
 366   int i;
 367   SvcInternal *const si = get_svc_internal(svc_ctx);
 368   if (svc_ctx == NULL || codec_ctx == NULL || iface == NULL ||
 369       enc_cfg == NULL) {
 370     return VPX_CODEC_INVALID_PARAM;
 371   }
 372   if (si == NULL) return VPX_CODEC_MEM_ERROR;
 373
 374   si->codec_ctx = codec_ctx;
 375
 376   si->width = enc_cfg->g_w;
 377   si->height = enc_cfg->g_h;
 378
 379   if (enc_cfg->kf_max_dist < 2) {
 380     svc_log(svc_ctx, SVC_LOG_ERROR, "key frame distance too small: %d\n",
 381             enc_cfg->kf_max_dist);
 382     return VPX_CODEC_INVALID_PARAM;
 383   }
 384   si->kf_dist = enc_cfg->kf_max_dist;
 385
 386   if (svc_ctx->spatial_layers == 0)
 387     svc_ctx->spatial_layers = VPX_SS_DEFAULT_LAYERS;
 388   if (svc_ctx->spatial_layers < 1 ||
 389       svc_ctx->spatial_layers > VPX_SS_MAX_LAYERS) {
 390     svc_log(svc_ctx, SVC_LOG_ERROR, "spatial layers: invalid value: %d\n",
 391             svc_ctx->spatial_layers);
 392     return VPX_CODEC_INVALID_PARAM;
 393   }
 394
 395   for (i = 0; i < VPX_SS_MAX_LAYERS; ++i) {
 396     si->svc_params.max_quantizers[i] = MAX_QUANTIZER;
 397     si->svc_params.min_quantizers[i] = 0;
 398     si->svc_params.scaling_factor_num[i] = DEFAULT_SCALE_FACTORS_NUM[i];
 399     si->svc_params.scaling_factor_den[i] = DEFAULT_SCALE_FACTORS_DEN[i];
 400   }
 401
 402   // Parse aggregate command line options. Options must start with
 403   // "layers=xx" then followed by other options
 404   res = parse_options(svc_ctx, si->options);
 405   if (res != VPX_CODEC_OK) return res;
 406
 407   if (svc_ctx->spatial_layers < 1)
 408     svc_ctx->spatial_layers = 1;
 409   if (svc_ctx->spatial_layers > VPX_SS_MAX_LAYERS)
 410     svc_ctx->spatial_layers = VPX_SS_MAX_LAYERS;
 411
 412   if (svc_ctx->temporal_layers < 1)
 413     svc_ctx->temporal_layers = 1;
 414   if (svc_ctx->temporal_layers > VPX_TS_MAX_LAYERS)
 415     svc_ctx->temporal_layers = VPX_TS_MAX_LAYERS;
 416
 417   assign_layer_bitrates(svc_ctx, enc_cfg);
 418
 419 #if CONFIG_SPATIAL_SVC
 420   for (i = 0; i < svc_ctx->spatial_layers; ++i)
 421     enc_cfg->ss_enable_auto_alt_ref[i] = si->enable_auto_alt_ref[i];
 422 #endif
 423
 424   if (svc_ctx->temporal_layers > 1) {
 425     int i;
 426     for (i = 0; i < svc_ctx->temporal_layers; ++i) {
 427       enc_cfg->ts_target_bitrate[i] = enc_cfg->rc_target_bitrate /
 428                                       svc_ctx->temporal_layers;
 429       enc_cfg->ts_rate_decimator[i] = 1 << (svc_ctx->temporal_layers - 1 - i);
 430     }
 431   }
 432
 433   // modify encoder configuration
 434   enc_cfg->ss_number_layers = svc_ctx->spatial_layers;
 435   enc_cfg->ts_number_layers = svc_ctx->temporal_layers;
 436
 437   if (enc_cfg->g_error_resilient == 0 && si->use_multiple_frame_contexts == 0)
 438     enc_cfg->g_error_resilient = 1;
 439
 440   // Initialize codec
 441   res = vpx_codec_enc_init(codec_ctx, iface, enc_cfg, VPX_CODEC_USE_PSNR);
 442   if (res != VPX_CODEC_OK) {
 443     svc_log(svc_ctx, SVC_LOG_ERROR, "svc_enc_init error\n");
 444     return res;
 445   }
 446
 447   vpx_codec_control(codec_ctx, VP9E_SET_SVC, 1);
 448   vpx_codec_control(codec_ctx, VP9E_SET_SVC_PARAMETERS, &si->svc_params);
 449
 450   return VPX_CODEC_OK;
 451 }
 452
 453 /**
 454  * Encode a frame into multiple layers
 455  * Create a superframe containing the individual layers
 456  */
 457 vpx_codec_err_t vpx_svc_encode(SvcContext *svc_ctx, vpx_codec_ctx_t *codec_ctx,
 458                                struct vpx_image *rawimg, vpx_codec_pts_t pts,
 459                                int64_t duration, int deadline) {
 460   vpx_codec_err_t res;
 461   vpx_codec_iter_t iter;
 462   const vpx_codec_cx_pkt_t *cx_pkt;
 463   SvcInternal *const si = get_svc_internal(svc_ctx);
 464   if (svc_ctx == NULL || codec_ctx == NULL || si == NULL) {
 465     return VPX_CODEC_INVALID_PARAM;
 466   }
 467
 468   svc_log_reset(svc_ctx);
 469
 470   res = vpx_codec_encode(codec_ctx, rawimg, pts, (uint32_t)duration, 0,
 471                          deadline);
 472   if (res != VPX_CODEC_OK) {
 473     return res;
 474   }
 475   // save compressed data
 476   iter = NULL;
 477   while ((cx_pkt = vpx_codec_get_cx_data(codec_ctx, &iter))) {
 478     switch (cx_pkt->kind) {
 479 #if CONFIG_SPATIAL_SVC
 480       case VPX_CODEC_SPATIAL_SVC_LAYER_PSNR: {
 481         int i;
 482         for (i = 0; i < svc_ctx->spatial_layers; ++i) {
 483           int j;
 484           svc_log(svc_ctx, SVC_LOG_DEBUG,
 485                   "SVC frame: %d, layer: %d, PSNR(Total/Y/U/V): "
 486                   "%2.3f  %2.3f  %2.3f  %2.3f \n",
 487                   si->psnr_pkt_received, i,
 488                   cx_pkt->data.layer_psnr[i].psnr[0],
 489                   cx_pkt->data.layer_psnr[i].psnr[1],
 490                   cx_pkt->data.layer_psnr[i].psnr[2],
 491                   cx_pkt->data.layer_psnr[i].psnr[3]);
 492           svc_log(svc_ctx, SVC_LOG_DEBUG,
 493                   "SVC frame: %d, layer: %d, SSE(Total/Y/U/V): "
 494                   "%2.3f  %2.3f  %2.3f  %2.3f \n",
 495                   si->psnr_pkt_received, i,
 496                   cx_pkt->data.layer_psnr[i].sse[0],
 497                   cx_pkt->data.layer_psnr[i].sse[1],
 498                   cx_pkt->data.layer_psnr[i].sse[2],
 499                   cx_pkt->data.layer_psnr[i].sse[3]);
 500
 501           for (j = 0; j < COMPONENTS; ++j) {
 502             si->psnr_sum[i][j] +=
 503                 cx_pkt->data.layer_psnr[i].psnr[j];
 504             si->sse_sum[i][j] += cx_pkt->data.layer_psnr[i].sse[j];
 505           }
 506         }
 507         ++si->psnr_pkt_received;
 508         break;
 509       }
 510       case VPX_CODEC_SPATIAL_SVC_LAYER_SIZES: {
 511         int i;
 512         for (i = 0; i < svc_ctx->spatial_layers; ++i)
 513           si->bytes_sum[i] += cx_pkt->data.layer_sizes[i];
 514         break;
 515       }
 516 #endif
 517       default: {
 518         break;
 519       }
 520     }
 521   }
 522
 523   return VPX_CODEC_OK;
 524 }
 525
 526 const char *vpx_svc_get_message(const SvcContext *svc_ctx) {
 527   const SvcInternal *const si = get_const_svc_internal(svc_ctx);
 528   if (svc_ctx == NULL || si == NULL) return NULL;
 529   return si->message_buffer;
 530 }
 531
 532 static double calc_psnr(double d) {
 533   if (d == 0) return 100;
 534   return -10.0 * log(d) / log(10.0);
 535 }
 536
 537 // dump accumulated statistics and reset accumulated values
 538 const char *vpx_svc_dump_statistics(SvcContext *svc_ctx) {
 539   int number_of_frames;
 540   int i, j;
 541   uint32_t bytes_total = 0;
 542   double scale[COMPONENTS];
 543   double psnr[COMPONENTS];
 544   double mse[COMPONENTS];
 545   double y_scale;
 546
 547   SvcInternal *const si = get_svc_internal(svc_ctx);
 548   if (svc_ctx == NULL || si == NULL) return NULL;
 549
 550   svc_log_reset(svc_ctx);
 551
 552   number_of_frames = si->psnr_pkt_received;
 553   if (number_of_frames <= 0) return vpx_svc_get_message(svc_ctx);
 554
 555   svc_log(svc_ctx, SVC_LOG_INFO, "\n");
 556   for (i = 0; i < svc_ctx->spatial_layers; ++i) {
 557
 558     svc_log(svc_ctx, SVC_LOG_INFO,
 559             "Layer %d Average PSNR=[%2.3f, %2.3f, %2.3f, %2.3f], Bytes=[%u]\n",
 560             i, (double)si->psnr_sum[i][0] / number_of_frames,
 561             (double)si->psnr_sum[i][1] / number_of_frames,
 562             (double)si->psnr_sum[i][2] / number_of_frames,
 563             (double)si->psnr_sum[i][3] / number_of_frames, si->bytes_sum[i]);
 564     // the following psnr calculation is deduced from ffmpeg.c#print_report
 565     y_scale = si->width * si->height * 255.0 * 255.0 * number_of_frames;
 566     scale[1] = y_scale;
 567     scale[2] = scale[3] = y_scale / 4;  // U or V
 568     scale[0] = y_scale * 1.5;           // total
 569
 570     for (j = 0; j < COMPONENTS; j++) {
 571       psnr[j] = calc_psnr(si->sse_sum[i][j] / scale[j]);
 572       mse[j] = si->sse_sum[i][j] * 255.0 * 255.0 / scale[j];
 573     }
 574     svc_log(svc_ctx, SVC_LOG_INFO,
 575             "Layer %d Overall PSNR=[%2.3f, %2.3f, %2.3f, %2.3f]\n", i, psnr[0],
 576             psnr[1], psnr[2], psnr[3]);
 577     svc_log(svc_ctx, SVC_LOG_INFO,
 578             "Layer %d Overall MSE=[%2.3f, %2.3f, %2.3f, %2.3f]\n", i, mse[0],
 579             mse[1], mse[2], mse[3]);
 580
 581     bytes_total += si->bytes_sum[i];
 582     // clear sums for next time
 583     si->bytes_sum[i] = 0;
 584     for (j = 0; j < COMPONENTS; ++j) {
 585       si->psnr_sum[i][j] = 0;
 586       si->sse_sum[i][j] = 0;
 587     }
 588   }
 589
 590   // only display statistics once
 591   si->psnr_pkt_received = 0;
 592
 593   svc_log(svc_ctx, SVC_LOG_INFO, "Total Bytes=[%u]\n", bytes_total);
 594   return vpx_svc_get_message(svc_ctx);
 595 }
 596
 597 void vpx_svc_release(SvcContext *svc_ctx) {
 598   SvcInternal *si;
 599   if (svc_ctx == NULL) return;
 600   // do not use get_svc_internal as it will unnecessarily allocate an
 601   // SvcInternal if it was not already allocated
 602   si = (SvcInternal *)svc_ctx->internal;
 603   if (si != NULL) {
 604     free(si);
 605     svc_ctx->internal = NULL;
 606   }
 607 }
 608