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

pyjallib/max/autoClavicleChain.py

@@ -0,0 +1,173 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+자동 쇄골 체인(AutoClavicle Chain) 관련 기능을 제공하는 클래스.
+AutoClavicle 클래스가 생성한 자동 쇄골 뼈대들을 관리하고 접근하는 인터페이스를 제공합니다.
+
+Examples:
+    # 자동 쇄골 체인 생성 예시
+    from pyjallib.max import AutoClavicle, AutoClavicleChain
+    from pymxs import runtime as rt
+    
+    # 선택된 쇄골과 상완 객체
+    clavicle = rt.selection[0]
+    upperArm = rt.selection[1]
+    
+    # AutoClavicle 클래스 인스턴스 생성
+    auto_clavicle = AutoClavicle()
+    
+    # 자동 쇄골 뼈대 생성
+    clavicle_bones = auto_clavicle.create_bones(clavicle, upperArm, liftScale=0.8)
+    
+    # 생성된 뼈대로 AutoClavicleChain 인스턴스 생성
+    chain = AutoClavicleChain.from_auto_clavicle_result(
+        {
+            "Bones": clavicle_bones,
+            "Helpers": auto_clavicle.helpers,
+            "Clavicle": clavicle,
+            "UpperArm": upperArm,
+            "LiftScale": 0.8
+        }
+    )
+    
+    # 체인 관리 기능 사용
+    bone = chain.get_bone()
+    
+    # 체인의 모든 뼈대 이름 변경
+    chain.rename_bones(prefix="character_", suffix="_autoClavicle")
+    
+    # 체인의 LiftScale 수정
+    chain.update_lift_scale(0.9)
+    
+    # 필요 없어지면 체인의 모든 뼈대와 헬퍼 삭제
+    # chain.delete_all()
+"""
+
+from pymxs import runtime as rt
+from pyjallib.max.header import get_pyjallibmaxheader
+jal = get_pyjallibmaxheader()
+
+
+class AutoClavicleChain:
+    def __init__(self, inResult):
+        """
+        클래스 초기화.
+        
+        Args:
+            inResult: AutoClavicle 클래스의 생성 결과 (딕셔너리)
+        """
+        self.bones = inResult.get("Bones", [])
+        self.helpers = inResult.get("Helpers", [])
+        self.clavicle = inResult.get("Clavicle", None)
+        self.upperArm = inResult.get("UpperArm", None)
+        self.liftScale = inResult.get("LiftScale", 0.8)
+    
+    def get_bones(self):
+        """
+        체인의 모든 뼈대 가져오기
+        
+        Returns:
+            모든 뼈대 객체의 배열
+        """
+        if self.is_empty():
+            return []
+        
+        return self.bones
+    
+    def get_helpers(self):
+        """
+        체인의 모든 헬퍼 가져오기
+        
+        Returns:
+            모든 헬퍼 객체의 배열
+        """
+        if self.is_empty():
+            return []
+        
+        return self.helpers
+    
+    def is_empty(self):
+        """
+        체인이 비어있는지 확인
+        
+        Returns:
+            체인이 비어있으면 True, 아니면 False
+        """
+        return len(self.bones) == 0
+    
+    def clear(self):
+        """체인의 모든 뼈대와 헬퍼 참조 제거"""
+        self.bones = []
+        self.helpers = []
+        self.clavicle = None
+        self.upperArm = None
+    
+    def delete_all(self):
+        """
+        체인의 모든 뼈대와 헬퍼를 3ds Max 씬에서 삭제
+        
+        Returns:
+            삭제 성공 여부 (boolean)
+        """
+        if self.is_empty():
+            return False
+            
+        try:
+            for bone in self.bones:
+                if rt.isValidNode(bone):
+                    rt.delete(bone)
+            
+            for helper in self.helpers:
+                if rt.isValidNode(helper):
+                    rt.delete(helper)
+                
+            self.clear()
+            return True
+        except:
+            return False
+    
+    def update_lift_scale(self, newLiftScale=0.8):
+        """
+        자동 쇄골의 들어올림 스케일 업데이트
+        
+        Args:
+            newLiftScale: 새로운 들어올림 스케일 (기본값: 0.8)
+            
+        Returns:
+            업데이트 성공 여부 (boolean)
+        """
+        if self.is_empty() or not rt.isValidNode(self.clavicle) or not rt.isValidNode(self.upperArm):
+            return False
+        
+        clavicle = self.clavicle
+        upperArm = self.upperArm
+        
+        # 기존 본과 헬퍼 삭제
+        self.delete_all()
+        
+        # 새로운 LiftScale 값 설정
+        self.liftScale = newLiftScale
+        self.clavicle = clavicle
+        self.upperArm = upperArm
+        
+        # 재생성
+        result = jal.autoClavicle.create_bones(self.clavicle, self.upperArm, self.liftScale)
+        if result:
+            return True
+        
+        return False
+    
+    @classmethod
+    def from_auto_clavicle_result(cls, inResult):
+        """
+        AutoClavicle 클래스의 결과로부터 AutoClavicleChain 인스턴스 생성
+        
+        Args:
+            inResult: AutoClavicle 클래스의 메서드가 반환한 결과값 딕셔너리
+            
+        Returns:
+            AutoClavicleChain 인스턴스
+        """
+        chain = cls(inResult)
+        return chain

pyjallib/max/bip.py

@@ -6,6 +6,9 @@ Biped 모듈 - 3ds Max의 Biped 객체 관련 기능 제공
 원본 MAXScript의 bip.ms를 Python으로 변환하였으며, pymxs 모듈 기반으로 구현됨
 """
 
