class _LWOBReader(lwob.LWOBReader):
"""Read a Lightwave Object file (*.lwo).
"""
def __init__(self, parent=None):
lwob.LWOBReader.__init__(self)
# Parent node for the Lightwave object
self.parent = parent
# The TriMeshGeom that receives the triangle mesh
self.trimeshgeom = TriMeshGeom()
# The PolyhedronGeom that receives the mesh (if it is no triangle mesh)
self.polyhedrongeom = PolyhedronGeom()
# The generated WorldObject
self.worldobj = None
# The number of surfaces in the file
self.numsurfaces = 0
# A mapping from surface name to material id
# Key: Surface name / Value: Material id (0-based)
self.surface_ids = {}
# Message flags so that warning messages are only output once
self.crv_msg = False
self.patch_msg = False
def handlePNTS(self, points):
"""Handle the points chunk.
Stores the points in the TriMeshGeom.
"""
verts = self.trimeshgeom.verts
verts.resize(len(points))
for i, p in enumerate(points):
verts[i] = p
def handleSRFS(self, names):
"""Handle the surface names chunk.
"""
self.numsurfaces = len(names)
for i, n in enumerate(names):
self.surface_ids[n] = i
def handlePOLS(self, polys):
"""Handle the polygons.
This method creates the actual object. It is assumed that the
points have been read before and are stored in the TriMeshGeom.
It is also assumed that a SRFS chunk was present and numsurfaces
is initialized.
"""
# Assume the mesh is a triangle mesh and initialize the TriMeshGeom
# first. If this fails, use a PolyhedronGeom instead...
if self._initTriMesh(polys):
geom = self.trimeshgeom
else:
# Copy the vertices into the polyhedron geom...
numverts = self.trimeshgeom.verts.size()
self.polyhedrongeom.verts.resize(numverts)
self.trimeshgeom.verts.copyValues(0, numverts, self.polyhedrongeom.verts, 0)
del self.trimeshgeom
# Initialize the polys...
self._initPolyhedron(polys)
geom = self.polyhedrongeom
w = WorldObject(name="lwob", parent=self.parent)
w.setNumMaterials(self.numsurfaces)
w.geom = geom
self.worldobj = w
def _initPolyhedron(self, polys):
"""Initialize the polys of the PolyhedronGeom.
Sets the faces a poly mesh and adds a matid slot with the
material indices.
"""
geom = self.polyhedrongeom
geom.setNumPolys(len(polys))
geom.newVariable("matid", UNIFORM, INT)
matid = geom.slot("matid")
for i, (verts, surfid) in enumerate(polys):
geom.setLoop(i, 0, verts)
matid[i] = max(0, surfid - 1)
def _initTriMesh(self, polys):
"""Initialize the faces of the TriMeshGeom.
Sets the faces of a triangle mesh and adds a matid slot with
the material indices.
If the mesh contains faces with more than 3 vertices the
method aborts and returns False.
"""
faces = self.trimeshgeom.faces
faces.resize(len(polys))
self.trimeshgeom.newVariable("matid", UNIFORM, INT)
matid = self.trimeshgeom.slot("matid")
for i, (verts, surfid) in enumerate(polys):
if len(verts) != 3:
self.trimeshgeom.deleteVariable("matid")
return False
faces[i] = verts
matid[i] = max(0, surfid - 1)
return True
def handleCRVS(self, curves):
if not self.crv_msg:
print("Curves are not yet supported.")
self.crv_msg = True
def handlePCHS(self, patches):
if not self.patch_msg:
print("Patches are not yet supported.")
self.patch_msg = True
def handleSURF(self, surface):
"""Handle a surface chunk.
Currently this just creates a GLMaterial with the base color of
the surface. Everything else is ignored so far.
"""
if surface.name not in self.surface_ids:
raise lwob.LWOBError('Invalid surface name "%s" (name not available in SRFS chunk)' % surface.name)
id = self.surface_ids[surface.name]
col = surface.color
if col == None:
col = (255, 255, 255)
mat = GLMaterial(diffuse=vec3(col) / 255)
self.worldobj.setMaterial(mat, id)
class _LWOBReader(lwob.LWOBReader):
"""Read a Lightwave Object file (*.lwo).
"""
def __init__(self, parent=None):
lwob.LWOBReader.__init__(self)
# Parent node for the Lightwave object
self.parent = parent
# The TriMeshGeom that receives the triangle mesh
self.trimeshgeom = TriMeshGeom()
# The PolyhedronGeom that receives the mesh (if it is no triangle mesh)
self.polyhedrongeom = PolyhedronGeom()
# The generated WorldObject
self.worldobj = None
# The number of surfaces in the file
self.numsurfaces = 0
# A mapping from surface name to material id
# Key: Surface name / Value: Material id (0-based)
self.surface_ids = {}
# Message flags so that warning messages are only output once
self.crv_msg = False
self.patch_msg = False
def handlePNTS(self, points):
"""Handle the points chunk.
Stores the points in the TriMeshGeom.
"""
verts = self.trimeshgeom.verts
verts.resize(len(points))
for i, p in enumerate(points):
verts[i] = p
def handleSRFS(self, names):
"""Handle the surface names chunk.
"""
self.numsurfaces = len(names)
for i, n in enumerate(names):
self.surface_ids[n] = i
def handlePOLS(self, polys):
"""Handle the polygons.
This method creates the actual object. It is assumed that the
points have been read before and are stored in the TriMeshGeom.
It is also assumed that a SRFS chunk was present and numsurfaces
is initialized.
"""
# Assume the mesh is a triangle mesh and initialize the TriMeshGeom
# first. If this fails, use a PolyhedronGeom instead...
if self._initTriMesh(polys):
geom = self.trimeshgeom
else:
# Copy the vertices into the polyhedron geom...
numverts = self.trimeshgeom.verts.size()
self.polyhedrongeom.verts.resize(numverts)
self.trimeshgeom.verts.copyValues(0, numverts,
self.polyhedrongeom.verts, 0)
del self.trimeshgeom
# Initialize the polys...
self._initPolyhedron(polys)
geom = self.polyhedrongeom
w = WorldObject(name="lwob", parent=self.parent)
w.setNumMaterials(self.numsurfaces)
w.geom = geom
self.worldobj = w
def _initPolyhedron(self, polys):
"""Initialize the polys of the PolyhedronGeom.
Sets the faces a poly mesh and adds a matid slot with the
material indices.
"""
geom = self.polyhedrongeom
geom.setNumPolys(len(polys))
geom.newVariable("matid", UNIFORM, INT)
matid = geom.slot("matid")
for i, (verts, surfid) in enumerate(polys):
geom.setLoop(i, 0, verts)
matid[i] = max(0, surfid - 1)
def _initTriMesh(self, polys):
"""Initialize the faces of the TriMeshGeom.
Sets the faces of a triangle mesh and adds a matid slot with
the material indices.
If the mesh contains faces with more than 3 vertices the
method aborts and returns False.
"""
faces = self.trimeshgeom.faces
faces.resize(len(polys))
self.trimeshgeom.newVariable("matid", UNIFORM, INT)
matid = self.trimeshgeom.slot("matid")
for i, (verts, surfid) in enumerate(polys):
if len(verts) != 3:
self.trimeshgeom.deleteVariable("matid")
return False
faces[i] = verts
matid[i] = max(0, surfid - 1)
return True
def handleCRVS(self, curves):
if not self.crv_msg:
print("Curves are not yet supported.")
self.crv_msg = True
def handlePCHS(self, patches):
if not self.patch_msg:
print("Patches are not yet supported.")
self.patch_msg = True
def handleSURF(self, surface):
"""Handle a surface chunk.
Currently this just creates a GLMaterial with the base color of
the surface. Everything else is ignored so far.
"""
if surface.name not in self.surface_ids:
raise lwob.LWOBError(
'Invalid surface name "%s" (name not available in SRFS chunk)'
% surface.name)
id = self.surface_ids[surface.name]
col = surface.color
if col == None:
col = (255, 255, 255)
mat = GLMaterial(diffuse=vec3(col) / 255)
self.worldobj.setMaterial(mat, id)
def createTriMesh(self, parent=None, name=None):
"""Create a triangle mesh from the current set of faces.
This method may only be called if the faces really have only
3 vertices.
Returns the TriMesh object.
"""
# Build lookup tables (so that only the verts that are really
# required are stored in the TriMesh)
#
# Key: Original vertex index - Value: New vertex index
vert_lut = {}
has_normals = True
has_tverts = True
iv = 0
for f in self.faces:
for v,tv,n in f:
if v not in vert_lut:
vert_lut[v] = iv
iv+=1
if tv==None:
has_tverts = False
if n==None:
has_normals = False
numfaces = len(self.faces)
numverts = len(vert_lut)
tm = TriMeshGeom()
tm.verts.resize(numverts)
tm.faces.resize(numfaces)
# Set vertices
for v in vert_lut:
newidx = vert_lut[v]
tm.verts[newidx] = self.verts[v]
# Set faces
idx = 0
for i,f in enumerate(self.faces):
fi = []
for v,tv,n in f:
fi.append(vert_lut[v])
tm.faces[i] = fi
# Create variable N for storing the normals
if has_normals:
tm.newVariable("N", FACEVARYING, NORMAL)
N = tm.slot("N")
idx = 0
for f in self.faces:
for v,tv,n in f:
N[idx] = self.normals[n]
idx += 1
# Set texture vertices
if has_tverts:
tm.newVariable("st", FACEVARYING, FLOAT, 2)
st = tm.slot("st")
idx = 0
for f in self.faces:
for v,tv,n in f:
st[idx] = self.tverts[tv]
idx += 1
obj = TriMesh(name=name, parent=parent)
obj.geom = tm
# Set the materials
self.initMaterial(obj)
return obj
def createTriMesh(self, parent=None, name=None):
"""Create a triangle mesh from the current set of faces.
This method may only be called if the faces really have only
3 vertices.
Returns the TriMesh object.
"""
# Build lookup tables (so that only the verts that are really
# required are stored in the TriMesh)
#
# Key: Original vertex index - Value: New vertex index
vert_lut = {}
has_normals = True
has_tverts = True
iv = 0
for f in self.faces:
for v, tv, n in f:
if v not in vert_lut:
vert_lut[v] = iv
iv += 1
if tv == None:
has_tverts = False
if n == None:
has_normals = False
numfaces = len(self.faces)
numverts = len(vert_lut)
tm = TriMeshGeom()
tm.verts.resize(numverts)
tm.faces.resize(numfaces)
# Set vertices
for v in vert_lut:
newidx = vert_lut[v]
tm.verts[newidx] = self.verts[v]
# Set faces
idx = 0
for i, f in enumerate(self.faces):
fi = []
for v, tv, n in f:
fi.append(vert_lut[v])
tm.faces[i] = fi
# Create variable N for storing the normals
if has_normals:
tm.newVariable("N", FACEVARYING, NORMAL)
N = tm.slot("N")
idx = 0
for f in self.faces:
for v, tv, n in f:
N[idx] = self.normals[n]
idx += 1
# Set texture vertices
if has_tverts:
tm.newVariable("st", FACEVARYING, FLOAT, 2)
st = tm.slot("st")
idx = 0
for f in self.faces:
for v, tv, n in f:
st[idx] = self.tverts[tv]
idx += 1
obj = TriMesh(name=name, parent=parent)
obj.geom = tm
# Set the materials
self.initMaterial(obj)
return obj
