private static bool AddDistrictParkServiced(List <DistrictPark>[] array, int buildingId, DistrictPark districtPark)
{
if (!TransferManagerInfo.IsDistrictServicesBuilding(buildingId))
{
var buildingName = TransferManagerInfo.GetBuildingName(buildingId);
Logger.LogWarning($"Constraints::AddDistrictParkServiced: Ignoring {districtPark.Name} restriction because {buildingName} ({buildingId}) is not a district services building.");
return(false);
}
if (!districtPark.Exists)
{
Logger.LogWarning($"Constraints::AddDistrictParkServiced: Ignoring {districtPark.Name} restriction because this district/park does not exist.");
return(false);
}
if (array[buildingId] == null)
{
array[buildingId] = new List <DistrictPark>();
}
if (!array[buildingId].Contains(districtPark))
{
array[buildingId].Add(districtPark);
}
return(true);
}
protected override void OnToolUpdate()
{
base.OnToolUpdate();
var building = m_hoverInstance.Building;
if (m_toolController.IsInsideUI || !Cursor.visible || building == 0)
{
ShowToolInfo(false, null, Vector3.zero);
return;
}
// Don't show info for dummy or sub buildings.
if (BuildingManager.instance.m_buildings.m_buffer[building].Info.GetAI() is DummyBuildingAI)
{
ShowToolInfo(false, null, Vector3.zero);
return;
}
if (TransferManagerInfo.IsDistrictServicesBuilding(building) || TransferManagerInfo.IsCustomVehiclesBuilding(building))
{
var position = BuildingManager.instance.m_buildings.m_buffer[building].m_position;
var txt = GetBuildingInfoText(building);
ShowToolInfo(true, txt, position);
}
}
private static void SetArrayStatus(bool[] array, int buildingId, bool status)
{
if (!TransferManagerInfo.IsDistrictServicesBuilding(buildingId))
{
return;
}
array[buildingId] = status;
}
/// <summary>
/// Called when a building is first created. If situated in a district or park, then automatically restricts that
/// building to serve its home district only.
/// </summary>
/// <param name="buildingId"></param>
public static void CreateBuilding(ushort buildingId)
{
if (!TransferManagerInfo.IsDistrictServicesBuilding(buildingId))
{
return;
}
var buildingInfo = BuildingManager.instance.m_buildings.m_buffer[buildingId].Info;
var service = buildingInfo.GetService();
var subService = buildingInfo.GetSubService();
var ai = buildingInfo.GetAI();
// Do not pack the homeDistrict and homePark into a single DistrictPark struct. Otherwise, it will make
// removing districts/parks a lot harder!!
var position = BuildingManager.instance.m_buildings.m_buffer[buildingId].m_position;
var homeDistrict = DistrictManager.instance.GetDistrict(position);
var homePark = DistrictManager.instance.GetPark(position);
Logger.Log($"Constraints::CreateBuilding: buildingId={buildingId}, homeDistrict={homeDistrict}, homePark={homePark}, service={service}, subService={subService}, ai={ai}");
// Set default input settings.
SetAllInputLocalAreas(buildingId, true);
SetAllInputOutsideConnections(buildingId, true);
m_inputBuildingToDistrictParkServiced[buildingId] = null;
// Do not set the home district for these types of buildings.
if (ai is ChildcareAI || ai is EldercareAI)
{
homeDistrict = 0;
homePark = 0;
}
// Serve all areas if the building doesn't belong to any district or park.
SetAllOutputLocalAreas(buildingId, homeDistrict == 0 && homePark == 0);
SetAllOutputOutsideConnections(buildingId, homeDistrict == 0 && homePark == 0);
m_outputBuildingToDistrictParkServiced[buildingId] = null;
if (homeDistrict != 0)
{
AddOutputDistrictParkServiced(buildingId, DistrictPark.FromDistrict(homeDistrict));
}
if (homePark != 0)
{
AddOutputDistrictParkServiced(buildingId, DistrictPark.FromPark(homePark));
}
}
/// <summary>
/// Returns the input type of the building, used to determine which GUI elements are shown in the main panel.
/// </summary>
/// <param name="building"></param>
/// <returns></returns>
public static InputType GetBuildingInputType(int building)
{
if (building == 0)
{
return(InputType.NONE);
}
var result = InputType.NONE;
// The only building type for which we will not show an outgoing tab is coal and heating power plants.
var info = BuildingManager.instance.m_buildings.m_buffer[building].Info;
if (TransferManagerInfo.IsDistrictServicesBuilding(building))
{
if ((info?.GetService() == ItemClass.Service.Electricity && info?.GetAI() is PowerPlantAI) ||
(info?.GetService() == ItemClass.Service.Water && info?.GetAI() is HeatingPlantAI) ||
(info?.GetService() == ItemClass.Service.Monument && info?.gameObject?.name == "ChirpX Launch Control Center"))
{
}
else if (!Settings.enableIndustriesControl && info?.GetService() == ItemClass.Service.PlayerIndustry)
{
}
else
{
result |= InputType.OUTGOING;
}
}
if (Settings.enableIndustriesControl && TransferManagerInfo.IsSupplyChainBuilding(building))
{
result |= InputType.SUPPLY_CHAIN;
if (!(info?.GetAI() is ExtractingFacilityAI || info?.GetAI() is FishFarmAI || info?.GetAI() is FishingHarborAI))
{
result |= InputType.INCOMING;
}
}
if (TransferManagerInfo.IsCustomVehiclesBuilding(building))
{
result |= InputType.VEHICLES;
}
return(result);
}
/// <summary>
/// Lists all buildings that can be configured using ESD.
/// </summary>
/// <returns></returns>
public static List <ushort> GetSupportedServiceBuildings()
{
var bs = new List <Building>();
for (ushort buildingId = 0; buildingId < BuildingManager.MAX_BUILDING_COUNT; buildingId++)
{
if (TransferManagerInfo.IsDistrictServicesBuilding(buildingId))
{
bs.Add(new Building
{
Name = TransferManagerInfo.GetBuildingName(buildingId),
Id = buildingId
});
}
}
bs.Sort();
return(bs.Select(b => b.Id).ToList());
}
/// <summary>
/// Helper method for displaying information, including district and supply chain constraints, about the
/// building with given building id.
/// </summary>
/// <param name="building"></param>
/// <returns></returns>
private static string GetBuildingInfoText(ushort building)
{
var inputType = TransferManagerInfo.GetBuildingInputType(building);
var txtItems = new List <string>();
txtItems.Add($"{TransferManagerInfo.GetBuildingName(building)} ({building})");
txtItems.Add(TransferManagerInfo.GetDistrictParkText(building));
// Early return. Rest of info pertains to building types that we deal with in the mod.
if (!(TransferManagerInfo.IsDistrictServicesBuilding(building) || TransferManagerInfo.IsCustomVehiclesBuilding(building)))
{
return(string.Join("\n", txtItems.ToArray()));
}
txtItems.Add(TransferManagerInfo.GetBuildingInputTypeText(building));
txtItems.Add(TransferManagerInfo.GetServicesText(building));
if (!TransferManagerInfo.IsSupplyChainBuilding(building))
{
if (TransferManagerInfo.IsDistrictServicesBuilding(building))
{
txtItems.Add("");
txtItems.Add(TransferManagerInfo.GetOutputDistrictsServedText(building));
}
if (Settings.enableCustomVehicles &&
!VehicleManagerMod.BuildingUseDefaultVehicles[building] &&
VehicleManagerMod.BuildingToVehicles[building] != null &&
(inputType & InputType.VEHICLES) != InputType.NONE)
{
txtItems.Add("");
txtItems.Add(TransferManagerInfo.GetCustomVehiclesText(building));
}
return(string.Join("\n", txtItems.ToArray()));
}
if (Settings.enableIndustriesControl)
{
// From this point forth, we know this is a supply chain building ...
txtItems.Add($"Supply Reserve: {Constraints.InternalSupplyBuffer(building)}");
if ((inputType & InputType.INCOMING) != InputType.NONE)
{
txtItems.Add("");
txtItems.Add(TransferManagerInfo.GetSupplyBuildingSourcesText(building));
}
if ((inputType & InputType.OUTGOING) != InputType.NONE)
{
txtItems.Add("");
txtItems.Add(TransferManagerInfo.GetSupplyBuildingDestinationsText(building));
}
if (Settings.enableCustomVehicles &&
!VehicleManagerMod.BuildingUseDefaultVehicles[building] &&
VehicleManagerMod.BuildingToVehicles[building] != null &&
(inputType & InputType.VEHICLES) != InputType.NONE)
{
txtItems.Add("");
txtItems.Add(TransferManagerInfo.GetCustomVehiclesText(building));
}
var problemText = TransferManagerInfo.GetSupplyBuildingProblemsText(building);
if (problemText != string.Empty)
{
txtItems.Add("");
txtItems.Add($"<<WARNING: Cannot find the following materials to procure!>>");
txtItems.Add(problemText);
}
}
return(string.Join("\n", txtItems.ToArray()));
}
protected override void OnToolGUI(Event e)
{
try
{
var hoverInstance = this.m_hoverInstance;
var building = hoverInstance.Building;
WorldInfoPanel.HideAllWorldInfoPanels();
if (Settings.keyCopy.IsPressed(e))
{
if (building == 0 ||
BuildingManager.instance.m_buildings.m_buffer[building].Info.GetAI() is DummyBuildingAI ||
!(TransferManagerInfo.IsDistrictServicesBuilding(building) || TransferManagerInfo.IsCustomVehiclesBuilding(building)))
{
Utils.DisplayMessage(
str1: "Enhanced District Services",
str2: $"Cannot copy policy from this building!",
str3: "IconMessage");
return;
}
CopyPaste.BuildingTemplate = building;
}
if (Settings.keyPaste.IsPressed(e))
{
if (CopyPaste.BuildingTemplate == 0)
{
Utils.DisplayMessage(
str1: "Enhanced District Services",
str2: $"Please hover over a valid building and press Ctrl-C to copy its policy first!",
str3: "IconMessage");
return;
}
if (building == 0 ||
BuildingManager.instance.m_buildings.m_buffer[building].Info.GetAI() is DummyBuildingAI ||
!(TransferManagerInfo.IsDistrictServicesBuilding(building) || TransferManagerInfo.IsCustomVehiclesBuilding(building)))
{
Utils.DisplayMessage(
str1: "Enhanced District Services",
str2: $"Cannot copy policy to this unsupported building!",
str3: "IconMessage");
return;
}
var inputType1 = TransferManagerInfo.GetBuildingInputType(CopyPaste.BuildingTemplate);
var inputType2 = TransferManagerInfo.GetBuildingInputType(building);
if (inputType1 != inputType2)
{
Utils.DisplayMessage(
str1: "Enhanced District Services",
str2: $"Can only copy-paste policy between buildings of the same policy type!",
str3: "IconMessage");
return;
}
var success = CopyPaste.CopyPolicyTo(building);
if (!success)
{
Utils.DisplayMessage(
str1: "Enhanced District Services",
str2: $"Could not copy certain supply chain restrictions. Please check results of copy operation!",
str3: "IconMessage");
return;
}
var position = BuildingManager.instance.m_buildings.m_buffer[building].m_position;
var txt = GetBuildingInfoText(building);
ShowToolInfo(true, txt, position);
}
if (!m_toolController.IsInsideUI && e.type == UnityEngine.EventType.MouseDown && e.button == 0)
{
if (!(TransferManagerInfo.IsDistrictServicesBuilding(building) || TransferManagerInfo.IsCustomVehiclesBuilding(building)))
{
return;
}
if (this.m_selectErrors == ToolBase.ToolErrors.None || this.m_selectErrors == ToolBase.ToolErrors.RaycastFailed)
{
Vector3 mousePosition = this.m_mousePosition;
UIInput.MouseUsed();
if (!Singleton <InstanceManager> .instance.SelectInstance(hoverInstance))
{
return;
}
SimulationManager.instance.AddAction(() =>
{
var panel = EnhancedDistrictServicesUIPanel.Instance;
if (panel != null)
{
panel.SetBuilding(hoverInstance.Building);
panel.UpdatePositionToBuilding(hoverInstance.Building);
panel.UpdatePanelToBuilding(hoverInstance.Building);
panel.opacity = 1f;
}
Singleton <GuideManager> .instance.m_worldInfoNotUsed.Disable();
});
}
}
}
catch (Exception ex)
{
Logger.LogWarning($"EnhancedDistrictServicesTool::OnToolGUI: ...");
Logger.LogException(ex);
}
}