OSGEarth绘制动态水效果

在OSGEarth上绘制水效果的思路为：

1. 使用OSG::Geometry的方式绘制多边形；

2. 在绘制的多边形上贴一个水的纹理；

3. 对纹理使用shader效果，使之动态展示。

此种方式适用于小范围的不严格的水效果，若需要大范围或者有水底效果的请使用OSGOcean。

代码如下：water.h

class Water : public HandleAdapter
{
public:
Water(GraphicsView* view);
~Water();

protected:
virtual void slotPickedXYZ(osg::Vec3d pos);
virtual void slotMoveingXYZ(osg::Vec3d pos);
virtual void slotRightHandle();

private:
void drawWater(osg::Vec3d pos = osg::Vec3d());
osg::Texture2D* creatText2D(const QString& strFile);
void initShader();

private:
osg::ref_ptr<osg::Vec3dArray> m_vecPoints;
osg::Geode* m_pWater;
char* m_waterFrag;
char* m_waterVert;
};

实现代码如下：water.cpp

Water::Water(GraphicsView* view)
: HandleAdapter(view)
{
m_pWater = NULL;
m_vecPoints = new osg::Vec3dArray;
m_vecPoints->clear();

initShader();
}

Water::~Water()
{

}

void Water::slotPickedXYZ(osg::Vec3d pos)
{
m_vecPoints->push_back(pos);
if (m_vecPoints->size() <= 2)
{
return;
}

drawWater();
}

void Water::slotMoveingXYZ(osg::Vec3d pos)
{
if (m_vecPoints->size() < 2)
{
return;
}

drawWater(pos);
}

void Water::slotRightHandle()
{
endHandle();
m_pWater = nullptr;
}

void Water::drawWater(osg::Vec3d pos /*= osg::Vec3d()*/)
{
osg::ref_ptr<osg::Vec3dArray> vecPoints = new osg::Vec3dArray;
vecPoints = m_vecPoints;
if (pos == osg::Vec3d() && m_vecPoints->size() <= 2)
{
return;
}
if (pos != osg::Vec3d())
{
vecPoints->push_back(pos);
}
osg::ref_ptr<osg::Geometry> pWater = new osg::Geometry;
pWater->setVertexArray(vecPoints);
pWater->addPrimitiveSet(
new osg::DrawArrays(osg::PrimitiveSet::TRIANGLE_FAN, 0, vecPoints->size()));

//纹理
osg::ref_ptr<osg::Texture2D> pText2D = new osg::Texture2D;
pText2D->setTextureSize(1024, 1024);
pText2D->setInternalFormat(GL_RGBA);

pWater->getOrCreateStateSet()->setTextureAttributeAndModes(0, pText2D);
pWater->getOrCreateStateSet()->setTextureAttributeAndModes(1,
creatText2D("shui_5.jpg"));
pWater->getOrCreateStateSet()->setTextureAttributeAndModes(2,
creatText2D("water_DUDV.jpg"));
pWater->getOrCreateStateSet()->setTextureAttributeAndModes(3,
creatText2D("water_NM.jpg"));

// shader
osg::ref_ptr<osg::Shader> pVeter = new osg::Shader(osg::Shader::VERTEX, m_waterVert);
osg::ref_ptr<osg::Shader> pFrag = new osg::Shader(osg::Shader::FRAGMENT, m_waterFrag);
osg::ref_ptr<osg::Program> pProgram = new osg::Program;
pProgram->addShader(pVeter);
pProgram->addShader(pFrag);
pWater->getOrCreateStateSet()->addUniform(new osg::Uniform("reflection", 0));
pWater->getOrCreateStateSet()->addUniform(new osg::Uniform("defaultTex", 1));
pWater->getOrCreateStateSet()->addUniform(new osg::Uniform("refraction", 2));
pWater->getOrCreateStateSet()->addUniform(new osg::Uniform("normalTex", 3));
pWater->getOrCreateStateSet()->setAttributeAndModes(pProgram);

if (m_pWater == NULL)
{
m_pWater = new osg::Geode;
m_pLayerGroup->addChild(m_pWater);
}
m_pWater->addDrawable(pWater);
m_pWater->setStateSet(pWater->getOrCreateStateSet());
}

