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;
38 //////////////////////////////////////////////////////////////////////////
40 void Fleet::setName(string sName)
45 //////////////////////////////////////////////////////////////////////////
47 string Fleet::name() const
52 //////////////////////////////////////////////////////////////////////////
54 /** This function first sets the race, then it iterates through the the
55 * s_Races and checks if it finds the race it returns true, if it reaches
56 * the end without finding it it returns false.
58 bool Fleet::setRace(string sRace)
61 for (map<string, vector<int> >::iterator i = s_Races.begin(); i != s_Races.end(); i++)
63 if (m_sRace == (*i).first)
69 //////////////////////////////////////////////////////////////////////////
71 string Fleet::race() const
76 //////////////////////////////////////////////////////////////////////////
78 /** This function iterates through m_Fleet and adds all numbers together to
81 int Fleet::numberOfShips() const
85 for (map<string, vector<int> >::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
87 if (i->second.size() != 0)
88 total += i->second[0];
94 //////////////////////////////////////////////////////////////////////////
96 void Fleet::setETA(int eta)
101 //////////////////////////////////////////////////////////////////////////
103 int Fleet::ETA() const
108 //////////////////////////////////////////////////////////////////////////
110 void Fleet::setRaces(map<string, vector<int> >& races)
115 //////////////////////////////////////////////////////////////////////////
117 void Fleet::setUnits(UnitList& units)
123 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); i++)
125 cerr << s_Units[(*i).first].Name() << "\t\t"
126 << s_Units[(*i).first].race() <<"\t"
127 << s_Units[(*i).first].unitClass() << "\t"
128 << s_Units[(*i).first].target(0) << "\t"
129 << s_Units[(*i).first].target(1) << "\t"
130 << s_Units[(*i).first].target(2) << "\t"
131 << s_Units[(*i).first].initiative() << "\t"
132 << s_Units[(*i).first].agility() << "\t"
133 << s_Units[(*i).first].weaponSpeed() << "\t"
134 << s_Units[(*i).first].guns() << "\t"
135 << s_Units[(*i).first].power() << "\t"
136 << s_Units[(*i).first].armor() << "\t"
137 << s_Units[(*i).first].EMP() << "\t"
138 << s_Units[(*i).first].totRes() << "\t"
139 << s_Units[(*i).first].fuel() << "\t"
140 << s_Units[(*i).first].ETA() << "\t"
141 << s_Units[(*i).first].type() << endl;
146 //////////////////////////////////////////////////////////////////////////
148 const map<string, vector<int> >& Fleet::Races()
153 //////////////////////////////////////////////////////////////////////////
155 const UnitList& Fleet::Units()
160 //////////////////////////////////////////////////////////////////////////
162 vector<int> Fleet::RacesAllowed() const
164 return s_Races[m_sRace];
167 //////////////////////////////////////////////////////////////////////////
169 unsigned Fleet::score(int tick) const
171 unsigned tot_score = 0;
173 for (FleetList::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
175 int ticks = i->second.size();
180 tot_score += i->second[ticks] * s_Units[i->first].totRes() / 10;
182 tot_score += i->second[tick] * s_Units[i->first].totRes() / 10;
188 //////////////////////////////////////////////////////////////////////////
190 void Fleet::setFleet(string unittype, int number, int tick)
193 int ticks = m_Fleet[unittype].size();
196 earlier = m_Fleet[unittype][ticks - 1];
198 for (int i = ticks; i <= tick; ++i)
200 m_Fleet[unittype].push_back(earlier);
202 m_Fleet[unittype][tick] = number;
205 //////////////////////////////////////////////////////////////////////////
207 void Fleet::addFleet(std::string unittype, int number, int tick)
210 int ticks = m_Fleet[unittype].size();
213 earlier = m_Fleet[unittype][ticks - 1];
215 for (int i = ticks; i <= tick; ++i)
217 m_Fleet[unittype].push_back(earlier);
219 m_Fleet[unittype][tick] += number;
222 //////////////////////////////////////////////////////////////////////////
224 int Fleet::fleet(string unittype, int tick)
227 int ticks = m_Fleet[unittype].size();
232 return m_Fleet[unittype][0];
237 return m_Fleet[unittype][ticks];
239 return m_Fleet[unittype][tick];
242 //////////////////////////////////////////////////////////////////////////
244 int Fleet::blockedFleet(std::string unittype, int tick)
246 int ticks = m_BlockedFleet[unittype].size();
255 return m_BlockedFleet[unittype][tick];
258 //////////////////////////////////////////////////////////////////////////
260 void Fleet::setBlockedFleet(std::string unittype, int number, int tick)
262 int ticks = m_BlockedFleet[unittype].size();
264 for (int i = ticks; i <= tick; ++i)
266 m_BlockedFleet[unittype].push_back(0);
268 m_BlockedFleet[unittype][tick] = number;
270 cerr << "This fleet got " << m_BlockedFleet[unittype][tick] << " blocked units tick: " << tick << endl;
274 //////////////////////////////////////////////////////////////////////////
276 void Fleet::addToThis(std::vector<Fleet*> fleets, int tick)
278 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
280 if (m_Fleet[i->first].size() == 0)
281 m_Fleet[i->first].push_back(0);
283 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
285 int num = (*j)->fleet(i->first, tick);
286 m_Fleet[i->first][0] += num;
288 cerr << (*j)->name() << " adding " << num << " units of type " << i->first << endl;
293 //////////////////////////////////////////////////////////////////////////
295 void Fleet::distributeLossesGains(std::vector<Fleet*> fleets, int tick)
297 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
299 string unittype = i->first;
302 if (m_Fleet[unittype].size() < 1)
304 if (m_Fleet[unittype][0] == 0)
308 int totallost = fleet(unittype,1) - fleet(unittype, 0);
311 cerr << "Distributing type: " << unittype << " with a total loss of " << totallost << " units" << endl;
313 cerr << "Total number of units before: " << fleet(unittype, 0) << " and after : " << fleet(unittype, 1) << endl;
315 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
317 int fl1 = (*j)->fleet(unittype, tick - 1);
318 float part = float(fl1) / fleet(unittype, 0) ;
319 int lost = int(totallost * part);
320 (*j)->setFleet(unittype, (*j)->fleet(unittype, tick - 1) + lost, tick);
322 cerr << (*j)->name() << " gaining " << lost << " " << unittype << " since it's " << part * 100 << "% of the whole fleet, and it had : " << fl1 << " units last tick.." << endl;
324 lost = int(part * blockedFleet(unittype, 0));
326 cerr << (*j)->name() << " got " << lost << " blocked " << unittype << ", the total number of blocked ships was: " << blockedFleet(unittype, 0) << endl;
328 (*j)->setBlockedFleet(unittype, lost, tick);
333 //////////////////////////////////////////////////////////////////////////
335 std::vector<Fleet*> Fleet::calculateSide(std::vector<Fleet*> fleets, int tick)
338 for (vector<Fleet*>::iterator i = fleets.begin(); i != fleets.end(); ++i)
340 if ((*i)->stays() == 0)
342 else if (( tick - (*i)->ETA()) >= 0 && (tick - (*i)->ETA()) < (*i)->stays())
345 cerr << "Using fleet " << (*i)->name() << " for tick " << tick << endl;
347 else if ((*i)->stays() < 0)
353 //////////////////////////////////////////////////////////////////////////
355 int Fleet::freeFleet(std:: string unittype, int tick)
357 int bticks = m_BlockedFleet[unittype].size();
362 return fleet(unittype, tick);
365 int free = fleet(unittype,tick) - m_BlockedFleet[unittype][tick];
372 //////////////////////////////////////////////////////////////////////////
374 void Fleet::takeShoot(std::string unittype, int number, std::map<std::string, int>& hitunits)
377 float guns = s_Units[unittype].guns() * number;
383 //cerr << number << " " << unittype << ": with " << guns << " guns\n";
385 float gunsleft = guns;
386 for (int count = 0; count < 3; ++count)//vector<string>::iterator i = s_Units[unittype].target().begin(); i != s_Units[unittype].target().end(); ++i)
388 string ta = s_Units[unittype].target(count);
389 //cerr << "Shooting at target class: " << ta << endl;
393 map<string, int*> targets = findTargets(ta,total, 0);
395 if (targets.size() == 0)
398 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
400 float maxguns = float((*j->second))/total * guns;
401 //cerr << "Now shooting at target: " << j->first << endl;
403 if (m_Armor[j->first] <= 0 || m_Armor[j->first] > s_Units[j->first].armor())
404 m_Armor[j->first] = s_Units[j->first].armor();
407 //cerr << "Targets agility: " << s_Units[j->first].agility() << endl;
408 //cerr << "Weaponspeed: " << s_Units[unittype].weaponSpeed() << endl;
412 if (*(j->second) <= 0)
415 int wpsp = s_Units[unittype].weaponSpeed();
416 int agil = s_Units[j->first].agility();
418 k -= float(100)/(25 + wpsp - agil);
419 //cout << "Used " << blaha << " guns to hit with one shot.\n";
420 //cout << "WPSP: " << wpsp << "\nAgil: " << agil << endl;
422 m_Armor[j->first] -= s_Units[unittype].power();
423 if (m_Armor[j->first] <= 0)
425 m_Armor[j->first] = s_Units[j->first].armor();
427 hitunits[j->first]++;
429 //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]--;
448 //cerr << hitunits[j->first] << " units of type: " << j->first << "killed\n";
452 gunsleft -= maxguns - k;
458 //////////////////////////////////////////////////////////////////////////
460 void Fleet::takeEMP(std::string unittype, int number, std::map<std::string, int>& hitunits)
462 int guns = s_Units[unittype].guns() * number;
466 //cerr << unittype << ": with " << guns << " guns\n";
468 float gunsleft = guns;
469 for (int count = 0; count < 3; ++count)//vector<string>::iterator i = s_Units[unittype].target().begin(); i != s_Units[unittype].target().end(); ++i)
471 string ta = s_Units[unittype].target(count);
472 //cerr << "Shooting at target class: " << ta << endl;
476 map<string, int*> targets = findTargets(ta, total, 1);
478 if (targets.size() == 0)
481 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
483 float maxguns = float((*j->second))/total * gunsleft;
484 //cerr << "Now shooting at target: " << j->first << endl;
493 if (*(j->second) <= blockedFleet(j->first, 1))
496 int eres = s_Units[j->first].EMP();
498 k -= float(100)/(100-eres);
500 hitunits[j->first]++;
501 blockFleet(j->first, 1);
504 //cerr << hits << " units of type: " << j->first << " blocked\n";
508 gunsleft -= maxguns - k;
514 //////////////////////////////////////////////////////////////////////////
516 void Fleet::killFleet(std::string unittype, int number, int tick)
519 int ticks = m_Fleet[unittype].size();
522 earlier = m_Fleet[unittype][ticks - 1];
524 for (int i = ticks; i <= tick; ++i)
526 m_Fleet[unittype].push_back(earlier);
528 m_Fleet[unittype][tick] -= number;
531 //////////////////////////////////////////////////////////////////////////
533 void Fleet::setResource(std::string type, int number, int tick)
536 int ticks = m_Resources[type].size();
537 for (int i = ticks; i <= tick; ++i)
538 m_Resources[type].push_back(number);
539 m_Resources[type][tick] = number;
542 //////////////////////////////////////////////////////////////////////////
544 void Fleet::addResource(std::string type, int number, int tick)
547 int ticks = m_Resources[type].size();
548 int latest = resource(type, tick - 1);
550 for (int i = ticks; i <= tick; ++i)
551 m_Resources[type].push_back(latest);
552 m_Resources[type][tick] += number;
555 //////////////////////////////////////////////////////////////////////////
557 int Fleet::resource(std::string type, int tick) const
562 vector<int>const* resource = 0;
563 for (ResourceList::const_iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
565 if (i->first == type)
567 resource = &i->second;
574 int ticks = resource->size();
582 return resource->at(ticks);
583 return resource->at(tick);
586 //////////////////////////////////////////////////////////////////////////
588 void Fleet::resetResources()
590 m_Resources.clear() ;
593 //////////////////////////////////////////////////////////////////////////
595 void Fleet::printFleet()
597 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
599 for (int tick = 0; tick < 5 ;++tick)
601 int num = fleet(i->first, tick);
605 cerr << num << " " << i->first << " during tick: " << tick << endl;
610 //////////////////////////////////////////////////////////////////////////
612 void Fleet::blockFleet(std::string unittype, int number, int tick)
614 if (m_BlockedFleet[unittype].size() >= 1)
616 m_BlockedFleet[unittype][0] += number;
617 if (m_BlockedFleet[unittype].size() > 1)
618 m_BlockedFleet[unittype][1] += number;
620 m_BlockedFleet[unittype].push_back(m_BlockedFleet[unittype][0]);
624 m_BlockedFleet[unittype].push_back(number);
625 m_BlockedFleet[unittype].push_back(number);
629 //////////////////////////////////////////////////////////////////////////
631 void Fleet::distributeCappedRoids(std::vector<Fleet*> fleets, int tick)
633 for (ResourceList::iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
635 string res = i->first;
638 cerr << "Distributing type: " << res << endl;
639 for (vector<int>::iterator j = i->second.begin(); j != i->second.end(); ++j)
640 cout << (*j) << endl;
642 if (m_Resources[res].size() < 2)
644 if (m_Resources[res][1] == 0)
648 int totcapped = resource(res,1) - resource(res, 0);
651 cerr << "Distributing type: " << res << " with a total gain of " << totcapped << " roids" << endl;
653 cerr << "Total number of roids before: " << resource(res, 0) << " and after : " << resource(res, 1) << endl;
655 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
657 unsigned fl1 = (*j)->score(tick - 1);
658 float part = float(fl1) / score(0) ;
659 int lost = int(totcapped * part);
661 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;
663 //(*j)->setResource(res, (*j)->resource(res,tick-1) + lost, tick);
664 (*j)->addResource(res,lost, tick);
669 //////////////////////////////////////////////////////////////////////////
671 void Fleet::addFleet(std::map<string, int> units, int tick)
673 for (map<string, int>::iterator i = units.begin(); i != units.end(); ++i)
674 addFleet(i->first, i->second, tick);
677 //////////////////////////////////////////////////////////////////////////
679 int Fleet::stays() const
684 //////////////////////////////////////////////////////////////////////////
686 void Fleet::setStays(int ticks)
691 //////////////////////////////////////////////////////////////////////////
693 void Fleet::calculateLostStealships(string unittype, std::map<std::string, int> stolen, int tick)
696 for (map<string, int>::iterator i = stolen.begin(); i != stolen.end(); ++i)
698 stealscore += int(stolen[i->first] * (s_Units[i->first].totRes() / 10.0));
701 int lost = int(stealscore / (s_Units[unittype].totRes() / 10.0));
703 cerr << "Lost " << lost << " " << unittype << " due to stealing ships worth: " << stealscore << endl;
704 killFleet(unittype, lost, tick);
707 //////////////////////////////////////////////////////////////////////////
709 void Fleet::distributeStolenShips(std::map<std::string, std::map<std::string, int> > stolen, std::vector<Fleet*> fleets, int tick)
711 for(map<string, map<string, int> >::iterator i = stolen.begin(); i != stolen.end(); ++i)
713 int totalstealers = 0;
714 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
715 totalstealers += (*j)->fleet(i->first, tick - 1);
717 for (map<string, int>::iterator j = i->second.begin(); j != i->second.end(); ++j)
719 for (vector<Fleet*>::iterator k = fleets.begin(); k != fleets.end(); ++k)
721 int stolen = int(float ((*k)->fleet(i->first, tick - 1)) / totalstealers * j->second);
722 (*k)->addFleet(j->first, stolen, tick);
728 //////////////////////////////////////////////////////////////////////////
730 void Fleet::calculateSalvage()
732 for (FleetList::iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
735 map<string, int> res = s_Units[i->first].resources();
737 if (i->second.size() > 0)
738 cerr << endl << i->first << ": ";
741 for (vector<int>::iterator j = i->second.begin(); j != i->second.end(); ++j, ++tick)
743 int lostunits = fleet(i->first, tick - 1) - fleet(i->first, tick);
747 cerr << "(" << tick << ":" << fleet(i->first, tick) << ") ";
748 for (map<string, int>::iterator k = res.begin(); k != res.end(); ++k)
749 addResource(k->first, int(lostunits * k->second * 0.25), tick);
754 //////////////////////////////////////////////////////////////////////////
756 void Fleet::resetTicks()
758 for (FleetList::iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
760 if ( i->second.size() < 2)
763 int temp = i->second[0];
767 i->second.push_back(temp);
772 //////////////////////////////////////////////////////////////////////////
774 int Fleet::freePodGuns(int tick)
777 for (FleetList::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
779 if (s_Units[i->first].type() == "Pod")
781 guns += freeFleet(i->first, tick) * s_Units[i->first].guns();
784 guns -= usedPodGuns(tick);
788 //////////////////////////////////////////////////////////////////////////
790 int Fleet::usedPodGuns(int tick) const
792 int ticks = m_UsedPodGuns.size();
798 return m_UsedPodGuns[0];
803 return m_UsedPodGuns[ticks];
805 return m_UsedPodGuns[tick];
809 //////////////////////////////////////////////////////////////////////////
811 void Fleet::usePodGuns(int tick, int guns)
813 int ticks = m_UsedPodGuns.size();
815 for (int i = ticks; i <= tick; ++i)
817 m_UsedPodGuns.push_back(0);
819 m_UsedPodGuns[tick] += guns;
822 //////////////////////////////////////////////////////////////////////////
824 std::map<std::string, int*> Fleet::findTargets(std::string targetClass,int& total, int attackerType)
826 map<string, int*> targets;
828 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
830 if (m_Fleet[i->first].size() == 0)
834 if (m_Fleet[i->first].size() == 1 )
835 m_Fleet[i->first].push_back(m_Fleet[i->first][0]);
837 //cerr << "Target is class: " << j->second.type() << endl;
839 if (m_Fleet[i->first][1] > 0 && ( targetClass == i->second.unitClass() || targetClass == "All"))
841 if (i->second.unitClass() == "PDS" && attackerType == 1)
843 // cerr << "Looking at target: " << j->first << endl;
844 targets[i->first] = &m_Fleet[i->first][1];
845 total += m_Fleet[i->first][1];