[MeshMatrixMass] Fix init from density

sofa-framework · Jun 18, 2021 · ea36995 · ea36995
1 parent ce505bf
commit ea36995
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/modules/SofaMiscForceField/src/SofaMiscForceField/MeshMatrixMass.inl b/modules/SofaMiscForceField/src/SofaMiscForceField/MeshMatrixMass.inl
@@ -1749,6 +1749,14 @@ void MeshMatrixMass<DataTypes, MassType>::initFromMassDensity()
     {
         sumMass += vertexMassInfo[i]*m_massLumpingCoeff;
     }
+
+    // Same for edgeMass
+    helper::WriteAccessor<Data<MassVector> > edgeMass = d_edgeMass;
+    for (size_t i = 0; i < edgeMass.size(); i++)
+    {
+        sumMass += edgeMass[i] * 2;
+    }
+
     d_totalMass.setValue(sumMass);
 }