mirror of https://github.com/OpenMW/openmw.git
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.
330 lines
16 KiB
C++
330 lines
16 KiB
C++
#include "tes4.hpp"
|
|
#include "arguments.hpp"
|
|
#include "labels.hpp"
|
|
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <type_traits>
|
|
|
|
#include <components/esm/esmcommon.hpp>
|
|
#include <components/esm4/reader.hpp>
|
|
#include <components/esm4/records.hpp>
|
|
|
|
namespace EsmTool
|
|
{
|
|
namespace
|
|
{
|
|
struct Params
|
|
{
|
|
const bool mQuite;
|
|
|
|
explicit Params(const Arguments& info)
|
|
: mQuite(info.quiet_given || info.mode == "clone")
|
|
{}
|
|
};
|
|
|
|
std::string toString(ESM4::GroupType type)
|
|
{
|
|
switch (type)
|
|
{
|
|
case ESM4::Grp_RecordType: return "RecordType";
|
|
case ESM4::Grp_WorldChild: return "WorldChild";
|
|
case ESM4::Grp_InteriorCell: return "InteriorCell";
|
|
case ESM4::Grp_InteriorSubCell: return "InteriorSubCell";
|
|
case ESM4::Grp_ExteriorCell: return "ExteriorCell";
|
|
case ESM4::Grp_ExteriorSubCell: return "ExteriorSubCell";
|
|
case ESM4::Grp_CellChild: return "CellChild";
|
|
case ESM4::Grp_TopicChild: return "TopicChild";
|
|
case ESM4::Grp_CellPersistentChild: return "CellPersistentChild";
|
|
case ESM4::Grp_CellTemporaryChild: return "CellTemporaryChild";
|
|
case ESM4::Grp_CellVisibleDistChild: return "CellVisibleDistChild";
|
|
}
|
|
|
|
return "Unknown (" + std::to_string(type) + ")";
|
|
}
|
|
|
|
template <class T, class = std::void_t<>>
|
|
struct HasFormId : std::false_type {};
|
|
|
|
template <class T>
|
|
struct HasFormId<T, std::void_t<decltype(T::mFormId)>> : std::true_type {};
|
|
|
|
template <class T>
|
|
constexpr bool hasFormId = HasFormId<T>::value;
|
|
|
|
template <class T, class = std::void_t<>>
|
|
struct HasFlags : std::false_type {};
|
|
|
|
template <class T>
|
|
struct HasFlags<T, std::void_t<decltype(T::mFlags)>> : std::true_type {};
|
|
|
|
template <class T>
|
|
constexpr bool hasFlags = HasFlags<T>::value;
|
|
|
|
template <class T>
|
|
void readTypedRecord(const Params& params, ESM4::Reader& reader)
|
|
{
|
|
reader.getRecordData();
|
|
|
|
T value;
|
|
value.load(reader);
|
|
|
|
if (params.mQuite)
|
|
return;
|
|
|
|
std::cout << "\n Record: " << ESM::NAME(reader.hdr().record.typeId).toStringView();
|
|
if constexpr (hasFormId<T>)
|
|
std::cout << ' ' << value.mFormId;
|
|
if constexpr (hasFlags<T>)
|
|
std::cout << "\n Record flags: " << recordFlags(value.mFlags);
|
|
std::cout << '\n';
|
|
}
|
|
|
|
void readRecord(const Params& params, ESM4::Reader& reader)
|
|
{
|
|
switch (static_cast<ESM4::RecordTypes>(reader.hdr().record.typeId))
|
|
{
|
|
case ESM4::REC_AACT: break;
|
|
case ESM4::REC_ACHR: return readTypedRecord<ESM4::ActorCharacter>(params, reader);
|
|
case ESM4::REC_ACRE: return readTypedRecord<ESM4::ActorCreature>(params, reader);
|
|
case ESM4::REC_ACTI: return readTypedRecord<ESM4::Activator>(params, reader);
|
|
case ESM4::REC_ADDN: break;
|
|
case ESM4::REC_ALCH: return readTypedRecord<ESM4::Potion>(params, reader);
|
|
case ESM4::REC_ALOC: return readTypedRecord<ESM4::MediaLocationController>(params, reader);
|
|
case ESM4::REC_AMMO: return readTypedRecord<ESM4::Ammunition>(params, reader);
|
|
case ESM4::REC_ANIO: return readTypedRecord<ESM4::AnimObject>(params, reader);
|
|
case ESM4::REC_APPA: return readTypedRecord<ESM4::Apparatus>(params, reader);
|
|
case ESM4::REC_ARMA: return readTypedRecord<ESM4::ArmorAddon>(params, reader);
|
|
case ESM4::REC_ARMO: return readTypedRecord<ESM4::Armor>(params, reader);
|
|
case ESM4::REC_ARTO: break;
|
|
case ESM4::REC_ASPC: return readTypedRecord<ESM4::AcousticSpace>(params, reader);
|
|
case ESM4::REC_ASTP: break;
|
|
case ESM4::REC_AVIF: break;
|
|
case ESM4::REC_BOOK: return readTypedRecord<ESM4::Book>(params, reader);
|
|
case ESM4::REC_BPTD: return readTypedRecord<ESM4::BodyPartData>(params, reader);
|
|
case ESM4::REC_CAMS: break;
|
|
case ESM4::REC_CCRD: break;
|
|
case ESM4::REC_CELL: return readTypedRecord<ESM4::Cell>(params, reader);
|
|
case ESM4::REC_CLAS: return readTypedRecord<ESM4::Class>(params, reader);
|
|
case ESM4::REC_CLFM: return readTypedRecord<ESM4::Colour>(params, reader);
|
|
case ESM4::REC_CLMT: break;
|
|
case ESM4::REC_CLOT: return readTypedRecord<ESM4::Clothing>(params, reader);
|
|
case ESM4::REC_CMNY: break;
|
|
case ESM4::REC_COBJ: break;
|
|
case ESM4::REC_COLL: break;
|
|
case ESM4::REC_CONT: return readTypedRecord<ESM4::Container>(params, reader);
|
|
case ESM4::REC_CPTH: break;
|
|
case ESM4::REC_CREA: return readTypedRecord<ESM4::Creature>(params, reader);
|
|
case ESM4::REC_CSTY: break;
|
|
case ESM4::REC_DEBR: break;
|
|
case ESM4::REC_DIAL: return readTypedRecord<ESM4::Dialogue>(params, reader);
|
|
case ESM4::REC_DLBR: break;
|
|
case ESM4::REC_DLVW: break;
|
|
case ESM4::REC_DOBJ: return readTypedRecord<ESM4::DefaultObj>(params, reader);
|
|
case ESM4::REC_DOOR: return readTypedRecord<ESM4::Door>(params, reader);
|
|
case ESM4::REC_DUAL: break;
|
|
case ESM4::REC_ECZN: break;
|
|
case ESM4::REC_EFSH: break;
|
|
case ESM4::REC_ENCH: break;
|
|
case ESM4::REC_EQUP: break;
|
|
case ESM4::REC_EXPL: break;
|
|
case ESM4::REC_EYES: return readTypedRecord<ESM4::Eyes>(params, reader);
|
|
case ESM4::REC_FACT: break;
|
|
case ESM4::REC_FLOR: return readTypedRecord<ESM4::Flora>(params, reader);
|
|
case ESM4::REC_FLST: return readTypedRecord<ESM4::FormIdList>(params, reader);
|
|
case ESM4::REC_FSTP: break;
|
|
case ESM4::REC_FSTS: break;
|
|
case ESM4::REC_FURN: return readTypedRecord<ESM4::Furniture>(params, reader);
|
|
case ESM4::REC_GLOB: return readTypedRecord<ESM4::GlobalVariable>(params, reader);
|
|
case ESM4::REC_GMST: break;
|
|
case ESM4::REC_GRAS: return readTypedRecord<ESM4::Grass>(params, reader);
|
|
case ESM4::REC_GRUP: break;
|
|
case ESM4::REC_HAIR: return readTypedRecord<ESM4::Hair>(params, reader);
|
|
case ESM4::REC_HAZD: break;
|
|
case ESM4::REC_HDPT: return readTypedRecord<ESM4::HeadPart>(params, reader);
|
|
case ESM4::REC_IDLE:
|
|
// FIXME: ESM4::IdleAnimation::load does not work with Oblivion.esm
|
|
// return readTypedRecord<ESM4::IdleAnimation>(params, reader);
|
|
break;
|
|
case ESM4::REC_IDLM: return readTypedRecord<ESM4::IdleMarker>(params, reader);
|
|
case ESM4::REC_IMAD: break;
|
|
case ESM4::REC_IMGS: break;
|
|
case ESM4::REC_IMOD: return readTypedRecord<ESM4::ItemMod>(params, reader);
|
|
case ESM4::REC_INFO: return readTypedRecord<ESM4::DialogInfo>(params, reader);
|
|
case ESM4::REC_INGR: return readTypedRecord<ESM4::Ingredient>(params, reader);
|
|
case ESM4::REC_IPCT: break;
|
|
case ESM4::REC_IPDS: break;
|
|
case ESM4::REC_KEYM: return readTypedRecord<ESM4::Key>(params, reader);
|
|
case ESM4::REC_KYWD: break;
|
|
case ESM4::REC_LAND: return readTypedRecord<ESM4::Land>(params, reader);
|
|
case ESM4::REC_LCRT: break;
|
|
case ESM4::REC_LCTN: break;
|
|
case ESM4::REC_LGTM: return readTypedRecord<ESM4::LightingTemplate>(params, reader);
|
|
case ESM4::REC_LIGH: return readTypedRecord<ESM4::Light>(params, reader);
|
|
case ESM4::REC_LSCR: break;
|
|
case ESM4::REC_LTEX: return readTypedRecord<ESM4::LandTexture>(params, reader);
|
|
case ESM4::REC_LVLC: return readTypedRecord<ESM4::LevelledCreature>(params, reader);
|
|
case ESM4::REC_LVLI: return readTypedRecord<ESM4::LevelledItem>(params, reader);
|
|
case ESM4::REC_LVLN: return readTypedRecord<ESM4::LevelledNpc>(params, reader);
|
|
case ESM4::REC_LVSP: break;
|
|
case ESM4::REC_MATO: return readTypedRecord<ESM4::Material>(params, reader);
|
|
case ESM4::REC_MATT: break;
|
|
case ESM4::REC_MESG: break;
|
|
case ESM4::REC_MGEF: break;
|
|
case ESM4::REC_MISC: return readTypedRecord<ESM4::MiscItem>(params, reader);
|
|
case ESM4::REC_MOVT: break;
|
|
case ESM4::REC_MSET: return readTypedRecord<ESM4::MediaSet>(params, reader);
|
|
case ESM4::REC_MSTT: return readTypedRecord<ESM4::MovableStatic>(params, reader);
|
|
case ESM4::REC_MUSC: return readTypedRecord<ESM4::Music>(params, reader);
|
|
case ESM4::REC_MUST: break;
|
|
case ESM4::REC_NAVI: return readTypedRecord<ESM4::Navigation>(params, reader);
|
|
case ESM4::REC_NAVM: return readTypedRecord<ESM4::NavMesh>(params, reader);
|
|
case ESM4::REC_NOTE: return readTypedRecord<ESM4::Note>(params, reader);
|
|
case ESM4::REC_NPC_: return readTypedRecord<ESM4::Npc>(params, reader);
|
|
case ESM4::REC_OTFT: return readTypedRecord<ESM4::Outfit>(params, reader);
|
|
case ESM4::REC_PACK: return readTypedRecord<ESM4::AIPackage>(params, reader);
|
|
case ESM4::REC_PERK: break;
|
|
case ESM4::REC_PGRD: return readTypedRecord<ESM4::Pathgrid>(params, reader);
|
|
case ESM4::REC_PGRE: return readTypedRecord<ESM4::PlacedGrenade>(params, reader);
|
|
case ESM4::REC_PHZD: break;
|
|
case ESM4::REC_PROJ: break;
|
|
case ESM4::REC_PWAT: return readTypedRecord<ESM4::PlaceableWater>(params, reader);
|
|
case ESM4::REC_QUST: return readTypedRecord<ESM4::Quest>(params, reader);
|
|
case ESM4::REC_RACE: return readTypedRecord<ESM4::Race>(params, reader);
|
|
case ESM4::REC_REFR: return readTypedRecord<ESM4::Reference>(params, reader);
|
|
case ESM4::REC_REGN: return readTypedRecord<ESM4::Region>(params, reader);
|
|
case ESM4::REC_RELA: break;
|
|
case ESM4::REC_REVB: break;
|
|
case ESM4::REC_RFCT: break;
|
|
case ESM4::REC_ROAD: return readTypedRecord<ESM4::Road>(params, reader);
|
|
case ESM4::REC_SBSP: return readTypedRecord<ESM4::SubSpace>(params, reader);
|
|
case ESM4::REC_SCEN: break;
|
|
case ESM4::REC_SCOL: return readTypedRecord<ESM4::StaticCollection>(params, reader);
|
|
case ESM4::REC_SCPT: return readTypedRecord<ESM4::Script>(params, reader);
|
|
case ESM4::REC_SCRL: return readTypedRecord<ESM4::Scroll>(params, reader);
|
|
case ESM4::REC_SGST: return readTypedRecord<ESM4::SigilStone>(params, reader);
|
|
case ESM4::REC_SHOU: break;
|
|
case ESM4::REC_SLGM: return readTypedRecord<ESM4::SoulGem>(params, reader);
|
|
case ESM4::REC_SMBN: break;
|
|
case ESM4::REC_SMEN: break;
|
|
case ESM4::REC_SMQN: break;
|
|
case ESM4::REC_SNCT: break;
|
|
case ESM4::REC_SNDR: return readTypedRecord<ESM4::SoundReference>(params, reader);
|
|
case ESM4::REC_SOPM: break;
|
|
case ESM4::REC_SOUN: return readTypedRecord<ESM4::Sound>(params, reader);
|
|
case ESM4::REC_SPEL: break;
|
|
case ESM4::REC_SPGD: break;
|
|
case ESM4::REC_STAT: return readTypedRecord<ESM4::Static>(params, reader);
|
|
case ESM4::REC_TACT: return readTypedRecord<ESM4::TalkingActivator>(params, reader);
|
|
case ESM4::REC_TERM: return readTypedRecord<ESM4::Terminal>(params, reader);
|
|
case ESM4::REC_TES4: return readTypedRecord<ESM4::Header>(params, reader);
|
|
case ESM4::REC_TREE: return readTypedRecord<ESM4::Tree>(params, reader);
|
|
case ESM4::REC_TXST: return readTypedRecord<ESM4::TextureSet>(params, reader);
|
|
case ESM4::REC_VTYP: break;
|
|
case ESM4::REC_WATR: break;
|
|
case ESM4::REC_WEAP: return readTypedRecord<ESM4::Weapon>(params, reader);
|
|
case ESM4::REC_WOOP: break;
|
|
case ESM4::REC_WRLD: return readTypedRecord<ESM4::World>(params, reader);
|
|
case ESM4::REC_WTHR: break;
|
|
}
|
|
|
|
if (!params.mQuite)
|
|
std::cout << "\n Unsupported record: " << ESM::NAME(reader.hdr().record.typeId).toStringView() << '\n';
|
|
|
|
reader.skipRecordData();
|
|
}
|
|
|
|
bool readItem(const Params& params, ESM4::Reader& reader);
|
|
|
|
bool readGroup(const Params& params, ESM4::Reader& reader)
|
|
{
|
|
const ESM4::RecordHeader& header = reader.hdr();
|
|
|
|
if (!params.mQuite)
|
|
std::cout << "\nGroup: " << toString(static_cast<ESM4::GroupType>(header.group.type))
|
|
<< " " << ESM::NAME(header.group.typeId).toStringView() << '\n';
|
|
|
|
switch (static_cast<ESM4::GroupType>(header.group.type))
|
|
{
|
|
case ESM4::Grp_RecordType:
|
|
case ESM4::Grp_InteriorCell:
|
|
case ESM4::Grp_InteriorSubCell:
|
|
case ESM4::Grp_ExteriorCell:
|
|
case ESM4::Grp_ExteriorSubCell:
|
|
reader.enterGroup();
|
|
return readItem(params, reader);
|
|
case ESM4::Grp_WorldChild:
|
|
case ESM4::Grp_CellChild:
|
|
case ESM4::Grp_TopicChild:
|
|
case ESM4::Grp_CellPersistentChild:
|
|
case ESM4::Grp_CellTemporaryChild:
|
|
case ESM4::Grp_CellVisibleDistChild:
|
|
reader.adjustGRUPFormId();
|
|
reader.enterGroup();
|
|
if (!reader.hasMoreRecs())
|
|
return false;
|
|
return readItem(params, reader);
|
|
}
|
|
|
|
reader.skipGroup();
|
|
|
|
return true;
|
|
}
|
|
|
|
bool readItem(const Params& params, ESM4::Reader& reader)
|
|
{
|
|
if (!reader.getRecordHeader() || !reader.hasMoreRecs())
|
|
return false;
|
|
|
|
const ESM4::RecordHeader& header = reader.hdr();
|
|
|
|
if (header.record.typeId == ESM4::REC_GRUP)
|
|
return readGroup(params, reader);
|
|
|
|
readRecord(params, reader);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
int loadTes4(const Arguments& info, std::unique_ptr<std::ifstream>&& stream)
|
|
{
|
|
std::cout << "Loading TES4 file: " << info.filename << '\n';
|
|
|
|
try
|
|
{
|
|
const ToUTF8::StatelessUtf8Encoder encoder(ToUTF8::calculateEncoding(info.encoding));
|
|
ESM4::Reader reader(std::move(stream), info.filename);
|
|
reader.setEncoder(&encoder);
|
|
const Params params(info);
|
|
|
|
if (!params.mQuite)
|
|
{
|
|
std::cout << "Author: " << reader.getAuthor() << '\n'
|
|
<< "Description: " << reader.getDesc() << '\n'
|
|
<< "File format version: " << reader.esmVersion() << '\n';
|
|
|
|
if (const std::vector<ESM::MasterData>& masterData = reader.getGameFiles(); !masterData.empty())
|
|
{
|
|
std::cout << "Masters:" << '\n';
|
|
for (const auto& master : masterData)
|
|
std::cout << " " << master.name << ", " << master.size << " bytes\n";
|
|
}
|
|
}
|
|
|
|
while (reader.hasMoreRecs())
|
|
{
|
|
reader.exitGroupCheck();
|
|
if (!readItem(params, reader))
|
|
break;
|
|
}
|
|
}
|
|
catch (const std::exception& e)
|
|
{
|
|
std::cout << "\nERROR:\n\n " << e.what() << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
}
|