KeosMeshGeom.cpp

Go to the documentation of this file.

/*
 * This source file is part of KEOS (Free 3D Engine)
 * For the latest info, see http://www.keosengine.org/
 * E-mails : thierry.vouriot@keosengine.org, yeri@keosengine.org
 *
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free Software
 * Foundation; either version 2 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
 * Place - Suite 330, Boston, MA 02111-1307, USA, or go to
 * http://www.gnu.org/copyleft/lesser.txt.
 *
 */

#include "KeosMeshGeom.h"
#include "KeosException.h"
#include "KeosRoot.h"
#include "KeosRenderSystem.h"
#include "KeosMesh.h"

namespace Keos
{
  //=======================================================================
  // CMeshGeom implementation
  //=======================================================================

  //-----------------------------------------------------------------------
  CMeshGeom::CMeshGeom(const TVertex* pVertices, ulong nVerticesCount,
                       const TIndex* pIndices, ulong nIndicesCount, TPrimitiveType PrimitiveType)
  {
    Assert(pVertices != NULL);
    Assert(pIndices != NULL);

    TDeclarationElement Elements[] =
      {
        {0, ELT_USAGE_POSITION,     ELT_TYPE_FLOAT3},
        {0, ELT_USAGE_NORMAL,       ELT_TYPE_FLOAT3},
        {0, ELT_USAGE_DIFFUSE,      ELT_TYPE_COLOR},
        {0, ELT_USAGE_TEXCOORD0,    ELT_TYPE_FLOAT2}
      };

    IRenderSystem* pRenderSystem = CRoot::Instance().GetRenderSystem();

    m_pDeclaration  = pRenderSystem->CreateVertexDeclaration(Elements);
    m_VertexBuffer  = pRenderSystem->CreateVertexBuffer(nVerticesCount, 0, pVertices);
    m_IndexBuffer   = pRenderSystem->CreateIndexBuffer(nIndicesCount, 0, pIndices);

    m_nMaterialIndex = -1;
    m_bUseSharedVertexBuffer = false;

    m_PrimitiveType = PrimitiveType;
    m_nPrimitivesCount = GetPrimitivesCount(PrimitiveType);
  }

  //-----------------------------------------------------------------------
#if KEOS_MESH_SHARED_VERTEX_BUFFER
  CMeshGeom::CMeshGeom(const TIndex* pIndices, ulong nIndicesCount, TPrimitiveType PrimitiveType)
  {
    Assert(pIndices != NULL);

    TDeclarationElement Elements[] =
      {
        {0, ELT_USAGE_POSITION,     ELT_TYPE_FLOAT3},
        {0, ELT_USAGE_NORMAL,       ELT_TYPE_FLOAT3},
        {0, ELT_USAGE_DIFFUSE,      ELT_TYPE_COLOR},
        {0, ELT_USAGE_TEXCOORD0,    ELT_TYPE_FLOAT2}
      };

    IRenderSystem* pRenderSystem = CRoot::Instance().GetRenderSystem();

    m_pDeclaration  = pRenderSystem->CreateVertexDeclaration(Elements);
    m_IndexBuffer   = pRenderSystem->CreateIndexBuffer(nIndicesCount, 0, pIndices);

    m_nMaterialIndex = -1;
    m_bUseSharedVertexBuffer = true;

    m_PrimitiveType = PrimitiveType;
    m_nPrimitivesCount = GetPrimitivesCount(PrimitiveType);
  }
#endif

  //-----------------------------------------------------------------------
  CMeshGeom::~CMeshGeom()
  {}

  //-----------------------------------------------------------------------
#if KEOS_DEBUG_MODE
  void CMeshGeom::SetMaterialIndex(int nIndex)
  {
    Assert(nIndex < (int)(m_pParentMesh->GetMaterialsCount()));
    m_nMaterialIndex = nIndex;

  }
#endif

  //-----------------------------------------------------------------------
  int CMeshGeom::GetPrimitivesCount(TPrimitiveType PrimitiveType)
  {
    switch (PrimitiveType)
    {
      case PT_TRIANGLELIST:
        return m_IndexBuffer.GetCount() / 3;
      case PT_TRIANGLESTRIP:
        return m_IndexBuffer.GetCount() - 2;
      case PT_LINELIST:
        return m_IndexBuffer.GetCount() / 2;
      case PT_LINESTRIP:
        return m_IndexBuffer.GetCount() - 2;
      case PT_POINTLIST:
        return m_IndexBuffer.GetCount();
    }

    return 0;
  }

  //-----------------------------------------------------------------------
  void CMeshGeom::AssignColor(const CColor& Color)
  {
    IRenderSystem* pRenderSystem = CRoot::Instance().GetRenderSystem();

    if (!m_bUseSharedVertexBuffer)
    {
      TVertex* Vertices = m_VertexBuffer.Lock(0, 0, LOCK_WRITEONLY);
      for (uint i = 0; i < m_VertexBuffer.GetCount(); i++)
      {
        Vertices[i].Color = pRenderSystem->ConvertColor(Color);
      }
      m_VertexBuffer.Unlock();
    }
    else
    {
      TVertex* Vertices = m_pParentMesh->m_SharedVertexBuffer.Lock(0, 0, LOCK_WRITEONLY);
      for (uint i = 0; i < m_pParentMesh->m_SharedVertexBuffer.GetCount(); i++)
      {
        Vertices[i].Color = pRenderSystem->ConvertColor(Color);
      }
      m_pParentMesh->m_SharedVertexBuffer.Unlock();
    }
  }

  //-----------------------------------------------------------------------
  void CMeshGeom::AssignMaterialColor()
  {
    if (m_nMaterialIndex == -1) return;
    Assert(m_nMaterialIndex < (int)(m_pParentMesh->GetMaterialsCount()));
    CMaterial* pMaterial = m_pParentMesh->GetMaterial(m_nMaterialIndex);
    AssignColor(pMaterial->GetDiffuseColor());
  }

} // namespace Keos

---