]> ruin.nu Git - hbs.git/blobdiff - bs/planet.cpp
Fixed a bug in the shipstealing code..
[hbs.git] / bs / planet.cpp
index 443b3b13695a31c2ba7d71854750acfcc2bd3241..4133277d62d7cc080abfee462df05db499a9756e 100644 (file)
@@ -23,7 +23,7 @@ using namespace std;
 Planet::Planet()
 {
        m_sRace = "Planet";
-       m_iScore = 0;
+       m_iStays = -1;
 }
 
 Planet::~Planet(){
@@ -31,18 +31,53 @@ Planet::~Planet(){
 
 //////////////////////////////////////////////////////////////////////////
 //
-unsigned Planet::planetScore() const
+unsigned Planet::planetScore(int tick = 0) const
 {
-       return m_iScore;
+       int ticks = m_Score.size();
+       if (ticks == 0)
+               return 0;
+
+       --ticks;
+
+       if (ticks < tick)
+               return m_Score[ticks];
+
+       return m_Score[tick];
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-void Planet::setPlanetScore(unsigned i)
+void Planet::setPlanetScore(unsigned number, int tick = 0)
 {
-       m_iScore = i;
+       int earlier = 0;
+       int ticks = m_Score.size();
+
+       if (ticks != 0)
+               earlier = m_Score[ticks - 1];
+
+       for (int i = ticks; i <= tick; ++i)
+       {
+               m_Score.push_back(earlier);
+       }
+       m_Score[tick] = number;
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+void Planet::addPlanetScore(unsigned number, int tick = 0)
+{
+       int earlier = 0;
+       int ticks = m_Score.size();
+
+       if (ticks != 0)
+               earlier = m_Score[ticks - 1];
+
+       for (int i = ticks; i <= tick; ++i)
+       {
+               m_Score.push_back(earlier);
+       }
+       m_Score[tick] += number;
+}
 //////////////////////////////////////////////////////////////////////////
 //
 int Planet::roids(std::string type, int tick = 0) const
@@ -97,6 +132,9 @@ void Planet::takeRoids(std::string type, int number, int tick = 0)
        for (int i = ticks; i <= tick; ++i )
                m_Roids[type].push_back(roids);
        m_Roids[type][tick] -= number;
+
+       if (type != "uninit")
+               addPlanetScore(-3000*number, tick);
 }
 
 
@@ -109,6 +147,13 @@ void Planet::runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile
                        return;
 
        int skipped = 0;
+
+       for (vector<Fleet*>::iterator i = friendly.begin(); i != friendly.end(); ++i)
+               (*i)->resetTicks();
+
+       for (vector<Fleet*>::iterator i = hostile.begin(); i != hostile.end(); ++i)
+               (*i)->resetTicks();
+
        for(int tick = 1; skipped < 20; ++tick)
        {
                //See who's in the battle at the current tick
@@ -137,18 +182,36 @@ void Planet::runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile
 
                map<string, map<string, int> > stealfriendly;
                map<string, map<string, int> > stealhostile;
+
+               //Reset roids
+               for (RoidList::iterator roid = m_Roids.begin(); roid != m_Roids.end(); ++roid)
+               {
+                       setRoids(roid->first, roids(roid->first, tick-1), tick);
+               }
+
+               setPlanetScore(planetScore(tick - 1), tick);
+               
                
                calcOneTick(&allFriends, &allHostiles, stealfriendly, stealhostile, tick );
 
                //allFriends.printFleet();
 
                allFriends.distributeLossesGains(friends, tick);
+               allFriends.distributeStolenShips(stealfriendly, friends, tick);
+
                allHostiles.distributeLossesGains(hostiles, tick);
+               allHostiles.distributeCappedRoids(hostiles, tick);
+               allHostiles.distributeStolenShips(stealhostile, hostiles, tick);
+
+               calculateScoreLoss(friends, tick);
 
        /*      for (RoidList::iterator i = m_Roids.begin(); i != m_Roids.end(); ++i)
                        setRoids(i->first, roids(i->first, 1), tick);*/
        }
 
+       for (vector<Fleet*>::iterator i = friendly.begin(); i != friendly.end(); ++i)
+               (*i)->calculateSalvage();
+
        for (RoidList::iterator i = m_Roids.begin(); i != m_Roids.end(); ++i)
        {
                for (vector<int>::iterator j = i->second.begin(); j != i->second.end(); ++j)
@@ -158,8 +221,17 @@ void Planet::runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile
 
 //////////////////////////////////////////////////////////////////////////
 //
-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, int tick = 0)
+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, int tick = 1)
 {
+       if (planetScore(tick - 1) > 0)
+               setCapping(float(planetScore(tick - 1)) / hostile->score() /  10, tick);
+       else
+               setCapping(0, tick);
+                       
+       map<string, int> pods;
+
+
+       //FIXME: Need to change this and allow multiple shiptypes with the same initiative.
        map<int, string> unitsinit; // order units after their ininitiative
        for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
                unitsinit[i->second.initiative()] = i->first;
@@ -182,6 +254,9 @@ void Planet::calcOneTick(Planet* friendly, Fleet* hostile, std::map<std::string,
                {
                        hostiletemp->takeShoot(unittype, friendly->freeFleet(unittype, 1), stealfriendly[unittype]);
                        friendlytemp->takeShoot(unittype, hostile->freeFleet(unittype, 1), stealhostile[unittype]);
+
+                       friendlytemp->calculateLostStealships(unittype, stealfriendly[unittype], 1);
+                       hostiletemp->calculateLostStealships(unittype, stealhostile[unittype], 1);
                }
                else
                {
@@ -192,38 +267,36 @@ void Planet::calcOneTick(Planet* friendly, Fleet* hostile, std::map<std::string,
 
                if (s_Units[unittype].type() == "Pod")
                {
-                       if (m_iScore <= 0) break;
-                       float capping = float(m_iScore) / hostile->score() /  10;
-                       
-                       cerr << "Capping is: " << capping << endl;
-
-                       if (capping > 0.15)
-                               capping = 0.15;
-                       if (capping <= 0)
-                               break;
-
-                       for (RoidList::iterator roid = m_Roids.begin(); roid != m_Roids.end(); ++roid)
+                       //FIXME: Prolly need to recode the whole capping section for r7, due to multiple pods thingy
+                       cerr << "Capping is: " << capping(tick) << endl;
+       
+                       if (capping(tick) > 0)
                        {
-                               int caproids = capping * roids(roid->first, tick - 1);
-                               int freepods = hostiletemp->freeFleet(unittype, 1);
-
-                               cerr << "Possible to steal " << caproids << " " << roid->first << " roids\n";
-                               cerr << freepods << " free pods available\n";
-                               
-                               if (freepods <= 0)
-                                       break;
-                               if (freepods < caproids)
-                                       caproids = freepods;
-
-                               cerr << caproids << " roids stolen\n";
-                                       
-                               takeRoids(roid->first, caproids, tick);
-
-                               hostiletemp->killFleet(unittype, caproids, 1);
-                               int totroids = caproids + hostiletemp->resource(roid->first, 0);
-                               hostiletemp->setResource(roid->first, totroids, 1);
+                               for (RoidList::iterator roid = m_Roids.begin(); roid != m_Roids.end(); ++roid)
+                               {
+                                       int caproids = capping(tick) * roids(roid->first, tick - 1);
+                                       int freepods = hostiletemp->freeFleet(unittype, 1);
+               
+                                       cerr << "Possible to steal " << caproids << " " << roid->first << " roids\n";
+                                       cerr << freepods << " free pods available\n";
                                
-                               cerr << totroids << " stolen " << roid->first << " roids\n";
+                                       if (freepods <= 0)
+                                               break;
+                                       if (freepods < caproids)
+                                               caproids = freepods;
+               
+                                       cerr << caproids << " roids stolen\n";
+
+                                       takeRoids(roid->first, caproids, tick);
+               
+                                       //FIXME: Going to move this to the distribute roids section instead.. Not really move, I'll keep this, but "regenerate" the pods in the distribute ships function.
+                                       hostiletemp->killFleet(unittype, caproids, 1);
+                                       pods[unittype] += caproids;
+                                       //int totroids = caproids + hostiletemp->resource(roid->first, 0);
+                                       hostiletemp->addResource(roid->first, caproids, 1);
+                                               
+                                       cerr << caproids << " stolen " << roid->first << " roids\n";
+                               }
                        }
                }
 
@@ -233,5 +306,53 @@ void Planet::calcOneTick(Planet* friendly, Fleet* hostile, std::map<std::string,
                *hostile = *hostiletemp;
                delete hostiletemp;
        }
+       hostile->addFleet(pods, 1);
+       
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void Planet::calculateScoreLoss(std::vector<Fleet*> friendly, int tick = 1)
+{
+       for (vector<Fleet*>::iterator i = friendly.begin(); i != friendly.end(); ++i)
+       {
+               if ((*i)->name().find("Home") != string::npos)
+               {
+                       int scorechange = (*i)->score(tick) - (*i)->score(tick - 1);
+                       addPlanetScore(scorechange, tick);
+               }
+       }
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+float Planet::capping(int tick = 0)
+{
+       int ticks = m_Capping.size();
+
+       --ticks;
+
+       if (ticks < tick)
+               return 0;
+
+       return m_Capping[tick];
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+void Planet::setCapping(float capping, int tick = 0)
+{
+       int ticks = m_Capping.size();
+
+       for (int i = ticks; i <= tick; ++i)
+       {
+               m_Capping.push_back(0);
+       }
+
+       if (capping <= 0.15 && capping >= 0)
+               m_Capping[tick] = capping;
+       else if (capping >= 0.15)
+               m_Capping[tick] = 0.15;
+       else
+               m_Capping[tick] = 0;
+}