// note this doesnt store hub types
public void StoreFunctionalBuildingRef(BuildingType btype, FunctionalBuilding building)
{
if (btype == BuildingType.Disk)
{
shieldGens.Add((ShieldGenerator)building);
}
if (btype == BuildingType.Disk2)
{
greenHouses.Add((GreenHouse)building);
}
if (btype == BuildingType.SmallDisk)
{
powerStations.Add((PowerStation)building);
}
if (btype == BuildingType.Room)
{
residences.Add((Residence)building);
}
}
public void InitializeFunctional(BuildingType btype)
{
if (btype == BuildingType.Disk)
{
functional = new ShieldGenerator();
}
if (btype == BuildingType.Disk2)
{
functional = new GreenHouse();
}
if (btype == BuildingType.SmallDisk)
{
functional = new PowerStation();
}
if (btype == BuildingType.Room)
{
functional = new Residence();
}
}
