WiiFlow_Lite/source/gui/GameTDB.cpp

1024 lines
18 KiB
C++

/****************************************************************************
* Copyright (C) 2010
* by Dimok
*
* This software is provided 'as-is', without any express or implied
* warranty. In no event will the authors be held liable for any
* damages arising from the use of this software.
*
* Permission is granted to anyone to use this software for any
* purpose, including commercial applications, and to alter it and
* redistribute it freely, subject to the following restrictions:
*
* 1. The origin of this software must not be misrepresented; you
* must not claim that you wrote the original software. If you use
* this software in a product, an acknowledgment in the product
* documentation would be appreciated but is not required.
*
* 2. Altered source versions must be plainly marked as such, and
* must not be misrepresented as being the original software.
*
* 3. This notice may not be removed or altered from any source
* distribution.
***************************************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string>
#include <cstring>
#include "GameTDB.hpp"
#include "video.hpp"
#include "text.hpp"
#include "config/config.hpp"
#include "gecko/gecko.hpp"
#include "memory/mem2.hpp"
#include "types.h"
#include "gui/coverflow.hpp"
#define NAME_OFFSET_DB "gametdb_offsets.bin"
#define MAXREADSIZE 1024*1024 //Cache size only for parsing the offsets: 1MB
typedef struct _ReplaceStruct
{
const char *orig;
char replace;
short size;
} ReplaceStruct;
//! More replacements can be added if needed
static const ReplaceStruct Replacements[] =
{
{ "&gt;", '>', 4 },
{ "&lt;", '<', 4 },
{ "&quot;", '\"', 6 },
{ "&apos;", '\'', 6 },
{ "&amp;", '&', 5 },
{ NULL, '\0', 0 }
};
GameTDB::GameTDB()
: isLoaded(false), file(0), LangCode("EN"), GameNodeCache(NULL)
{
}
GameTDB::GameTDB(const char *filepath)
: isLoaded(false), file(0), LangCode("EN"), GameNodeCache(NULL)
{
OpenFile(filepath);
}
GameTDB::~GameTDB()
{
CloseFile();
}
bool GameTDB::OpenFile(const char *filepath)
{
if(!filepath)
return false;
file = fopen(filepath, "rb");
if(file)
{
int pos;
string OffsetsPath = filepath;
if((pos = OffsetsPath.find_last_of('/')) != (int) string::npos)
OffsetsPath[pos] = '\0';
else
OffsetsPath.clear(); //! Relative path
LoadGameOffsets(OffsetsPath.c_str());
}
isLoaded = (file != NULL);
return isLoaded;
}
void GameTDB::CloseFile()
{
OffsetMap.clear();
vector<GameOffsets>().swap(OffsetMap);
if(GameNodeCache)
MEM2_free(GameNodeCache);
GameNodeCache = NULL;
if(file)
fclose(file);
file = NULL;
}
bool GameTDB::LoadGameOffsets(const char *path)
{
if(!path)
return false;
string OffsetDBPath = path;
if(strlen(path) > 0 && path[strlen(path)-1] != '/')
OffsetDBPath += '/';
OffsetDBPath += NAME_OFFSET_DB;
FILE *fp = fopen(OffsetDBPath.c_str(), "rb");
if(!fp)
{
bool result = ParseFile();
if(result)
SaveGameOffsets(OffsetDBPath.c_str());
return result;
}
u64 ExistingVersion = GetGameTDBVersion();
u64 Version = 0;
u32 NodeCount = 0;
fread(&Version, 1, sizeof(Version), fp);
if(ExistingVersion != Version)
{
fclose(fp);
bool result = ParseFile();
if(result)
SaveGameOffsets(OffsetDBPath.c_str());
return result;
}
fread(&NodeCount, 1, sizeof(NodeCount), fp);
if(NodeCount == 0)
{
fclose(fp);
bool result = ParseFile();
if(result)
SaveGameOffsets(OffsetDBPath.c_str());
return result;
}
OffsetMap.resize(NodeCount);
if(fread(&OffsetMap[0], 1, NodeCount*sizeof(GameOffsets), fp) != NodeCount*sizeof(GameOffsets))
{
fclose(fp);
bool result = ParseFile();
if(result)
SaveGameOffsets(OffsetDBPath.c_str());
return result;
}
fclose(fp);
return true;
}
bool GameTDB::SaveGameOffsets(const char *path)
{
if(OffsetMap.size() == 0 || !path)
return false;
FILE *fp = fopen(path, "wb");
if(!fp)
return false;
u64 ExistingVersion = GetGameTDBVersion();
u32 NodeCount = OffsetMap.size();
if(fwrite(&ExistingVersion, 1, sizeof(ExistingVersion), fp) != sizeof(ExistingVersion))
{
fclose(fp);
return false;
}
if(fwrite(&NodeCount, 1, sizeof(NodeCount), fp) != sizeof(NodeCount))
{
fclose(fp);
return false;
}
if(fwrite(&OffsetMap[0], 1, NodeCount*sizeof(GameOffsets), fp) != NodeCount*sizeof(GameOffsets))
{
fclose(fp);
return false;
}
fclose(fp);
return true;
}
u64 GameTDB::GetGameTDBVersion()
{
if(!file)
return 0;
char TmpText[1024];
if(GetData(TmpText, 0, sizeof(TmpText)) < 0)
return 0;
char *VersionText = GetNodeText(TmpText, "<GameTDB version=\"", "/>");
if(!VersionText)
return 0;
return strtoull(VersionText, NULL, 10);
}
int GameTDB::GetData(char *data, int offset, int size)
{
if(!file || !data)
return -1;
fseek(file, offset, SEEK_SET);
return fread(data, 1, size, file);
}
char *GameTDB::LoadGameNode(const char *id)
{
u32 read = 0;
GameOffsets *offset = this->GetGameOffset(id);
if(!offset)
return NULL;
char *data = (char*)MEM2_alloc(offset->nodesize+1);
if(!data)
return NULL;
if((read = GetData(data, offset->gamenode, offset->nodesize)) != offset->nodesize)
{
MEM2_free(data);
return NULL;
}
data[read] = '\0';
return data;
}
char *GameTDB::GetGameNode(const char *id)
{
char *data = NULL;
if(GameNodeCache != NULL && strncmp(id, GameIDCache, strlen(GameIDCache)) == 0)
{
data = (char*)MEM2_alloc(strlen(GameNodeCache)+1);
if(data)
strcpy(data, GameNodeCache);
}
else
{
if(GameNodeCache)
MEM2_free(GameNodeCache);
GameNodeCache = LoadGameNode(id);
if(GameNodeCache)
{
snprintf(GameIDCache, sizeof(GameIDCache), id);
data = (char*)MEM2_alloc(strlen(GameNodeCache)+1);
if(data)
strcpy(data, GameNodeCache);
}
}
return data;
}
GameOffsets *GameTDB::GetGameOffset(const char *gameID)
{
for(u32 i = 0; i < OffsetMap.size(); ++i)
{
if(strncmp(gameID, OffsetMap[i].gameID, strlen(OffsetMap[i].gameID)) == 0)
return &OffsetMap[i];
}
return 0;
}
static inline char *CleanText(char * &in_text)
{
if(!in_text)
return NULL;
const char *ptr = in_text;
char *text = in_text;
while(*ptr != '\0')
{
for(int i = 0; Replacements[i].orig != 0; ++i)
{
if(strncmp(ptr, Replacements[i].orig, Replacements[i].size) == 0)
{
ptr += Replacements[i].size;
*text = Replacements[i].replace;
++text;
i = 0;
continue;
}
}
if(*ptr == '\r')
{
++ptr;
continue;
}
*text = *ptr;
++ptr;
++text;
}
*text = '\0';
return in_text;
}
char *GameTDB::GetNodeText(char *data, const char *nodestart, const char *nodeend)
{
if(!data || !nodestart || !nodeend)
return NULL;
char *position = strstr(data, nodestart);
if(!position)
return NULL;
position += strlen(nodestart);
char *end = strstr(position, nodeend);
if(!end)
return NULL;
*end = '\0';
return CleanText(position);
}
char *GameTDB::SeekLang(char *text, const char *langcode)
{
if(!text || !langcode)
return NULL;
char *ptr = text;
while((ptr = strstr(ptr, "<locale lang=")) != NULL)
{
ptr += strlen("<locale lang=\"");
if(strncmp(ptr, langcode, strlen(langcode)) == 0)
{
//! Cut off all the other languages
char *end = strstr(ptr, "</locale>");
if(!end)
return NULL;
end += strlen("</locale>");
*end = '\0';
return ptr;
}
}
return NULL;
}
bool GameTDB::ParseFile()
{
OffsetMap.clear();
if(!file)
return false;
char *Line = (char*)MEM2_alloc(MAXREADSIZE+1);
if(!Line)
return false;
bool readnew = false;
int i, currentPos = 0;
int read = 0;
const char *gameNode = NULL;
const char *idNode = NULL;
const char *gameEndNode = NULL;
while((read = GetData(Line, currentPos, MAXREADSIZE)) > 0)
{
gameNode = Line;
readnew = false;
//! Ensure the null termination at the end
Line[read] = '\0';
while((gameNode = strstr(gameNode, "<game name=\"")) != NULL)
{
idNode = strstr(gameNode, "<id>");
gameEndNode = strstr(gameNode, "</game>");
if(!idNode || !gameEndNode)
{
//! We are in the middle of the game node, reread complete node and more
currentPos += (gameNode-Line);
fseek(file, currentPos, SEEK_SET);
readnew = true;
break;
}
idNode += strlen("<id>");
gameEndNode += strlen("</game>");
int size = OffsetMap.size();
OffsetMap.resize(size+1);
for(i = 0; i < 7 && *idNode != '<'; ++i, ++idNode)
OffsetMap[size].gameID[i] = *idNode;
OffsetMap[size].gameID[i] = '\0';
OffsetMap[size].gamenode = currentPos+(gameNode-Line);
OffsetMap[size].nodesize = (gameEndNode-gameNode);
gameNode = gameEndNode;
}
if(readnew)
continue;
currentPos += read;
}
MEM2_free(Line);
return true;
}
bool GameTDB::GetTitle(const char *id, const char * &title, bool plugin)
{
title = NULL;
if(id == NULL)
return false;
char *data = GetGameNode(id);
if(data == NULL)
return false;
char *language = SeekLang(data, LangCode.c_str());
if(language == NULL)
{
language = SeekLang(data, "EN");
if(language == NULL)
{
MEM2_free(data);
return false;
}
}
title = GetNodeText(language, "<title>", "</title>");
if(title == NULL && plugin)// If current language doesn't have Title then try in English
{
language = SeekLang(data, "EN");
if(language == NULL)
{
MEM2_free(data);
return false;
}
title = GetNodeText(language, "<title>", "</title>");
}
MEM2_free(data);
if(title == NULL)
return false;
return true;
}
/* used for plugins database files */
/* gets no-intro filename used for snapshots and cart/disk images */
bool GameTDB::GetName(const char *id, const char * &name)
{
name = NULL;
if(!id)
return false;
char *data = GetGameNode(id);
if(!data)
return false;
name = GetNodeText(data, "<game name=\"", "\"");
MEM2_free(data);
if(name == NULL)
return false;
return true;
}
bool GameTDB::GetSynopsis(const char *id, const char * &synopsis)
{
synopsis = NULL;
if(!id)
return false;
char *data = GetGameNode(id);
if(!data)
return false;
char *language = SeekLang(data, LangCode.c_str());
if(language == NULL)
{
language = SeekLang(data, "EN");
if(language == NULL)
{
MEM2_free(data);
return false;
}
}
synopsis = GetNodeText(language, "<synopsis>", "</synopsis>");
if(CoverFlow.getHdr()->type == TYPE_PLUGIN)
{
// Default to English
if(synopsis == NULL)
{
language = SeekLang(data, "EN");
if(language == NULL)
{
MEM2_free(data);
return false;
}
synopsis = GetNodeText(language, "<synopsis>", "</synopsis>");
}
}
MEM2_free(data);
if(synopsis == NULL)
return false;
return true;
}
bool GameTDB::GetRegion(const char *id, const char * &region)
{
region = NULL;
if(!id)
return false;
char *data = GetGameNode(id);
if(!data)
return false;
region = GetNodeText(data, "<region>", "</region>");
MEM2_free(data);
if(region == NULL)
return false;
return true;
}
bool GameTDB::GetDeveloper(const char *id, const char * &dev)
{
dev = NULL;
if(id == NULL)
return false;
char *data = GetGameNode(id);
if(data == NULL)
return false;
dev = GetNodeText(data, "<developer>", "</developer>");
MEM2_free(data);
if(dev == NULL)
return false;
return true;
}
bool GameTDB::GetPublisher(const char *id, const char * &pub)
{
pub = NULL;
if(!id)
return false;
char *data = GetGameNode(id);
if(data == NULL)
return false;
pub = GetNodeText(data, "<publisher>", "</publisher>");
MEM2_free(data);
if(pub == NULL)
return false;
return true;
}
u32 GameTDB::GetPublishDate(const char *id)
{
if(!id)
return 0;
char *data = GetGameNode(id);
if(!data)
return 0;
char *year_string = GetNodeText(data, "<date year=\"", "/>");
if(!year_string)
{
MEM2_free(data);
return 0;
}
u32 year, day, month;
year = atoi(year_string);
char *month_string = strstr(year_string, "month=\"");
if(!month_string)
{
MEM2_free(data);
return 0;
}
month_string += strlen("month=\"");
month = atoi(month_string);
char *day_string = strstr(month_string, "day=\"");
if(!day_string)
{
MEM2_free(data);
return 0;
}
day_string += strlen("day=\"");
day = atoi(day_string);
MEM2_free(data);
return ((year & 0xFFFF) << 16 | (month & 0xFF) << 8 | (day & 0xFF));
}
bool GameTDB::GetGenres(const char *id, const char * &gen)
{
gen = NULL;
if(CoverFlow.getHdr()->type == TYPE_PLUGIN)
{
if(!id)
return false;
char *data = GetGameNode(id);
if(!data)
return false;
char *language = SeekLang(data, LangCode.c_str());
if(language == NULL)
{
language = SeekLang(data, "EN");
if(language == NULL)
{
MEM2_free(data);
return false;
}
}
gen = GetNodeText(language, "<genre>", "</genre>");
// If not found try in English
if(gen == NULL)
{
language = SeekLang(data, "EN");
if(language == NULL)
{
MEM2_free(data);
return false;
}
gen = GetNodeText(language, "<genre>", "</genre>");
}
MEM2_free(data);
if(gen == NULL)
return false;
return true;
}
if(id == NULL)
return false;
char *data = GetGameNode(id);
if(data == NULL)
return false;
gen = GetNodeText(data, "<genre>", "</genre>");
MEM2_free(data);
if(gen == NULL)
return false;
return true;
}
const char *GameTDB::RatingToString(int rating)
{
switch(rating)
{
case GAMETDB_RATING_TYPE_CERO:
return "CERO";
case GAMETDB_RATING_TYPE_ESRB:
return "ESRB";
case GAMETDB_RATING_TYPE_PEGI:
return "PEGI";
case GAMETDB_RATING_TYPE_GRB:
return "GRB";
default:
break;
}
return NULL;
}
int GameTDB::GetRating(const char *id)
{
int rating = -1;
if(!id)
return rating;
char *data = GetGameNode(id);
if(!data)
return rating;
char *rating_text = GetNodeText(data, "<rating type=\"", "/>");
if(!rating_text)
{
MEM2_free(data);
return rating;
}
if(strncmp(rating_text, "CERO", 4) == 0)
rating = GAMETDB_RATING_TYPE_CERO;
else if(strncmp(rating_text, "ESRB", 4) == 0)
rating = GAMETDB_RATING_TYPE_ESRB;
else if(strncmp(rating_text, "PEGI", 4) == 0)
rating = GAMETDB_RATING_TYPE_PEGI;
else if(strncmp(rating_text, "GRB", 4) == 0)
rating = GAMETDB_RATING_TYPE_GRB;
MEM2_free(data);
return rating;
}
bool GameTDB::GetRatingValue(const char *id, const char * &rating_value)
{
rating_value = NULL;
if(!id)
return false;
char *data = GetGameNode(id);
if(!data)
return false;
char *rating_text = GetNodeText(data, "<rating type=\"", "/>");
if(!rating_text)
{
MEM2_free(data);
return false;
}
rating_value = GetNodeText(rating_text, "value=\"", "\"");
MEM2_free(data);
if(rating_value == NULL)
return false;
return true;
}
int GameTDB::GetRatingDescriptors(const char *id, vector<string> & desc_list)
{
desc_list.clear();
if(!id)
return -1;
char *data = GetGameNode(id);
if(!data)
return -1;
char *descriptor_text = GetNodeText(data, "<descriptor>", "</rating>");
if(!descriptor_text)
{
MEM2_free(data);
return -1;
}
u32 list_num = 0;
while(*descriptor_text != '\0')
{
if(strncmp(descriptor_text, "</descriptor>", strlen("</descriptor>")) == 0)
{
desc_list[list_num].push_back('\0');
descriptor_text = strstr(descriptor_text, "<descriptor>");
if(!descriptor_text)
break;
descriptor_text += strlen("<descriptor>");
list_num++;
}
if(list_num >= desc_list.size())
desc_list.resize(list_num+1);
desc_list[list_num].push_back(*descriptor_text);
++descriptor_text;
}
MEM2_free(data);
return desc_list.size();
}
int GameTDB::GetWifiPlayers(const char *id)
{
int players = -1;
if(!id)
return players;
char *data = GetGameNode(id);
if(!data)
return players;
char *PlayersNode = GetNodeText(data, "<wi-fi players=\"", "\">");
if(!PlayersNode)
{
MEM2_free(data);
return players;
}
players = atoi(PlayersNode);
return players;
}
int GameTDB::GetWifiFeatures(const char *id, vector<string> & feat_list)
{
feat_list.clear();
if(!id)
return -1;
char *data = GetGameNode(id);
if(!data)
return -1;
char *feature_text = GetNodeText(data, "<feature>", "</wi-fi>");
if(!feature_text)
{
MEM2_free(data);
return -1;
}
u32 list_num = 0;
while(*feature_text != '\0')
{
if(strncmp(feature_text, "</feature>", strlen("</feature>")) == 0)
{
feat_list[list_num].push_back('\0');
feature_text = strstr(feature_text, "<feature>");
if(!feature_text)
break;
feature_text += strlen("<feature>");
list_num++;
}
if(list_num >= feat_list.size())
feat_list.resize(list_num+1);
if(feat_list[list_num].size() == 0)
feat_list[list_num].push_back(toupper((int)*feature_text));
else
feat_list[list_num].push_back(*feature_text);
++feature_text;
}
MEM2_free(data);
return feat_list.size();
}
int GameTDB::GetPlayers(const char *id)
{
int players = -1;
if(!id)
return players;
char *data = GetGameNode(id);
if(!data)
return players;
char *PlayersNode = GetNodeText(data, "<input players=\"", "\">");
if(!PlayersNode)
{
MEM2_free(data);
return players;
}
players = atoi(PlayersNode);
return players;
}
int GameTDB::GetAccessories(const char *id, vector<Accessory> & acc_list)
{
acc_list.clear();
if(!id)
return -1;
char *data = GetGameNode(id);
if(!data)
return -1;
char *ControlsNode = GetNodeText(data, "<control type=\"", "</input>");
if(!ControlsNode)
{
MEM2_free(data);
return -1;
}
u32 list_num = 0;
while(ControlsNode && *ControlsNode != '\0')
{
if(list_num >= acc_list.size())
acc_list.resize(list_num+1);
for(const char *ptr = ControlsNode; *ptr != '"' && *ptr != '\0'; ptr++)
acc_list[list_num].Name.push_back(*ptr);
acc_list[list_num].Name.push_back('\0');
char *requiredField = strstr(ControlsNode, "required=\"");
if(!requiredField)
{
MEM2_free(data);
return -1;
}
requiredField += strlen("required=\"");
acc_list[list_num].Required = strncmp(requiredField, "true", 4) == 0;
ControlsNode = strstr(requiredField, "<control type=\"");
if(ControlsNode)
ControlsNode += strlen("<control type=\"");
list_num++;
}
MEM2_free(data);
return acc_list.size();
}
u32 GameTDB::GetCaseColor(const char *id)
{
u32 color = -1;// -1 = ffffffff
if(!id)
return color;
char *data = GetGameNode(id);
if(!data)
return color;
char *ColorNode = GetNodeText(data, "<case color=\"", "\"");
if(!ColorNode)
{
MEM2_free(data);
return color;
}
color = strtoul(ColorNode, NULL, 16);
//if(color != 0xffffffff)
// gprintf("GameTDB: Found alternate color(%x) for: %s\n", color, id);
MEM2_free(data);
return color;
}
int GameTDB::GetCaseVersions(const char *id)
{
int altcase = -1;
if(!id)
return altcase;
char *data = GetGameNode(id);
if(!data)
{
//gprintf("GameTDB: GameNode for %s not found\n", id);
return altcase;
}
char *PlayersNode = GetNodeText(data, "case versions=\"", "\"");
if(!PlayersNode)
{
MEM2_free(data);
return altcase;
}
altcase = atoi(PlayersNode);
return altcase;
}
bool GameTDB::IsLoaded()
{
return isLoaded;
}