From 698075105886658abdcc38a2eee8f921d9f90e03 Mon Sep 17 00:00:00 2001
From: Romain Boman <romain.boman@gmail.com>
Date: Fri, 17 May 2024 13:23:30 +0200
Subject: [PATCH] fix extraction when fixation are named with custom strings
 (as in feito)

---
 models/bonemodel2.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/models/bonemodel2.py b/models/bonemodel2.py
index 1bc8373..417ae52 100644
--- a/models/bonemodel2.py
+++ b/models/bonemodel2.py
@@ -189,7 +189,8 @@ def solve(p={}):
 
     print('extracting results...')
     print(f'\tinternal energy = {solver.int_energy:.3f} N.mm')
-    for grpname in ['axis_pt1', 'axis_pt2', 'contact_pts']:
+    for fix in p['fixations']:
+        grpname = fix['name']
         v = np.array(post.probe('force_vector', grpname))
         v = np.reshape(v, (v.size//3, 3))
         v = np.sum(v, axis=0)
-- 
GitLab