/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  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
 *
 *         https://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  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.
 *
 *=========================================================================*/
#include <iostream>
#include <set>

#include "itkVideoIOFactory.h"
#include "itkFileListVideoIOFactory.h"
#include "itkTestingMacros.h"

///////////////////////////////////////////////////////////////////////////////
// This tests all of the functionality of the FileListVideoIO
//
// Usage: [Video Input] [Non-Video Input] [Video Output] [Width] [Height]
//            [Num Frames] [FpS]

int
test_FileListVideoIOFactory(const char * input, char * output, itk::SizeValueType itkNotUsed(cameraNumber))
{

  int ret = EXIT_SUCCESS;

  //////
  // Register the FileListVideoIOFactory.
  //
  // There's something strange going on here that makes the factories not be
  // registered by default because of the order in which the includes happen.
  // The real strangeness seems to be in ITK's system with the modularized
  // framework since none of the factories get reigstered by default.
  itk::ObjectFactoryBase::RegisterFactory(itk::FileListVideoIOFactory::New());

  //////
  // Create the VideoIOBase for reading from a file
  //////
  std::cout << "Trying to create IO for reading from file..." << std::endl;
  itk::VideoIOBase::Pointer ioReadFile =
    itk::VideoIOFactory::CreateVideoIO(itk::VideoIOFactory::IOModeEnum::ReadFileMode, input);
  if (!ioReadFile)
  {
    std::cerr << "Did not create valid VideoIO for reading from file " << std::endl;
    ret = EXIT_FAILURE;
  }

  //////
  // Create the VideoIOBase for writing to a file
  //////
  std::cout << "Trying to create IO for writing to file..." << std::endl;
  itk::VideoIOBase::Pointer ioWrite =
    itk::VideoIOFactory::CreateVideoIO(itk::VideoIOFactory::IOModeEnum::WriteMode, output);
  if (!ioWrite)
  {
    std::cerr << "Did not create valid VideoIO for writing " << std::endl;
    ret = EXIT_FAILURE;
  }

  std::cout << "Done !" << std::endl;
  return ret;
}

int
itkFileListVideoIOFactoryTest(int argc, char * argv[])
{
  if (argc != 8)
  {
    std::cerr << "Usage: " << itkNameOfTestExecutableMacro(argv) << " [Video Input] [Video Output] [Webcam Number]"
              << std::endl;
    return EXIT_FAILURE;
  }

  std::string inFile = "";
  for (int i = 1; i <= 5; ++i)
  {
    inFile = inFile + std::string(argv[i]);
    if (i != 5)
    {
      inFile = inFile + std::string(",");
    }
  }

  // Test streaming enumeration for VideoIOFactoryEnums::IOMode elements
  const std::set<itk::VideoIOFactoryEnums::IOMode> allIOMode{ itk::VideoIOFactoryEnums::IOMode::ReadFileMode,
                                                              itk::VideoIOFactoryEnums::IOMode::ReadCameraMode,
                                                              itk::VideoIOFactoryEnums::IOMode::WriteMode };
  for (const auto & ee : allIOMode)
  {
    std::cout << "STREAMED ENUM VALUE VideoIOFactoryEnums::IOMode: " << ee << std::endl;
  }

  return test_FileListVideoIOFactory(inFile.c_str(), argv[6], std::stoi(argv[7]));
}
