pyjallib 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

pyjallib/max/groinBone.py

@@ -7,7 +7,14 @@
 
 from pymxs import runtime as rt
 
-from .header import jal
+# 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
+
+from .boneChain import BoneChain
 
 class GroinBone:
     """
@@ -15,107 +22,175 @@ class GroinBone:
     3DS Max에서 고간 부 본을 생성하고 관리하는 기능을 제공합니다.
     """
     
-    def __init__(self):
+    def __init__(self, nameService=None, animService=None, constraintService=None, boneService=None, helperService=None):
         """
         클래스 초기화.
         
         Args:
             nameService: 이름 처리 서비스 (제공되지 않으면 새로 생성)
             animService: 애니메이션 서비스 (제공되지 않으면 새로 생성)
-            constService: 제약 서비스 (제공되지 않으면 새로 생성)
-            bipService: 바이페드 서비스 (제공되지 않으면 새로 생성)
+            constraintService: 제약 서비스 (제공되지 않으면 새로 생성)
+            bipService: Biped 서비스 (제공되지 않으면 새로 생성)
+            boneService: 뼈대 서비스 (제공되지 않으면 새로 생성)
+            twistBoneService: 트위스트 본 서비스 (제공되지 않으면 새로 생성)
+            helperService: 헬퍼 객체 서비스 (제공되지 않으면 새로 생성)
+        """
+        # 서비스 인스턴스 설정 또는 생성
+        self.name = nameService if nameService else Name()
+        self.anim = animService if animService else Anim()
+        
+        # 종속성이 있는 서비스들은 이미 생성된 서비스들을 전달
+        self.const = constraintService if constraintService else Constraint(nameService=self.name)
+        self.bone = boneService if boneService else Bone(nameService=self.name, animService=self.anim)
+        self.helper = helperService if helperService else Helper(nameService=self.name)
+        
+        # 초기화된 결과를 저장할 변수들
+        self.pelvis = None
+        self.lThighTwist = None
+        self.rThighTwist = None
+        self.bones = []
+        self.helpers = []
+        self.pelvisWeight = 40.0
+        self.thighWeight = 60.0
+    
+    def reset(self):
+        """
+        클래스의 주요 컴포넌트들을 초기화합니다.
+        서비스가 아닌 클래스 자체의 작업 데이터를 초기화하는 함수입니다.
+        
+        Returns:
+            self: 메소드 체이닝을 위한 자기 자신 반환
         """
-        self.name = jal.name
-        self.anim = jal.anim
-        # Ensure dependent services use the potentially newly created instances
-        self.const = jal.constraint
-        self.bip = jal.bip
-        self.bone = jal.bone
-        self.twistBone = jal.twistBone
-        self.helper = jal.helper
-        
-        self.bipObj = None
-        self.genBones = []
-        self.genHelpers = []
+        self.pelvis = None
+        self.lThighTwist = None
+        self.rThighTwist = None
+        self.bones = []
+        self.helpers = []
+        self.pelvisWeight = 40.0
+        self.thighWeight = 60.0
+        
+        return self
     
-    def create_bone(self, inObj, inPelvisWeight=40.0, inThighWeight=60.0):
+    def create_bone(self, inPelvis, inLThighTwist, inRThighTwist, inPelvisWeight=40.0, inThighWeight=60.0):
         """
         고간 부 본을 생성하는 메소드.
         
         Args:
-            name: 생성할 본의 이름
-            parent: 부모 본 (제공되지 않으면 None)
+            inPelvis: Biped 객체
+            inLThighTwist: 왼쪽 허벅지 트위스트 본
+            inRThighTwist: 오른쪽 허벅지 트위스트 본
+            inPelvisWeight: 골반 가중치 (기본값: 40.0)
+            inThighWeight: 허벅지 가중치 (기본값: 60.0)
         
         Returns:
-            생성된 본 객체
+            BoneChain: 생성된 고간 부 본 체인 객체 또는 실패 시 False
         """
-        if self.bip.is_biped_object(inObj) == False:
-            rt.messageBox("This is not a biped object.")
-            return False
-        
-        bipObj = self.bip.get_com(inObj)
-        self.bipObj = bipObj
         
