implement pmx.writer

[meshio/pymeshio.git] / pymeshio / common.py

diff --git a/pymeshio/common.py b/pymeshio/common.py
index e67580d..7177506 100644 (file)
--- a/pymeshio/common.py
+++ b/pymeshio/common.py
@@ -21,6 +21,9 @@ class Vector2(object):
     def __str__(self):\r
         return "<%f %f>" % (self.x, self.y)\r
 \r
+    def __eq__(self, rhs):\r
+        return self.x==rhs.x and self.y==rhs.y\r
+\r
     def __getitem__(self, key):\r
         if key==0:\r
             return self.x\r
@@ -32,6 +35,10 @@ class Vector2(object):
     def to_tuple(self):\r
         return (self.x, self.y)\r
 \r
+    def cross(self, rhs):\r
+        """cross(outer) product"""\r
+        return self.x*rhs.y-self.y*rhs.x\r
+\r
 \r
 class Vector3(object):\r
     """\r
@@ -46,6 +53,9 @@ class Vector3(object):
     def __str__(self):\r
         return "<%f %f %f>" % (self.x, self.y, self.z)\r
 \r
+    def __eq__(self, rhs):\r
+        return self.x==rhs.x and self.y==rhs.y and self.z==rhs.z\r
+\r
     def __getitem__(self, key):\r
         if key==0:\r
             return self.x\r
@@ -59,8 +69,39 @@ class Vector3(object):
     def to_tuple(self):\r
         return (self.x, self.y, self.z)\r
 \r
-    def __add__(l, r):\r
-        return Vector3(l.x+r.x, l.y+r.y, l.z+r.z)\r
+    def __add__(self, r):\r
+        return Vector3(self.x+r.x, self.y+r.y, self.z+r.z)\r
+\r
+    def __sub__(self, rhs):\r
+        return Vector3(self.x-rhs.x, self.y-rhs.y, self.z-rhs.z)\r
+\r
+    def getSqNorm(self):\r
+        return self.x*self.x + self.y*self.y + self.z*self.z\r
+\r
+    def getNorm(self):\r
+        return math.sqrt(self.getSqNorm())\r
+\r
+    def normalize(self):\r
+        factor=1.0/self.getNorm()\r
+        self.x*=factor\r
+        self.y*=factor\r
+        self.z*=factor\r
+        return self\r
+\r
+    def to_a(self):\r
+        return [self.x, self.y, self.z]\r
+\r
+    def dot(self, rhs):\r
+        """dot(inner) product"""\r
+        return self.x*rhs.x + self.y*rhs.y + self.z*rhs.z\r
+\r
+    def cross(self, rhs):\r
+        """cross(outer) product"""\r
+        return Vector3(\r
+                self.y*rhs.z - rhs.y*self.z,\r
+                self.z*rhs.x - rhs.z*self.x,\r
+                self.x*rhs.y - rhs.x*self.y,\r
+                )\r
 \r
 \r
 class Quaternion(object):\r
@@ -177,6 +218,9 @@ class RGB(object):
         self.g=g\r
         self.b=b\r
 \r
+    def __eq__(self, rhs):\r
+        return self.r==rhs.r and self.g==rhs.g and self.b==rhs.b\r
+\r
     def __getitem__(self, key):\r
         if key==0:\r
             return self.r\r
@@ -199,6 +243,9 @@ class RGBA(object):
         self.b=b\r
         self.a=a\r
 \r
+    def __eq__(self, rhs):\r
+        return self.r==rhs.r and self.g==rhs.g and self.b==rhs.b and self.a==rhs.a\r
+\r
     def __getitem__(self, key):\r
         if key==0:\r
             return self.r\r
@@ -225,17 +272,24 @@ class ParseException(Exception):
     pass\r
 \r
 \r
-def readall(path: str) -> bytes:\r
+def readall(path):\r
+    """read all bytes from path\r
+    """\r
     with open(path, "rb") as f:\r
         return f.read()\r
 \r
 \r
-class BinaryLoader(object):\r
-    def __init__(self, io):\r
-        self.io=io\r
+class BinaryReader(object):\r
+    """general BinaryReader\r
+    """\r
+    def __init__(self, ios):\r
+        self.ios=ios\r
+\r
+    def is_end(self):\r
+        return not self.ios.readable()\r
 \r
-    def unpack(self, fmt: str, size: int) -> "read value as format":\r
-        result=struct.unpack(fmt, self.io.read(size))\r
+    def unpack(self, fmt, size):\r
+        result=struct.unpack(fmt, self.ios.read(size))\r
         return result[0]\r
 \r
     def read_uint(self, size):\r
@@ -280,3 +334,43 @@ class BinaryLoader(object):
                 )\r
 \r
 \r
+class WriteException(Exception):\r
+    pass\r
+\r
+\r
+class BinaryWriter(object):\r
+    def __init__(self, ios):\r
+        self.ios=ios\r
+\r
+    def write_bytes(self, v, size=None):\r
+        if size:\r
+            self.ios.write(struct.pack("={0}s".format(size), v))\r
+        else:\r
+            self.ios.write(v)\r
+\r
+    def write_float(self, v):\r
+        self.ios.write(struct.pack("f", v))\r
+\r
+    def write_uint(self, v, size):\r
+        if size==1:\r
+            self.ios.write(struct.pack("B", v))\r
+        elif size==2:\r
+            self.ios.write(struct.pack("H", v))\r
+        elif size==4:\r
+            self.ios.write(struct.pack("I", v))\r
+        else:\r
+            raise WriteError("invalid int uint size")\r
+\r
+    def write_vector2(self, v):\r
+        self.ios.write(struct.pack("=2f", v.x, v.y))\r
+\r
+    def write_vector3(self, v):\r
+        self.ios.write(struct.pack("=3f", v.x, v.y, v.z))\r
+\r
+    def write_rgb(self, v):\r
+        self.ios.write(struct.pack("=3f", v.r, v.g, v.b))\r
+\r
+    def write_rgba(self, v):\r
+        self.ios.write(struct.pack("=4f", v.r, v.g, v.b, v.a))\r
+\r
+\r