2
0

Model.cpp 25 KB

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