pymeshio/pmx/__init__.py

   1 #!/usr/bin/env python\r
   2 # coding: utf-8\r
   3 """\r
   4 pmx file io library.\r
   5 \r
   6 pmx file format:\r
   7     PMDEditor's Lib/PMX仕様/PMX仕様.txt\r
   8 """\r
   9 __author__="ousttrue"\r
  10 __license__="zlib"\r
  11 __versioon__="1.0.0"\r
  12 \r
  13 \r
  14 import io\r
  15 import os\r
  16 import struct\r
  17 from pymeshio import common\r
  18 \r
  19 \r
  20 \r
  21 class Ik(object):\r
  22     """ik info\r
  23     """\r
  24     __slots__=[\r
  25             'target_index',\r
  26             'loop',\r
  27             'limit_radian',\r
  28             'link',\r
  29             ]\r
  30     def __init__(self, target_index, loop, limit_radian):\r
  31         self.target_index=target_index\r
  32         self.loop=loop\r
  33         self.limit_radian=limit_radian\r
  34         self.link=[]\r
  35 \r
  36 \r
  37 class IkLink(object):\r
  38     """ik link info\r
  39     """\r
  40     __slots__=[\r
  41             'bone_index',\r
  42             'limit_angle',\r
  43             'limit_min',\r
  44             'limit_max',\r
  45             ]\r
  46     def __init__(self, bone_index, limit_angle):\r
  47         self.bone_index=bone_index\r
  48         self.limit_angle=limit_angle\r
  49         self.limit_min=None\r
  50         self.limit_max=None\r
  51 \r
  52 \r
  53 class Bone(object):\r
  54     """material\r
  55 \r
  56     Bone: see __init__\r
  57     """\r
  58     __slots__=[\r
  59             'name',\r
  60             'english_name',\r
  61             'position',\r
  62             'parent_index',\r
  63             'layer',\r
  64             'flag',\r
  65 \r
  66             'tail_positoin',\r
  67             'tail_index',\r
  68             'effect_index',\r
  69             'effect_factor',\r
  70             'fixed_axis',\r
  71             'local_x_vector',\r
  72             'local_z_vector',\r
  73             'external_key',\r
  74             'ik',\r
  75             ]\r
  76     def __init__(self,\r
  77             name: str,\r
  78             english_name: str,\r
  79             position: common.Vector3,\r
  80             parent_index: int,\r
  81             layer: int,\r
  82             flag: int\r
  83             ):\r
  84         self.name=name,\r
  85         self.english_name=english_name\r
  86         self.position=position\r
  87         self.parent_index=parent_index\r
  88         self.layer=layer\r
  89         self.flag=flag\r
  90 \r
  91     def getConnectionFlag(self) -> int:\r
  92         return self.flag & 0x0001\r
  93 \r
  94     def getIkFlag(self) -> int:\r
  95         return (self.flag & 0x0020) >> 5\r
  96 \r
  97     def getRotationFlag(self) -> int:\r
  98         return (self.flag & 0x0100) >> 8\r
  99 \r
 100     def getTranslationFlag(self) -> int:\r
 101         return (self.flag & 0x0200) >> 9\r
 102 \r
 103     def getFixedAxisFlag(self) -> int:\r
 104         return (self.flag & 0x0400) >> 10\r
 105 \r
 106     def getLocalCoordinateFlag(self) -> int:\r
 107         return (self.flag &  0x0800) >> 11\r
 108     \r
 109     def getExternalParentDeformFlag(self) -> int:\r
 110         return (self.flag &  0x2000) >> 13\r
 111 \r
 112  \r
 113 class Material(object):\r
 114     """material\r
 115 \r
 116     Attributes: see __init__\r
 117     """\r
 118     __slots__=[\r
 119             'name',\r
 120             'english_name',\r
 121             'diffuse_color',\r
 122             'diffuse_alpha',\r
 123             'specular_color',\r
 124             'specular_factor',\r
 125             'ambient_color',\r
 126             'flag',\r
 127             'edge_color',\r
 128             'edge_size',\r
 129             'texture_index',\r
 130             'sphia_texture_index',\r
 131             'sphia_mode',\r
 132             'toon_sharing_flag',\r
 133             'toon_texture_index',\r
 134             'comment',\r
 135             'index_count',\r
 136             ]\r
 137     def __init__(self,\r
 138             name: str,\r
 139             english_name: str,\r
 140             diffuse_color: common.RGB,\r
 141             diffuse_alpha: float,\r
 142             specular_color: common.RGB,\r
 143             specular_factor: float,\r
 144             ambient_color: common.RGB,\r
 145             flag: int,\r
 146             edge_color: common.RGBA,\r
 147             edge_size: float,\r
 148             texture_index: int,\r
 149             sphia_texture_index: int,\r
 150             sphia_mode: int,\r
 151             toon_sharing_flag: int\r
 152             ):\r
 153         self.name=name\r
 154         self.english_name=english_name\r
 155         self.diffuse_color=diffuse_color\r
 156         self.diffuse_alpha=diffuse_alpha\r
 157         self.specular_color=specular_color\r
 158         self.specular_factor=specular_factor\r
 159         self.ambient_color=ambient_color\r
 160         self.flag=flag\r
 161         self.edge_color=edge_color\r
 162         self.edge_size=edge_size\r
 163         self.texture_index=texture_index\r
 164         self.sphia_texture_index=sphia_texture_index\r
 165         self.sphia_mode=sphia_mode\r
 166         self.toon_sharing_flag=toon_sharing_flag\r
 167         #\r
 168         self.toon_texture_index=None\r
 169         self.comment=''\r
 170         self.index_count=0\r
 171 \r
 172 \r
 173 class Deform(object):\r
 174     pass\r
 175 \r
 176 \r
 177 class Bdef1(object):\r
 178     """bone deform. use a weight\r
 179 \r
 180     Attributes: see __init__\r
 181     """\r
 182     __slots__=[ 'bone_index']\r
 183     def __init__(self, bone_index: int):\r
 184         self.bone_index=bone_index\r
 185 \r
 186 \r
 187 class Bdef2(object):\r
 188     """bone deform. use two weights\r
 189 \r
 190     Attributes: see __init__\r
 191     """\r
 192     __slots__=[ 'index0', 'index1', 'weight0']\r
 193     def __init__(self, \r
 194             index0: int,\r
 195             index1: int,\r
 196             weight0: float):\r
 197         self.index0=index0\r
 198         self.index1=index1\r
 199         self.weight0=weight0\r
 200 \r
 201 \r
 202 class Vertex(object):\r
 203     """pmx vertex\r
 204 \r
 205     Attributes: see __init__\r
 206     """\r
 207     __slots__=[ 'position', 'normal', 'uv', 'deform', 'edge_factor' ]\r
 208     def __init__(self, \r
 209             position: common.Vector3, \r
 210             normal: common.Vector3, \r
 211             uv: common.Vector2, \r
 212             deform: Deform, \r
 213             edge_factor: float):\r
 214         self.position=position \r
 215         self.normal=normal\r
 216         self.uv=uv\r
 217         self.deform=deform\r
 218         self.edge_factor=edge_factor\r
 219 \r
 220 \r
 221 class Model(object):\r
 222     """pmx data representation\r
 223 \r
 224     Attributes:\r
 225         version: pmx version(expected 2.0)\r
 226         name: \r
 227         english_name: \r
 228         comment: \r
 229         english_comment: \r
 230         vertices:\r
 231         textures:\r
 232         materials:\r
 233         bones:\r
 234     """\r
 235     __slots__=[\r
 236             'version', # pmx version\r
 237             'name', # model name\r
 238             'english_name', # model name in english\r
 239             'comment', # model comment\r
 240             'english_comment', # model comment in english\r
 241             'vertices',\r
 242             'indices',\r
 243             'textures',\r
 244             'materials',\r
 245             'bones',\r
 246             ]\r
 247     def __init__(self):\r
 248         self.version=0.0\r
 249         self.name=''\r
 250         self.english_name=''\r
 251         self.comment=''\r
 252         self.english_comment=''\r
 253         self.vertices=[]\r
 254         self.indices=[]\r
 255         self.textures=[]\r
 256         self.materials=[]\r
 257         self.bones=[]\r
 258 \r