<reponame>JaneliaSciComp/osgpyplusplus
#!/bin/env python

# Automatically translated python version of 
# OpenSceneGraph example program "osganimationnode"
# !!! This program will need manual tuning before it will work. !!!

import sys

from osgpypp import osg
from osgpypp import osgAnimation
from osgpypp import osgGA
from osgpypp import osgViewer


# Translated from file 'osganimationnode.cpp'

#  -*-c++-*-
# *  Copyright (C) 2008 <NAME> <<EMAIL>>
# *
# * This library is open source and may be redistributed and/or modified under
# * the terms of the OpenSceneGraph Public License (OSGPL) version 0.0 or
# * (at your option) any later version.  The full license is in LICENSE file
# * included with this distribution, and on the openscenegraph.org website.
# *
# * This library 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
# * OpenSceneGraph Public License for more details.
#

#include <iostream>
#include <osg/Geometry>
#include <osg/Shape>
#include <osg/ShapeDrawable>
#include <osgViewer/Viewer>
#include <osgGA/TrackballManipulator>
#include <osg/MatrixTransform>
#include <osg/Material>
#include <osgAnimation/Sampler>

class AnimtkUpdateCallback (osg.NodeCallback) :
    META_Object(osgAnimation, AnimtkUpdateCallback)

    AnimtkUpdateCallback()
        _sampler = osgAnimation.Vec3CubicBezierSampler()
        _playing = False
        _lastUpdate = 0
    AnimtkUpdateCallback( AnimtkUpdateCallback val,  osg.CopyOp copyop = osg.CopyOp.SHALLOW_COPY):
        osg.Object(val, copyop),
        osg.NodeCallback(val, copyop),
        _sampler(val._sampler),
        _startTime(val._startTime),
        _currentTime(val._currentTime),
        _playing(val._playing),
        _lastUpdate(val._lastUpdate)

    #* Callback method called by the NodeVisitor when visiting a node.
    virtual void operator()(osg.Node* node, osg.NodeVisitor* nv)
        if nv.getVisitorType() == osg.NodeVisitor.UPDATE_VISITOR  and 
            nv.getFrameStamp()  and 
            nv.getFrameStamp().getFrameNumber()  not = _lastUpdate : 

            _lastUpdate = nv.getFrameStamp().getFrameNumber()
            _currentTime = osg.Timer.instance().tick()

            if _playing  and  _sampler  and  _sampler.getKeyframeContainer() : 
                transform = dynamic_cast<osg.MatrixTransform*>(node)
                if transform : 
                    result = osg.Vec3()
                    t = osg.Timer.instance().delta_s(_startTime, _currentTime)
                    duration = _sampler.getEndTime() - _sampler.getStartTime()
                    t = fmod(t, duration)
                    t += _sampler.getStartTime()
                    _sampler.getValueAt(t, result)
                    transform.setMatrix(osg.Matrix.translate(result))
        # note, callback is responsible for scenegraph traversal so
        # they must call traverse(node,nv) to ensure that the
        # scene graph subtree (and associated callbacks) are traversed.
        traverse(node,nv)

    def start():

         _startTime = osg.Timer.instance().tick() _currentTime = _startTime _playing = True
    def stop():
         _currentTime = _startTime _playing = False

    _sampler = osgAnimation.Vec3CubicBezierSampler()
    _startTime = osg.Timer_t()
    _currentTime = osg.Timer_t()
    _playing = bool()
    _lastUpdate = unsigned int()



class AnimtkStateSetUpdateCallback (osg.StateSet.Callback) :
    META_Object(osgAnimation, AnimtkStateSetUpdateCallback)

    AnimtkStateSetUpdateCallback()
        _sampler = osgAnimation.Vec4LinearSampler()
        _playing = False
        _lastUpdate = 0

    AnimtkStateSetUpdateCallback( AnimtkStateSetUpdateCallback val,  osg.CopyOp copyop = osg.CopyOp.SHALLOW_COPY):
        osg.Object(val, copyop),
        osg.StateSet.Callback(val, copyop),
        _sampler(val._sampler),
        _startTime(val._startTime),
        _currentTime(val._currentTime),
        _playing(val._playing),
        _lastUpdate(val._lastUpdate)

    #* Callback method called by the NodeVisitor when visiting a node.
    virtual void operator()(osg.StateSet* state, osg.NodeVisitor* nv)
        if state  and 
            nv.getVisitorType() == osg.NodeVisitor.UPDATE_VISITOR  and 
            nv.getFrameStamp()  and 
            nv.getFrameStamp().getFrameNumber()  not = _lastUpdate :

            _lastUpdate = nv.getFrameStamp().getFrameNumber()
            _currentTime = osg.Timer.instance().tick()

            if _playing  and  _sampler  and  _sampler.getKeyframeContainer() :
                material = dynamic_cast<osg.Material*>(state.getAttribute(osg.StateAttribute.MATERIAL))
                if material :
                    result = osg.Vec4()
                    t = osg.Timer.instance().delta_s(_startTime, _currentTime)
                    duration = _sampler.getEndTime() - _sampler.getStartTime()
                    t = fmod(t, duration)
                    t += _sampler.getStartTime()
                    _sampler.getValueAt(t, result)
                    material.setDiffuse(osg.Material.FRONT_AND_BACK, result)

    def start():

         _startTime = osg.Timer.instance().tick() _currentTime = _startTime _playing = True
    def stop():
         _currentTime = _startTime _playing = False

    _sampler = osgAnimation.Vec4LinearSampler()
    _startTime = osg.Timer_t()
    _currentTime = osg.Timer_t()
    _playing = bool()
    _lastUpdate = unsigned int()



