/*    This file is a part of plaYUVer project
 *    Copyright (C) 2014-2021  by Joao Carreira   (jfmcarreira@gmail.com)
 *                                Luis Lucas      (luisfrlucas@gmail.com)
 *
 *    This program is free software; you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation; either version 2 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License along
 *   with this program; if not, write to the Free Software Foundation, Inc.,
 *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

/**
 * \file     CalypStreamTests.cpp
 * \brief    CalypStream general tests
 */

#include <catch2/catch.hpp>
#include <filesystem>
#include <variant>

#include "CalypFrame.h"
#include "CalypStream.h"

constexpr int kFrameRate{ 30 };
constexpr bool kIsInput{ true };

auto getFileTestFilename( const char* test_sequence ) -> std::string
{
  return std::string( CALYP_TEST_DATA_DIR ) + std::string( "/" ) + std::string( test_sequence );
}

TEST_CASE( "Can open a YUV 420p file", "CalypStream" )
{
  constexpr int kWidth{ 352 };
  constexpr int kHeight{ 288 };
  constexpr auto kInputFormat{ CalypPixelFormats::CLP_YUV420P };
  constexpr int kBitsPel{ 8 };
  constexpr auto KEndianness{ CLP_INVALID_ENDIANESS };

  const auto kFilename = getFileTestFilename( "Foreman.yuv" );

  REQUIRE( std::filesystem::exists( std::filesystem::path( kFilename ) ) );

  CalypStream test_stream;
  auto result = test_stream.open( kFilename, kWidth, kHeight, kInputFormat, kBitsPel, KEndianness, kFrameRate, true );

  REQUIRE( result );

  CHECK( test_stream.getWidth() == kWidth );
  CHECK( test_stream.getHeight() == kHeight );
  CHECK( test_stream.getBitsPerPixel() == kBitsPel );
  CHECK( test_stream.getFormatName() == "YUV" );
  CHECK( test_stream.getCodecName() == "Raw Video" );
  CHECK( test_stream.getEndianess() == CLP_BIG_ENDIAN );

  const auto* frame = test_stream.getCurrFrame();

  CHECK( frame->getPixel( 2, 0 ) == CalypPixel{ CLP_COLOR_YUV, 201, 129, 125 } );
  CHECK( frame->getPixel( 336, 278 ) == CalypPixel{ CLP_COLOR_YUV, 99, 111, 142 } );
}

TEST_CASE( "Can open .batatas file as RAW video", "CalypStream" )
{
  constexpr int kWidth{ 352 };
  constexpr int kHeight{ 288 };
  constexpr auto kInputFormat{ CalypPixelFormats::CLP_YUV420P };
  constexpr int kBitsPel{ 8 };
  constexpr auto KEndianness{ CLP_INVALID_ENDIANESS };

  const auto kFilename = getFileTestFilename( "Foreman.yuv" );

  REQUIRE( std::filesystem::exists( std::filesystem::path( kFilename ) ) );

  bool force_raw{ false };

  SECTION( "Without forcing raw video" )
  {
    CalypStream test_stream;
    bool result{ false };
    try
    {
      result = test_stream.open( kFilename, kWidth, kHeight, kInputFormat, kBitsPel, KEndianness, kFrameRate, kIsInput, force_raw );
    }
    catch( ... )
    {
      // NOLINT
    }

    CHECK( !result );
  }

  SECTION( "Forcing raw video" )
  {
    force_raw = true;

    CalypStream test_stream;
    auto result = test_stream.open( kFilename, kWidth, kHeight, kInputFormat, kBitsPel, KEndianness, kFrameRate, kIsInput, force_raw );

    REQUIRE( result );

    CHECK( test_stream.getWidth() == kWidth );
    CHECK( test_stream.getHeight() == kHeight );
    CHECK( test_stream.getBitsPerPixel() == kBitsPel );
    CHECK( test_stream.getFormatName() == "YUV" );
    CHECK( test_stream.getCodecName() == "Raw Video" );
    CHECK( test_stream.getEndianess() == CLP_BIG_ENDIAN );

    const auto* frame = test_stream.getCurrFrame();

    CHECK( frame->getPixel( 2, 0 ) == CalypPixel{ CLP_COLOR_YUV, 201, 129, 125 } );
    CHECK( frame->getPixel( 336, 278 ) == CalypPixel{ CLP_COLOR_YUV, 99, 111, 142 } );
  }
}