khanat-opennel-code/code/nel/src/3d/flare_model.cpp
kaetemi e3dead19e4 Replace temporary CMaterial code
--HG--
branch : multipass-stereo
2013-09-09 21:44:14 +02:00

724 lines
23 KiB
C++

// NeL - MMORPG Framework <http://dev.ryzom.com/projects/nel/>
// Copyright (C) 2010 Winch Gate Property Limited
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as
// published by the Free Software Foundation, either version 3 of the
// License, or (at your option) any later version.
//
// This program 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
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
#include "std3d.h"
#include "nel/3d/flare_model.h"
#include "nel/3d/flare_shape.h"
#include "nel/3d/driver.h"
#include "nel/3d/material.h"
#include "nel/3d/dru.h"
#include "nel/3d/scene.h"
#include "nel/3d/render_trav.h"
#include "nel/3d/occlusion_query.h"
#include "nel/3d/mesh.h"
#include "nel/3d/viewport.h"
#include "nel/misc/common.h"
namespace NL3D {
CMaterial CFlareModel::_OcclusionQueryMaterial;
CMaterial CFlareModel::_DrawQueryMaterial;
bool CFlareModel::_OcclusionQuerySettuped = false;
CVertexBuffer CFlareModel::_OcclusionQueryVB;
using NLMISC::CVector;
// ********************************************************************************************************************
CFlareModel::CFlareModel()
{
std::fill(_Intensity, _Intensity + MaxNumContext, 0.f);
setTransparency(true);
setOpacity(false);
// RenderFilter: We are a flare
_RenderFilterType= UScene::FilterFlare;
resetOcclusionQuerries();
std::fill(_LastRenderIntervalBegin, _LastRenderIntervalBegin + MaxNumContext, (uint64) -2);
std::fill(_LastRenderIntervalEnd, _LastRenderIntervalEnd + MaxNumContext, (uint64) -2);
std::fill(_NumFrameForOcclusionQuery, _NumFrameForOcclusionQuery + MaxNumContext, 1);
Next = NULL;
}
// ********************************************************************************************************************
void CFlareModel::resetOcclusionQuerries()
{
for(uint k = 0; k < MaxNumContext; ++k)
{
for(uint l = 0; l < OcclusionTestFrameDelay; ++l)
{
_OcclusionQuery[k][l] = NULL;
_DrawQuery[k][l] = NULL;
}
}
}
// ********************************************************************************************************************
CFlareModel::~CFlareModel()
{
// if driver hasn't changed, delete all querries
if (_LastDrv)
{
for(uint k = 0; k < MaxNumContext; ++k)
{
for(uint l = 0; l < OcclusionTestFrameDelay; ++l)
{
if (_OcclusionQuery[k][l])
{
_LastDrv->deleteOcclusionQuery(_OcclusionQuery[k][l]);
}
if (_DrawQuery[k][l])
{
_LastDrv->deleteOcclusionQuery(_DrawQuery[k][l]);
}
}
}
}
}
// ********************************************************************************************************************
void CFlareModel::registerBasic()
{
// register the model
CScene::registerModel(FlareModelClassId, TransformShapeId, CFlareModel::creator);
}
// write a vector in a vertex buffer
static inline void vbWrite(uint8 *&dest, const CVector &v)
{
((float *) dest)[0] = v.x;
((float *) dest)[1] = v.y;
((float *) dest)[2] = v.z;
dest += 3 * sizeof(float);
}
// write uvs in a vertex buffer
static inline void vbWrite(uint8 *&dest, float uCoord, float vCoord)
{
((float *) dest)[0] = uCoord;
((float *) dest)[1] = vCoord;
dest += 2 * sizeof(float);
}
// ********************************************************************************************************************
void CFlareModel::traverseRender()
{
CRenderTrav &renderTrav = getOwnerScene()->getRenderTrav();
if (renderTrav.isCurrentPassOpaque()) return;
IDriver *drv = renderTrav.getDriver();
nlassert(drv);
// For now, don't render flare if occlusion query is not supported (direct read of z-buffer is far too slow)
if (!drv->supportOcclusionQuery()) return;
if (drv != _LastDrv)
{
// occlusion queries have been deleted by the driver
resetOcclusionQuerries();
_LastDrv = drv;
}
uint flareContext = _Scene ? _Scene->getFlareContext() : 0;
// transform the flare on screen
const CVector upt = getWorldMatrix().getPos(); // untransformed pos
const CVector pt = renderTrav.ViewMatrix * upt;
if (pt.y <= renderTrav.Near)
{
return; // flare behind us
}
nlassert(Shape);
CFlareShape *fs = NLMISC::safe_cast<CFlareShape *>((IShape *) Shape);
if (pt.y > fs->getMaxViewDist())
{
return; // flare too far away
}
float distIntensity;
if (fs->getFlareAtInfiniteDist())
{
distIntensity = 1.f;
}
else
{
// compute a color ratio for attenuation with distance
const float distRatio = pt.y / fs->getMaxViewDist();
distIntensity = distRatio > fs->getMaxViewDistRatio() ? 1.f - (distRatio - fs->getMaxViewDistRatio()) / (1.f - fs->getMaxViewDistRatio()) : 1.f;
}
//
uint32 width, height;
drv->getWindowSize(width, height);
// Compute position on screen
const float middleX = .5f * (renderTrav.Left + renderTrav.Right);
const float middleZ = .5f * (renderTrav.Bottom + renderTrav.Top);
const sint xPos = (width>>1) + (sint) (width * (((renderTrav.Near * pt.x) / pt.y) - middleX) / (renderTrav.Right - renderTrav.Left));
const sint yPos = (height>>1) - (sint) (height * (((renderTrav.Near * pt.z) / pt.y) - middleZ) / (renderTrav.Top - renderTrav.Bottom));
// See if the flare was inside the frustum during the last frame
// We can't use the scene frame counter because a flare can be rendered in several viewport during the same frame
// The swapBuffer counter is called only once per frame
uint64 currFrame = drv->getSwapBufferCounter();
//
bool visibilityRetrieved = false;
float visibilityRatio = 0.f;
// if driver support occlusion query mechanism, use it
CMesh *occlusionTestMesh = NULL;
if (_Scene->getShapeBank())
{
occlusionTestMesh = fs->getOcclusionTestMesh(*_Scene->getShapeBank());
}
if (drv->supportOcclusionQuery())
{
bool issueNewQuery = true;
IOcclusionQuery *lastOQ = _OcclusionQuery[flareContext][OcclusionTestFrameDelay - 1];
IOcclusionQuery *lastDQ = _DrawQuery[flareContext][OcclusionTestFrameDelay - 1];
if (_LastRenderIntervalEnd[flareContext] + 1 == currFrame)
{
if (_LastRenderIntervalEnd[flareContext] - _LastRenderIntervalBegin[flareContext] >= OcclusionTestFrameDelay - 1)
{
// occlusion test are possibles if at least OcclusionTestFrameDelay frames have ellapsed
if (lastOQ)
{
switch(lastOQ->getOcclusionType())
{
case IOcclusionQuery::NotAvailable:
issueNewQuery = false;
++ _NumFrameForOcclusionQuery[flareContext];
break;
case IOcclusionQuery::Occluded:
visibilityRetrieved = true;
visibilityRatio = 0.f;
break;
case IOcclusionQuery::NotOccluded:
if (occlusionTestMesh)
{
if (lastDQ)
{
if (lastDQ->getOcclusionType() != IOcclusionQuery::NotAvailable)
{
visibilityRetrieved = true;
// eval the percentage of samples that are visible
//nlinfo("%d / %d", lastOQ->getVisibleCount(), lastDQ->getVisibleCount());
visibilityRatio = (float) lastOQ->getVisibleCount() / (float) lastDQ->getVisibleCount();
NLMISC::clamp(visibilityRatio, 0.f, 1.f);
}
}
else
{
visibilityRetrieved = true;
visibilityRatio = 1.f;
}
}
else
{
// visibility test is done on a single point
visibilityRetrieved = true;
visibilityRatio = 1.f;
}
break;
}
}
}
}
if (issueNewQuery)
{
// shift the queries list
for(uint k = OcclusionTestFrameDelay - 1; k > 0; --k)
{
_OcclusionQuery[flareContext][k] = _OcclusionQuery[flareContext][k - 1];
_DrawQuery[flareContext][k] = _DrawQuery[flareContext][k - 1];
}
_OcclusionQuery[flareContext][0] = lastOQ;
_DrawQuery[flareContext][0] = lastDQ;
if (occlusionTestMesh)
{
occlusionTest(*occlusionTestMesh, *drv);
}
else
{
// Insert in list of waiting flare. Don't do it now to avoid repeated setup of test material (a material that don't write to color/zbuffer,
// and that is used for the sole purpose of the occlusion query)
_Scene->insertInOcclusionQueryList(this);
}
}
}
else
{
_NumFrameForOcclusionQuery[flareContext] = 1;
visibilityRetrieved = true;
// The device doesn't support asynchronous query -> must read the z-buffer directly in a slow fashion
CViewport vp;
drv->getViewport(vp);
// Read z-buffer value at the pos we are
static std::vector<float> v(1);
NLMISC::CRect rect((sint32) (vp.getX() * width + vp.getWidth() * xPos),
(sint32) (vp.getY() * height + vp.getHeight() * (height - yPos)), 1, 1);
drv->getZBufferPart(v, rect);
// Project in screen space
float z = (float) (1.0 - (1.0 / pt.y - 1.0 / renderTrav.Far) / (1.0 /renderTrav.Near - 1.0 / renderTrav.Far));
//
float depthRangeNear, depthRangeFar;
drv->getDepthRange(depthRangeNear, depthRangeFar);
z = (depthRangeFar - depthRangeNear) * z + depthRangeNear;
if (!v.size() || z > v[0]) // test against z-buffer
{
visibilityRatio = 0.f;
}
else
{
visibilityRatio = 1.f;
}
}
// Update render interval
// nlwarning("frame = %d, last frame = %d", (int) currFrame, (int) _LastRenderIntervalEnd[flareContext]);
if (_LastRenderIntervalEnd[flareContext] + 1 != currFrame)
{
//nlwarning("*");
_Intensity[flareContext] = 0.f;
_LastRenderIntervalBegin[flareContext] = currFrame;
}
_LastRenderIntervalEnd[flareContext] = currFrame;
// Update intensity depending on visibility
if (visibilityRetrieved)
{
nlassert(visibilityRatio >= 0.f);
nlassert(visibilityRatio <= 1.f);
_NumFrameForOcclusionQuery[flareContext] = 1; // reset number of frame needed to do the occlusion query
if (visibilityRatio < _Intensity[flareContext])
{
float p = fs->getPersistence();
if (p == 0.f)
{
_Intensity[flareContext] = visibilityRatio; // instant update
}
else
{
_Intensity[flareContext] -= 1.f / p * (float)_Scene->getEllapsedTime() * (float) _NumFrameForOcclusionQuery[flareContext];
if (_Intensity[flareContext] < visibilityRatio)
{
_Intensity[flareContext] = visibilityRatio;
}
}
//nlwarning("intensity update < of %x : %f", (int) this, _Intensity[flareContext]);
}
else if (visibilityRatio > _Intensity[flareContext])
{
float p = fs->getPersistence();
if (p == 0.f)
{
_Intensity[flareContext] = visibilityRatio; // instant update
}
else
{
//nlwarning("num frame = %d, currFrame = %d, ", (int) _NumFrameForOcclusionQuery[flareContext], (int) currFrame);
_Intensity[flareContext] += 1.f / p * (float)_Scene->getEllapsedTime() * (float) _NumFrameForOcclusionQuery[flareContext];
if (_Intensity[flareContext] > visibilityRatio)
{
_Intensity[flareContext] = visibilityRatio;
}
}
//nlwarning("intensity update > of %x : %f", (int) this, _Intensity[flareContext]);
}
}
if (_Intensity[flareContext] == 0.f) return;
//
static CMaterial material;
static CVertexBuffer vb;
static bool setupDone = false;
if (!setupDone)
{
material.setBlend(true);
material.setBlendFunc(CMaterial::one, CMaterial::one);
material.setZWrite(false);
material.setZFunc(CMaterial::always);
material.setLighting(false);
material.setDoubleSided(true);
// setup vertex buffer
vb.setVertexFormat(CVertexBuffer::PositionFlag | CVertexBuffer::TexCoord0Flag);
vb.setPreferredMemory(CVertexBuffer::RAMVolatile, false);
vb.setNumVertices(4);
vb.setName("CFlareModel");
{
CVertexBufferReadWrite vba;
vb.lock (vba);
vba.setTexCoord(0, 0, NLMISC::CUV(1, 0));
vba.setTexCoord(1, 0, NLMISC::CUV(1, 1));
vba.setTexCoord(2, 0, NLMISC::CUV(0, 1));
vba.setTexCoord(3, 0, NLMISC::CUV(0, 0));
}
setupDone = true;
}
// setup driver
drv->activeVertexProgram(NULL);
drv->activePixelProgram(NULL);
drv->activeGeometryProgram(NULL);
drv->setupModelMatrix(fs->getLookAtMode() ? CMatrix::Identity : getWorldMatrix());
// we don't change the fustrum to draw 2d shapes : it is costly, and we need to restore it after the drawing has been done
// we setup Z to be (near + far) / 2, and setup x and y to get the screen coordinates we want
const float zPos = 0.5f * (renderTrav.Near + renderTrav.Far);
const float zPosDivNear = zPos / renderTrav.Near;
// compute the coeff so that x = ax * px + bx; y = ax * py + by
const float aX = ( (renderTrav.Right - renderTrav.Left) / (float) width) * zPosDivNear;
const float bX = zPosDivNear * (middleX - 0.5f * (renderTrav.Right - renderTrav.Left));
//
const float aY = - ( (renderTrav.Top - renderTrav.Bottom) / (float) height) * zPosDivNear;
const float bY = zPosDivNear * (middleZ + 0.5f * (renderTrav.Top - renderTrav.Bottom));
const CVector I = renderTrav.CamMatrix.getI();
const CVector J = renderTrav.CamMatrix.getJ();
const CVector K = renderTrav.CamMatrix.getK();
//
CRGBA col;
CRGBA flareColor = fs->getColor();
const float norm = sqrtf((float) (((xPos - (width>>1)) * (xPos - (width>>1)) + (yPos - (height>>1))*(yPos - (height>>1)))))
/ (float) (width>>1);
// check for dazzle and draw it
/*if (fs->hasDazzle())
{
if (norm < fs->getDazzleAttenuationRange())
{
float dazzleIntensity = 1.f - norm / fs->getDazzleAttenuationRange();
CRGBA dazzleColor = fs->getDazzleColor();
col.modulateFromui(dazzleColor, (uint) (255.f * _Intensity * dazzleIntensity));
material.setColor(col);
material.setTexture(0, NULL);
const CVector dazzleCenter = renderTrav.CamPos + zPos * J;
const CVector dI = (width>>1) * aX * I;
const CVector dK = (height>>1) * bX * K;
vb.setVertexCoord(0, dazzleCenter + dI + dK);
vb.setVertexCoord(1, dazzleCenter + dI - dK);
vb.setVertexCoord(2, dazzleCenter - dI - dK);
vb.setVertexCoord(3, dazzleCenter - dI + dK);
drv->renderRawQuads(material, 0, 1);
}
} */
if (!fs->getAttenuable() )
{
col.modulateFromui(flareColor, (uint) (255.f * distIntensity * _Intensity[flareContext]));
}
else
{
if (norm > fs->getAttenuationRange() || fs->getAttenuationRange() == 0.f)
{
return; // nothing to draw;
}
col.modulateFromui(flareColor, (uint) (255.f * distIntensity * _Intensity[flareContext] * (1.f - norm / fs->getAttenuationRange() )));
}
col.modulateFromColor(col, getMeanColor());
if (col == CRGBA::Black) return; // not visible
material.setColor(col);
CVector scrPos; // vector that will map to the center of the flare on screen
// process each flare
// delta for each new Pos
const float dX = fs->getFlareSpacing() * ((sint) (width >> 1) - xPos);
const float dY = fs->getFlareSpacing() * ((sint) (height >> 1) - yPos);
ITexture *tex;
// special case for first flare
tex = fs->getTexture(0);
if (tex)
{
{
CVertexBufferReadWrite vba;
vb.lock (vba);
float size;
if (fs->getScaleWhenDisappear())
{
size = _Intensity[flareContext] * fs->getSize(0) + (1.f - _Intensity[flareContext]) * fs->getSizeDisappear();
}
else
{
size = fs->getSize(0);
}
CVector rI, rK;
if (fs->getFirstFlareKeepSize())
{
size *= renderTrav.Near * (getWorldMatrix().getPos() - renderTrav.CamMatrix.getPos()) * J;
}
if (fs->getAngleDisappear() == 0.f)
{
if (fs->getLookAtMode())
{
rI = I;
rK = K;
}
else
{
rI = NLMISC::CVector::I;
rK = NLMISC::CVector::K;
}
}
else
{
float angle = (1.f - _Intensity[flareContext]) * fs->getAngleDisappear() * (float) (NLMISC::Pi / 180);
float cosTheta = cosf(angle);
float sinTheta = sinf(angle);
if (fs->getLookAtMode())
{
rI = cosTheta * I + sinTheta * K;
rK = -sinTheta * I + cosTheta * K;
}
else
{
rI.set(cosTheta, 0.f, sinTheta);
rK.set(-sinTheta, 0.f, cosTheta);
}
}
uint8 *vbPtr = (uint8 *) vba.getVertexCoordPointer();
CHECK_VBA_RANGE(vba, vbPtr, vb.getVertexSize());
if (fs->getLookAtMode())
{
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, upt + size * (rI + rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 1.f, 0.f); // uvs
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, upt + size * (rI - rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 1.f, 1.f); // uvs
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, upt + size * (-rI - rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 0.f, 1.f); // uvs
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, upt + size * (-rI + rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 0.f, 0.f); // uvs
}
else
{
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, size * (rI + rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 1.f, 0.f); // uvs
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, size * (rI - rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 1.f, 1.f); // uvs
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, size * (-rI - rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 0.f, 1.f); // uvs
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, size * (-rI + rK));
CHECK_VBA(vba, vbPtr); vbWrite(vbPtr, 0.f, 0.f); // uvs
}
}
material.setTexture(0, tex);
drv->activeVertexBuffer(vb);
drv->renderRawQuads(material, 0, 1);
}
if (fs->_LookAtMode)
{
drv->setupModelMatrix(CMatrix::Identity); // look at mode is applied only to first flare
}
for (uint k = 1; k < MaxFlareNum; ++k)
{
tex = fs->getTexture(k);
if (tex)
{
// compute vector that map to the center of the flare
scrPos = (aX * (xPos + dX * fs->getRelativePos(k)) + bX) * I
+ zPos * J + (aY * (yPos + dY * fs->getRelativePos(k)) + bY) * K + renderTrav.CamMatrix.getPos();
{
CVertexBufferReadWrite vba;
vb.lock (vba);
uint8 *vbPtr = (uint8 *) vba.getVertexCoordPointer();
float size = fs->getSize(k) * zPos * renderTrav.Near;
vbWrite(vbPtr, scrPos + size * (I + K));
vbWrite(vbPtr, 1.f, 0.f); // uvs
vbWrite(vbPtr, scrPos + size * (I - K));
vbWrite(vbPtr, 1.f, 1.f); // uvs
vbWrite(vbPtr, scrPos + size * (-I - K));
vbWrite(vbPtr, 0.f, 1.f); // uvs
vbWrite(vbPtr, scrPos + size * (-I + K));
vbWrite(vbPtr, 0.f, 0.f); // uvs
}
material.setTexture(0, tex);
drv->activeVertexBuffer(vb);
drv->renderRawQuads(material, 0, 1);
}
}
}
// ********************************************************************************************************************
void CFlareModel::initStatics()
{
if (!_OcclusionQuerySettuped)
{
// setup materials
_OcclusionQueryMaterial.initUnlit();
_OcclusionQueryMaterial.setZWrite(false);
_DrawQueryMaterial.initUnlit();
_DrawQueryMaterial.setZWrite(false);
_DrawQueryMaterial.setZFunc(CMaterial::always);
// setup vbs
_OcclusionQueryVB.setVertexFormat(CVertexBuffer::PositionFlag);
_OcclusionQueryVB.setName("CFlareModel::_OcclusionQueryVB");
_OcclusionQueryVB.setPreferredMemory(CVertexBuffer::RAMVolatile, false); // use ram to avoid stall, and don't want to setup a VB per flare!
_OcclusionQueryVB.setNumVertices(1);
_OcclusionQuerySettuped = true;
}
}
// ********************************************************************************************************************
void CFlareModel::updateOcclusionQueryBegin(IDriver *drv)
{
nlassert(drv);
drv->activeVertexProgram(NULL);
drv->activePixelProgram(NULL);
drv->activeGeometryProgram(NULL);
drv->setupModelMatrix(CMatrix::Identity);
initStatics();
drv->setColorMask(false, false, false, false); // don't write any pixel during the test
}
// ********************************************************************************************************************
void CFlareModel::updateOcclusionQueryEnd(IDriver *drv)
{
drv->setColorMask(true, true, true, true);
}
// ********************************************************************************************************************
void CFlareModel::updateOcclusionQuery(IDriver *drv)
{
nlassert(drv);
nlassert(drv == _LastDrv); // driver shouldn't change during CScene::render
// allocate a new occlusion if nit already done
nlassert(_Scene);
IOcclusionQuery *oq = _OcclusionQuery[_Scene->getFlareContext()][0];
if (!oq)
{
nlassert(drv->supportOcclusionQuery());
oq = drv->createOcclusionQuery();
if (!oq) return;
_OcclusionQuery[_Scene->getFlareContext()][0] = oq;
}
{
CVertexBufferReadWrite vbrw;
_OcclusionQueryVB.lock(vbrw);
*vbrw.getVertexCoordPointer(0) = getWorldMatrix().getPos();
}
drv->activeVertexBuffer(_OcclusionQueryVB);
oq->begin();
// draw a single point
drv->renderRawPoints(_OcclusionQueryMaterial, 0, 1);
oq->end();
}
// ********************************************************************************************************************
void CFlareModel::renderOcclusionMeshPrimitives(CMesh &mesh, IDriver &drv)
{
uint numMatrixBlock = mesh.getNbMatrixBlock();
for(uint k = 0; k < numMatrixBlock; ++k)
{
uint numRdrPass = mesh.getNbRdrPass(k);
for(uint l = 0; l < numRdrPass; ++l)
{
CIndexBuffer &ib = const_cast<CIndexBuffer &>(mesh.getRdrPassPrimitiveBlock(k, l));
drv.activeIndexBuffer(ib);
drv.renderSimpleTriangles(0, ib.getNumIndexes() / 3);
}
}
}
// ********************************************************************************************************************
void CFlareModel::setupOcclusionMeshMatrix(IDriver &drv, CScene &scene) const
{
nlassert(Shape);
CFlareShape *fs = NLMISC::safe_cast<CFlareShape *>((IShape *) Shape);
if (fs->getOcclusionTestMeshInheritScaleRot())
{
drv.setupModelMatrix(getWorldMatrix());
}
else
{
nlassert(scene.getCam());
CMatrix m = scene.getCam()->getWorldMatrix();
m.setPos(getWorldMatrix().getPos());
drv.setupModelMatrix(m);
}
}
// ********************************************************************************************************************
void CFlareModel::occlusionTest(CMesh &mesh, IDriver &drv)
{
nlassert(_Scene);
initStatics();
IOcclusionQuery *oq = _OcclusionQuery[_Scene->getFlareContext()][0];
if (!oq)
{
nlassert(drv.supportOcclusionQuery());
oq = drv.createOcclusionQuery();
if (!oq) return;
_OcclusionQuery[_Scene->getFlareContext()][0] = oq;
}
IOcclusionQuery *dq = _DrawQuery[_Scene->getFlareContext()][0];
if (!dq)
{
nlassert(drv.supportOcclusionQuery());
dq = drv.createOcclusionQuery();
if (!dq) return;
_DrawQuery[_Scene->getFlareContext()][0] = dq;
}
drv.setColorMask(false, false, false, false); // don't write any pixel during the test
drv.activeVertexProgram(NULL);
drv.activePixelProgram(NULL);
drv.activeGeometryProgram(NULL);
setupOcclusionMeshMatrix(drv, *_Scene);
drv.activeVertexBuffer(const_cast<CVertexBuffer &>(mesh.getVertexBuffer()));
// query drawn count
drv.setupMaterial(_OcclusionQueryMaterial);
oq->begin();
renderOcclusionMeshPrimitives(mesh, drv);
oq->end();
// query total count
drv.setupMaterial(_DrawQueryMaterial);
dq->begin();
renderOcclusionMeshPrimitives(mesh, drv);
dq->end();
drv.setColorMask(true, true, true, true); // restore pixel writes
}
// ********************************************************************************************************************
void CFlareModel::renderOcclusionTestMesh(IDriver &drv)
{
nlassert(_Scene);
if (!_Scene->getShapeBank()) return;
nlassert(Shape);
CFlareShape *fs = NLMISC::safe_cast<CFlareShape *>((IShape *) Shape);
CMesh *occlusionTestMesh = fs->getOcclusionTestMesh(*_Scene->getShapeBank());
if (!occlusionTestMesh) return;
setupOcclusionMeshMatrix(drv, *_Scene);
drv.activeVertexBuffer(const_cast<CVertexBuffer &>(occlusionTestMesh->getVertexBuffer()));
renderOcclusionMeshPrimitives(*occlusionTestMesh, drv);
}
} // NL3D