def createAxis():


    
    geode = osg.Geode()
    geometry = osg.Geometry(osg.Geometry())

    vertices = osg.Vec3Array(osg.Vec3Array())
    vertices.push_back (osg.Vec3 ( 0.0, 0.0, 0.0))
    vertices.push_back (osg.Vec3 ( 10.0, 0.0, 0.0))
    vertices.push_back (osg.Vec3 ( 0.0, 0.0, 0.0))
    vertices.push_back (osg.Vec3 ( 0.0, 10.0, 0.0))
    vertices.push_back (osg.Vec3 ( 0.0, 0.0, 0.0))
    vertices.push_back (osg.Vec3 ( 0.0, 0.0, 10.0))
    geometry.setVertexArray (vertices)

    colors = osg.Vec4Array(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, osg.Array.BIND_PER_VERTEX)
    geometry.addPrimitiveSet(osg.DrawArrays(osg.PrimitiveSet.LINES,0,6))

    geode.addDrawable( geometry )
    geode.getOrCreateStateSet().setMode(GL_LIGHTING, False)
    return geode

def setupStateSet():

    
    st = osg.StateSet()
    st.setAttributeAndModes(osg.Material, True)()
    st.setMode(GL_BLEND, True)
    callback = AnimtkStateSetUpdateCallback()
    keys = callback._sampler.getOrCreateKeyframeContainer()
    keys.push_back(osgAnimation.Vec4Keyframe(0, osg.Vec4(0,0,0,0)))
    keys.push_back(osgAnimation.Vec4Keyframe(2, osg.Vec4(0.5,0,0,0.5)))
    keys.push_back(osgAnimation.Vec4Keyframe(4, osg.Vec4(0,0.5,0,1)))
    keys.push_back(osgAnimation.Vec4Keyframe(6, osg.Vec4(0,0,0.5,1)))
    keys.push_back(osgAnimation.Vec4Keyframe(8, osg.Vec4(1,1,1,0.5)))
    keys.push_back(osgAnimation.Vec4Keyframe(10, osg.Vec4(0,0,0,0)))
    callback.start()
    st.setUpdateCallback(callback)
    return st

def setupCube():

    
    geode = osg.Geode()
    geode.addDrawable(osg.ShapeDrawable(osg.Box(osg.Vec3(0.0,0.0,0.0),2)))
    geode.setStateSet(setupStateSet())
    return geode

def setupAnimtkNode():

    
    osg.Vec3 v[5]
    v[0] = osg.Vec3(0,0,0)
    v[1] = osg.Vec3(10,-50,0)
    v[2] = osg.Vec3(30,-10,20)
    v[3] = osg.Vec3(-10,20,-20)
    v[4] = osg.Vec3(0,0,0)
    node = osg.MatrixTransform()
    callback = AnimtkUpdateCallback()
    keys = callback._sampler.getOrCreateKeyframeContainer()
    keys.push_back(osgAnimation.Vec3CubicBezierKeyframe(0, osgAnimation.Vec3CubicBezier(
                                                        v[0], # pos
                                                        v[0] + (v[0] - v[3]), # p1
                                                        v[1] - (v[1] - v[0]) # p2
                                                        )))
    keys.push_back(osgAnimation.Vec3CubicBezierKeyframe(2, osgAnimation.Vec3CubicBezier(
                                                        v[1], # pos
                                                        v[1] + (v[1] - v[0]),
                                                        v[2] - (v[2] - v[1])
                                                        )))
    keys.push_back(osgAnimation.Vec3CubicBezierKeyframe(4, osgAnimation.Vec3CubicBezier(
                                                        v[2], # pos
                                                        v[2] + (v[2] - v[1]),
                                                        v[3] - (v[3] - v[2])
                                                        )))
    keys.push_back(osgAnimation.Vec3CubicBezierKeyframe(6, osgAnimation.Vec3CubicBezier(
                                                        v[3], # pos
                                                        v[3] + (v[3] - v[2]),
                                                        v[4] - (v[4] - v[3])
                                                        )))
    keys.push_back(osgAnimation.Vec3CubicBezierKeyframe(8, osgAnimation.Vec3CubicBezier(
                                                        v[4], # pos
                                                        v[4] + (v[4] - v[3]),
                                                        v[0] - (v[0] - v[4])
                                                        )))

    callback.start()
    node.setUpdateCallback(callback)
    node.addChild(setupCube())
    return node

int main (int argc, char* argv[])
    arguments = osg.ArgumentParser(argv)
    viewer = osgViewer.Viewer(arguments)

    manipulator = osgGA.TrackballManipulator()
    viewer.setCameraManipulator(manipulator)

    root = osg.Group()
    root.setInitialBound(osg.BoundingSphere(osg.Vec3(10,0,10), 30))
    root.addChild(createAxis())

    node = setupAnimtkNode()
    node.addChild(createAxis())
    root.addChild(node)

    viewer.setSceneData( root )
    viewer.realize()

    while  not viewer.done() :
        viewer.frame()



if __name__ == "__main__":
    main(sys.argv)
