Update visualization.py

visualization.py

@@ -56,14 +56,15 @@ def plotly_full_figure(
     single_speaker=False,
 ):
     fig = make_subplots(
-        rows=3,
+        rows=4,
         cols=1,
-        shared_xaxes=True,
-        vertical_spacing=0.05,
+        shared_xaxes=False,
+        vertical_spacing=0.1,
         specs=[
             [{"secondary_y": True}],
             [{"secondary_y": True}],
             [{"secondary_y": True}],
+            [{"secondary_y": True}],
         ],  # Enable secondary y-axis for both rows
     )
 
@@ -166,6 +167,38 @@ def plotly_full_figure(
         col=1,
         secondary_y=True,
     )
+    ### Mechanical displacement
+    fig.add_trace(
+        go.Scatter(
+            x=freq_array,
+            y=np.abs(loudspeaker_responses_1["mechanical_displacement"]),
+            mode="lines",
+            name="Speaker 1",
+            line=dict(color="blue"),
+            hovertemplate="%{x:.2f} Hz<br>%{y:.2f} " + "mm",
+            legendgroup="Speaker 1 (Magnitude)",
+            showlegend=False,
+        ),
+        row=4,
+        col=1,
+        secondary_y=False,
+    )
+
+    fig.add_trace(
+        go.Scatter(
+            x=freq_array,
+            y=np.angle(loudspeaker_responses_1["mechanical_displacement"], deg=True),
+            mode="lines",
+            name="Speaker 1",
+            line=dict(color="red"),
+            hovertemplate="%{x:.2f} Hz<br>%{y:.2f} º",
+            legendgroup="Speaker 1 (Phase)",
+            showlegend=False,
+        ),
+        row=4,
+        col=1,
+        secondary_y=True,
+    )
 
     ###### Speaker 2
     if not single_speaker:
@@ -267,6 +300,40 @@ def plotly_full_figure(
             col=1,
             secondary_y=True,
         )
+        ### Mechanical displacement
+        fig.add_trace(
+            go.Scatter(
+                x=freq_array,
+                y=np.abs(loudspeaker_responses_2["mechanical_displacement"]),
+                mode="lines",
+                name="Speaker 2",
+                line=dict(color="blue", dash="dot"),
+                hovertemplate="%{x:.1f} Hz<br>%{y:.1f} " + "mm",
+                legendgroup="Speaker 2 (Magnitude)",
+                showlegend=False,
+            ),
+            row=4,
+            col=1,
+            secondary_y=False,
+        )
+
+        fig.add_trace(
+            go.Scatter(
+                x=freq_array,
+                y=np.angle(
+                    loudspeaker_responses_2["mechanical_displacement"], deg=True
+                ),
+                mode="lines",
+                name="Speaker 2",
+                line=dict(color="red", dash="dot"),
+                hovertemplate="%{x:.1f} Hz<br>%{y:.1f} º",
+                legendgroup="Speaker 2 (Phase)",
+                showlegend=False,
+            ),
+            row=4,
+            col=1,
+            secondary_y=True,
+        )
 
     ### Update layout
     fig.update_layout(
@@ -282,6 +349,11 @@ def plotly_full_figure(
             range=[np.log10(10), np.log10(4000)],
         ),
         xaxis3=dict(
+            type="log",
+            tickvals=[16, 31, 63, 125, 250, 500, 1000, 2000, 4000],
+            range=[np.log10(10), np.log10(4000)],
+        ),
+        xaxis4=dict(
             title="Frequency [Hz]",
             titlefont=dict(size=14),
             tickfont=dict(size=12),
@@ -313,9 +385,24 @@ def plotly_full_figure(
             ticktext=["-180º", "-90º", "0º", "90º", "180º"],
             range=[-180, 180],
         ),
-        legend=dict(font=dict(size=10), x=0.85, y=0.99),
+        yaxis7=dict(title="Displacement [mm]", color="blue"),
+        yaxis8=dict(
+            title="Phase [º]",
+            color="red",
+            tickvals=[-180, -90, 0, 90, 180],
+            ticktext=["-180º", "-90º", "0º", "90º", "180º"],
+            range=[-180, 180],
+        ),
+        legend=dict(
+            font=dict(size=10),
+            x=0.245,
+            y=1.06,
+            orientation="h",
+            itemwidth=30,
+        ),
+        showlegend=not single_speaker,
         width=1200,
-        height=800,
+        height=1000,
         margin=dict(l=20, r=20, t=30, b=20),
         annotations=[
             dict(
@@ -330,7 +417,7 @@ def plotly_full_figure(
             dict(
                 text="<b>Electrical Impedance</b>",
                 x=0.47,
-                y=0.68,
+                y=0.755,
                 xref="paper",
                 yref="paper",
                 showarrow=False,
@@ -339,7 +426,16 @@ def plotly_full_figure(
             dict(
                 text="<b>Mechanical Velocity</b>",
                 x=0.47,
-                y=0.30,
+                y=0.47,
+                xref="paper",
+                yref="paper",
+                showarrow=False,
+                font=dict(size=18),
+            ),
+            dict(
+                text="<b>Mechanical Displacement</b>",
+                x=0.47,
+                y=0.175,
                 xref="paper",
                 yref="paper",
                 showarrow=False,