usbloadergx/source/usbloader/GameList.cpp

285 lines
7.7 KiB
C++
Raw Normal View History

/****************************************************************************
2010-09-24 02:48:03 +02:00
* 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 <algorithm>
#include <string>
#include <malloc.h>
#include "usbloader/wbfs.h"
#include "settings/newtitles.h"
#include "settings/CSettings.h"
#include "settings/CGameSettings.h"
#include "settings/CGameStatistics.h"
#include "settings/GameTitles.h"
#include "xml/xml.h"
#include "FreeTypeGX.h"
#include "GameList.h"
#include "memory/memory.h"
GameList gameList;
GameList::GameList()
{
}
void GameList::clear()
{
GameFilter.clear();
AvailableSearchChars.clear();
FullGameList.clear();
FilteredList.clear();
//! Clear memory of the vector completely
2010-09-24 02:48:03 +02:00
std::vector<struct discHdr *>().swap(FilteredList);
std::vector<struct discHdr>().swap(FullGameList);
}
2010-09-24 02:48:03 +02:00
struct discHdr * GameList::at(int i)
{
2010-09-24 02:48:03 +02:00
if (i < 0 || i >= (int) FilteredList.size()) return NULL;
return FilteredList[i];
}
struct discHdr * GameList::GetDiscHeader(const char * gameID)
{
for (u32 i = 0; i < FilteredList.size(); ++i)
{
if(strncasecmp(gameID, (const char *) FilteredList[i]->id, 6) == 0)
return FilteredList[i];
}
return NULL;
}
int GameList::ReadGameList()
{
// Clear list
clear();
// Retrieve all stuff from WBFS
u32 cnt;
2010-09-24 02:48:03 +02:00
int ret = WBFS_GetCount(&cnt);
if (ret < 0) return -1;
// We are done here if no games are there
if(cnt == 0)
return 0;
/* Buffer length */
2010-09-24 02:48:03 +02:00
u32 len = sizeof(struct discHdr) * cnt;
/* Allocate memory */
2010-09-24 02:48:03 +02:00
struct discHdr *buffer = (struct discHdr *) allocate_memory( len );
if (!buffer) return -1;
/* Clear buffer */
2010-09-24 02:48:03 +02:00
memset(buffer, 0, len);
/* Get header list */
2010-09-24 02:48:03 +02:00
ret = WBFS_GetHeaders(buffer, cnt, sizeof(struct discHdr));
if (ret < 0)
{
2010-09-24 02:48:03 +02:00
if (buffer) free(buffer);
return -1;
}
2010-09-24 02:48:03 +02:00
FullGameList.resize(cnt);
memcpy(&FullGameList[0], buffer, len);
2010-09-24 02:48:03 +02:00
free(buffer);
return LoadUnfiltered();
}
static bool WCharSortCallback(const wchar_t char1, const wchar_t char2)
{
2010-09-24 02:48:03 +02:00
if (char2 == 0) return true;
if (char1 == 0) return false;
return char2 > char1;
}
2010-09-24 02:48:03 +02:00
int GameList::FilterList(const wchar_t * gameFilter)
{
2010-09-24 02:48:03 +02:00
if (FullGameList.size() == 0) ReadGameList();
if (gameFilter) GameFilter.assign(gameFilter);
FilteredList.clear();
AvailableSearchChars.clear();
2010-09-24 02:48:03 +02:00
for (u32 i = 0; i < FullGameList.size(); ++i)
{
struct discHdr *header = &FullGameList[i];
/* Register game */
2010-09-24 02:48:03 +02:00
NewTitles::Instance()->CheckGame(header->id);
/* Filters */
if (Settings.GameSort & SORT_FAVORITE)
{
GameStatus * GameStats = GameStatistics.GetGameStatus(header->id);
if (!GameStats || GameStats->FavoriteRank == 0) continue;
}
//ignore uLoader cfg "iso". i was told it is "__CFG_" but not confirmed
if (strncasecmp((char*) header->id, "__CFG_", 6) == 0)
continue;
GameCFG * GameConfig = GameSettings.GetGameCFG(header);
if (Settings.parentalcontrol && !Settings.godmode)
{
if (GameConfig && GameConfig->parentalcontrol >= Settings.parentalcontrol)
continue;
}
/* Rating based parental control method */
if (Settings.parentalcontrol != 4 && Settings.godmode == 0)
{
// Check game rating in WiiTDB, since the default Wii parental control setting is enabled
int rating = GameTitles.GetParentalRating((char *) header->id);
if (rating > Settings.parentalcontrol)
continue;
}
//! Per game lock method
if(!Settings.godmode && Settings.lockedgames && GameConfig && GameConfig->Locked)
continue;
wchar_t *gameName = charToWideChar(GameTitles.GetTitle(header));
2010-09-24 02:48:03 +02:00
if (gameName && *GameFilter.c_str())
{
2010-09-24 02:48:03 +02:00
if (wcsnicmp(gameName, GameFilter.c_str(), GameFilter.size()) != 0)
{
2010-09-24 02:48:03 +02:00
delete[] gameName;
continue;
}
}
2010-09-24 02:48:03 +02:00
if (gameName)
{
2010-09-24 02:48:03 +02:00
if (wcslen(gameName) > GameFilter.size() && AvailableSearchChars.find(gameName[GameFilter.size()])
== std::string::npos) AvailableSearchChars.push_back(gameName[GameFilter.size()]);
delete[] gameName;
}
2010-09-24 02:48:03 +02:00
FilteredList.push_back(header);
}
NewTitles::Instance()->Save();
2010-09-24 02:48:03 +02:00
AvailableSearchChars.push_back(L'\0');
2010-09-24 02:48:03 +02:00
if (FilteredList.size() < 2) AvailableSearchChars.clear();
SortList();
return FilteredList.size();
}
int GameList::LoadUnfiltered()
{
2010-09-24 02:48:03 +02:00
if (FullGameList.size() == 0) ReadGameList();
GameFilter.clear();
AvailableSearchChars.clear();
FilteredList.clear();
2010-09-24 02:48:03 +02:00
for (u32 i = 0; i < FullGameList.size(); ++i)
{
struct discHdr *header = &FullGameList[i];
/* Register game */
2010-09-24 02:48:03 +02:00
NewTitles::Instance()->CheckGame(header->id);
wchar_t *gameName = charToWideChar(GameTitles.GetTitle(header));
2010-09-24 02:48:03 +02:00
if (gameName)
{
2010-09-24 02:48:03 +02:00
if (wcslen(gameName) > GameFilter.size() && AvailableSearchChars.find(gameName[GameFilter.size()])
== std::string::npos) AvailableSearchChars.push_back(gameName[GameFilter.size()]);
delete[] gameName;
}
2010-09-24 02:48:03 +02:00
FilteredList.push_back(header);
}
NewTitles::Instance()->Save();
2010-09-24 02:48:03 +02:00
AvailableSearchChars.push_back(L'\0');
2010-09-24 02:48:03 +02:00
if (FilteredList.size() < 2) AvailableSearchChars.clear();
SortList();
return FilteredList.size();
}
void GameList::SortList()
{
2010-09-24 02:48:03 +02:00
if (FilteredList.size() < 2) return;
if (Settings.GameSort & SORT_PLAYCOUNT)
{
2010-09-24 02:48:03 +02:00
std::sort(FilteredList.begin(), FilteredList.end(), PlaycountSortCallback);
}
else if(Settings.GameSort & SORT_RANKING)
{
std::sort(FilteredList.begin(), FilteredList.end(), RankingSortCallback);
}
else
{
2010-09-24 02:48:03 +02:00
std::sort(FilteredList.begin(), FilteredList.end(), NameSortCallback);
}
2010-09-24 02:48:03 +02:00
if (AvailableSearchChars.size() > 1) std::sort(AvailableSearchChars.begin(), AvailableSearchChars.end(),
WCharSortCallback);
}
bool GameList::NameSortCallback(const struct discHdr *a, const struct discHdr *b)
{
return (strcasecmp(GameTitles.GetTitle((struct discHdr *) a), GameTitles.GetTitle((struct discHdr *) b)) < 0);
}
bool GameList::PlaycountSortCallback(const struct discHdr *a, const struct discHdr *b)
{
int count1 = GameStatistics.GetPlayCount(a->id);
int count2 = GameStatistics.GetPlayCount(b->id);
2010-09-24 02:48:03 +02:00
if (count1 == count2) return NameSortCallback(a, b);
return (count1 > count2);
}
bool GameList::RankingSortCallback(const struct discHdr *a, const struct discHdr *b)
{
int fav1 = GameStatistics.GetFavoriteRank(a->id);
int fav2 = GameStatistics.GetFavoriteRank(b->id);
2010-09-24 02:48:03 +02:00
if (fav1 == fav2) return NameSortCallback(a, b);
return (fav1 > fav2);
}