Model.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728
  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. bool async = GetAsyncLoadState() == ASYNC_LOADING;
  82. // Read vertex buffers
  83. unsigned numVertexBuffers = source.ReadUInt();
  84. vertexBuffers_.Reserve(numVertexBuffers);
  85. morphRangeStarts_.Resize(numVertexBuffers);
  86. morphRangeCounts_.Resize(numVertexBuffers);
  87. loadVBData_.Resize(numVertexBuffers);
  88. for (unsigned i = 0; i < numVertexBuffers; ++i)
  89. {
  90. unsigned vertexCount = source.ReadUInt();
  91. unsigned elementMask = source.ReadUInt();
  92. morphRangeStarts_[i] = source.ReadUInt();
  93. morphRangeCounts_[i] = source.ReadUInt();
  94. SharedPtr<VertexBuffer> buffer(new VertexBuffer(context_));
  95. unsigned vertexSize = VertexBuffer::GetVertexSize(elementMask);
  96. // Prepare vertex buffer data to be uploaded during EndLoad()
  97. if (async)
  98. {
  99. loadVBData_[i].vertexCount_ = vertexCount;
  100. loadVBData_[i].elementMask_ = elementMask;
  101. loadVBData_[i].dataSize_ = vertexCount * vertexSize;
  102. loadVBData_[i].data_ = new unsigned char[loadVBData_[i].dataSize_];
  103. source.Read(loadVBData_[i].data_.Get(), loadVBData_[i].dataSize_);
  104. }
  105. else
  106. {
  107. // If not async loading, use locking to avoid extra allocation & copy
  108. loadVBData_[i].data_.Reset(); // Make sure no previous data
  109. buffer->SetShadowed(true);
  110. buffer->SetSize(vertexCount, elementMask);
  111. void* dest = buffer->Lock(0, vertexCount);
  112. source.Read(dest, vertexCount * vertexSize);
  113. buffer->Unlock();
  114. }
  115. memoryUse += sizeof(VertexBuffer) + vertexCount * vertexSize;
  116. vertexBuffers_.Push(buffer);
  117. }
  118. // Read index buffers
  119. unsigned numIndexBuffers = source.ReadUInt();
  120. indexBuffers_.Reserve(numIndexBuffers);
  121. loadIBData_.Resize(numIndexBuffers);
  122. for (unsigned i = 0; i < numIndexBuffers; ++i)
  123. {
  124. unsigned indexCount = source.ReadUInt();
  125. unsigned indexSize = source.ReadUInt();
  126. SharedPtr<IndexBuffer> buffer(new IndexBuffer(context_));
  127. // Prepare index buffer data to be uploaded during EndLoad()
  128. if (async)
  129. {
  130. loadIBData_[i].indexCount_ = indexCount;
  131. loadIBData_[i].indexSize_ = indexSize;
  132. loadIBData_[i].dataSize_ = indexCount * indexSize;
  133. loadIBData_[i].data_ = new unsigned char[loadIBData_[i].dataSize_];
  134. source.Read(loadIBData_[i].data_.Get(), loadIBData_[i].dataSize_);
  135. }
  136. else
  137. {
  138. // If not async loading, use locking to avoid extra allocation & copy
  139. loadIBData_[i].data_.Reset(); // Make sure no previous data
  140. buffer->SetShadowed(true);
  141. buffer->SetSize(indexCount, indexSize > sizeof(unsigned short));
  142. void* dest = buffer->Lock(0, indexCount);
  143. source.Read(dest, indexCount * indexSize);
  144. buffer->Unlock();
  145. }
  146. memoryUse += sizeof(IndexBuffer) + indexCount * indexSize;
  147. indexBuffers_.Push(buffer);
  148. }
  149. // Read geometries
  150. unsigned numGeometries = source.ReadUInt();
  151. geometries_.Reserve(numGeometries);
  152. geometryBoneMappings_.Reserve(numGeometries);
  153. geometryCenters_.Reserve(numGeometries);
  154. loadGeometries_.Resize(numGeometries);
  155. for (unsigned i = 0; i < numGeometries; ++i)
  156. {
  157. // Read bone mappings
  158. unsigned boneMappingCount = source.ReadUInt();
  159. PODVector<unsigned> boneMapping(boneMappingCount);
  160. for (unsigned j = 0; j < boneMappingCount; ++j)
  161. boneMapping[j] = source.ReadUInt();
  162. geometryBoneMappings_.Push(boneMapping);
  163. unsigned numLodLevels = source.ReadUInt();
  164. Vector<SharedPtr<Geometry> > geometryLodLevels;
  165. geometryLodLevels.Reserve(numLodLevels);
  166. loadGeometries_[i].Resize(numLodLevels);
  167. for (unsigned j = 0; j < numLodLevels; ++j)
  168. {
  169. float distance = source.ReadFloat();
  170. PrimitiveType type = (PrimitiveType)source.ReadUInt();
  171. unsigned vbRef = source.ReadUInt();
  172. unsigned ibRef = source.ReadUInt();
  173. unsigned indexStart = source.ReadUInt();
  174. unsigned indexCount = source.ReadUInt();
  175. if (vbRef >= vertexBuffers_.Size())
  176. {
  177. LOGERROR("Vertex buffer index out of bounds");
  178. loadVBData_.Clear();
  179. loadIBData_.Clear();
  180. loadGeometries_.Clear();
  181. return false;
  182. }
  183. if (ibRef >= indexBuffers_.Size())
  184. {
  185. LOGERROR("Index buffer index out of bounds");
  186. loadVBData_.Clear();
  187. loadIBData_.Clear();
  188. loadGeometries_.Clear();
  189. return false;
  190. }
  191. SharedPtr<Geometry> geometry(new Geometry(context_));
  192. geometry->SetLodDistance(distance);
  193. // Prepare geometry to be defined during EndLoad()
  194. loadGeometries_[i][j].type_ = type;
  195. loadGeometries_[i][j].vbRef_ = vbRef;
  196. loadGeometries_[i][j].ibRef_ = ibRef;
  197. loadGeometries_[i][j].indexStart_ = indexStart;
  198. loadGeometries_[i][j].indexCount_ = indexCount;
  199. geometryLodLevels.Push(geometry);
  200. memoryUse += sizeof(Geometry);
  201. }
  202. geometries_.Push(geometryLodLevels);
  203. }
  204. // Read morphs
  205. unsigned numMorphs = source.ReadUInt();
  206. morphs_.Reserve(numMorphs);
  207. for (unsigned i = 0; i < numMorphs; ++i)
  208. {
  209. ModelMorph newMorph;
  210. newMorph.name_ = source.ReadString();
  211. newMorph.nameHash_ = newMorph.name_;
  212. newMorph.weight_ = 0.0f;
  213. unsigned numBuffers = source.ReadUInt();
  214. for (unsigned j = 0; j < numBuffers; ++j)
  215. {
  216. VertexBufferMorph newBuffer;
  217. unsigned bufferIndex = source.ReadUInt();
  218. newBuffer.elementMask_ = source.ReadUInt();
  219. newBuffer.vertexCount_ = source.ReadUInt();
  220. // Base size: size of each vertex index
  221. unsigned vertexSize = sizeof(unsigned);
  222. // Add size of individual elements
  223. if (newBuffer.elementMask_ & MASK_POSITION)
  224. vertexSize += sizeof(Vector3);
  225. if (newBuffer.elementMask_ & MASK_NORMAL)
  226. vertexSize += sizeof(Vector3);
  227. if (newBuffer.elementMask_ & MASK_TANGENT)
  228. vertexSize += sizeof(Vector3);
  229. newBuffer.dataSize_ = newBuffer.vertexCount_ * vertexSize;
  230. newBuffer.morphData_ = new unsigned char[newBuffer.dataSize_];
  231. source.Read(&newBuffer.morphData_[0], newBuffer.vertexCount_ * vertexSize);
  232. newMorph.buffers_[bufferIndex] = newBuffer;
  233. memoryUse += sizeof(VertexBufferMorph) + newBuffer.vertexCount_ * vertexSize;
  234. }
  235. morphs_.Push(newMorph);
  236. memoryUse += sizeof(ModelMorph);
  237. }
  238. // Read skeleton
  239. skeleton_.Load(source);
  240. memoryUse += skeleton_.GetNumBones() * sizeof(Bone);
  241. // Read bounding box
  242. boundingBox_ = source.ReadBoundingBox();
  243. // Read geometry centers
  244. for (unsigned i = 0; i < geometries_.Size() && !source.IsEof(); ++i)
  245. geometryCenters_.Push(source.ReadVector3());
  246. while (geometryCenters_.Size() < geometries_.Size())
  247. geometryCenters_.Push(Vector3::ZERO);
  248. memoryUse += sizeof(Vector3) * geometries_.Size();
  249. SetMemoryUse(memoryUse);
  250. return true;
  251. }
  252. bool Model::EndLoad()
  253. {
  254. // Upload vertex buffer data
  255. for (unsigned i = 0; i < vertexBuffers_.Size(); ++i)
  256. {
  257. VertexBuffer* buffer = vertexBuffers_[i];
  258. VertexBufferDesc& desc = loadVBData_[i];
  259. if (desc.data_)
  260. {
  261. buffer->SetShadowed(true);
  262. buffer->SetSize(desc.vertexCount_, desc.elementMask_);
  263. buffer->SetData(desc.data_.Get());
  264. }
  265. }
  266. // Upload index buffer data
  267. for (unsigned i = 0; i < indexBuffers_.Size(); ++i)
  268. {
  269. IndexBuffer* buffer = indexBuffers_[i];
  270. IndexBufferDesc& desc = loadIBData_[i];
  271. if (desc.data_)
  272. {
  273. buffer->SetShadowed(true);
  274. buffer->SetSize(desc.indexCount_, desc.indexSize_ > sizeof(unsigned short));
  275. buffer->SetData(desc.data_.Get());
  276. }
  277. }
  278. // Set up geometries
  279. for (unsigned i = 0; i < geometries_.Size(); ++i)
  280. {
  281. for (unsigned j = 0; j < geometries_[i].Size(); ++j)
  282. {
  283. Geometry* geometry = geometries_[i][j];
  284. GeometryDesc& desc = loadGeometries_[i][j];
  285. geometry->SetVertexBuffer(0, vertexBuffers_[desc.vbRef_]);
  286. geometry->SetIndexBuffer(indexBuffers_[desc.ibRef_]);
  287. geometry->SetDrawRange(desc.type_, desc.indexStart_, desc.indexCount_);
  288. }
  289. }
  290. loadVBData_.Clear();
  291. loadIBData_.Clear();
  292. loadGeometries_.Clear();
  293. return true;
  294. }
  295. bool Model::Save(Serializer& dest) const
  296. {
  297. // Write ID
  298. if (!dest.WriteFileID("UMDL"))
  299. return false;
  300. // Write vertex buffers
  301. dest.WriteUInt(vertexBuffers_.Size());
  302. for (unsigned i = 0; i < vertexBuffers_.Size(); ++i)
  303. {
  304. VertexBuffer* buffer = vertexBuffers_[i];
  305. dest.WriteUInt(buffer->GetVertexCount());
  306. dest.WriteUInt(buffer->GetElementMask());
  307. dest.WriteUInt(morphRangeStarts_[i]);
  308. dest.WriteUInt(morphRangeCounts_[i]);
  309. dest.Write(buffer->GetShadowData(), buffer->GetVertexCount() * buffer->GetVertexSize());
  310. }
  311. // Write index buffers
  312. dest.WriteUInt(indexBuffers_.Size());
  313. for (unsigned i = 0; i < indexBuffers_.Size(); ++i)
  314. {
  315. IndexBuffer* buffer = indexBuffers_[i];
  316. dest.WriteUInt(buffer->GetIndexCount());
  317. dest.WriteUInt(buffer->GetIndexSize());
  318. dest.Write(buffer->GetShadowData(), buffer->GetIndexCount() * buffer->GetIndexSize());
  319. }
  320. // Write geometries
  321. dest.WriteUInt(geometries_.Size());
  322. for (unsigned i = 0; i < geometries_.Size(); ++i)
  323. {
  324. // Write bone mappings
  325. dest.WriteUInt(geometryBoneMappings_[i].Size());
  326. for (unsigned j = 0; j < geometryBoneMappings_[i].Size(); ++j)
  327. dest.WriteUInt(geometryBoneMappings_[i][j]);
  328. // Write the LOD levels
  329. dest.WriteUInt(geometries_[i].Size());
  330. for (unsigned j = 0; j < geometries_[i].Size(); ++j)
  331. {
  332. Geometry* geometry = geometries_[i][j];
  333. dest.WriteFloat(geometry->GetLodDistance());
  334. dest.WriteUInt(geometry->GetPrimitiveType());
  335. dest.WriteUInt(LookupVertexBuffer(geometry->GetVertexBuffer(0), vertexBuffers_));
  336. dest.WriteUInt(LookupIndexBuffer(geometry->GetIndexBuffer(), indexBuffers_));
  337. dest.WriteUInt(geometry->GetIndexStart());
  338. dest.WriteUInt(geometry->GetIndexCount());
  339. }
  340. }
  341. // Write morphs
  342. dest.WriteUInt(morphs_.Size());
  343. for (unsigned i = 0; i < morphs_.Size(); ++i)
  344. {
  345. dest.WriteString(morphs_[i].name_);
  346. dest.WriteUInt(morphs_[i].buffers_.Size());
  347. // Write morph vertex buffers
  348. for (HashMap<unsigned int, VertexBufferMorph>::ConstIterator j = morphs_[i].buffers_.Begin();
  349. j != morphs_[i].buffers_.End(); ++j)
  350. {
  351. dest.WriteUInt(j->first_);
  352. dest.WriteUInt(j->second_.elementMask_);
  353. dest.WriteUInt(j->second_.vertexCount_);
  354. // Base size: size of each vertex index
  355. unsigned vertexSize = sizeof(unsigned);
  356. // Add size of individual elements
  357. if (j->second_.elementMask_ & MASK_POSITION)
  358. vertexSize += sizeof(Vector3);
  359. if (j->second_.elementMask_ & MASK_NORMAL)
  360. vertexSize += sizeof(Vector3);
  361. if (j->second_.elementMask_ & MASK_TANGENT)
  362. vertexSize += sizeof(Vector3);
  363. dest.Write(j->second_.morphData_.Get(), vertexSize * j->second_.vertexCount_);
  364. }
  365. }
  366. // Write skeleton
  367. skeleton_.Save(dest);
  368. // Write bounding box
  369. dest.WriteBoundingBox(boundingBox_);
  370. // Write geometry centers
  371. for (unsigned i = 0; i < geometryCenters_.Size(); ++i)
  372. dest.WriteVector3(geometryCenters_[i]);
  373. return true;
  374. }
  375. void Model::SetBoundingBox(const BoundingBox& box)
  376. {
  377. boundingBox_ = box;
  378. }
  379. bool Model::SetVertexBuffers(const Vector<SharedPtr<VertexBuffer> >& buffers, const PODVector<unsigned>& morphRangeStarts, const PODVector<unsigned>& morphRangeCounts)
  380. {
  381. for (unsigned i = 0; i < buffers.Size(); ++i)
  382. {
  383. if (!buffers[i])
  384. {
  385. LOGERROR("Null model vertex buffers specified");
  386. return false;
  387. }
  388. if (!buffers[i]->IsShadowed())
  389. {
  390. LOGERROR("Model vertex buffers must be shadowed");
  391. return false;
  392. }
  393. }
  394. vertexBuffers_ = buffers;
  395. morphRangeStarts_.Resize(buffers.Size());
  396. morphRangeCounts_.Resize(buffers.Size());
  397. // If morph ranges are not specified for buffers, assume to be zero
  398. for (unsigned i = 0; i < buffers.Size(); ++i)
  399. {
  400. morphRangeStarts_[i] = i < morphRangeStarts.Size() ? morphRangeStarts[i] : 0;
  401. morphRangeCounts_[i] = i < morphRangeCounts.Size() ? morphRangeCounts[i] : 0;
  402. }
  403. return true;
  404. }
  405. bool Model::SetIndexBuffers(const Vector<SharedPtr<IndexBuffer> >& buffers)
  406. {
  407. for (unsigned i = 0; i < buffers.Size(); ++i)
  408. {
  409. if (!buffers[i])
  410. {
  411. LOGERROR("Null model index buffers specified");
  412. return false;
  413. }
  414. if (!buffers[i]->IsShadowed())
  415. {
  416. LOGERROR("Model index buffers must be shadowed");
  417. return false;
  418. }
  419. }
  420. indexBuffers_ = buffers;
  421. return true;
  422. }
  423. void Model::SetNumGeometries(unsigned num)
  424. {
  425. geometries_.Resize(num);
  426. geometryBoneMappings_.Resize(num);
  427. geometryCenters_.Resize(num);
  428. // For easier creation of from-scratch geometry, ensure that all geometries start with at least 1 LOD level (0 makes no sense)
  429. for (unsigned i = 0; i < geometries_.Size(); ++i)
  430. {
  431. if (geometries_[i].Empty())
  432. geometries_[i].Resize(1);
  433. }
  434. }
  435. bool Model::SetNumGeometryLodLevels(unsigned index, unsigned num)
  436. {
  437. if (index >= geometries_.Size())
  438. {
  439. LOGERROR("Geometry index out of bounds");
  440. return false;
  441. }
  442. if (!num)
  443. {
  444. LOGERROR("Zero LOD levels not allowed");
  445. return false;
  446. }
  447. geometries_[index].Resize(num);
  448. return true;
  449. }
  450. bool Model::SetGeometry(unsigned index, unsigned lodLevel, Geometry* geometry)
  451. {
  452. if (index >= geometries_.Size())
  453. {
  454. LOGERROR("Geometry index out of bounds");
  455. return false;
  456. }
  457. if (lodLevel >= geometries_[index].Size())
  458. {
  459. LOGERROR("LOD level index out of bounds");
  460. return false;
  461. }
  462. geometries_[index][lodLevel] = geometry;
  463. return true;
  464. }
  465. bool Model::SetGeometryCenter(unsigned index, const Vector3& center)
  466. {
  467. if (index >= geometryCenters_.Size())
  468. {
  469. LOGERROR("Geometry index out of bounds");
  470. return false;
  471. }
  472. geometryCenters_[index] = center;
  473. return true;
  474. }
  475. void Model::SetSkeleton(const Skeleton& skeleton)
  476. {
  477. skeleton_ = skeleton;
  478. }
  479. void Model::SetGeometryBoneMappings(const Vector<PODVector<unsigned> >& geometryBoneMappings)
  480. {
  481. geometryBoneMappings_ = geometryBoneMappings;
  482. }
  483. void Model::SetMorphs(const Vector<ModelMorph>& morphs)
  484. {
  485. morphs_ = morphs;
  486. }
  487. SharedPtr<Model> Model::Clone(const String& cloneName) const
  488. {
  489. SharedPtr<Model> ret(new Model(context_));
  490. ret->SetName(cloneName);
  491. ret->boundingBox_ = boundingBox_;
  492. ret->skeleton_ = skeleton_;
  493. ret->geometryBoneMappings_ = geometryBoneMappings_;
  494. ret->geometryCenters_ = geometryCenters_;
  495. ret->morphs_ = morphs_;
  496. ret->morphRangeStarts_ = morphRangeStarts_;
  497. ret->morphRangeCounts_ = morphRangeCounts_;
  498. // Deep copy vertex/index buffers
  499. HashMap<VertexBuffer*, VertexBuffer*> vbMapping;
  500. for (Vector<SharedPtr<VertexBuffer> >::ConstIterator i = vertexBuffers_.Begin(); i != vertexBuffers_.End(); ++i)
  501. {
  502. VertexBuffer* origBuffer = *i;
  503. SharedPtr<VertexBuffer> cloneBuffer;
  504. if (origBuffer)
  505. {
  506. cloneBuffer = new VertexBuffer(context_);
  507. cloneBuffer->SetSize(origBuffer->GetVertexCount(), origBuffer->GetElementMask(), origBuffer->IsDynamic());
  508. cloneBuffer->SetShadowed(origBuffer->IsShadowed());
  509. if (origBuffer->IsShadowed())
  510. cloneBuffer->SetData(origBuffer->GetShadowData());
  511. else
  512. {
  513. void* origData = origBuffer->Lock(0, origBuffer->GetVertexCount());
  514. if (origData)
  515. cloneBuffer->SetData(origData);
  516. else
  517. LOGERROR("Failed to lock original vertex buffer for copying");
  518. }
  519. vbMapping[origBuffer] = cloneBuffer;
  520. }
  521. ret->vertexBuffers_.Push(cloneBuffer);
  522. }
  523. HashMap<IndexBuffer*, IndexBuffer*> ibMapping;
  524. for (Vector<SharedPtr<IndexBuffer> >::ConstIterator i = indexBuffers_.Begin(); i != indexBuffers_.End(); ++i)
  525. {
  526. IndexBuffer* origBuffer = *i;
  527. SharedPtr<IndexBuffer> cloneBuffer;
  528. if (origBuffer)
  529. {
  530. cloneBuffer = new IndexBuffer(context_);
  531. cloneBuffer->SetSize(origBuffer->GetIndexCount(), origBuffer->GetIndexSize() == sizeof(unsigned), origBuffer->IsDynamic());
  532. cloneBuffer->SetShadowed(origBuffer->IsShadowed());
  533. if (origBuffer->IsShadowed())
  534. cloneBuffer->SetData(origBuffer->GetShadowData());
  535. else
  536. {
  537. void* origData = origBuffer->Lock(0, origBuffer->GetIndexCount());
  538. if (origData)
  539. cloneBuffer->SetData(origData);
  540. else
  541. LOGERROR("Failed to lock original index buffer for copying");
  542. }
  543. ibMapping[origBuffer] = cloneBuffer;
  544. }
  545. ret->indexBuffers_.Push(cloneBuffer);
  546. }
  547. // Deep copy all the geometry LOD levels and refer to the copied vertex/index buffers
  548. ret->geometries_.Resize(geometries_.Size());
  549. for (unsigned i = 0; i < geometries_.Size(); ++i)
  550. {
  551. ret->geometries_[i].Resize(geometries_[i].Size());
  552. for (unsigned j = 0; j < geometries_[i].Size(); ++j)
  553. {
  554. SharedPtr<Geometry> cloneGeometry;
  555. Geometry* origGeometry = geometries_[i][j];
  556. if (origGeometry)
  557. {
  558. cloneGeometry = new Geometry(context_);
  559. cloneGeometry->SetIndexBuffer(ibMapping[origGeometry->GetIndexBuffer()]);
  560. unsigned numVbs = origGeometry->GetNumVertexBuffers();
  561. for (unsigned k = 0; k < numVbs; ++k)
  562. {
  563. cloneGeometry->SetVertexBuffer(k, vbMapping[origGeometry->GetVertexBuffer(k)],
  564. origGeometry->GetVertexElementMask(k));
  565. }
  566. cloneGeometry->SetDrawRange(origGeometry->GetPrimitiveType(), origGeometry->GetIndexStart(),
  567. origGeometry->GetIndexCount(), origGeometry->GetVertexStart(), origGeometry->GetVertexCount(), false);
  568. cloneGeometry->SetLodDistance(origGeometry->GetLodDistance());
  569. }
  570. ret->geometries_[i][j] = cloneGeometry;
  571. }
  572. }
  573. // Deep copy the morph data (if any) to allow modifying it
  574. for (Vector<ModelMorph>::Iterator i = ret->morphs_.Begin(); i != ret->morphs_.End(); ++i)
  575. {
  576. ModelMorph& morph = *i;
  577. for (HashMap<unsigned, VertexBufferMorph>::Iterator j = morph.buffers_.Begin(); j != morph.buffers_.End(); ++j)
  578. {
  579. VertexBufferMorph& vbMorph = j->second_;
  580. if (vbMorph.dataSize_)
  581. {
  582. SharedArrayPtr<unsigned char> cloneData(new unsigned char[vbMorph.dataSize_]);
  583. memcpy(cloneData.Get(), vbMorph.morphData_.Get(), vbMorph.dataSize_);
  584. vbMorph.morphData_ = cloneData;
  585. }
  586. }
  587. }
  588. ret->SetMemoryUse(GetMemoryUse());
  589. return ret;
  590. }
  591. unsigned Model::GetNumGeometryLodLevels(unsigned index) const
  592. {
  593. return index < geometries_.Size() ? geometries_[index].Size() : 0;
  594. }
  595. Geometry* Model::GetGeometry(unsigned index, unsigned lodLevel) const
  596. {
  597. if (index >= geometries_.Size() || geometries_[index].Empty())
  598. return 0;
  599. if (lodLevel >= geometries_[index].Size())
  600. lodLevel = geometries_[index].Size() - 1;
  601. return geometries_[index][lodLevel];
  602. }
  603. const ModelMorph* Model::GetMorph(unsigned index) const
  604. {
  605. return index < morphs_.Size() ? &morphs_[index] : 0;
  606. }
  607. const ModelMorph* Model::GetMorph(const String& name) const
  608. {
  609. return GetMorph(StringHash(name));
  610. }
  611. const ModelMorph* Model::GetMorph(StringHash nameHash) const
  612. {
  613. for (Vector<ModelMorph>::ConstIterator i = morphs_.Begin(); i != morphs_.End(); ++i)
  614. {
  615. if (i->nameHash_ == nameHash)
  616. return &(*i);
  617. }
  618. return 0;
  619. }
  620. unsigned Model::GetMorphRangeStart(unsigned bufferIndex) const
  621. {
  622. return bufferIndex < vertexBuffers_.Size() ? morphRangeStarts_[bufferIndex] : 0;
  623. }
  624. unsigned Model::GetMorphRangeCount(unsigned bufferIndex) const
  625. {
  626. return bufferIndex < vertexBuffers_.Size() ? morphRangeCounts_[bufferIndex] : 0;
  627. }
  628. }