| Index: third_party/libvpx/source/libvpx/test/set_roi.cc
|
| diff --git a/third_party/libvpx/source/libvpx/test/set_roi.cc b/third_party/libvpx/source/libvpx/test/set_roi.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..fea8cca7a15e4e9de0dd3f9da277bf93d6677d50
|
| --- /dev/null
|
| +++ b/third_party/libvpx/source/libvpx/test/set_roi.cc
|
| @@ -0,0 +1,184 @@
|
| +/*
|
| + * Copyright (c) 2012 The WebM project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +
|
| +#include <math.h>
|
| +#include <stddef.h>
|
| +#include <stdio.h>
|
| +#include <stdlib.h>
|
| +#include <string.h>
|
| +#include <sys/types.h>
|
| +
|
| +#include "third_party/googletest/src/include/gtest/gtest.h"
|
| +#include "test/acm_random.h"
|
| +#include "vp8/encoder/onyx_int.h"
|
| +#include "vpx/vpx_integer.h"
|
| +#include "vpx_mem/vpx_mem.h"
|
| +
|
| +using libvpx_test::ACMRandom;
|
| +
|
| +namespace {
|
| +
|
| +TEST(VP8RoiMapTest, ParameterCheck) {
|
| + ACMRandom rnd(ACMRandom::DeterministicSeed());
|
| + int delta_q[MAX_MB_SEGMENTS] = { -2, -25, 0, 31 };
|
| + int delta_lf[MAX_MB_SEGMENTS] = { -2, -25, 0, 31 };
|
| + unsigned int threshold[MAX_MB_SEGMENTS] = { 0, 100, 200, 300 };
|
| +
|
| + const int internalq_trans[] = {
|
| + 0, 1, 2, 3, 4, 5, 7, 8,
|
| + 9, 10, 12, 13, 15, 17, 18, 19,
|
| + 20, 21, 23, 24, 25, 26, 27, 28,
|
| + 29, 30, 31, 33, 35, 37, 39, 41,
|
| + 43, 45, 47, 49, 51, 53, 55, 57,
|
| + 59, 61, 64, 67, 70, 73, 76, 79,
|
| + 82, 85, 88, 91, 94, 97, 100, 103,
|
| + 106, 109, 112, 115, 118, 121, 124, 127,
|
| + };
|
| +
|
| + // Initialize elements of cpi with valid defaults.
|
| + VP8_COMP cpi;
|
| + cpi.mb.e_mbd.mb_segement_abs_delta = SEGMENT_DELTADATA;
|
| + cpi.cyclic_refresh_mode_enabled = 0;
|
| + cpi.mb.e_mbd.segmentation_enabled = 0;
|
| + cpi.mb.e_mbd.update_mb_segmentation_map = 0;
|
| + cpi.mb.e_mbd.update_mb_segmentation_data = 0;
|
| + cpi.common.mb_rows = 240 >> 4;
|
| + cpi.common.mb_cols = 320 >> 4;
|
| + const int mbs = (cpi.common.mb_rows * cpi.common.mb_cols);
|
| + memset(cpi.segment_feature_data, 0, sizeof(cpi.segment_feature_data));
|
| +
|
| + // Segment map
|
| + cpi.segmentation_map = reinterpret_cast<unsigned char *>(vpx_calloc(mbs, 1));
|
| +
|
| + // Allocate memory for the source memory map.
|
| + unsigned char *roi_map =
|
| + reinterpret_cast<unsigned char *>(vpx_calloc(mbs, 1));
|
| + memset(&roi_map[mbs >> 2], 1, (mbs >> 2));
|
| + memset(&roi_map[mbs >> 1], 2, (mbs >> 2));
|
| + memset(&roi_map[mbs -(mbs >> 2)], 3, (mbs >> 2));
|
| +
|
| + // Do a test call with valid parameters.
|
| + int roi_retval = vp8_set_roimap(&cpi, roi_map, cpi.common.mb_rows,
|
| + cpi.common.mb_cols, delta_q, delta_lf,
|
| + threshold);
|
| + EXPECT_EQ(0, roi_retval)
|
| + << "vp8_set_roimap roi failed with default test parameters";
|
| +
|
| + // Check that the values in the cpi structure get set as expected.
|
| + if (roi_retval == 0) {
|
| + // Check that the segment map got set.
|
| + const int mapcompare = memcmp(roi_map, cpi.segmentation_map, mbs);
|
| + EXPECT_EQ(0, mapcompare) << "segment map error";
|
| +
|
| + // Check the q deltas (note the need to translate into
|
| + // the interanl range of 0-127.
|
| + for (int i = 0; i < MAX_MB_SEGMENTS; ++i) {
|
| + const int transq = internalq_trans[abs(delta_q[i])];
|
| + if (abs(cpi.segment_feature_data[MB_LVL_ALT_Q][i]) != transq) {
|
| + EXPECT_EQ(transq, cpi.segment_feature_data[MB_LVL_ALT_Q][i])
|
| + << "segment delta_q error";
|
| + break;
|
| + }
|
| + }
|
| +
|
| + // Check the loop filter deltas
|
| + for (int i = 0; i < MAX_MB_SEGMENTS; ++i) {
|
| + if (cpi.segment_feature_data[MB_LVL_ALT_LF][i] != delta_lf[i]) {
|
| + EXPECT_EQ(delta_lf[i], cpi.segment_feature_data[MB_LVL_ALT_LF][i])
|
| + << "segment delta_lf error";
|
| + break;
|
| + }
|
| + }
|
| +
|
| + // Check the breakout thresholds
|
| + for (int i = 0; i < MAX_MB_SEGMENTS; ++i) {
|
| + unsigned int breakout =
|
| + static_cast<unsigned int>(cpi.segment_encode_breakout[i]);
|
| +
|
| + if (threshold[i] != breakout) {
|
| + EXPECT_EQ(threshold[i], breakout)
|
| + << "breakout threshold error";
|
| + break;
|
| + }
|
| + }
|
| +
|
| + // Segmentation, and segmentation update flages should be set.
|
| + EXPECT_EQ(1, cpi.mb.e_mbd.segmentation_enabled)
|
| + << "segmentation_enabled error";
|
| + EXPECT_EQ(1, cpi.mb.e_mbd.update_mb_segmentation_map)
|
| + << "update_mb_segmentation_map error";
|
| + EXPECT_EQ(1, cpi.mb.e_mbd.update_mb_segmentation_data)
|
| + << "update_mb_segmentation_data error";
|
| +
|
| +
|
| + // Try a range of delta q and lf parameters (some legal, some not)
|
| + for (int i = 0; i < 1000; ++i) {
|
| + int rand_deltas[4];
|
| + int deltas_valid;
|
| + rand_deltas[0] = rnd(160) - 80;
|
| + rand_deltas[1] = rnd(160) - 80;
|
| + rand_deltas[2] = rnd(160) - 80;
|
| + rand_deltas[3] = rnd(160) - 80;
|
| +
|
| + deltas_valid = ((abs(rand_deltas[0]) <= 63) &&
|
| + (abs(rand_deltas[1]) <= 63) &&
|
| + (abs(rand_deltas[2]) <= 63) &&
|
| + (abs(rand_deltas[3]) <= 63)) ? 0 : -1;
|
| +
|
| + // Test with random delta q values.
|
| + roi_retval = vp8_set_roimap(&cpi, roi_map, cpi.common.mb_rows,
|
| + cpi.common.mb_cols, rand_deltas,
|
| + delta_lf, threshold);
|
| + EXPECT_EQ(deltas_valid, roi_retval) << "dq range check error";
|
| +
|
| + // One delta_q error shown at a time
|
| + if (deltas_valid != roi_retval)
|
| + break;
|
| +
|
| + // Test with random loop filter values.
|
| + roi_retval = vp8_set_roimap(&cpi, roi_map, cpi.common.mb_rows,
|
| + cpi.common.mb_cols, delta_q,
|
| + rand_deltas, threshold);
|
| + EXPECT_EQ(deltas_valid, roi_retval) << "dlf range check error";
|
| +
|
| + // One delta loop filter error shown at a time
|
| + if (deltas_valid != roi_retval)
|
| + break;
|
| + }
|
| +
|
| + // Test that we report and error if cyclic refresh is enabled.
|
| + cpi.cyclic_refresh_mode_enabled = 1;
|
| + roi_retval = vp8_set_roimap(&cpi, roi_map, cpi.common.mb_rows,
|
| + cpi.common.mb_cols, delta_q,
|
| + delta_lf, threshold);
|
| + EXPECT_EQ(-1, roi_retval) << "cyclic refresh check error";
|
| + cpi.cyclic_refresh_mode_enabled = 0;
|
| +
|
| + // Test invalid number of rows or colums.
|
| + roi_retval = vp8_set_roimap(&cpi, roi_map, cpi.common.mb_rows + 1,
|
| + cpi.common.mb_cols, delta_q,
|
| + delta_lf, threshold);
|
| + EXPECT_EQ(-1, roi_retval) << "MB rows bounds check error";
|
| +
|
| + roi_retval = vp8_set_roimap(&cpi, roi_map, cpi.common.mb_rows,
|
| + cpi.common.mb_cols - 1, delta_q,
|
| + delta_lf, threshold);
|
| + EXPECT_EQ(-1, roi_retval) << "MB cols bounds check error";
|
| + }
|
| +
|
| + // Free allocated memory
|
| + if (cpi.segmentation_map)
|
| + vpx_free(cpi.segmentation_map);
|
| + if (roi_map)
|
| + vpx_free(roi_map);
|
| +};
|
| +
|
| +} // namespace
|
|
|