importPolylines must work with only one polyline in shape

diff --git a/src/HYDROTools/hydroGeoMeshUtils.py b/src/HYDROTools/hydroGeoMeshUtils.py
index 990dd9c477e20ec1ea9253e89267316953d5ea99..e23d44c61b874cb135010de18850894635b3840a 100644 (file)
--- a/src/HYDROTools/hydroGeoMeshUtils.py
+++ b/src/HYDROTools/hydroGeoMeshUtils.py
@@ -35,7 +35,14 @@ def importPolylines(document, shapeFile, shapeName, iSpline, displayLevel):
         shapeType = 1 # polyline
     shapes = []
     isShapeFound = True
-    index = 0
+    shape = document.FindObjectByName(shapeName) # single shape
+    if shape is not None:
+        for i in range(shape.NbSections()):
+            shape.SetSectionType(i, shapeType)
+            shape.Update()
+            shape.SetZLevel( displayLevel )
+        shapes.append(shape)    
+    index = 0  # for multiple shapes
     while isShapeFound:
         shapeNameIndex = shapeName + "_%d" % index
         index = index + 1