#!/bin/bash
# Copyright 2023 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ------------------------------------------------------------------------------

source $(dirname "$0")/cmd_test_common.sh || exit

# Basic calls.
"${AVIFENC}" --version
"${AVIFDEC}" --version

# Input file paths.
INPUT_Y4M="${TESTDATA_DIR}/kodim03_yuv420_8bpc.y4m"
INPUT_SMALL_Y4M="${TESTDATA_DIR}/kodim03_yuv420_8bpc_384x256.y4m"
# Output file names.
ENCODED_FILE="avif_test_cmd_encoded.avif"
DECODED_FILE="avif_test_cmd_decoded.png"
INFO_FILE="avif_test_cmd_info.txt"

# Cleanup
cleanup() {
  pushd ${TMP_DIR}
    rm -- "${ENCODED_FILE}" "${DECODED_FILE}" "${INFO_FILE}"
  popd
}
trap cleanup EXIT

pushd ${TMP_DIR}
  echo "Testing automatic layered encoding"
  "${AVIFENC}" --progressive -s 8 "${INPUT_Y4M}" -o "${ENCODED_FILE}"
  "${AVIFDEC}" "${ENCODED_FILE}" "${DECODED_FILE}"
  "${AVIFDEC}" --progressive "${ENCODED_FILE}" "${DECODED_FILE}"

  echo "Testing manual layered encoding"
  "${AVIFENC}" -s 8 --layered -q:u 2 "${INPUT_Y4M}" -q:u 60 "${INPUT_Y4M}" -o "${ENCODED_FILE}"
  "${AVIFDEC}" "${ENCODED_FILE}" "${DECODED_FILE}"
  "${AVIFDEC}" --progressive "${ENCODED_FILE}" "${DECODED_FILE}"

  # libavif relies on libyuv to do scaling
  echo "Testing layered encoding with frame scaling"
  if avifenc -V | grep -o "libyuv : available" --quiet; then
    for SCALE in 1/4 1/2 1; do
      "${AVIFENC}" -s 8 --layered --scaling-mode:u ${SCALE} "${INPUT_Y4M}" --scaling-mode:u 1 "${INPUT_Y4M}" -o "${ENCODED_FILE}"
      "${AVIFDEC}" "${ENCODED_FILE}" "${DECODED_FILE}"
      "${AVIFDEC}" --progressive "${ENCODED_FILE}" "${DECODED_FILE}"
    done
  fi

  echo "Testing manual layered encoding with pre-scaled input"
  "${AVIFENC}" -s 8 --layered \
    "${INPUT_SMALL_Y4M}" "${INPUT_Y4M}" -o "${ENCODED_FILE}"
  "${AVIFDEC}" --info "${ENCODED_FILE}" > "${INFO_FILE}"
  grep -F "[768x512]" "${INFO_FILE}"
  "${AVIFDEC}" "${ENCODED_FILE}" "${DECODED_FILE}"
  "${AVIFDEC}" --progressive "${ENCODED_FILE}" "${DECODED_FILE}"

  echo "Testing an earlier layer larger than the last layer"
  "${AVIFENC}" -s 8 --layered \
    "${INPUT_Y4M}" "${INPUT_SMALL_Y4M}" -o "${ENCODED_FILE}" && exit 1

  echo "Testing layered encoding with pre-scaled input and frame scaling (rejected)"
  "${AVIFENC}" -s 8 --layered --scaling-mode:u 1/2 \
    "${INPUT_SMALL_Y4M}" "${INPUT_Y4M}" -o "${ENCODED_FILE}" && exit 1

  echo "Testing too few layers"
  "${AVIFENC}" -s 8 --layered -q:u 60 "${INPUT_Y4M}" -o "${ENCODED_FILE}" && exit 1

  echo "Testing too many layers"
  "${AVIFENC}" -s 8 --layered \
    -q:u 10 "${INPUT_Y4M}" \
    -q:u 20 "${INPUT_Y4M}" \
    -q:u 30 "${INPUT_Y4M}" \
    -q:u 40 "${INPUT_Y4M}" \
    -q:u 50 "${INPUT_Y4M}" \
    -o "${ENCODED_FILE}" && exit 1
popd

exit 0