diff --git a/apps/mwiniimporter/importer.cpp b/apps/mwiniimporter/importer.cpp index 24646b844..66b56f8a6 100644 --- a/apps/mwiniimporter/importer.cpp +++ b/apps/mwiniimporter/importer.cpp @@ -3,6 +3,7 @@ #include #include #include +#include #include #include @@ -824,33 +825,73 @@ void MwIniImporter::importArchives(multistrmap &cfg, const multistrmap &ini) con } } -void MwIniImporter::importGameFiles(multistrmap &cfg, const multistrmap &ini, const boost::filesystem::path& iniFilename) const { - std::vector > contentFiles; +void MwIniImporter::dependencySortStep(std::string& el, MwIniImporter::deplist& src, std::vector& ret) +{ + auto it = std::find_if(src.begin(), src.end(), [&el](std::pair< std::string, std::vector >& o) + { + return o.first == el; + }); + if (it != src.end()) + { + auto o = std::move(*it); + src.erase(it); + for (auto name : o.second) + { + MwIniImporter::dependencySortStep(name, src, ret); + } + ret.push_back(std::move(o.first)); + } +} + +std::vector MwIniImporter::dependencySort(MwIniImporter::deplist src) +{ + std::vector ret; + while (!src.empty()) + { + MwIniImporter::dependencySortStep(src.begin()->first, src, ret); + } + return ret; +} + +std::vector::iterator MwIniImporter::findString(std::vector& v, const std::string& s) +{ + return std::find_if(v.begin(), v.end(), [&s](const std::string& str) + { + return Misc::StringUtils::ciEqual(str, s); + }); +} + +void MwIniImporter::importGameFiles(multistrmap &cfg, const multistrmap &ini, const boost::filesystem::path& iniFilename) const +{ + std::vector> contentFiles; std::string baseGameFile("Game Files:GameFile"); std::string gameFile(""); std::time_t defaultTime = 0; + ToUTF8::Utf8Encoder encoder(mEncoding); // assume the Game Files are all in a "Data Files" directory under the directory holding Morrowind.ini const boost::filesystem::path gameFilesDir(iniFilename.parent_path() /= "Data Files"); multistrmap::const_iterator it = ini.begin(); - for(int i=0; it != ini.end(); i++) { + for (int i=0; it != ini.end(); i++) + { gameFile = baseGameFile; gameFile.append(this->numberToString(i)); it = ini.find(gameFile); - if(it == ini.end()) { + if(it == ini.end()) break; - } - for(std::vector::const_iterator entry = it->second.begin(); entry!=it->second.end(); ++entry) { + for(std::vector::const_iterator entry = it->second.begin(); entry!=it->second.end(); ++entry) + { std::string filetype(entry->substr(entry->length()-3)); Misc::StringUtils::lowerCaseInPlace(filetype); - if(filetype.compare("esm") == 0 || filetype.compare("esp") == 0) { + if(filetype.compare("esm") == 0 || filetype.compare("esp") == 0) + { boost::filesystem::path filepath(gameFilesDir); filepath /= *entry; - contentFiles.push_back(std::make_pair(lastWriteTime(filepath, defaultTime), *entry)); + contentFiles.push_back({lastWriteTime(filepath, defaultTime), filepath}); } } } @@ -858,11 +899,46 @@ void MwIniImporter::importGameFiles(multistrmap &cfg, const multistrmap &ini, co cfg.erase("content"); cfg.insert( std::make_pair("content", std::vector() ) ); - // this will sort files by time order first, then alphabetical (maybe), I suspect non ASCII filenames will be stuffed. + // sort by timestamp sort(contentFiles.begin(), contentFiles.end()); - for(std::vector >::const_iterator iter=contentFiles.begin(); iter!=contentFiles.end(); ++iter) { - cfg["content"].push_back(iter->second); + + MwIniImporter::deplist unsortedFiles; + + ESM::ESMReader reader; + reader.setEncoder(&encoder); + for (auto& file : contentFiles) + { + reader.open(file.second.string()); + std::vector deps; + for (auto& depFile : reader.getGameFiles()) + { + deps.push_back(depFile.name); + } + unsortedFiles.emplace_back(boost::filesystem::path(reader.getName()).filename().string(), deps); + reader.close(); + } + + auto sortedFiles = dependencySort(unsortedFiles); + + // hard-coded dependency Morrowind - Tribunal - Bloodmoon + if(findString(sortedFiles, "Morrowind.esm") != sortedFiles.end()) + { + auto foundTribunal = findString(sortedFiles, "Tribunal.esm"); + auto foundBloodmoon = findString(sortedFiles, "Bloodmoon.esm"); + + if (foundBloodmoon != sortedFiles.end() && foundTribunal != sortedFiles.end()) + { + size_t dstBloodmoon = std::distance(sortedFiles.begin(), foundBloodmoon); + size_t dstTribunal = std::distance(sortedFiles.begin(), foundTribunal); + if (dstBloodmoon < dstTribunal) + dstTribunal++; + sortedFiles.insert(foundBloodmoon, *foundTribunal); + sortedFiles.erase(sortedFiles.begin() + dstTribunal); + } } + + for (auto& file : sortedFiles) + cfg["content"].push_back(file); } void MwIniImporter::writeToFile(std::ostream &out, const multistrmap &cfg) { diff --git a/apps/mwiniimporter/importer.hpp b/apps/mwiniimporter/importer.hpp index c73cc65b5..44b2af83c 100644 --- a/apps/mwiniimporter/importer.hpp +++ b/apps/mwiniimporter/importer.hpp @@ -14,6 +14,7 @@ class MwIniImporter { public: typedef std::map strmap; typedef std::map > multistrmap; + typedef std::vector< std::pair< std::string, std::vector > > deplist; MwIniImporter(); void setInputEncoding(const ToUTF8::FromType& encoding); @@ -22,12 +23,17 @@ class MwIniImporter { static multistrmap loadCfgFile(const boost::filesystem::path& filename); void merge(multistrmap &cfg, const multistrmap &ini) const; void mergeFallback(multistrmap &cfg, const multistrmap &ini) const; - void importGameFiles(multistrmap &cfg, const multistrmap &ini, + void importGameFiles(multistrmap &cfg, const multistrmap &ini, const boost::filesystem::path& iniFilename) const; void importArchives(multistrmap &cfg, const multistrmap &ini) const; static void writeToFile(std::ostream &out, const multistrmap &cfg); + static std::vector dependencySort(MwIniImporter::deplist src); + private: + static void dependencySortStep(std::string& el, MwIniImporter::deplist& src, std::vector& ret); + static std::vector::iterator findString(std::vector& v, const std::string& s); + static void insertMultistrmap(multistrmap &cfg, const std::string& key, const std::string& value); static std::string numberToString(int n); @@ -40,5 +46,4 @@ class MwIniImporter { ToUTF8::FromType mEncoding; }; - #endif