osg着色器uniform回调实现高亮部分闪烁效果的实例分析

效果：牛身上圈圈里的颜色在闪烁

代码：

#include <osg/Program>
#include <osgDB/ReadFile>
#include <osgViewer/Viewer>

static const char* vertSource = {
    "varying vec3 normal;
"//易变量 用于着色器之间的传值
    "void main()
"
    "{
"
    "    normal = normalize(gl_NormalMatrix * gl_Normal);
" //把法线向量从物体空间转化到视觉空间。
    "    gl_Position = ftransform();
" //相当于 gl_Position = ModelViewProjectionMatrix * gl_Vertex,
                                                //模型视图投影矩阵与空间顶点位置相乘，得到裁剪空间顶点位置。
    "}
"
};

static const char* fragSource = {
    "uniform vec4 mainColor;
"  //外面程序输入的颜色
    "varying vec3 normal;
"    //上面的顶点着色器传的法线
    "void main()
"
    "{
"
    "    float intensity = dot(vec3(gl_LightSource[0].position), normal);
" //计算光照位置与法线的点积，为什么这么做，知道的朋友不吝赐教下。
                                                                            //按照我的理解就是计算了一下亮度值。
    //给输出的颜色赋值
    "    if (intensity > 0.95) gl_FragColor = mainColor;
" //如果这个值大于0.95，则输出颜色为mainColor，而uniform回调中每帧在来回改这个值，所
                                                            //以这一部分会出现闪烁的效果。
    "    else if (intensity > 0.5) gl_FragColor = vec4(0.6,0.3,0.3,1.0);
"
    "    else if (intensity > 0.25) gl_FragColor = vec4(0.4,0.2,0.2,1.0);
"
    "    else gl_FragColor = vec4(0.2,0.1,0.1,1.0);
"
    "}
"
};


/* 一直变量uniform回调类，主要用于每帧更新着色器中的用户数据，从而改变渲染的行为和输出结果。     */


class ColorCallback : public osg::Uniform::Callback
{
public:
    ColorCallback() : _incRed(false) {}
    
    virtual void operator()( osg::Uniform* uniform, osg::NodeVisitor* nv )
    {
        if ( !uniform ) return;
        
        osg::Vec4 color;
        uniform->get( color );//获取通过uniform 设置进去的值mainColor
        
        if ( _incRed==true )//3、如果r减到0呢，就给他一直加。
        {
            if ( color.x()<1.0 ) color.x() += 0.1;
            else _incRed = false;//4、如果加到1就给他再减。
        }
        else
        {
            if ( color.x()>0.0 ) color.x() -= 0.1;//1、如果RGB中的r分量大于0，则一直减小。
            else _incRed = true;// 2、r减到0了。
        }
        uniform->set( color );
    }

protected:
    bool _incRed;
};

void createShaders( osg::StateSet& ss )
{
    osg::ref_ptr<osg::Shader> vertShader = new osg::Shader( osg::Shader::VERTEX, vertSource );
    osg::ref_ptr<osg::Shader> fragShader = new osg::Shader( osg::Shader::FRAGMENT, fragSource );
    
    osg::ref_ptr<osg::Program> program = new osg::Program;
    program->addShader( vertShader.get() );
    program->addShader( fragShader.get() );
    
    osg::ref_ptr<osg::Uniform> mainColor = new osg::Uniform( "mainColor", osg::Vec4(0.5,0.5,0.5,1.0) );//这个部分的颜色是闪烁部分的颜色值。
    mainColor->setUpdateCallback( new ColorCallback );
    
    ss.addUniform( mainColor.get() );
    ss.setAttributeAndModes( program.get() );
}

int main( int argc, char** argv )
{
    osg::ArgumentParser arguments( &argc, argv );
    osg::Node* model = osgDB::readNodeFiles( arguments );
    if ( !model ) model = osgDB::readNodeFile( "cow.osg" );
    
    createShaders( *(model->getOrCreateStateSet()) );
    
    osgViewer::Viewer viewer;
    viewer.setSceneData( model );
    viewer.setUpViewInWindow(20,20,600,600);
    return viewer.run();
}