Samples\Math\FreezeLocalRotation.py

# Copyright 2023 Autodesk, Inc.  All rights reserved.
# Use of this software is subject to the terms of the Autodesk license agreement 
# provided at the time of installation or download, or which otherwise accompanies
# this software in either electronic or hard copy form.
#
# Script description:
# Show how to transfer, in a hierarchy of selected objects, all their local rotation properties to their pre-rotation properties
# and then zero-out their local rotation properties. This can be seen as "freezing" the location rotation properties of the selected objects.
# Tested only with the Rotation Order being Euler XYZ and with limited scenarios.
#
# Topic: operators and utility functions for FBVector3d, FBMatrix, 
# FBModelList, FBModelTransformationType, FBGetSelectedModels, FBMatrixMult.
#
 
 
from pyfbsdk import FBVector3d, FBMatrix, FBModelList, FBModelTransformationType, FBGetSelectedModels, FBMatrixMult
 
 
def freeze_local_rotation(model):
    model.PropertyList.Find("Enable Rotation DOF").Data = True   
    
    result_vector = FBVector3d()
    
    if not model.Parent:
        gRot = FBVector3d()
        model.GetVector(gRot, FBModelTransformationType.kModelRotation, True)
    
        result_vector = gRot
    else:
        prerot = model.PreRotation
        
        model_DOF2LRM = FBMatrix()
        model.DofToLRM(model_DOF2LRM, prerot)
        
        model_globalMatrix = FBMatrix()
        model.GetMatrix(model_globalMatrix, FBModelTransformationType.kModelRotation, False)
        
        model_multMatrix = FBMatrix()
        FBMatrixMult(model_multMatrix, model_DOF2LRM, model_globalMatrix)
        
        model.LRMToDof(result_vector, model_multMatrix)
        
    model.PropertyList.Find("Pre Rotation").Data = result_vector
    model.Rotation = FBVector3d()
           
 
modelList = FBModelList()
FBGetSelectedModels(modelList)
for model in modelList:
    freeze_local_rotation(model)