plotting: accept trimeshes directly

navis-org · Oct 14, 2020 · 2d75b0c · 2d75b0c
1 parent c1e8af6
commit 2d75b0c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/navis/utils/misc.py b/navis/utils/misc.py
@@ -301,7 +301,9 @@ def parse_objects(x) -> Tuple['core.NeuronList',
         dotprops = pd.concat(dps)
 
     # Collect and parse volumes
-    volumes = [ob for ob in x if isinstance(ob, core.Volume)]
+    volumes = [ob for ob in x if isinstance(ob, tm.Trimesh)]
+    # Converts trimeshes into Volumes
+    volumes = [core.Volume(v) if not isinstance(v, core.Volume) else v for v in volumes]
 
     # Collect dataframes with X/Y/Z coordinates
     # Note: dotprops and volumes are instances of pd.DataFrames