openmw-tes3mp/apps/openmw/mwrender/water.cpp

#include "water.hpp"

namespace MWRender {
  Water::Water (Ogre::Camera *camera, const ESM::Cell* cell) : mCamera (camera), mViewport (camera->getViewport()), mSceneManager (camera->getSceneManager()) {
    try {
      Ogre::CompositorManager::getSingleton().addCompositor(mViewport, "Water", -1);
      Ogre::CompositorManager::getSingleton().setCompositorEnabled(mViewport, "Water", false);
    } catch(...) {
    }
    mTop = cell->water;
    
   
    mIsUnderwater = false;
    //mCamera->addListener(this);
        
    mWaterPlane = Ogre::Plane(Ogre::Vector3::UNIT_Y, 0);
    
    Ogre::MeshManager::getSingleton().createPlane("water", Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME,  mWaterPlane, CELL_SIZE*5, CELL_SIZE * 5, 10, 10, true, 1, 3,5, Ogre::Vector3::UNIT_Z);
    
    
    mWater = mSceneManager->createEntity("water");
    
      mWater->setMaterialName("Examples/Water0");
    
        
      
    
    mWaterNode = mSceneManager->getRootSceneNode()->createChildSceneNode();
    mWaterNode->setPosition(0, mTop, 0);
    
    
   if(!(cell->data.flags & cell->Interior))
   {
       mWaterNode->setPosition(getSceneNodeCoordinates(cell->data.gridX, cell->data.gridY));
   }
    mWaterNode->attachObject(mWater);
   
  }


  Water::~Water() {
      Ogre::MeshManager::getSingleton().remove("water");
    
      mWaterNode->detachObject(mWater);
      mSceneManager->destroyEntity(mWater);
      mSceneManager->destroySceneNode(mWaterNode);
     
   
    Ogre::CompositorManager::getSingleton().removeCompositorChain(mViewport);
  }

  void Water::changeCell(const ESM::Cell* cell){
      mTop = cell->water;
      
      
   
        if(!(cell->data.flags & cell->Interior))
            mWaterNode->setPosition(getSceneNodeCoordinates(cell->data.gridX, cell->data.gridY));
        else
            mWaterNode->setPosition(0, mTop, 0);
  }



  void Water::checkUnderwater(float  y) {

    if (mIsUnderwater && y > mTop) {
      try {
	Ogre::CompositorManager::getSingleton().setCompositorEnabled(mViewport, "Water", false);
      } catch(...) {
      }
      mIsUnderwater = false;
    } 

    if (!mIsUnderwater && y < mTop) {
      try {
	Ogre::CompositorManager::getSingleton().setCompositorEnabled(mViewport, "Water", true);
      } catch(...) {
      }
      mIsUnderwater = true;
    }
  }


  
  Ogre::Vector3 Water::getSceneNodeCoordinates(int gridX, int gridY){
      Ogre::Vector3 out = Ogre::Vector3(gridX * CELL_SIZE + (CELL_SIZE / 2), mTop, -gridY * CELL_SIZE - (CELL_SIZE / 2));
      
      return out;
  }

}