blob: d72eab0883df22246351ba84ff416d3752ff3006 [file] [log] [blame]
/*
* Copyright (c) 2021, 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 "av1/common/common.h"
#include "av1/encoder/external_partition.h"
aom_codec_err_t av1_ext_part_create(aom_ext_part_funcs_t funcs,
aom_ext_part_config_t config,
ExtPartController *ext_part_controller) {
if (ext_part_controller == NULL) {
return AOM_CODEC_INVALID_PARAM;
}
ext_part_controller->funcs = funcs;
ext_part_controller->config = config;
const aom_ext_part_status_t status = ext_part_controller->funcs.create_model(
ext_part_controller->funcs.priv, &ext_part_controller->config,
&ext_part_controller->model);
if (status == AOM_EXT_PART_ERROR) {
return AOM_CODEC_ERROR;
} else if (status == AOM_EXT_PART_TEST) {
ext_part_controller->test_mode = 1;
ext_part_controller->ready = 0;
return AOM_CODEC_OK;
}
assert(status == AOM_EXT_PART_OK);
ext_part_controller->ready = 1;
return AOM_CODEC_OK;
}
aom_codec_err_t av1_ext_part_init(ExtPartController *ext_part_controller) {
if (ext_part_controller == NULL) {
return AOM_CODEC_INVALID_PARAM;
}
av1_zero(ext_part_controller);
return AOM_CODEC_OK;
}
aom_codec_err_t av1_ext_part_delete(ExtPartController *ext_part_controller) {
if (ext_part_controller == NULL) {
return AOM_CODEC_INVALID_PARAM;
}
if (ext_part_controller->ready) {
const aom_ext_part_status_t status =
ext_part_controller->funcs.delete_model(ext_part_controller->model);
if (status != AOM_EXT_PART_OK) {
return AOM_CODEC_ERROR;
}
}
return av1_ext_part_init(ext_part_controller);
}
bool av1_ext_part_get_partition_decision(ExtPartController *ext_part_controller,
aom_partition_decision_t *decision) {
assert(ext_part_controller != NULL);
assert(ext_part_controller->ready);
assert(decision != NULL);
const aom_ext_part_status_t status =
ext_part_controller->funcs.get_partition_decision(
ext_part_controller->model, decision);
if (status != AOM_EXT_PART_OK) return false;
return true;
}
bool av1_ext_part_send_partition_stats(ExtPartController *ext_part_controller,
const aom_partition_stats_t *stats) {
assert(ext_part_controller != NULL);
assert(ext_part_controller->ready);
assert(stats != NULL);
const aom_ext_part_status_t status =
ext_part_controller->funcs.send_partition_stats(
ext_part_controller->model, stats);
if (status != AOM_EXT_PART_OK) return false;
return true;
}
bool av1_ext_part_send_features(ExtPartController *ext_part_controller,
const aom_partition_features_t *features) {
assert(ext_part_controller != NULL);
assert(ext_part_controller->ready);
assert(features != NULL);
const aom_ext_part_status_t status = ext_part_controller->funcs.send_features(
ext_part_controller->model, features);
if (status != AOM_EXT_PART_OK) return false;
return true;
}
aom_ext_part_decision_mode_t av1_get_ext_part_decision_mode(
const ExtPartController *ext_part_controller) {
return ext_part_controller->funcs.decision_mode;
}