Fem: Update CalculiX writer of the constraint rigid body

Author: marioalexis84

src/Mod/Fem/femsolver/calculix/write_constraint_rigidbody.py

@@ -55,43 +55,24 @@ def get_after_write_constraint():
 
 
 def write_meshdata_constraint(f, femobj, rb_obj, ccxwriter):
+
     f.write("*NSET,NSET=" + rb_obj.Name + "\n")
     for n in femobj["Nodes"]:
         f.write("{},\n".format(n))
 
 
 def write_constraint(f, femobj, rb_obj, ccxwriter):
 
-    # floats read from ccx should use {:.13G}, see comment in writer module
-    is_ref_bc_defined = any((rb_obj.xDisplacement,
-                             rb_obj.yDisplacement,
-                             rb_obj.zDisplacement,
-                             rb_obj.xForce,
-                             rb_obj.yForce,
-                             rb_obj.zForce))
-
-    is_rot_bc_defined = any((rb_obj.xRotation,
-                             rb_obj.yRotation,
-                             rb_obj.zRotation,
-                             rb_obj.xMoment,
-                             rb_obj.yMoment,
-                             rb_obj.zMoment))
-
-    # FIXME: This needs to be implemented
-    ref_node_idx = 10000000
-    rot_node_idx = 20000000
+    rb_obj_idx = ccxwriter.analysis.Group.index(rb_obj)
+    node_count = ccxwriter.mesh_object.FemMesh.NodeCount
+    ref_node_idx = node_count + rb_obj_idx + 1
+    rot_node_idx = node_count + rb_obj_idx + 2
 
     f.write("*NODE\n")
-    f.write("{},{},{},{}\n".format(ref_node_idx, rb_obj.xRefNode, rb_obj.yRefNode, rb_obj.zRefNode))
-    f.write("{},{},{},{}\n".format(rot_node_idx, rb_obj.xRefNode, rb_obj.yRefNode, rb_obj.zRefNode))
+    f.write("{},{},{},{}\n".format(ref_node_idx, *rb_obj.ReferenceNode))
+    f.write("{},{},{},{}\n".format(rot_node_idx, *rb_obj.ReferenceNode))
 
     kw_line = "*RIGID BODY, NSET={}, REF NODE={}, ROT NODE={}".format(rb_obj.Name, ref_node_idx, rot_node_idx)
 
-    if is_ref_bc_defined:
-        kw_line = kw_line + ",REF NODE={}".format(ref_node_idx)
-
-    if is_rot_bc_defined:
-        kw_line = kw_line + ",ROT NODE={}".format(rot_node_idx)
-
     f.write(kw_line + "\n")
 

src/Mod/Fem/femsolver/calculix/write_constraint_rigidbody_step.py

@@ -26,6 +26,9 @@
 __url__ = "https://www.freecadweb.org"
 
 
+import FreeCAD
+
+
 def get_analysis_types():
     return "all"    # write for all analysis types
 
@@ -56,33 +59,41 @@ def get_after_write_constraint():
 
 def write_constraint(f, femobj, rb_obj, ccxwriter):
 
-    # floats read from ccx should use {:.13G}, see comment in writer module
-    is_ref_bc_defined = any((rb_obj.xDisplacement,
-                             rb_obj.yDisplacement,
-                             rb_obj.zDisplacement,
-                             rb_obj.xForce,
-                             rb_obj.yForce,
-                             rb_obj.zForce))
-
-    is_rot_bc_defined = any((rb_obj.xRotation,
-                             rb_obj.yRotation,
-                             rb_obj.zRotation,
-                             rb_obj.xMoment,
-                             rb_obj.yMoment,
-                             rb_obj.zMoment))
-
-    # FIXME: This needs to be implemented
-    ref_node_idx = 10000000
-    rot_node_idx = 20000000
-
-
-    # TODO: Displacement definitions need fixing
-    f.write("*CLOAD\n")
-    f.write("{},1,{}\n".format(ref_node_idx, rb_obj.xForce))
-    f.write("{},2,{}\n".format(ref_node_idx, rb_obj.yForce))
-    f.write("{},3,{}\n".format(ref_node_idx, rb_obj.zForce))
-
-    f.write("*CLOAD\n")
-    f.write("{},1,{}\n".format(rot_node_idx, rb_obj.xMoment))
-    f.write("{},2,{}\n".format(rot_node_idx, rb_obj.yMoment))
-    f.write("{},3,{}\n".format(rot_node_idx, rb_obj.zMoment))
+    rb_obj_idx = ccxwriter.analysis.Group.index(rb_obj)
+    node_count = ccxwriter.mesh_object.FemMesh.NodeCount
+    ref_node_idx = node_count + rb_obj_idx + 1
+    rot_node_idx = node_count + rb_obj_idx + 2
+
+    def write_mode(mode, node, dof, constraint, load):
+        if mode == "Constraint":
+            f.write("*BOUNDARY\n")
+            f.write("{},{},{},{:.13G}\n".format(node, dof, dof, constraint))
+        elif mode == "Load":
+            f.write("*CLOAD\n")
+            f.write("{},{},{:.13G}\n".format(node, dof, load))
+
+    mode = [rb_obj.TranslationalModeX, rb_obj.TranslationalModeY, rb_obj.TranslationalModeZ]
+    constraint = rb_obj.Displacement
+    load = [rb_obj.ForceX, rb_obj.ForceY, rb_obj.ForceZ]
+
+    for i in range(3):
+        write_mode(mode[i], ref_node_idx, i + 1, constraint[i], load[i].getValueAs("N").Value)
+
+
+    mode = [rb_obj.RotationalModeX, rb_obj.RotationalModeY, rb_obj.RotationalModeZ]
+    load = [rb_obj.MomentX,rb_obj.MomentY, rb_obj.MomentZ]
+
+    # write rotation components according to rotational mode
+    rot = rb_obj.Rotation
+    proj_axis = [rot.Axis[i] if mode[i] == "Constraint" else 0 for i in range(3)]
+    # proj_axis could be null
+    try:
+        constraint = FreeCAD.Vector(proj_axis).normalize() * rot.Angle
+    except:
+        constraint = FreeCAD.Vector(0, 0, 0)
+
+    for i in range(3):
+        write_mode(mode[i], rot_node_idx, i + 1, constraint[i], load[i].getValueAs("N*mm").Value)
+
+
+    f.write("\n")