pyjallib 0.1.8__py3-none-any.whl → 0.1.10__py3-none-any.whl

pyjallib/max/hip.py

@@ -0,0 +1,239 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+Hip 모듈 - 3ds Max용 Hip 관련 기능 제공
+원본 MAXScript의 hip.ms를 Python으로 변환하였으며, pymxs 모듈 기반으로 구현됨
+"""
+
+from pymxs import runtime as rt
+
+# Import necessary service classes for default initialization
+from .name import Name
+from .anim import Anim
+from .helper import Helper
+from .bone import Bone
+from .constraint import Constraint
+
+
+class Hip:
+    """
+    Hip 관련 기능을 제공하는 클래스.
+    MAXScript의 _Hip 구조체 개념을 Python으로 재구현한 클래스이며,
+    3ds Max의 기능들을 pymxs API를 통해 제어합니다.
+    """
+    
+    def __init__(self, nameService=None, animService=None, helperService=None, boneService=None, constraintService=None):
+        """
+        클래스 초기화.
+        
+        Args:
+            nameService: 이름 처리 서비스 (제공되지 않으면 새로 생성)
+            animService: 애니메이션 서비스 (제공되지 않으면 새로 생성)
+            helperService: 헬퍼 객체 관련 서비스 (제공되지 않으면 새로 생성)
+            boneService: 뼈대 관련 서비스 (제공되지 않으면 새로 생성)
+            constraintService: 제약 관련 서비스 (제공되지 않으면 새로 생성)
+            bipService: Biped 관련 서비스 (제공되지 않으면 새로 생성)
+        """
+        # 서비스 인스턴스 설정 또는 생성
+        self.name = nameService if nameService else Name()
+        self.anim = animService if animService else Anim()
+        
+        # 종속성이 있는 서비스들은 이미 생성된 서비스들을 전달
+        self.helper = helperService if helperService else Helper(nameService=self.name)
+        self.const = constraintService if constraintService else Constraint(nameService=self.name, helperService=self.helper)
+        self.bone = boneService if boneService else Bone(nameService=self.name, animService=self.anim, helperService=self.helper, constraintService=self.const)
+        
+        # 기본 속성 초기화
+        self.pelvisWeight = 0.4
+        self.thighWeight = 0.6
+        self.pushAmount = 20
+        
+        self.pelvis = None
+        self.thigh = None
+        self.thighTwist = None
+        self.calf = None
+        
+        self.pelvisHelper = None
+        self.thighHelper = None
+        self.thighTwistHelper = None
+        self.thighRotHelper = None
+        self.thighPosHelper = None
+        self.thighRotRootHelper = None
+        
+        self.helpers = []
+        self.bones = []
+        
+        self.posScriptExpression = (
+            "localLimbTm = limb.transform * inverse limbParent.transform\n"
+            "localDeltaTm = localLimbTm * inverse localRotRefTm\n"
+            "\n"
+            "q = localDeltaTm.rotation\n"
+            "\n"
+            "eulerRot = (quatToEuler q order:5)\n"
+            "swizzledRot = (eulerAngles eulerRot.y eulerRot.z eulerRot.x)\n"
+            "\n"
+            "axis = [0,0,1]\n"
+            "\n"
+            "saturatedTwistZ = (swizzledRot.x*axis.x + swizzledRot.y*axis.y + swizzledRot.z*axis.z)/180.0\n"
+            "pushScaleY = (amax 0.0 saturatedTwistZ) * 0.5\n"
+            "\n"
+            "axis = [0,1,0]\n"
+            "saturatedTwistY = (swizzledRot.x*axis.x + swizzledRot.y*axis.y + swizzledRot.z*axis.z)/180.0\n"
+            "pushScaleZ = amax 0.0 saturatedTwistY\n"
+            "\n"
+            "\n"
+            "[0, pushAmount * pushScaleY, -pushAmount * pushScaleZ]\n"
+        )
+        
+    def reset(self):
+        """
+        클래스의 주요 컴포넌트들을 초기화합니다.
+        서비스가 아닌 클래스 자체의 작업 데이터를 초기화하는 함수입니다.
+        
+        Returns:
+            self: 메소드 체이닝을 위한 자기 자신 반환
+        """
+        self.pelvisWeight = 0.4
+        self.thighWeight = 0.6
+        self.pushAmount = 20
+        
+        self.pelvis = None
+        self.thigh = None
+        self.thighTwist = None
+        self.calf = None
+        
+        self.pelvisHelper = None
+        self.thighHelper = None
+        self.thighTwistHelper = None
+        self.thighRotHelper = None
+        self.thighPosHelper = None
+        self.thighRotRootHelper = None
+        
+        self.helpers = []
+        self.bones = []
+        
+        return self
+    
+    def create_helper(self, inPelvis, inThigh, inThighTwist):
+        if not rt.isValidNode(inPelvis) or not rt.isValidNode(inThigh) or not rt.isValidNode(inThighTwist):
+            return False
+        
+        self.pelvis = inPelvis
+        self.thigh = inThigh
+        self.thighTwist = inThighTwist
+        
+        filteringChar = self.name._get_filtering_char(inThigh.name)
+        isLower = inThigh.name[0].islower()
+        
+        pelvisHelperName = self.name.replace_name_part("RealName", inThigh.name, self.name.get_RealName(inPelvis.name)+filteringChar+"Hip")
+        pelvisHelperName = self.name.replace_name_part("Type", pelvisHelperName, self.name.get_name_part_value_by_description("Type", "Dummy"))
+        pelvisHelper = self.helper.create_point(pelvisHelperName)
+        rt.setProperty(pelvisHelper, "transform", inThigh.transform)
+        pelvisHelper.parent = inPelvis
+        
+        tihgTwistHeleprName = self.name.replace_name_part("RealName", inThigh.name, self.name.get_RealName(inThighTwist.name)+filteringChar+"Hip")
+        tihgTwistHeleprName = self.name.replace_name_part("Type", tihgTwistHeleprName, self.name.get_name_part_value_by_description("Type", "Dummy"))
+        thighTwistHelper = self.helper.create_point(tihgTwistHeleprName)
+        rt.setProperty(thighTwistHelper, "transform", inThighTwist.transform)
+        thighTwistHelper.parent = inThighTwist
+        
+        tihghRotHelperName = self.name.replace_name_part("RealName", inThigh.name, self.name.get_RealName(inThigh.name)+filteringChar+"Hip")
+        tihghRotHelperName = self.name.replace_name_part("Type", tihghRotHelperName, self.name.get_name_part_value_by_description("Type", "Rotation"))
+        thighRotHelper = self.helper.create_point(tihghRotHelperName)
+        rt.setProperty(thighRotHelper, "transform", inThighTwist.transform)
+        thighRotHelper.parent = inThigh
+        
+        thighPosHelperName = self.name.replace_name_part("RealName", inThigh.name, self.name.get_RealName(inThigh.name)+filteringChar+"Hip")
+        thighPosHelperName = self.name.replace_name_part("Type", thighPosHelperName, self.name.get_name_part_value_by_description("Type", "Position"))
+        thighPosHelper = self.helper.create_point(thighPosHelperName)
+        rt.setProperty(thighPosHelper, "transform", inThighTwist.transform)
+        thighPosHelper.parent = thighRotHelper
+        
+        thighRotRootHelperName = self.name.replace_name_part("RealName", inThigh.name, self.name.get_RealName(inThigh.name)+filteringChar+"Hip")
+        thighRotRootHelperName = self.name.replace_name_part("Type", thighRotRootHelperName, self.name.get_name_part_value_by_description("Type", "Dummy"))
+        thighRotRootHelper = self.helper.create_point(thighRotRootHelperName)
+        rt.setProperty(thighRotRootHelper, "transform", thighRotHelper.transform)
+        thighRotRootHelper.parent = inThighTwist
+        
+        if isLower:
+            pelvisHelper.name = pelvisHelper.name.lower()
+            thighTwistHelper.name = thighTwistHelper.name.lower()
+            thighRotHelper.name = thighRotHelper.name.lower()
+            thighPosHelper.name = thighPosHelper.name.lower()
+            thighRotRootHelper.name = thighRotRootHelper.name.lower()
+            
+        self.pelvisHelper = pelvisHelper
+        self.thighTwistHelper = thighTwistHelper
+        self.thighRotHelper = thighRotHelper
+        self.thighPosHelper = thighPosHelper
+        self.thighRotRootHelper = thighRotRootHelper
+        
+        self.helpers.append(pelvisHelper)
+        self.helpers.append(thighTwistHelper)
+        self.helpers.append(thighRotHelper)
+        self.helpers.append(thighPosHelper)
+        self.helpers.append(thighRotRootHelper)
+    
+    def assing_constraint(self, inCalf, inPelvisWeight=0.6, inThighWeight=0.4, inPushAmount=5.0):
+        self.calf = inCalf
+        self.pelvisWeight = inPelvisWeight
+        self.thighWeight = inThighWeight
+        self.pushAmount = rt.Float(inPushAmount)
+        
+        facingDirVec = self.calf.transform.position - self.thigh.transform.position
+        inObjXAxisVec = self.thigh.objectTransform.row1
+        distanceDir = -1.0 if rt.dot(inObjXAxisVec, facingDirVec) > 0 else 1.0
+        
+        rotConst = self.const.assign_rot_const_multi(self.thighRotHelper, [self.pelvisHelper, self.thighTwistHelper])
+        rotConst.setWeight(1, self.pelvisWeight * 100.0)
+        rotConst.setWeight(2, self.thighWeight * 100.0)
+        
+        localRotRefTm = self.thighRotHelper.transform * rt.inverse(self.thighRotRootHelper.transform)
+        posConst = self.const.assign_pos_script_controller(self.thighPosHelper)
+        posConst.addNode("limb", self.thighRotHelper)
+        posConst.addNode("limbParent", self.thighRotRootHelper)
+        posConst.addConstant("localRotRefTm", localRotRefTm)
+        posConst.addConstant("pushAmount", self.pushAmount*distanceDir)
+        posConst.setExpression(self.posScriptExpression)
+        posConst.update()
+        
+    def create_bone(self, inPelvis, inThigh, inThighTwist, inCalf, pushAmount=5.0, inPelvisWeight=0.6, inThighWeight=0.4):
+        if not rt.isValidNode(inPelvis) or not rt.isValidNode(inThigh) or not rt.isValidNode(inThighTwist):
+            return False
+        
+        self.create_helper(inPelvis, inThigh, inThighTwist)
+        self.assing_constraint(inCalf, inPelvisWeight, inThighWeight, inPushAmount=pushAmount)
+        
+        isLower = inThigh.name[0].islower()
+        hipBoneName = self.name.replace_name_part("RealName", inThigh.name, "Hip")
+        hipBone = self.bone.create_nub_bone(hipBoneName, 2)
+        hipBone.name = self.name.remove_name_part("Nub", hipBone.name)
+        if isLower:
+            hipBone.name = hipBone.name.lower()
+        
+        rt.setProperty(hipBone, "transform", inThighTwist.transform)
+        hipBone.parent = inThigh
+        
+        self.const.assign_rot_const(hipBone, self.thighRotHelper)
+        self.const.assign_pos_const(hipBone, self.thighPosHelper)
+        
+        self.bones.append(hipBone)
+        
+        # 결과를 딕셔너리 형태로 준비
+        result = {
+            "Pelvis": inPelvis,
+            "Thigh": inThigh,
+            "ThighTwist": inThighTwist,
+            "Bones": self.bones,
+            "Helpers": self.helpers,
+            "PelvisWeight": inPelvisWeight,
+            "ThighWeight": inThighWeight,
+            "PushAmount": pushAmount
+        }
+        
+        # 메소드 호출 후 데이터 초기화
+        self.reset()
+        
+        return result
+