-        lThigh = self.bip.get_grouped_nodes(inObj, "lLeg")[0]
-        rThigh = self.bip.get_grouped_nodes(inObj, "rLeg")[0]
-        pelvis = self.bip.get_grouped_nodes(inObj, "pelvis")[0]
+        if rt.isValidNode(inPelvis) == False or rt.isValidNode(inLThighTwist) == False or rt.isValidNode(inRThighTwist) == False:
+            rt.messageBox("There is no valid node.")
+            return False
         
-        lThighTwists = self.twistBone.get_thigh_type(lThigh)
-        rThighTwists = self.twistBone.get_thigh_type(rThigh)
+        groinName = "Groin"
+        if inPelvis.name[0].islower():
+            groinName = groinName.lower()
         
-        if len(lThighTwists) == 0 or len(rThighTwists) == 0:
-            rt.messageBox("There is no twist bone.")
-            return False
+        groinBaseName = self.name.replace_name_part("RealName", inLThighTwist.name, groinName)
         
-        pelvisHelper = self.helper.create_point(bipObj.name + " Dum Groin 00")
-        pelvisHelper.transform = bipObj.transform
-        self.anim.rotate_local(pelvisHelper, 90, 0, 0)
-        self.anim.rotate_local(pelvisHelper, 0, 0, -90)
-        pelvisHelper.parent = pelvis
+        pelvisHelperName = self.name.replace_name_part("Type", groinBaseName, self.name.get_name_part_value_by_description("Type", "Dummy"))
+        pelvisHelperName = self.name.replace_name_part("Index", pelvisHelperName, "0")
+        pelvisHelperName = self.name.remove_name_part("Side", pelvisHelperName)
+        pelvisHelper = self.helper.create_point(pelvisHelperName)
+        pelvisHelper.transform = inPelvis.transform
+        self.anim.rotate_local(pelvisHelper, 0.0, 0.0, -180.0)
+        pelvisHelper.parent = inPelvis
         self.helper.set_shape_to_box(pelvisHelper)
-        self.genHelpers.append(pelvisHelper)
         
-        groinBones = self.bone.create_simple_bone(3.0, bipObj.name +" Groin 00", size=2)
-        groinBones[0].transform = pelvisHelper.transform
-        groinBones[0].parent = pelvis
-        for groinBone in groinBones:
-            self.genBones.append(groinBone)
-        
-        self.const.assign_rot_const_multi(groinBones[0], [pelvisHelper, lThigh, rThigh])
-        rotConst = self.const.get_rot_list_controller(groinBones[0])[1]
+        lThighTwistHelperName = self.name.replace_name_part("Type", groinBaseName, self.name.get_name_part_value_by_description("Type", "Dummy"))
+        lThighTwistHelperName = self.name.replace_name_part("Side", lThighTwistHelperName, self.name.get_name_part_value_by_description("Side", "Left"))
+        lThighTwistHelperName = self.name.replace_name_part("Index", lThighTwistHelperName, "0")
+        lThighTwistHelper = self.helper.create_point(lThighTwistHelperName)
+        lThighTwistHelper.transform = pelvisHelper.transform
+        lThighTwistHelper.position = inLThighTwist.position
+        lThighTwistHelper.parent = inLThighTwist
+        self.helper.set_shape_to_box(lThighTwistHelper)
+        
+        rThighTwistHelperName = self.name.replace_name_part("Type", groinBaseName, self.name.get_name_part_value_by_description("Type", "Dummy"))
+        rThighTwistHelperName = self.name.replace_name_part("Side", rThighTwistHelperName, self.name.get_name_part_value_by_description("Side", "Right"))
+        rThighTwistHelperName = self.name.replace_name_part("Index", rThighTwistHelperName, "0")
+        rThighTwistHelper = self.helper.create_point(rThighTwistHelperName)
+        rThighTwistHelper.transform = pelvisHelper.transform
+        rThighTwistHelper.position = inRThighTwist.position
+        rThighTwistHelper.parent = inRThighTwist
+        self.helper.set_shape_to_box(rThighTwistHelper)
+        
+        groinBoneName = self.name.replace_name_part("Index", groinBaseName, "0")
+        groinBoneName = self.name.remove_name_part("Side", groinBoneName)
+        groinBone = self.bone.create_nub_bone(groinBoneName, 2)
+        groinBone.name = groinBoneName
+        groinBone.transform = pelvisHelper.transform
+        groinBone.parent = inPelvis
+        
+        self.const.assign_rot_const_multi(groinBone, [pelvisHelper, lThighTwistHelper, rThighTwistHelper])
+        rotConst = self.const.get_rot_list_controller(groinBone)[1]
         rotConst.setWeight(1, inPelvisWeight)
         rotConst.setWeight(2, inThighWeight/2.0)
         rotConst.setWeight(3, inThighWeight/2.0)
         
