/*=========================================================================
 *
 *  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.
 *
 *=========================================================================*/

#ifndef itkOFFMeshIO_h
#define itkOFFMeshIO_h
#include "ITKIOMeshOFFExport.h"

#include "itkMeshIOBase.h"
#include "itkMakeUniqueForOverwrite.h"

#include <fstream>

namespace itk
{
/**
 * \class OFFMeshIO
 * \brief this class defines how to read and write Object file format.
 * \ingroup IOFilters
 * \ingroup ITKIOMeshOFF
 */

class ITKIOMeshOFF_EXPORT OFFMeshIO : public MeshIOBase
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(OFFMeshIO);

  /** Standard class type aliases. */
  using Self = OFFMeshIO;
  using Superclass = MeshIOBase;
  using ConstPointer = SmartPointer<const Self>;
  using Pointer = SmartPointer<Self>;

  using SizeValueType = Superclass::SizeValueType;
  using StreamOffsetType = Superclass::StreamOffsetType;

  /** Method for creation through the object factory. */
  itkNewMacro(Self);

  /** \see LightObject::GetNameOfClass() */
  itkOverrideGetNameOfClassMacro(OFFMeshIO);

  /*-------- This part of the interfaces deals with reading data. ----- */

  /** Determine if the file can be read with this MeshIO implementation.
   * \param fileName The name of the file to test for reading.
   * \post Sets classes MeshIOBase::m_FileName variable to be FileNameToWrite
   * \return Returns true if this MeshIO can read the file specified.
   */
  bool
  CanReadFile(const char * fileName) override;

  /** Set the spacing and dimension information for the set filename. */
  void
  ReadMeshInformation() override;

  /** Reads the data from disk into the memory buffer provided. */
  void
  ReadPoints(void * buffer) override;

  void
  ReadCells(void * buffer) override;

  void
  ReadPointData(void * buffer) override;

  void
  ReadCellData(void * buffer) override;

  /*-------- This part of the interfaces deals with writing data. ----- */

  /** Determine if the file can be written with this MeshIO implementation.
   * \param fileName The name of the file to test for writing.
   * \post Sets classes MeshIOBase::m_FileName variable to be FileNameToWrite
   * \return Returns true if this MeshIO can write the file specified.
   */
  bool
  CanWriteFile(const char * fileName) override;

  /** Set the spacing and dimension information for the set filename. */
  void
  WriteMeshInformation() override;

  /** Writes the data to disk from the memory buffer provided. Make sure
   * that the IORegions has been set properly. */
  void
  WritePoints(void * buffer) override;

  void
  WriteCells(void * buffer) override;

  void
  WritePointData(void * buffer) override;

  void
  WriteCellData(void * buffer) override;

  void
  Write() override;

protected:
  /** Read buffer as ascii stream */
  template <typename T>
  void
  ReadCellsBufferAsAscii(T * buffer, std::ifstream & inputFile)
  {
    SizeValueType index = 0;
    unsigned int  numberOfPoints = 0;
    std::string   line;

    for (SizeValueType ii = 0; ii < this->m_NumberOfCells; ++ii)
    {
      inputFile >> numberOfPoints;
      buffer[index++] = static_cast<T>(numberOfPoints);
      for (unsigned int jj = 0; jj < numberOfPoints; ++jj)
      {
        inputFile >> buffer[index++];
      }
      std::getline(inputFile, line, '\n');
    }
  }

  /** Read cells from a data buffer, used when writing cells. This function
    write all kind of cells as it is stored in cells container. It is used when
    cells container have only one kind of cells */
  template <typename TInput, typename TOutput>
  void
  ReadCellsBuffer(TInput * input, TOutput * output)
  {
    if (input && output)
    {
      SizeValueType indInput = 0;
      SizeValueType indOutput = 0;
      for (SizeValueType ii = 0; ii < this->m_NumberOfCells; ++ii)
      {
        ++indInput; // ignore the cell type
        auto numberOfPoints = static_cast<unsigned int>(input[indInput++]);
        output[indOutput++] = static_cast<TOutput>(numberOfPoints);
        for (unsigned int jj = 0; jj < numberOfPoints; ++jj)
        {
          output[indOutput++] = static_cast<TOutput>(input[indInput++]);
        }
      }
    }
  }

  template <typename T>
  void
  WriteCellsAsAscii(T * buffer, std::ofstream & outputFile)
  {
    SizeValueType index = 0;

    for (SizeValueType ii = 0; ii < this->m_NumberOfCells; ++ii)
    {
      ++index;
      auto numberOfCellPoints = static_cast<unsigned int>(buffer[index++]);
      outputFile << numberOfCellPoints << "  ";

      for (unsigned int jj = 0; jj < numberOfCellPoints; ++jj)
      {
        outputFile << buffer[index++] << "  ";
      }

      outputFile << '\n';
    }
  }

  template <typename TOutput, typename TInput>
  void
  WriteCellsAsBinary(TInput * buffer, std::ofstream & outputFile)
  {
    const auto data = make_unique_for_overwrite<TOutput[]>(m_CellBufferSize - this->m_NumberOfCells);

    ReadCellsBuffer(buffer, data.get());
    WriteBufferAsBinary<TOutput>(data.get(), outputFile, m_CellBufferSize - this->m_NumberOfCells);
  }

protected:
  OFFMeshIO();
  ~OFFMeshIO() override;

  void
  PrintSelf(std::ostream & os, Indent indent) const override;

  void
  OpenFile();

  void
  CloseFile();

private:
  std::ifstream    m_InputFile{};
  StreamOffsetType m_PointsStartPosition{}; // file position for points relative to std::ios::beg
  bool             m_TriangleCellType{};    // if all cells are triangle it is true. otherwise, it is false.
};
} // end namespace itk

#endif
