Test upscaling as well as downscaling
Fixes a bug in vp9_set_internal_size() that prevented returning to the unscaled state. Updated the ResizeInternalTest to scale both down and up. Added a check that all frames are within 2.5% of the quality of the initial keyframe. Change-Id: I3b7ef17cdac144ed05b9148dce6badfa75cff5c8
This commit is contained in:
@@ -101,7 +101,7 @@ TEST_P(ResizeTest, TestExternalResizeWorks) {
|
||||
|
||||
class ResizeInternalTest : public ResizeTest {
|
||||
protected:
|
||||
ResizeInternalTest() : ResizeTest() {}
|
||||
ResizeInternalTest() : ResizeTest(), frame0_psnr_(0.0) {}
|
||||
|
||||
virtual void PreEncodeFrameHook(libvpx_test::VideoSource *video,
|
||||
libvpx_test::Encoder *encoder) {
|
||||
@@ -109,19 +109,33 @@ class ResizeInternalTest : public ResizeTest {
|
||||
struct vpx_scaling_mode mode = {VP8E_FOURFIVE, VP8E_THREEFIVE};
|
||||
encoder->Control(VP8E_SET_SCALEMODE, &mode);
|
||||
}
|
||||
if (video->frame() == 6) {
|
||||
struct vpx_scaling_mode mode = {VP8E_NORMAL, VP8E_NORMAL};
|
||||
encoder->Control(VP8E_SET_SCALEMODE, &mode);
|
||||
}
|
||||
}
|
||||
|
||||
virtual void PSNRPktHook(const vpx_codec_cx_pkt_t *pkt) {
|
||||
if (!frame0_psnr_)
|
||||
frame0_psnr_ = pkt->data.psnr.psnr[0];
|
||||
ASSERT_NEAR(pkt->data.psnr.psnr[0], frame0_psnr_, 0.025*frame0_psnr_);
|
||||
}
|
||||
|
||||
double frame0_psnr_;
|
||||
};
|
||||
|
||||
TEST_P(ResizeInternalTest, TestInternalResizeWorks) {
|
||||
::libvpx_test::I420VideoSource video("hantro_collage_w352h288.yuv", 352, 288,
|
||||
30, 1, 0, 6);
|
||||
cfg_.rc_target_bitrate = 5000;
|
||||
30, 1, 0, 10);
|
||||
init_flags_ = VPX_CODEC_USE_PSNR;
|
||||
// q picked such that initial keyframe on this clip is ~30dB PSNR
|
||||
cfg_.rc_min_quantizer = cfg_.rc_max_quantizer = 48;
|
||||
ASSERT_NO_FATAL_FAILURE(RunLoop(&video));
|
||||
|
||||
for (std::vector<FrameInfo>::iterator info = frame_info_list_.begin();
|
||||
info != frame_info_list_.end(); ++info) {
|
||||
const vpx_codec_pts_t pts = info->pts;
|
||||
if (pts >= 3) {
|
||||
if (pts >= 3 && pts < 6) {
|
||||
ASSERT_EQ(282U, info->w) << "Frame " << pts << " had unexpected width";
|
||||
ASSERT_EQ(173U, info->h) << "Frame " << pts << " had unexpected height";
|
||||
} else {
|
||||
|
||||
@@ -4123,20 +4123,19 @@ int vp9_set_internal_size(VP9_PTR comp,
|
||||
VP9_COMP *cpi = (VP9_COMP *) comp;
|
||||
VP9_COMMON *cm = &cpi->common;
|
||||
|
||||
if (horiz_mode <= ONETWO)
|
||||
cm->horiz_scale = horiz_mode;
|
||||
else
|
||||
if (horiz_mode > ONETWO)
|
||||
return -1;
|
||||
|
||||
if (vert_mode <= ONETWO)
|
||||
cm->vert_scale = vert_mode;
|
||||
else
|
||||
if (vert_mode > ONETWO)
|
||||
return -1;
|
||||
|
||||
if (cm->horiz_scale != NORMAL || cm->vert_scale != NORMAL) {
|
||||
if (cm->horiz_scale != horiz_mode || cm->vert_scale != vert_mode) {
|
||||
int UNINITIALIZED_IS_SAFE(hr), UNINITIALIZED_IS_SAFE(hs);
|
||||
int UNINITIALIZED_IS_SAFE(vr), UNINITIALIZED_IS_SAFE(vs);
|
||||
|
||||
cm->horiz_scale = horiz_mode;
|
||||
cm->vert_scale = vert_mode;
|
||||
|
||||
Scale2Ratio(cm->horiz_scale, &hr, &hs);
|
||||
Scale2Ratio(cm->vert_scale, &vr, &vs);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user