-    def delete(self):
-        """
-        생성된 고간 부 본과 헬퍼를 삭제하는 메소드.
-        
-        Returns:
-            None
-        """
-        rt.delete(self.genBones)
-        rt.delete(self.genHelpers)
-        
-        self.genBones = []
-        self.genHelpers = []
+        # 결과를 멤버 변수에 저장
+        self.pelvis = inPelvis
+        self.lThighTwist = inLThighTwist
+        self.rThighTwist = inRThighTwist
+        self.bones = [groinBone]
+        self.helpers = [pelvisHelper, lThighTwistHelper, rThighTwistHelper]
+        self.pelvisWeight = inPelvisWeight
+        self.thighWeight = inThighWeight
+        
+        # BoneChain 구조에 맞는 결과 딕셔너리 생성
+        result = {
+            "Bones": [groinBone],
+            "Helpers": [pelvisHelper, lThighTwistHelper, rThighTwistHelper],
+            "SourceBones": [inPelvis, inLThighTwist, inRThighTwist],
+            "Parameters": [inPelvisWeight, inThighWeight]
+        }
+        
+        # 메소드 호출 후 데이터 초기화
+        self.reset()
+        
+        # BoneChain 객체 반환
+        return BoneChain.from_result(result)
     
-    def update_weight(self, inPelvisWeight=40.0, inThighWeight=60.0):
+    def create_bones_from_chain(self, inBoneChain: BoneChain):
         """
-        고간 부 본의 가중치를 업데이트하는 메소드.
+        기존 BoneChain 객체에서 고간 부 본을 생성합니다.
+        기존 설정을 복원하거나 저장된 데이터에서 고간 부 본 셋업을 재생성할 때 사용합니다.
         
         Args:
-            inPelvisWeight: 골반 가중치
-            inThighWeight: 허벅지 가중치
+            inBoneChain (BoneChain): 고간 부 본 정보를 포함한 BoneChain 객체
         
         Returns:
-            None
+            BoneChain: 업데이트된 BoneChain 객체 또는 실패 시 None
         """
-        if len(self.genBones) == 0:
-            return False
-        
-        self.delete()
-        self.create_bone(self.bipObj, inPelvisWeight, inThighWeight)
-        
-        
-        
+        if not inBoneChain or inBoneChain.is_empty():
+            return None
+        
+        # 기존 객체 삭제
+        inBoneChain.delete()
+            
+        # BoneChain에서 필요한 정보 추출
+        sourceBones = inBoneChain.sourceBones
+        parameters = inBoneChain.parameters
+        
+        # 필수 소스 본 확인 (최소 3개: 골반, 좌허벅지트위스트, 우허벅지트위스트)
+        if len(sourceBones) < 3 or not rt.isValidNode(sourceBones[0]) or not rt.isValidNode(sourceBones[1]) or not rt.isValidNode(sourceBones[2]):
+            return None
+            
+        # 파라미터 가져오기 (또는 기본값 사용)
+        pelvisWeight = parameters[0] if len(parameters) > 0 else 40.0
+        thighWeight = parameters[1] if len(parameters) > 1 else 60.0
+        
+        # 새로운 고간 부 본 생성
+        inPelvis = sourceBones[0]
+        inLThighTwist = sourceBones[1]
+        inRThighTwist = sourceBones[2]
+        
+        return self.create_bone(inPelvis, inLThighTwist, inRThighTwist, pelvisWeight, thighWeight)

pyjallib/max/header.py

@@ -25,8 +25,11 @@ from .bip import Bip
 from .skin import Skin
 
 from .twistBone import TwistBone
