From 5e849456a5105083766f9ff7f10a53527605915c Mon Sep 17 00:00:00 2001
From: Duccio <ducciolenkowicz@gmail.com>
Date: Mon, 14 May 2018 23:09:42 +0200
Subject: [PATCH] renamed file, added plugin reader implementation

---
 maya/draco_maya.py        | 59 -----------------------
 maya/draco_maya_plugin.py | 98 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 98 insertions(+), 59 deletions(-)
 delete mode 100644 maya/draco_maya.py
 create mode 100644 maya/draco_maya_plugin.py

diff --git a/maya/draco_maya.py b/maya/draco_maya.py
deleted file mode 100644
index f5a670a..0000000
--- a/maya/draco_maya.py
+++ /dev/null
@@ -1,59 +0,0 @@
-import maya.api.OpenMaya as OpenMaya
-import maya.OpenMayaMPx as OpenMayaMPx
-import sys
-
-__author__ = "Mattia Pezzano, Federico de Felici, Duccio Lenkowicz"
-__version__ = "0.1"
-
-class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
-    def __init__(self):
-        OpenMayaMPx.MPxFileTranslator.__init__(self)
-
-    def maya_useNewAPI():
-        pass
-
-    def haveWriteMethod(self):
-        return False
-
-    def haveReadMethod(self):
-        return True
-
-    def filter(self):
-        return "*.drc"
-
-    def defaultExtension(self):
-        return "drc"
-
-    def reader(self, fileObject, optionString, accessMode):
-        print(fileObject.fullName())
-        vertices = [OpenMaya.MPoint(10, 10, 10), OpenMaya.MPoint(-10, -10, -10), OpenMaya.MPoint(0, 10, 0)]
-        indices = [0, 1, 2]
-        polyCount = [3]
-
-        fnMesh = OpenMaya.MFnMesh()
-        newMesh = fnMesh.create(vertices, polyCount, indices)
-        fnMesh.updateSurface()
-
-def translatorCreator():
-    return OpenMayaMPx.asMPxPtr(DracoTranslator())
-    
-
-def initializePlugin(mobject):
-    print('init')
-    mplugin = OpenMayaMPx.MFnPlugin(
-        mobject, "Autodesk", '0.1', "Any")
-
-    try:
-        mplugin.registerFileTranslator(
-            'drc',
-            None,
-            translatorCreator)  # ,
-        # kPluginTranslatorDefaultOptions)
-
-    except Exception as e:
-        sys.stderr.write("Failed to register command: %s\n" %
-                        'drc')
-        raise
-
-def uninitializePlugin(mobject):
-print('uninit')
\ No newline at end of file
diff --git a/maya/draco_maya_plugin.py b/maya/draco_maya_plugin.py
new file mode 100644
index 0000000..e400710
--- /dev/null
+++ b/maya/draco_maya_plugin.py
@@ -0,0 +1,98 @@
+import sys
+
+import maya.api.OpenMaya as OpenMaya
+import maya.OpenMayaMPx as OpenMayaMPx
+
+from draco_maya_wrapper import Draco
+
+__author__ = "Mattia Pezzano, Federico de Felici, Duccio Lenkowicz"
+__version__ = "0.1"
+
+class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
+    def __init__(self):
+        OpenMayaMPx.MPxFileTranslator.__init__(self)
+
+    def maya_useNewAPI():
+        pass
+
+    def haveWriteMethod(self):
+        return False
+
+    def haveReadMethod(self):
+        return True
+
+    def filter(self):
+        return "*.drc"
+
+    def defaultExtension(self):
+        return "drc"
+
+    def reader(self, fileObject, optionString, accessMode):
+        drc = Draco()
+        mesh = drc.decode(fileObject.fullName())
+
+        #vertices
+        vertices = []
+        for i in range(0, mesh.vertices_len, 3):
+            vertices.append(OpenMaya.MPoint(mesh.vertices[i], mesh.vertices[i + 1], mesh.vertices[i + 2]))
+
+        #indices
+        indices = []
+        for index in mesh.faces:
+            indices.append(index)
+        poly_count = [3] * mesh.faces_num
+
+        fnMesh = OpenMaya.MFnMesh()
+
+        newMesh = None
+        #uv
+        if mesh.uvs:
+            us = OpenMaya.MFloatArray()
+            vs = OpenMaya.MFloatArray()
+            for i in range(0, mesh.uvs_len, 2):
+                us.append(mesh.uvs[i])
+                vs.append(mesh.uvs[i + 1])
+            #TODO: ensure the mesh actually uses the uvs inside maya
+            newMesh = fnMesh.create(vertices, poly_count, indices, uValues=us, vValues=vs)
+        else:
+            newMesh = fnMesh.create(vertices, poly_count, indices)
+
+        #normals
+        if mesh.normals:
+            normals = []
+            for i in range(0, mesh.normals_len, 3):
+                normals.append(OpenMaya.MFloatVector(mesh.normals[i], mesh.normals[i + 1], mesh.normals[i + 2]))
+            fnMesh.setVertexNormals(normals, range(len(vertices)))
+
+        fnMesh.updateSurface()
+
+        # Assign initial shading group
+        slist = OpenMaya.MGlobal.getSelectionListByName("initialShadingGroup")
+        initialSG = slist.getDependNode(0)
+
+        fnSG = OpenMaya.MFnSet(initialSG)
+        if fnSG.restriction() == OpenMaya.MFnSet.kRenderableOnly:
+            fnSG.addMember(newMesh)
+
+
+def translatorCreator():
+    return OpenMayaMPx.asMPxPtr(DracoTranslator())
+    
+
+def initializePlugin(mobject):
+    print('init')
+    mplugin = OpenMayaMPx.MFnPlugin(
+        mobject, "Autodesk", __version__, "Any")
+
+    try:
+        mplugin.registerFileTranslator(
+            'drc',
+            None,
+            translatorCreator)  
+
+    except Exception as e:
+        sys.stderr.write("Failed to register command: drc\n")
+        raise
+
+def uninitializePlugin(mobject):
+    print('uninit')
\ No newline at end of file