Merge pull request #17 from chahak13/dvol_strain

Add dvolumetric strain to Particle init

Author: chahak13

diffmpm/particle.py

@@ -66,6 +66,7 @@ def __init__(
             self.f_ext = jnp.zeros_like(self.loc)
             self.traction = jnp.zeros_like(self.loc)
             self.reference_loc = jnp.zeros_like(self.loc)
+            self.dvolumetric_strain = jnp.zeros((self.loc.shape[0], 1))
             self.volumetric_strain_centroid = jnp.zeros((self.loc.shape[0], 1))
         else:
             (
@@ -83,6 +84,7 @@ def __init__(
                 self.f_ext,
                 self.traction,
                 self.reference_loc,
+                self.dvolumetric_strain,
                 self.volumetric_strain_centroid,
             ) = data
         self.initialized = True
@@ -106,6 +108,7 @@ def tree_flatten(self):
             self.f_ext,
             self.traction,
             self.reference_loc,
+            self.dvolumetric_strain,
             self.volumetric_strain_centroid,
         )
         aux_data = (self.material,)