Model/Model.cpp

   1 //
   2 // Copyright (c) Microsoft. All rights reserved.
   3 // This code is licensed under the MIT License (MIT).
   4 // THIS CODE IS PROVIDED *AS IS* WITHOUT WARRANTY OF
   5 // ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING ANY
   6 // IMPLIED WARRANTIES OF FITNESS FOR A PARTICULAR
   7 // PURPOSE, MERCHANTABILITY, OR NON-INFRINGEMENT.
   8 //
   9 // Developed by Minigraph
  10 //
  11 // Author:  Alex Nankervis
  12 //
  13
  14 #include "pch.h"
  15 #include "Model.h"
  16 #include <string.h>
  17 #include <float.h>
  18
  19
  20 namespace Graphics
  21 {
  22
  23 // must match the format enum in model.h
  24 const char* Model::s_FormatString[] =
  25 {
  26     "none",
  27     "h3d",
  28 };
  29 static_assert(sizeof(Model::s_FormatString) / sizeof(const char*) == Model::formats, "s_FormatString doesn't match format enum");
  30
  31 int Model::FormatFromFilename(const char *filename)
  32 {
  33     const char *p = strrchr(filename, '.');
  34     if (!p || *p == 0)
  35         return format_none;
  36
  37     for (int n = 1; n < formats; n++)
  38     {
  39         if (_stricmp(p + 1, s_FormatString[n]) == 0)
  40             return n;
  41     }
  42
  43     return format_none;
  44 }
  45
  46
  47 Model::Model()
  48     : m_pMesh(nullptr)
  49     , m_pMaterial(nullptr)
  50     , m_pVertexData(nullptr)
  51     , m_pIndexData(nullptr)
  52     , m_pVertexDataDepth(nullptr)
  53     , m_pIndexDataDepth(nullptr)
  54     , m_SRVs(nullptr)
  55 {
  56     Clear();
  57 }
  58
  59 Model::~Model()
  60 {
  61     Clear();
  62 }
  63
  64 void Model::Clear()
  65 {
  66     m_VertexBuffer.Destroy();
  67     m_IndexBuffer.Destroy();
  68     m_VertexBufferDepth.Destroy();
  69     m_IndexBufferDepth.Destroy();
  70
  71     delete [] m_pMesh;
  72     m_pMesh = nullptr;
  73     m_Header.meshCount = 0;
  74
  75     delete [] m_pMaterial;
  76     m_pMaterial = nullptr;
  77     m_Header.materialCount = 0;
  78
  79     delete [] m_pVertexData;
  80     delete [] m_pIndexData;
  81     delete [] m_pVertexDataDepth;
  82     delete [] m_pIndexDataDepth;
  83
  84     m_pVertexData = nullptr;
  85     m_Header.vertexDataByteSize = 0;
  86     m_pIndexData = nullptr;
  87     m_Header.indexDataByteSize = 0;
  88     m_pVertexDataDepth = nullptr;
  89     m_Header.vertexDataByteSizeDepth = 0;
  90     m_pIndexDataDepth = nullptr;
  91
  92     ReleaseTextures();
  93
  94     m_Header.boundingBox.min = Vector3(0.0f);
  95     m_Header.boundingBox.max = Vector3(0.0f);
  96 }
  97
  98 bool Model::Load(const char *filename)
  99 {
 100     Clear();
 101
 102     int format = FormatFromFilename(filename);
 103
 104     bool rval = false;
 105     bool needToOptimize = true;
 106     switch (format)
 107     {
 108     case format_none:
 109 #ifdef MODEL_ENABLE_ASSIMP
 110         rval = LoadAssimp(filename);
 111 #endif
 112         break;
 113
 114     case format_h3d:
 115         rval = LoadH3D(filename);
 116         needToOptimize = false;
 117         break;
 118     }
 119
 120     if (!rval)
 121         return false;
 122
 123     LoadPostProcess(needToOptimize);
 124
 125     return true;
 126 }
 127
 128 bool Model::Save(const char *filename) const
 129 {
 130     int format = FormatFromFilename(filename);
 131
 132     bool rval = false;
 133     switch (format)
 134     {
 135     case format_none:
 136         break;
 137
 138     case format_h3d:
 139         rval = SaveH3D(filename);
 140         break;
 141     }
 142
 143     return rval;
 144 }
 145
 146 // assuming at least 3 floats for position
 147 void Model::ComputeMeshBoundingBox(unsigned int meshIndex, BoundingBox &bbox) const
 148 {
 149     const Mesh *mesh = m_pMesh + meshIndex;
 150
 151     if (mesh->vertexCount > 0)
 152     {
 153         unsigned int vertexStride = mesh->vertexStride;
 154
 155         const float *p = (float*)(m_pVertexData + mesh->vertexDataByteOffset + mesh->attrib[attrib_position].offset);
 156         const float *pEnd = (float*)(m_pVertexData + mesh->vertexDataByteOffset + mesh->vertexCount * mesh->vertexStride + mesh->attrib[attrib_position].offset);
 157         bbox.min = Scalar(FLT_MAX);
 158         bbox.max = Scalar(-FLT_MAX);
 159
 160         while (p < pEnd)
 161         {
 162             Vector3 pos(*(p + 0), *(p + 1), *(p + 2));
 163
 164             bbox.min = Min(bbox.min, pos);
 165             bbox.max = Max(bbox.max, pos);
 166
 167             (*(uint8_t**)&p) += vertexStride;
 168         }
 169     }
 170     else
 171     {
 172         bbox.min = Scalar(0.0f);
 173         bbox.max = Scalar(0.0f);
 174     }
 175 }
 176
 177 void Model::ComputeGlobalBoundingBox(BoundingBox &bbox) const
 178 {
 179     if (m_Header.meshCount > 0)
 180     {
 181         bbox.min = Scalar(FLT_MAX);
 182         bbox.max = Scalar(-FLT_MAX);
 183         for (unsigned int meshIndex = 0; meshIndex < m_Header.meshCount; meshIndex++)
 184         {
 185             const Mesh *mesh = m_pMesh + meshIndex;
 186
 187             bbox.min = Min(bbox.min, mesh->boundingBox.min);
 188             bbox.max = Max(bbox.max, mesh->boundingBox.max);
 189         }
 190     }
 191     else
 192     {
 193         bbox.min = Scalar(0.0f);
 194         bbox.max = Scalar(0.0f);
 195     }
 196 }
 197
 198 void Model::ComputeAllBoundingBoxes()
 199 {
 200     for (unsigned int meshIndex = 0; meshIndex < m_Header.meshCount; meshIndex++)
 201     {
 202         Mesh *mesh = m_pMesh + meshIndex;
 203         ComputeMeshBoundingBox(meshIndex, mesh->boundingBox);
 204     }
 205     ComputeGlobalBoundingBox(m_Header.boundingBox);
 206 }
 207
 208 void Model::LoadPostProcess(bool needToOptimize)
 209 {
 210     if (needToOptimize)
 211     {
 212 #ifdef MODEL_ENABLE_OPTIMIZER
 213         Optimize();
 214 #else
 215         assert(0);
 216 #endif
 217     }
 218 }
 219
 220 } // namespace Graphics