Blame extras/quality_estimate.c

Packit 9c6abc
// Copyright 2016 Google Inc. All Rights Reserved.
Packit 9c6abc
//
Packit 9c6abc
// Use of this source code is governed by a BSD-style license
Packit 9c6abc
// that can be found in the COPYING file in the root of the source
Packit 9c6abc
// tree. An additional intellectual property rights grant can be found
Packit 9c6abc
// in the file PATENTS. All contributing project authors may
Packit 9c6abc
// be found in the AUTHORS file in the root of the source tree.
Packit 9c6abc
// -----------------------------------------------------------------------------
Packit 9c6abc
//
Packit 9c6abc
//  VP8EstimateQuality(): rough encoding quality estimate
Packit 9c6abc
//
Packit 9c6abc
// Author: Skal (pascal.massimino@gmail.com)
Packit 9c6abc
Packit 9c6abc
#include "extras/extras.h"
Packit 9c6abc
#include "webp/decode.h"
Packit 9c6abc
Packit 9c6abc
#include <math.h>
Packit 9c6abc
Packit 9c6abc
//------------------------------------------------------------------------------
Packit 9c6abc
Packit 9c6abc
#define INVALID_BIT_POS (1ull << 63)
Packit 9c6abc
Packit 9c6abc
// In most cases, we don't need to use a full arithmetic decoder, since
Packit 9c6abc
// all the header's bits are written using a uniform probability of 128.
Packit 9c6abc
// We can just parse the header as if it was bits (works in 99.999% cases).
Packit 9c6abc
static WEBP_INLINE uint32_t GetBit(const uint8_t* const data, size_t nb,
Packit 9c6abc
                                   uint64_t max_size, uint64_t* const bit_pos) {
Packit 9c6abc
  uint32_t val = 0;
Packit 9c6abc
  if (*bit_pos + nb <= 8 * max_size) {
Packit 9c6abc
    while (nb-- > 0) {
Packit 9c6abc
      const uint64_t p = (*bit_pos)++;
Packit 9c6abc
      const int bit = !!(data[p >> 3] & (128 >> ((p & 7))));
Packit 9c6abc
      val = (val << 1) | bit;
Packit 9c6abc
    }
Packit 9c6abc
  } else {
Packit 9c6abc
    *bit_pos = INVALID_BIT_POS;
Packit 9c6abc
  }
Packit 9c6abc
  return val;
Packit 9c6abc
}
Packit 9c6abc
Packit 9c6abc
#define GET_BIT(n) GetBit(data, (n), size, &bit_pos)
Packit 9c6abc
#define CONDITIONAL_SKIP(n) (GET_BIT(1) ? GET_BIT((n)) : 0)
Packit 9c6abc
Packit 9c6abc
int VP8EstimateQuality(const uint8_t* const data, size_t size) {
Packit 9c6abc
  size_t pos = 0;
Packit 9c6abc
  uint64_t bit_pos;
Packit 9c6abc
  uint64_t sig = 0x00;
Packit 9c6abc
  int ok = 0;
Packit 9c6abc
  int Q = -1;
Packit 9c6abc
  WebPBitstreamFeatures features;
Packit 9c6abc
Packit 9c6abc
  if (data == NULL) return -1;
Packit 9c6abc
Packit 9c6abc
  if (WebPGetFeatures(data, size, &features) != VP8_STATUS_OK) {
Packit 9c6abc
    return -1;   // invalid file
Packit 9c6abc
  }
Packit 9c6abc
  if (features.format == 2) return 101;  // lossless
Packit 9c6abc
  if (features.format == 0 || features.has_animation) return -1;   // mixed
Packit 9c6abc
Packit 9c6abc
  while (pos < size) {
Packit 9c6abc
    sig = (sig >> 8) | ((uint64_t)data[pos++] << 40);
Packit 9c6abc
    if ((sig >> 24) == 0x2a019dull) {
Packit 9c6abc
      ok = 1;
Packit 9c6abc
      break;
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  if (!ok) return -1;
Packit 9c6abc
  if (pos + 4 > size) return -1;
Packit 9c6abc
Packit 9c6abc
  // Skip main Header
Packit 9c6abc
  // width  = (data[pos + 0] | (data[pos + 1] << 8)) & 0x3fff;
Packit 9c6abc
  // height = (data[pos + 2] | (data[pos + 3] << 8)) & 0x3fff;
Packit 9c6abc
  pos += 4;
Packit 9c6abc
  bit_pos = pos * 8;
Packit 9c6abc
Packit 9c6abc
  GET_BIT(2);  // colorspace + clamp type
Packit 9c6abc
Packit 9c6abc
  // Segment header
Packit 9c6abc
  if (GET_BIT(1)) {       // use_segment_
Packit 9c6abc
    int s;
Packit 9c6abc
    const int update_map = GET_BIT(1);
Packit 9c6abc
    if (GET_BIT(1)) {     // update data
Packit 9c6abc
      const int absolute_delta = GET_BIT(1);
Packit 9c6abc
      int q[4]  = { 0, 0, 0, 0 };
Packit 9c6abc
      for (s = 0; s < 4; ++s) {
Packit 9c6abc
        if (GET_BIT(1)) {
Packit 9c6abc
          q[s] = GET_BIT(7);
Packit 9c6abc
          if (GET_BIT(1)) q[s] = -q[s];   // sign
Packit 9c6abc
        }
Packit 9c6abc
      }
Packit 9c6abc
      if (absolute_delta) Q = q[0];  // just use the first segment's quantizer
Packit 9c6abc
      for (s = 0; s < 4; ++s) CONDITIONAL_SKIP(7);   //  filter strength
Packit 9c6abc
    }
Packit 9c6abc
    if (update_map) {
Packit 9c6abc
      for (s = 0; s < 3; ++s) CONDITIONAL_SKIP(8);
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  // Filter header
Packit 9c6abc
  GET_BIT(1 + 6 + 3);     // simple + level + sharpness
Packit 9c6abc
  if (GET_BIT(1)) {       // use_lf_delta
Packit 9c6abc
    if (GET_BIT(1)) {     // update lf_delta?
Packit 9c6abc
      int n;
Packit 9c6abc
      for (n = 0; n < 4 + 4; ++n) CONDITIONAL_SKIP(6);
Packit 9c6abc
    }
Packit 9c6abc
  }
Packit 9c6abc
  // num partitions
Packit 9c6abc
  GET_BIT(2);
Packit 9c6abc
Packit 9c6abc
  // ParseQuant
Packit 9c6abc
  {
Packit 9c6abc
    const int base_q = GET_BIT(7);
Packit 9c6abc
    /* dqy1_dc = */ CONDITIONAL_SKIP(5);
Packit 9c6abc
    /* dqy2_dc = */ CONDITIONAL_SKIP(5);
Packit 9c6abc
    /* dqy2_ac = */ CONDITIONAL_SKIP(5);
Packit 9c6abc
    /* dquv_dc = */ CONDITIONAL_SKIP(5);
Packit 9c6abc
    /* dquv_ac = */ CONDITIONAL_SKIP(5);
Packit 9c6abc
Packit 9c6abc
    if (Q < 0) Q = base_q;
Packit 9c6abc
  }
Packit 9c6abc
  if (bit_pos == INVALID_BIT_POS) return -1;
Packit 9c6abc
Packit 9c6abc
  // base mapping
Packit 9c6abc
  Q = (127 - Q) * 100 / 127;
Packit 9c6abc
  // correction for power-law behavior in low range
Packit 9c6abc
  if (Q < 80) {
Packit 9c6abc
    Q = (int)(pow(Q / 80., 1. / 0.38) * 80);
Packit 9c6abc
  }
Packit 9c6abc
  return Q;
Packit 9c6abc
}