From ee4f64b776f833f94a86f76359bb6f47700ab32b Mon Sep 17 00:00:00 2001 From: Saurabh Mogre Date: Tue, 18 Jul 2023 14:24:26 -0700 Subject: [PATCH] * use absolute value for distances * use normalization function for weights --- cellpack/autopack/Gradient.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/cellpack/autopack/Gradient.py b/cellpack/autopack/Gradient.py index a74ed878e..4ef47cb44 100644 --- a/cellpack/autopack/Gradient.py +++ b/cellpack/autopack/Gradient.py @@ -191,7 +191,7 @@ def build_directional_weight_map(self, bb, master_grid_positions): direction = self.normalize_vector(direction) self.weight = [] center = self.mode_settings.get("center", self.get_center()) - self.distances = numpy.dot(master_grid_positions - center, direction) + self.distances = numpy.abs(numpy.dot(master_grid_positions - center, direction)) self.set_weights_by_mode() def build_axis_weight_map(self, bb, master_grid_positions): @@ -232,9 +232,7 @@ def set_weights_by_mode(self): -self.scaled_distances / self.weight_mode_settings["decay_length"] ) # normalize the weight - self.weight = (self.weight - numpy.min(self.weight)) / ( - numpy.max(self.weight) - numpy.min(self.weight) - ) + self.weight = self.get_normalized_values(self.weight) self.weight[numpy.isnan(self.weight)] = 0