some changes

[hbs.git] / bs / bsdoc.cpp

diff --git a/bs/bsdoc.cpp b/bs/bsdoc.cpp
index a782c08412f9f8dcf1c70a9480cb84489feded9b..47688a7bfd06e20b37cf312b818fcc8e676a973c 100644 (file)
--- a/bs/bsdoc.cpp
+++ b/bs/bsdoc.cpp
@@ -17,6 +17,7 @@
 
 #include "bsdoc.h"
 
+using namespace std;
 
 BSDoc::BSDoc()
 {
@@ -57,7 +58,8 @@ bool BSDoc::isModified() const
 
 int BSDoc::newBattle(QString name)
 {
-       m_Battles[name]["Friendly"]["Home Planet"];
+       m_Battles[name]["Friendly"]["Home Planet"].setRace("Planet");
+       m_Battles[name]["Friendly"]["Home Fleet"];
        m_Battles[name]["Hostile"];
 
        modified = true;
@@ -68,8 +70,35 @@ int BSDoc::newBattle(QString name)
 /////////////////////////////////////////////////////////////////////////
 //
 
-std::map<QString, std::map<QString, std::map<QString, Fleet> > >& BSDoc::Battles()
+const std::map<QString, std::map<QString, std::map<QString, Fleet> > >& BSDoc::battles() const
 {
        return m_Battles;
 }
 
+//////////////////////////////////////////////////////////////////////////
+//
+Fleet BSDoc::specificFleet(QString battle, QString group, QString fleet) const
+{
+       for (map<QString, map<QString, map<QString, Fleet> > >::const_iterator i = m_Battles.begin(); i != m_Battles.end(); ++i)
+       {
+               if (i->first == battle)
+               {
+                       for (map<QString, map<QString, Fleet> >::const_iterator j = i->second.begin(); j != i->second.end(); j++)
+                       {
+                               if (j->first == group)
+                               {
+                                       for (map<QString, Fleet>::const_iterator k = j->second.begin(); k != j->second.end(); ++k)
+                                       {
+                                               if (k->first == fleet)
+                                               {
+                                                       return k->second;
+                                               }
+                                       }
+                               }
+                       }
+               }
+       }
+       return Fleet();
+}
+
+