added FloorGraphicExtension and ThingWithFloorGraphic classes

AndroidQuazar · Dec 21, 2021 · 429b215 · 429b215
1 parent 8d368ec
commit 429b215
Show file tree

Hide file tree

Showing 4 changed files with 53 additions and 1 deletion.
diff --git a/1.3/Assemblies/VFECore.dll b/1.3/Assemblies/VFECore.dll
diff --git a/Source/VFECore/VFECore.csproj b/Source/VFECore/VFECore.csproj
@@ -755,6 +755,7 @@
     <Compile Include="VFECore\ScenParts\ScenPart_ArriveAtEdge.cs" />
     <Compile Include="VFECore\ScenParts\ScenPart_ForcedFactionGoodwill.cs" />
     <Compile Include="VFECore\StaticConstructorClass.cs" />
+    <Compile Include="VFECore\Things\ThingWithFloorGraphic.cs" />
     <Compile Include="VFECore\Utilities\VerbUtility.cs" />
     <Compile Include="VFECore\Traders\StockGenerator_ThingSetMakerTags.cs" />
     <Compile Include="VFECore\Utilities\CustomSiegeUtility.cs" />

diff --git a/Source/VFECore/VFECore.csproj.user b/Source/VFECore/VFECore.csproj.user
@@ -1,7 +1,7 @@
 <?xml version="1.0" encoding="utf-8"?>
 <Project ToolsVersion="Current" xmlns="http://schemas.microsoft.com/developer/msbuild/2003">
   <PropertyGroup>
-    <ProjectView>ShowAllFiles</ProjectView>
+    <ProjectView>ProjectFiles</ProjectView>
   </PropertyGroup>
   <PropertyGroup>
     <!-- <ReferencePath>E:\Spiele\Steam\SteamApps\common\RimWorld\RimWorldWin64_Data\Managed\;E:\Spiele\Steam\SteamApps\workshop\content\294100\2009463077\v1.1\Assemblies\</ReferencePath> -->

diff --git a/Source/VFECore/VFECore/Things/ThingWithFloorGraphic.cs b/Source/VFECore/VFECore/Things/ThingWithFloorGraphic.cs
@@ -0,0 +1,51 @@
+using RimWorld;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Verse;
+using Verse.AI;
+using VFE.Mechanoids;
+using VFE.Mechanoids.Needs;
+
+namespace VFECore
+{
+    public class FloorGraphicExtension : DefModExtension
+    {
+        public GraphicData graphicData;
+        public override IEnumerable<string> ConfigErrors()
+        {
+			graphicData.ResolveReferencesSpecial();
+            return base.ConfigErrors();
+        }
+    }
+    public class ThingWithFloorGraphic : ThingWithComps
+    {
+		public Graphic graphicIntOverride;
+		public Graphic FloorGraphic(FloorGraphicExtension floorGraphicExtension)
+		{
+			if (graphicIntOverride == null)
+			{
+				if (floorGraphicExtension.graphicData == null)
+				{
+					return BaseContent.BadGraphic;
+				}
+				graphicIntOverride = floorGraphicExtension.graphicData.GraphicColoredFor(this);
+			}
+			return graphicIntOverride;
+		}
+		public override Graphic Graphic
+		{
+			get
+			{
+				var extension = this.def.GetModExtension<FloorGraphicExtension>();
+				if (extension != null)
+                {
+					return FloorGraphic(extension);
+				}
+				return base.Graphic;
+			}
+		}
+	}
+}