battle algorithms are getting closer.. ;)

[hbs.git] / bs / planet.cpp

diff --git a/bs/planet.cpp b/bs/planet.cpp
index c72fb4e61c39e5b9450cac19f3c97a82ac2b723f..4ae9c59e4d517e87b9047c7eb9b00dee837e3dad 100644 (file)
--- a/bs/planet.cpp
+++ b/bs/planet.cpp
@@ -78,4 +78,91 @@ void Planet::setRoids(std::string type, int number)
        m_Roids[type][0] = number;
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+void Planet::runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile)
+{
+       for(int tick = 1; ; ++tick)
+       {
+               //See who's in the battle at the current tick
+               vector<Fleet*> friends = calculateSide(friendly, 6, tick);
+               vector<Fleet*> hostiles = calculateSide(hostile, 3, tick);
+
+               // No idea to calculate anything if noone is there.. ;)
+               if (hostiles.size() == 0)
+                       break;
+
+               Planet allFriends;
+               allFriends.addToThis(friends);
+               
+               Fleet allHostiles;
+               allHostiles.addToThis(hostiles);
+
+               map<string, map<string, int> > stealfriendly;
+               map<string, map<string, int> > stealhostile;
+               
+               calcOneTick(&allFriends, &allHostiles, stealfriendly, stealhostile );
+
+               allFriends.distributeLossesGains(friends, tick);
+               allHostiles.distributeLossesGains(friends, tick);
+       }
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Planet::calcOneTick(Planet* friendly, Fleet* hostile, std::map<std::string, std::map<std::string, int> >& stealfriendly, std::map<std::string, std::map<std::string, int> >&  stealhostile )
+{
+       map<int, string> unitsinit; // order units after their ininitiative
+       for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
+               unitsinit[i->second.ETA()] = i->first;
+
+       for (map<int, string>::iterator i = unitsinit.begin(); i != unitsinit.end(); ++i)
+       {
+               Fleet* hostiletemp = new Fleet(*hostile);
+               Planet* friendlytemp = new Planet(*friendly);
+               
+               string unittype = i->second;
+
+               if (s_Units[unittype].type() == "EMP")  
+               {
+                       hostiletemp->takeEMP(unittype, friendly->freeFleet(unittype, 1));
+                       friendlytemp->takeEMP(unittype, hostile->freeFleet(unittype, 1));
+               }
+               else if (s_Units[unittype].type() == "Steal")
+               {
+                       hostiletemp->takeShoot(unittype, friendly->freeFleet(unittype, 1), stealfriendly[unittype]);
+                       friendlytemp->takeShoot(unittype, hostile->freeFleet(unittype, 1), stealhostile[unittype]);
+               }
+               else
+               {
+                       map<string, int> temp;
+                       hostiletemp->takeShoot(unittype, friendly->freeFleet(unittype, 1), temp);
+                       friendlytemp->takeShoot(unittype, hostile->freeFleet(unittype, 1), temp);
+               }
+
+               if (s_Units[unittype].type() == "Pod")
+               {
+                       float capping = friendly->m_iScore / hostile->score() /  10;
+      for (RoidList::iterator roids = m_Roids.begin(); roids != m_Roids.end(); ++roids)
+                       {
+                               int caproids = capping * roids->second[0];
+                               int freepods = hostiletemp->freeFleet(unittype, 1);
+                               
+                               if (freepods == 0)
+                                       break;
+                               if (freepods < caproids)
+                                       caproids = caproids - freepods;
+                                       
+                               roids->second.push_back(roids->second[0] - caproids);
+                               hostiletemp->killFleet(unittype, caproids, 1);
+                       }
+               }
+
+               //set the the objects so they point at the modified objects
+               delete friendly;
+               friendly = friendlytemp;
+               delete hostile;
+               hostile = hostiletemp;
+       }
+}