view api/Factories/Xml/WarFoundryXmlArmyParser.cs @ 119:0cdc0b07fe11

Re #54: Add army loading * Add unit to army once it is loaded
author IBBoard <dev@ibboard.co.uk>
date Sun, 23 Aug 2009 10:32:09 +0000
parents 093ee2da0f6e
children 53654e938982
line wrap: on
line source

//  This file (WarFoundryXmlArmyParser.cs) is a part of the IBBoard.WarFoundry.API project and is copyright 2009 IBBoard
// 
// The file and the library/program it is in are licensed and distributed, without warranty, under the GNU Affero GPL license, either version 3 of the License or (at your option) any later version. Please see COPYING for more information and the full license.

using System;
using System.Collections.Generic;
using System.Xml;
using IBBoard.Xml;
using ICSharpCode.SharpZipLib.Zip;
using IBBoard.WarFoundry.API.Objects;

namespace IBBoard.WarFoundry.API.Factories.Xml
{
	public class WarFoundryXmlArmyParser
	{
		private ZipFile file;
		private XmlElement elem;
		private Army army;
		private Dictionary<String, Unit> units;
		
		public WarFoundryXmlArmyParser(ZipFile file, XmlElement elem)
		{
			this.file = file;
			this.elem = elem;
		}

		public Army GetArmy()
		{
			if (army == null)
			{
				ParseArmy();
			}

			return army;
		}

		private void ParseArmy()
		{
			string name = elem.GetAttribute("name");
			string systemID = elem.GetAttribute("system");
			GameSystem system = WarFoundryLoader.GetDefault().GetGameSystem(systemID);
			
			if (system == null)
			{
				throw new RequiredDataMissingException(file.Name, "gameSystem", systemID);
			}
			
			string raceID = elem.GetAttribute("race");
			Race race = WarFoundryLoader.GetDefault().GetRace(system, raceID);
			
			if (race == null)
			{
				throw new RequiredDataMissingException(file.Name, "race", raceID);
			}
			
			int points = XmlTools.GetIntValueFromAttribute(elem, "maxPoints");			
			army = new Army(race, name, points, file);
			LoadUnits();
		}

		private void LoadUnits()
		{
			units = new Dictionary<string, Unit>();

			foreach (XmlElement unitElem in WarFoundryXmlFactoryUtils.SelectNodes(elem, "/army:army/army:units/army:unit"))
			{
				string id = unitElem.GetAttribute("id");

				if (!units.ContainsKey(id))
				{
					string unitTypeId = unitElem.GetAttribute("unitType");
					UnitType unitType = army.Race.GetUnitType(unitTypeId);

					if (unitType == null)
					{
						throw new RequiredDataMissingException(file.Name, "unitType", unitTypeId);
					}

					Unit unit = new Unit(unitType, army.GetCategory(unitType.MainCategory));
					army.AddUnit(unit);
					units.Add(id, unit);

					//LoadUnitEquipment(
				}
			}
		}
	}
}