video_geometry_aligner.h

Enable keyboard shortcuts

/*

 *  Copyright (c) 2018 The WebRTC 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.

*/

#ifndef RTC_TOOLS_FRAME_ANALYZER_VIDEO_GEOMETRY_ALIGNER_H_

#define RTC_TOOLS_FRAME_ANALYZER_VIDEO_GEOMETRY_ALIGNER_H_

#include "api/video/video_frame_buffer.h"

#include "rtc_tools/video_file_reader.h"

namespace webrtc {

namespace test {

struct CropRegion {

  // Each value represents how much to crop from each side. Left is where x=0,

  // and top is where y=0. All values equal to zero represents no cropping.

  int left = 0;

  int right = 0;

  int top = 0;

  int bottom = 0;

};

// Crops and zooms in on the cropped region so that the returned frame has the

// same resolution as the input frame.

rtc::scoped_refptr<I420BufferInterface> CropAndZoom(

    const CropRegion& crop_region,

    const rtc::scoped_refptr<I420BufferInterface>& frame);

// Calculate the optimal cropping region on the reference frame to maximize SSIM

// to the test frame.

CropRegion CalculateCropRegion(

    const rtc::scoped_refptr<I420BufferInterface>& reference_frame,

    const rtc::scoped_refptr<I420BufferInterface>& test_frame);

// Returns a cropped and zoomed version of the reference frame that matches up

// to the test frame. This is a simple helper function on top of

// CalculateCropRegion() and CropAndZoom().

rtc::scoped_refptr<I420BufferInterface> AdjustCropping(

    const rtc::scoped_refptr<I420BufferInterface>& reference_frame,

    const rtc::scoped_refptr<I420BufferInterface>& test_frame);

// Returns a cropped and zoomed version of the reference video that matches up

// to the test video. Frames are individually adjusted for cropping.

rtc::scoped_refptr<Video> AdjustCropping(

    const rtc::scoped_refptr<Video>& reference_video,

    const rtc::scoped_refptr<Video>& test_video);

}  // namespace test

}  // namespace webrtc

#endif  // RTC_TOOLS_FRAME_ANALYZER_VIDEO_GEOMETRY_ALIGNER_H_