From 52fc7bd32e6241fe41515f7ce3fd08756284d5ce Mon Sep 17 00:00:00 2001
From: papush! <contact@ccolin.fr>
Date: Mon, 14 Mar 2022 14:30:11 +0100
Subject: [PATCH] :)

---
 CMakeLists.txt | 1 +
 src/main.cc    | 6 +++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 5413eec..b259f06 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -12,6 +12,7 @@ set(VTK_COMPONENTS
   VTK::IOGeometry
   VTK::IOXML
   VTK::FiltersModeling
+  VTK::FiltersPoints
   VTK::vtksys)
 set(ENABLE_VIEWER OFF CACHE BOOL "Enable the 3D viewer, depends on Qt.")
 if(ENABLE_VIEWER)
diff --git a/src/main.cc b/src/main.cc
index 6d54bcb..edb10b0 100644
--- a/src/main.cc
+++ b/src/main.cc
@@ -15,6 +15,7 @@
 #include <vtkXMLUnstructuredGridReader.h>
 #include <vtkXMLUnstructuredGridWriter.h>
 #include <vtksys/SystemTools.hxx>
+#include <vtkPointSmoothingFilter.h>
 
 #ifdef USE_VIEWER
 #include <vtkCamera.h>
@@ -82,8 +83,11 @@ int main(int argc, char **argv) {
 	vtkNew<DihedralAnglesFilter> dihedralAnglesFilter;
 	dihedralAnglesFilter->SetInputConnection(project->GetOutputPort());
 
+	vtkNew<vtkPointSmoothingFilter> pointSmoothingFilter;
+	pointSmoothingFilter->SetInputConnection(dihedralAnglesFilter->GetOutputPoort());
+
 	vtkNew<vtkXMLUnstructuredGridWriter> writer;
-	writer->SetInputConnection(dihedralAnglesFilter->GetOutputPort());
+	writer->SetInputConnection(pointSmoothingFilter->GetOutputPort());
 	writer->SetDataModeToAscii();
 	writer->SetFileName("out.vtu");
 	writer->Write();