From 6d621b3bdf0c7d69163ef19b4d8389d88025a1c9 Mon Sep 17 00:00:00 2001 From: Sgr A* VMT <1611902585@qq.com> Date: Sun, 14 Apr 2024 10:47:25 +0800 Subject: [PATCH] fix --- idm.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/idm.py b/idm.py index 058a315..3214f63 100644 --- a/idm.py +++ b/idm.py @@ -1755,7 +1755,7 @@ class IDMMeshHelper: x += xo y += yo - if math.isinf(d): + if d is None math.isinf(d): if self._is_valid_position(x, y): invalid_samples[0] += 1 return @@ -1864,10 +1864,10 @@ class IDMMeshHelper: return None mask = np.full((self.res_y, self.res_x), True) for r in self.faulty_regions: - r_xmin = int(math.ceil((r.x_min - self.min_x) / self.step_x)) - r_ymin = int(math.ceil((r.y_min - self.min_y) / self.step_y)) - r_xmax = int(math.floor((r.x_max - self.min_x) / self.step_x)) - r_ymax = int(math.floor((r.y_max - self.min_y) / self.step_y)) + r_xmin = max(0,int(math.ceil((r.x_min - self.min_x) / self.step_x))) + r_ymin = max(0,int(math.ceil((r.y_min - self.min_y) / self.step_y))) + r_xmax = min(self.res_x-1,int(math.floor((r.x_max - self.min_x) / self.step_x))) + r_ymax = min(self.res_y-1,int(math.floor((r.y_max - self.min_y) / self.step_y))) for y in range(r_ymin, r_ymax + 1): for x in range(r_xmin, r_xmax + 1): mask[(y, x)] = False