/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  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
 *
 *         http://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 "itkRegularSphereMeshSource.h"
#include "itkVTKPolyDataWriter.h"

#include <iostream>

int itkVTKPolyDataWriterTest02( int argc, char * argv [] )
{
  if( argc != 2 )
    {
    std::cerr << "Usage: itkVTKPolyDataWriter outputFileName" << std::endl;
    return EXIT_FAILURE;
    }


  typedef itk::Mesh<float, 3>   MeshType;

  typedef itk::RegularSphereMeshSource< MeshType >  SphereMeshSourceType;

  SphereMeshSourceType::Pointer  mySphereMeshSource = SphereMeshSourceType::New();

  typedef SphereMeshSourceType::PointType   PointType;
  typedef SphereMeshSourceType::VectorType  VectorType;

  PointType center;
  center.Fill( 7.4 );

  const double radius = 1.5;
  const double tolerance = 1e-5;

  VectorType scale;
  scale.Fill( radius );

  mySphereMeshSource->SetCenter( center );
  mySphereMeshSource->SetResolution( 1 );
  mySphereMeshSource->SetScale( scale );

  mySphereMeshSource->Modified();

  try
    {
    mySphereMeshSource->Update();
    }
  catch( itk::ExceptionObject & excp )
    {
    std::cerr << "Error during Update() " << std::endl;
    std::cerr << excp << std::endl;
    }

  std::cout << "mySphereMeshSource: " << mySphereMeshSource;

  MeshType::Pointer myMesh = mySphereMeshSource->GetOutput();

  PointType pt;
  pt.Fill( 0. );

  bool testPassed = true;

  std::cout << "Testing itk::RegularSphereMeshSource "<< std::endl;

  for( unsigned int i=0; i< myMesh->GetNumberOfPoints(); i++)
    {
    myMesh->GetPoint(i, &pt);
    std::cout << "Point[" << i << "]: " << pt << std::endl;

    const double distanceToCenter = pt.EuclideanDistanceTo( center );

    if( itk::Math::abs( distanceToCenter - radius ) > tolerance )
      {
      std::cerr << "Distance to center " << distanceToCenter;
      std::cerr << " is too different from radius " << radius << std::endl;
      testPassed = false;
      }
    }

  typedef MeshType::CellsContainerPointer  CellsContainerPointer;
  typedef MeshType::CellType               CellType;

  CellsContainerPointer cells = myMesh->GetCells();

  unsigned faceId = 0;

  MeshType::CellsContainerIterator cellsItr = cells->Begin();


  while( cellsItr != cells->End() )
    {
    CellType * cellPointer = cellsItr.Value();

    if( cellPointer->GetType() != 1 )
      {
      const unsigned int numberOfPoints = cellPointer->GetNumberOfPoints();

      std::cout <<"Face " << faceId << " has " << numberOfPoints <<" points" << std::endl;

      if( numberOfPoints != 3 )
        {
        std::cerr << "Face with wrong number of points" << std::endl;
        testPassed = false;
        }
      }

    ++cellsItr;
    ++faceId;
    }


  typedef itk::VTKPolyDataWriter<MeshType>   WriterType;

  WriterType::Pointer writer = WriterType::New();
  writer->SetInput( myMesh  );
  writer->SetFileName( argv[1] );
  writer->Write();


  if( !testPassed )
    {
    std::cout << "Test FAILED! "<< std::endl;
    return EXIT_FAILURE;
    }

  std::cout << "Test PASSED! "<< std::endl;

  return EXIT_SUCCESS;

}
