add "fix axis" option in simulation

Author: bgraille

pylbm_ui/simulation.py

@@ -23,6 +23,7 @@ def __init__(self):
         self.fig.canvas.header_visible = False
         self.plot_type = None
         self.color_bar = None
+        self.fix_ylim = False
         # plt.ion()
 
     def plot(self, t, domain, field, data, transpose=True, properties=None):
@@ -63,11 +64,22 @@ def plot(self, t, domain, field, data, transpose=True, properties=None):
                     marker=marker,
                     markersize=markersize
                 )
+                ylim = None
+                if 'set_ylim' in properties:
+                    ylim = properties['set_ylim']
+                if ylim is not None:
+                    self.ax.set_ylim(
+                        ymin=float(ylim[0]),
+                        ymax=float(ylim[1]),
+                        auto=False
+                    )
+                    self.fix_ylim = True
             else:
                 self.plot_type.set_ydata(data)
-
-            self.ax.relim()
-            self.ax.autoscale_view()
+            
+            if not self.fix_ylim:
+                self.ax.relim()
+                self.ax.autoscale_view()
 
             if not properties:
                 self.ax.set_xlabel('x')
@@ -256,9 +268,12 @@ def save_one_field(f):
 
         h5.save()
 
-    def plot(self, fig, field):
+    def plot(self, fig, field, properties=None):
         data = self.get_data(field)
-        fig.plot(self.sol.t, self.sol.domain, field, data)
+        fig.plot(
+            self.sol.t, self.sol.domain, field, data,
+            properties=properties
+        )
 
     def save_config(self, filename='simu_config.json'):
         from .json import save_simu_config

pylbm_ui/widgets/simulation.py

@@ -77,6 +77,16 @@ def __init__(self, test_case_widget, lb_scheme_widget):
             list(lb_scheme_widget.get_case().equation.get_fields().keys())
         )
 
+        self.fix_axis = v.Switch(label='Fix axis', v_model=False)
+        self.fix_axis_ymin = v.TextField(
+            label='ymin', v_model='', class_="d-none"
+        )
+        self.fix_axis_ymax = v.TextField(
+            label='ymax', v_model='', class_="d-none"
+        )
+        # self.fix_axis_ymin.hide()
+        # self.fix_axis_ymax.hide()
+
         self.menu = [
             self.simulation_name,
             self.simu_cfg,
@@ -92,6 +102,13 @@ def __init__(self, test_case_widget, lb_scheme_widget):
                         self.save_fields.widget
                     ]),
                 ]),
+                v.ExpansionPanel(children=[
+                    v.ExpansionPanelHeader(children=['Graphic options']),
+                    v.ExpansionPanelContent(children=[
+                        self.fix_axis,
+                        self.fix_axis_ymin, self.fix_axis_ymax
+                    ]),
+                ]),
             ], multiple=True, class_='pa-0')
         ]
 
@@ -146,6 +163,7 @@ def __init__(self, test_case_widget, lb_scheme_widget):
         self.result.observe(self.replot, 'v_model')
         self.simu_cfg.observe(self.load_simu_cfg, 'v_model')
         self.discret['nt'].observe(self.change_period_by_nt, 'v_model')
+        self.fix_axis.observe(self.on_fix_axis_click, 'v_model')
 
         test_case_widget.select_case.observe(self.stop_simulation, 'v_model')
         lb_scheme_widget.select_case.observe(self.stop_simulation, 'v_model')
@@ -247,9 +265,20 @@ async def run_simu(self):
         self.plot = Plot()
         self.iplot = 0
         self.plot_output.children = [self.plot.fig.canvas]
+        if self.fix_axis.v_model:
+            ymin = self.fix_axis_ymin.v_model
+            ymax = self.fix_axis_ymax.v_model
+            self.plot_options = {
+                'set_ylim': [ymin, ymax]
+            }
+        else:
+            self.plot_options = None
 
         self.simu.save_data(self.result.v_model)
-        self.simu.plot(self.plot, self.result.v_model)
+        self.simu.plot(
+            self.plot, self.result.v_model,
+            properties=self.plot_options
+        )
         self.plot_output.children[0].draw_idle()
 
         ite_to_save = self.save_fields.get_save_time(
@@ -271,7 +300,10 @@ async def run_simu(self):
                     if nite > self.period.value:
                         nite = 1
                         self.simu.save_data(self.result.v_model)
-                        self.simu.plot(self.plot, self.result.v_model)
+                        self.simu.plot(
+                            self.plot, self.result.v_model,
+                            properties=self.plot_options
+                        )
                         self.plot_output.children[0].draw_idle()
                         ###### await asyncio.sleep(0.2)
 
@@ -333,6 +365,17 @@ def on_pause_click(self, widget, event, data):
         """
         self.pause.v_model = not self.pause.v_model
 
+    def on_fix_axis_click(self, change):
+        """
+        fix or not the ylim axis when click
+        """
+        if self.fix_axis.v_model:
+            self.fix_axis_ymin.show()
+            self.fix_axis_ymax.show()
+        else:
+            self.fix_axis_ymin.hide()
+            self.fix_axis_ymax.hide()
+
     def replot(self, change):
         """
         Update the plot.