From bdbd82e6722d871b6fa1f1b8919fbfd82215986b Mon Sep 17 00:00:00 2001 From: artistrea Date: Mon, 1 Sep 2025 16:10:42 -0300 Subject: [PATCH] update: uses_beamforming antenna attribute --- sharc/antenna/antenna.py | 1 + sharc/antenna/antenna_beamforming_imt.py | 1 + sharc/simulation.py | 5 +++-- 3 files changed, 5 insertions(+), 2 deletions(-) diff --git a/sharc/antenna/antenna.py b/sharc/antenna/antenna.py index 50cc33524..097c35ed3 100644 --- a/sharc/antenna/antenna.py +++ b/sharc/antenna/antenna.py @@ -21,6 +21,7 @@ class Antenna(ABC): def __init__(self): self.beams_list = [] self.w_vec_list = [] + self.uses_beamforming = False @abstractmethod def calculate_gain(self, *args, **kwargs) -> np.array: diff --git a/sharc/antenna/antenna_beamforming_imt.py b/sharc/antenna/antenna_beamforming_imt.py index d041c249e..a58bbc188 100644 --- a/sharc/antenna/antenna_beamforming_imt.py +++ b/sharc/antenna/antenna_beamforming_imt.py @@ -60,6 +60,7 @@ def __init__( referenced in the x axis """ super().__init__() + self.uses_beamforming = True self.param = par self.subarray = None diff --git a/sharc/simulation.py b/sharc/simulation.py index 120807227..06edf1f33 100644 --- a/sharc/simulation.py +++ b/sharc/simulation.py @@ -473,12 +473,13 @@ def select_ue(self, random_number_gen: np.random.RandomState): Select K UEs randomly from all the UEs linked to one BS as “chosen” UEs. These K “chosen” UEs will be scheduled during this snapshot. """ + if not self.bs.antenna[0].uses_beamforming: + return + if self.wrap_around_enabled: self.bs_to_ue_d_2D, self.bs_to_ue_d_3D, self.bs_to_ue_phi, self.bs_to_ue_theta = \ self.bs.get_dist_angles_wrap_around(self.ue) else: - self.bs_to_ue_d_2D = self.bs.get_distance_to(self.ue) - self.bs_to_ue_d_3D = self.bs.get_3d_distance_to(self.ue) self.bs_to_ue_phi, self.bs_to_ue_theta = self.bs.get_pointing_vector_to( self.ue, )