]> ruin.nu Git - hbs.git/blobdiff - bs/planet.cpp
Fixed a bug in the shipstealing code..
[hbs.git] / bs / planet.cpp
index 41d92ac51eb2ed6004bab3b01c8009a46b8fdefb..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
@@ -65,6 +100,7 @@ int Planet::roids(std::string type, int tick = 0) const
 
        if( ticks == 0)
                return 0;
+       --ticks;
        if (ticks < tick)
                return roids->at(ticks);
        return roids->at(tick);
@@ -72,13 +108,37 @@ int Planet::roids(std::string type, int tick = 0) const
 
 //////////////////////////////////////////////////////////////////////////
 //
-void Planet::setRoids(std::string type, int number)
+void Planet::setRoids(std::string type, int number, int tick = 0)
 {
-       if (m_Roids[type].size() == 0)
-               m_Roids[type].push_back(number);
-       m_Roids[type][0] = number;
+       int ticks = m_Roids[type].size();
+       int roids = 0;
+       if (m_Roids[type].size() > 0)
+               roids = m_Roids[type][m_Roids[type].size() - 1];
+
+       for (int i = ticks; i <= tick; ++i )
+               m_Roids[type].push_back(roids);
+       m_Roids[type][tick] = number;
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+void Planet::takeRoids(std::string type, int number, int tick = 0)
+{
+       int ticks = m_Roids[type].size();
+       int roids = 0;
+       if (m_Roids[type].size() > 0)
+               roids = m_Roids[type][m_Roids[type].size() - 1];
+
+       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);
+}
+
+
+
 //////////////////////////////////////////////////////////////////////////
 //
 void Planet::runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile)
@@ -87,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
@@ -103,25 +170,68 @@ void Planet::runBattle(std::vector<Fleet*> friendly, std::vector<Fleet*> hostile
                        skipped = 0;
 
                Planet allFriends;
-               allFriends.addToThis(friends);
+               allFriends.addToThis(friends, tick - 1);
+
+               /*for (RoidList::iterator i = m_Roids.begin(); i != m_Roids.end(); ++i)
+                       allFriends.setRoids(i->first, roids(i->first, tick));
+
+               allFriends.setPlanetScore(m_iScore);*/
                
                Fleet allHostiles;
-               allHostiles.addToThis(hostiles);
+               allHostiles.addToThis(hostiles, tick - 1);
 
                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 );
+               
+               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)
+                       cout << i->first << " roids : " << (*j) << endl;
        }
 }
 
 //////////////////////////////////////////////////////////////////////////
 //
-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 )
+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;
@@ -133,7 +243,7 @@ void Planet::calcOneTick(Planet* friendly, Fleet* hostile, std::map<std::string,
                
                string unittype = i->second;
 
-               cerr << "Initiative: " << s_Units[unittype].initiative() << " with unit: " << unittype << endl;
+               //cerr << "Initiative: " << s_Units[unittype].initiative() << " with unit: " << unittype << endl;
 
                if (s_Units[unittype].type() == "EMP")  
                {
@@ -144,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
                {
@@ -154,19 +267,36 @@ void Planet::calcOneTick(Planet* friendly, Fleet* hostile, std::map<std::string,
 
                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)
+                       //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->second[0];
-                               int freepods = hostiletemp->freeFleet(unittype, 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";
                                
-                               if (freepods == 0)
-                                       break;
-                               if (freepods < caproids)
-                                       caproids = caproids - freepods;
-                                       
-                               roids->second.push_back(roids->second[0] - caproids);
-                               hostiletemp->killFleet(unittype, caproids, 1);
+                                       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";
+                               }
                        }
                }
 
@@ -176,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;
+}