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