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

int
itkQuadEdgeMeshTest2(int, char *[])
{
  std::cout << "Testing points and simple edges... " << std::ends;

  using PixelType = double;
  using MeshType = itk::QuadEdgeMesh<PixelType, 3>;
  using CellType = MeshType::CellType;
  using LineType = itk::QuadEdgeMeshLineCell<CellType>;
  using QuadEdgeType = LineType::QEType;
  using CellAutoPointer = CellType::CellAutoPointer;

  auto mesh = MeshType::New();

  MeshType::PointType p0;
  MeshType::PointType p1;
  MeshType::PointType p2;

  p0[0] = -1.0;
  p0[1] = 0.0;
  p0[2] = 0.0;
  p1[0] = 1.0;
  p1[1] = 0.0;
  p1[2] = 0.0;
  p2[0] = 1.0;
  p2[1] = 1.0;
  p2[2] = 0.0;

  mesh->SetPoint(0, p0);
  mesh->SetPoint(1, p1);
  mesh->SetPoint(2, p2);

  CellAutoPointer line0;
  line0.TakeOwnership(new LineType);
  line0->SetPointId(0, 0);
  line0->SetPointId(1, 1);
  mesh->SetCell(0, line0);

  CellAutoPointer line1;
  line1.TakeOwnership(new LineType);
  line1->SetPointId(0, 1);
  line1->SetPointId(1, 2);
  mesh->SetCell(1, line1);

  CellAutoPointer line2;
  line2.TakeOwnership(new LineType);
  line2->SetPointId(0, 2);
  line2->SetPointId(1, 0);
  mesh->SetCell(2, line2);

  if (mesh->GetNumberOfPoints() != 3)
  {
    std::cout << "Not all points added." << std::endl;
    return EXIT_FAILURE;
  }

  if ((mesh->GetNumberOfCells() != 0) && (mesh->GetNumberOfEdges() != 3))
  {
    std::cout << "Not all cells added." << std::endl;
    return EXIT_FAILURE;
  }

  using CellIterator = MeshType::CellsContainer::Iterator;
  CellIterator cellIterator = mesh->GetCells()->Begin();
  unsigned int ids[] = { 0, 1, 2, 1, 2, 0, 2, 0, 1 };
  int          itIds = 0;

  while (cellIterator != mesh->GetCells()->End())
  {
    MeshType::CellType * cellptr = cellIterator.Value();
    auto *               lineCell = dynamic_cast<LineType *>(cellptr);
    lineCell->GetNameOfClass();
    QuadEdgeType *             QEGeom = lineCell->GetQEGeom();
    QuadEdgeType::IteratorGeom git = QEGeom->BeginGeomLnext();
    while (git != QEGeom->EndGeomLnext())
    {
      if (ids[itIds] != *git)
      {
        std::cout << "Problem with splicing edges." << std::endl;
        return EXIT_FAILURE;
      }
      itIds++;
      git++;
    }
    cellIterator++;
  }

  std::cout << "done!" << std::endl;
  return EXIT_SUCCESS;
}
