def createAxis():
geode = osg.Geode()
geometry = osg.Geometry()
vertices = osg.Vec3Array()
vertices.push_back(osg.Vec3(0.0, 0.0, 0.0))
vertices.push_back(osg.Vec3(1.0, 0.0, 0.0))
vertices.push_back(osg.Vec3(0.0, 0.0, 0.0))
vertices.push_back(osg.Vec3(0.0, 1.0, 0.0))
vertices.push_back(osg.Vec3(0.0, 0.0, 0.0))
vertices.push_back(osg.Vec3(0.0, 0.0, 1.0))
geometry.setVertexArray(vertices)
colors = osg.Vec4Array()
colors.push_back(osg.Vec4(1.0, 0.0, 0.0, 1.0))
colors.push_back(osg.Vec4(1.0, 0.0, 0.0, 1.0))
colors.push_back(osg.Vec4(0.0, 1.0, 0.0, 1.0))
colors.push_back(osg.Vec4(0.0, 1.0, 0.0, 1.0))
colors.push_back(osg.Vec4(0.0, 0.0, 1.0, 1.0))
colors.push_back(osg.Vec4(0.0, 0.0, 1.0, 1.0))
geometry.setColorArray(colors)
geometry.setColorBinding(osg.Geometry.BIND_PER_VERTEX)
geometry.addPrimitiveSet(osg.DrawArrays(osg.PrimitiveSet.LINES, 0, 6))
geode.addDrawable(geometry)
return geode
def createWireBox(node):
"""create a bounding box for the node """
#create an empty bounding box
bb = osg.BoundingBox()
#if we have a geode, expand by the drawables bounding box, else use the bounding sphere
geode = osg.NodeToGeode(node)
if geode:
print "geode found"
for i in range(geode.getNumDrawables()):
dwb = geode.getDrawable(0)
bb.expandBy(dwb.getBound())
else:
bb.expandBy(node.getBound())
center = node.getBound().center()
#create a geode for the wirebox
wbgeode = osg.Geode()
wbgeode.setName("ExtentsGeode")
#create a stateset for the wirebox
stateset = osg.StateSet()
wbgeode.setStateSet(stateset)
# stateset.thisown = 0
#create a polygonmode state attribute
polyModeObj = osg.PolygonMode()
polyModeObj.setMode(osg.PolygonMode.FRONT_AND_BACK, osg.PolygonMode.LINE)
stateset.setAttribute(polyModeObj)
#create a linewidth state attribute
lw = osg.LineWidth()
lw.setWidth(2.0)
stateset.setAttribute(lw)
stateset.setAttribute(selmat)
#create a drawablw box with the right position and size
lx = bb._max.x() - bb._min.x()
ly = bb._max.y() - bb._min.y()
lz = bb._max.z() - bb._min.z()
box = osg.Box(center, lx, ly, lz)
shape = osg.ShapeDrawable(box)
#shape.setColor(osg.Vec4(1.0, 0.0, 0.0, 1.0))
#add the drawable to the wirebox geode
wbgeode.addDrawable(shape)
for pointer in [stateset, box, polyModeObj, lw, shape]:
pointer.thisown = False
#return the wirebox geode
print "returning geode"
return wbgeode
def quad_create(self, texture):
stateset = osg.StateSet()
stateset.setTextureAttributeAndModes(0, texture)
corner = osg.Vec3(-self.aspect_ratio, -1.0, 0)
width = osg.Vec3(2 * self.aspect_ratio, 0, 0)
height = osg.Vec3(0, 2 * 1.0, 0)
geom = osg.createTexturedQuadGeometry(corner, width, height, 0.0, 0.0,
1.0, 1.0)
geom.setStateSet(stateset)
geode = osg.Geode()
geode.addDrawable(geom)
return geode
def createObjectFromImageList(pNode, pImageList, pBaseUrl=""):
"""
uses the image list to create a set of objects
"""
box_geometry = osg.Box()
box_drawable = osg.ShapeDrawable(box_geometry)
# create a random function for floating points
frandom = lambda x, y: random.randrange(x, y, int=types.FloatType)
print "Opening %d from %d images" % (maxphotos, len(pImageList))
for imgfile in [str(x) for x in pImageList[:maxphotos]]:
print "osgDB.readImageFile loading:", imgfile
image = osgDB.readImageFile(imgfile)
#check if the image is there, and not too small (icons and such)
if image and image.s() > 100.0 and image.t() > 100.0:
t = osg.TextureRectangle()
texmat = osg.TexMat()
texmat.setScaleByTextureRectangleSize(True)
t.setImage(image)
stateset = osg.StateSet()
stateset.setTextureAttributeAndModes(0, t, 1)
stateset.setTextureAttributeAndModes(0, texmat,
osg.StateAttribute.ON)
lbox_geode = osg.Geode()
lbox_geode.addDrawable(box_drawable)
lbox_geode.setStateSet(stateset)
lbox_node = osg.PositionAttitudeTransform()
lbox_node.addChild(lbox_geode)
rscale = frandom(5, 10)
lbox_node.setScale(
osg.Vec3d(image.s() / rscale, 0.5,
image.t() / rscale))
#set vertical
q = osg.Quat()
q.makeRotate(frandom(0, 2 * math.pi), 0, 0, 1)
lbox_node.setAttitude(q)
lbox_node.setPosition(
osg.Vec3d(frandom(-50, 50), frandom(-50, 50),
frandom(-100, 100)))
pNode.addChild(lbox_node)
def createPreRenderSubGraph(subgraph, tex_width, tex_height, renderImplementation, \
useImage, useTextureRectangle, useHDR, \
samples, colorSamples):
if not subgraph:
return 0
# create a group to contain the flag and the pre rendering camera.
parent = osg.Group()
#texture to render to and to use for rendering of flag.
texture = 0
if (useTextureRectangle):
textureRect = osg.TextureRectangle()
textureRect.setTextureSize(tex_width, tex_height)
textureRect.setInternalFormat(osg.GL_RGBA)
textureRect.setFilter(osg.Texture2D.MIN_FILTER, osg.Texture2D.LINEAR)
textureRect.setFilter(osg.Texture2D.MAG_FILTER, osg.Texture2D.LINEAR)
texture = textureRect
else:
texture2D = osg.Texture2D()
texture2D.setTextureSize(tex_width, tex_height)
texture2D.setInternalFormat(osg.GL_RGBA)
texture2D.setFilter(osg.Texture2D.MIN_FILTER, osg.Texture2D.LINEAR)
texture2D.setFilter(osg.Texture2D.MAG_FILTER, osg.Texture2D.LINEAR)
texture = texture2D
if useHDR:
texture.setInternalFormat(osg.GL_RGBA16F_ARB)
texture.setSourceFormat(osg.GL_RGBA)
texture.setSourceType(osg.GL_FLOAT)
#first create the geometry of the flag of which to view.
if True:
# create the to visualize.
polyGeom = osg.Geometry()
polyGeom.setName("PolyGeom")
polyGeom.setDataVariance(osg.Object.DYNAMIC)
polyGeom.setSupportsDisplayList(False)
origin = osg.Vec3(0.0, 0.0, 0.0)
xAxis = osg.Vec3(1.0, 0.0, 0.0)
yAxis = osg.Vec3(0.0, 0.0, 1.0)
zAxis = osg.Vec3(0.0, -1.0, 0.0)
height = 100.0
width = 200.0
noSteps = 20
vertices = osg.Vec3Array()
bottom = osg.Vec3(origin[0], origin[1], origin[2])
top = osg.Vec3(origin[0], origin[1], origin[2])
top[2] += height
dv = xAxis * (width / (float(noSteps - 1)))
texcoords = osg.Vec2Array()
# note, when we use TextureRectangle we have to scale the tex coords up to compensate.
bottom_texcoord = osg.Vec2(0.0, 0.0)
if useTextureRectangle:
ltex_height = tex_height
ltex_width = tex_width
else:
ltex_height = 1.0
ltex_width = 1.0
top_texcoord = osg.Vec2(0.0, ltex_height)
dv_texcoord = osg.Vec2(ltex_width / float((noSteps - 1)), 0.0)
for i in range(noSteps):
vertices.push_back(top)
vertices.push_back(bottom)
top += dv
bottom += dv
texcoords.push_back(top_texcoord)
texcoords.push_back(bottom_texcoord)
top_texcoord += dv_texcoord
bottom_texcoord += dv_texcoord
# pass the created vertex array to the points geometry object.
polyGeom.setVertexArray(vertices)
polyGeom.setTexCoordArray(0, texcoords)
colors = osg.Vec4Array()
colors.push_back(osg.Vec4(1.0, 1.0, 1.0, 1.0))
polyGeom.setColorArray(colors)
polyGeom.setColorBinding(osg.Geometry.BIND_OVERALL)
polyGeom.addPrimitiveSet(
osg.DrawArrays(osg.PrimitiveSet.QUAD_STRIP, 0, vertices.size()))
# new we need to add the texture to the Drawable, we do so by creating a
# StateSet to contain the Texture StateAttribute.
stateset = osg.StateSet()
stateset.setTextureAttributeAndModes(0, texture, osg.StateAttribute.ON)
polyGeom.setStateSet(stateset)
#changes to the geometry
polyGeom.setUpdateCallback(
MyGeometryCallback(origin, xAxis, yAxis, zAxis, 1.0, 1.0 / width,
0.2).__disown__())
geode = osg.Geode()
geode.addDrawable(polyGeom)
parent.addChild(geode)
# then create the camera node to do the render to texture
if True:
camera = osg.Camera()
# set up the background color and clear mask.
camera.setClearColor(osg.Vec4(0.1, 0.1, 0.3, 1.0))
camera.setClearMask(osg.GL_COLOR_BUFFER_BIT | osg.GL_DEPTH_BUFFER_BIT)
bs = subgraph.getBound()
if not bs.valid():
return subgraph
znear = 1.0 * bs.radius()
zfar = 3.0 * bs.radius()
# 2:1 aspect ratio as per flag geometry below.
proj_top = 0.25 * znear
proj_right = 0.5 * znear
znear *= 0.9
zfar *= 1.1
# set up projection.
camera.setProjectionMatrixAsFrustum(-proj_right, proj_right, -proj_top,
proj_top, znear, zfar)
# set view
camera.setReferenceFrame(osg.Transform.ABSOLUTE_RF)
camera.setViewMatrixAsLookAt(osg.Vec3d(bs.center())-osg.Vec3d(0.0,2.0,0.0)*bs.radius(), \
osg.Vec3d(bs.center()),\
osg.Vec3d(0.0,0.0,1.0))
# set viewport
camera.setViewport(0, 0, tex_width, tex_height)
# set the camera to render before the main camera.
camera.setRenderOrder(osg.Camera.PRE_RENDER)
# tell the camera to use OpenGL frame buffer object where supported.
camera.setRenderTargetImplementation(renderImplementation)
if useImage:
image = osg.Image()
#image.allocateImage(tex_width, tex_height, 1, GL_RGBA, GL_UNSIGNED_BYTE);
image.allocateImage(tex_width, tex_height, 1, osg.GL_RGBA,
osg.GL_FLOAT)
# attach the image so its copied on each frame.
camera.attach(osg.Camera.COLOR_BUFFER, image, samples,
colorSamples)
camera.setPostDrawCallback(MyCameraPostDrawCallback(image))
# Rather than attach the texture directly to illustrate the texture's ability to
# detect an image update and to subload the image onto the texture. You needn't
# do this when using an Image for copying to, as a separate camera.attach(..)
# would suffice as well, but we'll do it the long way round here just for demonstration
# purposes (long way round meaning we'll need to copy image to main memory, then
# copy it back to the graphics card to the texture in one frame).
# The long way round allows us to manually modify the copied image via the callback
# and then let this modified image by reloaded back.
texture.setImage(0, image)
else:
# attach the texture and use it as the color buffer.
camera.attach(osg.Camera.COLOR_BUFFER, texture, 0, 0, False,
samples, colorSamples)
# add subgraph to render
camera.addChild(subgraph)
parent.addChild(camera)
return parent
vert = osg.Shader(osg.Shader.VERTEX, vertProg) frag = osg.Shader(osg.Shader.FRAGMENT, fragProg) prog = osg.Program() prog.addShader(vert) prog.addShader(frag) geom.getOrCreateStateSet().setTextureAttributeAndModes(0, osg.Texture2D(image)) geom.getOrCreateStateSet().setAttributeAndModes(prog) ind = osg.Uniform(osg.Uniform.INT, "index") ind.set(ctypes.c_int(1)) uniform = osg.Uniform(osg.Uniform.FLOAT, "test") uniform.set(4.0) tex = osg.Uniform(osg.Uniform.SAMPLER_2D, "tex0") tex.set(ctypes.c_uint(0)) tex2 = osg.Uniform(osg.Uniform.SAMPLER_2D, "tex1") tex2.set(ctypes.c_uint(1)) geom.getOrCreateStateSet().addUniform(uniform) geom.getOrCreateStateSet().addUniform(ind) geom.getOrCreateStateSet().addUniform(tex) geom.getOrCreateStateSet().addUniform(tex2) geode = osg.Geode() geode.addDrawable(geom) group.addChild(geode) group.addChild(loadedModel) viewer = osgViewer.Viewer() viewer.setSceneData(group) viewer.addEventHandler(osgViewer.StatsHandler()) viewer.run()
def main():
#arguments = osg.ArgumentParses(len(sys.argv),sys.argv)
viewer = osgViewer.Viewer() #arguments)
viewer.setCameraManipulator(osgGA.TrackballManipulator())
skelroot = osgAnimation.Skeleton()
skelroot.setDefaultUpdateCallback()
root = osgAnimation.Bone()
root.setBindMatrixInBoneSpace(osg.Matrixd_identity())
root.setBindMatrixInBoneSpace(osg.Matrixd_translate(-1, 0, 0))
root.setName("root")
root.setDefaultUpdateCallback()
right0 = osgAnimation.Bone()
right0.setBindMatrixInBoneSpace(osg.Matrixd_translate(1, 0, 0))
right0.setName("right0")
right0.setDefaultUpdateCallback("right0")
right1 = osgAnimation.Bone()
right1.setBindMatrixInBoneSpace(osg.Matrixd_translate(1, 0, 0))
right1.setName("right1")
right1.setDefaultUpdateCallback("right1")
root.addChild(right0)
right0.addChild(right1)
skelroot.addChild(root)
scene = osg.Group()
manager = osgAnimation.BasicAnimationManager()
scene.setUpdateCallback(manager)
anim = osgAnimation.Animation()
keys0 = osgAnimation.QuatKeyframeContainer()
keys1 = osgAnimation.QuatKeyframeContainer()
pKeys0 = int(keys0.this)
pKeys1 = int(keys1.this)
rotate = osg.Quat()
rotate.makeRotate(PI_2, osg.Vec3(0, 0, 1))
keys0.push_back(osgAnimation.QuatKeyframe(0, osg.Quat(0, 0, 0, 1)))
keys0.push_back(osgAnimation.QuatKeyframe(3, rotate))
keys0.push_back(osgAnimation.QuatKeyframe(6, rotate))
keys1.push_back(osgAnimation.QuatKeyframe(0, osg.Quat(0, 0, 0, 1)))
keys1.push_back(osgAnimation.QuatKeyframe(3, osg.Quat(0, 0, 0, 1)))
keys1.push_back(osgAnimation.QuatKeyframe(6, rotate))
sampler0 = osgAnimation.QuatSphericalLinearSampler()
sampler0.setKeyframeContainer(keys0)
channel0 = osgAnimation.QuatSphericalLinearChannel(sampler0)
channel0.setName("quaternion")
channel0.setTargetName("right0")
anim.addChannel(channel0)
sampler1 = osgAnimation.QuatSphericalLinearSampler()
sampler1.setKeyframeContainer(keys1)
channel1 = osgAnimation.QuatSphericalLinearChannel(sampler1)
channel1.setName("quaternion")
channel1.setTargetName("right1")
anim.addChannel(channel1)
manager.registerAnimation(anim)
manager.buildTargetReference()
# let's start !
manager.playAnimation(anim)
# we will use local data from the skeleton
rootTransform = osg.MatrixTransform()
rootTransform.setMatrix(osg.Matrixd_rotate(PI_2, osg.Vec3(1, 0, 0)))
right0.addChild(createAxis())
right0.setDataVariance(osg.Object.DYNAMIC)
right1.addChild(createAxis())
right1.setDataVariance(osg.Object.DYNAMIC)
trueroot = osg.MatrixTransform()
trueroot.setMatrix(osg.Matrixd(root.getMatrixInBoneSpace()))
trueroot.addChild(createAxis())
trueroot.addChild(skelroot)
trueroot.setDataVariance(osg.Object.DYNAMIC)
rootTransform.addChild(trueroot)
scene.addChild(rootTransform)
geom = createTesselatedBox(4, 4.0)
geode = osg.Geode()
geode.addDrawable(geom)
skelroot.addChild(geode)
src = geom.getVertexArray().asVec3Array()
geom.getOrCreateStateSet().setMode(osg.GL_LIGHTING, False)
geom.setDataVariance(osg.Object.DYNAMIC)
initVertexMap(root, right0, right1, geom, src)
#let's run !
viewer.setSceneData(scene)
viewer.realize()
while not viewer.done():
viewer.frame()
# A simple example of how to create geometry # Pyramid geometry creation follows this tutorial: # http://www.openscenegraph.org/documentation/NPSTutorials/osgGeometry.html # import all necessary stuff import sys import osg import osgDB import osgGA import osgViewer # create a root node node = osg.Group() # Line Geometry lineGeode = osg.Geode() lineGeometry = osg.Geometry() lineGeode.addDrawable(lineGeometry) lineStateSet = lineGeode.getOrCreateStateSet() lineVertices = osg.Vec3Array() lineVertices.push_back(osg.Vec3(-10, 10, 0)) lineVertices.push_back(osg.Vec3(-10, -10, 0)) lineGeometry.setVertexArray(lineVertices) lineBase = osg.DrawElementsUInt(osg.PrimitiveSet.LINES, 0) lineBase.push_back(0) lineBase.push_back(1) lineGeometry.addPrimitiveSet(lineBase) node.addChild(lineGeode)
