/**
* Initializes a new savegame with
* everything available.
*/
void NewBattleState::initSave()
{
const Mod *mod = _game->getMod();
SavedGame *save = new SavedGame();
Base *base = new Base(mod);
const YAML::Node &starter = _game->getMod()->getStartingBase();
base->load(starter, save, true, true);
save->getBases()->push_back(base);
// Kill everything we don't want in this base
for (std::vector<Soldier*>::iterator i = base->getSoldiers()->begin(); i != base->getSoldiers()->end(); ++i) delete (*i);
base->getSoldiers()->clear();
for (std::vector<Craft*>::iterator i = base->getCrafts()->begin(); i != base->getCrafts()->end(); ++i) delete (*i);
base->getCrafts()->clear();
base->getStorageItems()->getContents()->clear();
_craft = new Craft(mod->getCraft(_crafts[_cbxCraft->getSelected()]), base, 1);
base->getCrafts()->push_back(_craft);
// Generate soldiers
for (int i = 0; i < 30; ++i)
{
int randomType = RNG::generate(0, _game->getMod()->getSoldiersList().size() - 1);
Soldier *soldier = mod->genSoldier(save, _game->getMod()->getSoldiersList().at(randomType));
for (int n = 0; n < 5; ++n)
{
if (RNG::percent(70))
continue;
soldier->promoteRank();
UnitStats* stats = soldier->getCurrentStats();
stats->tu += RNG::generate(0, 5);
stats->stamina += RNG::generate(0, 5);
stats->health += RNG::generate(0, 5);
stats->bravery += RNG::generate(0, 5);
stats->reactions += RNG::generate(0, 5);
stats->firing += RNG::generate(0, 5);
stats->throwing += RNG::generate(0, 5);
stats->strength += RNG::generate(0, 5);
stats->psiStrength += RNG::generate(0, 5);
stats->melee += RNG::generate(0, 5);
stats->psiSkill += RNG::generate(0, 20);
}
UnitStats* stats = soldier->getCurrentStats();
stats->bravery = (int)ceil(stats->bravery / 10.0) * 10; // keep it a multiple of 10
base->getSoldiers()->push_back(soldier);
if (i < _craft->getRules()->getSoldiers())
soldier->setCraft(_craft);
}
// Generate items
const std::vector<std::string> &items = mod->getItemsList();
for (std::vector<std::string>::const_iterator i = items.begin(); i != items.end(); ++i)
{
RuleItem *rule = _game->getMod()->getItem(*i);
if (rule->getBattleType() != BT_CORPSE && rule->isRecoverable())
{
base->getStorageItems()->addItem(*i, 1);
if (rule->getBattleType() != BT_NONE && !rule->isFixed() && rule->getBigSprite() > -1)
{
_craft->getItems()->addItem(*i, 1);
}
}
}
// Add research
const std::vector<std::string> &research = mod->getResearchList();
for (std::vector<std::string>::const_iterator i = research.begin(); i != research.end(); ++i)
{
save->addFinishedResearchSimple(mod->getResearch(*i));
}
_game->setSavedGame(save);
cbxMissionChange(0);
}
/**
* Loads new battle data from a YAML file.
* @param filename YAML filename.
*/
void NewBattleState::load(const std::string &filename)
{
std::string s = Options::getMasterUserFolder() + filename + ".cfg";
if (!CrossPlatform::fileExists(s))
{
initSave();
}
else
{
try
{
YAML::Node doc = YAML::LoadFile(s);
_cbxMission->setSelected(std::min(doc["mission"].as<size_t>(0), _missionTypes.size() - 1));
cbxMissionChange(0);
_cbxCraft->setSelected(std::min(doc["craft"].as<size_t>(0), _crafts.size() - 1));
_slrDarkness->setValue(doc["darkness"].as<size_t>(0));
_cbxTerrain->setSelected(std::min(doc["terrain"].as<size_t>(0), _terrainTypes.size() - 1));
cbxTerrainChange(0);
_cbxAlienRace->setSelected(std::min(doc["alienRace"].as<size_t>(0), _alienRaces.size() - 1));
_cbxDifficulty->setSelected(doc["difficulty"].as<size_t>(0));
_slrAlienTech->setValue(doc["alienTech"].as<size_t>(0));
if (doc["base"])
{
const Mod *mod = _game->getMod();
SavedGame *save = new SavedGame();
Base *base = new Base(mod);
base->load(doc["base"], save, false);
save->getBases()->push_back(base);
// Add research
const std::vector<std::string> &research = mod->getResearchList();
for (std::vector<std::string>::const_iterator i = research.begin(); i != research.end(); ++i)
{
save->addFinishedResearchSimple(mod->getResearch(*i));
}
// Generate items
base->getStorageItems()->getContents()->clear();
const std::vector<std::string> &items = mod->getItemsList();
for (std::vector<std::string>::const_iterator i = items.begin(); i != items.end(); ++i)
{
RuleItem *rule = _game->getMod()->getItem(*i);
if (rule->getBattleType() != BT_CORPSE && rule->isRecoverable())
{
base->getStorageItems()->addItem(*i, 1);
}
}
// Fix invalid contents
if (base->getCrafts()->empty())
{
std::string craftType = _crafts[_cbxCraft->getSelected()];
_craft = new Craft(_game->getMod()->getCraft(craftType), base, save->getId(craftType));
base->getCrafts()->push_back(_craft);
}
else
{
_craft = base->getCrafts()->front();
for (std::map<std::string, int>::iterator i = _craft->getItems()->getContents()->begin(); i != _craft->getItems()->getContents()->end(); ++i)
{
RuleItem *rule = _game->getMod()->getItem(i->first);
if (!rule)
{
i->second = 0;
}
}
}
_game->setSavedGame(save);
}
else
{
initSave();
}
}
catch (YAML::Exception &e)
{
Log(LOG_WARNING) << e.what();
initSave();
}
}
}
