# -------------------------------------------------------------------------
# Illusoft Collada 1.4 plugin for Blender version 0.2.65
# --------------------------------------------------------------------------
# ***** BEGIN GPL LICENSE BLOCK *****
#
# Copyright (C) 2006: Illusoft - colladablender@illusoft.com
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License,
# or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ***** END GPL LICENCE BLOCK *****
# --------------------------------------------------------------------------
from xml.dom.minidom import *
import xmlUtils
from cutils import *
from datetime import *
# The number of decimals to export floats to
ROUND = 5
# TODO: finish DaeDocument
class DaeDocument(object):
def __init__(self, debugM = False):
global debugMode
debugMode = debugM
self.colladaVersion = '1.4.0'
self.version = ''
self.xmlns = ''
self.asset = DaeAsset()
self.extras = []
# create all the libraries
self.animationsLibrary = DaeLibrary(DaeSyntax.LIBRARY_ANIMATIONS,DaeAnimation,DaeSyntax.ANIMATION)
self.animationClipsLibrary = DaeLibrary(DaeSyntax.LIBRARY_ANIMATION_CLIPS,DaeAnimationClip,DaeSyntax.ANIMATION_CLIP)
self.camerasLibrary = DaeLibrary(DaeSyntax.LIBRARY_CAMERAS,DaeCamera,DaeSyntax.CAMERA)
self.controllersLibrary = DaeLibrary(DaeSyntax.LIBRARY_CONTROLLERS,DaeController,DaeSyntax.CONTROLLER)
self.effectsLibrary = DaeLibrary(DaeSyntax.LIBRARY_EFFECTS,DaeFxEffect,DaeFxSyntax.EFFECT)
self.geometriesLibrary = DaeLibrary(DaeSyntax.LIBRARY_GEOMETRIES,DaeGeometry,DaeSyntax.GEOMETRY)
self.imagesLibrary = DaeLibrary(DaeSyntax.LIBRARY_IMAGES, DaeImage, DaeSyntax.IMAGE)
self.lightsLibrary = DaeLibrary(DaeSyntax.LIBRARY_LIGHTS,DaeLight,DaeSyntax.LIGHT)
self.materialsLibrary = DaeLibrary(DaeSyntax.LIBRARY_MATERIALS,DaeFxMaterial,DaeFxSyntax.MATERIAL)
self.nodesLibrary = DaeLibrary(DaeSyntax.LIBRARY_NODES, DaeNode, DaeSyntax.NODE)
self.visualScenesLibrary = DaeLibrary(DaeSyntax.LIBRARY_VISUAL_SCENES,DaeVisualScene,DaeSyntax.VISUAL_SCENE)
# Physics Support
self.physicsMaterialsLibrary = DaeLibrary(DaeSyntax.LIBRARY_PHYSICS_MATERIALS, DaePhysicsMaterial, DaePhysicsSyntax.PHYSICS_MATERIAL)
self.physicsScenesLibrary = DaeLibrary(DaeSyntax.LIBRARY_PHYSICS_SCENES, DaePhysicsScene, DaePhysicsSyntax.PHYSICS_SCENE)
self.physicsModelsLibrary = DaeLibrary(DaeSyntax.LIBRARY_PHYSICS_MODELS, DaePhysicsModel, DaePhysicsSyntax.PHYSICS_MODEL)
self.scene = None
self.physicsScene = None
def LoadDocumentFromFile(self, filename):
global debugMode
# Build DOM tree
doc = parse( filename )
# Get COLLADA element
colladaNode = doc.documentElement
# Get Attributes
self.version = colladaNode.getAttribute(DaeSyntax.VERSION)
if not IsVersionOk(self.version, self.colladaVersion):
Debug.Debug('The version of the file (%s) is older then the version supported by this plugin(%s).'%(self.version, self.colladaVersion),'ERROR')
doc.unlink()
return
self.xmlns = colladaNode.getAttribute(DaeSyntax.XMLNS)
# get the assets element
self.asset.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.ASSET))
# get the extra elements
self.extras = CreateObjectsFromXml(self,colladaNode,DaeSyntax.EXTRA,DaeExtra)
# parse all the libraries
self.imagesLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_IMAGES))
self.animationsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_IMAGES))
self.animationClipsLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_ANIMATION_CLIPS))
self.camerasLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_CAMERAS))
self.controllersLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_CONTROLLERS))
self.effectsLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_EFFECTS))
self.geometriesLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_GEOMETRIES))
self.lightsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_LIGHTS))
self.materialsLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_MATERIALS))
self.nodesLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_NODES))
self.visualScenesLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_VISUAL_SCENES))
self.physicsMaterialsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_PHYSICS_MATERIALS))
self.physicsModelsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_PHYSICS_MODELS))
self.physicsScenesLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_PHYSICS_SCENES))
# Get the sceneNodes
sceneNodes = colladaNode.getElementsByTagName(DaeSyntax.SCENE)
# Get the scene
sceneNode = xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.SCENE)
if sceneNode != None:
scene = DaeScene()
scene.LoadFromXml(self, sceneNode)
self.scene = scene
doc.unlink()
if debugMode:
Debug.Debug('Directly exporting this DaeDocument...','DEBUG')
self.SaveDocumentToFile(filename+'_out.dae')
def SaveDocumentToFile(self, filename):
self.version = '1.4.0'
self.xmlns = 'http://www.collada.org/2005/11/COLLADASchema'
colladaNode = Element(DaeSyntax.COLLADA)
colladaNode.setAttribute(DaeSyntax.VERSION, self.version)
colladaNode.setAttribute(DaeSyntax.XMLNS, self.xmlns)
colladaNode.appendChild(self.asset.SaveToXml(self))
# add the labraries
AppendChild(self,colladaNode,self.animationsLibrary)
AppendChild(self,colladaNode,self.animationClipsLibrary)
AppendChild(self,colladaNode,self.camerasLibrary)
AppendChild(self,colladaNode,self.controllersLibrary)
AppendChild(self,colladaNode,self.effectsLibrary)
AppendChild(self,colladaNode,self.geometriesLibrary)
AppendChild(self,colladaNode,self.imagesLibrary)
AppendChild(self,colladaNode,self.lightsLibrary)
AppendChild(self,colladaNode,self.materialsLibrary)
AppendChild(self,colladaNode,self.nodesLibrary)
AppendChild(self,colladaNode,self.visualScenesLibrary)
AppendChild(self,colladaNode,self.physicsMaterialsLibrary)
AppendChild(self,colladaNode,self.physicsModelsLibrary)
AppendChild(self,colladaNode,self.physicsScenesLibrary)
AppendChild(self,colladaNode,self.scene)
# write xml to the file
fileref = open(filename, 'w')
fileref.write(xmlUtils.ToXml(colladaNode))
fileref.flush()
fileref.close()
colladaNode.unlink()
def GetItemCount(self):
return (##self.animationClipsLibrary.GetItemCount()+
##self.animationsLibrary.GetItemCount()+
##self.camerasLibrary.GetItemCount()+
##self.controllersLibrary.GetItemCount()+
##self.effectsLibrary.GetItemCount()+
self.geometriesLibrary.GetItemCount()+
self.lightsLibrary.GetItemCount()+
##self.materialsLibrary.GetItemCount()+
self.nodesLibrary.GetItemCount()##+
##self.visualScenesLibrary.GetItemCount()
)
def __str__(self):
return '%s version: %s, xmlns: %s, asset: %s, extras: %s, scene: %s'%(type(self), self.version, self.xmlns, self.asset, self.extras, self.scene)
class DaeEntity(object):
def __init__(self):
self.syntax = 'UNKNOWN'
def LoadFromXml(self, daeDocument, xmlNode):
Debug.Debug('DaeEntity: Override this method for %s'%(type(self)),'WARNING')
def SaveToXml(self, daeDocument):
node = Element(self.syntax)
return node
def GetType(self):
return self.syntax
class DaeElement(DaeEntity):
def __init__(self):
super(DaeElement,self).__init__()
self.id = ''
self.name = ''
def LoadFromXml(self,daeDocument, xmlNode):
if xmlNode is None:
return
self.id = xmlNode.getAttribute(DaeSyntax.ID)
self.name = xmlNode.getAttribute(DaeSyntax.NAME)
def SaveToXml(self, daeDocument):
node = super(DaeElement,self).SaveToXml(daeDocument)
SetAttribute(node,DaeSyntax.ID,StripString(self.id))
SetAttribute(node,DaeSyntax.NAME, StripString(self.name))
return node
def __str__(self):
return super(DaeElement,self).__str__()+'id: %s, name: %s'%(self.id, self.name)
# TODO: finish DaeLibrary
class DaeLibrary(DaeElement):
def __init__(self, syntax, objectType, objectSyntax):
super(DaeLibrary,self).__init__()
self.extras = []
self.asset = None
self.items = []
self.__objectSyntax = objectSyntax
self.syntax = syntax
self.__objectType = objectType
def LoadFromXml(self,daeDocument, xmlNode):
if xmlNode is None:
return
super(DaeLibrary,self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
self.items = CreateObjectsFromXml(daeDocument, xmlNode, self.__objectSyntax, self.__objectType)
def SaveToXml(self,daeDocument):
if len(self.items) > 0:
node = super(DaeLibrary,self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the library_items
AppendChilds(daeDocument,node,self.items)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
else:
return None
def GetItemCount(self):
return len(self.items)
def FindObject(self,url):
for i in self.items:
if i.id == url:
return i
return None
def AddItem(self,item):
self.items.append(item)
def __str__(self):
return super(DaeLibrary,self).__str__() + 'extras: %s, asset: %s, items: %s'%(self.extras, self.asset, self.items)
class DaeAsset(DaeEntity):
def __init__(self):
super(DaeAsset,self).__init__()
self.contributors = []
self.created = None
self.modified = None
self.revision = None
self.title = None
self.subject = None
self.keywords = []
self.unit = DaeUnit()
self.upAxis = 'Y_UP'
self.syntax = DaeSyntax.ASSET
def LoadFromXml(self, daeDocument, xmlNode):
if xmlNode is None:
return
# Get the contributor(s)
self.contributors = CreateObjectsFromXml(daeDocument, xmlNode,DaeSyntax.CONTRIBUTOR,DaeContributor)
# Get created
self.created = xmlUtils.ReadDateTime(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.CREATED))
# Get modified
self.modified = xmlUtils.ReadDateTime(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.MODIFIED))
# Get revision
self.revision = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.REVISION))
# Get title
self.title = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.TITLE))
# Get subject
self.subject = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.SUBJECT))
# Get keywords
self.keywords = xmlUtils.GetStringArrayFromNodes(xmlNode.getElementsByTagName(DaeSyntax.KEYWORDS))
# Get Unit
self.unit.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.UNIT))
# Get upAxis
self.upAxis = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.UP_AXIS))
def SaveToXml(self, daeDocument):
node = Element(DaeSyntax.ASSET)
AppendChilds(daeDocument,node, self.contributors)
AppendTextChild(node, DaeSyntax.CREATED, self.created)
AppendTextChild(node, DaeSyntax.MODIFIED, self.modified)
AppendTextChild(node, DaeSyntax.REVISION, self.revision)
AppendTextChild(node, DaeSyntax.TITLE, self.title)
AppendTextChild(node, DaeSyntax.SUBJECT, self.subject)
AppendChild(daeDocument, node, self.unit)
AppendTextChild(node, DaeSyntax.UP_AXIS, self.upAxis)
return node
def __str__(self):
return super(DaeAsset,self).__str__()+'contributors: %s, created: %s, modified: %s, revision: %s, title: %s, subject: %s, keywords: %s, unit: %s, upAxis: %s'%(self.contributors, self.created, self.modified, self.revision, self.title, self.subject, self.keywords, self.unit, self.upAxis)
# TODO: finish DaeScene
class DaeScene(DaeEntity):
def __init__(self):
super(DaeScene,self).__init__()
self.extras = []
self.iVisualScenes = []
self.iPhysicsScenes = []
self.syntax = DaeSyntax.SCENE
def LoadFromXml(self, daeDocument, xmlNode):
if xmlNode is None:
return
self.iVisualScenes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_VISUAL_SCENE, DaeVisualSceneInstance)
self.iPhysicsScenes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_PHYSICS_SCENE, DaePhysicsSceneInstance)
def SaveToXml(self, daeDocument):
node = super(DaeScene,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.iVisualScenes)
AppendChilds(daeDocument, node, self.iPhysicsScenes)
return node
def GetVisualScenes(self):
result = []
for i in self.iVisualScenes:
result.append(i.object)
return result
def GetPhysicsScenes(self):
result = []
for i in self.iPhysicsScenes:
result.append(i.object)
return result
def __str__(self):
return super(DaeScene,self).__str__()+'extras: %s, visualScenes: %s, physicsScenes: %s'%(self.extras, self.iVisualScenes, self.iPhysicsScenes)
class DaeUnit(DaeEntity):
def __init__(self):
super(DaeUnit,self).__init__()
self.name = 'meter'
self.meter = 1.0
self.syntax = DaeSyntax.UNIT
def LoadFromXml(self, daeDocument, xmlNode):
if xmlNode is None:
return
name = xmlNode.getAttribute(DaeSyntax.NAME)
if name != '':
self.name = name
meter = xmlNode.getAttribute(DaeSyntax.METER)
if meter != '':
self.meter = float(meter)
def SaveToXml(self,daeDocument):
node = super(DaeUnit, self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.METER, self.meter)
SetAttribute(node, DaeSyntax.NAME, self.name)
return node
def __str__(self):
return super(DaeUnit, self).__str__()+' name: %s, meter: %s'%(self.name, self.meter)
class DaeContributor(DaeEntity):
def __init__(self):
super(DaeContributor, self).__init__()
self.author = ''
self.authoringTool = ''
self.comments = ''
self.copyright = ''
self.sourceData = ''
self.syntax = DaeSyntax.CONTRIBUTOR
def LoadFromXml(self, daeDocument, xmlNode):
self.author = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.AUTHOR))
self.authoringTool = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.AUTHORING_TOOL))
self.comments = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.COMMENTS))
self.copyright = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.COPYRIGHT))
self.sourceData = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.SOURCE_DATA))
def SaveToXml(self, daeDocument):
node = super(DaeContributor, self).SaveToXml(daeDocument)
AppendTextChild(node, DaeSyntax.AUTHOR, self.author)
AppendTextChild(node, DaeSyntax.AUTHORING_TOOL, self.authoringTool)
AppendTextChild(node, DaeSyntax.COMMENTS, self.comments)
AppendTextChild(node, DaeSyntax.COPYRIGHT, self.copyright)
AppendTextChild(node, DaeSyntax.SOURCE_DATA, self.sourceData)
return node
def __str__(self):
return super(DaeContributor,self).__str__() + 'author: %s, authoring_tool: %s, comments: %s, copyright: %s, sourceData: %s'%(self.author, self.authoringTool, self.comments, self.copyright, self.sourceData)
class DaeAnimation(DaeElement):
pass
class DaeAnimationClip(DaeElement):
pass
class DaeCamera(DaeElement):
def __init__(self):
super(DaeCamera,self).__init__()
self.asset = None
self.extras = []
self.optics = DaeOptics()
self.imager = None
self.syntax = DaeSyntax.CAMERA
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeCamera, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
self.optics.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.OPTICS))
self.imager = CreateObjectFromXml(daeDocument,xmlNode, DaeSyntax.IMAGER,DaeImager)
def SaveToXml(self, daeDocument):
node = super(DaeCamera, self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the optics
node.appendChild(self.optics.SaveToXml(daeDocument))
# Add the imager
AppendChild(daeDocument,node,self.imager)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaeCamera,self).__str__()+'asset: %s, optics: %s, imager: %s, extras: %s'%(self.asset, self.optics, self.imager, self.extras)
class DaeController(DaeElement):
pass
class DaeImage(DaeElement):
def __init__(self):
super(DaeImage,self).__init__()
self.format = None
self.height = None
self.width = None
self.depth = None
self.initFrom = None
self.syntax = DaeSyntax.IMAGE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeImage, self).LoadFromXml(daeDocument, xmlNode)
self.format = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.FORMAT)
self.height = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.HEIGHT)
self.width = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.WIDTH)
self.depth = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.DEPTH)
self.initFrom = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.INIT_FROM))
def SaveToXml(self, daeDocument):
node = super(DaeImage, self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.FORMAT, self.format)
SetAttribute(node, DaeSyntax.HEIGHT, self.height)
SetAttribute(node, DaeSyntax.WIDTH, self.width)
SetAttribute(node, DaeSyntax.DEPTH, self.depth)
AppendTextChild(node, DaeSyntax.INIT_FROM,self.initFrom, None)
return node
##class DaeMaterial(DaeElement):
## def __init__(self):
## super(DaeMaterial,self).__init__()
## self.asset = None
## self.iEffects = []
## self.extras = None
## self.syntax = DaeSyntax.MATERIAL
##
## def LoadFromXml(self, daeDocument, xmlNode):
## super(DaeMaterial, self).LoadFromXml(daeDocument, xmlNode)
## self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
## self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
## self.iEffects = CreateObjectsFromXml(daeDocument,xmlNode, DaeSyntax.INSTANCE_EFFECT, DaeEffectInstance)
##
## def SaveToXml(self, daeDocument):
## node = super(DaeMaterial, self).SaveToXml(daeDocument)
## # Add the assets
## AppendChild(daeDocument,node,self.asset)
## # Add the effect instances
## AppendChilds(daeDocument, node, self.iEffects)
## # Add the extra's
## AppendChilds(self,node,self.extras)
## return node
##
## def __str__(self):
## return super(DaeLight,self).__str__()+' assets: %s, data: %s, extras: %s'%(self.asset, self.data, self.extras)
class DaeGeometry(DaeElement):
def __init__(self):
super(DaeGeometry,self).__init__()
self.asset = None
self.data = None
self.extras = None
self.syntax = DaeSyntax.GEOMETRY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeGeometry, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
self.data = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.MESH, DaeMesh)
if self.data is None:
self.data = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.CONVEX_MESH, DaeConvexMesh)
if self.data is None:
self.data = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.SPLINE, DaeSpline)
def SaveToXml(self, daeDocument):
node = super(DaeGeometry, self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the data
AppendChild(daeDocument, node, self.data)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaeGeometry,self).__str__()+' assets: %s, data: %s, extras: %s'%(self.asset, self.data, self.extras)
class DaeConvexMesh(DaeEntity):
def __init__(self):
super(DaeConvexMesh, self).__init__()
self.syntax = DaeSyntax.CONVEX_MESH
self.convexHullOf = None
def LoadFromXml(self, daeDocument, xmlNode):
self.convexHullOf = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.CONVEX_HULL_OF)
def SaveToXml(self, daeDocument):
node = super(DaeConvexMesh, self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.CONVEX_HULL_OF, StripString(self.convexHullOf))
return node
class DaeMesh(DaeEntity):
def __init__(self):
super(DaeMesh, self).__init__()
self.sources = []
self.vertices = None
self.primitives = []
self.extras = []
self.syntax = DaeSyntax.MESH
def LoadFromXml(self, daeDocument, xmlNode):
if xmlNode is None:
return
self.vertices = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.VERTICES, DaeVertices)
self.sources = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.SOURCE, DaeSource)
lines = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.LINES, DaeLines)
linestrips = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.LINESTRIPS, DaeLineStrips)
polygons = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.POLYGONS, DaePolygons)
polylist = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.POLYLIST, DaePolylist)
triangles = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TRIANGLES, DaeTriangles)
trifans = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TRIFANS, DaeTriFans)
tristrips = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TRISTRIPS, DaeTriStrips)
if lines != None: self.primitives += lines
if linestrips != None: self.primitives += linestrips
if polygons != None: self.primitives += polygons
if polylist != None: self.primitives += polylist
if triangles != None: self.primitives += triangles
if trifans != None: self.primitives += trifans
if tristrips != None: self.primitives += tristrips
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
def FindSource(self,input):
for s in self.sources:
if s.id == input.source:
return s
return None
def SaveToXml(self, daeDocument):
node = super(DaeMesh, self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.sources)
AppendChild(daeDocument, node, self.vertices)
AppendChilds(daeDocument, node, self.primitives)
AppendChilds(daeDocument, node, self.extras)
return node
class DaeVertices(DaeElement):
def __init__(self):
super(DaeVertices,self).__init__()
self.inputs = []
self.extras = []
self.syntax = DaeSyntax.VERTICES
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeVertices,self).LoadFromXml(daeDocument, xmlNode)
self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
def SaveToXml(self, daeDocument):
node = super(DaeVertices, self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.inputs)
AppendChilds(daeDocument, node, self.extras)
return node
def FindInput(self, semantic):
for i in self.inputs:
if i.semantic == semantic:
return i
return None
def __str__(self):
return super(DaeVertices,self).__str__() + ' inputs: %s, extras: %s'%(self.inputs, self.extras)
class DaeInput(DaeEntity):
def __init__(self):
super(DaeInput, self).__init__()
self.offset = None
self.semantic = ''
self.source = ''
self.set = ''
self.syntax = DaeSyntax.INPUT
def LoadFromXml(self, daeDocument, xmlNode):
self.offset = CastAttributeFromXml(xmlNode, DaeSyntax.OFFSET, int)
self.semantic = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SEMANTIC)
self.source = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SOURCE)[1:]
self.set = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SET)
def SaveToXml(self, daeDocument):
node = super(DaeInput, self).SaveToXml(daeDocument)
SetAttribute(node,DaeSyntax.OFFSET, self.offset)
SetAttribute(node,DaeSyntax.SEMANTIC, self.semantic)
SetAttribute(node,DaeSyntax.SOURCE, StripString('#'+self.source))
SetAttribute(node,DaeSyntax.SET, self.set)
return node
class DaeSource(DaeElement):
def __init__(self):
super(DaeSource, self).__init__()
self.source = DaeArray()
self.vectors = []
self.techniqueCommon = None
self.techniques = []
self.syntax = DaeSyntax.SOURCE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeSource,self).LoadFromXml(daeDocument, xmlNode)
if xmlNode is None:
return
bools = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.BOOL_ARRAY, DaeBoolArray)
floats = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.FLOAT_ARRAY, DaeFloatArray)
ints = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.INT_ARRAY, DaeIntArray)
if bools != None:
self.source = bools
elif floats != None:
self.source = floats
elif ints != None:
self.source = ints
self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE_COMMON, DaeSource.DaeTechniqueCommon)
self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique)
for i in range(0,self.techniqueCommon.accessor.count):
vec = []
for j in range(0,self.techniqueCommon.accessor.stride):
vec.append(self.source.data[i*self.techniqueCommon.accessor.stride+j])
self.vectors.append(vec)
def SaveToXml(self, daeDocument):
node = super(DaeSource, self).SaveToXml(daeDocument)
## if len(self.vectors) > 0:
## if type(self.vectors[0][0]) == float:
## self.source = DaeFloatArray()
## self.source.id = self.id+'-array'
##
## for i in range(len(self.vectors)):
## for j in range(len(self.vectors[i])):
## self.source.data.append(self.vectors[i][j])
# Add the source
AppendChild(daeDocument, node, self.source)
# Add the technique common
AppendChild(daeDocument,node,self.techniqueCommon)
# Add the techniques
AppendChilds(daeDocument, node, self.techniques)
return node
def __str__(self):
return super(DaeSource,self).__str__()+' source: %s, techniqueCommon: %s, techniques: %s'%(self.source, self.techniqueCommon, self.techniques)
class DaeTechniqueCommon(DaeEntity):
def __init__(self):
self.accessor = None
self.syntax = DaeSyntax.TECHNIQUE_COMMON
def LoadFromXml(self, daeDocument, xmlNode):
self.accessor = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ACCESSOR, DaeAccessor)
def SaveToXml(self, daeDocument):
node = super(DaeSource.DaeTechniqueCommon,self).SaveToXml(daeDocument)
AppendChild(daeDocument,node, self.accessor)
return node
def __str__(self):
return super(DaeSource.DaeTechniqueCommon,self).__str__()+' accessor: %s'%(self.accessor)
class DaeLight(DaeElement):
def __init__(self):
super(DaeLight,self).__init__()
self.asset = None
self.techniqueCommon = DaeLight.DaeTechniqueCommon()
self.techniques = []
self.extras = None
self.syntax = DaeSyntax.LIGHT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLight, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
##self.techniqueCommon.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON))
self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique)
lightSourceNode = xmlUtils.RemoveWhiteSpaceNode(xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON)).firstChild
lightSourceName = lightSourceNode.localName
if lightSourceName == DaeSyntax.DIRECTIONAL:
self.techniqueCommon = DaeLight.DaeDirectional()
elif lightSourceName == DaeSyntax.SPOT:
self.techniqueCommon = DaeLight.DaeSpot()
elif lightSourceName == DaeSyntax.AMBIENT:
self.techniqueCommon = DaeLight.DaeAmbient()
elif lightSourceName == DaeSyntax.POINT:
self.techniqueCommon = DaeLight.DaePoint()
self.techniqueCommon.LoadFromXml(daeDocument,lightSourceNode)
def SaveToXml(self, daeDocument):
node = super(DaeLight, self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the technique common
AppendChild(daeDocument,node,self.techniqueCommon)
# Add the techniques
AppendChilds(daeDocument, node, self.techniques)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaeLight,self).__str__()+' techniqueCommon: %s, techniques: %s'%(self.techniqueCommon, self.techniques)
class DaeTechniqueCommon(DaeEntity):
def __init__(self):
self.color = []
self.lightSource = None
self.syntax = DaeSyntax.TECHNIQUE_COMMON
def LoadFromXml(self, daeDocument, xmlNode):
self.color = ToFloat3(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.COLOR)))
def SaveToXml(self, daeDocument):
node = Element(DaeSyntax.TECHNIQUE_COMMON)
child = super(DaeLight.DaeTechniqueCommon,self).SaveToXml(daeDocument)
AppendTextChild(child,DaeSyntax.COLOR, self.color)
node.appendChild(child)
return node
def __str__(self):
return super(DaeLight.DaeTechniqueCommon,self).__str__()+' color: %s'%(self.color)
class DaeAmbient(DaeTechniqueCommon):
def __init__(self):
super(DaeLight.DaeAmbient,self).__init__()
self.syntax = DaeSyntax.AMBIENT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLight.DaeAmbient,self).LoadFromXml(daeDocument, xmlNode)
def SaveToXml(self, daeDocument):
node = super(DaeLight.DaeAmbient,self).SaveToXml(daeDocument)
return node
def __str__(self):
return super(DaeLight.DaeAmbient,self).__str__()
class DaeSpot(DaeTechniqueCommon):
def __init__(self):
super(DaeLight.DaeSpot,self).__init__()
self.defConstantAttenuation = 0.0
self.defLinearAttenuation = 0.0
self.defQuadraticAttenuation = 0.0
self.defFalloffAngle = 180.0
self.defFalloffExponent = 0.0
self.constantAttenuation = 0.0
self.linearAttenuation = 0.0
self.quadraticAttenuation = 0.0
self.falloffAngle = 180.0
self.falloffExponent = 0.0
self.syntax = DaeSyntax.SPOT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLight.DaeSpot,self).LoadFromXml(daeDocument, xmlNode)
self.constantAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.CONSTANT_ATTENUATION, float, self.defConstantAttenuation)
self.linearAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.LINEAR_ATTENUATION,float, self.defLinearAttenuation)
self.quadraticAttenuation = CastFromXml(daeDocument, xmlNode,DaeSyntax.QUADRATIC_ATTENUATION, float, self.defQuadraticAttenuation)
self.falloffAngle = CastFromXml(daeDocument, xmlNode, DaeSyntax.FALLOFF_ANGLE, float, self.defFalloffAngle)
self.falloffExponent = CastFromXml(daeDocument, xmlNode, DaeSyntax.FALLOFF_EXPONENT, float, self.defFalloffExponent)
def SaveToXml(self, daeDocument):
node = super(DaeLight.DaeSpot,self).SaveToXml(daeDocument)
AppendTextChild(node.firstChild,DaeSyntax.CONSTANT_ATTENUATION, self.constantAttenuation, self.defConstantAttenuation)
AppendTextChild(node.firstChild,DaeSyntax.LINEAR_ATTENUATION, self.linearAttenuation, self.defLinearAttenuation)
AppendTextChild(node.firstChild,DaeSyntax.QUADRATIC_ATTENUATION, self.quadraticAttenuation, self.defQuadraticAttenuation)
AppendTextChild(node.firstChild,DaeSyntax.FALLOFF_ANGLE, self.falloffAngle, self.defFalloffAngle)
AppendTextChild(node.firstChild,DaeSyntax.FALLOFF_EXPONENT, self.falloffExponent, self.defFalloffExponent)
return node
def __str__(self):
return super(DaeLight.DaeSpot,self).__str__()+' const.att: %s, lin.att: %s, quad.att: %s, falloffAngle: %s, falloffExponent: %s'%(self.constantAttenuation, self.linearAttenuation, self.quadraticAttenuation, self.falloffAngle, self.falloffExponent)
class DaeDirectional(DaeTechniqueCommon):
# default direction is [0,0,-1] pointing down the -Z axis.
# To change the direction, change the transform of the parent DaeNode
def __init__(self):
super(DaeLight.DaeDirectional,self).__init__()
self.syntax = DaeSyntax.DIRECTIONAL
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLight.DaeDirectional,self).LoadFromXml(daeDocument, xmlNode)
def SaveToXml(self, daeDocument):
node = super(DaeLight.DaeDirectional,self).SaveToXml(daeDocument)
return node
def __str__(self):
return super(DaeLight.DaeDirectional,self).__str__()
class DaePoint(DaeTechniqueCommon):
def __init__(self):
super(DaeLight.DaePoint,self).__init__()
self.constantAttenuation = 0.0
self.linearAttenuation = 0.0
self.quadraticAttenuation = 0.0
self.syntax = DaeSyntax.POINT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLight.DaePoint,self).LoadFromXml(daeDocument, xmlNode)
self.constantAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.CONSTANT_ATTENUATION, float)
self.linearAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.LINEAR_ATTENUATION,float)
self.quadraticAttenuation = CastFromXml(daeDocument, xmlNode,DaeSyntax.QUADRATIC_ATTENUATION, float)
def SaveToXml(self, daeDocument):
node = super(DaeLight.DaePoint,self).SaveToXml(daeDocument)
AppendTextChild(node.firstChild,DaeSyntax.CONSTANT_ATTENUATION, self.constantAttenuation)
AppendTextChild(node.firstChild,DaeSyntax.LINEAR_ATTENUATION, self.linearAttenuation)
AppendTextChild(node.firstChild,DaeSyntax.QUADRATIC_ATTENUATION, self.quadraticAttenuation)
return node
def __str__(self):
return super(DaeLight.DaePoint,self).__str__()+' const.att: %s, lin.att: %s, quad.att: %s'%(self.constantAttenuation, self.linearAttenuation, self.quadraticAttenuation)
class DaeVisualScene(DaeElement):
def __init__(self):
super(DaeVisualScene,self).__init__()
self.asset = None
self.extras = None
self.nodes = []
self.syntax = DaeSyntax.VISUAL_SCENE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeVisualScene, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
self.nodes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.NODE, DaeNode)
def SaveToXml(self, daeDocument):
node = super(DaeVisualScene, self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the nodes
AppendChilds(daeDocument, node, self.nodes)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def FindNode(self, nodeUrl):
for n in self.nodes:
if n.id == nodeUrl:
return n
return None
def __str__(self):
return super(DaeVisualScene,self).__str__()+' asset: %s, nodes: %s, extras: %s'%(self.asset, self.nodes, self.extras)
class DaeNode(DaeElement):
NODE = 2
JOINT = 1
def __init__(self):
super(DaeNode,self).__init__()
self.sid = None
self.type = DaeNode.NODE
self.layer = None
self.transforms = []
self.nodes = [] #Doug: child nodes defined in nested xml
#Doug: instances - references to items defined elsewhere
# ie iNode <instance_node url="#someothernode"/>
self.iAnimations = []
self.iCameras = []
self.iControllers = []
self.iGeometries = []
self.iLights = []
self.iNodes = []
self.iVisualScenes = []
self.syntax = DaeSyntax.NODE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeNode, self).LoadFromXml(daeDocument, xmlNode)
self.sid = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SID)
type = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.TYPE)
if type == DaeSyntax.TYPE_JOINT:
self.type = DaeNode.JOINT
else:
self.type = DaeNode.NODE
self.layer = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.LAYER)
# Get transforms
xmlUtils.RemoveWhiteSpaceNode(xmlNode)
child = xmlNode.firstChild
while child != None:
name = child.localName
if name == DaeSyntax.TRANSLATE:
self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))])
elif name == DaeSyntax.ROTATE:
self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))])
elif name == DaeSyntax.SCALE:
self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))])
elif name == DaeSyntax.SKEW:
self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))])
elif name == DaeSyntax.LOOKAT:
self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))])
elif name == DaeSyntax.MATRIX:
self.transforms.append([name,ToMatrix4(xmlUtils.ReadContents(child))])
child = child.nextSibling
# Get the instances
self.iAnimations = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_ANIMATION, DaeAnimationInstance)
self.iCameras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_CAMERA, DaeCameraInstance)
self.iControllers = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_CONTROLLER, DaeControllerInstance)
self.iGeometries = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_GEOMETRY, DaeGeometryInstance)
self.iLights = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_LIGHT, DaeLightInstance)
self.iNodes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_NODE, DaeNodeInstance)
self.iVisualScenes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_VISUAL_SCENE, DaeVisualSceneInstance)
# Get childs nodes
self.nodes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.NODE, DaeNode)
def SaveToXml(self, daeDocument):
node = super(DaeNode, self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.SID, self.sid)
if self.type == DaeSyntax.TYPE_JOINT:
SetAttribute(node, DaeSyntax.TYPE, DaeNode.GetType(self.type))
SetAttribute(node, DaeSyntax.LAYER, self.layer)
for i in self.transforms:
writeTransform = False
el = Element(i[0])
val = i[1]
if i[0] == DaeSyntax.MATRIX:
val = MatrixToString(val,ROUND)
AppendTextChild(node,i[0],val)
else:
val = ListToString(RoundList(val, 5))
if i[0] == DaeSyntax.SCALE:
AppendTextChild(node,i[0],val,"1.0 1.0 1.0")
elif i[0] == DaeSyntax.TRANSLATE:
AppendTextChild(node,i[0],val,"0.0 0.0 0.0")
elif i[0] == DaeSyntax.ROTATE:
AppendTextChild(node,i[0],val,"0.0 0.0 0.0 0.0")
elif i[0] == DaeSyntax.SKEW:
AppendTextChild(node,i[0],val)
elif i[0] == DaeSyntax.MATRIX or i[0] == DaeSyntax.LOOKAT:
AppendTextChild(node,i[0],val)
AppendChilds(daeDocument, node, self.nodes)
AppendChilds(daeDocument, node, self.iAnimations)
AppendChilds(daeDocument, node, self.iCameras)
AppendChilds(daeDocument, node, self.iControllers)
AppendChilds(daeDocument, node, self.iGeometries)
AppendChilds(daeDocument, node, self.iLights)
AppendChilds(daeDocument, node, self.iNodes)
AppendChilds(daeDocument, node, self.iVisualScenes)
return node
def IsJoint(self):
return self.type == DaeNode.JOINT
def GetType(type):
if type == DaeNode.JOINT:
return DaeSyntax.TYPE_JOINT
else:
return DaeSyntax.TYPE_NODE
GetType = staticmethod(GetType)
def GetInstances(self):
return []+self.iAnimations+self.iCameras+self.iControllers+self.iGeometries+self.iLights+self.iNodes
# TODO: finish DaeTechnique
class DaeTechnique(DaeEntity):
def __init__(self):
super(DaeTechnique,self).__init__()
self.profile = ''
self.xmlns = ''
self.syntax = DaeSyntax.TECHNIQUE
def LoadFromXml(self, daeDocument, xmlNode):
self.profile = xmlNode.getAttribute(DaeSyntax.PROFILE)
self.xmlns = xmlNode.getAttribute(DaeSyntax.XMLNS)
def SaveToXml(self, daeDocument):
node = super(DaeTechnique,self).SaveToXml(daeDocument)
node.setAttribute(DaeSyntax.PROFILE, self.profile)
SetAttribute(node, DaeSyntax.XMLNS, self.xmlns)
return node
class DaeOptics(DaeEntity):
def __init__(self):
super(DaeOptics,self).__init__()
self.techniqueCommon = DaeOptics.DaeTechniqueCommon()
self.techniques = []
self.extras = None
self.syntax = DaeSyntax.OPTICS
def LoadFromXml(self, daeDocument, xmlNode):
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
#self.techniqueCommon.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON))
self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique)
opticsSourceNode = xmlUtils.RemoveWhiteSpaceNode(xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON)).firstChild
opticsSourceName = opticsSourceNode.localName
if opticsSourceName == DaeSyntax.PERSPECTIVE:
self.techniqueCommon = DaeOptics.DaePerspective()
elif opticsSourceName == DaeSyntax.ORTHOGRAPHIC:
self.techniqueCommon = DaeOptics.DaeOrthoGraphic()
self.techniqueCommon.LoadFromXml(daeDocument,opticsSourceNode)
def SaveToXml(self, daeDocument):
node = super(DaeOptics, self).SaveToXml(daeDocument)
# Add the technique common
AppendChild(daeDocument,node,self.techniqueCommon)
# Add the techniques
AppendChilds(daeDocument, node, self.techniques)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaeOptics,self).__str__()+' extras: %s, techniqueCommon: %s, techniques: %s'%(self.extras, self.techniqueCommon, self.techniques)
class DaeTechniqueCommon(DaeEntity):
def __init__(self):
self.znear = 0.0
self.zfar = 0.0
self.aspectRatio = None
self.syntax = DaeSyntax.TECHNIQUE_COMMON
def LoadFromXml(self, daeDocument, xmlNode):
self.znear = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.ZNEAR)))
self.zfar = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.ZFAR)))
self.aspectRatio = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.ASPECT_RATIO)))
def SaveToXml(self, daeDocument):
node = Element(DaeSyntax.TECHNIQUE_COMMON)
child = super(DaeOptics.DaeTechniqueCommon,self).SaveToXml(daeDocument)
## AppendTextChild(child,DaeSyntax.ZNEAR, self.znear)
## AppendTextChild(child,DaeSyntax.ZFAR, self.zfar)
## AppendTextChild(child,DaeSyntax.ASPECT_RATIO, self.aspectRatio)
node.appendChild(child)
return node
def SavePropertiesToXml(self, daeDocument, node):
AppendTextChild(node,DaeSyntax.ZNEAR, self.znear)
AppendTextChild(node,DaeSyntax.ZFAR, self.zfar)
AppendTextChild(node,DaeSyntax.ASPECT_RATIO, self.aspectRatio)
def __str__(self):
return super(DaeOptics.DaeTechniqueCommon,self).__str__()+' znear: %s, zfar: %s, aspectRatio: %s'%(self.znear, self.zfar, self.aspectRatio)
class DaePerspective(DaeTechniqueCommon):
def __init__(self):
super(DaeOptics.DaePerspective,self).__init__()
self.xfov = None
self.yfov = None
self.syntax = DaeSyntax.PERSPECTIVE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeOptics.DaePerspective,self).LoadFromXml(daeDocument, xmlNode)
self.xfov = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.XFOV)))
self.yfov = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.YFOV)))
def SaveToXml(self, daeDocument):
node = super(DaeOptics.DaePerspective,self).SaveToXml(daeDocument)
AppendTextChild(node.firstChild,DaeSyntax.XFOV, self.xfov)
AppendTextChild(node.firstChild,DaeSyntax.YFOV, self.yfov)
super(DaeOptics.DaePerspective,self).SavePropertiesToXml(daeDocument, node.firstChild)
return node
def __str__(self):
return super(DaeOptics.DaePerspective,self).__str__()+' xfov: %s, yfov: %s'%(self.xfov, self.yfov)
class DaeOrthoGraphic(DaeTechniqueCommon):
def __init__(self):
super(DaeOptics.DaeOrthoGraphic,self).__init__()
self.xmag = None
self.ymag = None
self.syntax = DaeSyntax.ORTHOGRAPHIC
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeOptics.DaeOrthoGraphic,self).LoadFromXml(daeDocument, xmlNode)
self.xmag = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.XMAG)))
self.ymag = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.YMAG)))
def SaveToXml(self, daeDocument):
node = super(DaeOptics.DaeOrthoGraphic,self).SaveToXml(daeDocument)
AppendTextChild(node.firstChild,DaeSyntax.XMAG, self.xmag)
AppendTextChild(node.firstChild,DaeSyntax.YMAG, self.ymag)
return node
def __str__(self):
return super(DaeOptics.DaeOrthoGraphic,self).__str__()+ 'xmag: %s, ymag: %s'%(self.xmag, self.ymag)
class DaeImager(DaeEntity):
def __init__(self):
super(DaeImager,self).__init__()
self.techniques = []
self.extras = None
self.syntax = DaeSyntax.IMAGER
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeImager, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique)
def SaveToXml(self, daeDocument):
node = super(DaeOptics, self).SaveToXml(daeDocument)
# Add the techniques
AppendChilds(daeDocument, node, self.techniques)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaeImager,self).__str__()+' extras: %s, techniques: %s'%(self.techniqueCommon, self.techniques)
class DaeExtra(DaeEntity):
def __init__(self):
super(DaeExtra,self).__init__()
self.type = None
self.techniques = []
self.syntax = DaeSyntax.EXTRA
def LoadFromXml(self, daeDocument, xmlNode):
self.type = xmlNode.getAttribute(DaeSyntax.TYPE)
self.techniques = CreateObjectsFromXml(daeDocument, xmlNode,DaeSyntax.TECHNIQUE,DaeTechnique)
def SaveToXml(self, daeDocument):
node = super(DaeExtra,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.techniques)
return node
def __str__(self):
return super(DaeExtra,self).__str__()+'techniques: %s'%(self.techniques)
class DaeAccessor(DaeEntity):
def __init__(self):
super(DaeAccessor,self).__init__()
self.count = 0
self.offset = None
self.source = ''
self.stride = None
self.params = []
self.syntax = DaeSyntax.ACCESSOR
def LoadFromXml(self,daeDocument, xmlNode):
self.count = CastAttributeFromXml(xmlNode, DaeSyntax.COUNT,int,0)
self.offset = CastAttributeFromXml(xmlNode, DaeSyntax.OFFSET,int)
self.source = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SOURCE)
self.stride = CastAttributeFromXml(xmlNode, DaeSyntax.STRIDE,int)
self.params = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.PARAM, DaeParam)
def SaveToXml(self, daeDocument):
node = super(DaeAccessor,self).SaveToXml(daeDocument)
node.setAttribute(DaeSyntax.COUNT, str(self.count))
SetAttribute(node,DaeSyntax.OFFSET, self.offset)
node.setAttribute(DaeSyntax.SOURCE, StripString('#'+self.source))
SetAttribute(node, DaeSyntax.STRIDE, len(self.params))
AppendChilds(daeDocument, node, self.params)
return node
def AddParam(self, name, type):
param = DaeParam()
param.name = name
param.type = type
self.params.append(param)
class DaeParam(DaeEntity):
def __init__(self):
super(DaeParam,self).__init__()
self.name = None
self.semantic = None
self.sid = None
self.type = ''
self.syntax = DaeSyntax.PARAM
def LoadFromXml(self, daeDocument, xmlNode):
self.semantic = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SEMANTIC)
self.sid = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SID)
self.name = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.NAME)
self.type = xmlNode.getAttribute(DaeSyntax.TYPE)
def SaveToXml(self, daeDocument):
node = super(DaeParam,self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.SEMANTIC, self.semantic)
SetAttribute(node, DaeSyntax.SID, self.sid)
SetAttribute(node, DaeSyntax.NAME, self.name)
node.setAttribute(DaeSyntax.TYPE, self.type)
return node
class DaeArray(DaeElement):
def __init__(self):
super(DaeArray, self).__init__()
self.count = 0
self.data = []
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeArray, self).LoadFromXml(daeDocument, xmlNode)
self.count = ToInt(xmlUtils.ReadAttribute(xmlNode, DaeSyntax.COUNT))
self.data = ToList(xmlUtils.ReadContents(xmlNode))
def SaveToXml(self, daeDocument):
node = super(DaeArray,self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.COUNT, len(self.data))
AppendTextInChild(node, self.data)
return node
def __str__(self):
return super(DaeArray,self).__str__()+' count: %s'%(self.count)
class DaeFloatArray(DaeArray):
def __init__(self):
super(DaeFloatArray, self).__init__()
self.syntax = DaeSyntax.FLOAT_ARRAY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFloatArray, self).LoadFromXml(daeDocument, xmlNode)
self.data = ToFloatList(self.data)
class DaeIntArray(DaeArray):
def __init__(self):
super(DaeIntArray, self).__init__()
self.syntax = DaeSyntax.INT_ARRAY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeIntArray, self).LoadFromXml(daeDocument, xmlNode)
self.data = ToIntList(self.data)
class DaeBoolArray(DaeArray):
def __init__(self):
super(DaeBoolArray, self).__init__()
self.syntax = DaeSyntax.BOOL_ARRAY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeBoolArray, self).LoadFromXml(daeDocument, xmlNode)
self.data = ToBoolList(self.data)
class DaeNameArray(DaeArray):
def __init__(self):
super(DaeNameArray, self).__init__()
self.syntax = DaeSyntax.NAME_ARRAY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeNameArray, self).LoadFromXml(daeDocument, xmlNode)
##self.data = ToFloatList(self.data)
class DaeIDREFArray(DaeArray):
def __init__(self):
super(DaeIDREFArray, self).__init__()
self.syntax = DaeSyntax.IDREF_ARRAY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeIDREFArray, self).LoadFromXml(daeDocument, xmlNode)
##self.data = ToFloatList(self.data)
#---Primitive Classes---
class DaePrimitive(DaeEntity):
def __init__(self):
super(DaePrimitive, self).__init__()
self.name = None
self.count = 0
self.material = ''
self.inputs = []
def LoadFromXml(self, daeDocument, xmlNode):
self.name = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.NAME)
self.count = int(xmlUtils.ReadAttribute(xmlNode, DaeSyntax.COUNT))
self.material = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.MATERIAL)
def SaveToXml(self, daeDocument):
node = super(DaePrimitive, self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.NAME, self.name)
SetAttribute(node, DaeSyntax.MATERIAL, StripString(self.material))
node.setAttribute(DaeSyntax.COUNT, str(self.count))
return node
def GetMaxOffset(self):
if self.inputs != []:
return max([i.offset for i in self.inputs])
else:
return None
def FindInput(self, semantic):
for i in self.inputs:
if i.semantic == semantic:
return i
return None
class DaeLines(DaePrimitive):
def __init__(self):
super(DaeLines, self).__init__()
self.syntax = DaeSyntax.LINES
self.lines = []
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLines,self).LoadFromXml(daeDocument, xmlNode)
self.syntax = DaeSyntax.LINES
self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput)
self.lines = ToIntList(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.P)))
def SaveToXml(self, daeDocument):
node = super(DaeLines,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.inputs)
AppendTextChild(node, DaeSyntax.P, self.lines)
return node
class DaeLineStrips(DaePrimitive):
def __init__(self):
super(DaeLineStrips, self).__init__()
self.syntax = DaeSyntax.LINESTRIPS
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLineStrips,self).LoadFromXml(daeDocument, xmlNode)
class DaePolygons(DaePrimitive):
def __init__(self):
super(DaePolygons, self).__init__()
self.syntax = DaeSyntax.POLYGONS
self.polygons = []
self.holedPolygons = []
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePolygons,self).LoadFromXml(daeDocument, xmlNode)
self.polygons = xmlUtils.GetListFromNodes(xmlNode.getElementsByTagName(DaeSyntax.P), int)
self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput)
def SaveToXml(self, daeDocument):
node = super(DaePolygons,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.inputs)
xmlUtils.AppendChilds(node, DaeSyntax.P, self.polygons)
return node
class DaePolylist(DaePrimitive):
def __init__(self):
super(DaePolylist, self).__init__()
self.syntax = DaeSyntax.POLYLIST
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePolylist,self).LoadFromXml(daeDocument, xmlNode)
class DaeTriangles(DaePrimitive):
def __init__(self):
super(DaeTriangles, self).__init__()
self.triangles = []
self.syntax = DaeSyntax.TRIANGLES
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeTriangles,self).LoadFromXml(daeDocument, xmlNode)
self.triangles = ToIntList(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.P)))
self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput)
def SaveToXml(self, daeDocument):
node = super(DaeTriangles,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.inputs)
AppendTextChild(node, DaeSyntax.P, self.triangles)
return node
class DaeTriFans(DaePrimitive):
def __init__(self):
super(DaeTriFans, self).__init__()
self.syntax = DaeSyntax.TRIFANS
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeTriFans,self).LoadFromXml(daeDocument, xmlNode)
class DaeTriStrips(DaePrimitive):
def __init__(self):
super(DaeTriStrips, self).__init__()
self.syntax = DaeSyntax.TRISTRIPS
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeTriStrips,self).LoadFromXml(daeDocument, xmlNode)
#---instance Classes---
class DaeInstance(DaeEntity):
def __init__(self):
super(DaeInstance, self).__init__()
self.url = ''
self.extras = []
self.object = None
def LoadFromXml(self, daeDocument, xmlNode):
self.url = ReadNodeUrl(xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
def SaveToXml(self, daeDocument):
node = super(DaeInstance,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.extras)
WriteNodeUrl(node, self.object.id)
return node
class DaeAnimationInstance(DaeInstance):
def __init__(self):
super(DaeAnimationInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_ANIMATION
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeAnimationInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.animationsLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeAnimationInstance,self).SaveToXml(daeDocument)
return node
class DaeCameraInstance(DaeInstance):
def __init__(self):
super(DaeCameraInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_CAMERA
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeCameraInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.camerasLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeCameraInstance,self).SaveToXml(daeDocument)
return node
class DaeControllerInstance(DaeInstance):
def __init__(self):
super(DaeControllerInstance, self).__init__()
self.skeletons = []
self.bindMaterials = []
self.syntax = DaeSyntax.INSTANCE_CONTROLLER
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeControllerInstance,self).LoadFromXml(daeDocument, xmlNode)
self.skeletons = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.SKELETON, DaeSkeleton)
self.bindMaterials = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.BIND_MATERIAL, DaeFxBindMaterial)
self.object = daeDocument.controllersLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeControllerInstance,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.skeletons)
AppendChilds(daeDocument, node, self.bindMaterials)
return node
# TODO: finish DaeEffectInstance
class DaeEffectInstance(DaeInstance):
def __init__(self):
super(DaeEffectInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_EFFECT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeEffectInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.effectsLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeEffectInstance,self).SaveToXml(daeDocument)
return node
class DaeGeometryInstance(DaeInstance):
def __init__(self):
super(DaeGeometryInstance, self).__init__()
self.bindMaterials = []
self.syntax = DaeSyntax.INSTANCE_GEOMETRY
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeGeometryInstance,self).LoadFromXml(daeDocument, xmlNode)
self.bindMaterials = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.BIND_MATERIAL, DaeFxBindMaterial)
self.object = daeDocument.geometriesLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeGeometryInstance,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.bindMaterials)
return node
class DaeLightInstance(DaeInstance):
def __init__(self):
super(DaeLightInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_LIGHT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeLightInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.lightsLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeLightInstance,self).SaveToXml(daeDocument)
return node
class DaeNodeInstance(DaeInstance):
def __init__(self):
super(DaeNodeInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_NODE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeNodeInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.nodesLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeNodeInstance,self).SaveToXml(daeDocument)
return node
class DaeVisualSceneInstance(DaeInstance):
def __init__(self):
super(DaeVisualSceneInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_VISUAL_SCENE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeVisualSceneInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.visualScenesLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeVisualSceneInstance,self).SaveToXml(daeDocument)
return node
class DaeSkeleton(DaeEntity):
def __init__(self):
super(DaeSkeleton,self).__init__()
self.iControllers = []
def LoadFromXml(self, daeDocument, xmlNode):
self.iControllers = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_CONTROLLER, DaeControllerInstance)
def SaveToXml(self, daeDocument):
node = super(DaeSkeleton, self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.iControllers)
return node
class DaeSyntax(object):
#---collada---
COLLADA = 'COLLADA'
VERSION = 'version'
XMLNS = 'xmlns'
BODY = 'body'
TARGET = 'target'
ASSET = 'asset'
ID = 'id'
NAME = 'name'
URL = 'url'
COUNT = 'count'
OFFSET = 'offset'
STRIDE = 'stride'
METER = 'meter'
SID = 'sid'
SEMANTIC = 'semantic'
PARAM = 'param'
PROFILE = 'profile'
TECHNIQUE = 'technique'
TECHNIQUE_COMMON = 'technique_common'
##BIND_MATERIAL = 'bind_material'
SKELETON = 'skeleton'
P = 'p'
PH = 'ph'
H = 'h'
INPUT = 'input'
SET = 'set'
#---light---
COLOR = 'color'
AMBIENT = 'ambient'
SPOT = 'spot'
DIRECTIONAL = 'directional'
POINT = 'point'
CONSTANT_ATTENUATION = 'constant_attenuation'
LINEAR_ATTENUATION = 'linear_attenuation'
QUADRATIC_ATTENUATION = 'quadratic_attenuation'
FALLOFF_ANGLE = 'falloff_angle'
FALLOFF_EXPONENT = 'falloff_exponent'
#---camera--
OPTICS = 'optics'
PERSPECTIVE = 'perspective'
ORTHOGRAPHIC = 'orthographic'
IMAGER = 'imager'
ZNEAR = 'znear'
ZFAR = 'zfar'
XFOV = 'xfov'
YFOV = 'yfov'
XMAG = 'xmag'
YMAG ='ymag'
ASPECT_RATIO = 'aspect_ratio'
#---geometry---
MESH = 'mesh'
CONVEX_MESH ='convex_mesh'
SPLINE = 'spline'
SOURCE ='source'
VERTICES = 'vertices'
ACCESSOR = 'accessor'
CONVEX_HULL_OF = 'convex_hull_of'
#---primitives---
LINES = 'lines'
LINESTRIPS = 'linestrips'
POLYGONS = 'polygons'
POLYLIST = 'polylist'
TRIANGLES = 'triangles'
TRIFANS = 'trifans'
TRISTRIPS = 'tristrips'
#---libraries---
LIBRARY_ANIMATIONS = 'library_animations'
LIBRARY_ANIMATION_CLIPS = 'library_animation_clips'
LIBRARY_CAMERAS = 'library_cameras'
LIBRARY_CONTROLLERS = 'library_controllers'
LIBRARY_EFFECTS = 'library_effects'
LIBRARY_FORCE_FIELDS = 'library_force_fields'
LIBRARY_GEOMETRIES = 'library_geometries'
LIBRARY_IMAGES = 'library_images'
LIBRARY_LIGHTS = 'library_lights'
LIBRARY_MATERIALS = 'library_materials'
#LIBRARY_NODES = 'library_NODES' #Doug: looks funny - should be library_nodes?
LIBRARY_NODES = 'library_nodes'
LIBRARY_PHYSICS_MATERIALS = 'library_physics_materials'
LIBRARY_PHYSICS_MODELS = 'library_physics_models'
LIBRARY_PHYSICS_SCENES = 'library_physics_scenes'
LIBRARY_VISUAL_SCENES = 'library_visual_scenes'
SCENE = 'scene'
EXTRA = 'extra'
TYPE = 'type'
LIGHT = 'light'
CAMERA = 'camera'
ANIMATION = 'animation'
ANIMATION_CLIP = 'animation_clip'
GEOMETRY = 'geometry'
IMAGE = 'image'
##EFFECT = 'effect'
VISUAL_SCENE = 'visual_scene'
CONTROLLER = 'controller'
MATERIAL = 'material'
#---asset---
CONTRIBUTOR = 'contributor'
CREATED = 'created'
MODIFIED = 'modified'
REVISION = 'revision'
TITLE = 'title'
SUBJECT = 'subject'
KEYWORDS = 'keywords'
UNIT = 'unit'
UP_AXIS = 'up_axis'
Y_UP = 'Y_UP'
Z_UP = 'Z_UP'
#---contributor---
AUTHOR = 'author'
AUTHORING_TOOL = 'authoring_tool'
COMMENTS = 'comments'
COPYRIGHT = 'copyright'
SOURCE_DATA = 'source_data'
#---array---
FLOAT_ARRAY = 'float_array'
NAME_ARRAY = 'name_array'
BOOL_ARRAY = 'bool_array'
INT_ARRAY = 'int_array'
IDREF_ARRAY = 'IDREF_array'
#---node---
NODE = 'node'
TYPE_JOINT = 'JOINT'
TYPE_NODE = 'NODE'
LAYER = 'layer'
#---transforms---
TRANSLATE = 'translate'
ROTATE = 'rotate'
SCALE = 'scale'
SKEW = 'skew'
MATRIX = 'matrix'
LOOKAT = 'lookat'
#---instances---
INSTANCE_ANIMATION = 'instance_animation'
INSTANCE_CAMERA = 'instance_camera'
INSTANCE_CONTROLLER = 'instance_controller'
##INSTANCE_EFFECT = 'instance_effect'
INSTANCE_GEOMETRY = 'instance_geometry'
INSTANCE_LIGHT = 'instance_light'
INSTANCE_NODE = 'instance_node'
INSTANCE_VISUAL_SCENE = 'instance_visual_scene'
INSTANCE_PHYSICS_SCENE = 'instance_physics_scene'
#---image---
FORMAT = 'format'
DEPTH = 'depth'
HEIGHT = 'height'
WIDTH = 'width'
INIT_FROM = 'init_from'
class DaeFxBindMaterial(DaeEntity):
def __init__(self):
super(DaeFxBindMaterial, self).__init__()
self.syntax = DaeFxSyntax.BIND_MATERIAL
self.techniqueCommon = DaeFxBindMaterial.DaeFxTechniqueCommon()
def LoadFromXml(self, daeDocument, xmlNode):
self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TECHNIQUE_COMMON, DaeFxBindMaterial.DaeFxTechniqueCommon)
def SaveToXml(self, daeDocument):
node = super(DaeFxBindMaterial,self).SaveToXml(daeDocument)
AppendChild(daeDocument, node, self.techniqueCommon)
return node
class DaeFxTechniqueCommon(DaeEntity):
def __init__(self):
self.syntax = DaeFxSyntax.TECHNIQUE_COMMON
self.iMaterials = []
def LoadFromXml(self, daeDocument, xmlNode):
self.iMaterials = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.INSTANCE_MATERIAL, DaeFxMaterialInstance)
def SaveToXml(self, daeDocument):
node = super(DaeFxBindMaterial.DaeFxTechniqueCommon,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.iMaterials)
return node
def __str__(self):
return super(DaeFxBindMaterial.DaeFxTechniqueCommon,self).__str__()
class DaeFxMaterialInstance(DaeEntity):
def __init__(self):
super(DaeFxMaterialInstance, self).__init__()
self.target = ''
self.symbol = ''
self.object = None
self.syntax = DaeFxSyntax.INSTANCE_MATERIAL
def LoadFromXml(self, daeDocument, xmlNode):
self.target = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.TARGET)[1:]
self.symbol = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.SYMBOL)
self.object = daeDocument.materialsLibrary.FindObject(self.target)
def SaveToXml(self, daeDocument):
node = super(DaeFxMaterialInstance,self).SaveToXml(daeDocument)
SetAttribute(node, DaeFxSyntax.TARGET, StripString('#'+self.object.id))
SetAttribute(node, DaeFxSyntax.SYMBOL, StripString(self.object.id))
return node
class DaeFxMaterial(DaeElement):
def __init__(self):
super(DaeFxMaterial, self).__init__()
self.asset = None
self.iEffects = []
self.extras = None
self.syntax = DaeFxSyntax.MATERIAL
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFxMaterial, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
self.iEffects = CreateObjectsFromXml(daeDocument,xmlNode, DaeFxSyntax.INSTANCE_EFFECT, DaeFxEffectInstance)
def SaveToXml(self, daeDocument):
node = super(DaeFxMaterial,self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the effect instances
AppendChilds(daeDocument, node, self.iEffects)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaeFxMaterial,self).__str__()+' assets: %s, iEffects: %s, extras: %s'%(self.asset, self.iEffects, self.extras)
class DaeFxEffectInstance(DaeEntity):
def __init__(self):
super(DaeFxEffectInstance, self).__init__()
self.url = ''
self.syntax = DaeFxSyntax.INSTANCE_EFFECT
def LoadFromXml(self, daeDocument, xmlNode):
self.url = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.URL)[1:]
self.object = daeDocument.effectsLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaeFxEffectInstance,self).SaveToXml(daeDocument)
SetAttribute(node, DaeFxSyntax.URL, StripString('#'+self.object.id))
return node
class DaeFxEffect(DaeElement):
def __init__(self):
super(DaeFxEffect, self).__init__()
self.profileCommon = DaeFxProfileCommon()
self.syntax = DaeFxSyntax.EFFECT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFxEffect, self).LoadFromXml(daeDocument, xmlNode)
self.profileCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PROFILE_COMMON, DaeFxProfileCommon)
def SaveToXml(self, daeDocument):
node = super(DaeFxEffect,self).SaveToXml(daeDocument)
AppendChild(daeDocument, node, self.profileCommon)
return node
def __str__(self):
return super(DaeFxEffect, self).__str__() + ', profileCommon: %s' % (self.profileCommon)
def AddShader(self, daeShader):
self.profileCommon.technique.shader = daeShader
def AddDoubleSided(self, val):
if self.profileCommon.extra == None:
self.profileCommon.AddExtra()
self.profileCommon.extra.technique.AddDoubleSided(val)
class DaeFxProfileCommon(DaeEntity):
def __init__(self):
super(DaeFxProfileCommon, self).__init__()
self.technique = DaeFxTechnique()
self.images = []
self.newParams = []
self.syntax = DaeFxSyntax.PROFILE_COMMON
self.extra = None #Doug: added Mar3008
def LoadFromXml(self, daeDocument, xmlNode):
self.images = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.IMAGE, DaeFxImage)
self.newParams = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.NEWPARAM, DaeFxNewParam)
self.technique = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TECHNIQUE, DaeFxTechnique)
self.extra = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.EXTRA, DaeFxExtra) #Doug: added Mar3008
def SaveToXml(self, daeDocument):
node = super(DaeFxProfileCommon,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.images)
AppendChilds(daeDocument, node, self.newParams)
AppendChild(daeDocument, node, self.technique)
AppendChild(daeDocument, node, self.extra) #Doug: added Mar3008
return node
def AddExtra(self):
if not self.extra:
self.extra = DaeFxExtra()
def __str__(self):
return super(DaeFxProfileCommon, self).__str__() + ', technique: %s, images: %s, newParams: %s' % (self.technique, self.images, self.newParams)
class DaeFxExtra(DaeEntity): #Doug: added Mar3008 - there is a DaeExtra which could be used instead or for guidance
def __init__(self):
super(DaeFxExtra, self).__init__()
self.technique = DaeFxTechnique()
#self.images = []
#self.newParams = []
self.syntax = DaeFxSyntax.EXTRA
#self.extra = [] #Doug: added Mar3008
def LoadFromXml(self, daeDocument, xmlNode):
#self.images = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.IMAGE, DaeFxImage)
#self.newParams = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.NEWPARAM, DaeFxNewParam)
self.technique = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TECHNIQUE, DaeFxTechnique)
#self.extra = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.EXTRA, DaeFxExtra)
def SaveToXml(self, daeDocument):
node = super(DaeFxExtra,self).SaveToXml(daeDocument)
#AppendChilds(daeDocument, node, self.images)
#AppendChilds(daeDocument, node, self.newParams)
AppendChild(daeDocument, node, self.technique)
return node
def __str__(self):
#return super(DaeFxExtra, self).__str__() + ', technique: %s, images: %s, newParams: %s' % (self.technique, self.images, self.newParams)
return super(DaeFxExtra, self).__str__() + ', technique: %s' % (self.technique)
class DaeFxImage(DaeEntity):
def __init__(self):
super(DaeFxImage, self).__init__()
self.syntax = DaeFxSyntax.IMAGE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFxImage, self).LoadFromXml(daeDocument, xmlNode)
def SaveToXml(self, daeDocument):
node = super(DaeFxImage,self).SaveToXml(daeDocument)
return node
class DaeFxNewParam(DaeEntity):
def __init__(self):
super(DaeFxNewParam, self).__init__()
self.syntax = DaeFxSyntax.NEWPARAM
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFxNewParam, self).LoadFromXml(daeDocument, xmlNode)
def SaveToXml(self, daeDocument):
node = super(DaeFxNewParam,self).SaveToXml(daeDocument)
return node
class DaeFxTechnique(DaeEntity):
def __init__(self):
super(DaeFxTechnique, self).__init__()
self.syntax = DaeFxSyntax.TECHNIQUE
self.shader = DaeFxShadeConstant()
self.sid = ''
self.doubleside = DaeFxDoubleSided() #Doug: added Mar3008
def LoadFromXml(self, daeDocument, xmlNode):
# TODO: add asset and extra?
self.sid = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.SID)
#Doug: added Mar3008 the following 5 lines about googleearth profile and doublesided
profile = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.PROFILE)
if profile == DaeFxSyntax.GOOGLEEARTH:
lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.DOUBLESIDED)
if lightSourceNode:
self.doubleside = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.DOUBLE_SIDED, DaeFxDoubleSided)
else:
lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.CONSTANT)
if lightSourceNode:
self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.CONSTANT, DaeFxShadeConstant)
else:
lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.LAMBERT)
if lightSourceNode:
self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.LAMBERT, DaeFxShadeLambert)
else:
lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.BLINN)
if lightSourceNode:
self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.BLINN, DaeFxShadeBlinn)
else:
lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.PHONG)
self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PHONG, DaeFxShadePhong)
def SaveToXml(self, daeDocument):
node = super(DaeFxTechnique,self).SaveToXml(daeDocument)
if self.doubleside.value:
node.setAttribute(DaeFxSyntax.PROFILE, DaeFxSyntax.GOOGLEEARTH)
AppendChild(daeDocument, node, self.doubleside)
else:
node.setAttribute(DaeFxSyntax.SID, self.sid)
AppendChild(daeDocument, node, self.shader)
return node
def AddDoubleSided(self, val): #Doug: added Mar3008
self.doubleside.AddValue(val)
def __str__(self):
if self.doubleside.doublesided:
return super(DaeFxTechnique,self).__str__()+' doublesided: %s'%(self.doubleside)
else:
return super(DaeFxTechnique,self).__str__()+' shader: %s'%(self.shader)
class DaeFxDoubleSided(DaeEntity): #Doug: added Mar3008
def __init__(self):
super(DaeFxDoubleSided, self).__init__()
self.value = None
self.syntax = DaeFxSyntax.DOUBLE_SIDED
def LoadFromXml(self, daeDocument, xmlNode):
#self.value = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.DOUBLE_SIDED, DaeFxCommonContainer, True)
self.value = CastFromXml(daeDocument, xmlNode, self.syntax, int)
def SaveToXml(self, daeDocument):
node = super(DaeFxDoubleSided,self).SaveToXml(daeDocument)
#def AppendTextInChild(xmlNode, object):
AppendTextInChild(node,int(self.value))
#AppendTextChild(node, self.syntax, int(self.value)) #not quie the right thing- dont need syntax again
return node
def AddValue(self, val):
self.value = int(val)
class DaeFxShadeConstant(DaeEntity):
def __init__(self):
super(DaeFxShadeConstant, self).__init__()
self.emission = None
self.reflective = None
self.reflectivity = None
self.transparent = None
self.transparency = None
self.indexOfRefraction = None
self.syntax = DaeFxSyntax.CONSTANT
def LoadFromXml(self, daeDocument, xmlNode):
self.emission = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.EMISSION, DaeFxCommonColorAndTextureContainer, True)
self.reflective = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.REFLECTIVE, DaeFxCommonColorAndTextureContainer, True)
self.reflectivity = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.REFLECTIVITY, DaeFxCommonFloatAndParamContainer, True)
self.transparent = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TRANSPARENT, DaeFxCommonColorAndTextureContainer, True)
self.transparency = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TRANSPARENCY, DaeFxCommonFloatAndParamContainer, True)
self.indexOfRefraction = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.INDEXOFREFRACTION, DaeFxCommonFloatAndParamContainer, True)
def SaveToXml(self, daeDocument):
node = super(DaeFxShadeConstant,self).SaveToXml(daeDocument)
AppendChild(daeDocument, node, self.emission)
if isinstance(self, DaeFxShadeLambert):
AppendChild(daeDocument, node, self.ambient)
AppendChild(daeDocument, node, self.diffuse)
if isinstance(self,DaeFxShadeBlinn):
AppendChild(daeDocument, node, self.specular)
AppendChild(daeDocument, node, self.shininess)
AppendChild(daeDocument, node, self.reflective)
AppendChild(daeDocument, node, self.reflectivity)
AppendChild(daeDocument, node, self.transparent)
AppendChild(daeDocument, node, self.transparency)
AppendChild(daeDocument, node, self.indexOfRefraction)
return node
def AddValue(self, type, val):
col = DaeFxColor()
col.rgba = val
if type == DaeFxSyntax.EMISSION:
if not self.emission:
self.emission = DaeFxCommonColorAndTextureContainer(type)
self.emission.color = col
elif type == DaeFxSyntax.REFLECTIVE:
if not self.reflective:
self.reflective = DaeFxCommonColorAndTextureContainer(type)
self.reflective.color = col
elif type == DaeFxSyntax.REFLECTIVITY:
if not self.reflectivity:
self.reflectivity = DaeFxCommonFloatAndParamContainer(type)
self.reflectivity.float = val
elif type == DaeFxSyntax.TRANSPARENT:
if not self.transparent:
self.transparent = DaeFxCommonColorAndTextureContainer(type)
self.transparent.color = col
elif type == DaeFxSyntax.TRANSPARENCY:
if not self.transparency:
self.transparency = DaeFxCommonFloatAndParamContainer(type)
self.transparency.float = val
elif type == DaeFxSyntax.INDEXOFREFRACTION:
if not self.indexOfRefraction:
self.indexOfRefraction = DaeFxCommonFloatAndParamContainer(type)
self.indexOfRefraction.float = val
else:
Debug.Debug('DaeFxShadeConstant: type: %s not recognised'%(type),'ERROR')
class DaeFxShadeLambert(DaeFxShadeConstant):
def __init__(self):
super(DaeFxShadeLambert, self).__init__()
self.ambient = None
self.diffuse = None
self.syntax = DaeFxSyntax.LAMBERT
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFxShadeLambert, self).LoadFromXml(daeDocument, xmlNode)
self.ambient = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.AMBIENT, DaeFxCommonColorAndTextureContainer, True)
self.diffuse = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.DIFFUSE, DaeFxCommonColorAndTextureContainer, True)
def SaveToXml(self, daeDocument):
node = super(DaeFxShadeLambert,self).SaveToXml(daeDocument)
#AppendChild(daeDocument, node, self.ambient)
#AppendChild(daeDocument, node, self.diffuse)
return node
def AddValue(self, type, val):
col = DaeFxColor()
col.rgba = val
if type == DaeFxSyntax.DIFFUSE:
if not self.diffuse:
self.diffuse = DaeFxCommonColorAndTextureContainer(type)
if isinstance(val, DaeFxTexture): # its a texture
self.diffuse.texture = val
else: # it's a color
self.diffuse.color = col
elif type == DaeFxSyntax.AMBIENT:
if not self.ambient:
self.ambient = DaeFxCommonColorAndTextureContainer(type)
self.ambient.color = col
else:
super(DaeFxShadeLambert,self).AddValue(type, val)
class DaeFxShadeBlinn(DaeFxShadeLambert):
def __init__(self):
super(DaeFxShadeBlinn, self).__init__()
self.specular = None
self.shininess = None
self.syntax = DaeFxSyntax.BLINN
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeFxShadeBlinn, self).LoadFromXml(daeDocument, xmlNode)
self.specular = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.SPECULAR, DaeFxCommonColorAndTextureContainer, True)
self.shininess = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.SHININESS, DaeFxCommonFloatAndParamContainer, True)
def SaveToXml(self, daeDocument):
node = super(DaeFxShadeBlinn,self).SaveToXml(daeDocument)
#AppendChild(daeDocument, node, self.specular)
#AppendChild(daeDocument, node, self.shininess)
return node
def AddValue(self, type, val):
col = DaeFxColor()
col.rgba = val
if type == DaeFxSyntax.SPECULAR:
if not self.specular:
self.specular = DaeFxCommonColorAndTextureContainer(type)
self.specular.color = col
elif type == DaeFxSyntax.SHININESS:
if not self.shininess:
self.shininess = DaeFxCommonFloatAndParamContainer(type)
self.shininess.float = val
else:
super(DaeFxShadeBlinn,self).AddValue(type, val)
class DaeFxShadePhong(DaeFxShadeBlinn):
def __init__(self):
super(DaeFxShadePhong, self).__init__()
self.syntax = DaeFxSyntax.PHONG
class DaeFxCommonColorAndTextureContainer(DaeEntity):
def __init__(self, syntax='UNKNOWN'):
super(DaeFxCommonColorAndTextureContainer, self).__init__()
self.color = None
self.texture = None
self.syntax = syntax
def LoadFromXml(self, daeDocument, xmlNode):
self.color = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.COLOR, DaeFxColor)
self.texture = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TEXTURE, DaeFxTexture)
def SaveToXml(self, daeDocument):
node = super(DaeFxCommonColorAndTextureContainer,self).SaveToXml(daeDocument)
AppendChild(daeDocument, node, self.color)
AppendChild(daeDocument, node, self.texture)
return node
class DaeFxCommonFloatAndParamContainer(DaeEntity):
def __init__(self, syntax = 'UNKNOWN'):
super(DaeFxCommonFloatAndParamContainer, self).__init__()
self.float = None
self.param = None
self.syntax = syntax
def LoadFromXml(self, daeDocument, xmlNode):
self.float = CastFromXml(daeDocument, xmlNode, DaeFxSyntax.FLOAT, float)
##self.param = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PARAM, DaeFxParam)
def SaveToXml(self, daeDocument):
node = super(DaeFxCommonFloatAndParamContainer,self).SaveToXml(daeDocument)
AppendTextChild(node, DaeFxSyntax.FLOAT, self.float)
AppendChild(daeDocument, node, self.param)
return node
class DaeFxCommonContainer(DaeEntity): #Doug: added Mar3008
def __init__(self, cast = str):
super(DaeFxCommonContainer, self).__init__()
self.value = None
self.cast = cast
self.syntax = syntax
def LoadFromXml(self, daeDocument, xmlNode):
self.value = CastFromXml(daeDocument, xmlNode, self.syntax, self.cast)
#def CastFromXml(colladaDocument, xmlNode, nodeType, cast, default=None):
##self.param = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PARAM, DaeFxParam)
def SaveToXml(self, daeDocument):
node = super(DaeFxCommonContainer,self).SaveToXml(daeDocument)
AppendTextChild(node, self.syntax, self.cast(self.value))
#def AppendTextChild(xmlNode,syntax, object, default = None):
#AppendChild(daeDocument, node, false)
return node
class DaeFxColor(DaeEntity):
def __init__(self):
super(DaeFxColor, self).__init__()
self.sid = ''
self.rgba = []
self.syntax = DaeFxSyntax.COLOR
def LoadFromXml(self, daeDocument, xmlNode):
self.sid = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.SID)
self.rgba = ToFloatList(xmlUtils.ReadContents(xmlNode))
def SaveToXml(self, daeDocument):
node = super(DaeFxColor,self).SaveToXml(daeDocument)
SetAttribute(node, DaeFxSyntax.SID, self.sid)
AppendTextInChild(node, self.rgba)
return node
class DaeFxTexture(DaeEntity):
def __init__(self):
super(DaeFxTexture, self).__init__()
self.texture = ''
self.textCoord = ''
self.syntax = DaeFxSyntax.TEXTURE
def LoadFromXml(self, daeDocument, xmlNode):
self.texture = daeDocument.imagesLibrary.FindObject(xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.TEXTURE))
self.textCoord = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.TEXCOORD)
def SaveToXml(self, daeDocument):
node = super(DaeFxTexture,self).SaveToXml(daeDocument)
SetAttribute(node, DaeFxSyntax.TEXTURE, StripString(self.texture))
SetAttribute(node, DaeFxSyntax.TEXCOORD, self.textCoord)
return node
class DaeFxSyntax(object):
COLOR = 'color'
INSTANCE_MATERIAL = 'instance_material'
INSTANCE_EFFECT = 'instance_effect'
TECHNIQUE_COMMON = 'technique_common'
SID = 'sid'
EMISSION = 'emission'
REFLECTIVE ='reflective'
REFLECTIVITY = 'reflectivity'
TRANSPARENT = 'transparent'
TRANSPARENCY = 'transparency'
INDEXOFREFRACTION = 'index_of_refraction'
TEXTURE = 'texture'
TEXCOORD = 'texcoord'
AMBIENT = 'ambient'
DIFFUSE = 'diffuse'
BIND_MATERIAL ='bind_material'
PROFILE_COMMON = 'profile_COMMON'
SYMBOL = 'symbol'
MATERIAL = 'material'
EFFECT = 'effect'
TARGET = 'target'
URL = 'url'
SYMBOL = 'symbol'
BLINN = 'blinn'
SHININESS = 'shininess'
SPECULAR = 'specular'
PHONG = 'phong'
IMAGE = 'image'
NEWPARAM = 'newparam'
TECHNIQUE = 'technique'
EXTRA = 'extra' #Doug: added Mar3008
DOUBLE_SIDED = 'double_sided' #Doug: added Mar3008
PROFILE = 'profile' #Doug: added Mar3008
GOOGLEEARTH = 'GOOGLEEARTH' #Doug: added Mar3008
CONSTANT = 'constant'
LAMBERT = 'lambert'
FLOAT ='float'
#---COLLADA PHYSICS---
class DaePhysicsScene(DaeElement):
def __init__(self):
super(DaePhysicsScene,self).__init__()
self.asset = None
self.extras = None
self.iPhysicsModels = []
self.syntax = DaePhysicsSyntax.PHYSICS_SCENE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePhysicsScene, self).LoadFromXml(daeDocument, xmlNode)
self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra)
self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset)
self.iPhysicsModels = CreateObjectsFromXml(daeDocument, xmlNode, DaePhysicsSyntax.INSTANCE_PHYSICS_MODEL, DaePhysicsModelInstance)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsScene, self).SaveToXml(daeDocument)
# Add the assets
AppendChild(daeDocument,node,self.asset)
# Add the phyics models
AppendChilds(daeDocument, node, self.iPhysicsModels)
# Add the extra's
AppendChilds(self,node,self.extras)
return node
def __str__(self):
return super(DaePhysicsScene,self).__str__()+' asset: %s, iPhysicsModels: %s, extras: %s'%(self.asset, self.iPhysicsModels, self.extras)
class DaePhysicsModelInstance(DaeInstance):
def __init__(self):
super(DaePhysicsModelInstance, self).__init__()
self.syntax = DaePhysicsSyntax.INSTANCE_PHYSICS_MODEL
self.iRigidBodies = []
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePhysicsModelInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.physicsModelsLibrary.FindObject(self.url)
self.iRigidBodies = self.CreateInstanceRigidBodies(daeDocument, xmlNode, self.object)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsModelInstance,self).SaveToXml(daeDocument)
AppendChilds(daeDocument, node, self.iRigidBodies)
return node
def CreateInstanceRigidBodies(self, daeDocument, xmlNode, physicsModel):
objects = []
CreateObjectsFromXml(daeDocument, xmlNode, DaePhysicsSyntax.INSTANCE_RIGID_BODY, DaeRigidBodyInstance)
nodes = xmlUtils.FindElementsByTagName(xmlNode,DaePhysicsSyntax.INSTANCE_RIGID_BODY)
for node in nodes:
object = DaeRigidBodyInstance()
object.LoadFromXml(daeDocument, node)
object.body = physicsModel.FindRigidBody(object.bodyString)
n = None
for visualScene in daeDocument.visualScenesLibrary.items:
n = visualScene.FindNode(object.targetString)
if not (n is None):
break
object.target = n
objects.append(object)
return objects
class DaePhysicsSceneInstance(DaeInstance):
def __init__(self):
super(DaePhysicsSceneInstance, self).__init__()
self.syntax = DaeSyntax.INSTANCE_PHYSICS_SCENE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePhysicsSceneInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.physicsScenesLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsSceneInstance,self).SaveToXml(daeDocument)
return node
class DaePhysicsMaterialInstance(DaeInstance):
def __init__(self):
super(DaePhysicsMaterialInstance, self).__init__()
self.syntax = DaePhysicsSyntax.INSTANCE_PHYSICS_MATERIAL
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePhysicsMaterialInstance,self).LoadFromXml(daeDocument, xmlNode)
self.object = daeDocument.physicsMaterialsLibrary.FindObject(self.url)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsMaterialInstance,self).SaveToXml(daeDocument)
return node
class DaeRigidBodyInstance(DaeEntity):
def __init__(self):
super(DaeRigidBodyInstance, self).__init__()
self.syntax = DaePhysicsSyntax.INSTANCE_RIGID_BODY
self.body = None
self.target = None
self.bodyString = ''
self.targetString = ''
def LoadFromXml(self, daeDocument, xmlNode):
self.bodyString = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.BODY)
self.targetString = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.TARGET)[1:]
def SaveToXml(self, daeDocument):
node = super(DaeRigidBodyInstance,self).SaveToXml(daeDocument)
SetAttribute(node, DaeSyntax.BODY, StripString(self.body.name))
SetAttribute(node, DaeSyntax.TARGET, StripString('#'+self.target.id))
return node
class DaePhysicsModel(DaeElement):
def __init__(self):
super(DaePhysicsModel,self).__init__()
self.syntax = DaePhysicsSyntax.PHYSICS_MODEL
self.rigidBodies = []
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePhysicsModel, self).LoadFromXml(daeDocument, xmlNode)
self.rigidBodies = CreateObjectsFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RIGID_BODY, DaeRigidBody)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsModel, self).SaveToXml(daeDocument)
# Add the rigid bodies
AppendChilds(daeDocument, node, self.rigidBodies)
return node
def FindRigidBody(self, url):
for rigidBody in self.rigidBodies:
if rigidBody.sid == url:
return rigidBody
return None
class DaePhysicsMaterial(DaeElement):
def __init__(self):
super(DaePhysicsMaterial,self).__init__()
self.syntax = DaePhysicsSyntax.PHYSICS_MATERIAL
self.techniqueCommon = DaePhysicsMaterial.DaeTechniqueCommon()
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePhysicsMaterial, self).LoadFromXml(daeDocument, xmlNode)
self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE_COMMON, DaePhysicsMaterial.DaeTechniqueCommon)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsMaterial, self).SaveToXml(daeDocument)
AppendChild(daeDocument,node,self.techniqueCommon)
return node
class DaeTechniqueCommon(DaeEntity):
def __init__(self):
super(DaePhysicsMaterial.DaeTechniqueCommon,self).__init__()
self.syntax = DaeSyntax.TECHNIQUE_COMMON
self.dynamicFriction = 0
self.restitution = 0
self.staticFriction = 0
def LoadFromXml(self, daeDocument, xmlNode):
self.dynamicFriction = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.DYNAMIC_FRICTION, float, 0)
self.restitution = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RESTITUTION, float, 0)
self.staticFriction = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.STATIC_FRICTION, float, 0)
def SaveToXml(self, daeDocument):
node = super(DaePhysicsMaterial.DaeTechniqueCommon,self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.DYNAMIC_FRICTION, self.dynamicFriction, None)
AppendTextChild(node, DaePhysicsSyntax.RESTITUTION, self.restitution, None)
AppendTextChild(node, DaePhysicsSyntax.STATIC_FRICTION, self.staticFriction, None)
return node
def __str__(self):
return super(DaePhysicsMaterial.DaeTechniqueCommon,self).__str__()
class DaeRigidBody(DaeEntity):
def __init__(self):
super(DaeRigidBody, self).__init__()
self.syntax = DaePhysicsSyntax.RIGID_BODY
self.name = ''
self.sid = ''
self.techniqueCommon = DaeRigidBody.DaeTechniqueCommon()
def LoadFromXml(self, daeDocument, xmlNode):
self.name = xmlNode.getAttribute(DaeSyntax.NAME)
self.sid = xmlNode.getAttribute(DaeSyntax.SID)
self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE_COMMON, DaeRigidBody.DaeTechniqueCommon)
def SaveToXml(self, daeDocument):
node = super(DaeRigidBody, self).SaveToXml(daeDocument)
SetAttribute(node,DaeSyntax.NAME, StripString(self.name))
SetAttribute(node,DaeSyntax.SID, StripString(self.name))
AppendChild(daeDocument,node,self.techniqueCommon)
return node
class DaeTechniqueCommon(DaeEntity):
def __init__(self):
super(DaeRigidBody.DaeTechniqueCommon, self).__init__()
self.syntax = DaeSyntax.TECHNIQUE_COMMON
self.iPhysicsMaterial = None
self.physicsMaterial = None
self.dynamic = True
self.mass = None
self.inertia = None
self.shapes = []
def LoadFromXml(self, daeDocument, xmlNode):
self.iPhysicsMaterial = CreateObjectFromXml(daeDocument, xmlNode, DaePhysicsSyntax.INSTANCE_PHYSICS_MATERIAL, DaePhysicsMaterialInstance)
self.physicsMaterial = CreateObjectFromXml(daeDocument, xmlNode, DaePhysicsSyntax.PHYSICS_MATERIAL, DaePhysicsMaterial)
self.dynamic = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.DYNAMIC,bool,True)
self.mass = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.MASS, float, 1)
self.inertia = ToFloat3(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaePhysicsSyntax.INERTIA)))
shapeNodes = xmlUtils.FindElementsByTagName(xmlNode, DaePhysicsSyntax.SHAPE)
for shapeNode in shapeNodes:
s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.BOX)
b = None
if not (s is None):
b = DaeBoxShape()
else:
s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.SPHERE)
if not (s is None):
b = DaeSphereShape()
else:
s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.PLANE)
if not (s is None):
b = DaePlaneShape()
else:
s = xmlUtils.FindElementByTagName(shapeNode, DaeSyntax.INSTANCE_GEOMETRY)
if not (s is None):
b = DaeGeometryShape()
else:
s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.CYLINDER)
if not (s is None):
b = DaeCylinderShape()
else:
s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.TAPERED_CYLINDER)
if not (s is None):
b = DaeTaperedCylinderShape()
else:
s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.CAPSULE)
if not (s is None):
b = DaeCapsule()
else: # TAPERED_CAPSULE
b = DaeTaperedCapsuleShape()
b.LoadFromXml(daeDocument, s)
self.shapes.append(b)
def SaveToXml(self, daeDocument):
node = super(DaeRigidBody.DaeTechniqueCommon,self).SaveToXml(daeDocument)
AppendChild(daeDocument,node,self.iPhysicsMaterial)
AppendChild(daeDocument,node,self.physicsMaterial)
shapes = Element(DaePhysicsSyntax.SHAPE)
AppendChilds(daeDocument, shapes, self.shapes)
node.appendChild(shapes)
AppendTextChild(node, DaePhysicsSyntax.DYNAMIC, self.dynamic, None)
AppendTextChild(node, DaePhysicsSyntax.MASS, self.mass, None)
AppendTextChild(node, DaePhysicsSyntax.INERTIA, self.inertia, None)
return node
def GetPhysicsMaterial(self):
if not (self.physicsMaterial is None):
return self.physicsMaterial
else:
return self.iPhysicsMaterial.object
def __str__(self):
return super(DaeRigidBody.DaeTechniqueCommon,self).__str__()
class DaeShape(DaeEntity):
def __init__(self):
super(DaeShape, self).__init__()
self.mass = None
self.density = None
self.syntax = DaePhysicsSyntax.SHAPE
def LoadFromXml(self, daeDocument, xmlNode):
self.iGeometry = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_GEOMETRY,DaeGeometryInstance)
self.mass = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.MASS, float, None)
self.density = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.DENSITY, float, None)
def SaveToXml(self, daeDocument):
node = super(DaeShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.MASS, self.mass, None)
AppendTextChild(node, DaePhysicsSyntax.DENSITY, self.density, None)
return node
class DaeBoxShape(DaeShape):
def __init__(self):
super(DaeBoxShape, self).__init__()
self.halfExtents = []
self.syntax = DaePhysicsSyntax.BOX
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeBoxShape, self).LoadFromXml(daeDocument, xmlNode)
self.halfExtents = ToFloat3(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaePhysicsSyntax.HALF_EXTENTS)))
def SaveToXml(self, daeDocument):
node = super(DaeBoxShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.HALF_EXTENTS, self.halfExtents)
return node
class DaeSphereShape(DaeShape):
def __init__(self):
super(DaeSphereShape, self).__init__()
self.radius = None
self.syntax = DaePhysicsSyntax.SPHERE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeSphereShape, self).LoadFromXml(daeDocument, xmlNode)
self.radius = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RADIUS, float)
def SaveToXml(self, daeDocument):
node = super(DaeSphereShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.RADIUS, self.radius)
return node
class DaeCylinderShape(DaeShape):
def __init__(self):
super(DaeCylinderShape, self).__init__()
self.radius = [1 , 1]
self.height = None
self.syntax = DaePhysicsSyntax.CYLINDER
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeCylinderShape, self).LoadFromXml(daeDocument, xmlNode)
self.radius = ToFloat2(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaePhysicsSyntax.RADIUS)),'Not a valid radius found. Must consist of 2 floats')
self.height = CastFromXml(daeDocument, xmlNode, DaeSyntax.HEIGHT, float)
def SaveToXml(self, daeDocument):
node = super(DaeCylinderShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.RADIUS, self.radius)
AppendTextChild(node, DaeSyntax.HEIGHT, self.height)
return node
class DaeTaperedCylinderShape(DaeShape):
def __init__(self):
super(DaeTaperedCylinderShape, self).__init__()
self.radius1 = [1 , 1]
self.radius2 = [1 , 1]
self.height = None
self.syntax = DaePhysicsSyntax.TAPERED_CYLINDER
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeTaperedCylinderShape, self).LoadFromXml(daeDocument, xmlNode)
self.radius1 = ToFloat2(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaePhysicsSyntax.RADIUS1)),'Not a valid radius found. Must consist of 2 floats')
self.radius2 = ToFloat2(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaePhysicsSyntax.RADIUS2)), 'Not a valid radius found. Must consist of 2 floats')
self.height = CastFromXml(daeDocument, xmlNode, DaeSyntax.HEIGHT, float)
def SaveToXml(self, daeDocument):
node = super(DaeTaperedCylinderShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.RADIUS1, self.radius1)
AppendTextChild(node, DaePhysicsSyntax.RADIUS2, self.radius2)
AppendTextChild(node, DaeSyntax.HEIGHT, self.height)
return node
class DaePlaneShape(DaeShape):
def __init__(self):
super(DaePlaneShape, self).__init__()
self.equation = []
self.syntax = DaePhysicsSyntax.PLANE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaePlaneShape, self).LoadFromXml(daeDocument, xmlNode)
self.equation = ToFloat4(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaePhysicsSyntax.EQUATION)))
def SaveToXml(self, daeDocument):
node = super(DaePlaneShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.EQUATION, self.equation)
return node
class DaeCapsuleShape(DaeShape):
def __init__(self):
super(DaeCapsuleShape, self).__init__()
self.radius = None
self.height = None
self.syntax = DaePhysicsSyntax.CAPSULE
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeCapsuleShape, self).LoadFromXml(daeDocument, xmlNode)
self.radius = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RADIUS, float)
self.height = CastFromXml(daeDocument, xmlNode, DaeSyntax.HEIGHT, float)
def SaveToXml(self, daeDocument):
node = super(DaeCapsuleShape, self).SaveToXml(daeDocument)
AppendTextChild(node, DaePhysicsSyntax.RADIUS, self.radius)
AppendTextChild(node, DaeSyntax.HEIGHT, self.height)
return node
class DaeGeometryShape(DaeShape):
def __init__(self):
super(DaeGeometryShape, self).__init__()
self.iGeometry = None
def LoadFromXml(self, daeDocument, xmlNode):
super(DaeGeometryShape, self).LoadFromXml(daeDocument, xmlNode)
self.iGeometry = DaeGeometryInstance()
self.iGeometry.LoadFromXml(daeDocument, xmlNode)
#self.iGeometry = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_GEOMETRY, DaeGeometryInstance)
#print self.iGeometry
#print xmlNode.toxml()
def SaveToXml(self, daeDocument):
#node = super(DaeGeometryShape, self).SaveToXml(daeDocument)
#AppendChild(daeDocument, node, self.iGeometry)
return self.iGeometry.SaveToXml(daeDocument)
class DaePhysicsSyntax(object):
PHYSICS_SCENE = 'physics_scene'
PHYSICS_MODEL = 'physics_model'
PHYSICS_MATERIAL = 'physics_material'
RIGID_BODY = 'rigid_body'
INSTANCE_PHYSICS_MODEL = 'instance_physics_model'
INSTANCE_PHYSICS_MATERIAL = 'instance_physics_material'
INSTANCE_RIGID_BODY = 'instance_rigid_body'
RESTITUTION = 'restitution'
STATIC_FRICTION = 'static_friction'
DYNAMIC_FRICTION = 'dynamic_friction'
DYNAMIC = 'dynamic'
MASS = 'mass'
INERTIA = 'inertia'
SHAPE = 'shape'
DENSITY = 'density'
RADIUS = 'radius'
RADIUS1 = 'radius1'
RADIUS2 = 'radius2'
BOX = 'box'
PLANE = 'plane'
CYLINDER = 'cylinder'
SPHERE = 'sphere'
CAPSULE = 'capsule'
TAPERED_CAPSULE ='tapered_capsule'
TAPERED_CYLINDER = 'tapered_cylinder'
HALF_EXTENTS = 'half_extents'
#---Functions---
def CreateObjectsFromXml(colladaDocument, xmlNode, nodeType, objectType):
if xmlNode is None:
return None
objects = []
nodes = xmlUtils.FindElementsByTagName(xmlNode,nodeType)
for node in nodes:
object = objectType()
object.LoadFromXml(colladaDocument, node)
objects.append(object)
return objects
def CreateObjectFromXml(colladaDocument, xmlNode, nodeType, objectType, setSyntax = False):
if xmlNode is None:
return None
node = xmlUtils.FindElementByTagName(xmlNode, nodeType)
object = None
if setSyntax:
object = objectType(nodeType)
else:
object = objectType()
if node != None:
object.LoadFromXml(colladaDocument, node)
return object
return None
def CastFromXml(colladaDocument, xmlNode, nodeType, cast, default=None):
if xmlNode is None:
return default
node = xmlUtils.FindElementByTagName(xmlNode, nodeType)
if node != None:
textValue = xmlUtils.ReadContents(node)
if cast == bool:
if textValue.lower() == 'false':
return False
else:
return True
return cast(textValue)
return default
def CastAttributeFromXml(xmlNode, nodeType, cast, default=None):
if xmlNode is None:
return default
val = xmlUtils.ReadAttribute(xmlNode, nodeType)
if val != None and val != '':
return cast(val)
return default
def AppendChild(daeDocument, xmlNode, daeEntity):
if daeEntity is None or xmlNode is None:
return
else:
child = daeEntity.SaveToXml(daeDocument)
if child is None:
return
else :
xmlNode.appendChild(child)
def AppendChilds(daeDocument, xmlNode, daeEntities):
if daeEntities is None or xmlNode is None:
return
else:
for daeEntity in daeEntities:
AppendChild(daeDocument, xmlNode, daeEntity)
def AppendTextChild(xmlNode,syntax, object, default = None):
if object is None:
return
if default != None and object == default:
return
node = Element(syntax)
xmlNode.appendChild(node)
return AppendTextInChild(node, object)
def AppendTextInChild(xmlNode, object):
if object is None:
return
text = Text()
if type(object) == datetime:
text.data = object.isoformat()##xmlUtils.ToDateTime(object)
elif type(object) == list:
if len(object) == 0: return
if object[0] is not None and type(object[0]) == float:
object = RoundList(object, ROUND)
text.data = ListToString(object)
elif type(object) == float:
text.data = round(object, ROUND)
elif type(object) == bool:
text.data = str(object).lower()
elif type(object) == int:
text.data = str(object) #Doug: added Mar3008 - same as default below so should have no effect
else:
text.data = str(object)
xmlNode.appendChild(text)
return xmlNode
def SetAttribute(xmlNode,syntax, object):
if xmlNode is None or object is None or str(object) == '':
return
xmlNode.setAttribute(syntax,str(object))
def ReadNodeUrl(node):
attribute = xmlUtils.ReadAttribute(node,DaeSyntax.URL)
if attribute == None: return None
else :
attribute = str(attribute)
if attribute.startswith('#'):
return attribute[1:]
return None
def WriteNodeUrl(node, url):
node.setAttribute(DaeSyntax.URL, StripString('#'+url))
def IsVersionOk(version, curVersion):
versionAr = version.split('.')
curVersionAr = curVersion.split('.')
for i in range(len(curVersionAr)):
if versionAr[i] != curVersionAr[i]:
return False
return True
def StripString(text):
return text.replace(' ','_').replace('.','_')
# ------------------------------------------------------------------------- # Illusoft Collada 1.4 plugin for Blender version 0.2.65 # -------------------------------------------------------------------------- # ***** BEGIN GPL LICENSE BLOCK ***** # # Copyright (C) 2006: Illusoft - colladablender@illusoft.com # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, # or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software Foundation, # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. # # ***** END GPL LICENCE BLOCK ***** # -------------------------------------------------------------------------- from xml.dom.minidom import * import xmlUtils from cutils import * from datetime import * # The number of decimals to export floats to ROUND = 5 # TODO: finish DaeDocument class DaeDocument(object): def __init__(self, debugM = False): global debugMode debugMode = debugM self.colladaVersion = '1.4.0' self.version = '' self.xmlns = '' self.asset = DaeAsset() self.extras = [] # create all the libraries self.animationsLibrary = DaeLibrary(DaeSyntax.LIBRARY_ANIMATIONS,DaeAnimation,DaeSyntax.ANIMATION) self.animationClipsLibrary = DaeLibrary(DaeSyntax.LIBRARY_ANIMATION_CLIPS,DaeAnimationClip,DaeSyntax.ANIMATION_CLIP) self.camerasLibrary = DaeLibrary(DaeSyntax.LIBRARY_CAMERAS,DaeCamera,DaeSyntax.CAMERA) self.controllersLibrary = DaeLibrary(DaeSyntax.LIBRARY_CONTROLLERS,DaeController,DaeSyntax.CONTROLLER) self.effectsLibrary = DaeLibrary(DaeSyntax.LIBRARY_EFFECTS,DaeFxEffect,DaeFxSyntax.EFFECT) self.geometriesLibrary = DaeLibrary(DaeSyntax.LIBRARY_GEOMETRIES,DaeGeometry,DaeSyntax.GEOMETRY) self.imagesLibrary = DaeLibrary(DaeSyntax.LIBRARY_IMAGES, DaeImage, DaeSyntax.IMAGE) self.lightsLibrary = DaeLibrary(DaeSyntax.LIBRARY_LIGHTS,DaeLight,DaeSyntax.LIGHT) self.materialsLibrary = DaeLibrary(DaeSyntax.LIBRARY_MATERIALS,DaeFxMaterial,DaeFxSyntax.MATERIAL) self.nodesLibrary = DaeLibrary(DaeSyntax.LIBRARY_NODES, DaeNode, DaeSyntax.NODE) self.visualScenesLibrary = DaeLibrary(DaeSyntax.LIBRARY_VISUAL_SCENES,DaeVisualScene,DaeSyntax.VISUAL_SCENE) # Physics Support self.physicsMaterialsLibrary = DaeLibrary(DaeSyntax.LIBRARY_PHYSICS_MATERIALS, DaePhysicsMaterial, DaePhysicsSyntax.PHYSICS_MATERIAL) self.physicsScenesLibrary = DaeLibrary(DaeSyntax.LIBRARY_PHYSICS_SCENES, DaePhysicsScene, DaePhysicsSyntax.PHYSICS_SCENE) self.physicsModelsLibrary = DaeLibrary(DaeSyntax.LIBRARY_PHYSICS_MODELS, DaePhysicsModel, DaePhysicsSyntax.PHYSICS_MODEL) self.scene = None self.physicsScene = None def LoadDocumentFromFile(self, filename): global debugMode # Build DOM tree doc = parse( filename ) # Get COLLADA element colladaNode = doc.documentElement # Get Attributes self.version = colladaNode.getAttribute(DaeSyntax.VERSION) if not IsVersionOk(self.version, self.colladaVersion): Debug.Debug('The version of the file (%s) is older then the version supported by this plugin(%s).'%(self.version, self.colladaVersion),'ERROR') doc.unlink() return self.xmlns = colladaNode.getAttribute(DaeSyntax.XMLNS) # get the assets element self.asset.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.ASSET)) # get the extra elements self.extras = CreateObjectsFromXml(self,colladaNode,DaeSyntax.EXTRA,DaeExtra) # parse all the libraries self.imagesLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_IMAGES)) self.animationsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_IMAGES)) self.animationClipsLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_ANIMATION_CLIPS)) self.camerasLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_CAMERAS)) self.controllersLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_CONTROLLERS)) self.effectsLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_EFFECTS)) self.geometriesLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_GEOMETRIES)) self.lightsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_LIGHTS)) self.materialsLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_MATERIALS)) self.nodesLibrary.LoadFromXml(self,xmlUtils.FindElementByTagName(colladaNode,DaeSyntax.LIBRARY_NODES)) self.visualScenesLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_VISUAL_SCENES)) self.physicsMaterialsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_PHYSICS_MATERIALS)) self.physicsModelsLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_PHYSICS_MODELS)) self.physicsScenesLibrary.LoadFromXml(self, xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.LIBRARY_PHYSICS_SCENES)) # Get the sceneNodes sceneNodes = colladaNode.getElementsByTagName(DaeSyntax.SCENE) # Get the scene sceneNode = xmlUtils.FindElementByTagName(colladaNode, DaeSyntax.SCENE) if sceneNode != None: scene = DaeScene() scene.LoadFromXml(self, sceneNode) self.scene = scene doc.unlink() if debugMode: Debug.Debug('Directly exporting this DaeDocument...','DEBUG') self.SaveDocumentToFile(filename+'_out.dae') def SaveDocumentToFile(self, filename): self.version = '1.4.0' self.xmlns = 'http://www.collada.org/2005/11/COLLADASchema' colladaNode = Element(DaeSyntax.COLLADA) colladaNode.setAttribute(DaeSyntax.VERSION, self.version) colladaNode.setAttribute(DaeSyntax.XMLNS, self.xmlns) colladaNode.appendChild(self.asset.SaveToXml(self)) # add the labraries AppendChild(self,colladaNode,self.animationsLibrary) AppendChild(self,colladaNode,self.animationClipsLibrary) AppendChild(self,colladaNode,self.camerasLibrary) AppendChild(self,colladaNode,self.controllersLibrary) AppendChild(self,colladaNode,self.effectsLibrary) AppendChild(self,colladaNode,self.geometriesLibrary) AppendChild(self,colladaNode,self.imagesLibrary) AppendChild(self,colladaNode,self.lightsLibrary) AppendChild(self,colladaNode,self.materialsLibrary) AppendChild(self,colladaNode,self.nodesLibrary) AppendChild(self,colladaNode,self.visualScenesLibrary) AppendChild(self,colladaNode,self.physicsMaterialsLibrary) AppendChild(self,colladaNode,self.physicsModelsLibrary) AppendChild(self,colladaNode,self.physicsScenesLibrary) AppendChild(self,colladaNode,self.scene) # write xml to the file fileref = open(filename, 'w') fileref.write(xmlUtils.ToXml(colladaNode)) fileref.flush() fileref.close() colladaNode.unlink() def GetItemCount(self): return (##self.animationClipsLibrary.GetItemCount()+ ##self.animationsLibrary.GetItemCount()+ ##self.camerasLibrary.GetItemCount()+ ##self.controllersLibrary.GetItemCount()+ ##self.effectsLibrary.GetItemCount()+ self.geometriesLibrary.GetItemCount()+ self.lightsLibrary.GetItemCount()+ ##self.materialsLibrary.GetItemCount()+ self.nodesLibrary.GetItemCount()##+ ##self.visualScenesLibrary.GetItemCount() ) def __str__(self): return '%s version: %s, xmlns: %s, asset: %s, extras: %s, scene: %s'%(type(self), self.version, self.xmlns, self.asset, self.extras, self.scene) class DaeEntity(object): def __init__(self): self.syntax = 'UNKNOWN' def LoadFromXml(self, daeDocument, xmlNode): Debug.Debug('DaeEntity: Override this method for %s'%(type(self)),'WARNING') def SaveToXml(self, daeDocument): node = Element(self.syntax) return node def GetType(self): return self.syntax class DaeElement(DaeEntity): def __init__(self): super(DaeElement,self).__init__() self.id = '' self.name = '' def LoadFromXml(self,daeDocument, xmlNode): if xmlNode is None: return self.id = xmlNode.getAttribute(DaeSyntax.ID) self.name = xmlNode.getAttribute(DaeSyntax.NAME) def SaveToXml(self, daeDocument): node = super(DaeElement,self).SaveToXml(daeDocument) SetAttribute(node,DaeSyntax.ID,StripString(self.id)) SetAttribute(node,DaeSyntax.NAME, StripString(self.name)) return node def __str__(self): return super(DaeElement,self).__str__()+'id: %s, name: %s'%(self.id, self.name) # TODO: finish DaeLibrary class DaeLibrary(DaeElement): def __init__(self, syntax, objectType, objectSyntax): super(DaeLibrary,self).__init__() self.extras = [] self.asset = None self.items = [] self.__objectSyntax = objectSyntax self.syntax = syntax self.__objectType = objectType def LoadFromXml(self,daeDocument, xmlNode): if xmlNode is None: return super(DaeLibrary,self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) self.items = CreateObjectsFromXml(daeDocument, xmlNode, self.__objectSyntax, self.__objectType) def SaveToXml(self,daeDocument): if len(self.items) > 0: node = super(DaeLibrary,self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the library_items AppendChilds(daeDocument,node,self.items) # Add the extra's AppendChilds(self,node,self.extras) return node else: return None def GetItemCount(self): return len(self.items) def FindObject(self,url): for i in self.items: if i.id == url: return i return None def AddItem(self,item): self.items.append(item) def __str__(self): return super(DaeLibrary,self).__str__() + 'extras: %s, asset: %s, items: %s'%(self.extras, self.asset, self.items) class DaeAsset(DaeEntity): def __init__(self): super(DaeAsset,self).__init__() self.contributors = [] self.created = None self.modified = None self.revision = None self.title = None self.subject = None self.keywords = [] self.unit = DaeUnit() self.upAxis = 'Y_UP' self.syntax = DaeSyntax.ASSET def LoadFromXml(self, daeDocument, xmlNode): if xmlNode is None: return # Get the contributor(s) self.contributors = CreateObjectsFromXml(daeDocument, xmlNode,DaeSyntax.CONTRIBUTOR,DaeContributor) # Get created self.created = xmlUtils.ReadDateTime(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.CREATED)) # Get modified self.modified = xmlUtils.ReadDateTime(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.MODIFIED)) # Get revision self.revision = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.REVISION)) # Get title self.title = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.TITLE)) # Get subject self.subject = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.SUBJECT)) # Get keywords self.keywords = xmlUtils.GetStringArrayFromNodes(xmlNode.getElementsByTagName(DaeSyntax.KEYWORDS)) # Get Unit self.unit.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.UNIT)) # Get upAxis self.upAxis = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.UP_AXIS)) def SaveToXml(self, daeDocument): node = Element(DaeSyntax.ASSET) AppendChilds(daeDocument,node, self.contributors) AppendTextChild(node, DaeSyntax.CREATED, self.created) AppendTextChild(node, DaeSyntax.MODIFIED, self.modified) AppendTextChild(node, DaeSyntax.REVISION, self.revision) AppendTextChild(node, DaeSyntax.TITLE, self.title) AppendTextChild(node, DaeSyntax.SUBJECT, self.subject) AppendChild(daeDocument, node, self.unit) AppendTextChild(node, DaeSyntax.UP_AXIS, self.upAxis) return node def __str__(self): return super(DaeAsset,self).__str__()+'contributors: %s, created: %s, modified: %s, revision: %s, title: %s, subject: %s, keywords: %s, unit: %s, upAxis: %s'%(self.contributors, self.created, self.modified, self.revision, self.title, self.subject, self.keywords, self.unit, self.upAxis) # TODO: finish DaeScene class DaeScene(DaeEntity): def __init__(self): super(DaeScene,self).__init__() self.extras = [] self.iVisualScenes = [] self.iPhysicsScenes = [] self.syntax = DaeSyntax.SCENE def LoadFromXml(self, daeDocument, xmlNode): if xmlNode is None: return self.iVisualScenes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_VISUAL_SCENE, DaeVisualSceneInstance) self.iPhysicsScenes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_PHYSICS_SCENE, DaePhysicsSceneInstance) def SaveToXml(self, daeDocument): node = super(DaeScene,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.iVisualScenes) AppendChilds(daeDocument, node, self.iPhysicsScenes) return node def GetVisualScenes(self): result = [] for i in self.iVisualScenes: result.append(i.object) return result def GetPhysicsScenes(self): result = [] for i in self.iPhysicsScenes: result.append(i.object) return result def __str__(self): return super(DaeScene,self).__str__()+'extras: %s, visualScenes: %s, physicsScenes: %s'%(self.extras, self.iVisualScenes, self.iPhysicsScenes) class DaeUnit(DaeEntity): def __init__(self): super(DaeUnit,self).__init__() self.name = 'meter' self.meter = 1.0 self.syntax = DaeSyntax.UNIT def LoadFromXml(self, daeDocument, xmlNode): if xmlNode is None: return name = xmlNode.getAttribute(DaeSyntax.NAME) if name != '': self.name = name meter = xmlNode.getAttribute(DaeSyntax.METER) if meter != '': self.meter = float(meter) def SaveToXml(self,daeDocument): node = super(DaeUnit, self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.METER, self.meter) SetAttribute(node, DaeSyntax.NAME, self.name) return node def __str__(self): return super(DaeUnit, self).__str__()+' name: %s, meter: %s'%(self.name, self.meter) class DaeContributor(DaeEntity): def __init__(self): super(DaeContributor, self).__init__() self.author = '' self.authoringTool = '' self.comments = '' self.copyright = '' self.sourceData = '' self.syntax = DaeSyntax.CONTRIBUTOR def LoadFromXml(self, daeDocument, xmlNode): self.author = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.AUTHOR)) self.authoringTool = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.AUTHORING_TOOL)) self.comments = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.COMMENTS)) self.copyright = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.COPYRIGHT)) self.sourceData = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.SOURCE_DATA)) def SaveToXml(self, daeDocument): node = super(DaeContributor, self).SaveToXml(daeDocument) AppendTextChild(node, DaeSyntax.AUTHOR, self.author) AppendTextChild(node, DaeSyntax.AUTHORING_TOOL, self.authoringTool) AppendTextChild(node, DaeSyntax.COMMENTS, self.comments) AppendTextChild(node, DaeSyntax.COPYRIGHT, self.copyright) AppendTextChild(node, DaeSyntax.SOURCE_DATA, self.sourceData) return node def __str__(self): return super(DaeContributor,self).__str__() + 'author: %s, authoring_tool: %s, comments: %s, copyright: %s, sourceData: %s'%(self.author, self.authoringTool, self.comments, self.copyright, self.sourceData) class DaeAnimation(DaeElement): pass class DaeAnimationClip(DaeElement): pass class DaeCamera(DaeElement): def __init__(self): super(DaeCamera,self).__init__() self.asset = None self.extras = [] self.optics = DaeOptics() self.imager = None self.syntax = DaeSyntax.CAMERA def LoadFromXml(self, daeDocument, xmlNode): super(DaeCamera, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) self.optics.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.OPTICS)) self.imager = CreateObjectFromXml(daeDocument,xmlNode, DaeSyntax.IMAGER,DaeImager) def SaveToXml(self, daeDocument): node = super(DaeCamera, self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the optics node.appendChild(self.optics.SaveToXml(daeDocument)) # Add the imager AppendChild(daeDocument,node,self.imager) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaeCamera,self).__str__()+'asset: %s, optics: %s, imager: %s, extras: %s'%(self.asset, self.optics, self.imager, self.extras) class DaeController(DaeElement): pass class DaeImage(DaeElement): def __init__(self): super(DaeImage,self).__init__() self.format = None self.height = None self.width = None self.depth = None self.initFrom = None self.syntax = DaeSyntax.IMAGE def LoadFromXml(self, daeDocument, xmlNode): super(DaeImage, self).LoadFromXml(daeDocument, xmlNode) self.format = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.FORMAT) self.height = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.HEIGHT) self.width = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.WIDTH) self.depth = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.DEPTH) self.initFrom = xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.INIT_FROM)) def SaveToXml(self, daeDocument): node = super(DaeImage, self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.FORMAT, self.format) SetAttribute(node, DaeSyntax.HEIGHT, self.height) SetAttribute(node, DaeSyntax.WIDTH, self.width) SetAttribute(node, DaeSyntax.DEPTH, self.depth) AppendTextChild(node, DaeSyntax.INIT_FROM,self.initFrom, None) return node ##class DaeMaterial(DaeElement): ## def __init__(self): ## super(DaeMaterial,self).__init__() ## self.asset = None ## self.iEffects = [] ## self.extras = None ## self.syntax = DaeSyntax.MATERIAL ## ## def LoadFromXml(self, daeDocument, xmlNode): ## super(DaeMaterial, self).LoadFromXml(daeDocument, xmlNode) ## self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) ## self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) ## self.iEffects = CreateObjectsFromXml(daeDocument,xmlNode, DaeSyntax.INSTANCE_EFFECT, DaeEffectInstance) ## ## def SaveToXml(self, daeDocument): ## node = super(DaeMaterial, self).SaveToXml(daeDocument) ## # Add the assets ## AppendChild(daeDocument,node,self.asset) ## # Add the effect instances ## AppendChilds(daeDocument, node, self.iEffects) ## # Add the extra's ## AppendChilds(self,node,self.extras) ## return node ## ## def __str__(self): ## return super(DaeLight,self).__str__()+' assets: %s, data: %s, extras: %s'%(self.asset, self.data, self.extras) class DaeGeometry(DaeElement): def __init__(self): super(DaeGeometry,self).__init__() self.asset = None self.data = None self.extras = None self.syntax = DaeSyntax.GEOMETRY def LoadFromXml(self, daeDocument, xmlNode): super(DaeGeometry, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) self.data = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.MESH, DaeMesh) if self.data is None: self.data = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.CONVEX_MESH, DaeConvexMesh) if self.data is None: self.data = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.SPLINE, DaeSpline) def SaveToXml(self, daeDocument): node = super(DaeGeometry, self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the data AppendChild(daeDocument, node, self.data) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaeGeometry,self).__str__()+' assets: %s, data: %s, extras: %s'%(self.asset, self.data, self.extras) class DaeConvexMesh(DaeEntity): def __init__(self): super(DaeConvexMesh, self).__init__() self.syntax = DaeSyntax.CONVEX_MESH self.convexHullOf = None def LoadFromXml(self, daeDocument, xmlNode): self.convexHullOf = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.CONVEX_HULL_OF) def SaveToXml(self, daeDocument): node = super(DaeConvexMesh, self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.CONVEX_HULL_OF, StripString(self.convexHullOf)) return node class DaeMesh(DaeEntity): def __init__(self): super(DaeMesh, self).__init__() self.sources = [] self.vertices = None self.primitives = [] self.extras = [] self.syntax = DaeSyntax.MESH def LoadFromXml(self, daeDocument, xmlNode): if xmlNode is None: return self.vertices = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.VERTICES, DaeVertices) self.sources = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.SOURCE, DaeSource) lines = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.LINES, DaeLines) linestrips = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.LINESTRIPS, DaeLineStrips) polygons = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.POLYGONS, DaePolygons) polylist = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.POLYLIST, DaePolylist) triangles = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TRIANGLES, DaeTriangles) trifans = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TRIFANS, DaeTriFans) tristrips = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TRISTRIPS, DaeTriStrips) if lines != None: self.primitives += lines if linestrips != None: self.primitives += linestrips if polygons != None: self.primitives += polygons if polylist != None: self.primitives += polylist if triangles != None: self.primitives += triangles if trifans != None: self.primitives += trifans if tristrips != None: self.primitives += tristrips self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) def FindSource(self,input): for s in self.sources: if s.id == input.source: return s return None def SaveToXml(self, daeDocument): node = super(DaeMesh, self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.sources) AppendChild(daeDocument, node, self.vertices) AppendChilds(daeDocument, node, self.primitives) AppendChilds(daeDocument, node, self.extras) return node class DaeVertices(DaeElement): def __init__(self): super(DaeVertices,self).__init__() self.inputs = [] self.extras = [] self.syntax = DaeSyntax.VERTICES def LoadFromXml(self, daeDocument, xmlNode): super(DaeVertices,self).LoadFromXml(daeDocument, xmlNode) self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) def SaveToXml(self, daeDocument): node = super(DaeVertices, self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.inputs) AppendChilds(daeDocument, node, self.extras) return node def FindInput(self, semantic): for i in self.inputs: if i.semantic == semantic: return i return None def __str__(self): return super(DaeVertices,self).__str__() + ' inputs: %s, extras: %s'%(self.inputs, self.extras) class DaeInput(DaeEntity): def __init__(self): super(DaeInput, self).__init__() self.offset = None self.semantic = '' self.source = '' self.set = '' self.syntax = DaeSyntax.INPUT def LoadFromXml(self, daeDocument, xmlNode): self.offset = CastAttributeFromXml(xmlNode, DaeSyntax.OFFSET, int) self.semantic = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SEMANTIC) self.source = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SOURCE)[1:] self.set = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SET) def SaveToXml(self, daeDocument): node = super(DaeInput, self).SaveToXml(daeDocument) SetAttribute(node,DaeSyntax.OFFSET, self.offset) SetAttribute(node,DaeSyntax.SEMANTIC, self.semantic) SetAttribute(node,DaeSyntax.SOURCE, StripString('#'+self.source)) SetAttribute(node,DaeSyntax.SET, self.set) return node class DaeSource(DaeElement): def __init__(self): super(DaeSource, self).__init__() self.source = DaeArray() self.vectors = [] self.techniqueCommon = None self.techniques = [] self.syntax = DaeSyntax.SOURCE def LoadFromXml(self, daeDocument, xmlNode): super(DaeSource,self).LoadFromXml(daeDocument, xmlNode) if xmlNode is None: return bools = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.BOOL_ARRAY, DaeBoolArray) floats = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.FLOAT_ARRAY, DaeFloatArray) ints = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.INT_ARRAY, DaeIntArray) if bools != None: self.source = bools elif floats != None: self.source = floats elif ints != None: self.source = ints self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE_COMMON, DaeSource.DaeTechniqueCommon) self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique) for i in range(0,self.techniqueCommon.accessor.count): vec = [] for j in range(0,self.techniqueCommon.accessor.stride): vec.append(self.source.data[i*self.techniqueCommon.accessor.stride+j]) self.vectors.append(vec) def SaveToXml(self, daeDocument): node = super(DaeSource, self).SaveToXml(daeDocument) ## if len(self.vectors) > 0: ## if type(self.vectors[0][0]) == float: ## self.source = DaeFloatArray() ## self.source.id = self.id+'-array' ## ## for i in range(len(self.vectors)): ## for j in range(len(self.vectors[i])): ## self.source.data.append(self.vectors[i][j]) # Add the source AppendChild(daeDocument, node, self.source) # Add the technique common AppendChild(daeDocument,node,self.techniqueCommon) # Add the techniques AppendChilds(daeDocument, node, self.techniques) return node def __str__(self): return super(DaeSource,self).__str__()+' source: %s, techniqueCommon: %s, techniques: %s'%(self.source, self.techniqueCommon, self.techniques) class DaeTechniqueCommon(DaeEntity): def __init__(self): self.accessor = None self.syntax = DaeSyntax.TECHNIQUE_COMMON def LoadFromXml(self, daeDocument, xmlNode): self.accessor = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ACCESSOR, DaeAccessor) def SaveToXml(self, daeDocument): node = super(DaeSource.DaeTechniqueCommon,self).SaveToXml(daeDocument) AppendChild(daeDocument,node, self.accessor) return node def __str__(self): return super(DaeSource.DaeTechniqueCommon,self).__str__()+' accessor: %s'%(self.accessor) class DaeLight(DaeElement): def __init__(self): super(DaeLight,self).__init__() self.asset = None self.techniqueCommon = DaeLight.DaeTechniqueCommon() self.techniques = [] self.extras = None self.syntax = DaeSyntax.LIGHT def LoadFromXml(self, daeDocument, xmlNode): super(DaeLight, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) ##self.techniqueCommon.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON)) self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique) lightSourceNode = xmlUtils.RemoveWhiteSpaceNode(xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON)).firstChild lightSourceName = lightSourceNode.localName if lightSourceName == DaeSyntax.DIRECTIONAL: self.techniqueCommon = DaeLight.DaeDirectional() elif lightSourceName == DaeSyntax.SPOT: self.techniqueCommon = DaeLight.DaeSpot() elif lightSourceName == DaeSyntax.AMBIENT: self.techniqueCommon = DaeLight.DaeAmbient() elif lightSourceName == DaeSyntax.POINT: self.techniqueCommon = DaeLight.DaePoint() self.techniqueCommon.LoadFromXml(daeDocument,lightSourceNode) def SaveToXml(self, daeDocument): node = super(DaeLight, self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the technique common AppendChild(daeDocument,node,self.techniqueCommon) # Add the techniques AppendChilds(daeDocument, node, self.techniques) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaeLight,self).__str__()+' techniqueCommon: %s, techniques: %s'%(self.techniqueCommon, self.techniques) class DaeTechniqueCommon(DaeEntity): def __init__(self): self.color = [] self.lightSource = None self.syntax = DaeSyntax.TECHNIQUE_COMMON def LoadFromXml(self, daeDocument, xmlNode): self.color = ToFloat3(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.COLOR))) def SaveToXml(self, daeDocument): node = Element(DaeSyntax.TECHNIQUE_COMMON) child = super(DaeLight.DaeTechniqueCommon,self).SaveToXml(daeDocument) AppendTextChild(child,DaeSyntax.COLOR, self.color) node.appendChild(child) return node def __str__(self): return super(DaeLight.DaeTechniqueCommon,self).__str__()+' color: %s'%(self.color) class DaeAmbient(DaeTechniqueCommon): def __init__(self): super(DaeLight.DaeAmbient,self).__init__() self.syntax = DaeSyntax.AMBIENT def LoadFromXml(self, daeDocument, xmlNode): super(DaeLight.DaeAmbient,self).LoadFromXml(daeDocument, xmlNode) def SaveToXml(self, daeDocument): node = super(DaeLight.DaeAmbient,self).SaveToXml(daeDocument) return node def __str__(self): return super(DaeLight.DaeAmbient,self).__str__() class DaeSpot(DaeTechniqueCommon): def __init__(self): super(DaeLight.DaeSpot,self).__init__() self.defConstantAttenuation = 0.0 self.defLinearAttenuation = 0.0 self.defQuadraticAttenuation = 0.0 self.defFalloffAngle = 180.0 self.defFalloffExponent = 0.0 self.constantAttenuation = 0.0 self.linearAttenuation = 0.0 self.quadraticAttenuation = 0.0 self.falloffAngle = 180.0 self.falloffExponent = 0.0 self.syntax = DaeSyntax.SPOT def LoadFromXml(self, daeDocument, xmlNode): super(DaeLight.DaeSpot,self).LoadFromXml(daeDocument, xmlNode) self.constantAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.CONSTANT_ATTENUATION, float, self.defConstantAttenuation) self.linearAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.LINEAR_ATTENUATION,float, self.defLinearAttenuation) self.quadraticAttenuation = CastFromXml(daeDocument, xmlNode,DaeSyntax.QUADRATIC_ATTENUATION, float, self.defQuadraticAttenuation) self.falloffAngle = CastFromXml(daeDocument, xmlNode, DaeSyntax.FALLOFF_ANGLE, float, self.defFalloffAngle) self.falloffExponent = CastFromXml(daeDocument, xmlNode, DaeSyntax.FALLOFF_EXPONENT, float, self.defFalloffExponent) def SaveToXml(self, daeDocument): node = super(DaeLight.DaeSpot,self).SaveToXml(daeDocument) AppendTextChild(node.firstChild,DaeSyntax.CONSTANT_ATTENUATION, self.constantAttenuation, self.defConstantAttenuation) AppendTextChild(node.firstChild,DaeSyntax.LINEAR_ATTENUATION, self.linearAttenuation, self.defLinearAttenuation) AppendTextChild(node.firstChild,DaeSyntax.QUADRATIC_ATTENUATION, self.quadraticAttenuation, self.defQuadraticAttenuation) AppendTextChild(node.firstChild,DaeSyntax.FALLOFF_ANGLE, self.falloffAngle, self.defFalloffAngle) AppendTextChild(node.firstChild,DaeSyntax.FALLOFF_EXPONENT, self.falloffExponent, self.defFalloffExponent) return node def __str__(self): return super(DaeLight.DaeSpot,self).__str__()+' const.att: %s, lin.att: %s, quad.att: %s, falloffAngle: %s, falloffExponent: %s'%(self.constantAttenuation, self.linearAttenuation, self.quadraticAttenuation, self.falloffAngle, self.falloffExponent) class DaeDirectional(DaeTechniqueCommon): # default direction is [0,0,-1] pointing down the -Z axis. # To change the direction, change the transform of the parent DaeNode def __init__(self): super(DaeLight.DaeDirectional,self).__init__() self.syntax = DaeSyntax.DIRECTIONAL def LoadFromXml(self, daeDocument, xmlNode): super(DaeLight.DaeDirectional,self).LoadFromXml(daeDocument, xmlNode) def SaveToXml(self, daeDocument): node = super(DaeLight.DaeDirectional,self).SaveToXml(daeDocument) return node def __str__(self): return super(DaeLight.DaeDirectional,self).__str__() class DaePoint(DaeTechniqueCommon): def __init__(self): super(DaeLight.DaePoint,self).__init__() self.constantAttenuation = 0.0 self.linearAttenuation = 0.0 self.quadraticAttenuation = 0.0 self.syntax = DaeSyntax.POINT def LoadFromXml(self, daeDocument, xmlNode): super(DaeLight.DaePoint,self).LoadFromXml(daeDocument, xmlNode) self.constantAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.CONSTANT_ATTENUATION, float) self.linearAttenuation = CastFromXml(daeDocument,xmlNode,DaeSyntax.LINEAR_ATTENUATION,float) self.quadraticAttenuation = CastFromXml(daeDocument, xmlNode,DaeSyntax.QUADRATIC_ATTENUATION, float) def SaveToXml(self, daeDocument): node = super(DaeLight.DaePoint,self).SaveToXml(daeDocument) AppendTextChild(node.firstChild,DaeSyntax.CONSTANT_ATTENUATION, self.constantAttenuation) AppendTextChild(node.firstChild,DaeSyntax.LINEAR_ATTENUATION, self.linearAttenuation) AppendTextChild(node.firstChild,DaeSyntax.QUADRATIC_ATTENUATION, self.quadraticAttenuation) return node def __str__(self): return super(DaeLight.DaePoint,self).__str__()+' const.att: %s, lin.att: %s, quad.att: %s'%(self.constantAttenuation, self.linearAttenuation, self.quadraticAttenuation) class DaeVisualScene(DaeElement): def __init__(self): super(DaeVisualScene,self).__init__() self.asset = None self.extras = None self.nodes = [] self.syntax = DaeSyntax.VISUAL_SCENE def LoadFromXml(self, daeDocument, xmlNode): super(DaeVisualScene, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) self.nodes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.NODE, DaeNode) def SaveToXml(self, daeDocument): node = super(DaeVisualScene, self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the nodes AppendChilds(daeDocument, node, self.nodes) # Add the extra's AppendChilds(self,node,self.extras) return node def FindNode(self, nodeUrl): for n in self.nodes: if n.id == nodeUrl: return n return None def __str__(self): return super(DaeVisualScene,self).__str__()+' asset: %s, nodes: %s, extras: %s'%(self.asset, self.nodes, self.extras) class DaeNode(DaeElement): NODE = 2 JOINT = 1 def __init__(self): super(DaeNode,self).__init__() self.sid = None self.type = DaeNode.NODE self.layer = None self.transforms = [] self.nodes = [] #Doug: child nodes defined in nested xml #Doug: instances - references to items defined elsewhere # ie iNode <instance_node url="#someothernode"/> self.iAnimations = [] self.iCameras = [] self.iControllers = [] self.iGeometries = [] self.iLights = [] self.iNodes = [] self.iVisualScenes = [] self.syntax = DaeSyntax.NODE def LoadFromXml(self, daeDocument, xmlNode): super(DaeNode, self).LoadFromXml(daeDocument, xmlNode) self.sid = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SID) type = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.TYPE) if type == DaeSyntax.TYPE_JOINT: self.type = DaeNode.JOINT else: self.type = DaeNode.NODE self.layer = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.LAYER) # Get transforms xmlUtils.RemoveWhiteSpaceNode(xmlNode) child = xmlNode.firstChild while child != None: name = child.localName if name == DaeSyntax.TRANSLATE: self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))]) elif name == DaeSyntax.ROTATE: self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))]) elif name == DaeSyntax.SCALE: self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))]) elif name == DaeSyntax.SKEW: self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))]) elif name == DaeSyntax.LOOKAT: self.transforms.append([name,ToFloatList(xmlUtils.ReadContents(child))]) elif name == DaeSyntax.MATRIX: self.transforms.append([name,ToMatrix4(xmlUtils.ReadContents(child))]) child = child.nextSibling # Get the instances self.iAnimations = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_ANIMATION, DaeAnimationInstance) self.iCameras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_CAMERA, DaeCameraInstance) self.iControllers = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_CONTROLLER, DaeControllerInstance) self.iGeometries = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_GEOMETRY, DaeGeometryInstance) self.iLights = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_LIGHT, DaeLightInstance) self.iNodes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_NODE, DaeNodeInstance) self.iVisualScenes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_VISUAL_SCENE, DaeVisualSceneInstance) # Get childs nodes self.nodes = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.NODE, DaeNode) def SaveToXml(self, daeDocument): node = super(DaeNode, self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.SID, self.sid) if self.type == DaeSyntax.TYPE_JOINT: SetAttribute(node, DaeSyntax.TYPE, DaeNode.GetType(self.type)) SetAttribute(node, DaeSyntax.LAYER, self.layer) for i in self.transforms: writeTransform = False el = Element(i[0]) val = i[1] if i[0] == DaeSyntax.MATRIX: val = MatrixToString(val,ROUND) AppendTextChild(node,i[0],val) else: val = ListToString(RoundList(val, 5)) if i[0] == DaeSyntax.SCALE: AppendTextChild(node,i[0],val,"1.0 1.0 1.0") elif i[0] == DaeSyntax.TRANSLATE: AppendTextChild(node,i[0],val,"0.0 0.0 0.0") elif i[0] == DaeSyntax.ROTATE: AppendTextChild(node,i[0],val,"0.0 0.0 0.0 0.0") elif i[0] == DaeSyntax.SKEW: AppendTextChild(node,i[0],val) elif i[0] == DaeSyntax.MATRIX or i[0] == DaeSyntax.LOOKAT: AppendTextChild(node,i[0],val) AppendChilds(daeDocument, node, self.nodes) AppendChilds(daeDocument, node, self.iAnimations) AppendChilds(daeDocument, node, self.iCameras) AppendChilds(daeDocument, node, self.iControllers) AppendChilds(daeDocument, node, self.iGeometries) AppendChilds(daeDocument, node, self.iLights) AppendChilds(daeDocument, node, self.iNodes) AppendChilds(daeDocument, node, self.iVisualScenes) return node def IsJoint(self): return self.type == DaeNode.JOINT def GetType(type): if type == DaeNode.JOINT: return DaeSyntax.TYPE_JOINT else: return DaeSyntax.TYPE_NODE GetType = staticmethod(GetType) def GetInstances(self): return []+self.iAnimations+self.iCameras+self.iControllers+self.iGeometries+self.iLights+self.iNodes # TODO: finish DaeTechnique class DaeTechnique(DaeEntity): def __init__(self): super(DaeTechnique,self).__init__() self.profile = '' self.xmlns = '' self.syntax = DaeSyntax.TECHNIQUE def LoadFromXml(self, daeDocument, xmlNode): self.profile = xmlNode.getAttribute(DaeSyntax.PROFILE) self.xmlns = xmlNode.getAttribute(DaeSyntax.XMLNS) def SaveToXml(self, daeDocument): node = super(DaeTechnique,self).SaveToXml(daeDocument) node.setAttribute(DaeSyntax.PROFILE, self.profile) SetAttribute(node, DaeSyntax.XMLNS, self.xmlns) return node class DaeOptics(DaeEntity): def __init__(self): super(DaeOptics,self).__init__() self.techniqueCommon = DaeOptics.DaeTechniqueCommon() self.techniques = [] self.extras = None self.syntax = DaeSyntax.OPTICS def LoadFromXml(self, daeDocument, xmlNode): self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) #self.techniqueCommon.LoadFromXml(daeDocument, xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON)) self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique) opticsSourceNode = xmlUtils.RemoveWhiteSpaceNode(xmlUtils.FindElementByTagName(xmlNode, DaeSyntax.TECHNIQUE_COMMON)).firstChild opticsSourceName = opticsSourceNode.localName if opticsSourceName == DaeSyntax.PERSPECTIVE: self.techniqueCommon = DaeOptics.DaePerspective() elif opticsSourceName == DaeSyntax.ORTHOGRAPHIC: self.techniqueCommon = DaeOptics.DaeOrthoGraphic() self.techniqueCommon.LoadFromXml(daeDocument,opticsSourceNode) def SaveToXml(self, daeDocument): node = super(DaeOptics, self).SaveToXml(daeDocument) # Add the technique common AppendChild(daeDocument,node,self.techniqueCommon) # Add the techniques AppendChilds(daeDocument, node, self.techniques) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaeOptics,self).__str__()+' extras: %s, techniqueCommon: %s, techniques: %s'%(self.extras, self.techniqueCommon, self.techniques) class DaeTechniqueCommon(DaeEntity): def __init__(self): self.znear = 0.0 self.zfar = 0.0 self.aspectRatio = None self.syntax = DaeSyntax.TECHNIQUE_COMMON def LoadFromXml(self, daeDocument, xmlNode): self.znear = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.ZNEAR))) self.zfar = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.ZFAR))) self.aspectRatio = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.ASPECT_RATIO))) def SaveToXml(self, daeDocument): node = Element(DaeSyntax.TECHNIQUE_COMMON) child = super(DaeOptics.DaeTechniqueCommon,self).SaveToXml(daeDocument) ## AppendTextChild(child,DaeSyntax.ZNEAR, self.znear) ## AppendTextChild(child,DaeSyntax.ZFAR, self.zfar) ## AppendTextChild(child,DaeSyntax.ASPECT_RATIO, self.aspectRatio) node.appendChild(child) return node def SavePropertiesToXml(self, daeDocument, node): AppendTextChild(node,DaeSyntax.ZNEAR, self.znear) AppendTextChild(node,DaeSyntax.ZFAR, self.zfar) AppendTextChild(node,DaeSyntax.ASPECT_RATIO, self.aspectRatio) def __str__(self): return super(DaeOptics.DaeTechniqueCommon,self).__str__()+' znear: %s, zfar: %s, aspectRatio: %s'%(self.znear, self.zfar, self.aspectRatio) class DaePerspective(DaeTechniqueCommon): def __init__(self): super(DaeOptics.DaePerspective,self).__init__() self.xfov = None self.yfov = None self.syntax = DaeSyntax.PERSPECTIVE def LoadFromXml(self, daeDocument, xmlNode): super(DaeOptics.DaePerspective,self).LoadFromXml(daeDocument, xmlNode) self.xfov = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.XFOV))) self.yfov = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.YFOV))) def SaveToXml(self, daeDocument): node = super(DaeOptics.DaePerspective,self).SaveToXml(daeDocument) AppendTextChild(node.firstChild,DaeSyntax.XFOV, self.xfov) AppendTextChild(node.firstChild,DaeSyntax.YFOV, self.yfov) super(DaeOptics.DaePerspective,self).SavePropertiesToXml(daeDocument, node.firstChild) return node def __str__(self): return super(DaeOptics.DaePerspective,self).__str__()+' xfov: %s, yfov: %s'%(self.xfov, self.yfov) class DaeOrthoGraphic(DaeTechniqueCommon): def __init__(self): super(DaeOptics.DaeOrthoGraphic,self).__init__() self.xmag = None self.ymag = None self.syntax = DaeSyntax.ORTHOGRAPHIC def LoadFromXml(self, daeDocument, xmlNode): super(DaeOptics.DaeOrthoGraphic,self).LoadFromXml(daeDocument, xmlNode) self.xmag = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.XMAG))) self.ymag = ToFloat(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.YMAG))) def SaveToXml(self, daeDocument): node = super(DaeOptics.DaeOrthoGraphic,self).SaveToXml(daeDocument) AppendTextChild(node.firstChild,DaeSyntax.XMAG, self.xmag) AppendTextChild(node.firstChild,DaeSyntax.YMAG, self.ymag) return node def __str__(self): return super(DaeOptics.DaeOrthoGraphic,self).__str__()+ 'xmag: %s, ymag: %s'%(self.xmag, self.ymag) class DaeImager(DaeEntity): def __init__(self): super(DaeImager,self).__init__() self.techniques = [] self.extras = None self.syntax = DaeSyntax.IMAGER def LoadFromXml(self, daeDocument, xmlNode): super(DaeImager, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.techniques = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE,DaeTechnique) def SaveToXml(self, daeDocument): node = super(DaeOptics, self).SaveToXml(daeDocument) # Add the techniques AppendChilds(daeDocument, node, self.techniques) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaeImager,self).__str__()+' extras: %s, techniques: %s'%(self.techniqueCommon, self.techniques) class DaeExtra(DaeEntity): def __init__(self): super(DaeExtra,self).__init__() self.type = None self.techniques = [] self.syntax = DaeSyntax.EXTRA def LoadFromXml(self, daeDocument, xmlNode): self.type = xmlNode.getAttribute(DaeSyntax.TYPE) self.techniques = CreateObjectsFromXml(daeDocument, xmlNode,DaeSyntax.TECHNIQUE,DaeTechnique) def SaveToXml(self, daeDocument): node = super(DaeExtra,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.techniques) return node def __str__(self): return super(DaeExtra,self).__str__()+'techniques: %s'%(self.techniques) class DaeAccessor(DaeEntity): def __init__(self): super(DaeAccessor,self).__init__() self.count = 0 self.offset = None self.source = '' self.stride = None self.params = [] self.syntax = DaeSyntax.ACCESSOR def LoadFromXml(self,daeDocument, xmlNode): self.count = CastAttributeFromXml(xmlNode, DaeSyntax.COUNT,int,0) self.offset = CastAttributeFromXml(xmlNode, DaeSyntax.OFFSET,int) self.source = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SOURCE) self.stride = CastAttributeFromXml(xmlNode, DaeSyntax.STRIDE,int) self.params = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.PARAM, DaeParam) def SaveToXml(self, daeDocument): node = super(DaeAccessor,self).SaveToXml(daeDocument) node.setAttribute(DaeSyntax.COUNT, str(self.count)) SetAttribute(node,DaeSyntax.OFFSET, self.offset) node.setAttribute(DaeSyntax.SOURCE, StripString('#'+self.source)) SetAttribute(node, DaeSyntax.STRIDE, len(self.params)) AppendChilds(daeDocument, node, self.params) return node def AddParam(self, name, type): param = DaeParam() param.name = name param.type = type self.params.append(param) class DaeParam(DaeEntity): def __init__(self): super(DaeParam,self).__init__() self.name = None self.semantic = None self.sid = None self.type = '' self.syntax = DaeSyntax.PARAM def LoadFromXml(self, daeDocument, xmlNode): self.semantic = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SEMANTIC) self.sid = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.SID) self.name = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.NAME) self.type = xmlNode.getAttribute(DaeSyntax.TYPE) def SaveToXml(self, daeDocument): node = super(DaeParam,self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.SEMANTIC, self.semantic) SetAttribute(node, DaeSyntax.SID, self.sid) SetAttribute(node, DaeSyntax.NAME, self.name) node.setAttribute(DaeSyntax.TYPE, self.type) return node class DaeArray(DaeElement): def __init__(self): super(DaeArray, self).__init__() self.count = 0 self.data = [] def LoadFromXml(self, daeDocument, xmlNode): super(DaeArray, self).LoadFromXml(daeDocument, xmlNode) self.count = ToInt(xmlUtils.ReadAttribute(xmlNode, DaeSyntax.COUNT)) self.data = ToList(xmlUtils.ReadContents(xmlNode)) def SaveToXml(self, daeDocument): node = super(DaeArray,self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.COUNT, len(self.data)) AppendTextInChild(node, self.data) return node def __str__(self): return super(DaeArray,self).__str__()+' count: %s'%(self.count) class DaeFloatArray(DaeArray): def __init__(self): super(DaeFloatArray, self).__init__() self.syntax = DaeSyntax.FLOAT_ARRAY def LoadFromXml(self, daeDocument, xmlNode): super(DaeFloatArray, self).LoadFromXml(daeDocument, xmlNode) self.data = ToFloatList(self.data) class DaeIntArray(DaeArray): def __init__(self): super(DaeIntArray, self).__init__() self.syntax = DaeSyntax.INT_ARRAY def LoadFromXml(self, daeDocument, xmlNode): super(DaeIntArray, self).LoadFromXml(daeDocument, xmlNode) self.data = ToIntList(self.data) class DaeBoolArray(DaeArray): def __init__(self): super(DaeBoolArray, self).__init__() self.syntax = DaeSyntax.BOOL_ARRAY def LoadFromXml(self, daeDocument, xmlNode): super(DaeBoolArray, self).LoadFromXml(daeDocument, xmlNode) self.data = ToBoolList(self.data) class DaeNameArray(DaeArray): def __init__(self): super(DaeNameArray, self).__init__() self.syntax = DaeSyntax.NAME_ARRAY def LoadFromXml(self, daeDocument, xmlNode): super(DaeNameArray, self).LoadFromXml(daeDocument, xmlNode) ##self.data = ToFloatList(self.data) class DaeIDREFArray(DaeArray): def __init__(self): super(DaeIDREFArray, self).__init__() self.syntax = DaeSyntax.IDREF_ARRAY def LoadFromXml(self, daeDocument, xmlNode): super(DaeIDREFArray, self).LoadFromXml(daeDocument, xmlNode) ##self.data = ToFloatList(self.data) #---Primitive Classes--- class DaePrimitive(DaeEntity): def __init__(self): super(DaePrimitive, self).__init__() self.name = None self.count = 0 self.material = '' self.inputs = [] def LoadFromXml(self, daeDocument, xmlNode): self.name = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.NAME) self.count = int(xmlUtils.ReadAttribute(xmlNode, DaeSyntax.COUNT)) self.material = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.MATERIAL) def SaveToXml(self, daeDocument): node = super(DaePrimitive, self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.NAME, self.name) SetAttribute(node, DaeSyntax.MATERIAL, StripString(self.material)) node.setAttribute(DaeSyntax.COUNT, str(self.count)) return node def GetMaxOffset(self): if self.inputs != []: return max([i.offset for i in self.inputs]) else: return None def FindInput(self, semantic): for i in self.inputs: if i.semantic == semantic: return i return None class DaeLines(DaePrimitive): def __init__(self): super(DaeLines, self).__init__() self.syntax = DaeSyntax.LINES self.lines = [] def LoadFromXml(self, daeDocument, xmlNode): super(DaeLines,self).LoadFromXml(daeDocument, xmlNode) self.syntax = DaeSyntax.LINES self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput) self.lines = ToIntList(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.P))) def SaveToXml(self, daeDocument): node = super(DaeLines,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.inputs) AppendTextChild(node, DaeSyntax.P, self.lines) return node class DaeLineStrips(DaePrimitive): def __init__(self): super(DaeLineStrips, self).__init__() self.syntax = DaeSyntax.LINESTRIPS def LoadFromXml(self, daeDocument, xmlNode): super(DaeLineStrips,self).LoadFromXml(daeDocument, xmlNode) class DaePolygons(DaePrimitive): def __init__(self): super(DaePolygons, self).__init__() self.syntax = DaeSyntax.POLYGONS self.polygons = [] self.holedPolygons = [] def LoadFromXml(self, daeDocument, xmlNode): super(DaePolygons,self).LoadFromXml(daeDocument, xmlNode) self.polygons = xmlUtils.GetListFromNodes(xmlNode.getElementsByTagName(DaeSyntax.P), int) self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput) def SaveToXml(self, daeDocument): node = super(DaePolygons,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.inputs) xmlUtils.AppendChilds(node, DaeSyntax.P, self.polygons) return node class DaePolylist(DaePrimitive): def __init__(self): super(DaePolylist, self).__init__() self.syntax = DaeSyntax.POLYLIST def LoadFromXml(self, daeDocument, xmlNode): super(DaePolylist,self).LoadFromXml(daeDocument, xmlNode) class DaeTriangles(DaePrimitive): def __init__(self): super(DaeTriangles, self).__init__() self.triangles = [] self.syntax = DaeSyntax.TRIANGLES def LoadFromXml(self, daeDocument, xmlNode): super(DaeTriangles,self).LoadFromXml(daeDocument, xmlNode) self.triangles = ToIntList(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaeSyntax.P))) self.inputs = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INPUT, DaeInput) def SaveToXml(self, daeDocument): node = super(DaeTriangles,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.inputs) AppendTextChild(node, DaeSyntax.P, self.triangles) return node class DaeTriFans(DaePrimitive): def __init__(self): super(DaeTriFans, self).__init__() self.syntax = DaeSyntax.TRIFANS def LoadFromXml(self, daeDocument, xmlNode): super(DaeTriFans,self).LoadFromXml(daeDocument, xmlNode) class DaeTriStrips(DaePrimitive): def __init__(self): super(DaeTriStrips, self).__init__() self.syntax = DaeSyntax.TRISTRIPS def LoadFromXml(self, daeDocument, xmlNode): super(DaeTriStrips,self).LoadFromXml(daeDocument, xmlNode) #---instance Classes--- class DaeInstance(DaeEntity): def __init__(self): super(DaeInstance, self).__init__() self.url = '' self.extras = [] self.object = None def LoadFromXml(self, daeDocument, xmlNode): self.url = ReadNodeUrl(xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) def SaveToXml(self, daeDocument): node = super(DaeInstance,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.extras) WriteNodeUrl(node, self.object.id) return node class DaeAnimationInstance(DaeInstance): def __init__(self): super(DaeAnimationInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_ANIMATION def LoadFromXml(self, daeDocument, xmlNode): super(DaeAnimationInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.animationsLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeAnimationInstance,self).SaveToXml(daeDocument) return node class DaeCameraInstance(DaeInstance): def __init__(self): super(DaeCameraInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_CAMERA def LoadFromXml(self, daeDocument, xmlNode): super(DaeCameraInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.camerasLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeCameraInstance,self).SaveToXml(daeDocument) return node class DaeControllerInstance(DaeInstance): def __init__(self): super(DaeControllerInstance, self).__init__() self.skeletons = [] self.bindMaterials = [] self.syntax = DaeSyntax.INSTANCE_CONTROLLER def LoadFromXml(self, daeDocument, xmlNode): super(DaeControllerInstance,self).LoadFromXml(daeDocument, xmlNode) self.skeletons = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.SKELETON, DaeSkeleton) self.bindMaterials = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.BIND_MATERIAL, DaeFxBindMaterial) self.object = daeDocument.controllersLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeControllerInstance,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.skeletons) AppendChilds(daeDocument, node, self.bindMaterials) return node # TODO: finish DaeEffectInstance class DaeEffectInstance(DaeInstance): def __init__(self): super(DaeEffectInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_EFFECT def LoadFromXml(self, daeDocument, xmlNode): super(DaeEffectInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.effectsLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeEffectInstance,self).SaveToXml(daeDocument) return node class DaeGeometryInstance(DaeInstance): def __init__(self): super(DaeGeometryInstance, self).__init__() self.bindMaterials = [] self.syntax = DaeSyntax.INSTANCE_GEOMETRY def LoadFromXml(self, daeDocument, xmlNode): super(DaeGeometryInstance,self).LoadFromXml(daeDocument, xmlNode) self.bindMaterials = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.BIND_MATERIAL, DaeFxBindMaterial) self.object = daeDocument.geometriesLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeGeometryInstance,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.bindMaterials) return node class DaeLightInstance(DaeInstance): def __init__(self): super(DaeLightInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_LIGHT def LoadFromXml(self, daeDocument, xmlNode): super(DaeLightInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.lightsLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeLightInstance,self).SaveToXml(daeDocument) return node class DaeNodeInstance(DaeInstance): def __init__(self): super(DaeNodeInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_NODE def LoadFromXml(self, daeDocument, xmlNode): super(DaeNodeInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.nodesLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeNodeInstance,self).SaveToXml(daeDocument) return node class DaeVisualSceneInstance(DaeInstance): def __init__(self): super(DaeVisualSceneInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_VISUAL_SCENE def LoadFromXml(self, daeDocument, xmlNode): super(DaeVisualSceneInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.visualScenesLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeVisualSceneInstance,self).SaveToXml(daeDocument) return node class DaeSkeleton(DaeEntity): def __init__(self): super(DaeSkeleton,self).__init__() self.iControllers = [] def LoadFromXml(self, daeDocument, xmlNode): self.iControllers = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_CONTROLLER, DaeControllerInstance) def SaveToXml(self, daeDocument): node = super(DaeSkeleton, self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.iControllers) return node class DaeSyntax(object): #---collada--- COLLADA = 'COLLADA' VERSION = 'version' XMLNS = 'xmlns' BODY = 'body' TARGET = 'target' ASSET = 'asset' ID = 'id' NAME = 'name' URL = 'url' COUNT = 'count' OFFSET = 'offset' STRIDE = 'stride' METER = 'meter' SID = 'sid' SEMANTIC = 'semantic' PARAM = 'param' PROFILE = 'profile' TECHNIQUE = 'technique' TECHNIQUE_COMMON = 'technique_common' ##BIND_MATERIAL = 'bind_material' SKELETON = 'skeleton' P = 'p' PH = 'ph' H = 'h' INPUT = 'input' SET = 'set' #---light--- COLOR = 'color' AMBIENT = 'ambient' SPOT = 'spot' DIRECTIONAL = 'directional' POINT = 'point' CONSTANT_ATTENUATION = 'constant_attenuation' LINEAR_ATTENUATION = 'linear_attenuation' QUADRATIC_ATTENUATION = 'quadratic_attenuation' FALLOFF_ANGLE = 'falloff_angle' FALLOFF_EXPONENT = 'falloff_exponent' #---camera-- OPTICS = 'optics' PERSPECTIVE = 'perspective' ORTHOGRAPHIC = 'orthographic' IMAGER = 'imager' ZNEAR = 'znear' ZFAR = 'zfar' XFOV = 'xfov' YFOV = 'yfov' XMAG = 'xmag' YMAG ='ymag' ASPECT_RATIO = 'aspect_ratio' #---geometry--- MESH = 'mesh' CONVEX_MESH ='convex_mesh' SPLINE = 'spline' SOURCE ='source' VERTICES = 'vertices' ACCESSOR = 'accessor' CONVEX_HULL_OF = 'convex_hull_of' #---primitives--- LINES = 'lines' LINESTRIPS = 'linestrips' POLYGONS = 'polygons' POLYLIST = 'polylist' TRIANGLES = 'triangles' TRIFANS = 'trifans' TRISTRIPS = 'tristrips' #---libraries--- LIBRARY_ANIMATIONS = 'library_animations' LIBRARY_ANIMATION_CLIPS = 'library_animation_clips' LIBRARY_CAMERAS = 'library_cameras' LIBRARY_CONTROLLERS = 'library_controllers' LIBRARY_EFFECTS = 'library_effects' LIBRARY_FORCE_FIELDS = 'library_force_fields' LIBRARY_GEOMETRIES = 'library_geometries' LIBRARY_IMAGES = 'library_images' LIBRARY_LIGHTS = 'library_lights' LIBRARY_MATERIALS = 'library_materials' #LIBRARY_NODES = 'library_NODES' #Doug: looks funny - should be library_nodes? LIBRARY_NODES = 'library_nodes' LIBRARY_PHYSICS_MATERIALS = 'library_physics_materials' LIBRARY_PHYSICS_MODELS = 'library_physics_models' LIBRARY_PHYSICS_SCENES = 'library_physics_scenes' LIBRARY_VISUAL_SCENES = 'library_visual_scenes' SCENE = 'scene' EXTRA = 'extra' TYPE = 'type' LIGHT = 'light' CAMERA = 'camera' ANIMATION = 'animation' ANIMATION_CLIP = 'animation_clip' GEOMETRY = 'geometry' IMAGE = 'image' ##EFFECT = 'effect' VISUAL_SCENE = 'visual_scene' CONTROLLER = 'controller' MATERIAL = 'material' #---asset--- CONTRIBUTOR = 'contributor' CREATED = 'created' MODIFIED = 'modified' REVISION = 'revision' TITLE = 'title' SUBJECT = 'subject' KEYWORDS = 'keywords' UNIT = 'unit' UP_AXIS = 'up_axis' Y_UP = 'Y_UP' Z_UP = 'Z_UP' #---contributor--- AUTHOR = 'author' AUTHORING_TOOL = 'authoring_tool' COMMENTS = 'comments' COPYRIGHT = 'copyright' SOURCE_DATA = 'source_data' #---array--- FLOAT_ARRAY = 'float_array' NAME_ARRAY = 'name_array' BOOL_ARRAY = 'bool_array' INT_ARRAY = 'int_array' IDREF_ARRAY = 'IDREF_array' #---node--- NODE = 'node' TYPE_JOINT = 'JOINT' TYPE_NODE = 'NODE' LAYER = 'layer' #---transforms--- TRANSLATE = 'translate' ROTATE = 'rotate' SCALE = 'scale' SKEW = 'skew' MATRIX = 'matrix' LOOKAT = 'lookat' #---instances--- INSTANCE_ANIMATION = 'instance_animation' INSTANCE_CAMERA = 'instance_camera' INSTANCE_CONTROLLER = 'instance_controller' ##INSTANCE_EFFECT = 'instance_effect' INSTANCE_GEOMETRY = 'instance_geometry' INSTANCE_LIGHT = 'instance_light' INSTANCE_NODE = 'instance_node' INSTANCE_VISUAL_SCENE = 'instance_visual_scene' INSTANCE_PHYSICS_SCENE = 'instance_physics_scene' #---image--- FORMAT = 'format' DEPTH = 'depth' HEIGHT = 'height' WIDTH = 'width' INIT_FROM = 'init_from' class DaeFxBindMaterial(DaeEntity): def __init__(self): super(DaeFxBindMaterial, self).__init__() self.syntax = DaeFxSyntax.BIND_MATERIAL self.techniqueCommon = DaeFxBindMaterial.DaeFxTechniqueCommon() def LoadFromXml(self, daeDocument, xmlNode): self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TECHNIQUE_COMMON, DaeFxBindMaterial.DaeFxTechniqueCommon) def SaveToXml(self, daeDocument): node = super(DaeFxBindMaterial,self).SaveToXml(daeDocument) AppendChild(daeDocument, node, self.techniqueCommon) return node class DaeFxTechniqueCommon(DaeEntity): def __init__(self): self.syntax = DaeFxSyntax.TECHNIQUE_COMMON self.iMaterials = [] def LoadFromXml(self, daeDocument, xmlNode): self.iMaterials = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.INSTANCE_MATERIAL, DaeFxMaterialInstance) def SaveToXml(self, daeDocument): node = super(DaeFxBindMaterial.DaeFxTechniqueCommon,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.iMaterials) return node def __str__(self): return super(DaeFxBindMaterial.DaeFxTechniqueCommon,self).__str__() class DaeFxMaterialInstance(DaeEntity): def __init__(self): super(DaeFxMaterialInstance, self).__init__() self.target = '' self.symbol = '' self.object = None self.syntax = DaeFxSyntax.INSTANCE_MATERIAL def LoadFromXml(self, daeDocument, xmlNode): self.target = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.TARGET)[1:] self.symbol = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.SYMBOL) self.object = daeDocument.materialsLibrary.FindObject(self.target) def SaveToXml(self, daeDocument): node = super(DaeFxMaterialInstance,self).SaveToXml(daeDocument) SetAttribute(node, DaeFxSyntax.TARGET, StripString('#'+self.object.id)) SetAttribute(node, DaeFxSyntax.SYMBOL, StripString(self.object.id)) return node class DaeFxMaterial(DaeElement): def __init__(self): super(DaeFxMaterial, self).__init__() self.asset = None self.iEffects = [] self.extras = None self.syntax = DaeFxSyntax.MATERIAL def LoadFromXml(self, daeDocument, xmlNode): super(DaeFxMaterial, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) self.iEffects = CreateObjectsFromXml(daeDocument,xmlNode, DaeFxSyntax.INSTANCE_EFFECT, DaeFxEffectInstance) def SaveToXml(self, daeDocument): node = super(DaeFxMaterial,self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the effect instances AppendChilds(daeDocument, node, self.iEffects) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaeFxMaterial,self).__str__()+' assets: %s, iEffects: %s, extras: %s'%(self.asset, self.iEffects, self.extras) class DaeFxEffectInstance(DaeEntity): def __init__(self): super(DaeFxEffectInstance, self).__init__() self.url = '' self.syntax = DaeFxSyntax.INSTANCE_EFFECT def LoadFromXml(self, daeDocument, xmlNode): self.url = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.URL)[1:] self.object = daeDocument.effectsLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaeFxEffectInstance,self).SaveToXml(daeDocument) SetAttribute(node, DaeFxSyntax.URL, StripString('#'+self.object.id)) return node class DaeFxEffect(DaeElement): def __init__(self): super(DaeFxEffect, self).__init__() self.profileCommon = DaeFxProfileCommon() self.syntax = DaeFxSyntax.EFFECT def LoadFromXml(self, daeDocument, xmlNode): super(DaeFxEffect, self).LoadFromXml(daeDocument, xmlNode) self.profileCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PROFILE_COMMON, DaeFxProfileCommon) def SaveToXml(self, daeDocument): node = super(DaeFxEffect,self).SaveToXml(daeDocument) AppendChild(daeDocument, node, self.profileCommon) return node def __str__(self): return super(DaeFxEffect, self).__str__() + ', profileCommon: %s' % (self.profileCommon) def AddShader(self, daeShader): self.profileCommon.technique.shader = daeShader def AddDoubleSided(self, val): if self.profileCommon.extra == None: self.profileCommon.AddExtra() self.profileCommon.extra.technique.AddDoubleSided(val) class DaeFxProfileCommon(DaeEntity): def __init__(self): super(DaeFxProfileCommon, self).__init__() self.technique = DaeFxTechnique() self.images = [] self.newParams = [] self.syntax = DaeFxSyntax.PROFILE_COMMON self.extra = None #Doug: added Mar3008 def LoadFromXml(self, daeDocument, xmlNode): self.images = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.IMAGE, DaeFxImage) self.newParams = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.NEWPARAM, DaeFxNewParam) self.technique = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TECHNIQUE, DaeFxTechnique) self.extra = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.EXTRA, DaeFxExtra) #Doug: added Mar3008 def SaveToXml(self, daeDocument): node = super(DaeFxProfileCommon,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.images) AppendChilds(daeDocument, node, self.newParams) AppendChild(daeDocument, node, self.technique) AppendChild(daeDocument, node, self.extra) #Doug: added Mar3008 return node def AddExtra(self): if not self.extra: self.extra = DaeFxExtra() def __str__(self): return super(DaeFxProfileCommon, self).__str__() + ', technique: %s, images: %s, newParams: %s' % (self.technique, self.images, self.newParams) class DaeFxExtra(DaeEntity): #Doug: added Mar3008 - there is a DaeExtra which could be used instead or for guidance def __init__(self): super(DaeFxExtra, self).__init__() self.technique = DaeFxTechnique() #self.images = [] #self.newParams = [] self.syntax = DaeFxSyntax.EXTRA #self.extra = [] #Doug: added Mar3008 def LoadFromXml(self, daeDocument, xmlNode): #self.images = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.IMAGE, DaeFxImage) #self.newParams = CreateObjectsFromXml(daeDocument, xmlNode, DaeFxSyntax.NEWPARAM, DaeFxNewParam) self.technique = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TECHNIQUE, DaeFxTechnique) #self.extra = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.EXTRA, DaeFxExtra) def SaveToXml(self, daeDocument): node = super(DaeFxExtra,self).SaveToXml(daeDocument) #AppendChilds(daeDocument, node, self.images) #AppendChilds(daeDocument, node, self.newParams) AppendChild(daeDocument, node, self.technique) return node def __str__(self): #return super(DaeFxExtra, self).__str__() + ', technique: %s, images: %s, newParams: %s' % (self.technique, self.images, self.newParams) return super(DaeFxExtra, self).__str__() + ', technique: %s' % (self.technique) class DaeFxImage(DaeEntity): def __init__(self): super(DaeFxImage, self).__init__() self.syntax = DaeFxSyntax.IMAGE def LoadFromXml(self, daeDocument, xmlNode): super(DaeFxImage, self).LoadFromXml(daeDocument, xmlNode) def SaveToXml(self, daeDocument): node = super(DaeFxImage,self).SaveToXml(daeDocument) return node class DaeFxNewParam(DaeEntity): def __init__(self): super(DaeFxNewParam, self).__init__() self.syntax = DaeFxSyntax.NEWPARAM def LoadFromXml(self, daeDocument, xmlNode): super(DaeFxNewParam, self).LoadFromXml(daeDocument, xmlNode) def SaveToXml(self, daeDocument): node = super(DaeFxNewParam,self).SaveToXml(daeDocument) return node class DaeFxTechnique(DaeEntity): def __init__(self): super(DaeFxTechnique, self).__init__() self.syntax = DaeFxSyntax.TECHNIQUE self.shader = DaeFxShadeConstant() self.sid = '' self.doubleside = DaeFxDoubleSided() #Doug: added Mar3008 def LoadFromXml(self, daeDocument, xmlNode): # TODO: add asset and extra? self.sid = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.SID) #Doug: added Mar3008 the following 5 lines about googleearth profile and doublesided profile = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.PROFILE) if profile == DaeFxSyntax.GOOGLEEARTH: lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.DOUBLESIDED) if lightSourceNode: self.doubleside = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.DOUBLE_SIDED, DaeFxDoubleSided) else: lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.CONSTANT) if lightSourceNode: self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.CONSTANT, DaeFxShadeConstant) else: lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.LAMBERT) if lightSourceNode: self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.LAMBERT, DaeFxShadeLambert) else: lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.BLINN) if lightSourceNode: self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.BLINN, DaeFxShadeBlinn) else: lightSourceNode = xmlUtils.FindElementByTagName(xmlNode, DaeFxSyntax.PHONG) self.shader = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PHONG, DaeFxShadePhong) def SaveToXml(self, daeDocument): node = super(DaeFxTechnique,self).SaveToXml(daeDocument) if self.doubleside.value: node.setAttribute(DaeFxSyntax.PROFILE, DaeFxSyntax.GOOGLEEARTH) AppendChild(daeDocument, node, self.doubleside) else: node.setAttribute(DaeFxSyntax.SID, self.sid) AppendChild(daeDocument, node, self.shader) return node def AddDoubleSided(self, val): #Doug: added Mar3008 self.doubleside.AddValue(val) def __str__(self): if self.doubleside.doublesided: return super(DaeFxTechnique,self).__str__()+' doublesided: %s'%(self.doubleside) else: return super(DaeFxTechnique,self).__str__()+' shader: %s'%(self.shader) class DaeFxDoubleSided(DaeEntity): #Doug: added Mar3008 def __init__(self): super(DaeFxDoubleSided, self).__init__() self.value = None self.syntax = DaeFxSyntax.DOUBLE_SIDED def LoadFromXml(self, daeDocument, xmlNode): #self.value = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.DOUBLE_SIDED, DaeFxCommonContainer, True) self.value = CastFromXml(daeDocument, xmlNode, self.syntax, int) def SaveToXml(self, daeDocument): node = super(DaeFxDoubleSided,self).SaveToXml(daeDocument) #def AppendTextInChild(xmlNode, object): AppendTextInChild(node,int(self.value)) #AppendTextChild(node, self.syntax, int(self.value)) #not quie the right thing- dont need syntax again return node def AddValue(self, val): self.value = int(val) class DaeFxShadeConstant(DaeEntity): def __init__(self): super(DaeFxShadeConstant, self).__init__() self.emission = None self.reflective = None self.reflectivity = None self.transparent = None self.transparency = None self.indexOfRefraction = None self.syntax = DaeFxSyntax.CONSTANT def LoadFromXml(self, daeDocument, xmlNode): self.emission = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.EMISSION, DaeFxCommonColorAndTextureContainer, True) self.reflective = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.REFLECTIVE, DaeFxCommonColorAndTextureContainer, True) self.reflectivity = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.REFLECTIVITY, DaeFxCommonFloatAndParamContainer, True) self.transparent = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TRANSPARENT, DaeFxCommonColorAndTextureContainer, True) self.transparency = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TRANSPARENCY, DaeFxCommonFloatAndParamContainer, True) self.indexOfRefraction = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.INDEXOFREFRACTION, DaeFxCommonFloatAndParamContainer, True) def SaveToXml(self, daeDocument): node = super(DaeFxShadeConstant,self).SaveToXml(daeDocument) AppendChild(daeDocument, node, self.emission) if isinstance(self, DaeFxShadeLambert): AppendChild(daeDocument, node, self.ambient) AppendChild(daeDocument, node, self.diffuse) if isinstance(self,DaeFxShadeBlinn): AppendChild(daeDocument, node, self.specular) AppendChild(daeDocument, node, self.shininess) AppendChild(daeDocument, node, self.reflective) AppendChild(daeDocument, node, self.reflectivity) AppendChild(daeDocument, node, self.transparent) AppendChild(daeDocument, node, self.transparency) AppendChild(daeDocument, node, self.indexOfRefraction) return node def AddValue(self, type, val): col = DaeFxColor() col.rgba = val if type == DaeFxSyntax.EMISSION: if not self.emission: self.emission = DaeFxCommonColorAndTextureContainer(type) self.emission.color = col elif type == DaeFxSyntax.REFLECTIVE: if not self.reflective: self.reflective = DaeFxCommonColorAndTextureContainer(type) self.reflective.color = col elif type == DaeFxSyntax.REFLECTIVITY: if not self.reflectivity: self.reflectivity = DaeFxCommonFloatAndParamContainer(type) self.reflectivity.float = val elif type == DaeFxSyntax.TRANSPARENT: if not self.transparent: self.transparent = DaeFxCommonColorAndTextureContainer(type) self.transparent.color = col elif type == DaeFxSyntax.TRANSPARENCY: if not self.transparency: self.transparency = DaeFxCommonFloatAndParamContainer(type) self.transparency.float = val elif type == DaeFxSyntax.INDEXOFREFRACTION: if not self.indexOfRefraction: self.indexOfRefraction = DaeFxCommonFloatAndParamContainer(type) self.indexOfRefraction.float = val else: Debug.Debug('DaeFxShadeConstant: type: %s not recognised'%(type),'ERROR') class DaeFxShadeLambert(DaeFxShadeConstant): def __init__(self): super(DaeFxShadeLambert, self).__init__() self.ambient = None self.diffuse = None self.syntax = DaeFxSyntax.LAMBERT def LoadFromXml(self, daeDocument, xmlNode): super(DaeFxShadeLambert, self).LoadFromXml(daeDocument, xmlNode) self.ambient = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.AMBIENT, DaeFxCommonColorAndTextureContainer, True) self.diffuse = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.DIFFUSE, DaeFxCommonColorAndTextureContainer, True) def SaveToXml(self, daeDocument): node = super(DaeFxShadeLambert,self).SaveToXml(daeDocument) #AppendChild(daeDocument, node, self.ambient) #AppendChild(daeDocument, node, self.diffuse) return node def AddValue(self, type, val): col = DaeFxColor() col.rgba = val if type == DaeFxSyntax.DIFFUSE: if not self.diffuse: self.diffuse = DaeFxCommonColorAndTextureContainer(type) if isinstance(val, DaeFxTexture): # its a texture self.diffuse.texture = val else: # it's a color self.diffuse.color = col elif type == DaeFxSyntax.AMBIENT: if not self.ambient: self.ambient = DaeFxCommonColorAndTextureContainer(type) self.ambient.color = col else: super(DaeFxShadeLambert,self).AddValue(type, val) class DaeFxShadeBlinn(DaeFxShadeLambert): def __init__(self): super(DaeFxShadeBlinn, self).__init__() self.specular = None self.shininess = None self.syntax = DaeFxSyntax.BLINN def LoadFromXml(self, daeDocument, xmlNode): super(DaeFxShadeBlinn, self).LoadFromXml(daeDocument, xmlNode) self.specular = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.SPECULAR, DaeFxCommonColorAndTextureContainer, True) self.shininess = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.SHININESS, DaeFxCommonFloatAndParamContainer, True) def SaveToXml(self, daeDocument): node = super(DaeFxShadeBlinn,self).SaveToXml(daeDocument) #AppendChild(daeDocument, node, self.specular) #AppendChild(daeDocument, node, self.shininess) return node def AddValue(self, type, val): col = DaeFxColor() col.rgba = val if type == DaeFxSyntax.SPECULAR: if not self.specular: self.specular = DaeFxCommonColorAndTextureContainer(type) self.specular.color = col elif type == DaeFxSyntax.SHININESS: if not self.shininess: self.shininess = DaeFxCommonFloatAndParamContainer(type) self.shininess.float = val else: super(DaeFxShadeBlinn,self).AddValue(type, val) class DaeFxShadePhong(DaeFxShadeBlinn): def __init__(self): super(DaeFxShadePhong, self).__init__() self.syntax = DaeFxSyntax.PHONG class DaeFxCommonColorAndTextureContainer(DaeEntity): def __init__(self, syntax='UNKNOWN'): super(DaeFxCommonColorAndTextureContainer, self).__init__() self.color = None self.texture = None self.syntax = syntax def LoadFromXml(self, daeDocument, xmlNode): self.color = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.COLOR, DaeFxColor) self.texture = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.TEXTURE, DaeFxTexture) def SaveToXml(self, daeDocument): node = super(DaeFxCommonColorAndTextureContainer,self).SaveToXml(daeDocument) AppendChild(daeDocument, node, self.color) AppendChild(daeDocument, node, self.texture) return node class DaeFxCommonFloatAndParamContainer(DaeEntity): def __init__(self, syntax = 'UNKNOWN'): super(DaeFxCommonFloatAndParamContainer, self).__init__() self.float = None self.param = None self.syntax = syntax def LoadFromXml(self, daeDocument, xmlNode): self.float = CastFromXml(daeDocument, xmlNode, DaeFxSyntax.FLOAT, float) ##self.param = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PARAM, DaeFxParam) def SaveToXml(self, daeDocument): node = super(DaeFxCommonFloatAndParamContainer,self).SaveToXml(daeDocument) AppendTextChild(node, DaeFxSyntax.FLOAT, self.float) AppendChild(daeDocument, node, self.param) return node class DaeFxCommonContainer(DaeEntity): #Doug: added Mar3008 def __init__(self, cast = str): super(DaeFxCommonContainer, self).__init__() self.value = None self.cast = cast self.syntax = syntax def LoadFromXml(self, daeDocument, xmlNode): self.value = CastFromXml(daeDocument, xmlNode, self.syntax, self.cast) #def CastFromXml(colladaDocument, xmlNode, nodeType, cast, default=None): ##self.param = CreateObjectFromXml(daeDocument, xmlNode, DaeFxSyntax.PARAM, DaeFxParam) def SaveToXml(self, daeDocument): node = super(DaeFxCommonContainer,self).SaveToXml(daeDocument) AppendTextChild(node, self.syntax, self.cast(self.value)) #def AppendTextChild(xmlNode,syntax, object, default = None): #AppendChild(daeDocument, node, false) return node class DaeFxColor(DaeEntity): def __init__(self): super(DaeFxColor, self).__init__() self.sid = '' self.rgba = [] self.syntax = DaeFxSyntax.COLOR def LoadFromXml(self, daeDocument, xmlNode): self.sid = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.SID) self.rgba = ToFloatList(xmlUtils.ReadContents(xmlNode)) def SaveToXml(self, daeDocument): node = super(DaeFxColor,self).SaveToXml(daeDocument) SetAttribute(node, DaeFxSyntax.SID, self.sid) AppendTextInChild(node, self.rgba) return node class DaeFxTexture(DaeEntity): def __init__(self): super(DaeFxTexture, self).__init__() self.texture = '' self.textCoord = '' self.syntax = DaeFxSyntax.TEXTURE def LoadFromXml(self, daeDocument, xmlNode): self.texture = daeDocument.imagesLibrary.FindObject(xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.TEXTURE)) self.textCoord = xmlUtils.ReadAttribute(xmlNode, DaeFxSyntax.TEXCOORD) def SaveToXml(self, daeDocument): node = super(DaeFxTexture,self).SaveToXml(daeDocument) SetAttribute(node, DaeFxSyntax.TEXTURE, StripString(self.texture)) SetAttribute(node, DaeFxSyntax.TEXCOORD, self.textCoord) return node class DaeFxSyntax(object): COLOR = 'color' INSTANCE_MATERIAL = 'instance_material' INSTANCE_EFFECT = 'instance_effect' TECHNIQUE_COMMON = 'technique_common' SID = 'sid' EMISSION = 'emission' REFLECTIVE ='reflective' REFLECTIVITY = 'reflectivity' TRANSPARENT = 'transparent' TRANSPARENCY = 'transparency' INDEXOFREFRACTION = 'index_of_refraction' TEXTURE = 'texture' TEXCOORD = 'texcoord' AMBIENT = 'ambient' DIFFUSE = 'diffuse' BIND_MATERIAL ='bind_material' PROFILE_COMMON = 'profile_COMMON' SYMBOL = 'symbol' MATERIAL = 'material' EFFECT = 'effect' TARGET = 'target' URL = 'url' SYMBOL = 'symbol' BLINN = 'blinn' SHININESS = 'shininess' SPECULAR = 'specular' PHONG = 'phong' IMAGE = 'image' NEWPARAM = 'newparam' TECHNIQUE = 'technique' EXTRA = 'extra' #Doug: added Mar3008 DOUBLE_SIDED = 'double_sided' #Doug: added Mar3008 PROFILE = 'profile' #Doug: added Mar3008 GOOGLEEARTH = 'GOOGLEEARTH' #Doug: added Mar3008 CONSTANT = 'constant' LAMBERT = 'lambert' FLOAT ='float' #---COLLADA PHYSICS--- class DaePhysicsScene(DaeElement): def __init__(self): super(DaePhysicsScene,self).__init__() self.asset = None self.extras = None self.iPhysicsModels = [] self.syntax = DaePhysicsSyntax.PHYSICS_SCENE def LoadFromXml(self, daeDocument, xmlNode): super(DaePhysicsScene, self).LoadFromXml(daeDocument, xmlNode) self.extras = CreateObjectsFromXml(daeDocument, xmlNode, DaeSyntax.EXTRA, DaeExtra) self.asset = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.ASSET, DaeAsset) self.iPhysicsModels = CreateObjectsFromXml(daeDocument, xmlNode, DaePhysicsSyntax.INSTANCE_PHYSICS_MODEL, DaePhysicsModelInstance) def SaveToXml(self, daeDocument): node = super(DaePhysicsScene, self).SaveToXml(daeDocument) # Add the assets AppendChild(daeDocument,node,self.asset) # Add the phyics models AppendChilds(daeDocument, node, self.iPhysicsModels) # Add the extra's AppendChilds(self,node,self.extras) return node def __str__(self): return super(DaePhysicsScene,self).__str__()+' asset: %s, iPhysicsModels: %s, extras: %s'%(self.asset, self.iPhysicsModels, self.extras) class DaePhysicsModelInstance(DaeInstance): def __init__(self): super(DaePhysicsModelInstance, self).__init__() self.syntax = DaePhysicsSyntax.INSTANCE_PHYSICS_MODEL self.iRigidBodies = [] def LoadFromXml(self, daeDocument, xmlNode): super(DaePhysicsModelInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.physicsModelsLibrary.FindObject(self.url) self.iRigidBodies = self.CreateInstanceRigidBodies(daeDocument, xmlNode, self.object) def SaveToXml(self, daeDocument): node = super(DaePhysicsModelInstance,self).SaveToXml(daeDocument) AppendChilds(daeDocument, node, self.iRigidBodies) return node def CreateInstanceRigidBodies(self, daeDocument, xmlNode, physicsModel): objects = [] CreateObjectsFromXml(daeDocument, xmlNode, DaePhysicsSyntax.INSTANCE_RIGID_BODY, DaeRigidBodyInstance) nodes = xmlUtils.FindElementsByTagName(xmlNode,DaePhysicsSyntax.INSTANCE_RIGID_BODY) for node in nodes: object = DaeRigidBodyInstance() object.LoadFromXml(daeDocument, node) object.body = physicsModel.FindRigidBody(object.bodyString) n = None for visualScene in daeDocument.visualScenesLibrary.items: n = visualScene.FindNode(object.targetString) if not (n is None): break object.target = n objects.append(object) return objects class DaePhysicsSceneInstance(DaeInstance): def __init__(self): super(DaePhysicsSceneInstance, self).__init__() self.syntax = DaeSyntax.INSTANCE_PHYSICS_SCENE def LoadFromXml(self, daeDocument, xmlNode): super(DaePhysicsSceneInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.physicsScenesLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaePhysicsSceneInstance,self).SaveToXml(daeDocument) return node class DaePhysicsMaterialInstance(DaeInstance): def __init__(self): super(DaePhysicsMaterialInstance, self).__init__() self.syntax = DaePhysicsSyntax.INSTANCE_PHYSICS_MATERIAL def LoadFromXml(self, daeDocument, xmlNode): super(DaePhysicsMaterialInstance,self).LoadFromXml(daeDocument, xmlNode) self.object = daeDocument.physicsMaterialsLibrary.FindObject(self.url) def SaveToXml(self, daeDocument): node = super(DaePhysicsMaterialInstance,self).SaveToXml(daeDocument) return node class DaeRigidBodyInstance(DaeEntity): def __init__(self): super(DaeRigidBodyInstance, self).__init__() self.syntax = DaePhysicsSyntax.INSTANCE_RIGID_BODY self.body = None self.target = None self.bodyString = '' self.targetString = '' def LoadFromXml(self, daeDocument, xmlNode): self.bodyString = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.BODY) self.targetString = xmlUtils.ReadAttribute(xmlNode, DaeSyntax.TARGET)[1:] def SaveToXml(self, daeDocument): node = super(DaeRigidBodyInstance,self).SaveToXml(daeDocument) SetAttribute(node, DaeSyntax.BODY, StripString(self.body.name)) SetAttribute(node, DaeSyntax.TARGET, StripString('#'+self.target.id)) return node class DaePhysicsModel(DaeElement): def __init__(self): super(DaePhysicsModel,self).__init__() self.syntax = DaePhysicsSyntax.PHYSICS_MODEL self.rigidBodies = [] def LoadFromXml(self, daeDocument, xmlNode): super(DaePhysicsModel, self).LoadFromXml(daeDocument, xmlNode) self.rigidBodies = CreateObjectsFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RIGID_BODY, DaeRigidBody) def SaveToXml(self, daeDocument): node = super(DaePhysicsModel, self).SaveToXml(daeDocument) # Add the rigid bodies AppendChilds(daeDocument, node, self.rigidBodies) return node def FindRigidBody(self, url): for rigidBody in self.rigidBodies: if rigidBody.sid == url: return rigidBody return None class DaePhysicsMaterial(DaeElement): def __init__(self): super(DaePhysicsMaterial,self).__init__() self.syntax = DaePhysicsSyntax.PHYSICS_MATERIAL self.techniqueCommon = DaePhysicsMaterial.DaeTechniqueCommon() def LoadFromXml(self, daeDocument, xmlNode): super(DaePhysicsMaterial, self).LoadFromXml(daeDocument, xmlNode) self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE_COMMON, DaePhysicsMaterial.DaeTechniqueCommon) def SaveToXml(self, daeDocument): node = super(DaePhysicsMaterial, self).SaveToXml(daeDocument) AppendChild(daeDocument,node,self.techniqueCommon) return node class DaeTechniqueCommon(DaeEntity): def __init__(self): super(DaePhysicsMaterial.DaeTechniqueCommon,self).__init__() self.syntax = DaeSyntax.TECHNIQUE_COMMON self.dynamicFriction = 0 self.restitution = 0 self.staticFriction = 0 def LoadFromXml(self, daeDocument, xmlNode): self.dynamicFriction = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.DYNAMIC_FRICTION, float, 0) self.restitution = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RESTITUTION, float, 0) self.staticFriction = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.STATIC_FRICTION, float, 0) def SaveToXml(self, daeDocument): node = super(DaePhysicsMaterial.DaeTechniqueCommon,self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.DYNAMIC_FRICTION, self.dynamicFriction, None) AppendTextChild(node, DaePhysicsSyntax.RESTITUTION, self.restitution, None) AppendTextChild(node, DaePhysicsSyntax.STATIC_FRICTION, self.staticFriction, None) return node def __str__(self): return super(DaePhysicsMaterial.DaeTechniqueCommon,self).__str__() class DaeRigidBody(DaeEntity): def __init__(self): super(DaeRigidBody, self).__init__() self.syntax = DaePhysicsSyntax.RIGID_BODY self.name = '' self.sid = '' self.techniqueCommon = DaeRigidBody.DaeTechniqueCommon() def LoadFromXml(self, daeDocument, xmlNode): self.name = xmlNode.getAttribute(DaeSyntax.NAME) self.sid = xmlNode.getAttribute(DaeSyntax.SID) self.techniqueCommon = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.TECHNIQUE_COMMON, DaeRigidBody.DaeTechniqueCommon) def SaveToXml(self, daeDocument): node = super(DaeRigidBody, self).SaveToXml(daeDocument) SetAttribute(node,DaeSyntax.NAME, StripString(self.name)) SetAttribute(node,DaeSyntax.SID, StripString(self.name)) AppendChild(daeDocument,node,self.techniqueCommon) return node class DaeTechniqueCommon(DaeEntity): def __init__(self): super(DaeRigidBody.DaeTechniqueCommon, self).__init__() self.syntax = DaeSyntax.TECHNIQUE_COMMON self.iPhysicsMaterial = None self.physicsMaterial = None self.dynamic = True self.mass = None self.inertia = None self.shapes = [] def LoadFromXml(self, daeDocument, xmlNode): self.iPhysicsMaterial = CreateObjectFromXml(daeDocument, xmlNode, DaePhysicsSyntax.INSTANCE_PHYSICS_MATERIAL, DaePhysicsMaterialInstance) self.physicsMaterial = CreateObjectFromXml(daeDocument, xmlNode, DaePhysicsSyntax.PHYSICS_MATERIAL, DaePhysicsMaterial) self.dynamic = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.DYNAMIC,bool,True) self.mass = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.MASS, float, 1) self.inertia = ToFloat3(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaePhysicsSyntax.INERTIA))) shapeNodes = xmlUtils.FindElementsByTagName(xmlNode, DaePhysicsSyntax.SHAPE) for shapeNode in shapeNodes: s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.BOX) b = None if not (s is None): b = DaeBoxShape() else: s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.SPHERE) if not (s is None): b = DaeSphereShape() else: s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.PLANE) if not (s is None): b = DaePlaneShape() else: s = xmlUtils.FindElementByTagName(shapeNode, DaeSyntax.INSTANCE_GEOMETRY) if not (s is None): b = DaeGeometryShape() else: s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.CYLINDER) if not (s is None): b = DaeCylinderShape() else: s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.TAPERED_CYLINDER) if not (s is None): b = DaeTaperedCylinderShape() else: s = xmlUtils.FindElementByTagName(shapeNode, DaePhysicsSyntax.CAPSULE) if not (s is None): b = DaeCapsule() else: # TAPERED_CAPSULE b = DaeTaperedCapsuleShape() b.LoadFromXml(daeDocument, s) self.shapes.append(b) def SaveToXml(self, daeDocument): node = super(DaeRigidBody.DaeTechniqueCommon,self).SaveToXml(daeDocument) AppendChild(daeDocument,node,self.iPhysicsMaterial) AppendChild(daeDocument,node,self.physicsMaterial) shapes = Element(DaePhysicsSyntax.SHAPE) AppendChilds(daeDocument, shapes, self.shapes) node.appendChild(shapes) AppendTextChild(node, DaePhysicsSyntax.DYNAMIC, self.dynamic, None) AppendTextChild(node, DaePhysicsSyntax.MASS, self.mass, None) AppendTextChild(node, DaePhysicsSyntax.INERTIA, self.inertia, None) return node def GetPhysicsMaterial(self): if not (self.physicsMaterial is None): return self.physicsMaterial else: return self.iPhysicsMaterial.object def __str__(self): return super(DaeRigidBody.DaeTechniqueCommon,self).__str__() class DaeShape(DaeEntity): def __init__(self): super(DaeShape, self).__init__() self.mass = None self.density = None self.syntax = DaePhysicsSyntax.SHAPE def LoadFromXml(self, daeDocument, xmlNode): self.iGeometry = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_GEOMETRY,DaeGeometryInstance) self.mass = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.MASS, float, None) self.density = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.DENSITY, float, None) def SaveToXml(self, daeDocument): node = super(DaeShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.MASS, self.mass, None) AppendTextChild(node, DaePhysicsSyntax.DENSITY, self.density, None) return node class DaeBoxShape(DaeShape): def __init__(self): super(DaeBoxShape, self).__init__() self.halfExtents = [] self.syntax = DaePhysicsSyntax.BOX def LoadFromXml(self, daeDocument, xmlNode): super(DaeBoxShape, self).LoadFromXml(daeDocument, xmlNode) self.halfExtents = ToFloat3(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaePhysicsSyntax.HALF_EXTENTS))) def SaveToXml(self, daeDocument): node = super(DaeBoxShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.HALF_EXTENTS, self.halfExtents) return node class DaeSphereShape(DaeShape): def __init__(self): super(DaeSphereShape, self).__init__() self.radius = None self.syntax = DaePhysicsSyntax.SPHERE def LoadFromXml(self, daeDocument, xmlNode): super(DaeSphereShape, self).LoadFromXml(daeDocument, xmlNode) self.radius = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RADIUS, float) def SaveToXml(self, daeDocument): node = super(DaeSphereShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.RADIUS, self.radius) return node class DaeCylinderShape(DaeShape): def __init__(self): super(DaeCylinderShape, self).__init__() self.radius = [1 , 1] self.height = None self.syntax = DaePhysicsSyntax.CYLINDER def LoadFromXml(self, daeDocument, xmlNode): super(DaeCylinderShape, self).LoadFromXml(daeDocument, xmlNode) self.radius = ToFloat2(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaePhysicsSyntax.RADIUS)),'Not a valid radius found. Must consist of 2 floats') self.height = CastFromXml(daeDocument, xmlNode, DaeSyntax.HEIGHT, float) def SaveToXml(self, daeDocument): node = super(DaeCylinderShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.RADIUS, self.radius) AppendTextChild(node, DaeSyntax.HEIGHT, self.height) return node class DaeTaperedCylinderShape(DaeShape): def __init__(self): super(DaeTaperedCylinderShape, self).__init__() self.radius1 = [1 , 1] self.radius2 = [1 , 1] self.height = None self.syntax = DaePhysicsSyntax.TAPERED_CYLINDER def LoadFromXml(self, daeDocument, xmlNode): super(DaeTaperedCylinderShape, self).LoadFromXml(daeDocument, xmlNode) self.radius1 = ToFloat2(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaePhysicsSyntax.RADIUS1)),'Not a valid radius found. Must consist of 2 floats') self.radius2 = ToFloat2(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode, DaePhysicsSyntax.RADIUS2)), 'Not a valid radius found. Must consist of 2 floats') self.height = CastFromXml(daeDocument, xmlNode, DaeSyntax.HEIGHT, float) def SaveToXml(self, daeDocument): node = super(DaeTaperedCylinderShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.RADIUS1, self.radius1) AppendTextChild(node, DaePhysicsSyntax.RADIUS2, self.radius2) AppendTextChild(node, DaeSyntax.HEIGHT, self.height) return node class DaePlaneShape(DaeShape): def __init__(self): super(DaePlaneShape, self).__init__() self.equation = [] self.syntax = DaePhysicsSyntax.PLANE def LoadFromXml(self, daeDocument, xmlNode): super(DaePlaneShape, self).LoadFromXml(daeDocument, xmlNode) self.equation = ToFloat4(xmlUtils.ReadContents(xmlUtils.FindElementByTagName(xmlNode,DaePhysicsSyntax.EQUATION))) def SaveToXml(self, daeDocument): node = super(DaePlaneShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.EQUATION, self.equation) return node class DaeCapsuleShape(DaeShape): def __init__(self): super(DaeCapsuleShape, self).__init__() self.radius = None self.height = None self.syntax = DaePhysicsSyntax.CAPSULE def LoadFromXml(self, daeDocument, xmlNode): super(DaeCapsuleShape, self).LoadFromXml(daeDocument, xmlNode) self.radius = CastFromXml(daeDocument, xmlNode, DaePhysicsSyntax.RADIUS, float) self.height = CastFromXml(daeDocument, xmlNode, DaeSyntax.HEIGHT, float) def SaveToXml(self, daeDocument): node = super(DaeCapsuleShape, self).SaveToXml(daeDocument) AppendTextChild(node, DaePhysicsSyntax.RADIUS, self.radius) AppendTextChild(node, DaeSyntax.HEIGHT, self.height) return node class DaeGeometryShape(DaeShape): def __init__(self): super(DaeGeometryShape, self).__init__() self.iGeometry = None def LoadFromXml(self, daeDocument, xmlNode): super(DaeGeometryShape, self).LoadFromXml(daeDocument, xmlNode) self.iGeometry = DaeGeometryInstance() self.iGeometry.LoadFromXml(daeDocument, xmlNode) #self.iGeometry = CreateObjectFromXml(daeDocument, xmlNode, DaeSyntax.INSTANCE_GEOMETRY, DaeGeometryInstance) #print self.iGeometry #print xmlNode.toxml() def SaveToXml(self, daeDocument): #node = super(DaeGeometryShape, self).SaveToXml(daeDocument) #AppendChild(daeDocument, node, self.iGeometry) return self.iGeometry.SaveToXml(daeDocument) class DaePhysicsSyntax(object): PHYSICS_SCENE = 'physics_scene' PHYSICS_MODEL = 'physics_model' PHYSICS_MATERIAL = 'physics_material' RIGID_BODY = 'rigid_body' INSTANCE_PHYSICS_MODEL = 'instance_physics_model' INSTANCE_PHYSICS_MATERIAL = 'instance_physics_material' INSTANCE_RIGID_BODY = 'instance_rigid_body' RESTITUTION = 'restitution' STATIC_FRICTION = 'static_friction' DYNAMIC_FRICTION = 'dynamic_friction' DYNAMIC = 'dynamic' MASS = 'mass' INERTIA = 'inertia' SHAPE = 'shape' DENSITY = 'density' RADIUS = 'radius' RADIUS1 = 'radius1' RADIUS2 = 'radius2' BOX = 'box' PLANE = 'plane' CYLINDER = 'cylinder' SPHERE = 'sphere' CAPSULE = 'capsule' TAPERED_CAPSULE ='tapered_capsule' TAPERED_CYLINDER = 'tapered_cylinder' HALF_EXTENTS = 'half_extents' #---Functions--- def CreateObjectsFromXml(colladaDocument, xmlNode, nodeType, objectType): if xmlNode is None: return None objects = [] nodes = xmlUtils.FindElementsByTagName(xmlNode,nodeType) for node in nodes: object = objectType() object.LoadFromXml(colladaDocument, node) objects.append(object) return objects def CreateObjectFromXml(colladaDocument, xmlNode, nodeType, objectType, setSyntax = False): if xmlNode is None: return None node = xmlUtils.FindElementByTagName(xmlNode, nodeType) object = None if setSyntax: object = objectType(nodeType) else: object = objectType() if node != None: object.LoadFromXml(colladaDocument, node) return object return None def CastFromXml(colladaDocument, xmlNode, nodeType, cast, default=None): if xmlNode is None: return default node = xmlUtils.FindElementByTagName(xmlNode, nodeType) if node != None: textValue = xmlUtils.ReadContents(node) if cast == bool: if textValue.lower() == 'false': return False else: return True return cast(textValue) return default def CastAttributeFromXml(xmlNode, nodeType, cast, default=None): if xmlNode is None: return default val = xmlUtils.ReadAttribute(xmlNode, nodeType) if val != None and val != '': return cast(val) return default def AppendChild(daeDocument, xmlNode, daeEntity): if daeEntity is None or xmlNode is None: return else: child = daeEntity.SaveToXml(daeDocument) if child is None: return else : xmlNode.appendChild(child) def AppendChilds(daeDocument, xmlNode, daeEntities): if daeEntities is None or xmlNode is None: return else: for daeEntity in daeEntities: AppendChild(daeDocument, xmlNode, daeEntity) def AppendTextChild(xmlNode,syntax, object, default = None): if object is None: return if default != None and object == default: return node = Element(syntax) xmlNode.appendChild(node) return AppendTextInChild(node, object) def AppendTextInChild(xmlNode, object): if object is None: return text = Text() if type(object) == datetime: text.data = object.isoformat()##xmlUtils.ToDateTime(object) elif type(object) == list: if len(object) == 0: return if object[0] is not None and type(object[0]) == float: object = RoundList(object, ROUND) text.data = ListToString(object) elif type(object) == float: text.data = round(object, ROUND) elif type(object) == bool: text.data = str(object).lower() elif type(object) == int: text.data = str(object) #Doug: added Mar3008 - same as default below so should have no effect else: text.data = str(object) xmlNode.appendChild(text) return xmlNode def SetAttribute(xmlNode,syntax, object): if xmlNode is None or object is None or str(object) == '': return xmlNode.setAttribute(syntax,str(object)) def ReadNodeUrl(node): attribute = xmlUtils.ReadAttribute(node,DaeSyntax.URL) if attribute == None: return None else : attribute = str(attribute) if attribute.startswith('#'): return attribute[1:] return None def WriteNodeUrl(node, url): node.setAttribute(DaeSyntax.URL, StripString('#'+url)) def IsVersionOk(version, curVersion): versionAr = version.split('.') curVersionAr = curVersion.split('.') for i in range(len(curVersionAr)): if versionAr[i] != curVersionAr[i]: return False return True def StripString(text): return text.replace(' ','_').replace('.','_')
|
