openmw-tes3mp/terrain/cpp_mesh.cpp

// The Ogre renderable used to hold and display the terrain meshes.
class TerrainMesh : public Ogre::Renderable, public Ogre::MovableObject
{
public:

  TerrainMesh(Ogre::SceneNode *parent, const MeshInfo &info,
              int level, float scale)
    : Ogre::Renderable(),
      Ogre::MovableObject()
  {
    TRACE("TerrainMesh()");

    mLevel = level;

    // This is a bit messy, with everything in one function. We could
    // split it up later.

    // Use MW coordinates all the way
    assert(info.worldWidth > 0);
    assert(info.minHeight <= info.maxHeight);
    mBounds.setExtents(0,0,info.minHeight,
                       info.worldWidth, info.worldWidth,
                       info.maxHeight);
    mCenter = mBounds.getCenter();
    mBoundingRadius = mBounds.getHalfSize().length();

    // TODO: VertexData has a clone() function. This probably means we
    // can set this up once and then clone it, to get a completely
    // unnoticable increase in performance :)
    mVertices = new VertexData();
    mVertices->vertexStart = 0;
    mVertices->vertexCount = info.vertRows*info.vertCols;

    VertexDeclaration* vertexDecl = mVertices->vertexDeclaration;
    size_t currOffset = 0;

    vertexDecl->addElement(0, currOffset, VET_FLOAT3, VES_POSITION);
    currOffset += VertexElement::getTypeSize(VET_FLOAT3);

    vertexDecl->addElement(0, currOffset, VET_FLOAT3, VES_NORMAL);
    currOffset += VertexElement::getTypeSize(VET_FLOAT3);

    vertexDecl->addElement(0, currOffset, VET_FLOAT2,
                           VES_TEXTURE_COORDINATES, 0);
    currOffset += VertexElement::getTypeSize(VET_FLOAT2);

    assert(vertexDecl->getVertexSize(0) == currOffset);

    HardwareVertexBufferSharedPtr mMainBuffer;
    mMainBuffer = HardwareBufferManager::getSingleton().createVertexBuffer
      (
       vertexDecl->getVertexSize(0), // size of one whole vertex
       mVertices->vertexCount,       // number of vertices
       HardwareBuffer::HBU_STATIC_WRITE_ONLY, // usage
       false); // no shadow buffer

    // Bind the data
    mVertices->vertexBufferBinding->setBinding(0, mMainBuffer);

    // Fill the buffer
    float* verts = static_cast<float*>
      (mMainBuffer->lock(HardwareBuffer::HBL_DISCARD));
    info.fillVertexBuffer(verts,8*mVertices->vertexCount);
    mMainBuffer->unlock();

    // Create the index data holder
    mIndices = new IndexData();
    mIndices->indexCount = 64*64*6; // TODO: Shouldn't be hard-coded
    mIndices->indexBuffer =
      HardwareBufferManager::getSingleton().createIndexBuffer
      ( HardwareIndexBuffer::IT_16BIT,
        mIndices->indexCount,
        HardwareBuffer::HBU_STATIC_WRITE_ONLY,
        false);

    // Fill the buffer with warm fuzzy archive data
    unsigned short* indices = static_cast<unsigned short*>
      (mIndices->indexBuffer->lock
       (0, mIndices->indexBuffer->getSizeInBytes(),
        HardwareBuffer::HBL_DISCARD));
    info.fillIndexBuffer(indices,mIndices->indexCount);
    mIndices->indexBuffer->unlock();

    // Finally, create the material
    const std::string texName = info.getTexName();

    // Set up the scene node.
    mNode = parent->createChildSceneNode();
    mNode->attachObject(this);

    // Finally, create or retrieve the material
    if(MaterialManager::getSingleton().resourceExists(texName))
      {
        mMaterial = MaterialManager::getSingleton().getByName
          (texName);
        return;
      }

    // No existing material. Create a new one.
    mMaterial = MaterialManager::getSingleton().create
      (texName, ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME);

    Pass* pass = mMaterial->getTechnique(0)->getPass(0);
    pass->setLightingEnabled(false);

    if(level > 1)
      {
        // This material just has a normal texture
        pass->createTextureUnitState(texName)
          //->setTextureAddressingMode(TextureUnitState::TAM_CLAMP)
          ;
      }
    else
      {
        assert(level == 1);

        // Get the background texture. TODO: We should get this from
        // somewhere, no file names should be hard coded. The texture
        // might exist as a .tga in earlier versions of the game, and
        // we might also want to specify a different background
        // texture on some meshes.
        //const char *bgTex = info.getBackgroundTex();

        const char *bgTex = "_land_default.dds";
        pass->createTextureUnitState(bgTex)
          ->setTextureScale(scale,scale);

        // Loop through all the textures in this mesh
        for(int tnum=0; tnum<info.alphaNum; tnum++)
          {
            const AlphaInfo &alpha = *info.getAlphaInfo(tnum);

            // Name of the alpha map texture to create
            std::string alphaName = alpha.getAlphaName();

            // Name of the texture
            std::string tname = alpha.getTexName();

            // Create the alpha texture if it doesn't exist
            if(!TextureManager::getSingleton().resourceExists(alphaName))
              {
                TexturePtr texPtr = Ogre::TextureManager::
                  getSingleton().createManual
                  (alphaName,
                   Ogre::ResourceGroupManager::DEFAULT_RESOURCE_GROUP_NAME,
                   Ogre::TEX_TYPE_2D,
                   g_alphaSize,g_alphaSize,
                   1,0, // depth, mipmaps
                   Ogre::PF_A8, // One-channel alpha
                   Ogre::TU_STATIC_WRITE_ONLY);

                // Get the pointer
                Ogre::HardwarePixelBufferSharedPtr pixelBuffer = texPtr->getBuffer();
                pixelBuffer->lock(Ogre::HardwareBuffer::HBL_DISCARD);
                const Ogre::PixelBox& pixelBox = pixelBuffer->getCurrentLock();
                Ogre::uint8* pDest = static_cast<Ogre::uint8*>(pixelBox.data);

                // Copy alpha data from file
                alpha.fillAlphaBuffer(pDest,g_alphaSize*g_alphaSize);

                // Close the buffer
                pixelBuffer->unlock();
              }

            pass = mMaterial->getTechnique(0)->createPass();
            pass->setSceneBlending(Ogre::SBT_TRANSPARENT_ALPHA);
            pass->setLightingEnabled(false);
            pass->setDepthFunction(Ogre::CMPF_EQUAL);

            Ogre::TextureUnitState* tus = pass->createTextureUnitState(alphaName);
            //tus->setTextureAddressingMode(Ogre::TextureUnitState::TAM_CLAMP);

            tus->setAlphaOperation(Ogre::LBX_BLEND_TEXTURE_ALPHA,
                                   Ogre::LBS_TEXTURE,
                                   Ogre::LBS_TEXTURE);
            tus->setColourOperationEx(Ogre::LBX_BLEND_DIFFUSE_ALPHA,
                                      Ogre::LBS_TEXTURE,
                                      Ogre::LBS_TEXTURE);
            tus->setIsAlpha(true);

            // Add the actual texture on top of the alpha map.
            tus = pass->createTextureUnitState(tname);
            tus->setColourOperationEx(Ogre::LBX_BLEND_DIFFUSE_ALPHA,
                                      Ogre::LBS_TEXTURE,
                                      Ogre::LBS_CURRENT);

            tus->setTextureScale(scale, scale);
          }
      }
  }

