Hackers Magazine 57

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 / dddsimport.py < prev next >

Wrap

Python Source | 2007-01-11 | 16.3 KB | 375 lines

# ***** BEGIN LICENSE BLOCK ***** # Version: MPL 1.1/GPL 2.0/LGPL 2.1 # # The contents of this file are subject to the Mozilla Public License Version # 1.1 (the "License"); you may not use this file except in compliance with # the License. You may obtain a copy of the License at # http://www.mozilla.org/MPL/ # # Software distributed under the License is distributed on an "AS IS" basis, # WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License # for the specific language governing rights and limitations under the # License. # # The Original Code is the Python Computer Graphics Kit. # # The Initial Developer of the Original Code is Matthias Baas. # Portions created by the Initial Developer are Copyright (C) 2004 # the Initial Developer. All Rights Reserved. # # Contributor(s): # # Alternatively, the contents of this file may be used under the terms of # either the GNU General Public License Version 2 or later (the "GPL"), or # the GNU Lesser General Public License Version 2.1 or later (the "LGPL"), # in which case the provisions of the GPL or the LGPL are applicable instead # of those above. If you wish to allow use of your version of this file only # under the terms of either the GPL or the LGPL, and not to allow others to # use your version of this file under the terms of the MPL, indicate your # decision by deleting the provisions above and replace them with the notice # and other provisions required by the GPL or the LGPL. If you do not delete # the provisions above, a recipient may use your version of this file under # the terms of any one of the MPL, the GPL or the LGPL. # # ***** END LICENSE BLOCK ***** # $Id: dddsimport.py,v 1.5 2005/06/07 07:43:03 mbaas Exp $ import os.path import _core from cgtypes import * from scene import getScene from worldobject import WorldObject from trimesh import TriMesh from trimeshgeom import TriMeshGeom from targetcamera import TargetCamera from glpointlight import GLPointLight from gltargetspotlight import GLTargetSpotLight from spotlight3ds import SpotLight3DS from glmaterial import GLMaterial from material3ds import Material3DS, TextureMap3DS import pluginmanager from sl import * TYPE_UNKNOWN = 0 TYPE_AMBIENT = 1 TYPE_OBJECT = 2 TYPE_CAMERA = 3 TYPE_TARGET = 4 TYPE_LIGHT = 5 TYPE_SPOT = 6 GEOM_INIT_NORMALS = 0x01 GEOM_INIT_MATIDS = 0x02 GEOM_INIT_TEXELS = 0x04 GEOM_INIT_FLAGS = 0x08 GEOM_INIT_SMOOTHING = 0x10 GEOM_INIT_ALL = 0xff # DDDSImporter class DDDSImporter: _protocols = ["Import"] # extension def extension(): """Return the file extensions for this format.""" return ["3ds"] extension = staticmethod(extension) # description def description(self): """Return a short description for the file dialog.""" return "3D Studio" description = staticmethod(description) # importFile def importFile(self, filename, flags=GEOM_INIT_ALL, parent=None): """Import a 3DS file.""" self.filename = filename self.ddds = _core.File3ds() self.ddds.load(filename) # f = self.ddds.current_frame f = getScene().timer().frame if f<self.ddds.segment_from: f = self.ddds.segment_from if f>self.ddds.segment_to: f = self.ddds.segment_to self.ddds.eval(f) # Create a dictionary containing all available meshes. # Key is the mesh name. This is used to check if all meshes have # been processed (i.e. if there were corresponding nodes). self.meshes = {} m = self.ddds.meshes() while m!=None: if self.meshes.has_key(m.name): print "Warning: Duplicate mesh names in 3ds file" self.meshes[m.name] = m m = m.next() # Create objects... self.materials = {} self.createObject(self.ddds.nodes(), parent=parent, flags=flags) del self.materials # Create TriMeshes for all left over meshes... for n in self.meshes: mesh = self.meshes[n] tm = TriMeshGeom() mesh.initGeom(tm, flags) worldobj = TriMesh(name = mesh.name, parent=parent) worldobj.geom = tm self.ddds.free() def createObject(self, node, parent=None, flags=GEOM_INIT_ALL): """ \param node (\c Lib3dsNode) Current node \param parent (\c WorldObject) Parent object or None \param flags (\c int) Flags for mesh creation \todo worldtransform als Slot \todo Pivot so setzen wie in 3DS """ if node==None: return worldobj = None auto_insert = True if parent!=None: auto_insert = False # Object? if node.type==TYPE_OBJECT: if node.name!="$$$DUMMY" and node.name!="_Quader01": data = node.object_data mesh = self.ddds.meshByName(node.name) if self.meshes.has_key(mesh.name): del self.meshes[mesh.name] # print "###",node.name,"###" # print "Node matrix:" # print node.matrix # print "Pivot:",data.pivot # print "Mesh matrix:" # print mesh.matrix tm = TriMeshGeom() mesh.initGeom(tm, flags) if parent==None: PT = mat4().translation(-data.pivot) m = node.matrix*PT*mesh.matrix.inverse() else: PT = mat4().translation(-data.pivot) m = node.matrix*PT*mesh.matrix.inverse() # worldtransform von Parent bestimmen... pworldtransform = parent.worldtransform # pworldtransform = parent.localTransform() # p = parent.parent # while p!=None: # pworldtransform = p.localTransform()*pworldtransform # p = p.parent m = pworldtransform.inverse()*m worldobj = TriMesh(name=node.name, # pivot=data.pivot, transform=m, auto_insert=auto_insert) worldobj.geom = tm # Set the materials matnames = mesh.materialNames() worldobj.setNumMaterials(len(matnames)) for i in range(len(matnames)): if matnames[i]=="": material = GLMaterial() # Was the material already instantiated? elif matnames[i] in self.materials: material = self.materials[matnames[i]] else: mat = self.ddds.materialByName(matnames[i]) # print "Material:",matnames[i] # print " self_illum:",mat.self_illum # print " self_ilpct:",mat.self_ilpct material = Material3DS(name = matnames[i], ambient = mat.ambient, diffuse = mat.diffuse, specular = mat.specular, shininess = mat.shininess, shin_strength = mat.shin_strength, use_blur = mat.use_blur, transparency = mat.transparency, falloff = mat.falloff, additive = mat.additive, use_falloff = mat.use_falloff, self_illum = mat.self_illum, self_ilpct = getattr(mat, "self_ilpct", 0), shading = mat.shading, soften = mat.soften, face_map = mat.face_map, two_sided = mat.two_sided, map_decal = mat.map_decal, use_wire = mat.use_wire, use_wire_abs = mat.use_wire_abs, wire_size = mat.wire_size, texture1_map = self._createTexMap(mat.texture1_map), texture2_map = self._createTexMap(mat.texture2_map), opacity_map = self._createTexMap(mat.opacity_map), bump_map = self._createTexMap(mat.bump_map), specular_map = self._createTexMap(mat.specular_map), shininess_map = self._createTexMap(mat.shininess_map), self_illum_map = self._createTexMap(mat.self_illum_map), reflection_map = self._createTexMap(mat.reflection_map) ) self.materials[matnames[i]] = material worldobj.setMaterial(material, i) # Camera? elif node.type==TYPE_CAMERA: cam = self.ddds.cameraByName(node.name) # print "Camera:",node.name # print " Roll:",cam.roll # print node.matrix # Convert the FOV from horizontal to vertical direction # (Todo: What aspect ratio to use?) fov = degrees(atan(480/640.0*tan(radians(cam.fov/2.0))))*2.0 worldobj = TargetCamera(name = node.name, pos = cam.position, target = cam.target, fov = fov, auto_insert = auto_insert) # Light? elif node.type==TYPE_LIGHT: lgt = self.ddds.lightByName(node.name) worldobj = self._createLight(node, lgt, auto_insert) if worldobj!=None and parent!=None: parent.addChild(worldobj) if worldobj!=None: self.createObject(node.childs(), worldobj, flags=flags) else: self.createObject(node.childs(), parent, flags=flags) self.createObject(node.next(), parent, flags=flags) ## protected: def _createLight(self, node, lgt, auto_insert): """Create a light source. Helper method for the createObject() method. """ shadow_size = lgt.shadow_size shadow_bias = lgt.shadow_bias shadow_filter = lgt.shadow_filter # Check if a shadow map is used but the parameters are all zero # (this happens when the Use Global Settings checkbox in MAX # is set). If it happens then use the same default values as MAX if lgt.shadowed and shadow_size==0 and shadow_bias==0 and shadow_filter==0: shadow_size = 256 shadow_bias = 3.0 shadow_filter = 5.0 if lgt.spot_light: worldobj = SpotLight3DS(name = node.name, pos = lgt.position, target = lgt.spot, enabled = not lgt.off, intensity = lgt.multiplier, see_cone = lgt.see_cone, roll = lgt.roll, outer_range = lgt.outer_range, inner_range = lgt.inner_range, attenuation = lgt.attenuation, rectangular_spot = lgt.rectangular_spot, shadowed = lgt.shadowed, shadow_bias = shadow_bias, shadow_filter = shadow_filter, shadow_size = shadow_size, spot_aspect = lgt.spot_aspect, use_projector = lgt.use_projector, projector = lgt.projector, overshoot = lgt.spot_overshoot, ray_shadows = lgt.ray_shadows, ray_bias = lgt.ray_bias, hotspot = lgt.hotspot, falloff = lgt.falloff, color = lgt.color, auto_insert = auto_insert) # worldobj = GLTargetSpotLight(name = node.name, # pos = lgt.position, # target = lgt.spot, # cutoff = lgt.falloff/2, # intensity = lgt.multiplier, # diffuse = lgt.color, # auto_insert = auto_insert) # if lgt.shadowed: # worldobj.shadowmap = (lgt.shadow_size, 1.0, 0.001) else: worldobj = SpotLight3DS(name = node.name, pos = lgt.position, target = lgt.spot, enabled = not lgt.off, intensity = lgt.multiplier, see_cone = lgt.see_cone, roll = lgt.roll, outer_range = lgt.outer_range, inner_range = lgt.inner_range, attenuation = lgt.attenuation, rectangular_spot = lgt.rectangular_spot, shadowed = False, shadow_bias = shadow_bias, shadow_filter = shadow_filter, shadow_size = shadow_size, spot_aspect = lgt.spot_aspect, use_projector = lgt.use_projector, projector = lgt.projector, overshoot = True, ray_shadows = lgt.ray_shadows, ray_bias = lgt.ray_bias, hotspot = lgt.hotspot, falloff = lgt.falloff, color = lgt.color, auto_insert = auto_insert) # worldobj = GLPointLight(name = node.name, # pos = lgt.position, # intensity = lgt.multiplier, # diffuse = lgt.color, # auto_insert = auto_insert) return worldobj def _createTexMap(self, texmap): """Create a TextureMap3DS object (or None). \param texmap (\c Lib3dsTextureMap) Texture map """ if texmap.name=="": return None dir = os.path.dirname(self.filename) dir = os.path.abspath(dir) return TextureMap3DS(name=os.path.join(dir, texmap.name), flags = texmap.flags, percent = texmap.percent, blur = texmap.blur, scale = texmap.scale, offset = texmap.offset, rotation = texmap.rotation, tint1 = texmap.tint_1, tint2 = texmap.tint_2, tintr = texmap.tint_r, tintg = texmap.tint_g, tintb = texmap.tint_b) ###################################################################### # Register the Importer class as a plugin class if hasattr(_core, "File3ds"): pluginmanager.register(DDDSImporter)