| /* |
| * Copyright (c) 2019, 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 <stdint.h> |
| #include <stdio.h> |
| |
| #include "av1/common/cnn.h" |
| #include "av1/common/cnn_wrapper.h" |
| #include "av1/common/onyxc_int.h" |
| |
| #include "av1/models/intra_frame_model/qp22.h" |
| #include "av1/models/intra_frame_model/qp32.h" |
| #include "av1/models/intra_frame_model/qp43.h" |
| #include "av1/models/intra_frame_model/qp53.h" |
| #include "av1/models/intra_frame_model/qp63.h" |
| |
| static void av1_restore_cnn_plane_Y_wrapper(AV1_COMMON *cm, int plane) { |
| // TODO(logangw): Add infrastructure to choose models. |
| int qindex = cm->base_qindex; |
| if (qindex <= MIN_CNN_Q_INDEX) { |
| return; |
| } else if (qindex < 108) { |
| av1_restore_cnn_plane(cm, &model22, plane); |
| } else if (qindex < 148) { |
| av1_restore_cnn_plane(cm, &model32, plane); |
| } else if (qindex < 192) { |
| av1_restore_cnn_plane(cm, &model43, plane); |
| } else if (qindex < 232) { |
| av1_restore_cnn_plane(cm, &model53, plane); |
| } else { |
| av1_restore_cnn_plane(cm, &model63, plane); |
| } |
| } |
| |
| void av1_encode_restore_cnn(AV1_COMMON *cm) { |
| // TODO(logangw): Add mechanism to restore AOM_PLANE_U and AOM_PLANE_V. |
| av1_restore_cnn_plane_Y_wrapper(cm, AOM_PLANE_Y); |
| } |
| |
| void av1_decode_restore_cnn(AV1_COMMON *cm) { |
| av1_restore_cnn_plane_Y_wrapper(cm, AOM_PLANE_Y); |
| } |