#include "dihedral_angles_filter.h"
#include "remove_external_cells_filter.h"
#include "surface_points_filter.h"
#include "project_surface_points_on_poly.h"

#include <vtkCellArrayIterator.h>
#include <vtkCellData.h>
#include <vtkDataSetReader.h>
#include <vtkOBJReader.h>
#include <vtkPolyData.h>
#include <vtkPolyDataReader.h>
#include <vtkUnstructuredGrid.h>
#include <vtkUnstructuredGridReader.h>
#include <vtkXMLPolyDataReader.h>
#include <vtkXMLUnstructuredGridReader.h>
#include <vtkXMLUnstructuredGridWriter.h>
#include <vtksys/SystemTools.hxx>

#ifdef USE_VIEWER
#include <vtkCamera.h>
#include <vtkNamedColors.h>
#include <vtkOpenGLProjectedTetrahedraMapper.h>
#include <vtkPiecewiseFunction.h>
#include <vtkRenderWindow.h>
#include <vtkRenderWindowInteractor.h>
#include <vtkRenderer.h>
#include <vtkVolume.h>
#include <vtkVolumeMapper.h>
#include <vtkVolumeProperty.h>
#endif // USE_VIEWER

#include <array>


vtkSmartPointer<vtkAlgorithm> readerFromFileName(const char *fileName) {
	std::string extension =
		vtksys::SystemTools::GetFilenameLastExtension(fileName);
	if (extension == ".vtk") {
		auto reader = vtkDataSetReader::New();
		reader->SetFileName(fileName);
		return {reader};
	} else if (extension == ".vtu") {
		auto reader = vtkXMLUnstructuredGridReader::New();
		reader->SetFileName(fileName);
		return {reader};
	} else if (extension == ".vtp") {
		auto reader = vtkXMLPolyDataReader::New();
		reader->SetFileName(fileName);
		return {reader};
	} else if (extension == ".obj") {
		auto reader = vtkOBJReader::New();
		reader->SetFileName(fileName);
		return {reader};
	}
	throw std::runtime_error("Invalid file extension.");
}


int main(int argc, char **argv) {
	if (argc != 3) {
		std::cerr << "Usage: " << argv[0] << " tetmesh polydata" << std::endl;
		return EXIT_FAILURE;
	}

	auto tetMeshReader = readerFromFileName(argv[1]);
	auto polyMeshReader = readerFromFileName(argv[2]);

	vtkNew<DihedralAnglesFilter> dihedralAnglesFilter;
	dihedralAnglesFilter->SetInputConnection(tetMeshReader->GetOutputPort());

	vtkNew<RemoveExternalCellsFilter> removeExternalCellsFilter;
	removeExternalCellsFilter->SetInputConnection(0,
		dihedralAnglesFilter->GetOutputPort());
	removeExternalCellsFilter->SetInputConnection(1,
		polyMeshReader->GetOutputPort());

	vtkNew<SurfacePointsFilter> surfacePointsFilter;
	surfacePointsFilter->SetInputConnection(
		removeExternalCellsFilter->GetOutputPort());

	vtkNew<ProjectSurfacePointsOnPoly> project;
	project->SetInputConnection(0, surfacePointsFilter->GetOutputPort());
	project->SetInputConnection(1, polyMeshReader->GetOutputPort());

	vtkNew<vtkXMLUnstructuredGridWriter> writer;
	writer->SetInputConnection(project->GetOutputPort());
	writer->SetDataModeToAscii();
	writer->SetFileName("out.vtu");
	writer->Write();

#ifdef USE_VIEWER
	/* Volume rendering properties */
	vtkNew<vtkOpenGLProjectedTetrahedraMapper> volumeMapper;
	volumeMapper->SetInputConnection(externalPointsFilter->GetOutputPort());
	vtkNew<vtkVolume> volume;
	volume->SetMapper(volumeMapper);
	vtkNew<vtkPiecewiseFunction> transferFunction;
	transferFunction->AddPoint(-1, 0);
	transferFunction->AddPoint(1, 1);
	volume->GetProperty()->SetScalarOpacity(transferFunction);
	volume->GetProperty()->SetColor(transferFunction);

	/* Renderer */
	vtkNew<vtkNamedColors> colors;
	std::array<unsigned char, 4> bkg{{26, 51, 102, 255}};
	colors->SetColor("BkgColor", bkg.data());
	vtkNew<vtkRenderer> renderer;
	renderer->AddVolume(volume);
	renderer->SetBackground(colors->GetColor3d("BkgColor").GetData());
	renderer->ResetCamera();
	renderer->GetActiveCamera()->Zoom(1.5);
	vtkNew<vtkRenderWindow> renderWindow;
	renderWindow->SetSize(300, 300);
	renderWindow->AddRenderer(renderer);
	renderWindow->SetWindowName("PFE");
	vtkNew<vtkRenderWindowInteractor> renderWindowInteractor;
	renderWindowInteractor->SetRenderWindow(renderWindow);
	renderWindow->Render();
	renderWindowInteractor->Start();
#endif
	return EXIT_SUCCESS;
}