Merge pull request #4 from EdelweissFE/tom.abaqusinputgenerator

Tom.abaqusinputgenerator

Author: matthiasneuner

edelweissmpm/constraints/particlelagrangianweakdirichlet.py

@@ -59,7 +59,7 @@ def _getConstraintLocation():
             elif location == "vertex":
                 if vertexID is None:
                     raise ValueError("vertexID must be specified when location is 'vertex'.")
-                constraintLocation = particle.getVertexCoordinates(vertexID)
+                constraintLocation = particle.getVertexCoordinates()[vertexID]
             return constraintLocation
 
         self._getConstraintLocation = _getConstraintLocation
@@ -176,14 +176,28 @@ def ParticleLagrangianWeakDirichletOnParticleSetFactory(
     model: MPMModel,
     location: str = "center",
     faceID: int = None,
-    vertexID: int = None,
+    vertexID: int | list[int] = None,
 ):
     constraints = dict()
     for i, p in enumerate(particleSet):
-        name = f"{baseName}_{i}"
-        constraint = ParticleLagrangianWeakDirichlet(
-            name, p, field, prescribedStepDelta, model, location, faceID, vertexID
-        )
-        constraints[name] = constraint
-
+        if location == "vertex":
+            if isinstance(vertexID, list):
+                for vertIdx in vertexID:
+                    name = f"{baseName}_{i}_v{vertIdx}"
+                    constraint = ParticleLagrangianWeakDirichlet(
+                        name, p, field, prescribedStepDelta, model, location, faceID, vertIdx
+                    )
+                    constraints[name] = constraint
+            else:
+                name = f"{baseName}_{i}_v{vertexID}"
+                constraint = ParticleLagrangianWeakDirichlet(
+                    name, p, field, prescribedStepDelta, model, location, faceID, vertexID
+                )
+                constraints[name] = constraint
+        else:
+            name = f"{baseName}_{i}"
+            constraint = ParticleLagrangianWeakDirichlet(
+                name, p, field, prescribedStepDelta, model, location, faceID, vertexID
+            )
+            constraints[name] = constraint
     return constraints