From f5e397c107c046dbc5a12094035d53714fc11d75 Mon Sep 17 00:00:00 2001
From: Gavin Macaulay <gavin@macaulay.co.nz>
Date: Mon, 2 Dec 2024 11:22:37 +1300
Subject: [PATCH] Add example of using high-pass model

---
 src/example_code.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/src/example_code.py b/src/example_code.py
index 913993a..ba81b79 100644
--- a/src/example_code.py
+++ b/src/example_code.py
@@ -6,6 +6,7 @@
 import trimesh
 
 from echosms import MSSModel, PSMSModel, DCMModel, ESModel, PTDWBAModel, KAModel, DWBAModel
+from echosms import HPModel
 from echosms import BenchmarkData
 from echosms import ReferenceModels
 from echosms import as_dataframe, as_dataarray
@@ -272,6 +273,26 @@ def plot_compare_angle(theta1, ts1, label1, theta2, ts2, label2, title):
 plt.title('WC12 calibration sphere')
 plt.show()
 
+# %% ###############################################################################################
+# Try the high-pass model
+mod = HPModel()
+p = {'boundary_type': 'fixed rigid', 'shape': 'sphere', 'medium_c': 1500, 'a': 0.01,
+     'f': np.arange(1, 400, 1)*1e3}
+fixed_rigid = mod.calculate_ts(p)
+p['boundary_type'] = 'elastic'
+p |= {'medium_rho': 1024, 'target_c': 1600, 'target_rho': 1600}
+elastic = mod.calculate_ts(p)
+p['boundary_type'] = 'fluid filled'
+p |= {'medium_rho': 1024, 'target_c': 1510, 'target_rho': 1025}
+fluid = mod.calculate_ts(p)
+
+plt.semilogx(p['f']/1e3, fixed_rigid, label='fixed rigid')
+plt.semilogx(p['f']/1e3, elastic, label='elastic')
+plt.semilogx(p['f']/1e3, fluid, label='fluid filled')
+plt.legend()
+plt.xlabel('Frequency [kHz]')
+plt.ylabel('TS re 1m$2$ [dB]')
+
 # %% ###############################################################################################
 # Some other ways to run models.