From 2304053625a8f765ef2ed746bcb3d3e72bedf51a Mon Sep 17 00:00:00 2001
From: Maximilian Gruber <maximilian.gruber@ptb.de>
Date: Fri, 31 May 2024 15:18:22 +0200
Subject: [PATCH] enhance result plot

---
 app/cocal_methods.py | 30 ++++++++++++++++++++----------
 1 file changed, 20 insertions(+), 10 deletions(-)

diff --git a/app/cocal_methods.py b/app/cocal_methods.py
index 3a5e737..3c5575b 100644
--- a/app/cocal_methods.py
+++ b/app/cocal_methods.py
@@ -444,7 +444,19 @@ class CocalMethods:
         # )
 
         # visualize result
-        fig, ax = plt.subplots(1, 1, sharex=True, squeeze=False)
+        fig, ax = plt.subplots(nrows=1, ncols=1, sharex=True, squeeze=False, figsize=(8,5))
+        fig.suptitle("Transfer behavior in frequency domain")
+
+        # empirical TF
+        ax[0, 0].scatter(
+            self.ref_frequency[mask],
+            np.abs(h_empirical[mask]),
+            label="empirical TF",
+            s=2,
+            color="black",
+        )
+
+        # fitted TF
         ax[0, 0].plot(
             f[mask],
             np.abs(h[mask]),
@@ -459,13 +471,8 @@ class CocalMethods:
             label="unc of fitted TF",
             color="blue",
         )
-        ax[0, 0].scatter(
-            self.ref_frequency[mask],
-            np.abs(h_empirical[mask]),
-            label="empirical TF",
-            s=2,
-            color="black",
-        )
+
+        # compensated TF
         ax[0, 0].scatter(
             self.ref_frequency[mask],
             np.abs(h_comp[mask]),
@@ -481,12 +488,15 @@ class CocalMethods:
             label="unc of compensated TF",
             color="red",
         )
-
+        
+        # ideal
         ax[0, 0].plot(f, np.ones_like(f), "--r", label="ideal")
+
         ax[0, 0].legend()
         ax[0, 0].set_xscale("log")
         ax[0, 0].set_yscale("log")
-        plt.savefig(self.result_image_path)
+        fig.tight_layout()
+        fig.savefig(self.result_image_path)
         # plt.show()
 
     def perform_dummy_computations(self):
-- 
GitLab