emp works now too..

[hbs.git] / bs / bsdoc.cpp

diff --git a/bs/bsdoc.cpp b/bs/bsdoc.cpp
index 70cfd899ba4459461f10ea3cc9e3ed7ebea0f420..047248f42dea9518a51f55fc5de106645ff89fac 100644 (file)
--- a/bs/bsdoc.cpp
+++ b/bs/bsdoc.cpp
@@ -64,8 +64,11 @@ int BSDoc::newBattle(QString name)
 {
        m_Battles[name]["Friendly"]["Home Planet"] = new Planet();;
        m_Battles[name]["Friendly"]["Home Fleet"] = new Fleet();
-       m_Battles[name]["Hostile"];
+       m_Battles[name]["Hostile"]["Evil guy"] = new Fleet();
 
+       m_Battles[name]["Friendly"]["Home Planet"]->setName("Home Planet");
+       m_Battles[name]["Friendly"]["Home Fleet"]->setName("Home Fleet");
+       m_Battles[name]["Hostile"]["Evil guy"]->setName("Evil guy");
        //Planet* pl = dynamic_cast<Planet*>(m_Battles[name]["Friendly"]["Home Planet"]);
        modified = true;
        emit documentChanged();
@@ -133,3 +136,35 @@ void BSDoc::newFleet(QString battle, QString group, QString fleet, Fleet* fl)
        emit documentChanged();
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+void BSDoc::removeFleet(QString battle, QString group, QString fleet)
+{
+       delete m_Battles[battle][group][fleet];
+       m_Battles[battle][group].erase(fleet);
+       emit documentChanged();
+}
+
+//////////////////////////////////////////////////////////////////////////
+//
+void BSDoc::runBattleSimulation()
+{
+       for(BattleList::iterator i = m_Battles.begin(); i != m_Battles.end(); ++i)
+       {
+               map<QString, vector<Fleet*> > battle;
+               for (map<QString, map<QString, Fleet*> >::iterator j = i->second.begin(); j != i->second.end(); ++j)
+               {
+                       for (map<QString, Fleet*>::iterator k = j->second.begin(); k != j->second.end(); ++k)
+                               battle[j->first].push_back(k->second);
+               }
+
+               Planet* pl = dynamic_cast<Planet*>(i->second["Friendly"]["Home Planet"]);
+
+               if (pl)
+               {
+                       pl->runBattle(battle["Friendly"], battle["Hostile"]);
+               }
+       }
+
+       emit documentChanged();
+}