home *** CD-ROM | disk | FTP | other *** search
/ Hackers Magazine 57 / CdHackersMagazineNr57.iso / Software / Multimedia / k3d-setup-0.7.11.0.exe / lib / site-packages / cgkit / lwobimport.py < prev    next >
Encoding:
Python Source  |  2007-01-11  |  7.3 KB  |  215 lines
  1. # ***** BEGIN LICENSE BLOCK *****
  2. # Version: MPL 1.1/GPL 2.0/LGPL 2.1
  3. #
  4. # The contents of this file are subject to the Mozilla Public License Version
  5. # 1.1 (the "License"); you may not use this file except in compliance with
  6. # the License. You may obtain a copy of the License at
  7. # http://www.mozilla.org/MPL/
  8. #
  9. # Software distributed under the License is distributed on an "AS IS" basis,
  10. # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11. # for the specific language governing rights and limitations under the
  12. # License.
  13. #
  14. # The Original Code is the Python Computer Graphics Kit.
  15. #
  16. # The Initial Developer of the Original Code is Matthias Baas.
  17. # Portions created by the Initial Developer are Copyright (C) 2004
  18. # the Initial Developer. All Rights Reserved.
  19. #
  20. # Contributor(s):
  21. #
  22. # Alternatively, the contents of this file may be used under the terms of
  23. # either the GNU General Public License Version 2 or later (the "GPL"), or
  24. # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  25. # in which case the provisions of the GPL or the LGPL are applicable instead
  26. # of those above. If you wish to allow use of your version of this file only
  27. # under the terms of either the GPL or the LGPL, and not to allow others to
  28. # use your version of this file under the terms of the MPL, indicate your
  29. # decision by deleting the provisions above and replace them with the notice
  30. # and other provisions required by the GPL or the LGPL. If you do not delete
  31. # the provisions above, a recipient may use your version of this file under
  32. # the terms of any one of the MPL, the GPL or the LGPL.
  33. #
  34. # ***** END LICENSE BLOCK *****
  35. # $Id: lwobimport.py,v 1.1 2006/03/12 22:41:42 mbaas Exp $
  36.  
  37. import os.path, sys
  38. from cgtypes import *
  39. from worldobject import WorldObject
  40. from trimesh import TriMesh
  41. from trimeshgeom import TriMeshGeom
  42. from polyhedron import Polyhedron
  43. from polyhedrongeom import PolyhedronGeom
  44. import pluginmanager
  45. import cmds
  46. from cgkit.all import UNIFORM, INT, GLMaterial
  47. import lwob
  48.  
  49.  
  50. class _LWOBReader(lwob.LWOBReader):
  51.     """Read a Lightwave Object file (*.lwo).
  52.     """
  53.     
  54.     def __init__(self, parent=None):
  55.         lwob.LWOBReader.__init__(self)
  56.  
  57.         # Parent node for the Lightwave object
  58.         self.parent = parent
  59.  
  60.         # The TriMeshGeom that receives the triangle mesh
  61.         self.trimeshgeom = TriMeshGeom()
  62.         # The PolyhedronGeom that receives the mesh (if it is no triangle mesh)
  63.         self.polyhedrongeom = PolyhedronGeom()
  64.         # The generated WorldObject
  65.         self.worldobj = None
  66.         # The number of surfaces in the file
  67.         self.numsurfaces = 0
  68.         # A mapping from surface name to material id
  69.         # Key: Surface name / Value: Material id (0-based)
  70.         self.surface_ids = {}
  71.  
  72.         # Message flags so that warning messages are only output once
  73.         self.crv_msg = False
  74.         self.patch_msg = False
  75.  
  76.     def handlePNTS(self, points):
  77.         """Handle the points chunk.
  78.  
  79.         Stores the points in the TriMeshGeom.
  80.         """
  81.         verts = self.trimeshgeom.verts
  82.         verts.resize(len(points))
  83.         for i,p in enumerate(points):
  84.             verts[i] = p
  85.     
  86.     def handleSRFS(self, names):
  87.         """Handle the surface names chunk.
  88.         """
  89.         self.numsurfaces = len(names)
  90.         for i,n in enumerate(names):
  91.             self.surface_ids[n] = i
  92.     
  93.     def handlePOLS(self, polys):
  94.         """Handle the polygons.
  95.  
  96.         This method creates the actual object. It is assumed that the
  97.         points have been read before and are stored in the TriMeshGeom.
  98.         It is also assumed that a SRFS chunk was present and numsurfaces
  99.         is initialized.
  100.         """
  101.         # Assume the mesh is a triangle mesh and initialize the TriMeshGeom
  102.         # first. If this fails, use a PolyhedronGeom instead...
  103.         if self._initTriMesh(polys):
  104.             geom = self.trimeshgeom
  105.         else:
  106.             # Copy the vertices into the polyhedron geom...
  107.             numverts = self.trimeshgeom.verts.size()
  108.             self.polyhedrongeom.verts.resize(numverts)
  109.             self.trimeshgeom.verts.copyValues(0, numverts, self.polyhedrongeom.verts, 0)
  110.             del self.trimeshgeom
  111.             # Initialize the polys...
  112.             self._initPolyhedron(polys)
  113.             geom = self.polyhedrongeom
  114.  
  115.         w = WorldObject(name="lwob", parent=self.parent)
  116.         w.setNumMaterials(self.numsurfaces)
  117.         w.geom = geom
  118.         self.worldobj = w
  119.  
  120.     def _initPolyhedron(self, polys):
  121.         """Initialize the polys of the PolyhedronGeom.
  122.  
  123.         Sets the faces a poly mesh and adds a matid slot with the
  124.         material indices.
  125.         """
  126.         geom = self.polyhedrongeom
  127.         geom.setNumPolys(len(polys))
  128.         geom.newVariable("matid", UNIFORM, INT)
  129.         matid = geom.slot("matid")
  130.         for i,(verts,surfid) in enumerate(polys):
  131.             geom.setLoop(i, 0, verts)
  132.             matid[i] = max(0, surfid-1)
  133.  
  134.     def _initTriMesh(self, polys):
  135.         """Initialize the faces of the TriMeshGeom.
  136.  
  137.         Sets the faces of a triangle mesh and adds a matid slot with
  138.         the material indices.
  139.         If the mesh contains faces with more than 3 vertices the
  140.         method aborts and returns False.
  141.         """
  142.         faces = self.trimeshgeom.faces
  143.         faces.resize(len(polys))
  144.         self.trimeshgeom.newVariable("matid", UNIFORM, INT)
  145.         matid = self.trimeshgeom.slot("matid")
  146.         for i,(verts,surfid) in enumerate(polys):
  147.             if len(verts)!=3:
  148.                 self.trimeshgeom.deleteVariable("matid")
  149.                 return False
  150.             faces[i] = verts
  151.             matid[i] = max(0, surfid-1)
  152.             
  153.         return True
  154.         
  155.  
  156.     def handleCRVS(self, curves):
  157.         if not self.crv_msg:
  158.             print "Curves are not yet supported."
  159.             self.crv_msg = True
  160.     
  161.     def handlePCHS(self, patches):
  162.         if not self.patch_msg:
  163.             print "Patches are not yet supported."
  164.             self.patch_msg = True
  165.     
  166.     def handleSURF(self, surface):
  167.         """Handle a surface chunk.
  168.  
  169.         Currently this just creates a GLMaterial with the base color of
  170.         the surface. Everything else is ignored so far.
  171.         """
  172.         if surface.name not in self.surface_ids:
  173.             raise lwob.LWOBError, 'Invalid surface name "%s" (name not available in SRFS chunk)'%surface.name
  174.         
  175.         id = self.surface_ids[surface.name]
  176.         
  177.         col = surface.color
  178.         if col==None:
  179.             col = (255,255,255)
  180.             
  181.         mat = GLMaterial(diffuse=vec3(col)/255)
  182.         self.worldobj.setMaterial(mat, id)
  183.  
  184.  
  185. # LWOBImporter
  186. class LWOBImporter:
  187.  
  188.     _protocols = ["Import"]
  189.  
  190.     # extension
  191.     def extension():
  192.         """Return the file extensions for this format."""
  193.         return ["lwo"]
  194.     extension = staticmethod(extension)
  195.  
  196.     # description
  197.     def description(self):
  198.         """Return a short description for the file dialog."""
  199.         return "Lightwave object file"
  200.     description = staticmethod(description)
  201.  
  202.     # importFile
  203.     def importFile(self, filename, parent=None):
  204.         """Import an LWOB file."""
  205.  
  206.         f = file(filename, "rb")
  207.         reader = _LWOBReader(parent=parent)
  208.         reader.read(f)
  209.         f.close()
  210.  
  211. ######################################################################
  212.  
  213. # Register the Importer class as a plugin class
  214. pluginmanager.register(LWOBImporter)
  215.