File: export_obj.py

package info (click to toggle)
opensubdiv 3.6.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 75,132 kB
  • sloc: cpp: 137,820; python: 1,069; objc: 412; javascript: 361; lisp: 216; ansic: 170; makefile: 24
file content (108 lines) | stat: -rw-r--r-- 3,354 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
#
#   Copyright 2013 Pixar
#
#   Licensed under the Apache License, Version 2.0 (the "Apache License")
#   with the following modification; you may not use this file except in
#   compliance with the Apache License and the following modification to it:
#   Section 6. Trademarks. is deleted and replaced with:
#
#   6. Trademarks. This License does not grant permission to use the trade
#      names, trademarks, service marks, or product names of the Licensor
#      and its affiliates, except as required to comply with Section 4(c) of
#      the License and to reproduce the content of the NOTICE file.
#
#   You may obtain a copy of the Apache License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the Apache License with the above modification is
#   distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
#   KIND, either express or implied. See the Apache License for the specific
#   language governing permissions and limitations under the Apache License.
#

import maya.OpenMaya as OpenMaya

selectionList = OpenMaya.MSelectionList()
OpenMaya.MGlobal.getActiveSelectionList(selectionList)

path = OpenMaya.MDagPath()
selectionList.getDagPath(0, path)

meshFn = OpenMaya.MFnMesh(path)

points = OpenMaya.MPointArray()
normals = OpenMaya.MFloatVectorArray()
u = OpenMaya.MFloatArray()
v = OpenMaya.MFloatArray()
meshFn.getPoints(points)
meshFn.getVertexNormals(True, normals);
meshFn.getUVs(u, v, "map1")
vertexCount = OpenMaya.MIntArray()
vertexList = OpenMaya.MIntArray()
meshFn.getVertices(vertexCount, vertexList)
edgeIds = OpenMaya.MUintArray()
edgeCreaseData = OpenMaya.MDoubleArray()
vtxIds = OpenMaya.MUintArray()
vtxCreaseData = OpenMaya.MDoubleArray()
try:
    meshFn.getCreaseEdges(edgeIds, edgeCreaseData)
except:
    pass
try:
    meshFn.getCreaseVertices(vtxIds, vtxCreaseData)
except:
    pass

f = open('out.obj', 'w')

for i in range(0,points.length()):
    f.write('v %f %f %f\n' % (points[i].x, points[i].y, points[i].z))

for i in range(0,u.length()):
    f.write('vt %f %f \n' % (u[i], v[i]))

for i in range(0,normals.length()):
    f.write('vn %f %f %f\n' % (normals[i].x, normals[i].y, normals[i].z))

f.write('s off\n')

vindex = 0
for i in range(0,vertexCount.length()):
    f.write('f')
    for j in range(0, vertexCount[i]):
        v = vertexList[vindex] + 1

        sutil = OpenMaya.MScriptUtil()
        sutil.createFromInt(0)
        uvptr = sutil.asIntPtr()
        meshFn.getPolygonUVid(i, j, uvptr, 'map1')
        uv = sutil.getInt(uvptr) + 1

        f.write(' %d/%d/%d' % (v, uv, v))
        vindex = vindex+1
    f.write('\n')

if vtxIds.length() > 0:
    f.write('t corner %d/%d/0' % (vtxIds.length(), vtxIds.length()))
    for i in range(0,vtxIds.length()):
        f.write(' %d' % vtxIds[i])
    for i in range(0,vtxCreaseData.length()):
        f.write(' %f' % vtxCreaseData[i])
    f.write('\n')

for i in range(0, edgeIds.length()):
    edgeIt = OpenMaya.MItMeshEdge(path)
    dummy = OpenMaya.MScriptUtil().asIntPtr()
    edgeIt.setIndex(edgeIds[i], dummy)
    faceList = OpenMaya.MIntArray()
    edgeIt.getConnectedFaces(faceList)
    vid0 = edgeIt.index(0)
    vid1 = edgeIt.index(1)

    f.write('t crease 2/1/0 %d %d %f\n' % (vid0, vid1, edgeCreaseData[i]))


f.close()