2012-01-14 18:34:14 -05:00

105 lines
3.1 KiB
C++

#include "water.hpp"
namespace MWRender {
Water::Water (Ogre::Camera *camera) : mCamera (camera), mViewport (camera->getViewport()), mSceneManager (camera->getSceneManager()) {
try {
Ogre::CompositorManager::getSingleton().addCompositor(mViewport, "Water", -1);
Ogre::CompositorManager::getSingleton().setCompositorEnabled(mViewport, "Water", false);
} catch(...) {
}
mIsUnderwater = false;
mCamera->addListener(this);
for (unsigned int i = 0; i < 2; i++) {
Ogre::TexturePtr tex = Ogre::TextureManager::getSingleton().createManual(i == 0 ? "refraction" : "reflection", Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME, Ogre::TEX_TYPE_2D, 512, 512, 0, Ogre::PF_R8G8B8, Ogre::TU_RENDERTARGET);
Ogre::RenderTarget* rtt = tex->getBuffer()->getRenderTarget();
rtt->addViewport(mCamera)->setOverlaysEnabled(false);
rtt->addListener(this);
if (i == 0) mRefractionTarget = rtt;
else mReflectionTarget = rtt;
}
mWaterPlane = Ogre::Plane(Ogre::Vector3::UNIT_Y, 0);
Ogre::MeshManager::getSingleton().createPlane("water", Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME, mWaterPlane, 7000, 7000, 10, 10, true, 1, 3, 5, Ogre::Vector3::UNIT_Z);
mWater = mSceneManager->createEntity("Water", "water");
mWater->setMaterialName("Water/ReflectionRefraction");
mWaterNode = mSceneManager->getRootSceneNode()->createChildSceneNode();
mWaterNode->attachObject(mWater);
}
Water::~Water() {
Ogre::MeshManager::getSingleton().remove("water");
Ogre::TextureManager::getSingleton().remove("refraction");
Ogre::TextureManager::getSingleton().remove("reflection");
Ogre::CompositorManager::getSingleton().removeCompositorChain(mViewport);
}
void Water::preRenderTargetUpdate(const Ogre::RenderTargetEvent& evt) {
mWater->setVisible(false);
if (evt.source == mReflectionTarget) {
mCamera->enableReflection(mWaterPlane);
} else {
}
}
void Water::postRenderTargetUpdate(const Ogre::RenderTargetEvent& evt) {
mWater->setVisible(true);
if (evt.source == mReflectionTarget) {
mCamera->disableReflection();
} else {
}
}
void Water::checkUnderwater() {
Ogre::Vector3 pos = mCamera->getPosition();
if (mIsUnderwater && pos.y > 0) {
try {
Ogre::CompositorManager::getSingleton().setCompositorEnabled(mViewport, "Water", false);
} catch(...) {
}
std::cout << "Removing water compositor" << "\n";
mIsUnderwater = false;
}
if (!mIsUnderwater && pos.y < 0) {
try {
Ogre::CompositorManager::getSingleton().setCompositorEnabled(mViewport, "Water", true);
} catch(...) {
}
mIsUnderwater = true;
std::cout << "Adding water compositor" << "\n";
}
}
void Water::cameraPreRenderScene(Ogre::Camera *cam) {
Ogre::Vector3 pos = cam->getPosition();
if (pos != mOldCameraPos) {
mWaterNode->setPosition(pos.x, 0, pos.z);
mOldCameraPos = pos;
}
}
void Water::cameraPostRenderScene(Ogre::Camera *cam) {
}
void Water::cameraDestroyed(Ogre::Camera *cam) {
}
}