mirror of
https://github.com/rwengine/openrw.git
synced 2024-11-07 11:22:45 +01:00
399 lines
9.1 KiB
C++
399 lines
9.1 KiB
C++
#include <renderwure/engine/GTAData.hpp>
|
|
#include <renderwure/engine/GTAEngine.hpp>
|
|
#include <renderwure/loaders/LoaderIPL.hpp>
|
|
#include <renderwure/loaders/LoaderDFF.hpp>
|
|
#include <renderwure/loaders/LoaderIDE.hpp>
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <algorithm>
|
|
|
|
#include <dirent.h>
|
|
#include <sys/types.h>
|
|
|
|
/**
|
|
* Finds the 'real' case for a path, to get around the fact that Rockstar's data is usually the wrong case.
|
|
* @param base The base of the path to start looking from.
|
|
* @param path the lowercase path.
|
|
*/
|
|
std::string findPathRealCase(const std::string& base, const std::string& path)
|
|
{
|
|
#ifdef __unix__
|
|
size_t endslash = path.find("/");
|
|
bool isDirectory = true;
|
|
if(endslash == path.npos) {
|
|
isDirectory = false;
|
|
}
|
|
std::string orgFileName = isDirectory ? path.substr(0, endslash) : path;
|
|
std::string realName;
|
|
|
|
// Open the current "base" path (i.e. the real path)
|
|
DIR* dp = opendir(base.c_str());
|
|
dirent* ep;
|
|
|
|
if( dp != NULL) {
|
|
while( ep = readdir(dp)) {
|
|
realName = ep->d_name;
|
|
std::string lowerRealName = realName;
|
|
std::transform(lowerRealName.begin(), lowerRealName.end(), lowerRealName.begin(), ::tolower);
|
|
if( lowerRealName == orgFileName) {
|
|
closedir(dp);
|
|
if( isDirectory) {
|
|
return findPathRealCase(base + "/" + realName, path.substr(endslash+1));
|
|
}
|
|
else {
|
|
return base + "/" + realName;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return "";
|
|
#else
|
|
// Is anything other than Windows likely to fall here?
|
|
return path;
|
|
#endif
|
|
}
|
|
|
|
// Yet another hack function to fix these paths
|
|
std::string fixPath(std::string path) {
|
|
for( size_t t = 0; t < path.size(); ++t) {
|
|
path[t] = tolower(path[t]);
|
|
if(path[t] == '\\') {
|
|
path[t] = '/';
|
|
}
|
|
}
|
|
return path;
|
|
}
|
|
|
|
|
|
GTAData::GTAData(const std::string& path)
|
|
: datpath(path), engine(nullptr)
|
|
{
|
|
|
|
}
|
|
|
|
void GTAData::load()
|
|
{
|
|
parseDAT(datpath+"/data/default.dat");
|
|
parseDAT(datpath+"/data/gta3.dat");
|
|
|
|
fileLocations.insert({"wheels.DFF", {false, datpath+"/models/Generic/wheels.DFF"}});
|
|
fileLocations.insert({"particle.txd", {false, datpath+"/models/particle.txd"}});
|
|
loadDFF("wheels.DFF");
|
|
loadTXD("particle.txd");
|
|
|
|
loadCarcols(datpath+"/data/carcols.dat");
|
|
loadWeather(datpath+"/data/timecyc.dat");
|
|
loadWaterpro(datpath+"/data/waterpro.dat");
|
|
loadWater(datpath+"/data/water.dat");
|
|
}
|
|
|
|
void GTAData::parseDAT(const std::string& path)
|
|
{
|
|
std::ifstream datfile(path.c_str());
|
|
|
|
if(!datfile.is_open())
|
|
{
|
|
std::cerr << "Failed to open gta.dat" << std::endl;
|
|
}
|
|
else
|
|
{
|
|
for(std::string line, cmd; std::getline(datfile, line);)
|
|
{
|
|
if(line.size() == 0 || line[0] == '#') continue;
|
|
line.erase(line.size()-1);
|
|
|
|
size_t space = line.find_first_of(' ');
|
|
if(space != line.npos)
|
|
{
|
|
cmd = line.substr(0, space);
|
|
if(cmd == "IDE")
|
|
{
|
|
loadIDE(line.substr(space+1));
|
|
}
|
|
else if(cmd == "SPLASH")
|
|
{
|
|
splash = line.substr(space+1);
|
|
}
|
|
else if(cmd == "COLFILE")
|
|
{
|
|
int zone = atoi(line.substr(space+1,1).c_str());
|
|
std::string file = line.substr(space+3);
|
|
loadCOL(zone, file);
|
|
}
|
|
else if(cmd == "IPL")
|
|
{
|
|
std::string fixedpath = fixPath(line.substr(space+1));
|
|
fixedpath = findPathRealCase(datpath, fixedpath);
|
|
loadIPL(fixedpath);
|
|
}
|
|
else if(cmd == "TEXDICTION")
|
|
{
|
|
std::string texpath = line.substr(space+1);
|
|
for( size_t t = 0; t < texpath.size(); ++t) {
|
|
texpath[t] = tolower(texpath[t]);
|
|
if(texpath[t] == '\\') {
|
|
texpath[t] = '/';
|
|
}
|
|
}
|
|
texpath = findPathRealCase(datpath, texpath);
|
|
std::string texname = texpath.substr(texpath.find_last_of("/")+1);
|
|
fileLocations.insert({ texname, { false, texpath }});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void GTAData::loadIDE(const std::string& name)
|
|
{
|
|
std::string lowername = name;
|
|
for(size_t t = 0; t < lowername.size(); ++t)
|
|
{
|
|
lowername[t] = tolower(lowername[t]);
|
|
if(lowername[t] == '\\') {
|
|
lowername[t] = '/';
|
|
}
|
|
}
|
|
|
|
ideLocations.insert({lowername, datpath+"/"+lowername});
|
|
}
|
|
|
|
void GTAData::loadCOL(const size_t zone, const std::string& name)
|
|
{
|
|
LoaderCOL col;
|
|
|
|
std::string realPath = fixPath(name);
|
|
realPath = findPathRealCase(datpath, realPath);
|
|
|
|
if(col.load(realPath)) {
|
|
for( size_t i = 0; i < col.instances.size(); ++i ) {
|
|
collisions[std::string(col.instances[i].header.name)] = std::move(std::unique_ptr<CollisionInstance>( new CollisionInstance(col.instances[i]) ));
|
|
}
|
|
}
|
|
}
|
|
|
|
void GTAData::loadIMG(const std::string& name)
|
|
{
|
|
LoaderIMG imgLoader;
|
|
std::string archivePath = datpath + name;
|
|
|
|
if (imgLoader.load(archivePath)) {
|
|
for (int i = 0; i < imgLoader.getAssetCount(); i++) {
|
|
auto &asset = imgLoader.getAssetInfoByIndex(i);
|
|
|
|
std::string filename = asset.name;
|
|
|
|
if(asset.size == 0)
|
|
{
|
|
std::cerr << "Asset " << filename << " has no size, ignoring." << std::endl;
|
|
}
|
|
else
|
|
{
|
|
// Enter the asset twice..
|
|
fileLocations.insert({ filename, { true, archivePath }});
|
|
for(size_t t = 0; t < filename.size(); ++t)
|
|
{
|
|
filename[t] = tolower(filename[t]);
|
|
}
|
|
fileLocations.insert({ filename, { true, archivePath }});
|
|
}
|
|
}
|
|
archives.insert({archivePath, imgLoader});
|
|
}
|
|
}
|
|
|
|
void GTAData::loadIPL(const std::string& name)
|
|
{
|
|
iplLocations.insert({name, name});
|
|
}
|
|
|
|
enum ColSection {
|
|
Unknown,
|
|
COL,
|
|
CAR,
|
|
CAR3, ///> Used in GTASA, contains extra specular color
|
|
CAR4 ///> Used in GTASA, contains quadruple colors
|
|
};
|
|
|
|
void GTAData::loadCarcols(const std::string& path)
|
|
{
|
|
std::ifstream fstream(path.c_str());
|
|
|
|
std::string line;
|
|
ColSection currentSection = Unknown;
|
|
while( std::getline(fstream, line)) {
|
|
if( line.substr(0, 1) == "#") { // Comment
|
|
continue;
|
|
}
|
|
else if( currentSection == Unknown) {
|
|
if( line.substr(0, 3) == "col") {
|
|
currentSection = COL;
|
|
}
|
|
else if( line.substr(0, 3) == "car") {
|
|
currentSection = CAR;
|
|
}
|
|
}
|
|
else if( line.substr(0, 3) == "end") {
|
|
currentSection = Unknown;
|
|
}
|
|
else {
|
|
if( currentSection == COL) {
|
|
std::string r, g, b;
|
|
std::stringstream ss(line);
|
|
|
|
if( std::getline(ss, r, ',') && std::getline(ss, g, ',') && std::getline(ss, b)) {
|
|
vehicleColours.push_back(glm::vec3(
|
|
atoi(r.c_str())/255.f,
|
|
atoi(g.c_str())/255.f,
|
|
atoi(b.c_str())/255.f
|
|
));
|
|
}
|
|
}
|
|
else if( currentSection == CAR) {
|
|
std::string vehicle, p, s;
|
|
std::stringstream ss(line);
|
|
size_t commapos = line.find(',');
|
|
|
|
std::getline(ss, vehicle, ',');
|
|
std::vector<std::pair<size_t, size_t>> colours;
|
|
|
|
while( std::getline(ss, p, ',') && std::getline(ss, s, ',') ) {
|
|
colours.push_back({ atoi(p.c_str()), atoi(s.c_str()) });
|
|
}
|
|
|
|
vehiclePalettes.insert({vehicle, colours});
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void GTAData::loadWeather(const std::string &path)
|
|
{
|
|
weatherLoader.load(path);
|
|
}
|
|
|
|
void GTAData::loadWaterpro(const std::string& path)
|
|
{
|
|
std::ifstream ifstr(path.c_str());
|
|
|
|
if(ifstr.is_open()) {
|
|
uint32_t numlevels;
|
|
ifstr.read(reinterpret_cast<char*>(&numlevels), sizeof(uint32_t));
|
|
ifstr.read(reinterpret_cast<char*>(&waterHeights), sizeof(float)*48);
|
|
|
|
ifstr.seekg(0x03C4);
|
|
ifstr.read(reinterpret_cast<char*>(&visibleWater), sizeof(float)*64*64);
|
|
}
|
|
}
|
|
|
|
void GTAData::loadWater(const std::string& path)
|
|
{
|
|
std::ifstream ifstr(path.c_str());
|
|
|
|
std::string line;
|
|
while( std::getline(ifstr, line)) {
|
|
if( line[0] == ';') {
|
|
continue;
|
|
}
|
|
|
|
std::stringstream ss(line);
|
|
|
|
std::string a, b, c, d, e;
|
|
|
|
if( std::getline(ss, a, ',') && std::getline(ss, b, ',') && std::getline(ss, c, ',') && std::getline(ss, d, ',') && std::getline(ss, e, ',')) {
|
|
waterRects.push_back({
|
|
atof(a.c_str()),
|
|
atof(b.c_str()),
|
|
atof(c.c_str()),
|
|
atof(d.c_str()),
|
|
atof(e.c_str())
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
void GTAData::loadTXD(const std::string& name)
|
|
{
|
|
if( loadedFiles.find(name) != loadedFiles.end()) {
|
|
return;
|
|
}
|
|
|
|
char* file = loadFile(name);
|
|
if(file) {
|
|
textureLoader.loadFromMemory(file);
|
|
delete[] file;
|
|
}
|
|
}
|
|
|
|
void GTAData::loadDFF(const std::string& name)
|
|
{
|
|
if( loadedFiles.find(name) != loadedFiles.end()) {
|
|
return;
|
|
}
|
|
|
|
char *file = loadFile(name);
|
|
if(file)
|
|
{
|
|
LoaderDFF dffLoader;
|
|
models[name.substr(0, name.size() - 4)] = dffLoader.loadFromMemory(file);
|
|
delete[] file;
|
|
}
|
|
}
|
|
|
|
char* GTAData::loadFile(const std::string& name)
|
|
{
|
|
if( loadedFiles.find(name) != loadedFiles.end()) {
|
|
std::cerr << "File " << name << " already loaded!" << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
auto i = fileLocations.find(name);
|
|
if(i != fileLocations.end())
|
|
{
|
|
if(i->second.archived)
|
|
{
|
|
// Find the archive
|
|
auto ai = archives.find(i->second.path);
|
|
if(ai != archives.end())
|
|
{
|
|
loadedFiles[name] = true;
|
|
return ai->second.loadToMemory(name);
|
|
}
|
|
else
|
|
{
|
|
std::cerr << "Archive not found " << i->second.path << std::endl;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::ifstream dfile(i->second.path);
|
|
if ( ! dfile.is_open()) {
|
|
std::cerr << "Error opening file " << i->second.path << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
dfile.seekg(0, std::ios_base::end);
|
|
size_t length = dfile.tellg();
|
|
dfile.seekg(0);
|
|
char *data = new char[length];
|
|
dfile.read(data, length);
|
|
|
|
loadedFiles[name] = true;
|
|
|
|
return data;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::stringstream err;
|
|
err << "Unable to locate file " << name;
|
|
engine->logError(err.str());
|
|
std::cerr << err.str() << std::endl;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|