// 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(); } }