import settings.map_lib;
import settings.game_settings;
import empire_data;
import ftl;
import orbitals;
import object_creation;
import traits;
import maps;
from empire import Creeps, Pirates, majorEmpireCount, initEmpireDesigns;
import void addModifierToEmpire(Empire@ emp, const string& spec) from "bonus_effects";

#priority init 5000
void init() {
	if(isLoadedSave)
		return;

	//Modify settings based on maps
	for(uint i = 0, cnt = gameSettings.galaxies.length; i < cnt; ++i) {
		Map@ desc = getMap(gameSettings.galaxies[i].map_id);
		if(desc !is null)
			desc.modSettings(gameSettings);
	}

	//Generate the empires
	EmpirePortraitCreation portraits;
	uint empCnt = gameSettings.empires.length;
	majorEmpireCount = empCnt;
	for(uint i = 0; i < empCnt; ++i) {
		EmpireSettings@ settings = gameSettings.empires[i];

		//Create the empire
		Empire@ emp = Empire();
		emp.name = settings.name;
		emp.major = true;
		emp.team = settings.team;
		emp.handicap = settings.handicap;
		emp.effectorSkin = settings.effectorSkin;
		emp.ContactMask.value = emp.mask;
		emp.TradeMask.value = emp.mask;
		@emp.shipset = getShipset(settings.shipset);
		emp.RaceName = settings.raceName;
		emp.cheatLevel = 0;
		if(emp.shipset is null || !emp.shipset.available)
			@emp.shipset = getShipset(DEFAULT_SHIPSET);

		//Check if we have a player
		if(settings.playerId != -1) {
			Player@ pl = getPlayer(settings.playerId);
			if(pl.emp is null && emp.player is null)
				pl.linkEmpire(emp);
		}

		//Create the portrait from settings data
		portraits.apply(settings, emp);

		//First empire created is the player
		if(playerEmpire is null)
			@playerEmpire = emp;

		//Add all the traits
		int points = settings.getTraitPoints();
		for(uint n = 0, ncnt = settings.traits.length; n < ncnt; ++n) {
			auto@ trait = settings.traits[n];

			//Skip traits with conflicts
			if(trait.hasConflicts(settings.traits))
				continue;

			//Skip traits if the points are invalid
			if(points < 0) {
				if(trait.cost > 0) {
					points += trait.cost;
					continue;
				}
			}

			emp.addTrait(trait.id);
		}

		//Apply generic AI cheats
		if(settings.type != ET_Player) {
			if(settings.cheatWealth > 0) {
				double factor = settings.cheatWealth;
				emp.modResearchRate(0.75 * factor);
				emp.modTotalBudget(100 * factor);
				emp.modEnergyIncome(0.5 * factor);
				emp.modDefenseRate(1.0 * factor / 60.0);
				emp.modInfluenceIncome(0.5 * factor);
				emp.FactoryLaborMod += factor;
				emp.PopulationGrowthFactor += 0.1 * factor;

				emp.modFTLCapacity(125.0 * factor);
				emp.modFTLIncome(0.5 * factor);

				emp.cheatLevel += settings.cheatWealth / 10;
			}
			if(settings.cheatAbundance > 0) {
				double factor = 1.0 + double(settings.cheatAbundance) * 0.5;
				emp.MoneyGenerationFactor *= factor;
				emp.LaborGenerationFactor *= factor;
				emp.ResearchGenerationFactor *= factor;
				emp.EnergyGenerationFactor *= factor;
				emp.DefenseGenerationFactor *= factor;
				emp.PopulationGrowthFactor *= factor;
				emp.modInfluenceFactor(sqrt(factor) - 1.0);

				emp.cheatLevel += settings.cheatAbundance;
			}
			if(settings.cheatStrength > 0) {
				double factor = 1.0 + double(settings.cheatStrength) * 0.5;
				factor = sqrt(factor);

				addModifierToEmpire(emp, "HpFactor("+factor+")");
				addModifierToEmpire(emp, "tag/Weapon::DamageFactor("+factor+")");

				emp.cheatLevel += settings.cheatStrength;
			}
			if(settings.aiFlags & AIF_CheatPrivileged != 0) {
				const Trait@ trait;
				uint checked = 0;
				for(uint j = 0, jcnt = getTraitCount(); j < jcnt; ++j) {
					auto@ chk = getTrait(j);
					if(chk.category is null || chk.category.ident != "Privilege")
						continue;
					if(chk.hasConflicts(settings.traits))
						continue;

					checked += 1;
					if(randomd() < 1.0 / double(checked))
						@trait = chk;
				}

				if(trait !is null) {
					emp.addTrait(trait.id);
				}
				else {
					error("Error: could not find privilege trait to add to AI "+emp.name);
				}

				emp.cheatLevel += 1;
			}
		}
	}

	//Create game empires
	{
		@Creeps = Empire();
		Creeps.name = locale::REMNANTS;
		Creeps.color = Color(0xaaaaaaff);
		Creeps.major = false;
		Creeps.visionMask = ~0;

		auto@ flag = getEmpireFlag(randomi(0, getEmpireFlagCount()-1));
		Creeps.flagDef = flag.flagDef;
		Creeps.flagID = flag.id;
		@Creeps.flag = getMaterial(Creeps.flagDef);
		@Creeps.shipset = getShipset("Tyrant");

		@Pirates = Empire();
		Pirates.name = locale::PIRATES;
		Pirates.color = Color(0xff0000ff);
		Pirates.visionMask = ~0;
		Pirates.major = false;
		@flag = getEmpireFlag(randomi(0, getEmpireFlagCount()-1));
		Pirates.flagDef = flag.flagDef;
		Pirates.flagID = flag.id;
		@Pirates.flag = getMaterial(Creeps.flagDef);
		@Pirates.shipset = getShipset("Tyrant");

		//Everyone hates creeps and pirates
		for(uint i = 0, cnt = getEmpireCount(); i < cnt; ++i) {
			Empire@ other = getEmpire(i);
			if(other !is Creeps) {
				Creeps.setHostile(other, true);
				other.setHostile(Creeps, true);
			}
			if(other !is Pirates) {
				Pirates.setHostile(other, true);
				other.setHostile(Pirates, true);
			}
		}

		auto@ hd = getSubsystemDef("Hyperdrive");
		for(uint i = 0, cnt = getSubsystemDefCount(); i < cnt; ++i) {
			auto@ def = getSubsystemDef(i);
			if(def !is hd)
				Creeps.setUnlocked(def, true);
			Pirates.setUnlocked(def, true);
			
			for(uint n = 0, ncnt = def.moduleCount; n < ncnt; ++n) {
				Creeps.setUnlocked(def, def.modules[n], true);
				Pirates.setUnlocked(def, def.modules[n], true);
			}
		}

		spectatorEmpire.ContactMask.value = int(~0);
		defaultEmpire.ContactMask.value = int(~0);
		Creeps.ContactMask.value = int(~0);
		Pirates.ContactMask.value = int(~0);
	}

	//Pre-initialize traits
	for(uint i = 0, cnt = getEmpireCount(); i < cnt; ++i)
		getEmpire(i).preInitTraits();

	//Initialize default designs into empires
	initEmpireDesigns();
}