NoPillars mod for Cities:Skylines

NoPillars.cs

using ICities;
using UnityEngine;
using ColossalFramework.UI;
using ColossalFramework.Plugins;
using System.Collections;
using System.Reflection;
using System;

namespace NoPillars {
	public class NoPillarsMod : IUserMod {
		public string Name { get { return "No Pillars"; } }
		public string Description { get { return "Toggle pillars and collision, and use railroad tracks in the asset editor"; } }
	}

	public class NoPillarsLoadingExtension : LoadingExtensionBase {
		public static UIButton b_pillars = null;
		public static UIButton b_collide = null;
		public static UIButton b_track = null;
		public static UIButton b_strack = null;

		public class SaveInfo {
			public NetInfo prefab;

			public bool collide;
			public bool zoning;
			public BuildingInfo bpi;
			public BuildingInfo bmi;
		}
		public static FastList<SaveInfo> saveList = null;
		public static SaveInfo track = null;

		public static int pillars = 0;
		public static bool collide = true;

		public override void OnLevelLoaded(LoadMode mode) {
			var uiView = UIView.GetAView();

			float aeo = 0f;
			if(mode == LoadMode.LoadAsset || mode == LoadMode.NewAsset) {
				aeo = 0.2f;

				b_track = makeButton(uiView, "Train Track");
				b_track.transformPosition = new Vector3(-1.25f+aeo, -0.82f);
				b_track.eventClick += setTrainTrack;

				b_strack = makeButton(uiView, "Station Track");
				b_strack.transformPosition = new Vector3(-1.25f+aeo, -0.88f);
				b_strack.eventClick += setStationTrack;
			}

			b_pillars = makeButton(uiView, "Pillars");
			b_pillars.transformPosition = new Vector3(-1.45f+aeo, -0.82f);
			b_pillars.eventClick += togglePillars;

			b_collide = makeButton(uiView, "Collide");
			b_collide.transformPosition = new Vector3(-1.45f+aeo, -0.88f);
			b_collide.eventClick += toggleColliding;
		}

		public override void OnLevelUnloading() {
			revert();
		}

		private UIButton makeButton(UIView uiView, string t) {
			UIButton b = (UIButton)uiView.AddUIComponent(typeof(UIButton));
			b.text = t;
			b.width = 100;
			b.height = 30;
			b.normalBgSprite = "ButtonMenu";
			b.disabledBgSprite = "ButtonMenuDisabled";
			b.hoveredBgSprite = "ButtonMenuHovered";
			b.focusedBgSprite = "ButtonMenuFocused";
			b.pressedBgSprite = "ButtonMenuPressed";
			b.textColor = new Color32(255, 255, 255, 255);
			b.disabledTextColor = new Color32(7, 7, 7, 255);
			b.hoveredTextColor = new Color32(7, 132, 255, 255);
			b.focusedTextColor = new Color32(255, 255, 255, 255);
			b.pressedTextColor = new Color32(30, 30, 44, 255);
			b.playAudioEvents = true;
			return b;
		}

		private void switchToRoadByName(string name) {
			NetTool nt = ToolsModifierControl.SetTool<NetTool>();
			if(nt != null) {
				foreach(NetCollection collection in NetCollection.FindObjectsOfType<NetCollection>()) {
					foreach(NetInfo prefab in collection.m_prefabs) {
						if(prefab.name == name) {
							nt.m_prefab = prefab;
							return;
						}
					}
				}
			}
		}

		private void setTrainTrack(UIComponent component, UIMouseEventParameter eventParam) {
			switchToRoadByName("Train Track");
		}

		private void setStationTrack(UIComponent component, UIMouseEventParameter eventParam) {
			switchToRoadByName("Train Station Track");
		}

		private void modify() {
			if(saveList != null) revert();
			saveList = new FastList<SaveInfo>();
			foreach(NetCollection collection in NetCollection.FindObjectsOfType<NetCollection>()) {
				foreach(NetInfo prefab in collection.m_prefabs) {
					SaveInfo si = new SaveInfo();

					if(prefab.name == "Train Track Elevated") {
						track = si;
					}

					si.prefab = prefab;
					si.collide = prefab.m_canCollide;
					prefab.m_canCollide = collide && prefab.m_canCollide;

					TrainTrackBridgeAI ta = prefab.m_netAI as TrainTrackBridgeAI;
					if(ta != null) {
						si.bpi = ta.m_bridgePillarInfo;
						si.bmi = ta.m_middlePillarInfo;
						if(pillars == 1) {
							ta.m_bridgePillarInfo = null;
							ta.m_middlePillarInfo = null;
						} else if(pillars == 2 && track != null) {
							ta.m_bridgePillarInfo = track.bpi;
							ta.m_middlePillarInfo = track.bmi;
						}
					}
					RoadBridgeAI ra = prefab.m_netAI as RoadBridgeAI;
					if(ra != null) {
						si.bpi = ra.m_bridgePillarInfo;
						si.bmi = ra.m_middlePillarInfo;
						if(pillars == 1) {
							ra.m_bridgePillarInfo = null;
							ra.m_middlePillarInfo = null;
						} else if(pillars == 2 && track != null) {
							ra.m_bridgePillarInfo = track.bpi;
							ra.m_middlePillarInfo = track.bmi;
						}
					}
					PedestrianBridgeAI pa = prefab.m_netAI as PedestrianBridgeAI;
					if(pa != null) {
						si.bpi = pa.m_bridgePillarInfo;
						if(pillars != 0) {
							pa.m_bridgePillarInfo = null;
						}
					}
					RoadAI r2 = prefab.m_netAI as RoadAI;
					if(r2 != null) {
						si.zoning = r2.m_enableZoning;
						if(!collide) r2.m_enableZoning = false;
					}

					saveList.Add(si);
				}
			}
		}

		private void revert() {
			if(saveList == null) return;
			foreach(SaveInfo si in saveList) {
				si.prefab.m_canCollide = si.collide;
				TrainTrackBridgeAI ta = si.prefab.m_netAI as TrainTrackBridgeAI;
				if(ta != null) {
					ta.m_bridgePillarInfo = si.bpi;
					ta.m_middlePillarInfo = si.bmi;
				}
				RoadBridgeAI ra = si.prefab.m_netAI as RoadBridgeAI;
				if(ra != null) {
					ra.m_bridgePillarInfo = si.bpi;
					ra.m_middlePillarInfo = si.bmi;
				}
				PedestrianBridgeAI pa = si.prefab.m_netAI as PedestrianBridgeAI;
				if(pa != null) {
					pa.m_bridgePillarInfo = si.bpi;
				}
				RoadAI r2 = si.prefab.m_netAI as RoadAI;
				if(r2 != null) {
					r2.m_enableZoning = si.zoning;
				}

			}
			saveList = null;
		}

		private void togglePillars(UIComponent component, UIMouseEventParameter eventParam) {
			pillars = (pillars + 1) % 3;
			modify();
			if(pillars == 0) {
				b_pillars.text = "Pillars";
			} else if(pillars == 1) {
				b_pillars.text = "Floating";
			} else if(pillars == 2) {
				b_pillars.text = "Pillars (Side)";
			}
		}

		private void toggleColliding(UIComponent component, UIMouseEventParameter eventParam) {
			collide = !collide;
			modify();
			if(collide) {
				b_collide.text = "Collide";
			} else {
				b_collide.text = "Overlap";
			}
		}
	}
}