X-Git-Url: https://ruin.nu/git/?a=blobdiff_plain;f=bs%2Ffleet.cpp;h=73923b5c22449e43c8702fdf9b376be5a39c201a;hb=HEAD;hp=ee22ae7a2759f58b97e35a98b8dffc8dfd4e8728;hpb=6a5c3106cf6dc30d4624df99473e18df2b1d50ed;p=hbs.git

diff --git a/bs/fleet.cpp b/bs/fleet.cpp
index ee22ae7..73923b5 100644
--- a/bs/fleet.cpp
+++ b/bs/fleet.cpp
@@ -1,9 +1,9 @@
 /***************************************************************************
-                          fleet.cpp  -  description
-                             -------------------
-    begin                : Tue Jan 22 2002
-    copyright            : (C) 2002 by Michael Andreen
-    email                : whale@linux.nu
+  fleet.cpp  -  description
+  -------------------
+begin                : Tue Jan 22 2002
+copyright            : (C) 2002 by Michael Andreen
+email                : whale@linux.nu
  ***************************************************************************/
 
 /***************************************************************************
@@ -18,6 +18,7 @@
 #include "fleet.h"
 
 #include <iostream>
+#include <cstdlib>
 using namespace std;
 
 //Static variables
@@ -27,6 +28,7 @@ UnitList Fleet::s_Units;
 Fleet::Fleet()
 {
 	m_iETA = 0;
+	m_iStays = 3;
 	m_sRace = "Cathaar";
 }
 
@@ -116,29 +118,29 @@ void Fleet::setUnits(UnitList& units)
 {
 	s_Units = units;
 
-	/*
 
-	for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); i++)
-	{
-		cerr << s_Units[(*i).first].Name() << "\t\t"
-			<< s_Units[(*i).first].race() <<"\t"
-			<< s_Units[(*i).first].unitClass() << "\t"
-			<< s_Units[(*i).first].target(0) << "\t"
-			<< s_Units[(*i).first].target(1) << "\t"
-			<< s_Units[(*i).first].target(2) << "\t"
-			<< s_Units[(*i).first].initiative() << "\t"
-			<< s_Units[(*i).first].agility() << "\t"
-			<< s_Units[(*i).first].weaponSpeed() << "\t"
-			<< s_Units[(*i).first].guns() << "\t"
-			<< s_Units[(*i).first].power() << "\t"
-			<< s_Units[(*i).first].armor() << "\t"
-			<< s_Units[(*i).first].EMP() << "\t"
-			<< s_Units[(*i).first].totRes() << "\t"
-			<< s_Units[(*i).first].fuel() << "\t"
-			<< s_Units[(*i).first].ETA() << "\t"
-			<< s_Units[(*i).first].type() << endl;
-	}
-	*/
+/*	
+	   for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); i++)
+	   {
+		   cerr << s_Units[(*i).first].Name() << "\t\t"
+		   << s_Units[(*i).first].race() <<"\t"
+		   << s_Units[(*i).first].unitClass() << "\t"
+		   << s_Units[(*i).first].target(0) << "\t"
+		   << s_Units[(*i).first].target(1) << "\t"
+		   << s_Units[(*i).first].target(2) << "\t"
+		   << s_Units[(*i).first].initiative() << "\t"
+		   << s_Units[(*i).first].agility() << "\t"
+		   << s_Units[(*i).first].weaponSpeed() << "\t"
+		   << s_Units[(*i).first].guns() << "\t"
+		   << s_Units[(*i).first].power() << "\t"
+		   << s_Units[(*i).first].armor() << "\t"
+		   << s_Units[(*i).first].EMP() << "\t"
+		   << s_Units[(*i).first].totRes() << "\t"
+		   << s_Units[(*i).first].fuel() << "\t"
+		   << s_Units[(*i).first].ETA() << "\t"
+		   << s_Units[(*i).first].type() << endl;
+	   }
+	   */
 }
 
 //////////////////////////////////////////////////////////////////////////
@@ -164,14 +166,19 @@ vector<int> Fleet::RacesAllowed() const
 
 //////////////////////////////////////////////////////////////////////////
 //
