changed acceleration_collection and alpha_collection to be more consi…

…stent with velocity_collection and omega_collection
Ali-7800 · Apr 24, 2024 · 71e19c6 · 71e19c6
1 parent 665f55a
commit 71e19c6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/elastica/rigidbody/cylinder.py b/elastica/rigidbody/cylinder.py
@@ -64,8 +64,8 @@ def __init__(self, start, direction, normal, base_length, base_radius, density):
 
         self.velocity_collection = np.zeros((MaxDimension.value(), 1))
         self.omega_collection = np.zeros((MaxDimension.value(), 1))
-        self.acceleration_collection = 0.0 * self.velocity_collection
-        self.alpha_collection = 0.0 * self.omega_collection
+        self.acceleration_collection = np.zeros((MaxDimension.value(), 1))
+        self.alpha_collection = np.zeros((MaxDimension.value(), 1))
 
         self.director_collection = np.zeros(
             (MaxDimension.value(), MaxDimension.value(), 1)

diff --git a/elastica/rigidbody/sphere.py b/elastica/rigidbody/sphere.py
@@ -54,8 +54,8 @@ def __init__(self, center, base_radius, density):
 
         self.velocity_collection = np.zeros((MaxDimension.value(), 1))
         self.omega_collection = np.zeros((MaxDimension.value(), 1))
-        self.acceleration_collection = 0.0 * self.velocity_collection
-        self.alpha_collection = 0.0 * self.omega_collection
+        self.acceleration_collection = np.zeros((MaxDimension.value(), 1))
+        self.alpha_collection = np.zeros((MaxDimension.value(), 1))
 
         self.director_collection = np.zeros(
             (MaxDimension.value(), MaxDimension.value(), 1)