mirror of
				https://github.com/OpenMW/openmw.git
				synced 2025-10-30 23:56:38 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			162 lines
		
	
	
	
		
			6.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			162 lines
		
	
	
	
		
			6.2 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| #include "foreachbulletobject.hpp"
 | |
| 
 | |
| #include <components/debug/debuglog.hpp>
 | |
| #include <components/esm3/cellref.hpp>
 | |
| #include <components/esm3/esmreader.hpp>
 | |
| #include <components/esm3/loadcell.hpp>
 | |
| #include <components/esm3/loadland.hpp>
 | |
| #include <components/esmloader/esmdata.hpp>
 | |
| #include <components/esmloader/lessbyid.hpp>
 | |
| #include <components/esmloader/record.hpp>
 | |
| #include <components/misc/resourcehelpers.hpp>
 | |
| #include <components/misc/strings/lower.hpp>
 | |
| #include <components/resource/bulletshapemanager.hpp>
 | |
| #include <components/settings/settings.hpp>
 | |
| #include <components/vfs/manager.hpp>
 | |
| #include <components/esm3/readerscache.hpp>
 | |
| 
 | |
| #include <osg/ref_ptr>
 | |
| 
 | |
| #include <algorithm>
 | |
| #include <memory>
 | |
| #include <stdexcept>
 | |
| #include <string>
 | |
| #include <string_view>
 | |
| #include <utility>
 | |
| #include <vector>
 | |
| 
 | |
| namespace Resource
 | |