-unsigned Fleet::score(int tick = 0) const
+unsigned Fleet::score(int tick) const
 {
 	unsigned tot_score = 0;
 
 	for (FleetList::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
 	{
-		  if (i->second.size() >= tick)
-				break;
+		int ticks =	i->second.size();
+		if (ticks == 0)
+			continue;
+		--ticks;
+		if ( ticks < tick)
+			tot_score += i->second[ticks] * s_Units[i->first].totRes() / 10;
+		else
 			tot_score += i->second[tick] * s_Units[i->first].totRes() / 10;
 	}
 
@@ -180,24 +187,627 @@ unsigned Fleet::score(int tick = 0) const
 
 //////////////////////////////////////////////////////////////////////////
 //
-void Fleet::setFleet(string unittype, int number)
+void Fleet::setFleet(string unittype, int number, int tick)
 {
-	if (m_Fleet[unittype].size() == 0)
+	int earlier = 0;
+	int ticks = m_Fleet[unittype].size();
+
+	if (ticks != 0)
+		earlier = m_Fleet[unittype][ticks - 1];
+
+	for (int i = ticks; i <= tick; ++i)
 	{
-		m_Fleet[unittype].push_back(number);
-		return;
+		m_Fleet[unittype].push_back(earlier);
 	}
-	m_Fleet[unittype][0] = number;
+	m_Fleet[unittype][tick] = number;
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-int	 Fleet::fleet(string unittype, int tick = 0)
+void Fleet::addFleet(std::string unittype, int number, int tick)
 {
-	if (m_Fleet[unittype].size() == 0)
+	int earlier = 0;
+	int ticks = m_Fleet[unittype].size();
+
+	if (ticks != 0)
+		earlier = m_Fleet[unittype][ticks - 1];
+
+	for (int i = ticks; i <= tick; ++i)
+	{
+		m_Fleet[unittype].push_back(earlier);
+	}
+	m_Fleet[unittype][tick] += number;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int	 Fleet::fleet(string unittype, int tick)
+{
+
+	int ticks = m_Fleet[unittype].size();
+	if (ticks == 0)
 		return 0;
 
+	if (tick < 0)
+		return m_Fleet[unittype][0];
+
+	--ticks;
+
+	if (ticks < tick)
+		return m_Fleet[unittype][ticks];
+
 	return m_Fleet[unittype][tick];
 }
 
-	
+//////////////////////////////////////////////////////////////////////////
+//
+int Fleet::blockedFleet(std::string unittype, int tick)
+{
+	int ticks = m_BlockedFleet[unittype].size();
+	if (ticks == 0)
+		return 0;
+
+	--ticks;
+
+	if (ticks < tick)
+		return 0;
+
+	return m_BlockedFleet[unittype][tick];
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::setBlockedFleet(std::string unittype, int number, int tick)
+{
+	int ticks = m_BlockedFleet[unittype].size();
+
+	for (int i = ticks; i <= tick; ++i)
+	{
+		m_BlockedFleet[unittype].push_back(0);
+	}
+	m_BlockedFleet[unittype][tick] = number;
+
+	cerr << "This fleet got " << m_BlockedFleet[unittype][tick] << " blocked units tick: " << tick << endl;
+
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::addToThis(std::vector<Fleet*> fleets, int tick)
+{
+	for (UnitList::iterator i = s_Units.begin();  i != s_Units.end(); ++i)
+	{
+		if (m_Fleet[i->first].size() == 0)
+			m_Fleet[i->first].push_back(0);
+
+		for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
+		{
+			int num = (*j)->fleet(i->first, tick);
+			m_Fleet[i->first][0] += num;
+			if (num > 0)
+				cerr << (*j)->name() <<  " adding " << num << " units of type " << i->first << endl;
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::distributeLossesGains(std::vector<Fleet*> fleets, int tick)
+{
+	for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
+	{
+		string unittype = i->first;
+
+
+		if (m_Fleet[unittype].size() < 1)
+			continue;
+		if (m_Fleet[unittype][0] == 0)
+			continue;
+
+
+		int totallost = fleet(unittype,1) - fleet(unittype, 0);
+
+
+		cerr << "Distributing type: " << unittype << " with a total loss of " << totallost << " units" << endl;
+
+		cerr << "Total number of units before: " << fleet(unittype, 0)  << " and after : " << fleet(unittype, 1) << endl;
+
+		for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
+		{
+			int fl1 = (*j)->fleet(unittype, tick - 1);
+			float part = float(fl1) / fleet(unittype, 0) ;
+			int lost =  int(totallost * part);
+			(*j)->setFleet(unittype, (*j)->fleet(unittype, tick - 1) + lost, tick);
+
+			cerr << (*j)->name() << " gaining " << lost << " " << unittype  << " since it's " << part * 100 << "% of the whole fleet, and it had : " << fl1 << " units last tick.." <<  endl;
+
+			lost = int(part * blockedFleet(unittype, 0));
+
+			cerr << (*j)->name() << " got " << lost << " blocked " << unittype  << ", the total number of blocked ships was: " << blockedFleet(unittype, 0) << endl; 
+
+			(*j)->setBlockedFleet(unittype, lost, tick);
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+std::vector<Fleet*> Fleet::calculateSide(std::vector<Fleet*> fleets, int tick)
+{
+	vector<Fleet*> fl;
+	for (vector<Fleet*>::iterator i = fleets.begin(); i != fleets.end(); ++i)
+	{
+		if ((*i)->stays() == 0)
+			continue;
+		else if (( tick - (*i)->ETA()) >= 0 && (tick - (*i)->ETA()) < (*i)->stays())
+		{
+			fl.push_back((*i));
+			cerr << "Using fleet " << (*i)->name() << " for tick " << tick << endl;
+		}
+		else if ((*i)->stays() < 0)
+			fl.push_back((*i));
+	}
+	return fl;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int Fleet::freeFleet(std:: string unittype, int tick)
+{
+	int bticks = m_BlockedFleet[unittype].size();
+
+	--bticks;
+
+	if (bticks < tick)
+		return fleet(unittype, tick);
+
+
+	int free = fleet(unittype,tick) - m_BlockedFleet[unittype][tick];
+	if (free < 0)
+		return 0;
+	return free;
+}
+
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::takeShoot(std::string unittype, int number, std::map<std::string, int>& hitunits)
+{
+
+	float guns = s_Units[unittype].guns() * number;
+
+
+	if (guns == 0)
+		return;
+
+	float gunsleft = guns;
+	for (int count = 0; count < 3; ++count)
+	{
+		string ta = s_Units[unittype].target(count);
+
+		while (gunsleft > 0)
+		{
+			int total = 0;
+			map<string, int*> targets = findTargets(ta,total, 0);
+
+			if (targets.size() == 0)
+				break;
+
+			for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
+			{
+				float maxguns = float((*j->second))/total * guns;
+
+				if (m_Armor[j->first] <= 0 || m_Armor[j->first] > s_Units[j->first].armor())
+					m_Armor[j->first] = s_Units[j->first].armor();
+				double k = maxguns;
+
+				while (k > 0)	
+				{
+
+					if (*(j->second) <= 0)
+						break;
+
+					int wpsp = s_Units[unittype].weaponSpeed();
+					int agil = s_Units[j->first].agility();
+
+					k -= float(100)/(25 + wpsp - agil);
+
+					m_Armor[j->first] -= s_Units[unittype].power();
+					if (m_Armor[j->first] <= 0)
+					{
+						m_Armor[j->first] = s_Units[j->first].armor();
+						(*j->second)--;
+						hitunits[j->first]++;
+					}
+
+				}
+				if (k <= 0)
+					gunsleft -= maxguns;
+				else
+					gunsleft -= maxguns - k;
+			}
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::takeEMP(std::string unittype, int number, std::map<std::string, int>& hitunits)
+{
+	int guns = s_Units[unittype].guns() * number;
+	if (guns == 0)
+		return;
+
+	float gunsleft = guns;
+	for (int count = 0; count < 3; ++count)
+	{
+		string ta = s_Units[unittype].target(count);
+
+		while (gunsleft > 0)
+		{
+			int total = 0;
+			map<string, int*> targets = findTargets(ta, total, 1);
+
+			if (targets.size() == 0)
+				break;
+
+			for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
+			{
+				float maxguns = float((*j->second))/total * gunsleft;
+				double k = maxguns;
+				int hits = 0;
+
+				while (k > 0)	
+				{
+
+					if (*(j->second) <= blockedFleet(j->first, 1))
+						break;
+
+					int eres = s_Units[j->first].EMP();
+
+					k -= float(100)/(100-eres);
+					hits++;
+					hitunits[j->first]++;
+					blockFleet(j->first, 1);
+				}
+				if (k <= 0)
+					gunsleft -= maxguns;
+				else
+					gunsleft -= maxguns - k;
+			}
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::killFleet(std::string unittype, int number, int tick)
+{
+	int earlier = 0;
+	int ticks = m_Fleet[unittype].size();
+
+	if (ticks != 0)
+		earlier = m_Fleet[unittype][ticks - 1];
+
+	for (int i = ticks; i <= tick; ++i)
+	{
+		m_Fleet[unittype].push_back(earlier);
+	}
+	m_Fleet[unittype][tick] -= number;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::setResource(std::string type, int number, int tick)
+{
+
+	int ticks = m_Resources[type].size();
+	for (int i = ticks; i <= tick; ++i)
+		m_Resources[type].push_back(number);
+	m_Resources[type][tick] = number;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::addResource(std::string type, int number, int tick)
+{
+
+	int ticks = m_Resources[type].size();
+	int latest = resource(type, tick - 1);
+
+	for (int i = ticks; i <= tick; ++i)
+		m_Resources[type].push_back(latest);
+	m_Resources[type][tick] += number;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int Fleet::resource(std::string type, int tick) const
+{
+	if (tick < 0)
+		return 0;
+
+	vector<int>const* resource = 0;
+	for (ResourceList::const_iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
+	{
+		if (i->first == type)
+		{
+			resource = &i->second;
+			break;
+		}
+	}
+	if (resource == 0)
+		return 0;
+
+	int ticks = resource->size();
+
+	if( ticks == 0)
+		return 0;
+
+	--ticks;
+
+	if (ticks < tick)
+		return resource->at(ticks);
+	return resource->at(tick);
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::resetResources()
+{
+	m_Resources.clear() ;	
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::printFleet()
+{
+	for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
+	{
+		for (int tick = 0; tick < 5 ;++tick)
+		{
+			int num = fleet(i->first, tick);
+
+			if (num <= 0)
+				break;
+			cerr << num << " " << i->first << " during tick: " << tick << endl;
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::blockFleet(std::string unittype, int number, int tick)
+{
+	if (m_BlockedFleet[unittype].size() >= 1)
+	{
+		m_BlockedFleet[unittype][0] += number;
+		if (m_BlockedFleet[unittype].size() > 1)
+			m_BlockedFleet[unittype][1] += number;
+		else
+			m_BlockedFleet[unittype].push_back(m_BlockedFleet[unittype][0]);
+	}
+	else
+	{
+		m_BlockedFleet[unittype].push_back(number);
+		m_BlockedFleet[unittype].push_back(number);
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::distributeCappedRoids(std::vector<Fleet*> fleets, int tick)
+{
+	for (ResourceList::iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
+	{
+		string res = i->first;
+
+
+		cerr << "Distributing type: " << res << endl;
+		for (vector<int>::iterator j = i->second.begin(); j != i->second.end(); ++j)
+			cout << (*j) << endl;
+
+		if (m_Resources[res].size() < 2)
+			continue;
+		if (m_Resources[res][1] == 0)
+			continue;
+
+
+		int totcapped = resource(res,1) - resource(res, 0);
+
+
+		cerr << "Distributing type: " << res << " with a total gain of " << totcapped << " roids" << endl;
+
+		cerr << "Total number of roids before: " << resource(res, 0)  << " and after : " << resource(res, 1) << endl;
+
+		for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
+		{
+			unsigned fl1 = (*j)->score(tick - 1);
+			float part = float(fl1) / score(0) ;
+			int lost =  int(totcapped * part);
+
+			cerr << (*j)->name() << " gaining " << lost << " " << res  << " since it's " << part * 100 << "% of the whole score, and it had : " << fl1 << " score last tick.. compared to fleet total of: " << score(0) <<  endl;
+
+			//(*j)->setResource(res, (*j)->resource(res,tick-1) + lost, tick);
+			(*j)->addResource(res,lost, tick);
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::addFleet(std::map<string, int> units, int tick)
+{
+	for (map<string, int>::iterator i = units.begin(); i != units.end(); ++i)
+		addFleet(i->first, i->second, tick);
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int Fleet::stays() const
+{
+	return m_iStays;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::setStays(int ticks)
+{
+	m_iStays = ticks;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::calculateLostStealships(string unittype, std::map<std::string, int> stolen, int tick)
+{
+	int stealscore = 0;
+	for (map<string, int>::iterator i = stolen.begin(); i != stolen.end(); ++i)
+	{
+		stealscore += int(stolen[i->first] * (s_Units[i->first].totRes() / 10.0));	
+	}
+
+	int lost = int(stealscore / (s_Units[unittype].totRes() / 10.0));
+
+	cerr << "Lost " << lost << " " << unittype << " due to stealing ships worth: " << stealscore << endl; 
+	killFleet(unittype, lost, tick);
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::distributeStolenShips(std::map<std::string, std::map<std::string, int> > stolen, std::vector<Fleet*> fleets, int tick)
+{
+	for(map<string, map<string, int> >::iterator i = stolen.begin(); i != stolen.end(); ++i)
+	{
+		int totalstealers = 0;
+		for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
+			totalstealers += (*j)->fleet(i->first, tick - 1);
+
+		for (map<string, int>::iterator j = i->second.begin(); j != i->second.end(); ++j)
+		{
+			for (vector<Fleet*>::iterator k = fleets.begin(); k != fleets.end(); ++k)
+			{
+				int stolen = int(float ((*k)->fleet(i->first, tick - 1)) / totalstealers * j->second);
+				(*k)->addFleet(j->first, stolen, tick);
+			}
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::calculateSalvage()
+{
+	for (FleetList::iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
+	{
+
+		map<string, int> res = s_Units[i->first].resources();
+
+		if (i->second.size() > 0)
+			cerr << endl << i->first << ": ";
+
+		int tick = 0;
+		for (vector<int>::iterator j = i->second.begin(); j != i->second.end(); ++j, ++tick)
+		{
+			int lostunits = fleet(i->first, tick - 1) - fleet(i->first, tick);
+
+			if (lostunits <= 0)
+				continue;
+			cerr << "(" << tick << ":" << fleet(i->first, tick) << ") ";
+			for (map<string, int>::iterator k = res.begin(); k != res.end(); ++k)
+				addResource(k->first, int(lostunits * k->second * 0.25), tick);
+		}
+	}
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::resetTicks()
+{
+	for (FleetList::iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
+	{
+		if ( i->second.size() < 2)
+			continue;
+
+		int temp = i->second[0];
+		i->second.clear();
+
+		if (temp > 0)
+			i->second.push_back(temp);
+	}
+	resetResources();
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int Fleet::freePodGuns(int tick) 
+{
+	int guns = 0;
+	for (FleetList::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
+	{
+		if (s_Units[i->first].type() == "Pod")
+		{
+			guns += freeFleet(i->first, tick) * s_Units[i->first].guns();
+		}
+	}
+	guns -= usedPodGuns(tick);
+	return guns;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+int Fleet::usedPodGuns(int tick) const
+{
+	int ticks = m_UsedPodGuns.size();
+
+	if (ticks == 0)
+		return 0;
+
+	if (tick < 0)
+		return m_UsedPodGuns[0];
+
+	--ticks;
+
+	if (ticks < tick)
+		return m_UsedPodGuns[ticks];
+
+	return m_UsedPodGuns[tick];
+
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Fleet::usePodGuns(int tick, int guns)
+{
+	int ticks = m_UsedPodGuns.size();
+
+	for (int i = ticks; i <= tick; ++i)
+	{
+		m_UsedPodGuns.push_back(0);
+	}
+	m_UsedPodGuns[tick] += guns;
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+std::map<std::string, int*> Fleet::findTargets(std::string targetClass,int& total, int attackerType)
+{
+	map<string, int*> targets;
+
+	for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
+	{
+		if (m_Fleet[i->first].size() == 0)
+			continue;
+
+		if (m_Fleet[i->first].size() == 1 )
+			m_Fleet[i->first].push_back(m_Fleet[i->first][0]);
+
+		if (m_Fleet[i->first][1] > 0  && ( targetClass == i->second.unitClass() || targetClass == "All"))
+		{
+			if (i->second.unitClass() == "PDS" && attackerType == 1)
+				continue;
+			targets[i->first] = &m_Fleet[i->first][1];
+			total += m_Fleet[i->first][1];
+		}
+	}
+	return targets;
+}