-# from .autoClavicle import AutoClavicle
-# from .volumePreserveBone import VolumePreserveBone
+from .autoClavicle import AutoClavicle
+from .groinBone import GroinBone
+from .volumeBone import VolumeBone
+from .kneeBone import KneeBone
+from .hip import Hip
 
 from .morph import Morph
 
@@ -67,13 +70,46 @@ class Header:
         self.bip = Bip(animService=self.anim, nameService=self.name, boneService=self.bone)
         self.skin = Skin()
 
-        self.twistBone = TwistBone(nameService=self.name, animService=self.anim, constService=self.constraint, bipService=self.bip, boneService=self.bone)
-        # self.autoClavicle = AutoClavicle(nameService=self.name, animService=self.anim, helperService=self.helper, boneService=self.bone, constraintService=self.constraint, bipService=self.bip)
-        # self.volumePreserveBone = VolumePreserveBone(nameService=self.name, animService=self.anim, constService=self.constraint, boneService=self.bone, helperService=self.helper)
+        self.twistBone = TwistBone(nameService=self.name, animService=self.anim, constraintService=self.constraint, bipService=self.bip, boneService=self.bone)
+        self.groinBone = GroinBone(nameService=self.name, animService=self.anim, constraintService=self.constraint, boneService=self.bone, helperService=self.helper)
+        self.autoClavicle = AutoClavicle(nameService=self.name, animService=self.anim, helperService=self.helper, boneService=self.bone, constraintService=self.constraint, bipService=self.bip)
+        self.volumeBone = VolumeBone(nameService=self.name, animService=self.anim, constraintService=self.constraint, boneService=self.bone, helperService=self.helper)
+        self.kneeBone = KneeBone(nameService=self.name, animService=self.anim, constraintService=self.constraint, boneService=self.bone, helperService=self.helper, volumeBoneService=self.volumeBone)
+        self.hip = Hip(nameService=self.name, animService=self.anim, helperService=self.helper, boneService=self.bone, constraintService=self.constraint)
         
         self.morph = Morph()
         
         self.tools = []
+    
+    def update_nameConifg(self, configPath):
+        """
+        이름 설정을 업데이트합니다.
+        
+        Args:
+            configPath: ConfigPath 인스턴스
+        """
+        self.name.load_from_config_file(configPath)
+    
+    def add_tool(self, tool):
+        """
+        도구를 추가합니다.
+        
+        Args:
+            tool: 추가할 도구
+        """
+        if tool in self.tools:
+            self.tools.remove(tool)
+        
+        self.tools.append(tool)
 
 # 모듈 레벨에서 전역 인스턴스 생성
-jal = Header.get_instance()
+_pyjallibmaxheader = Header.get_instance()
+
+def get_pyjallibmaxheader():
+    """
+    jal 인스턴스를 반환합니다.
+    
+    Returns:
+        Header 인스턴스
+    """
+    return _pyjallibmaxheader

pyjallib/max/helper.py

@@ -93,30 +93,12 @@ class Helper:
             찾은 Type namePart 값
         """
         typePart = self.name.get_name_part("Type")
-        predefinedValues = typePart.get_predefined_values()
         firstTypeValue = typePart.get_value_by_min_weight()
         
+        helperTypeName = self.name.get_name_part_value_by_description("Type", helperType)
+        if helperTypeName != "":
+            return helperTypeName
         
-        # 헬퍼 타입 패턴 정의
-        helperNamePatterns = {
-            "Dummy": ["dum", "Dum", "Dummy", "Helper", "Hpr", "Dmy"],
-            "IK": ["ik", "IK", "Ik"],
-            "Target": ["Tgt", "Target", "TG", "Tg", "T"],
-            "Parent": ["Prn", "PRN", "Parent", "P"],
-            "ExposeTm": ["Exp", "Etm", "EXP", "ETM"]
-        }
-        
-        # 타입 패턴 가져오기
-        patterns = helperNamePatterns.get(helperType, [])
-        if not patterns:
-            return firstTypeValue
-        
-        # 패턴과 일치하는 값 찾기
-        for value in predefinedValues:
-            if value in patterns:
-                return value
-        
-        # 일치하는 값이 없으면 기본값 반환
         return firstTypeValue
     
     def gen_helper_name_from_obj(self, inObj, make_two=False, is_exp=False):