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;
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 = 0) const
171 unsigned tot_score = 0;
173 for (FleetList::const_iterator i = m_Fleet.begin(); i != m_Fleet.end(); ++i)
175 if (i->second.size() >= tick)
177 tot_score += i->second[tick] * s_Units[i->first].totRes() / 10;
183 //////////////////////////////////////////////////////////////////////////
185 void Fleet::setFleet(string unittype, int number, int tick = 0)
188 int ticks = m_Fleet[unittype].size();
191 earlier = m_Fleet[unittype][ticks - 1];
193 for (int i = ticks; i <= tick; ++i)
195 m_Fleet[unittype].push_back(earlier);
197 m_Fleet[unittype][tick] = number;
200 //////////////////////////////////////////////////////////////////////////
202 int Fleet::fleet(string unittype, int tick = 0)
204 int ticks = m_Fleet[unittype].size();
211 return m_Fleet[unittype][ticks];
213 return m_Fleet[unittype][tick];
216 //////////////////////////////////////////////////////////////////////////
218 void Fleet::addToThis(std::vector<Fleet*> fleets, int tick = 0)
220 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
222 if (m_Fleet[i->first].size() == 0)
223 m_Fleet[i->first].push_back(0);
225 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
227 m_Fleet[i->first][0] += (*j)->fleet(i->first, tick);
232 //////////////////////////////////////////////////////////////////////////
234 void Fleet::distributeLossesGains(std::vector<Fleet*> fleets, int tick = 0)
236 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
238 string unittype = i->first;
240 cerr << "Distributing type: " << unittype << endl;
242 if (m_Fleet[unittype].size() < 1)
244 if (m_Fleet[unittype][0] == 0)
247 int totallost = m_Fleet[unittype][1] - m_Fleet[unittype][0];
250 for (vector<Fleet*>::iterator j = fleets.begin(); j != fleets.end(); ++j)
252 int lost = totallost * ( (*j)->fleet(unittype, tick - 1) / m_Fleet[unittype][0] );
253 cerr << (*j)->name() << " gaining " << lost << " " << unittype << endl;
254 (*j)->setFleet(unittype, (*j)->fleet(unittype, tick - 1) + lost, tick);
259 //////////////////////////////////////////////////////////////////////////
261 std::vector<Fleet*> Fleet::calculateSide(std::vector<Fleet*> fleets, int stays = 0, int tick = 0)
264 for (vector<Fleet*>::iterator i = fleets.begin(); i != fleets.end(); ++i)
266 if (( tick - (*i)->ETA()) >= 0 && (tick - (*i)->ETA()) < stays)
269 cerr << "Using fleet " << (*i)->name() << " for tick " << tick;
271 else if ((*i)->name() == "Home Planet")
277 //////////////////////////////////////////////////////////////////////////
279 int Fleet::freeFleet(std:: string unittype, int tick = 0)
281 int bticks = m_BlockedFleet[unittype].size();
286 return fleet(unittype, tick);
289 int free = fleet(unittype,tick) - m_BlockedFleet[unittype][tick];
296 //////////////////////////////////////////////////////////////////////////
298 void Fleet::takeShoot(std::string unittype, int number, std::map<std::string, int>& hitunits)
301 cerr << unittype << ": " << number << " in number\n";
302 int guns = s_Units[unittype].guns() * number;
306 cerr << unittype << " with " << guns << " guns\n";
309 for (int count = 0; count < 3; ++count)//vector<string>::iterator i = s_Units[unittype].target().begin(); i != s_Units[unittype].target().end(); ++i)
311 string ta = s_Units[unittype].target(count);
312 cerr << "Shooting at target class: " << ta << endl;
316 map<string, int*> targets;
318 for (UnitList::iterator j = s_Units.begin(); j != s_Units.end(); ++j)
321 if (m_Fleet[j->first].size() == 0)
324 if (m_Fleet[j->first].size() == 1)
325 m_Fleet[j->first].push_back(m_Fleet[j->first][0]);
327 //cerr << "Target is class: " << j->second.type() << endl;
329 if (m_Fleet[j->first][1] > 0 && ( ta == j->second.unitClass() || ta == "All"))
332 cerr << "Looking at target: " << j->first << endl;
333 targets[j->first] = &m_Fleet[j->first][1];
338 if (targets.size() == 0)
342 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
343 total += (*j->second);
345 for (map<string, int*>::iterator j = targets.begin(); j != targets.end(); ++j)
347 int maxguns = (*j->second)/total * guns;
348 cerr << "Now shooting at target: " << j->first << endl;
350 if (m_Armor[j->first] <= 0 || m_Armor[j->first] > s_Units[j->first].armor())
351 m_Armor[j->first] = s_Units[j->first].armor();
355 cerr << "Targets agility: " << s_Units[j->first].agility() << endl;
356 cerr << "Weaponspeed: " << s_Units[unittype].weaponSpeed() << endl;
360 if (*(j->second) <= 0)
362 k -= 100/(25+s_Units[unittype].weaponSpeed() - s_Units[j->first].agility());
363 m_Armor[j->first] -= s_Units[unittype].power();
364 if (m_Armor[j->first] <= 0)
366 m_Armor[j->first] = s_Units[j->first].armor();
368 hitunits[j->first]++;
370 //There is a chance that we're hitting a blocked ship.
371 if (m_BlockedFleet[j->first].size() >= 1)
373 int test = rand() % m_BlockedFleet[j->first][0];
375 && m_BlockedFleet[j->first][0] > 0)
377 if (m_BlockedFleet[j->first].size() == 1)
378 m_BlockedFleet[j->first].push_back(m_BlockedFleet[j->first][0] - 1);
379 else if (m_BlockedFleet[j->first][1] > 0)
380 m_BlockedFleet[j->first][1]--;
387 cerr << hitunits[j->first] << " units of type: " << j->first << "killed\n";
391 gunsleft -= maxguns - k;
397 //////////////////////////////////////////////////////////////////////////
399 void Fleet::takeEMP(std::string unittype, int number)
404 //////////////////////////////////////////////////////////////////////////
406 void Fleet::killFleet(std::string unittype, int number, int tick = 0)
408 if (m_Fleet[unittype].size() <= tick)
410 m_Fleet[unittype].push_back(m_Fleet[unittype][m_Fleet[unittype].size()] - number);
413 m_Fleet[unittype][tick] -= number;
416 //////////////////////////////////////////////////////////////////////////
418 void Fleet::setResource(std::string type, int number, int tick = 0)
421 if (m_Resources[type].size() <= tick)
422 m_Resources[type].push_back(number);
423 m_Resources[type][tick] = number;
426 //////////////////////////////////////////////////////////////////////////
428 int Fleet::resource(std::string type, int tick = 0) const
430 vector<int>const* resource = 0;
431 for (ResourceList::const_iterator i = m_Resources.begin(); i != m_Resources.end(); ++i)
433 if (i->first == type)
435 resource = &i->second;
442 int ticks = resource->size();
450 return resource->at(ticks);
451 return resource->at(tick);
454 //////////////////////////////////////////////////////////////////////////
456 void Fleet::printFleet()
458 for (UnitList::iterator i = s_Units.begin(); i != s_Units.end(); ++i)
460 for (int tick = 0; tick < 5 ;++tick)
462 int num = fleet(i->first, tick);
466 cerr << num << " " << i->first << " during tick: " << tick << endl;