apply_pivot_offset_to_translation added. Store and apply world space updated #501

Author: miquelcampos

release/scripts/mgear/core/transform.py

@@ -1083,6 +1083,29 @@ def rotate_180(axis="Y", rotation_order="XYZ", objects=None):
 
 
 # Wolrd Transform Functions: Transform data IO with pivot offset
+def apply_pivot_offset_to_translation(obj):
+    """Add the local rotate pivot to the object's current translation.
+
+    Args:
+        obj (str): Name of the object.
+
+    Raises:
+        RuntimeError: If the object does not exist.
+    """
+    if not cmds.objExists(obj):
+        raise RuntimeError("Object '{}' does not exist.".format(obj))
+
+    # Check if any component of .translate is locked or connected
+    for axis in ["X", "Y", "Z"]:
+        attr = "{}.translate{}".format(obj, axis)
+        if cmds.getAttr(attr, lock=True):
+            return
+
+    translation = cmds.getAttr(obj + ".translate")[0]
+    rotate_pivot = cmds.getAttr(obj + ".rotatePivot")[0]
+
+    new_translation = [t - rp for t, rp in zip(translation, rotate_pivot)]
+    cmds.setAttr(obj + ".translate", *new_translation, type="double3")
 
 def get_world_transform_data(obj):
     """Get world transform matrix and pivot of a given object.
@@ -1099,9 +1122,14 @@ def get_world_transform_data(obj):
     if not cmds.objExists(obj):
         raise RuntimeError("Object '{}' does not exist.".format(obj))
 
-    matrix = cmds.xform(obj, q=True, ws=True, m=True)
-    pivot = cmds.xform(obj, q=True, ws=True, rotatePivot=True)
-    return {'matrix': matrix, 'rotatePivot': pivot}
+    transform_data = {
+        'matrix': cmds.xform(obj, q=True, ws=True, m=True),
+        "translation": cmds.xform(obj, q=True, ws=True, t=True),
+        "rotation": cmds.xform(obj, q=True, ws=True, rotation=True),
+        "scale": cmds.xform(obj, q=True, ws=True, s=True),
+        "rotatePivot": cmds.xform(obj, q=True, ws=True, rotatePivot=True),
+    }
+    return transform_data
 
 
 def set_world_transform_data(obj, transform_data):
@@ -1117,14 +1145,31 @@ def set_world_transform_data(obj, transform_data):
     if not cmds.objExists(obj):
         raise RuntimeError("Object '{}' does not exist.".format(obj))
 
-    matrix = transform_data.get('matrix')
-    pivot = transform_data.get('rotatePivot')
-
-    if matrix:
-        cmds.xform(obj, ws=True, m=matrix)
-    if pivot:
-        # restore both rotate- and scale-pivot in world space
-        cmds.xform(obj, ws=True, rotatePivot=pivot)
-        cmds.xform(obj, ws=True, scalePivot=pivot)
+    # matrix = transform_data.get('matrix')
+    # pivot = transform_data.get('rotatePivot')
+
+    if "matrix" in transform_data:
+        cmds.xform(obj, ws=True, m=transform_data["matrix"])
+        # reset rotatePivotTranslate to neutral value
+        cmds.setAttr(obj + ".rotatePivotTranslate", *[0,0,0], type="double3")
+    # re-apply the final position in world space
+    if "translation" in transform_data:
+        cmds.xform(
+            obj,
+            ws=True,
+            t=vector.subtract_3Dvectors_list(
+                transform_data["translation"], transform_data["rotatePivot"]
+            ),
+        )
+        v1 = vector.subtract_3Dvectors_list(
+            transform_data["rotatePivot"], transform_data["translation"]
+        )
+        v2 = vector.add_3Dvectors_list(transform_data["translation"], v1)
+        cmds.xform(obj, ws=True, t=v2)
+        apply_pivot_offset_to_translation(obj)
+    # if pivot:
+    #     # restore both rotate- and scale-pivot in world space
+    #     cmds.xform(obj, ws=True, rotatePivot=pivot)
+    #     cmds.xform(obj, ws=True, scalePivot=pivot)