/*=========================================================================
 *
 *  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 <cstdlib>
#include <cmath>
#include <metaMesh.h>
#include <metaScene.h>
#include "itksys/SystemTools.hxx"
#include "itkMacro.h"
#include "itkMath.h"


bool
TestingMetaMesh(MetaMesh * _mesh)
{
  int j;
  // Testing Points
  std::cout << "Testing Points : ";
  using PointListType = MetaMesh::PointListType;
  PointListType::const_iterator it2 = _mesh->GetPoints().begin();
  for (j = 0; j < static_cast<int>(_mesh->GetPoints().size()); ++j)
  {
    if (((*it2)->m_Id != j) || (itk::Math::NotExactlyEquals((*it2)->m_X[0], j)) ||
        (itk::Math::NotExactlyEquals((*it2)->m_X[1], j)) || (itk::Math::NotExactlyEquals((*it2)->m_X[2], j)))
    {
      std::cout << (*it2)->m_Id << " : " << (*it2)->m_X[0] << " " << (*it2)->m_X[1] << " " << (*it2)->m_X[2]
                << std::endl;
      std::cout << "[FAILED]" << std::endl;
      return EXIT_FAILURE;
    }
    ++it2;
  }
  std::cout << "[PASSED]" << std::endl;

  // Testing cells
  std::cout << "Testing Cells : ";
  using CellListType = MetaMesh::CellListType;
  CellListType::const_iterator it3 = _mesh->GetCells(MET_TETRAHEDRON_CELL).begin();
  for (j = 0; j < static_cast<int>(_mesh->GetCells(MET_TETRAHEDRON_CELL).size()); ++j)
  {
    if (((*it3)->m_Dim != 4) || ((*it3)->m_Id != j))
    {
      std::cout << "Cell Type = " << (*it3)->m_Dim << " : " << (*it3)->m_Id << " : ";
      std::cout << "[FAILED]" << std::endl;
      return EXIT_FAILURE;
    }

    for (int k = 0; k < static_cast<int>((*it3)->m_Dim); ++k)
    {
      if ((*it3)->m_PointsId[k] != j + k)
      {
        std::cout << (*it3)->m_PointsId[k] << " ";
        std::cout << "[FAILED]" << std::endl;
        return EXIT_FAILURE;
      }
    }
    ++it3;
  }
  it3 = _mesh->GetCells(MET_TRIANGLE_CELL).begin();
  for (j = 0; j < static_cast<int>(_mesh->GetCells(MET_TRIANGLE_CELL).size()); ++j)
  {
    if (((*it3)->m_Dim != 3) || ((*it3)->m_Id != j))
    {
      std::cout << "Cell Type = " << (*it3)->m_Dim << " : " << (*it3)->m_Id << " : ";
      std::cout << "[FAILED]" << std::endl;
      return EXIT_FAILURE;
    }
    for (int k = 0; k < static_cast<int>((*it3)->m_Dim); ++k)
    {
      if ((*it3)->m_PointsId[k] != j + k)
      {
        std::cout << (*it3)->m_PointsId[k] << " ";
        std::cout << "[FAILED]" << std::endl;
        return EXIT_FAILURE;
      }
    }
    ++it3;
  }
  std::cout << "[PASSED]" << std::endl;

  // Testing cell links
  std::cout << "Testing CellLinks : ";
  using CellLinkListType = MetaMesh::CellLinkListType;
  CellLinkListType::const_iterator it_link = _mesh->GetCellLinks().begin();
  for (j = 0; j < static_cast<int>(_mesh->GetCellLinks().size()); ++j)
  {
    if ((*it_link)->m_Id != j)
    {
      std::cout << "CellLink ID = " << (*it_link)->m_Id << " : ";
      std::cout << "[FAILED]" << std::endl;
      return EXIT_FAILURE;
    }
    std::list<int>::const_iterator it_link2 = (*it_link)->m_Links.begin();
    while (it_link2 != (*it_link)->m_Links.end())
    {
      if (*it_link2 != j + 1)
      {
        std::cout << "[FAILED]" << std::endl;
        return EXIT_FAILURE;
      }
      ++it_link2;
    }
    ++it_link;
  }
  std::cout << "[PASSED]" << std::endl;

  // Testing PointData
  std::cout << "Testing PointData : ";
  using PointDataListType = MetaMesh::PointDataListType;
  PointDataListType::const_iterator it_pd = _mesh->GetPointData().begin();
  for (j = 0; j < static_cast<int>(_mesh->GetPointData().size()); ++j)
  {
    if (((*it_pd)->m_Id != j) || (static_cast<int>(static_cast<MeshData<int> *>(*it_pd)->m_Data) != j))
    {
      std::cout << "PointData ID = " << (*it_pd)->m_Id << " : "
                << static_cast<int>(static_cast<MeshData<int> *>(*it_pd)->m_Data) << std::endl;
      std::cout << "[FAILED]" << std::endl;
      return EXIT_FAILURE;
    }
    ++it_pd;
  }
  std::cout << "[PASSED]" << std::endl;

  // Testing CellData
  std::cout << "Testing CellData : ";
  using CellDataListType = MetaMesh::CellDataListType;
  CellDataListType::const_iterator it_cd = _mesh->GetCellData().begin();
  auto                             f = static_cast<float>(0.1);
  for (j = 0; j < static_cast<int>(_mesh->GetCellData().size()); ++j)
  {
    if (((*it_cd)->m_Id != j) ||
        (itk::Math::abs(static_cast<float>(static_cast<MeshData<float> *>(*it_cd)->m_Data) - f) > 0.001))
    {
      std::cout << "CellData ID = " << (*it_cd)->m_Id << " : "
                << static_cast<float>(static_cast<MeshData<float> *>(*it_cd)->m_Data) << " : " << f << std::endl;
      std::cout << "[FAILED]" << std::endl;
      return EXIT_FAILURE;
    }
    f += static_cast<float>(0.2);
    ++it_cd;
  }
  std::cout << "[PASSED]" << std::endl;
  return EXIT_SUCCESS;
}


/** Main */
int
testMetaMesh(int argc, char * argv[])
{
  if (argc > 1)
  {
    itksys::SystemTools::ChangeDirectory(argv[1]);
  }

  MetaScene myScene = MetaScene(3);

  std::cout << "Creating mesh: ";
  auto * mesh = new MetaMesh(3);
  mesh->ID(0);

  // Add Points
  MeshPoint * pnt;
  int         i;
  for (i = 0; i < 10; ++i)
  {
    pnt = new MeshPoint(3);
    pnt->m_X[0] = pnt->m_X[1] = pnt->m_X[2] = static_cast<float>(i);
    pnt->m_Id = i;
    mesh->GetPoints().push_back(pnt);
  }

  // Add Cells
  MeshCell * cell;
  for (i = 0; i < 6; ++i)
  {
    cell = new MeshCell(4); // tetrahedra
    cell->m_Id = i;
    cell->m_PointsId[0] = i;
    cell->m_PointsId[1] = i + 1;
    cell->m_PointsId[2] = i + 2;
    cell->m_PointsId[3] = i + 3;
    mesh->GetCells(MET_TETRAHEDRON_CELL).push_back(cell);
  }

  // Add other type of cells
  for (i = 0; i < 4; ++i)
  {
    cell = new MeshCell(3); // triangle
    cell->m_Id = i;
    cell->m_PointsId[0] = i;
    cell->m_PointsId[1] = i + 1;
    cell->m_PointsId[2] = i + 2;
    mesh->GetCells(MET_TRIANGLE_CELL).push_back(cell);
  }

  // Add cell links
  for (i = 0; i < 3; ++i)
  {
    auto * link = new MeshCellLink();
    link->m_Id = i;
    link->m_Links.push_back(i + 1);
    mesh->GetCellLinks().push_back(link);
  }

  // Add point data
  for (i = 0; i < 5; ++i)
  {
    auto * pd = new MeshData<int>();
    pd->m_Id = i;
    pd->m_Data = i;
    mesh->GetPointData().push_back(pd);
  }

  // Add cell data
  auto f = static_cast<float>(0.1);
  for (i = 0; i < 4; ++i)
  {
    auto * cd = new MeshData<float>();
    cd->m_Id = i;
    cd->m_Data = f;
    f += static_cast<float>(0.2);
    mesh->GetCellData().push_back(cd);
  }
  std::cout << "[PASSED]" << std::endl;

  // Write the mesh
  std::cout << "Writing non binary Mesh : ";
  myScene.AddObject(mesh);
  myScene.BinaryData(false);
  myScene.Write("metamesh.msh");
  std::cout << "[PASSED]" << std::endl;

  std::cout << "Reading non binary Mesh : ";
  // Read the mesh
  MetaScene myScene2 = MetaScene();
  myScene2.InitializeEssential(3);
  myScene2.Read("metamesh.msh");
  std::cout << "[PASSED]" << std::endl;

  using ListType = MetaScene::ObjectListType;
  ListType * list = myScene2.GetObjectList();
  auto       it = list->begin();

  for (i = 0; i < static_cast<int>(list->size()); ++i)
  {
    if (!strncmp((*it)->ObjectTypeName(), "Mesh", 4))
    {
      auto * mesh2 = dynamic_cast<MetaMesh *>(*it);
      if (TestingMetaMesh(mesh2))
      {
        std::cout << "[FAILED]" << std::endl;
        return EXIT_FAILURE;
      }
      (mesh2)->PrintInfo();
    }
    ++it;
  }

  // Now testing Binary mesh
  std::cout << "Writing binary Mesh : ";
  myScene.BinaryData(true);
  myScene.Write("metamesh.msh");
  std::cout << "[PASSED]" << std::endl;

  std::cout << "Reading binary Mesh : ";
  // Read the mesh
  MetaScene myScene3 = MetaScene();
  myScene3.InitializeEssential(3);
  myScene3.Read("metamesh.msh");
  std::cout << "[PASSED]" << std::endl;

  list = myScene3.GetObjectList();
  it = list->begin();

  for (i = 0; i < static_cast<int>(list->size()); ++i)
  {
    if (!strncmp((*it)->ObjectTypeName(), "Mesh", 4))
    {
      auto * mesh2 = dynamic_cast<MetaMesh *>(*it);
      if (TestingMetaMesh(mesh2))
      {
        std::cout << "[FAILED]" << std::endl;
        return EXIT_FAILURE;
      }
      (mesh2)->PrintInfo();
    }
    ++it;
  }
  std::cout << "[DONE]" << std::endl;
  return EXIT_SUCCESS;
}