From 344882303a26b77b18452d88ab2db39f80cb3c7e Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Fri, 21 Apr 2017 12:01:46 -0500
Subject: [PATCH] Add option to generate neighbor lists (Closes #25)

---
 meshpy/triangle.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/meshpy/triangle.py b/meshpy/triangle.py
index a4f310d..3a5e385 100644
--- a/meshpy/triangle.py
+++ b/meshpy/triangle.py
@@ -126,7 +126,7 @@ def build(mesh_info, verbose=False, refinement_func=None, attributes=False,
         volume_constraints=False, max_volume=None, allow_boundary_steiner=True,
         allow_volume_steiner=True, quality_meshing=True,
         generate_edges=None, generate_faces=False, min_angle=None,
-        mesh_order=None):
+        mesh_order=None, generate_neighbor_lists=False):
     """Triangulate the domain given in `mesh_info'."""
     opts = "pzj"
     if quality_meshing:
@@ -158,6 +158,8 @@ def build(mesh_info, verbose=False, refinement_func=None, attributes=False,
         from warnings import warn
         warn("generate_edges is deprecated--use generate_faces instead")
         generate_faces = generate_edges
+    if generate_neighbor_lists is not None:
+        opts += "n"
 
     if generate_faces:
         opts += "e"
@@ -193,7 +195,7 @@ def build(mesh_info, verbose=False, refinement_func=None, attributes=False,
 
 
 def refine(input_p, verbose=False, refinement_func=None,  quality_meshing=True,
-        min_angle=None):
+        min_angle=None, generate_neighbor_lists=False):
     opts = "razj"
 
     if quality_meshing:
@@ -210,6 +212,8 @@ def refine(input_p, verbose=False, refinement_func=None,  quality_meshing=True,
         opts += "Q"
     if refinement_func is not None:
         opts += "u"
+    if generate_neighbor_lists is not None:
+        opts += "n"
 
     output_p = MeshInfo()
     internals.triangulate(opts, input_p, output_p, MeshInfo(), refinement_func)
-- 
GitLab