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.
428 lines
12 KiB
C++
428 lines
12 KiB
C++
/*
|
|
OpenMW - The completely unofficial reimplementation of Morrowind
|
|
Copyright (C) 2008-2010 Nicolay Korslund
|
|
Email: < korslund@gmail.com >
|
|
WWW: http://openmw.sourceforge.net/
|
|
|
|
This file (cpp_bsaarchive.cpp) is part of the OpenMW package.
|
|
|
|
OpenMW is distributed as free software: you can redistribute it
|
|
and/or modify it under the terms of the GNU General Public License
|
|
version 3, as published by the Free Software Foundation.
|
|
|
|
This program is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
version 3 along with this program. If not, see
|
|
http://www.gnu.org/licenses/ .
|
|
|
|
*/
|
|
|
|
#include "bsa_archive.hpp"
|
|
|
|
#include <OgreFileSystem.h>
|
|
#include <OgreArchive.h>
|
|
#include <OgreArchiveFactory.h>
|
|
#include <OgreArchiveManager.h>
|
|
#include "bsa_file.hpp"
|
|
#include <libs/mangle/stream/clients/ogre_datastream.hpp>
|
|
|
|
namespace
|
|
{
|
|
|
|
using namespace Ogre;
|
|
using namespace Mangle::Stream;
|
|
using namespace Bsa;
|
|
|
|
struct ciLessBoost : std::binary_function<std::string, std::string, bool>
|
|
{
|
|
bool operator() (const std::string & s1, const std::string & s2) const {
|
|
//case insensitive version of is_less
|
|
return boost::ilexicographical_compare(s1, s2);
|
|
}
|
|
};
|
|
|
|
struct pathComparer
|
|
{
|
|
private:
|
|
std::string find;
|
|
|
|
public:
|
|
pathComparer(const std::string& toFind) : find(toFind) { }
|
|
|
|
bool operator() (const std::string& other)
|
|
{
|
|
return boost::iequals(find, other);
|
|
}
|
|
};
|
|
|
|
static bool fsstrict = false;
|
|
|
|
/// An OGRE Archive wrapping a BSAFile archive
|
|
class DirArchive: public Ogre::FileSystemArchive
|
|
{
|
|
|
|
boost::filesystem::path currentdir;
|
|
std::map<std::string, std::vector<std::string>, ciLessBoost> m;
|
|
unsigned int cutoff;
|
|
|
|
bool findFile(const String& filename, std::string& copy) const
|
|
{
|
|
{
|
|
String passed = filename;
|
|
if(filename.at(filename.length() - 1) == '*' || filename.at(filename.length() - 1) == '?' || filename.at(filename.length() - 1) == '<'
|
|
|| filename.at(filename.length() - 1) == '"' || filename.at(filename.length() - 1) == '>' || filename.at(filename.length() - 1) == ':'
|
|
|| filename.at(filename.length() - 1) == '|')
|
|
{
|
|
passed = filename.substr(0, filename.length() - 2);
|
|
}
|
|
if(filename.at(filename.length() - 2) == '>' || filename.at(filename.length() - 2) == ':')
|
|
passed = filename.substr(0, filename.length() - 6);
|
|
copy = passed;
|
|
}
|
|
|
|
std::replace(copy.begin(), copy.end(), '\\', '/');
|
|
|
|
if(copy.at(0) == '/')
|
|
copy.erase(0, 1);
|
|
|
|
if(fsstrict == true)
|
|
return true;
|
|
|
|
std::string folder;
|
|
int delimiter = 0;
|
|
size_t lastSlash = copy.rfind('/');
|
|
if (lastSlash != std::string::npos)
|
|
{
|
|
folder = copy.substr(0, lastSlash);
|
|
delimiter = lastSlash+1;
|
|
}
|
|
|
|
std::vector<std::string> current;
|
|
{
|
|
std::map<std::string,std::vector<std::string>,ciLessBoost>::const_iterator found = m.find(folder);
|
|
|
|
if (found == m.end())
|
|
{
|
|
return false;
|
|
}
|
|
else
|
|
current = found->second;
|
|
}
|
|
|
|
std::vector<std::string>::iterator find = std::lower_bound(current.begin(), current.end(), copy, ciLessBoost());
|
|
if (find != current.end() && !ciLessBoost()(copy, current.front()))
|
|
{
|
|
if (!boost::iequals(copy, *find))
|
|
if ((find = std::find_if(current.begin(), current.end(), pathComparer(copy))) == current.end()) //\todo Check if this line is actually needed
|
|
return false;
|
|
|
|
copy = *find;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
public:
|
|
|
|
DirArchive(const String& name)
|
|
: FileSystemArchive(name, "Dir"), currentdir (name)
|
|
{
|
|
mType = "Dir";
|
|
std::string s = name;
|
|
cutoff = s.size() + 1;
|
|
if(fsstrict == false)
|
|
populateMap(currentdir);
|
|
|
|
}
|
|
void populateMap(boost::filesystem::path d){
|
|
//need to cut off first
|
|
boost::filesystem::directory_iterator dir_iter(d), dir_end;
|
|
std::vector<std::string> filesind;
|
|
for(;dir_iter != dir_end; dir_iter++)
|
|
{
|
|
if(boost::filesystem::is_directory(*dir_iter))
|
|
populateMap(*dir_iter);
|
|
else
|
|
{
|
|
std::string s = dir_iter->path().string();
|
|
std::replace(s.begin(), s.end(), '\\', '/');
|
|
|
|
std::string small;
|
|
if(cutoff < s.size())
|
|
small = s.substr(cutoff, s.size() - cutoff);
|
|
else
|
|
small = s.substr(cutoff - 1, s.size() - cutoff);
|
|
|
|
filesind.push_back(small);
|
|
}
|
|
}
|
|
std::sort(filesind.begin(), filesind.end(), ciLessBoost());
|
|
|
|
std::string small;
|
|
std::string original = d.string();
|
|
std::replace(original.begin(), original.end(), '\\', '/');
|
|
if(cutoff < original.size())
|
|
small = original.substr(cutoff, original.size() - cutoff);
|
|
else
|
|
small = original.substr(cutoff - 1, original.size() - cutoff);
|
|
|
|
m[small] = filesind;
|
|
}
|
|
|
|
bool isCaseSensitive() const { return fsstrict; }
|
|
|
|
// The archive is loaded in the constructor, and never unloaded.
|
|
void load() {}
|
|
void unload() {}
|
|
|
|
bool exists(const String& filename) {
|
|
std::string copy;
|
|
|
|
if (findFile(filename, copy))
|
|
return FileSystemArchive::exists(copy);
|
|
|
|
return false;
|
|
}
|
|
|
|
DataStreamPtr open(const String& filename, bool readonly = true) const
|
|
{
|
|
std::string copy;
|
|
|
|
if (findFile(filename, copy))
|
|
return FileSystemArchive::open(copy, readonly);
|
|
|
|
DataStreamPtr p;
|
|
return p;
|
|
}
|
|
|
|
};
|
|
|
|
class BSAArchive : public Archive
|
|
{
|
|
BSAFile arc;
|
|
|
|
public:
|
|
BSAArchive(const String& name)
|
|
: Archive(name, "BSA")
|
|
{ arc.open(name); }
|
|
|
|
bool isCaseSensitive() const { return false; }
|
|
|
|
// The archive is loaded in the constructor, and never unloaded.
|
|
void load() {}
|
|
void unload() {}
|
|
|
|
DataStreamPtr open(const String& filename, bool recursive = true) const
|
|
{
|
|
// Get a non-const reference to arc. This is a hack and it's all
|
|
// OGRE's fault. You should NOT expect an open() command not to
|
|
// have any side effects on the archive, and hence this function
|
|
// should not have been declared const in the first place.
|
|
BSAFile *narc = (BSAFile*)&arc;
|
|
|
|
String passed = filename;
|
|
if(filename.at(filename.length() - 1) == '*' || filename.at(filename.length() - 1) == '?' || filename.at(filename.length() - 1) == '<'
|
|
|| filename.at(filename.length() - 1) == '"' || filename.at(filename.length() - 1) == '>' || filename.at(filename.length() - 1) == ':'
|
|
|| filename.at(filename.length() - 1) == '|')
|
|
{
|
|
passed = filename.substr(0, filename.length() - 2);
|
|
}
|
|
if(filename.at(filename.length() - 2) == '>' || filename.at(filename.length() - 2) == ':')
|
|
passed = filename.substr(0, filename.length() - 6);
|
|
// Open the file
|
|
StreamPtr strm = narc->getFile(passed.c_str());
|
|
|
|
// Wrap it into an Ogre::DataStream.
|
|
return DataStreamPtr(new Mangle2OgreStream(strm));
|
|
}
|
|
|
|
bool exists(const String& filename) {
|
|
return cexists(filename);
|
|
}
|
|
|
|
// Check if the file exists.
|
|
bool cexists(const String& filename) const {
|
|
String passed = filename;
|
|
if(filename.at(filename.length() - 1) == '*' || filename.at(filename.length() - 1) == '?' || filename.at(filename.length() - 1) == '<'
|
|
|| filename.at(filename.length() - 1) == '"' || filename.at(filename.length() - 1) == '>' || filename.at(filename.length() - 1) == ':'
|
|
|| filename.at(filename.length() - 1) == '|')
|
|
{
|
|
passed = filename.substr(0, filename.length() - 2);
|
|
}
|
|
if(filename.at(filename.length() - 2) == '>' || filename.at(filename.length() - 2) == ':')
|
|
passed = filename.substr(0, filename.length() - 6);
|
|
|
|
return arc.exists(passed.c_str());
|
|
}
|
|
time_t getModifiedTime(const String&) { return 0; }
|
|
|
|
// This is never called as far as I can see.
|
|
StringVectorPtr list(bool recursive = true, bool dirs = false)
|
|
{
|
|
//std::cout << "list(" << recursive << ", " << dirs << ")\n";
|
|
StringVectorPtr ptr = StringVectorPtr(new StringVector());
|
|
return ptr;
|
|
}
|
|
|
|
// Also never called.
|
|
FileInfoListPtr listFileInfo(bool recursive = true, bool dirs = false)
|
|
{
|
|
//std::cout << "listFileInfo(" << recursive << ", " << dirs << ")\n";
|
|
FileInfoListPtr ptr = FileInfoListPtr(new FileInfoList());
|
|
return ptr;
|
|
}
|
|
|
|
// After load() is called, find("*") is called once. It doesn't seem
|
|
// to matter that we return an empty list, exists() gets called on
|
|
// the correct files anyway.
|
|
StringVectorPtr find(const String& pattern, bool recursive = true,
|
|
bool dirs = false)
|
|
{
|
|
//std::cout << "find(" << pattern << ", " << recursive
|
|
// << ", " << dirs << ")\n";
|
|
StringVectorPtr ptr = StringVectorPtr(new StringVector());
|
|
return ptr;
|
|
}
|
|
|
|
/* Gets called once for each of the ogre formats, *.program,
|
|
*.material etc. We ignore all these.
|
|
|
|
However, it's also called by MyGUI to find individual textures,
|
|
and we can't ignore these since many of the GUI textures are
|
|
located in BSAs. So instead we channel it through exists() and
|
|
set up a single-element result list if the file is found.
|
|
*/
|
|
FileInfoListPtr findFileInfo(const String& pattern, bool recursive = true,
|
|
bool dirs = false) const
|
|
{
|
|
FileInfoListPtr ptr = FileInfoListPtr(new FileInfoList());
|
|
|
|
// Check if the file exists (only works for single files - wild
|
|
// cards and recursive search isn't implemented.)
|
|
if(cexists(pattern))
|
|
{
|
|
FileInfo fi;
|
|
fi.archive = this;
|
|
fi.filename = pattern;
|
|
// It apparently doesn't matter that we return bogus
|
|
// information
|
|
fi.path = "";
|
|
fi.compressedSize = fi.uncompressedSize = 0;
|
|
|
|
ptr->push_back(fi);
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
FileInfoListPtr findFileInfo(const String& pattern, bool recursive = true,
|
|
bool dirs = false)
|
|
{
|
|
FileInfoListPtr ptr = FileInfoListPtr(new FileInfoList());
|
|
|
|
// Check if the file exists (only works for single files - wild
|
|
// cards and recursive search isn't implemented.)
|
|
if(cexists(pattern))
|
|
{
|
|
FileInfo fi;
|
|
fi.archive = this;
|
|
fi.filename = pattern;
|
|
// It apparently doesn't matter that we return bogus
|
|
// information
|
|
fi.path = "";
|
|
fi.compressedSize = fi.uncompressedSize = 0;
|
|
|
|
ptr->push_back(fi);
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
};
|
|
|
|
// An archive factory for BSA archives
|
|
class BSAArchiveFactory : public ArchiveFactory
|
|
{
|
|
public:
|
|
const String& getType() const
|
|
{
|
|
static String name = "BSA";
|
|
return name;
|
|
}
|
|
|
|
Archive *createInstance( const String& name )
|
|
{
|
|
return new BSAArchive(name);
|
|
}
|
|
|
|
void destroyInstance( Archive* arch) { delete arch; }
|
|
};
|
|
|
|
class DirArchiveFactory : public FileSystemArchiveFactory
|
|
{
|
|
public:
|
|
const String& getType() const
|
|
{
|
|
static String name = "Dir";
|
|
return name;
|
|
}
|
|
|
|
Archive *createInstance( const String& name )
|
|
{
|
|
return new DirArchive(name);
|
|
}
|
|
|
|
void destroyInstance( Archive* arch) { delete arch; }
|
|
};
|
|
|
|
|
|
static bool init = false;
|
|
static bool init2 = false;
|
|
|
|
static void insertBSAFactory()
|
|
{
|
|
if(!init)
|
|
{
|
|
ArchiveManager::getSingleton().addArchiveFactory( new BSAArchiveFactory );
|
|
init = true;
|
|
}
|
|
}
|
|
|
|
static void insertDirFactory()
|
|
{
|
|
if(!init2)
|
|
{
|
|
ArchiveManager::getSingleton().addArchiveFactory( new DirArchiveFactory );
|
|
init2 = true;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
namespace Bsa
|
|
{
|
|
|
|
// The function below is the only publicly exposed part of this file
|
|
|
|
void addBSA(const std::string& name, const std::string& group)
|
|
{
|
|
insertBSAFactory();
|
|
ResourceGroupManager::getSingleton().
|
|
addResourceLocation(name, "BSA", group);
|
|
}
|
|
|
|
void addDir(const std::string& name, const bool& fs, const std::string& group)
|
|
{
|
|
fsstrict = fs;
|
|
insertDirFactory();
|
|
|
|
ResourceGroupManager::getSingleton().
|
|
addResourceLocation(name, "Dir", group);
|
|
}
|
|
|
|
}
|