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)