  ~TerrainMesh()
  {
    assert(mNode);
    mNode->detachAllObjects();
    mNode->getCreator()->destroySceneNode(mNode);

    // TODO: This still crashes on level1 meshes. Find out why!
    if(mLevel!=1)delete mVertices;
    delete mIndices;
  }

  //-----------------------------------------------------------------------
  // These are all Ogre functions that we have to override
  //-----------------------------------------------------------------------

  // Internal Ogre function. We should call visitor->visit on all
  // Renderables that are part of this object. In our case, this is
  // only ourselves.
  void visitRenderables(Renderable::Visitor* visitor,
                        bool debugRenderables = false) {
    visitor->visit(this, 0, false);
  }

  void getRenderOperation( Ogre::RenderOperation& op ) {
    op.useIndexes = true;
    op.operationType = Ogre::RenderOperation::OT_TRIANGLE_LIST;
    op.vertexData = mVertices;
    op.indexData = mIndices;
  }

  void getWorldTransforms( Ogre::Matrix4* xform ) const {
    *xform = mNode->_getFullTransform();
  }

  const Ogre::Quaternion& getWorldOrientation(void) const {
    return mNode->_getDerivedOrientation();
  }

  const Ogre::Vector3& getWorldPosition(void) const {
    return mNode->_getDerivedPosition();
  }

  Ogre::Real getSquaredViewDepth(const Ogre::Camera *cam) const {
    Ogre::Vector3 diff = mCenter - cam->getDerivedPosition();
    // Use squared length to avoid square root
    return diff.squaredLength();
  }

  const Ogre::LightList& getLights(void) const {
    if (mLightListDirty) {
      getParentSceneNode()->getCreator()->_populateLightList
        (mCenter, mBoundingRadius, mLightList);
      mLightListDirty = false;
    }
    return mLightList;
  }
  virtual const Ogre::String& getMovableType( void ) const {
    static Ogre::String t = "MW_TERRAIN";
    return t;
  }
  void _updateRenderQueue( Ogre::RenderQueue* queue ) {
    mLightListDirty = true;
    queue->addRenderable(this, mRenderQueueID);
  }
  const Ogre::AxisAlignedBox& getBoundingBox( void ) const
  {
    return mBounds;
  }

  Ogre::Real getBoundingRadius(void) const {
    return mBoundingRadius;
  }
  virtual const MaterialPtr& getMaterial(void) const
  { return mMaterial; }

  //-----------------------------------------------------------------------
  //-----------------------------------------------------------------------

private:

  int mLevel;

  Ogre::SceneNode* mNode;

  Ogre::MaterialPtr mMaterial;

  Ogre::VertexData* mVertices;
  Ogre::IndexData* mIndices;

  mutable bool mLightListDirty;

  Ogre::Real mBoundingRadius;
  Ogre::AxisAlignedBox mBounds;
  Ogre::Vector3 mCenter;
};