Model.cpp 25 KB

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