1 /***************************************************************************
2 fleet.cpp - description
4 begin : Tue Jan 22 2002
5 copyright : (C) 2002 by Michael Andreen
7 ***************************************************************************/
9 /***************************************************************************
11 * This program is free software; you can redistribute it and/or modify *
12 * it under the terms of the GNU General Public License as published by *
13 * the Free Software Foundation; either version 2 of the License, or *
14 * (at your option) any later version. *
16 ***************************************************************************/
25 map<string, vector<int> > Fleet::s_Races;
26 UnitList Fleet::s_Units;
37 //////////////////////////////////////////////////////////////////////////
39 void Fleet::setName(string sName)
44 //////////////////////////////////////////////////////////////////////////
46 string Fleet::name() const
51 //////////////////////////////////////////////////////////////////////////
53 /** This function first sets the race, then it iterates through the the
54 * s_Races and checks if it finds the race it returns true, if it reaches
55 * the end without finding it it returns false.
57 bool Fleet::setRace(string sRace)
60 for (map<string, vector<int> >::iterator i = s_Races.begin(); i != s_Races.end(); i++)
62 if (m_sRace == (*i).first)
68 //////////////////////////////////////////////////////////////////////////
70 string Fleet::race() const
75 //////////////////////////////////////////////////////////////////////////
77 /** This function iterates through m_Fleet and adds all numbers together to
80 int Fleet::numberOfShips() const
84 for (map<string, vector<int> >::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
86 if (i->second.size() != 0)
87 total += i->second[0];
93 //////////////////////////////////////////////////////////////////////////
95 void Fleet::setETA(int eta)
100 //////////////////////////////////////////////////////////////////////////
102 int Fleet::ETA() const
107 //////////////////////////////////////////////////////////////////////////
109 void Fleet::setRaces(map<string, vector<int> >& races)
114 //////////////////////////////////////////////////////////////////////////
116 void Fleet::setUnits(UnitList& units)
122 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); i++)
124 cerr << s_Units[(*i).first].Name() << "\t\t"
125 << s_Units[(*i).first].race() <<"\t"
126 << s_Units[(*i).first].unitClass() << "\t"
127 << s_Units[(*i).first].target(0) << "\t"
128 << s_Units[(*i).first].target(1) << "\t"
129 << s_Units[(*i).first].target(2) << "\t"
130 << s_Units[(*i).first].initiative() << "\t"
131 << s_Units[(*i).first].agility() << "\t"
132 << s_Units[(*i).first].weaponSpeed() << "\t"
133 << s_Units[(*i).first].guns() << "\t"
134 << s_Units[(*i).first].power() << "\t"
135 << s_Units[(*i).first].armor() << "\t"
136 << s_Units[(*i).first].EMP() << "\t"
137 << s_Units[(*i).first].totRes() << "\t"
138 << s_Units[(*i).first].fuel() << "\t"
139 << s_Units[(*i).first].ETA() << "\t"
140 << s_Units[(*i).first].type() << endl;
145 //////////////////////////////////////////////////////////////////////////
147 const map<string, vector<int> >& Fleet::Races()
152 //////////////////////////////////////////////////////////////////////////
154 const UnitList& Fleet::Units()
159 //////////////////////////////////////////////////////////////////////////
161 vector<int> Fleet::RacesAllowed() const
163 return s_Races[m_sRace];
166 //////////////////////////////////////////////////////////////////////////
168 unsigned Fleet::score(int tick = 0) const
170 unsigned tot_score = 0;
172 for (FleetList::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
174 int ticks = i->second.size();
179 tot_score += i->second[ticks] * s_Units[i->first].totRes() / 10;
181 tot_score += i->second[tick] * s_Units[i->first].totRes() / 10;
187 //////////////////////////////////////////////////////////////////////////
189 void Fleet::setFleet(string unittype, int number, int tick = 0)
192 int ticks = m_Fleet[unittype].size();
195 earlier = m_Fleet[unittype][ticks - 1];
197 for (int i = ticks; i <= tick; ++i)
199 m_Fleet[unittype].push_back(earlier);
201 m_Fleet[unittype][tick] = number;
204 //////////////////////////////////////////////////////////////////////////
206 int Fleet::fleet(string unittype, int tick = 0)
208 int ticks = m_Fleet[unittype].size();
215 return m_Fleet[unittype][ticks];
217 return m_Fleet[unittype][tick];
220 //////////////////////////////////////////////////////////////////////////
222 int Fleet::blockedFleet(std::string unittype, int tick = 0)
224 int ticks = m_BlockedFleet[unittype].size();
233 return m_BlockedFleet[unittype][tick];
236 //////////////////////////////////////////////////////////////////////////
238 void Fleet::setBlockedFleet(std::string unittype, int number, int tick = 0)
240 int ticks = m_BlockedFleet[unittype].size();
242 for (int i = ticks; i <= tick; ++i)
244 m_BlockedFleet[unittype].push_back(0);
246 m_BlockedFleet[unittype][tick] = number;
248 cerr << "This fleet got " << m_BlockedFleet[unittype][tick] << " blocked units tick: " << tick << endl;
252 //////////////////////////////////////////////////////////////////////////
254 void Fleet::addToThis(std::vector<Fleet*> fleets, int tick = 0)
256 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
258 if (m_Fleet[i->first].size() == 0)
259 m_Fleet[i->first].push_back(0);
261 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
263 int num = (*j)->fleet(i->first, tick);
264 m_Fleet[i->first][0] += num;
266 cerr << (*j)->name() << " adding " << num << " units of type " << i->first << endl;
271 //////////////////////////////////////////////////////////////////////////
273 void Fleet::distributeLossesGains(std::vector<Fleet*> fleets, int tick = 0)
275 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
277 string unittype = i->first;
280 if (m_Fleet[unittype].size() < 1)
282 if (m_Fleet[unittype][0] == 0)
286 int totallost = fleet(unittype,1) - fleet(unittype, 0);
289 cerr << "Distributing type: " << unittype << " with a total loss of " << totallost << " units" << endl;
291 cerr << "Total number of units before: " << fleet(unittype, 0) << " and after : " << fleet(unittype, 1) << endl;
293 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
295 int fl1 = (*j)->fleet(unittype, tick - 1);
296 float part = float(fl1) / fleet(unittype, 0) ;
297 int lost = totallost * part;
298 (*j)->setFleet(unittype, (*j)->fleet(unittype, tick - 1) + lost, tick);
300 cerr << (*j)->name() << " gaining " << lost << " " << unittype << " since it's " << part * 100 << "% of the whole fleet, and it had : " << fl1 << " units last tick.." << endl;
302 lost = part * blockedFleet(unittype, 0);
304 cerr << (*j)->name() << " got " << lost << " blocked " << unittype << ", the total number of blocked ships was: " << blockedFleet(unittype, 0) << endl;
306 (*j)->setBlockedFleet(unittype, lost, tick);
311 //////////////////////////////////////////////////////////////////////////
313 std::vector<Fleet*> Fleet::calculateSide(std::vector<Fleet*> fleets, int stays = 0, int tick = 0)
316 for (vector<Fleet*>::iterator i = fleets.begin(); i != fleets.end(); ++i)
318 if (( tick - (*i)->ETA()) >= 0 && (tick - (*i)->ETA()) < stays)
321 cerr << "Using fleet " << (*i)->name() << " for tick " << tick << endl;
323 else if ((*i)->name() == "Home Planet")
329 //////////////////////////////////////////////////////////////////////////
331 int Fleet::freeFleet(std:: string unittype, int tick = 0)
333 int bticks = m_BlockedFleet[unittype].size();
338 return fleet(unittype, tick);
341 int free = fleet(unittype,tick) - m_BlockedFleet[unittype][tick];
348 //////////////////////////////////////////////////////////////////////////
350 void Fleet::takeShoot(std::string unittype, int number, std::map<std::string, int>& hitunits)
353 float guns = s_Units[unittype].guns() * number;
359 cerr << number << " " << unittype << ": with " << guns << " guns\n";
361 float gunsleft = guns;
362 for (int count = 0; count < 3; ++count)//vector<string>::iterator i = s_Units[unittype].target().begin(); i != s_Units[unittype].target().end(); ++i)
364 string ta = s_Units[unittype].target(count);
365 cerr << "Shooting at target class: " << ta << endl;
369 map<string, int*> targets;
371 for (UnitList::iterator j = s_Units.begin(); j != s_Units.end(); ++j)
374 if (m_Fleet[j->first].size() == 0)
378 if (m_Fleet[j->first].size() == 1 )
379 m_Fleet[j->first].push_back(m_Fleet[j->first][0]);
381 //cerr << "Target is class: " << j->second.type() << endl;
383 if (m_Fleet[j->first][1] > 0 && ( ta == j->second.unitClass() || ta == "All"))
386 // cerr << "Looking at target: " << j->first << endl;
387 targets[j->first] = &m_Fleet[j->first][1];
392 if (targets.size() == 0)
396 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
397 total += (*j->second);
399 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
401 float maxguns = float((*j->second))/total * guns;
402 //cerr << "Now shooting at target: " << j->first << endl;
404 if (m_Armor[j->first] <= 0 || m_Armor[j->first] > s_Units[j->first].armor())
405 m_Armor[j->first] = s_Units[j->first].armor();
408 //cerr << "Targets agility: " << s_Units[j->first].agility() << endl;
409 //cerr << "Weaponspeed: " << s_Units[unittype].weaponSpeed() << endl;
413 if (*(j->second) <= 0)
416 int wpsp = s_Units[unittype].weaponSpeed();
417 int agil = s_Units[j->first].agility();
419 k -= float(100)/(25 + wpsp - agil);
420 //cout << "Used " << blaha << " guns to hit with one shot.\n";
421 //cout << "WPSP: " << wpsp << "\nAgil: " << agil << endl;
423 m_Armor[j->first] -= s_Units[unittype].power();
424 if (m_Armor[j->first] <= 0)
426 m_Armor[j->first] = s_Units[j->first].armor();
428 hitunits[j->first]++;
430 //There is a chance that we're hitting a blocked ship.
431 if (m_BlockedFleet[j->first].size() >= 1)
433 int test = rand() % m_BlockedFleet[j->first][0];
435 && m_BlockedFleet[j->first][0] > 0)
437 if (m_BlockedFleet[j->first].size() == 1)
438 m_BlockedFleet[j->first].push_back(m_BlockedFleet[j->first][0] - 1);
439 else if (m_BlockedFleet[j->first][1] > 0)
440 m_BlockedFleet[j->first][1]--;
447 cerr << hitunits[j->first] << " units of type: " << j->first << "killed\n";
451 gunsleft -= maxguns - k;
457 //////////////////////////////////////////////////////////////////////////
459 void Fleet::takeEMP(std::string unittype, int number)
461 int guns = s_Units[unittype].guns() * number;
465 cerr << unittype << ": with " << guns << " guns\n";
467 float gunsleft = guns;
468 for (int count = 0; count < 3; ++count)//vector<string>::iterator i = s_Units[unittype].target().begin(); i != s_Units[unittype].target().end(); ++i)
470 string ta = s_Units[unittype].target(count);
471 cerr << "Shooting at target class: " << ta << endl;
475 map<string, int*> targets;
477 for (UnitList::iterator j = s_Units.begin(); j != s_Units.end(); ++j)
479 if (j->second.type() == "PDS")
482 if (freeFleet(j->first, 1) <= 0)
485 if (m_Fleet[j->first].size() == 0)
488 if (m_Fleet[j->first].size() == 1)
489 m_Fleet[j->first].push_back(m_Fleet[j->first][0]);
491 //cerr << "Target is class: " << j->second.type() << endl;
493 if (m_Fleet[j->first][1] > 0 && ( ta == j->second.unitClass() || ta == "All"))
496 // cerr << "Looking at target: " << j->first << endl;
497 targets[j->first] = &m_Fleet[j->first][1];
502 if (targets.size() == 0)
506 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
507 total += (*j->second);
509 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
511 float maxguns = float((*j->second))/total * guns;
512 cerr << "Now shooting at target: " << j->first << endl;
521 if (*(j->second) <= blockedFleet(j->first, 1))
524 int eres = s_Units[j->first].EMP();
526 k -= float(100)/(100-eres);
528 blockFleet(j->first, 1);
531 cerr << hits << " units of type: " << j->first << " blocked\n";
535 gunsleft -= maxguns - k;
541 //////////////////////////////////////////////////////////////////////////
543 void Fleet::killFleet(std::string unittype, int number, int tick = 0)
546 int ticks = m_Fleet[unittype].size();
549 earlier = m_Fleet[unittype][ticks - 1];
551 for (int i = ticks; i <= tick; ++i)
553 m_Fleet[unittype].push_back(earlier);
555 m_Fleet[unittype][tick] -= number;
558 //////////////////////////////////////////////////////////////////////////
560 void Fleet::setResource(std::string type, int number, int tick = 0)
563 int ticks = m_Resources[type].size();
564 for (int i = ticks; i <= tick; ++i)
565 m_Resources[type].push_back(number);
566 m_Resources[type][tick] = number;
569 //////////////////////////////////////////////////////////////////////////
571 void Fleet::addResource(std::string type, int number, int tick = 0)
574 int ticks = m_Resources[type].size();
578 latest = m_Resources[type][ticks - 1];
580 for (int i = ticks; i <= tick; ++i)
581 m_Resources[type].push_back(latest);
582 m_Resources[type][tick] += number;
585 //////////////////////////////////////////////////////////////////////////
587 int Fleet::resource(std::string type, int tick = 0) const
589 vector<int>const* resource = 0;
590 for (ResourceList::const_iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
592 if (i->first == type)
594 resource = &i->second;
601 int ticks = resource->size();
609 return resource->at(ticks);
610 return resource->at(tick);
613 //////////////////////////////////////////////////////////////////////////
615 void Fleet::printFleet()
617 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
619 for (int tick = 0; tick < 5 ;++tick)
621 int num = fleet(i->first, tick);
625 cerr << num << " " << i->first << " during tick: " << tick << endl;
630 //////////////////////////////////////////////////////////////////////////
632 void Fleet::blockFleet(std::string unittype, int number, int tick = 0)
634 if (m_BlockedFleet[unittype].size() >= 1)
636 m_BlockedFleet[unittype][0] += number;
637 if (m_BlockedFleet[unittype].size() > 1)
638 m_BlockedFleet[unittype][1] += number;
640 m_BlockedFleet[unittype].push_back(m_BlockedFleet[unittype][0]);
644 m_BlockedFleet[unittype].push_back(number);
645 m_BlockedFleet[unittype].push_back(number);
649 //////////////////////////////////////////////////////////////////////////
651 void Fleet::distributeCappedRoids(std::vector<Fleet*> fleets, int tick = 0)
653 for (ResourceList::iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
655 string res = i->first;
657 cerr << "Distributing type: " << res << endl;
658 for (vector<int>::iterator j = i->second.begin(); j != i->second.end(); ++j)
659 cout << (*j) << endl;
661 if (m_Resources[res].size() < 2)
663 if (m_Resources[res][1] == 0)
667 int totcapped = resource(res,1) - resource(res, 0);
670 cerr << "Distributing type: " << res << " with a total gain of " << totcapped << " roids" << endl;
672 cerr << "Total number of roids before: " << resource(res, 0) << " and after : " << resource(res, 1) << endl;
674 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
676 unsigned fl1 = (*j)->score(tick - 1);
677 float part = float(fl1) / score(0) ;
678 int lost = totcapped * part;
680 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;
681 (*j)->setResource(res, (*j)->resource(res,tick-1) + lost, tick);
686 //////////////////////////////////////////////////////////////////////////
688 void Fleet::addPodsForLostRoids(int tick = 1)
690 for (ResourceList::iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
692 setFleet("Astro Pod", fleet("Astro Pod", tick) + (resource(i->first, tick) - resource(i->first, tick - 1)), tick);