| {
 | |
|     namespace
 | |
|     {
 | |
|         struct CellRef
 | |
|         {
 | |
|             ESM::RecNameInts mType;
 | |
|             ESM::RefNum mRefNum;
 | |
|             std::string mRefId;
 | |
|             float mScale;
 | |
|             ESM::Position mPos;
 | |
| 
 | |
|             CellRef(ESM::RecNameInts type, ESM::RefNum refNum, std::string&& refId, float scale, const ESM::Position& pos)
 | |
|                 : mType(type), mRefNum(refNum), mRefId(std::move(refId)), mScale(scale), mPos(pos) {}
 | |
|         };
 | |
| 
 | |
|         ESM::RecNameInts getType(const EsmLoader::EsmData& esmData, std::string_view refId)
 | |
|         {
 | |
|             const auto it = std::lower_bound(esmData.mRefIdTypes.begin(), esmData.mRefIdTypes.end(),
 | |
|                                              refId, EsmLoader::LessById {});
 | |
|             if (it == esmData.mRefIdTypes.end() || it->mId != refId)
 | |
|                 return {};
 | |
|             return it->mType;
 | |
|         }
 | |
| 
 | |
|         std::vector<CellRef> loadCellRefs(const ESM::Cell& cell, const EsmLoader::EsmData& esmData,
 | |
|             ESM::ReadersCache& readers)
 | |
|         {
 | |
|             std::vector<EsmLoader::Record<CellRef>> cellRefs;
 | |
| 
 | |
|             for (std::size_t i = 0; i < cell.mContextList.size(); i++)
 | |
|             {
 | |
|                 const ESM::ReadersCache::BusyItem reader = readers.get(static_cast<std::size_t>(cell.mContextList[i].index));
 | |
|                 cell.restore(*reader, static_cast<int>(i));
 | |
|                 ESM::CellRef cellRef;
 | |
|                 bool deleted = false;
 | |
|                 while (ESM::Cell::getNextRef(*reader, cellRef, deleted))
 | |
|                 {
 | |
|                     Misc::StringUtils::lowerCaseInPlace(cellRef.mRefID);
 | |
|                     const ESM::RecNameInts type = getType(esmData, cellRef.mRefID);
 | |
|                     if (type == ESM::RecNameInts {})
 | |
|                         continue;
 | |
|                     cellRefs.emplace_back(deleted, type, cellRef.mRefNum, std::move(cellRef.mRefID),
 | |
|                                         cellRef.mScale, cellRef.mPos);
 | |
|                 }
 | |
|             }
 | |
| 
 | |
|             Log(Debug::Debug) << "Loaded " << cellRefs.size() << " cell refs";
 | |
| 
 | |
|             const auto getKey = [] (const EsmLoader::Record<CellRef>& v) -> const ESM::RefNum& { return v.mValue.mRefNum; };
 | |
|             std::vector<CellRef> result = prepareRecords(cellRefs, getKey);
 | |
| 
 | |
|             Log(Debug::Debug) << "Prepared " << result.size() << " unique cell refs";
 | |
| 
 | |
|             return result;
 | |
|         }
 | |
| 
 | |
|         template <class F>
 | |
|         void forEachObject(const ESM::Cell& cell, const EsmLoader::EsmData& esmData, const VFS::Manager& vfs,
 | |
|             Resource::BulletShapeManager& bulletShapeManager, ESM::ReadersCache& readers,
 | |
|             F&& f)
 | |
|         {
 | |
|             std::vector<CellRef> cellRefs = loadCellRefs(cell, esmData, readers);
 | |
| 
 | |
|             Log(Debug::Debug) << "Prepared " << cellRefs.size() << " unique cell refs";
 | |
| 
 | |
|             for (CellRef& cellRef : cellRefs)
 | |
|             {
 | |
|                 std::string model(getModel(esmData, cellRef.mRefId, cellRef.mType));
 | |
|                 if (model.empty())
 | |
|                     continue;
 | |
| 
 | |
|                 if (cellRef.mType != ESM::REC_STAT)
 | |
|                     model = Misc::ResourceHelpers::correctActorModelPath(model, &vfs);
 | |
| 
 | |
|                 osg::ref_ptr<const Resource::BulletShape> shape = [&]
 | |
|                 {
 | |
|                     try
 | |
|                     {
 | |
|                         return bulletShapeManager.getShape("meshes/" + model);
 | |
|                     }
 | |
|                     catch (const std::exception& e)
 | |
|                     {
 | |
|                         Log(Debug::Warning) << "Failed to load cell ref \"" << cellRef.mRefId << "\" model \"" << model << "\": " << e.what();
 | |
|                         return osg::ref_ptr<const Resource::BulletShape>();
 | |
|                     }
 | |
|                 } ();
 | |
| 
 | |
|                 if (shape == nullptr)
 | |
|                     continue;
 | |
| 
 | |
|                 switch (cellRef.mType)
 | |
|                 {
 | |
|                     case ESM::REC_ACTI:
 | |
|                     case ESM::REC_CONT:
 | |
|                     case ESM::REC_DOOR:
 | |
|                     case ESM::REC_STAT:
 | |
|                         f(BulletObject {std::move(shape), cellRef.mPos, cellRef.mScale});
 | |
|                         break;
 | |
|                     default:
 | |
|                         break;
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     void forEachBulletObject(ESM::ReadersCache& readers, const VFS::Manager& vfs,
 | |
|         Resource::BulletShapeManager& bulletShapeManager, const EsmLoader::EsmData& esmData,
 | |
|         std::function<void (const ESM::Cell& cell, const BulletObject& object)> callback)
 | |
|     {
 | |
|         Log(Debug::Info) << "Processing " << esmData.mCells.size() << " cells...";
 | |
| 
 | |
|         for (std::size_t i = 0; i < esmData.mCells.size(); ++i)
 | |
|         {
 | |
|             const ESM::Cell& cell = esmData.mCells[i];
 | |
|             const bool exterior = cell.isExterior();
 | |
| 
 | |
|             Log(Debug::Debug) << "Processing " << (exterior ? "exterior" : "interior")
 | |
|                 << " cell (" << (i + 1) << "/" << esmData.mCells.size() << ") \"" << cell.getDescription() << "\"";
 | |
| 
 | |
|             std::size_t objects = 0;
 | |
| 
 | |
|             forEachObject(cell, esmData, vfs, bulletShapeManager, readers,
 | |
|                 [&] (const BulletObject& object)
 | |
|                 {
 | |
|                     callback(cell, object);
 | |
|                     ++objects;
 | |
|                 });
 | |
| 
 | |
|             Log(Debug::Info) << "Processed " << (exterior ? "exterior" : "interior")
 | |
|                 << " cell (" << (i + 1) << "/" << esmData.mCells.size() << ") " << cell.getDescription()
 | |
|                 << " with " << objects << " objects";
 | |
|         }
 | |
|     }
 | |
| }
 |