From 25e259f0382b60e2b6533db26478b0ea5b1556f7 Mon Sep 17 00:00:00 2001 From: Christos Falas Date: Sun, 29 Dec 2024 12:22:47 +0000 Subject: [PATCH] fix constant spectrum for all tof --- src/aoa.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/src/aoa.py b/src/aoa.py index 06d78c7..b7a8bd8 100644 --- a/src/aoa.py +++ b/src/aoa.py @@ -79,16 +79,21 @@ class AoA: ) def steering_vector(self, theta: float, tof: float): - omega_t = self.omega_base**tof - phi_theta = self.phi_base ** (1 - np.cos(theta)) - - assert phi_theta.shape == omega_t.shape + omega_t = np.exp(-2j * np.pi * config.DELTA_F * tof) + phi_theta = np.exp( + 2j + * np.pi + * config.CENTRAL_FREQUENCY_HZ + * config.ANTENNA_SPACING + * (1 - np.cos(theta)) + / config.C + ) omega_t = np.expand_dims(omega_t, axis=-1) phi_theta = np.expand_dims(phi_theta, axis=-1) antenna_v = omega_t ** np.arange(self.N_subcarriers // 2) - phis = phi_theta ** np.arange(self.N_rx) + phis = phi_theta ** np.arange(self.N_rx // 2) antenna_v = np.expand_dims(antenna_v, axis=-1) steering = antenna_v * phis return steering.T.reshape(-1)