add logo to 3d plots when possible

ImpedanCEI · Jan 22, 2025 · 9a6e5cf · 9a6e5cf
1 parent b31f73f
commit 9a6e5cf
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/wakis/plotting.py b/wakis/plotting.py
@@ -98,7 +98,9 @@ def plot3D(self, field='E', component='z', clim=None, hide_solids=None,
             pl.camera_position = 'zx'
             pl.camera.azimuth += 30
             pl.camera.elevation += 30
-            #pl.background_color = "grey"
+            pl.set_background('mistyrose', top='white')
+            try: pl.add_logo_widget('../docs/img/wakis-logo-pink.png')
+            except: pass
             pl.camera.zoom(zoom)
             pl.add_axes()
             pl.enable_3_lights()
@@ -386,7 +388,9 @@ def plot3DonSTL(self, field='E', component='z', clim=None, cmap='jet', log_scale
         pl.camera_position = 'zx'
         pl.camera.azimuth += 20
         pl.camera.elevation += 30
-        #pl.background_color = "grey"
+        pl.set_background('mistyrose', top='white')
+        try: pl.add_logo_widget('../docs/img/wakis-logo-pink.png')
+        except: pass
         pl.camera.zoom(zoom)
         pl.add_axes()
         pl.enable_anti_aliasing()