Reading obj mesh files

Thiago Franco de Moraes
1 parent 65bab099
Showing 2 changed files with 7 additions and 4 deletions Show diff stats
invesalius/data/surface.py
invesalius/gui/dialogs.py
@@ -245,7 +245,7 @@ class SurfaceManager():
     def OnImportSurfaceFile(self, pubsub_evt):
         """
-        Creates a new surface from a surface file (STL, PLY or VTP)
+        Creates a new surface from a surface file (STL, PLY, OBJ or VTP)
         """
         filename = pubsub_evt.data
         self.CreateSurfaceFromFile(filename)
@@ -255,6 +255,8 @@ class SurfaceManager():
             reader = vtk.vtkSTLReader()
         elif filename.lower().endswith('.ply'):
             reader = vtk.vtkPLYReader()
+        elif filename.lower().endswith('.obj'):
+            reader = vtk.vtkOBJReader()
         elif filename.lower().endswith('.vtp'):
             reader = vtk.vtkXMLPolyDataReader()
         else:
@@ -223,9 +223,10 @@ WILDCARD_NIFTI = &quot;NIfTI 1 (*.nii)|*.nii|&quot; \
 WILDCARD_PARREC = "PAR/REC (*.par)|*.par|" \
                   "All files (*.*)|*.*"
-WILDCARD_MESH_FILES = "STL (*.stl)|*.stl|" \
-                      "PLY (*.ply)|*.ply|" \
-                      "VTP (*.vtp)|*.vtp|" \
+WILDCARD_MESH_FILES = "STL File format (*.stl)|*.stl|" \
+                      "Standard Polygon File Format (*.ply)|*.ply|" \
+                      "Alias Wavefront Object (*.obj)|*.obj|" \
+                      "VTK Polydata File Format (*.vtp)|*.vtp|" \
                       "All files (*.*)|*.*"