+
+import os
+
 from pymxs import runtime as rt
 
 # Import necessary service classes for default initialization
@@ -452,11 +455,11 @@ class Bip:
             if rt.isValidObj(baseSkelObj):
                 baseSkel[i] = baseSkelObj
         
-        self.anim.save_xform(bipSkel)
-        self.anim.set_xform(bipSkel)
-        
-        self.anim.save_xform(baseSkel)
-        self.anim.set_xform(baseSkel)
+            self.anim.save_xform(bipSkel[i])
+            self.anim.set_xform(bipSkel[i])
+            
+            self.anim.save_xform(baseSkel[i])
+            self.anim.set_xform(baseSkel[i])
         
         for i in range(len(baseSkel)):
             if baseSkel[i] is not None:
@@ -478,21 +481,25 @@ class Bip:
         """
         rt.setWaitCursor()
         
-        bipSkel = self.get_bips()
+        bipComs = self.get_coms()
+        allBips = self.get_nodes(bipComs[0])
+        bipSkel = [item for item in allBips if item != bipComs[0]]
         baseSkel = [None] * len(bipSkel)
         
         for i in range(len(bipSkel)):
-            baseSkeletonName = self.name.replace_base(bipSkel[i].name, skinBoneBaseName)
-            baseSkeletonName = self.name.replace_filteringChar(baseSkeletonName, "_")
+            baseSkeletonName = self.name.replace_name_part("Base", bipSkel[i].name, skinBoneBaseName)
+            baseSkeletonName = self.name.replace_filtering_char(baseSkeletonName, "_")
+            print("baseSkeletonName", baseSkeletonName)
             baseSkelObj = rt.getNodeByName(baseSkeletonName)
+            print("baseSkelObj", baseSkelObj)
             if rt.isValidObj(baseSkelObj):
                 baseSkel[i] = baseSkelObj
         
-        self.anim.save_xform(bipSkel)
-        self.anim.set_xform(bipSkel)
-        
-        self.anim.save_xform(baseSkel)
-        self.anim.set_xform(baseSkel)
+            self.anim.save_xform(bipSkel[i])
+            self.anim.set_xform(bipSkel[i])
+            
+            self.anim.save_xform(baseSkel[i])
+            self.anim.set_xform(baseSkel[i])
         
         for i in range(len(baseSkel)):
             if baseSkel[i] is not None:
@@ -503,4 +510,194 @@ class Bip:
             if baseSkel[i] is not None:
                 baseSkel[i].boneEnable = True
                 
-        rt.setArrowCursor()
+        rt.setArrowCursor()
+        
+    def convert_name_for_ue5(self, inBipRoot, inBipNameConfigFile):
+        """
+        Biped 이름을 UE5에 맞게 변환
+        
+        Args:
+            inBipRoot: 변환할 Biped 객체
+            
+        Returns:
+            변환된 Biped 객체
+        """
+        bipComs = self.get_coms()
+    
+        if len(bipComs) > 1:
+            rt.messageBox("Please select only one Biped object.")
+            return False
+        
+        from pyjallib.max.name import Name
+        
+        bipNameTool = Name(configPath=inBipNameConfigFile)
+        
+        bipObj = bipComs[0]
+        bipNodes = self.get_all(bipObj)
+        for bipNode in bipNodes:
+            if bipNode.name == bipObj.controller.rootName:
+                bipNode.name = bipNode.name.lower()
+                continue
+            
+            bipNodeNameDict = bipNameTool.convert_to_dictionary(bipNode.name)
+            
+            newNameDict = {}
+            for namePartName, value in bipNodeNameDict.items():
+                namePart = bipNameTool.get_name_part(namePartName)
+                desc = namePart.get_description_by_value(value)
+                
+                if namePartName == "RealName" or namePartName == "Index" or namePartName == "Nub":
+                    newNameDict[namePartName] = value
+                else:
+                    newNameDict[namePartName] = self.name.get_name_part(namePartName).get_value_by_description(desc)
+            
+            if newNameDict["Index"] == "" and self.name._has_digit(newNameDict["RealName"]):
+                if "Finger" not in newNameDict["RealName"]:
+                    splitedRealName = self.name._split_into_string_and_digit(newNameDict["RealName"])
+                    newNameDict["RealName"] = splitedRealName[0]
+                    newNameDict["Index"] = splitedRealName[1]
+            if newNameDict["Nub"] == "" and bipNameTool.get_name_part_value_by_description("Nub", "Nub") in (newNameDict["RealName"]):
+                newNameDict["RealName"] = newNameDict["RealName"].replace(bipNameTool.get_name_part_value_by_description("Nub", "Nub"), "")
+                newNameDict["Nub"] = self.name.get_name_part_value_by_description("Nub", "Nub")
+            
+            if newNameDict["RealName"] == "Forearm":
+                newNameDict["RealName"] = "Lowerarm"
+            
+            if newNameDict["RealName"] == "Spine" or newNameDict["RealName"] == "Neck":
+                if newNameDict["Index"] == "":
+                    newNameDict["Index"] = str(int(1)).zfill(self.name.get_padding_num())
+                else:
+                    newNameDict["Index"] = str(int(newNameDict["Index"]) + 1).zfill(self.name.get_padding_num())
+                
+            newBipName = self.name.combine(newNameDict)
+            
+            bipNode.name = newBipName.lower()
+            
+        # 손가락 바꾸는 부분
+        # 5개가 아닌 손가락은 지원하지 않음
+        # 손가락 하나의 최대 링크는 3개
+        indices = []
+        if bipObj.controller.knuckles:
+            pass
+        else:
+            indices = list(range(0, 15, 3))
+            
+        fingerNum = bipObj.controller.fingers
+        fingerLinkNum = bipObj.controller.fingerLinks
+            
+        lFingersList = []
+        rFingersList = []
+        
+        for i in range(1, fingerNum+1):
+            fingers = []
+            for j in range(1, fingerLinkNum+1):
+                linkIndex = (i-1)*fingerLinkNum + j
+                fingerNode = rt.biped.getNode(bipObj.controller, rt.name("lFingers"), link=linkIndex)
+                fingers.append(fingerNode)
+            lFingersList.append(fingers)
+        for i in range(1, fingerNum+1):
+            fingers = []
+            for j in range(1, fingerLinkNum+1):
+                linkIndex = (i-1)*fingerLinkNum + j
+                fingerNode = rt.biped.getNode(bipObj.controller, rt.name("rFingers"), link=linkIndex)
+                fingers.append(fingerNode)
+            rFingersList.append(fingers)
+            
+        fingerName = ["thumb", "index", "middle", "ring", "pinky"]
+        
+        for i, fingers in enumerate(lFingersList):
+            for j, item in enumerate(fingers):
+                item.name = self.name.replace_name_part("RealName", item.name, fingerName[i])
+                item.name = self.name.replace_name_part("Index", item.name, str(j+1))
+            
+            fingerNub = self.bone.get_every_children(fingers[-1])[0]
+            fingerNub.name = self.name.replace_name_part("RealName", fingerNub.name, fingerName[i])
+            fingerNub.name = self.name.remove_name_part("Index", fingerNub.name)
+            fingerNub.name = self.name.replace_name_part("Nub", fingerNub.name, self.name.get_name_part_value_by_description("Nub", "Nub"))
+        
+        for i, fingers in enumerate(rFingersList):
+            for j, item in enumerate(fingers):
+                item.name = self.name.replace_name_part("RealName", item.name, fingerName[i])
+                item.name = self.name.replace_name_part("Index", item.name, str(j+1))
+            
+            fingerNub = self.bone.get_every_children(fingers[-1])[0]
+            fingerNub.name = self.name.replace_name_part("RealName", fingerNub.name, fingerName[i])
+            fingerNub.name = self.name.remove_name_part("Index", fingerNub.name)
+            fingerNub.name = self.name.replace_name_part("Nub", fingerNub.name, self.name.get_name_part_value_by_description("Nub", "Nub"))
+        
+        # Toe 이름 바꾸는 부분
+        lToesList = []
+        rToesList = []
+        
+        toeNum = bipObj.controller.toes
+        toeLinkNum = bipObj.controller.toeLinks
+        
+        # Use the same sequential indexing pattern as fingers
+        for i in range(1, toeNum+1):
+            toes = []
+            for j in range(1, toeLinkNum+1):
+                linkIndex = (i-1)*toeLinkNum + j
+                toeNode = rt.biped.getNode(bipObj.controller, rt.name("lToes"), link=linkIndex)
+                if toeNode:
+                    toes.append(toeNode)
+            if toes:
+                lToesList.append(toes)
+
+        for i in range(1, toeNum+1):
+            toes = []
+            for j in range(1, toeLinkNum+1):
+                linkIndex = (i-1)*toeLinkNum + j
+                toeNode = rt.biped.getNode(bipObj.controller, rt.name("rToes"), link=linkIndex)
+                if toeNode:
+                    toes.append(toeNode)
+            if toes:
+                rToesList.append(toes)
+                
+        for i, toes in enumerate(lToesList):
+            for j, item in enumerate(toes):
+                item.name = self.name.replace_name_part("RealName", item.name, "ball"+str(i+1))
+                item.name = self.name.replace_name_part("Index", item.name, str(j+1))
+            
+            toeNub = self.bone.get_every_children(toes[-1])[0]
+            toeNub.name = self.name.replace_name_part("RealName", toeNub.name, "ball"+str(i+1))
+            toeNub.name = self.name.remove_name_part("Index", toeNub.name)
+            toeNub.name = self.name.replace_name_part("Nub", toeNub.name, self.name.get_name_part_value_by_description("Nub", "Nub"))
+            
+        for i, toes in enumerate(rToesList):
+            for j, item in enumerate(toes):
+                item.name = self.name.replace_name_part("RealName", item.name, "ball"+str(i+1))
+                item.name = self.name.replace_name_part("Index", item.name, str(j+1))
+            
+            toeNub = self.bone.get_every_children(toes[-1])[0]
+            toeNub.name = self.name.replace_name_part("RealName", toeNub.name, "ball"+str(i+1))
+            toeNub.name = self.name.remove_name_part("Index", toeNub.name)
+            toeNub.name = self.name.replace_name_part("Nub", toeNub.name, self.name.get_name_part_value_by_description("Nub", "Nub"))
+        
+        if toeNum == 1:
+            if toeLinkNum == 1:
+                lToesList[0][0].name = self.name.replace_name_part("RealName", lToesList[0][0].name, "ball")
+                lToesList[0][0].name = self.name.remove_name_part("Index", lToesList[0][0].name)
+            else:
+                for i, item in enumerate(lToesList[0]):
+                    item.name = self.name.replace_name_part("RealName", item.name, "ball")
+                    item.name = self.name.replace_name_part("Index", item.name, str(i+1))
+            
+            toeNub = self.bone.get_every_children(lToesList[0][-1])[0]
+            toeNub.name = self.name.replace_name_part("RealName", toeNub.name, "ball")
+            toeNub.name = self.name.remove_name_part("Index", toeNub.name)
+            toeNub.name = self.name.replace_name_part("Nub", toeNub.name, self.name.get_name_part_value_by_description("Nub", "Nub"))
+            
+            if toeLinkNum == 1:
+                rToesList[0][0].name = self.name.replace_name_part("RealName", lToesList[0][0].name, "ball")
+                rToesList[0][0].name = self.name.remove_name_part("Index", lToesList[0][0].name)
+            else:
+                for i, item in enumerate(rToesList[0]):
+                    item.name = self.name.replace_name_part("RealName", item.name, "ball")
+                    item.name = self.name.replace_name_part("Index", item.name, str(i+1))
+            
+            toeNub = self.bone.get_every_children(rToesList[0][-1])[0]
+            toeNub.name = self.name.replace_name_part("RealName", toeNub.name, "ball")
+            toeNub.name = self.name.remove_name_part("Index", toeNub.name)
+            toeNub.name = self.name.replace_name_part("Nub", toeNub.name, self.name.get_name_part_value_by_description("Nub", "Nub"))
+        
+        return True