60 lines
2.4 KiB
C
60 lines
2.4 KiB
C
/*
|
|
* Copyright (c) 2016, Alliance for Open Media. All rights reserved
|
|
*
|
|
* This source code is subject to the terms of the BSD 2 Clause License and
|
|
* the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
|
|
* was not distributed with this source code in the LICENSE file, you can
|
|
* obtain it at www.aomedia.org/license/software. If the Alliance for Open
|
|
* Media Patent License 1.0 was not distributed with this source code in the
|
|
* PATENTS file, you can obtain it at www.aomedia.org/license/patent.
|
|
*/
|
|
|
|
#include <assert.h>
|
|
|
|
#include "aom_dsp/flow_estimation/corner_match.h"
|
|
#include "aom_dsp/flow_estimation/disflow.h"
|
|
#include "aom_dsp/flow_estimation/flow_estimation.h"
|
|
#include "aom_ports/mem.h"
|
|
#include "aom_scale/yv12config.h"
|
|
|
|
int aom_compute_global_motion(TransformationType type,
|
|
unsigned char *src_buffer, int src_width,
|
|
int src_height, int src_stride, int *src_corners,
|
|
int num_src_corners, YV12_BUFFER_CONFIG *ref,
|
|
int bit_depth,
|
|
GlobalMotionEstimationType gm_estimation_type,
|
|
int *num_inliers_by_motion,
|
|
MotionModel *params_by_motion, int num_motions) {
|
|
switch (gm_estimation_type) {
|
|
case GLOBAL_MOTION_FEATURE_BASED:
|
|
return av1_compute_global_motion_feature_based(
|
|
type, src_buffer, src_width, src_height, src_stride, src_corners,
|
|
num_src_corners, ref, bit_depth, num_inliers_by_motion,
|
|
params_by_motion, num_motions);
|
|
case GLOBAL_MOTION_DISFLOW_BASED:
|
|
return av1_compute_global_motion_disflow_based(
|
|
type, src_buffer, src_width, src_height, src_stride, src_corners,
|
|
num_src_corners, ref, bit_depth, num_inliers_by_motion,
|
|
params_by_motion, num_motions);
|
|
default: assert(0 && "Unknown global motion estimation type");
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
unsigned char *av1_downconvert_frame(YV12_BUFFER_CONFIG *frm, int bit_depth) {
|
|
int i, j;
|
|
uint16_t *orig_buf = CONVERT_TO_SHORTPTR(frm->y_buffer);
|
|
uint8_t *buf_8bit = frm->y_buffer_8bit;
|
|
assert(buf_8bit);
|
|
if (!frm->buf_8bit_valid) {
|
|
for (i = 0; i < frm->y_height; ++i) {
|
|
for (j = 0; j < frm->y_width; ++j) {
|
|
buf_8bit[i * frm->y_stride + j] =
|
|
orig_buf[i * frm->y_stride + j] >> (bit_depth - 8);
|
|
}
|
|
}
|
|
frm->buf_8bit_valid = 1;
|
|
}
|
|
return buf_8bit;
|
|
}
|