You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
openmw/apps/openmw/mwlua/cellbindings.cpp

287 lines
13 KiB
C++

#include "cellbindings.hpp"
#include <components/esm/esmbridge.hpp>
#include <components/esm3/loadacti.hpp>
#include <components/esm3/loadalch.hpp>
#include <components/esm3/loadappa.hpp>
#include <components/esm3/loadarmo.hpp>
#include <components/esm3/loadbook.hpp>
#include <components/esm3/loadclot.hpp>
#include <components/esm3/loadcont.hpp>
#include <components/esm3/loadcrea.hpp>
#include <components/esm3/loaddoor.hpp>
#include <components/esm3/loadingr.hpp>
#include <components/esm3/loadlevlist.hpp>
#include <components/esm3/loadligh.hpp>
#include <components/esm3/loadlock.hpp>
#include <components/esm3/loadmisc.hpp>
#include <components/esm3/loadnpc.hpp>
#include <components/esm3/loadprob.hpp>
#include <components/esm3/loadrepa.hpp>
#include <components/esm3/loadstat.hpp>
#include <components/esm3/loadweap.hpp>
#include <components/esm4/loadacti.hpp>
#include <components/esm4/loadalch.hpp>
#include <components/esm4/loadammo.hpp>
#include <components/esm4/loadarmo.hpp>
#include <components/esm4/loadbook.hpp>
#include <components/esm4/loadcell.hpp>
#include <components/esm4/loadclot.hpp>
#include <components/esm4/loadcont.hpp>
#include <components/esm4/loaddoor.hpp>
#include <components/esm4/loadflor.hpp>
#include <components/esm4/loadfurn.hpp>
#include <components/esm4/loadimod.hpp>
#include <components/esm4/loadingr.hpp>
#include <components/esm4/loadligh.hpp>
#include <components/esm4/loadmisc.hpp>
#include <components/esm4/loadmstt.hpp>
#include <components/esm4/loadrefr.hpp>
#include <components/esm4/loadstat.hpp>
#include <components/esm4/loadtree.hpp>
#include <components/esm4/loadweap.hpp>
#include "../mwworld/cellstore.hpp"
#include "../mwworld/worldmodel.hpp"
#include "types/types.hpp"
namespace sol
{
template <>
struct is_automagical<MWLua::LCell> : std::false_type
{
};
template <>
struct is_automagical<MWLua::GCell> : std::false_type
{
};
}
namespace MWLua
{
template <class CellT, class ObjectT>
static void initCellBindings(const std::string& prefix, const Context& context)
{
sol::usertype<CellT> cellT = context.mLua->sol().new_usertype<CellT>(prefix + "Cell");
cellT[sol::meta_function::equal_to] = [](const CellT& a, const CellT& b) { return a.mStore == b.mStore; };
cellT[sol::meta_function::to_string] = [](const CellT& c) {
auto cell = c.mStore->getCell();
std::stringstream res;
if (cell->isExterior())
res << "exterior(" << cell->getGridX() << ", " << cell->getGridY() << ", "
<< cell->getWorldSpace().toDebugString() << ")";
else
res << "interior(" << cell->getNameId() << ")";
return res.str();
};
cellT["name"] = sol::readonly_property([](const CellT& c) { return c.mStore->getCell()->getNameId(); });
cellT["region"] = sol::readonly_property(
[](const CellT& c) -> std::string { return c.mStore->getCell()->getRegion().serializeText(); });
cellT["worldSpaceId"] = sol::readonly_property(
[](const CellT& c) -> std::string { return c.mStore->getCell()->getWorldSpace().serializeText(); });
cellT["gridX"] = sol::readonly_property([](const CellT& c) { return c.mStore->getCell()->getGridX(); });
cellT["gridY"] = sol::readonly_property([](const CellT& c) { return c.mStore->getCell()->getGridY(); });
cellT["hasWater"] = sol::readonly_property([](const CellT& c) { return c.mStore->getCell()->hasWater(); });
cellT["hasSky"] = sol::readonly_property([](const CellT& c) {
return c.mStore->getCell()->isExterior() || (c.mStore->getCell()->isQuasiExterior()) != 0;
});
cellT["isExterior"] = sol::readonly_property([](const CellT& c) { return c.mStore->isExterior(); });
// deprecated, use cell:hasTag("QuasiExterior") instead
cellT["isQuasiExterior"]
= sol::readonly_property([](const CellT& c) { return (c.mStore->getCell()->isQuasiExterior()) != 0; });
cellT["hasTag"] = [](const CellT& c, std::string_view tag) -> bool {
if (tag == "NoSleep")
return (c.mStore->getCell()->noSleep()) != 0;
else if (tag == "QuasiExterior")
return (c.mStore->getCell()->isQuasiExterior()) != 0;
return false;
};
cellT["isInSameSpace"] = [](const CellT& c, const ObjectT& obj) {
const MWWorld::Ptr& ptr = obj.ptr();
if (!ptr.isInCell())
return false;
MWWorld::CellStore* cell = ptr.getCell();
return cell == c.mStore || (cell->getCell()->getWorldSpace() == c.mStore->getCell()->getWorldSpace());
};
if constexpr (std::is_same_v<CellT, GCell>)
{ // only for global scripts
cellT["getAll"] = [ids = getPackageToTypeTable(context.mLua->sol())](
const CellT& cell, sol::optional<sol::table> type) {
if (cell.mStore->getState() != MWWorld::CellStore::State_Loaded)
cell.mStore->load();
ObjectIdList res = std::make_shared<std::vector<ObjectId>>();
auto visitor = [&](const MWWorld::Ptr& ptr) {
if (ptr.mRef->isDeleted())
return true;
MWBase::Environment::get().getWorldModel()->registerPtr(ptr);
if (getLiveCellRefType(ptr.mRef) == ptr.getType())
res->push_back(getId(ptr));
return true;
};
bool ok = true;
if (!type.has_value())
cell.mStore->forEach(std::move(visitor));
else if (ids[*type] == sol::nil)
ok = false;
else
{
uint32_t typeId = ids[*type];
switch (typeId)
{
case ESM::REC_INTERNAL_PLAYER:
{
MWWorld::Ptr player = MWBase::Environment::get().getWorld()->getPlayerPtr();
if (player.getCell() == cell.mStore)
res->push_back(getId(player));
}
break;
case ESM::REC_CREA:
cell.mStore->template forEachType<ESM::Creature>(visitor);
break;
case ESM::REC_NPC_:
cell.mStore->template forEachType<ESM::NPC>(visitor);
break;
case ESM::REC_ACTI:
cell.mStore->template forEachType<ESM::Activator>(visitor);
break;
case ESM::REC_DOOR:
cell.mStore->template forEachType<ESM::Door>(visitor);
break;
case ESM::REC_CONT:
cell.mStore->template forEachType<ESM::Container>(visitor);
break;
case ESM::REC_ALCH:
cell.mStore->template forEachType<ESM::Potion>(visitor);
break;
case ESM::REC_ARMO:
cell.mStore->template forEachType<ESM::Armor>(visitor);
break;
case ESM::REC_BOOK:
cell.mStore->template forEachType<ESM::Book>(visitor);
break;
case ESM::REC_CLOT:
cell.mStore->template forEachType<ESM::Clothing>(visitor);
break;
case ESM::REC_INGR:
cell.mStore->template forEachType<ESM::Ingredient>(visitor);
break;
case ESM::REC_LIGH:
cell.mStore->template forEachType<ESM::Light>(visitor);
break;
case ESM::REC_MISC:
cell.mStore->template forEachType<ESM::Miscellaneous>(visitor);
break;
case ESM::REC_WEAP:
cell.mStore->template forEachType<ESM::Weapon>(visitor);
break;
case ESM::REC_APPA:
cell.mStore->template forEachType<ESM::Apparatus>(visitor);
break;
case ESM::REC_LOCK:
cell.mStore->template forEachType<ESM::Lockpick>(visitor);
break;
case ESM::REC_PROB:
cell.mStore->template forEachType<ESM::Probe>(visitor);
break;
case ESM::REC_REPA:
cell.mStore->template forEachType<ESM::Repair>(visitor);
break;
case ESM::REC_STAT:
cell.mStore->template forEachType<ESM::Static>(visitor);
break;
case ESM::REC_LEVC:
cell.mStore->template forEachType<ESM::CreatureLevList>(visitor);
break;
case ESM::REC_ACTI4:
cell.mStore->template forEachType<ESM4::Activator>(visitor);
break;
case ESM::REC_AMMO4:
cell.mStore->template forEachType<ESM4::Ammunition>(visitor);
break;
case ESM::REC_ARMO4:
cell.mStore->template forEachType<ESM4::Armor>(visitor);
break;
case ESM::REC_BOOK4:
cell.mStore->template forEachType<ESM4::Book>(visitor);
break;
case ESM::REC_CLOT4:
cell.mStore->template forEachType<ESM4::Clothing>(visitor);
break;
case ESM::REC_CONT4:
cell.mStore->template forEachType<ESM4::Container>(visitor);
break;
case ESM::REC_DOOR4:
cell.mStore->template forEachType<ESM4::Door>(visitor);
break;
case ESM::REC_FLOR4:
cell.mStore->template forEachType<ESM4::Flora>(visitor);
break;
case ESM::REC_FURN4:
cell.mStore->template forEachType<ESM4::Furniture>(visitor);
break;
case ESM::REC_IMOD4:
cell.mStore->template forEachType<ESM4::ItemMod>(visitor);
break;
case ESM::REC_INGR4:
cell.mStore->template forEachType<ESM4::Ingredient>(visitor);
break;
case ESM::REC_LIGH4:
cell.mStore->template forEachType<ESM4::Light>(visitor);
break;
case ESM::REC_MISC4:
cell.mStore->template forEachType<ESM4::MiscItem>(visitor);
break;
case ESM::REC_MSTT4:
cell.mStore->template forEachType<ESM4::MovableStatic>(visitor);
break;
case ESM::REC_ALCH4:
cell.mStore->template forEachType<ESM4::Potion>(visitor);
break;
case ESM::REC_STAT4:
cell.mStore->template forEachType<ESM4::Static>(visitor);
break;
case ESM::REC_TREE4:
cell.mStore->template forEachType<ESM4::Tree>(visitor);
break;
case ESM::REC_WEAP4:
cell.mStore->template forEachType<ESM4::Weapon>(visitor);
break;
default:
ok = false;
}
}
if (!ok)
throw std::runtime_error(
std::string("Incorrect type argument in cell:getAll: " + LuaUtil::toString(*type)));
return GObjectList{ res };
};
}
}
void initCellBindingsForLocalScripts(const Context& context)
{
initCellBindings<LCell, LObject>("L", context);
}
void initCellBindingsForGlobalScripts(const Context& context)
{
initCellBindings<GCell, GObject>("G", context);
}
}