virtual void operator () (osg.RenderInfo renderInfo)
if not _snapImage : return
osg.notify(osg.NOTICE), "Camera callback"
camera = renderInfo.getCurrentCamera()
viewport = camera.getViewport() if (camera) else 0
osg.notify(osg.NOTICE), "Camera callback ", camera, " ", viewport
if viewport and _image.valid() :
_image.readPixels(int(viewport.x()),int(viewport.y()),int(viewport.width()),int(viewport.height()),
GL_RGBA,
GL_UNSIGNED_BYTE)
osgDB.writeImageFile(*_image, _filename)
osg.notify(osg.NOTICE), "Taken screenshot, and written to '", _filename, "'"
_snapImage = False
_filename = str()
mutable bool _snapImage
mutable osg.Image _image
class SnapeImageHandler (osgGA.GUIEventHandler) :
SnapeImageHandler(int key,SnapImage* si):
_key(key),
_snapImage(si)
OSG_NOTICE, "outputFileName=", outputFileName
camera = renderInfo.getCurrentCamera()
viewport = camera.getViewport() if (camera) else 0
if viewport :
OSG_NOTICE, "Doing read of =", viewport.x(), ", ", viewport.y(), ", ", viewport.width(), ", ", viewport.height(), " with pixelFormat=0x", std.hex, _pixelFormat, std.dec
glReadBuffer(camera.getDrawBuffer())
image = osg.Image()
image.readPixels(viewport.x(),viewport.y(),viewport.width(),viewport.height(),
_pixelFormat, GL_UNSIGNED_BYTE, 1)
if _flip : image.flipVertical()
osgDB.writeImageFile(*image, outputFileName)
typedef std.map< osg.Camera*, unsigned int> CameraNumMap
_pixelFormat = GLenum()
_flip = bool()
_frameCapture = gsc.CaptureSettings()
_cameraNumMap = CameraNumMap()
def main(argv):
# use an ArgumentParser object to manage the program arguments.
arguments = osg.ArgumentParser(argv)
_isRunning = True
_isFinishing = False
void PosterPrinter.frame( osg.FrameStamp* fs, osg.Node* node )
# Add cull callbacks to all existing paged nodes,
# and advance frame when all callbacks are dispatched.
if addCullCallbacks(fs, node) :
return
if _isFinishing :
if fs.getFrameNumber()-_lastBindingFrame :>2 :
# Record images and the final poster
recordImages()
if _finalPoster.valid() :
print "Writing final result to file..."
osgDB.writeImageFile( *_finalPoster, _outputPosterName )
# Release all cull callbacks to free unused paged nodes
removeCullCallbacks( node )
_visitor.clearNames()
_isFinishing = False
print "Recording images finished."
if _isRunning :
# Every "copy-to-image" process seems to be finished in 2 frames.
# So record them and dispatch camera to next tiles.
if fs.getFrameNumber()-_lastBindingFrame :>2 :
# Record images and unref them to free memory
recordImages()
def main(argv):
# use an ArgumentParser object to manage the program arguments.
arguments = osg.ArgumentParser(argv)
# set up the usage document, in case we need to print out how to use this program.
arguments.getApplicationUsage().setDescription(arguments.getApplicationName()+" is the example which demonstrates use of osg.AnimationPath and UpdateCallbacks for adding animation to your scenes.")
arguments.getApplicationUsage().setCommandLineUsage(arguments.getApplicationName()+" [options] filename ...")
arguments.getApplicationUsage().addCommandLineOption("-h or --help","Display this information")
arguments.getApplicationUsage().addCommandLineOption("-o <filename>","Object to be loaded")
if arguments.read( "-h" ) or arguments.read( "--help" ) :
print "Argumentlist:"
print "\t-o <filename> sets object to be loaded and sliced"
print "\t--slices <unsigned int> sets number of slices through the object"
print "\t--near <double> sets start for near clipping plane"
print "\t--far <double> sets start for far clipping plane"
return 1
outputName = str("volume_tex.dds")
while arguments.read( "-o", outputName ) :
numberSlices = 128
while arguments.read( "--slices", numberSlices) :
nearClip = 0.0
farClip = 0.0
while arguments.read( "--near",nearClip ) :
while arguments.read( "--far", farClip) :
# load the scene.
loadedModel = osgDB.readNodeFiles( arguments )
if not loadedModel :
print "No data loaded."
return 1
bs = loadedModel.getBound()
sp = SliceProcessor( (double)bs.radius(), numberSlices )
if nearClip not =0.0 : sp._nearPlane = nearClip
if farClip not =0.0 : sp._farPlane = farClip
# any option left unread are converted into errors to write out later.
arguments.reportRemainingOptionsAsUnrecognized()
# report any errors if they have occurred when parsing the program arguments.
if arguments.errors() :
arguments.writeErrorMessages(std.cout)
return 1
traits = osg.GraphicsContext.Traits()
traits.x = 100
traits.y = 100
traits.width = numberSlices
traits.height = numberSlices
traits.alpha = 8
traits.windowDecoration = True
traits.doubleBuffer = True
traits.sharedContext = 0
traits.pbuffer = False
gc = osg.GraphicsContext.createGraphicsContext(traits)
if not gc or not gc.valid() :
osg.notify(osg.NOTICE), "Error: unable to create graphics window"
return 1
gc.realize()
gc.makeCurrent()
# create the view of the scene.
sceneView = osgUtil.SceneView()
sceneView.setDefaults()
sceneView.setSceneData(loadedModel)
# initialize the view to look at the center of the scene graph
viewMatrix = osg.Matrix()
# distance from viewport to object's center is set to be 2x bs.radius()
viewMatrix.makeLookAt(bs.center()-osg.Vec3(0.0,2.0*bs.radius(),0.0),bs.center(),osg.Vec3(0.0,0.0,1.0))
# turn off autocompution of near and far clipping planes
sceneView.setComputeNearFarMode(osgUtil.CullVisitor.DO_NOT_COMPUTE_NEAR_FAR)
# set the clear color of the background to make sure that the alpha is 0.0.
sceneView.setClearColor(osg.Vec4(0.0,0.0,0.0,0.0))
# record the timer tick at the start of rendering.
start_tick = osg.Timer.instance().tick()
print "radius: ", bs.radius()
frameNum = 0
double tmpNear, tmpFar
baseImageName = str("shot_")
tmpImageName = str()
tmpImage = osg.Image()
# main loop (note, window toolkits which take control over the main loop will require a window redraw callback containing the code below.)
for( unsigned int i = 0 i < sp._sliceNumber and gc.isRealized() ++i )
# set up the frame stamp for current frame to record the current time and frame number so that animtion code can advance correctly
frameStamp = osg.FrameStamp()
frameStamp.setReferenceTime(osg.Timer.instance().delta_s(start_tick,osg.Timer.instance().tick()))
frameStamp.setFrameNumber(frameNum++)
# pass frame stamp to the SceneView so that the update, cull and draw traversals all use the same FrameStamp
sceneView.setFrameStamp(frameStamp)
# update the viewport dimensions, incase the window has been resized.
sceneView.setViewport(0,0,traits.width,traits.height)
# set the view
sceneView.setViewMatrix(viewMatrix)
# set Projection Matrix
tmpNear = sp._nearPlane+i*sp._sliceDelta
tmpFar = sp._farPlane+(i*sp._sliceDelta)+sp._nearFarOffset
sceneView.setProjectionMatrixAsOrtho(-(bs.radius()+bs.radius()/2), bs.radius()+bs.radius()/2,-bs.radius(), bs.radius(), tmpNear, tmpFar)
# do the update traversal the scene graph - such as updating animations
sceneView.update()
# do the cull traversal, collect all objects in the view frustum into a sorted set of rendering bins
sceneView.cull()
# draw the rendering bins.
sceneView.draw()
# Swap Buffers
gc.swapBuffers()
print "before readPixels: _r = ", sp._image.r()
tmpImage.readPixels(static_cast<int>(sceneView.getViewport().x()),
static_cast<int>(sceneView.getViewport().y()),
static_cast<int>(sceneView.getViewport().width()),
static_cast<int>(sceneView.getViewport().height()),
GL_RGBA,GL_UNSIGNED_BYTE)
# print "vor copySubImage: _r = ", sp._image.r()
sp._image.copySubImage( 0, 0, i, tmpImage )
#
# o = std.ostringstream()
# o, baseImageName, i, ".rgba"
# tmpImageName = o.str()
# osgDB.writeImageFile( *(sp._image), tmpImageName )
# print "Wrote image to file: ", tmpImageName
#
osgDB.writeImageFile( *(sp._image), outputName)
return 0
if __name__ == "__main__":
main(sys.argv)