osg::Texture2D* Water::creatText2D(const QString& strFile)
{
if (strFile.isEmpty())
{
return NULL;
}

osg::ref_ptr<osg::Texture2D> pText2D = new osg::Texture2D;
pText2D->setImage(osgDB::readImageFile(strFile.toStdString()));
pText2D->setWrap(osg::Texture2D::WRAP_S, osg::Texture2D::REPEAT);
pText2D->setWrap(osg::Texture2D::WRAP_T, osg::Texture2D::REPEAT);
pText2D->setFilter(osg::Texture2D::MIN_FILTER, osg::Texture::LINEAR_MIPMAP_LINEAR);
pText2D->setFilter(osg::Texture2D::MAG_FILTER, osg::Texture::LINEAR);
return pText2D.release();
}

shader的内容如下：

void Water::initShader()
{
m_waterVert = {
"uniform float osg_FrameTime;\n"
"varying vec4 projCoords;\n"
"varying vec3 lightDir, eyeDir;\n"
"varying vec2 flowCoords, rippleCoords;\n"

"void main()\n"
"{\n"
"   vec3 T = vec3(0.0, 1.0, 0.0);\n"
"   vec3 N = vec3(0.0, 0.0, 1.0);\n"
"   vec3 B = vec3(1.0, 0.0, 0.0);\n"
"   T = normalize(gl_NormalMatrix * T);\n"
"   B = normalize(gl_NormalMatrix * B);\n"
"   N = normalize(gl_NormalMatrix * N);\n"

"   mat3 TBNmat;\n"
"   TBNmat[0][0] = T[0]; TBNmat[1][0] = T[1]; TBNmat[2][0] = T[2];\n"
"   TBNmat[0][1] = B[0]; TBNmat[1][1] = B[1]; TBNmat[2][1] = B[2];\n"
"   TBNmat[0][2] = N[0]; TBNmat[1][2] = N[1]; TBNmat[2][2] = N[2];\n"

"   vec3 vertexInEye = vec3(gl_ModelViewMatrix * gl_Vertex);\n"
"   lightDir =  gl_LightSource[0].position.xyz - vertexInEye;\n"
"   lightDir = normalize(TBNmat * lightDir);\n"
"   eyeDir = normalize(TBNmat * (-vertexInEye));\n"

"   vec2 t1 = vec2(osg_FrameTime*0.02, osg_FrameTime*0.02);\n"
"   vec2 t2 = vec2(osg_FrameTime*0.05, osg_FrameTime*0.05);\n"
"   flowCoords = gl_MultiTexCoord0.xy + t1/10.0;\n" //* 5.0 + t1
"   rippleCoords = gl_MultiTexCoord0.xy + t1;\n"   //

"   gl_TexCoord[0] = gl_MultiTexCoord0;\n"
"   gl_Position = ftransform();\n"
"   projCoords = gl_Position;\n"
"}\n"
};

m_waterFrag = {
"uniform sampler2D defaultTex;\n"
"uniform sampler2D reflection;\n"
"uniform sampler2D refraction;\n"
"uniform sampler2D normalTex;\n"
"varying vec4 projCoords;\n"
"varying vec3 lightDir, eyeDir;\n"
"varying vec2 flowCoords, rippleCoords;\n"

"void main()\n"
"{\n"
"   vec2 rippleEffect = 0.02 * texture2D(refraction, rippleCoords * 0.01).xy;\n"
"   vec4 N = texture2D(normalTex, flowCoords + rippleEffect);\n"
"   N = N * 2.0 - vec4(1.0);\n"
"   N.a = 1.0; N = normalize(N);\n"

"   vec3 refVec = normalize(reflect(-lightDir, vec3(N) * 0.6));\n"
"   float refAngle = clamp(dot(eyeDir, refVec), 0.0, 1.0);\n"
"   vec4 specular = vec4(pow(refAngle, 40.0));\n"

"   vec2 dist = texture2D(refraction, flowCoords + rippleEffect).xy;\n"
"   dist = (dist * 2.0 - vec2(1.0)) * 0.1;\n"
"   vec2 uv = projCoords.xy / projCoords.w;\n"
"   uv = clamp((uv + 1.0) * 0.5 + dist, 0.0, 1.0);\n"

"   vec4 base = texture2D(defaultTex, uv);\n"
"   vec4 refl = texture2D(reflection, uv);\n"
"   gl_FragColor = mix(base, refl + specular, 0.6);\n"
"}\n"
};
}