osg着色器uniform回調實現高亮部分閃爍效果的實例分析

效果：牛身上圈圈里的顏色在閃爍

代碼：

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

static const char* vertSource = {
    "varying vec3 normal;\n"//易變量 用於着色器之間的傳值
    "void main()\n"
    "{\n"
    "    normal = normalize(gl_NormalMatrix * gl_Normal);\n" //把法線向量從物體空間轉化到視覺空間。
    "    gl_Position = ftransform();\n" //相當於 gl_Position = ModelViewProjectionMatrix * gl_Vertex,
                                                //模型視圖投影矩陣與空間頂點位置相乘，得到裁剪空間頂點位置。
    "}\n"
};

static const char* fragSource = {
    "uniform vec4 mainColor;\n"  //外面程序輸入的顏色
    "varying vec3 normal;\n"    //上面的頂點着色器傳的法線
    "void main()\n"
    "{\n"
    "    float intensity = dot(vec3(gl_LightSource[0].position), normal);\n" //計算光照位置與法線的點積，為什么這么做，知道的朋友不吝賜教下。
                                                                            //按照我的理解就是計算了一下亮度值。
    //給輸出的顏色賦值
    "    if (intensity > 0.95) gl_FragColor = mainColor;\n" //如果這個值大於0.95，則輸出顏色為mainColor，而uniform回調中每幀在來回改這個值，所
                                                            //以這一部分會出現閃爍的效果。
    "    else if (intensity > 0.5) gl_FragColor = vec4(0.6,0.3,0.3,1.0);\n"
    "    else if (intensity > 0.25) gl_FragColor = vec4(0.4,0.2,0.2,1.0);\n"
    "    else gl_FragColor = vec4(0.2,0.1,0.1,1.0);\n"
    "}\n"
};


/* 一直變量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();
}