Model.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688
  1. //
  2. // Copyright (c) 2008-2014 the Urho3D project.
  3. //
  4. // Permission is hereby granted, free of charge, to any person obtaining a copy
  5. // of this software and associated documentation files (the "Software"), to deal
  6. // in the Software without restriction, including without limitation the rights
  7. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  8. // copies of the Software, and to permit persons to whom the Software is
  9. // furnished to do so, subject to the following conditions:
  10. //
  11. // The above copyright notice and this permission notice shall be included in
  12. // all copies or substantial portions of the Software.
  13. //
  14. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  15. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  16. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  17. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  18. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  19. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  20. // THE SOFTWARE.
  21. //
  22. #include "Precompiled.h"
  23. #include "Context.h"
  24. #include "Deserializer.h"
  25. #include "Geometry.h"
  26. #include "IndexBuffer.h"
  27. #include "Log.h"
  28. #include "Model.h"
  29. #include "Profiler.h"
  30. #include "Graphics.h"
  31. #include "Serializer.h"
  32. #include "VertexBuffer.h"
  33. #include <cstring>
  34. #include "DebugNew.h"
  35. namespace Urho3D
  36. {
  37. unsigned LookupVertexBuffer(VertexBuffer* buffer, const Vector<SharedPtr<VertexBuffer> >& buffers)
  38. {
  39. for (unsigned i = 0; i < buffers.Size(); ++i)
  40. {
  41. if (buffers[i] == buffer)
  42. return i;
  43. }
  44. return 0;
  45. }
  46. unsigned LookupIndexBuffer(IndexBuffer* buffer, const Vector<SharedPtr<IndexBuffer> >& buffers)
  47. {
  48. for (unsigned i = 0; i < buffers.Size(); ++i)
  49. {
  50. if (buffers[i] == buffer)
  51. return i;
  52. }
  53. return 0;
  54. }
  55. Model::Model(Context* context) :
  56. Resource(context)
  57. {
  58. }
  59. Model::~Model()
  60. {
  61. }
  62. void Model::RegisterObject(Context* context)
  63. {
  64. context->RegisterFactory<Model>();
  65. }
  66. bool Model::BeginLoad(Deserializer& source)
  67. {
  68. // Check ID
  69. if (source.ReadFileID() != "UMDL")
  70. {
  71. LOGERROR(source.GetName() + " is not a valid model file");
  72. return false;
  73. }
  74. geometries_.Clear();
  75. geometryBoneMappings_.Clear();
  76. geometryCenters_.Clear();
  77. morphs_.Clear();
  78. vertexBuffers_.Clear();
  79. indexBuffers_.Clear();
  80. unsigned memoryUse = sizeof(Model);
  81. // Read vertex buffers
  82. unsigned numVertexBuffers = source.ReadUInt();
  83. vertexBuffers_.Reserve(numVertexBuffers);
  84. morphRangeStarts_.Resize(numVertexBuffers);
  85. morphRangeCounts_.Resize(numVertexBuffers);
  86. loadVBData_.Resize(numVertexBuffers);
  87. for (unsigned i = 0; i < numVertexBuffers; ++i)
  88. {
  89. unsigned vertexCount = source.ReadUInt();
  90. unsigned elementMask = source.ReadUInt();
  91. morphRangeStarts_[i] = source.ReadUInt();
  92. morphRangeCounts_[i] = source.ReadUInt();
  93. SharedPtr<VertexBuffer> buffer(new VertexBuffer(context_));
  94. unsigned vertexSize = VertexBuffer::GetVertexSize(elementMask);
  95. // Prepare vertex buffer data to be uploaded during EndLoad()
  96. loadVBData_[i].vertexCount_ = vertexCount;
  97. loadVBData_[i].elementMask_ = elementMask;
  98. loadVBData_[i].dataSize_ = vertexCount * vertexSize;
  99. loadVBData_[i].data_ = new unsigned char[loadVBData_[i].dataSize_];
  100. source.Read(loadVBData_[i].data_.Get(), loadVBData_[i].dataSize_);
  101. memoryUse += sizeof(VertexBuffer) + vertexCount * vertexSize;
  102. vertexBuffers_.Push(buffer);
  103. }
  104. // Read index buffers
  105. unsigned numIndexBuffers = source.ReadUInt();
  106. indexBuffers_.Reserve(numIndexBuffers);
  107. loadIBData_.Resize(numIndexBuffers);
  108. for (unsigned i = 0; i < numIndexBuffers; ++i)
  109. {
  110. unsigned indexCount = source.ReadUInt();
  111. unsigned indexSize = source.ReadUInt();
  112. SharedPtr<IndexBuffer> buffer(new IndexBuffer(context_));
  113. // Prepare index buffer data to be uploaded during EndLoad()
  114. loadIBData_[i].indexCount_ = indexCount;
  115. loadIBData_[i].indexSize_ = indexSize;
  116. loadIBData_[i].dataSize_ = indexCount * indexSize;
  117. loadIBData_[i].data_ = new unsigned char[loadIBData_[i].dataSize_];
  118. source.Read(loadIBData_[i].data_.Get(), loadIBData_[i].dataSize_);
  119. memoryUse += sizeof(IndexBuffer) + indexCount * indexSize;
  120. indexBuffers_.Push(buffer);
  121. }
  122. // Read geometries
  123. unsigned numGeometries = source.ReadUInt();
  124. geometries_.Reserve(numGeometries);
  125. geometryBoneMappings_.Reserve(numGeometries);
  126. geometryCenters_.Reserve(numGeometries);
  127. loadGeometries_.Resize(numGeometries);
  128. for (unsigned i = 0; i < numGeometries; ++i)
  129. {
  130. // Read bone mappings
  131. unsigned boneMappingCount = source.ReadUInt();
  132. PODVector<unsigned> boneMapping(boneMappingCount);
  133. for (unsigned j = 0; j < boneMappingCount; ++j)
  134. boneMapping[j] = source.ReadUInt();
  135. geometryBoneMappings_.Push(boneMapping);
  136. unsigned numLodLevels = source.ReadUInt();
  137. Vector<SharedPtr<Geometry> > geometryLodLevels;
  138. geometryLodLevels.Reserve(numLodLevels);
  139. loadGeometries_[i].Resize(numLodLevels);
  140. for (unsigned j = 0; j < numLodLevels; ++j)
  141. {
  142. float distance = source.ReadFloat();
  143. PrimitiveType type = (PrimitiveType)source.ReadUInt();
  144. unsigned vbRef = source.ReadUInt();
  145. unsigned ibRef = source.ReadUInt();
  146. unsigned indexStart = source.ReadUInt();
  147. unsigned indexCount = source.ReadUInt();
  148. if (vbRef >= vertexBuffers_.Size())
  149. {
  150. LOGERROR("Vertex buffer index out of bounds");
  151. loadVBData_.Clear();
  152. loadIBData_.Clear();
  153. loadGeometries_.Clear();
  154. return false;
  155. }
  156. if (ibRef >= indexBuffers_.Size())
  157. {
  158. LOGERROR("Index buffer index out of bounds");
  159. loadVBData_.Clear();
  160. loadIBData_.Clear();
  161. loadGeometries_.Clear();
  162. return false;
  163. }
  164. SharedPtr<Geometry> geometry(new Geometry(context_));
  165. geometry->SetLodDistance(distance);
  166. // Prepare geometry to be defined during EndLoad()
  167. loadGeometries_[i][j].type_ = type;
  168. loadGeometries_[i][j].vbRef_ = vbRef;
  169. loadGeometries_[i][j].ibRef_ = ibRef;
  170. loadGeometries_[i][j].indexStart_ = indexStart;
  171. loadGeometries_[i][j].indexCount_ = indexCount;
  172. geometryLodLevels.Push(geometry);
  173. memoryUse += sizeof(Geometry);
  174. }
  175. geometries_.Push(geometryLodLevels);
  176. }
  177. // Read morphs
  178. unsigned numMorphs = source.ReadUInt();
  179. morphs_.Reserve(numMorphs);
  180. for (unsigned i = 0; i < numMorphs; ++i)
  181. {
  182. ModelMorph newMorph;
  183. newMorph.name_ = source.ReadString();
  184. newMorph.nameHash_ = newMorph.name_;
  185. newMorph.weight_ = 0.0f;
  186. unsigned numBuffers = source.ReadUInt();
  187. for (unsigned j = 0; j < numBuffers; ++j)
  188. {
  189. VertexBufferMorph newBuffer;
  190. unsigned bufferIndex = source.ReadUInt();
  191. newBuffer.elementMask_ = source.ReadUInt();
  192. newBuffer.vertexCount_ = source.ReadUInt();
  193. // Base size: size of each vertex index
  194. unsigned vertexSize = sizeof(unsigned);
  195. // Add size of individual elements
  196. if (newBuffer.elementMask_ & MASK_POSITION)
  197. vertexSize += sizeof(Vector3);
  198. if (newBuffer.elementMask_ & MASK_NORMAL)
  199. vertexSize += sizeof(Vector3);
  200. if (newBuffer.elementMask_ & MASK_TANGENT)
  201. vertexSize += sizeof(Vector3);
  202. newBuffer.dataSize_ = newBuffer.vertexCount_ * vertexSize;
  203. newBuffer.morphData_ = new unsigned char[newBuffer.dataSize_];
  204. source.Read(&newBuffer.morphData_[0], newBuffer.vertexCount_ * vertexSize);
  205. newMorph.buffers_[bufferIndex] = newBuffer;
  206. memoryUse += sizeof(VertexBufferMorph) + newBuffer.vertexCount_ * vertexSize;
  207. }
  208. morphs_.Push(newMorph);
  209. memoryUse += sizeof(ModelMorph);
  210. }
  211. // Read skeleton
  212. skeleton_.Load(source);
  213. memoryUse += skeleton_.GetNumBones() * sizeof(Bone);
  214. // Read bounding box
  215. boundingBox_ = source.ReadBoundingBox();
  216. // Read geometry centers
  217. for (unsigned i = 0; i < geometries_.Size() && !source.IsEof(); ++i)
  218. geometryCenters_.Push(source.ReadVector3());
  219. while (geometryCenters_.Size() < geometries_.Size())
  220. geometryCenters_.Push(Vector3::ZERO);
  221. memoryUse += sizeof(Vector3) * geometries_.Size();
  222. SetMemoryUse(memoryUse);
  223. return true;
  224. }
  225. bool Model::EndLoad()
  226. {
  227. // Upload vertex buffer data
  228. for (unsigned i = 0; i < vertexBuffers_.Size(); ++i)
  229. {
  230. VertexBuffer* buffer = vertexBuffers_[i];
  231. VertexBufferDesc& desc = loadVBData_[i];
  232. buffer->SetShadowed(true);
  233. buffer->SetSize(desc.vertexCount_, desc.elementMask_);
  234. buffer->SetData(desc.data_.Get());
  235. }
  236. // Upload index buffer data
  237. for (unsigned i = 0; i < indexBuffers_.Size(); ++i)
  238. {
  239. IndexBuffer* buffer = indexBuffers_[i];
  240. IndexBufferDesc& desc = loadIBData_[i];
  241. buffer->SetShadowed(true);
  242. buffer->SetSize(desc.indexCount_, desc.indexSize_ > sizeof(unsigned short));
  243. buffer->SetData(desc.data_.Get());
  244. }
  245. // Set up geometries
  246. for (unsigned i = 0; i < geometries_.Size(); ++i)
  247. {
  248. for (unsigned j = 0; j < geometries_[i].Size(); ++j)
  249. {
  250. Geometry* geometry = geometries_[i][j];
  251. GeometryDesc& desc = loadGeometries_[i][j];
  252. geometry->SetVertexBuffer(0, vertexBuffers_[desc.vbRef_]);
  253. geometry->SetIndexBuffer(indexBuffers_[desc.ibRef_]);
  254. geometry->SetDrawRange(desc.type_, desc.indexStart_, desc.indexCount_);
  255. }
  256. }
  257. loadVBData_.Clear();
  258. loadIBData_.Clear();
  259. loadGeometries_.Clear();
  260. return true;
  261. }
  262. bool Model::Save(Serializer& dest) const
  263. {
  264. // Write ID
  265. if (!dest.WriteFileID("UMDL"))
  266. return false;
  267. // Write vertex buffers
  268. dest.WriteUInt(vertexBuffers_.Size());
  269. for (unsigned i = 0; i < vertexBuffers_.Size(); ++i)
  270. {
  271. VertexBuffer* buffer = vertexBuffers_[i];
  272. dest.WriteUInt(buffer->GetVertexCount());
  273. dest.WriteUInt(buffer->GetElementMask());
  274. dest.WriteUInt(morphRangeStarts_[i]);
  275. dest.WriteUInt(morphRangeCounts_[i]);
  276. dest.Write(buffer->GetShadowData(), buffer->GetVertexCount() * buffer->GetVertexSize());
  277. }
  278. // Write index buffers
  279. dest.WriteUInt(indexBuffers_.Size());
  280. for (unsigned i = 0; i < indexBuffers_.Size(); ++i)
  281. {
  282. IndexBuffer* buffer = indexBuffers_[i];
  283. dest.WriteUInt(buffer->GetIndexCount());
  284. dest.WriteUInt(buffer->GetIndexSize());
  285. dest.Write(buffer->GetShadowData(), buffer->GetIndexCount() * buffer->GetIndexSize());
  286. }
  287. // Write geometries
  288. dest.WriteUInt(geometries_.Size());
  289. for (unsigned i = 0; i < geometries_.Size(); ++i)
  290. {
  291. // Write bone mappings
  292. dest.WriteUInt(geometryBoneMappings_[i].Size());
  293. for (unsigned j = 0; j < geometryBoneMappings_[i].Size(); ++j)
  294. dest.WriteUInt(geometryBoneMappings_[i][j]);
  295. // Write the LOD levels
  296. dest.WriteUInt(geometries_[i].Size());
  297. for (unsigned j = 0; j < geometries_[i].Size(); ++j)
  298. {
  299. Geometry* geometry = geometries_[i][j];
  300. dest.WriteFloat(geometry->GetLodDistance());
  301. dest.WriteUInt(geometry->GetPrimitiveType());
  302. dest.WriteUInt(LookupVertexBuffer(geometry->GetVertexBuffer(0), vertexBuffers_));
  303. dest.WriteUInt(LookupIndexBuffer(geometry->GetIndexBuffer(), indexBuffers_));
  304. dest.WriteUInt(geometry->GetIndexStart());
  305. dest.WriteUInt(geometry->GetIndexCount());
  306. }
  307. }
  308. // Write morphs
  309. dest.WriteUInt(morphs_.Size());
  310. for (unsigned i = 0; i < morphs_.Size(); ++i)
  311. {
  312. dest.WriteString(morphs_[i].name_);
  313. dest.WriteUInt(morphs_[i].buffers_.Size());
  314. // Write morph vertex buffers
  315. for (HashMap<unsigned int, VertexBufferMorph>::ConstIterator j = morphs_[i].buffers_.Begin();
  316. j != morphs_[i].buffers_.End(); ++j)
  317. {
  318. dest.WriteUInt(j->first_);
  319. dest.WriteUInt(j->second_.elementMask_);
  320. dest.WriteUInt(j->second_.vertexCount_);
  321. // Base size: size of each vertex index
  322. unsigned vertexSize = sizeof(unsigned);
  323. // Add size of individual elements
  324. if (j->second_.elementMask_ & MASK_POSITION)
  325. vertexSize += sizeof(Vector3);
  326. if (j->second_.elementMask_ & MASK_NORMAL)
  327. vertexSize += sizeof(Vector3);
  328. if (j->second_.elementMask_ & MASK_TANGENT)
  329. vertexSize += sizeof(Vector3);
  330. dest.Write(j->second_.morphData_.Get(), vertexSize * j->second_.vertexCount_);
  331. }
  332. }
  333. // Write skeleton
  334. skeleton_.Save(dest);
  335. // Write bounding box
  336. dest.WriteBoundingBox(boundingBox_);
  337. // Write geometry centers
  338. for (unsigned i = 0; i < geometryCenters_.Size(); ++i)
  339. dest.WriteVector3(geometryCenters_[i]);
  340. return true;
  341. }
  342. void Model::SetBoundingBox(const BoundingBox& box)
  343. {
  344. boundingBox_ = box;
  345. }
  346. bool Model::SetVertexBuffers(const Vector<SharedPtr<VertexBuffer> >& buffers, const PODVector<unsigned>& morphRangeStarts, const PODVector<unsigned>& morphRangeCounts)
  347. {
  348. for (unsigned i = 0; i < buffers.Size(); ++i)
  349. {
  350. if (!buffers[i])
  351. {
  352. LOGERROR("Null model vertex buffers specified");
  353. return false;
  354. }
  355. if (!buffers[i]->IsShadowed())
  356. {
  357. LOGERROR("Model vertex buffers must be shadowed");
  358. return false;
  359. }
  360. }
  361. vertexBuffers_ = buffers;
  362. morphRangeStarts_.Resize(buffers.Size());
  363. morphRangeCounts_.Resize(buffers.Size());
  364. // If morph ranges are not specified for buffers, assume to be zero
  365. for (unsigned i = 0; i < buffers.Size(); ++i)
  366. {
  367. morphRangeStarts_[i] = i < morphRangeStarts.Size() ? morphRangeStarts[i] : 0;
  368. morphRangeCounts_[i] = i < morphRangeCounts.Size() ? morphRangeCounts[i] : 0;
  369. }
  370. return true;
  371. }
  372. bool Model::SetIndexBuffers(const Vector<SharedPtr<IndexBuffer> >& buffers)
  373. {
  374. for (unsigned i = 0; i < buffers.Size(); ++i)
  375. {
  376. if (!buffers[i])
  377. {
  378. LOGERROR("Null model index buffers specified");
  379. return false;
  380. }
  381. if (!buffers[i]->IsShadowed())
  382. {
  383. LOGERROR("Model index buffers must be shadowed");
  384. return false;
  385. }
  386. }
  387. indexBuffers_ = buffers;
  388. return true;
  389. }
  390. void Model::SetNumGeometries(unsigned num)
  391. {
  392. geometries_.Resize(num);
  393. geometryBoneMappings_.Resize(num);
  394. geometryCenters_.Resize(num);
  395. }
  396. bool Model::SetNumGeometryLodLevels(unsigned index, unsigned num)
  397. {
  398. if (index >= geometries_.Size())
  399. {
  400. LOGERROR("Geometry index out of bounds");
  401. return false;
  402. }
  403. if (!num)
  404. {
  405. LOGERROR("Zero LOD levels not allowed");
  406. return false;
  407. }
  408. geometries_[index].Resize(num);
  409. return true;
  410. }
  411. bool Model::SetGeometry(unsigned index, unsigned lodLevel, Geometry* geometry)
  412. {
  413. if (index >= geometries_.Size())
  414. {
  415. LOGERROR("Geometry index out of bounds");
  416. return false;
  417. }
  418. if (lodLevel >= geometries_[index].Size())
  419. {
  420. LOGERROR("LOD level index out of bounds");
  421. return false;
  422. }
  423. geometries_[index][lodLevel] = geometry;
  424. return true;
  425. }
  426. bool Model::SetGeometryCenter(unsigned index, const Vector3& center)
  427. {
  428. if (index >= geometryCenters_.Size())
  429. {
  430. LOGERROR("Geometry index out of bounds");
  431. return false;
  432. }
  433. geometryCenters_[index] = center;
  434. return true;
  435. }
  436. void Model::SetSkeleton(const Skeleton& skeleton)
  437. {
  438. skeleton_ = skeleton;
  439. }
  440. void Model::SetGeometryBoneMappings(const Vector<PODVector<unsigned> >& geometryBoneMappings)
  441. {
  442. geometryBoneMappings_ = geometryBoneMappings;
  443. }
  444. void Model::SetMorphs(const Vector<ModelMorph>& morphs)
  445. {
  446. morphs_ = morphs;
  447. }
  448. SharedPtr<Model> Model::Clone(const String& cloneName) const
  449. {
  450. SharedPtr<Model> ret(new Model(context_));
  451. ret->SetName(cloneName);
  452. ret->boundingBox_ = boundingBox_;
  453. ret->skeleton_ = skeleton_;
  454. ret->geometryBoneMappings_ = geometryBoneMappings_;
  455. ret->geometryCenters_ = geometryCenters_;
  456. ret->morphs_ = morphs_;
  457. ret->morphRangeStarts_ = morphRangeStarts_;
  458. ret->morphRangeCounts_ = morphRangeCounts_;
  459. // Deep copy vertex/index buffers
  460. HashMap<VertexBuffer*, VertexBuffer*> vbMapping;
  461. for (Vector<SharedPtr<VertexBuffer> >::ConstIterator i = vertexBuffers_.Begin(); i != vertexBuffers_.End(); ++i)
  462. {
  463. VertexBuffer* origBuffer = *i;
  464. SharedPtr<VertexBuffer> cloneBuffer;
  465. if (origBuffer)
  466. {
  467. cloneBuffer = new VertexBuffer(context_);
  468. cloneBuffer->SetSize(origBuffer->GetVertexCount(), origBuffer->GetElementMask(), origBuffer->IsDynamic());
  469. cloneBuffer->SetShadowed(origBuffer->IsShadowed());
  470. if (origBuffer->IsShadowed())
  471. cloneBuffer->SetData(origBuffer->GetShadowData());
  472. else
  473. {
  474. void* origData = origBuffer->Lock(0, origBuffer->GetVertexCount());
  475. if (origData)
  476. cloneBuffer->SetData(origData);
  477. else
  478. LOGERROR("Failed to lock original vertex buffer for copying");
  479. }
  480. vbMapping[origBuffer] = cloneBuffer;
  481. }
  482. ret->vertexBuffers_.Push(cloneBuffer);
  483. }
  484. HashMap<IndexBuffer*, IndexBuffer*> ibMapping;
  485. for (Vector<SharedPtr<IndexBuffer> >::ConstIterator i = indexBuffers_.Begin(); i != indexBuffers_.End(); ++i)
  486. {
  487. IndexBuffer* origBuffer = *i;
  488. SharedPtr<IndexBuffer> cloneBuffer;
  489. if (origBuffer)
  490. {
  491. cloneBuffer = new IndexBuffer(context_);
  492. cloneBuffer->SetSize(origBuffer->GetIndexCount(), origBuffer->GetIndexSize() == sizeof(unsigned), origBuffer->IsDynamic());
  493. cloneBuffer->SetShadowed(origBuffer->IsShadowed());
  494. if (origBuffer->IsShadowed())
  495. cloneBuffer->SetData(origBuffer->GetShadowData());
  496. else
  497. {
  498. void* origData = origBuffer->Lock(0, origBuffer->GetIndexCount());
  499. if (origData)
  500. cloneBuffer->SetData(origData);
  501. else
  502. LOGERROR("Failed to lock original index buffer for copying");
  503. }
  504. ibMapping[origBuffer] = cloneBuffer;
  505. }
  506. ret->indexBuffers_.Push(cloneBuffer);
  507. }
  508. // Deep copy all the geometry LOD levels and refer to the copied vertex/index buffers
  509. ret->geometries_.Resize(geometries_.Size());
  510. for (unsigned i = 0; i < geometries_.Size(); ++i)
  511. {
  512. ret->geometries_[i].Resize(geometries_[i].Size());
  513. for (unsigned j = 0; j < geometries_[i].Size(); ++j)
  514. {
  515. SharedPtr<Geometry> cloneGeometry;
  516. Geometry* origGeometry = geometries_[i][j];
  517. if (origGeometry)
  518. {
  519. cloneGeometry = new Geometry(context_);
  520. cloneGeometry->SetIndexBuffer(ibMapping[origGeometry->GetIndexBuffer()]);
  521. unsigned numVbs = origGeometry->GetNumVertexBuffers();
  522. for (unsigned k = 0; k < numVbs; ++k)
  523. {
  524. cloneGeometry->SetVertexBuffer(k, vbMapping[origGeometry->GetVertexBuffer(k)],
  525. origGeometry->GetVertexElementMask(k));
  526. }
  527. cloneGeometry->SetDrawRange(origGeometry->GetPrimitiveType(), origGeometry->GetIndexStart(),
  528. origGeometry->GetIndexCount(), origGeometry->GetVertexStart(), origGeometry->GetVertexCount(), false);
  529. cloneGeometry->SetLodDistance(origGeometry->GetLodDistance());
  530. }
  531. ret->geometries_[i][j] = cloneGeometry;
  532. }
  533. }
  534. // Deep copy the morph data (if any) to allow modifying it
  535. for (Vector<ModelMorph>::Iterator i = ret->morphs_.Begin(); i != ret->morphs_.End(); ++i)
  536. {
  537. ModelMorph& morph = *i;
  538. for (HashMap<unsigned, VertexBufferMorph>::Iterator j = morph.buffers_.Begin(); j != morph.buffers_.End(); ++j)
  539. {
  540. VertexBufferMorph& vbMorph = j->second_;
  541. if (vbMorph.dataSize_)
  542. {
  543. SharedArrayPtr<unsigned char> cloneData(new unsigned char[vbMorph.dataSize_]);
  544. memcpy(cloneData.Get(), vbMorph.morphData_.Get(), vbMorph.dataSize_);
  545. vbMorph.morphData_ = cloneData;
  546. }
  547. }
  548. }
  549. ret->SetMemoryUse(GetMemoryUse());
  550. return ret;
  551. }
  552. unsigned Model::GetNumGeometryLodLevels(unsigned index) const
  553. {
  554. return index < geometries_.Size() ? geometries_[index].Size() : 0;
  555. }
  556. Geometry* Model::GetGeometry(unsigned index, unsigned lodLevel) const
  557. {
  558. if (index >= geometries_.Size() || geometries_[index].Empty())
  559. return 0;
  560. if (lodLevel >= geometries_[index].Size())
  561. lodLevel = geometries_[index].Size() - 1;
  562. return geometries_[index][lodLevel];
  563. }
  564. const ModelMorph* Model::GetMorph(unsigned index) const
  565. {
  566. return index < morphs_.Size() ? &morphs_[index] : 0;
  567. }
  568. const ModelMorph* Model::GetMorph(const String& name) const
  569. {
  570. return GetMorph(StringHash(name));
  571. }
  572. const ModelMorph* Model::GetMorph(StringHash nameHash) const
  573. {
  574. for (Vector<ModelMorph>::ConstIterator i = morphs_.Begin(); i != morphs_.End(); ++i)
  575. {
  576. if (i->nameHash_ == nameHash)
  577. return &(*i);
  578. }
  579. return 0;
  580. }
  581. unsigned Model::GetMorphRangeStart(unsigned bufferIndex) const
  582. {
  583. return bufferIndex < vertexBuffers_.Size() ? morphRangeStarts_[bufferIndex] : 0;
  584. }
  585. unsigned Model::GetMorphRangeCount(unsigned bufferIndex) const
  586. {
  587. return bufferIndex < vertexBuffers_.Size() ? morphRangeCounts_[bufferIndex] : 0;
  588. }
  589. }