GitHub-Proxy/draco
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/google/draco synced 2025-08-12 08:29:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed new api

Browse Source
This commit is contained in:
MattiaPezzanoAiv 2018-05-15 17:08:28 +02:00
parent fae76dd191
commit 8c88b5b2c8
1 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
maya/draco_maya_plugin.py
View File

@ -13,7 +13,7 @@ class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
OpenMayaMPx.MPxFileTranslator.__init__(self) OpenMayaMPx.MPxFileTranslator.__init__(self)
def maya_useNewAPI(): def maya_useNewAPI():
pass return True
def haveWriteMethod(self): def haveWriteMethod(self):
return True return True
@ -30,7 +30,7 @@ class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
def writer(self, fileObject, optionString, accessMode): def writer(self, fileObject, optionString, accessMode):
# Get the selection and create a selection list of all the nodes meshes # Get the selection and create a selection list of all the nodes meshes
selection = OpenMaya.MSelectionList() selection = OpenMaya.MSelectionList()
OpenMaya.MGlobal.getActiveSelectionList(selection) selection = OpenMaya.MGlobal.getActiveSelectionList()
# Create an MItSelectionList class to iterate over the selection # Create an MItSelectionList class to iterate over the selection
# Use the MFn class to as a filter to filter node types # Use the MFn class to as a filter to filter node types
@ -49,11 +49,11 @@ class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
# get dag path of current iterated selection # get dag path of current iterated selection
dagPath = OpenMaya.MDagPath() dagPath = OpenMaya.MDagPath()
iter.getDagPath(dagPath) dagPath = iter.getDagPath()
# get the selection as an MObject # get the selection as an MObject
mObj = OpenMaya.MObject() mObj = OpenMaya.MObject()
iter.getDependNode(mObj) mObj = iter.getDependNode()
# create iterator of current mesh polygons # create iterator of current mesh polygons
polygonsIterator = OpenMaya.MItMeshPolygon(mObj) polygonsIterator = OpenMaya.MItMeshPolygon(mObj)
@ -66,26 +66,33 @@ class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
# Get current polygons vertices # Get current polygons vertices
verts = OpenMaya.MIntArray() verts = OpenMaya.MIntArray()
polygonsIterator.getVertices(verts) verts = polygonsIterator.getVertices()
normalIndices = []
# Append the current polygons vertex indices # Append the current polygons vertex indices
for i in range(verts.length()): for i in range(len(verts)):
vertexList.append(verts[i]) vertexList.append(verts[i])
# NOT VALID CODE
# for i in range(0,len(vertexList), 3):
# need vertex this is a float
# vertex = OpenMaya.MFloatPoint(vertexList[i], vertexList[i+1], vertexList[i+2])
# normalIndices.append(polygonsIterator.normalIndex(vertex)) #return the index in the normals buffer
# Get current polygons edges # Get current polygons edges
edges = OpenMaya.MIntArray() edges = OpenMaya.MIntArray()
polygonsIterator.getEdges(edges) edges = polygonsIterator.getEdges()
# Append the current polygons edge indices # Append the current polygons edge indices
for i in range(edges.length()): for i in range(len(edges)):
edgeList.append(edges[i]) edgeList.append(edges[i])
# Get current polygons connected faces # Get current polygons connected faces
indexConnectedFaces = OpenMaya.MIntArray() indexConnectedFaces = OpenMaya.MIntArray()
polygonsIterator.getConnectedFaces(indexConnectedFaces) indexConnectedFaces = polygonsIterator.getConnectedFaces()
# Append the connected polygons indices # Append the connected polygons indices
for i in range(indexConnectedFaces.length()): for i in range(len(indexConnectedFaces)):
connectedPolyList.append(indexConnectedFaces[i]) connectedPolyList.append(indexConnectedFaces[i])
# Get current polygons triangles # Get current polygons triangles
@ -94,16 +101,15 @@ class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
space = OpenMaya.MSpace.kObject space = OpenMaya.MSpace.kObject
# Get the vertices and vertex positions of all the triangles in the current face's triangulation. # Get the vertices and vertex positions of all the triangles in the current face's triangulation.
polygonsIterator.getTriangles(pointArray, intArray, space) pointArray, intArray = polygonsIterator.getTriangles(space)
# Append vertices that are part of the triangles # Append vertices that are part of the triangles
for i in range(intArray.length()): for i in range(len(intArray)):
polytriVertsList.append(intArray[i]) polytriVertsList.append(intArray[i])
# next poligon # next poligon
polygonsIterator.next() polygonsIterator.next(None) # idk what arguments i need to pass here
#TODO: CREATE DRACOMESH
drcMesh = DrcMesh() drcMesh = DrcMesh()
drcMesh.faces_num = len(polytriVertsList) / 3 drcMesh.faces_num = len(polytriVertsList) / 3
drcMesh.faces_len = len(polytriVertsList) drcMesh.faces_len = len(polytriVertsList)
@ -113,7 +119,8 @@ class DracoTranslator(OpenMayaMPx.MPxFileTranslator):
drcMesh.vertices_len = len(vertexList) drcMesh.vertices_len = len(vertexList)
drcMesh.vertices = vertexList drcMesh.vertices = vertexList
#SAVE FILE draco = Draco()
draco.encode(drcMesh, fileObject)
# print data for current selection being iterated on, # print data for current selection being iterated on,
print ("Object name: {}".format(dagPath.fullPathName())) print ("Object name: {}".format(dagPath.fullPathName()))