use road piece flags enum in roadextra

OpenLoco · Mar 20, 2024 · 6bcd59b · 6bcd59b
1 parent 7da5af2
commit 6bcd59b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Core/Objects/RoadExtraObject.cs b/Core/Objects/RoadExtraObject.cs
@@ -10,7 +10,7 @@ namespace OpenLoco.ObjectEditor.Objects
 	[LocoStringTable("Name")]
 	public record RoadExtraObject(
 		[property: LocoStructOffset(0x00), LocoString, Browsable(false)] string_id Name,
-		[property: LocoStructOffset(0x02)] uint16_t RoadPieces,
+		[property: LocoStructOffset(0x02)] RoadObjectPieceFlags RoadPieces,
 		[property: LocoStructOffset(0x04)] uint8_t PaintStyle,
 		[property: LocoStructOffset(0x05)] uint8_t CostIndex,
 		[property: LocoStructOffset(0x06)] int16_t BuildCostFactor,

diff --git a/Tests/LoadSaveTests.cs b/Tests/LoadSaveTests.cs
@@ -1,12 +1,14 @@
 using Core.Objects;
 using Core.Objects.Sound;
+using Microsoft.VisualBasic;
 using NUnit.Framework;
 using NUnit.Framework.Internal;
 using OpenLoco.ObjectEditor.Data;
 using OpenLoco.ObjectEditor.DatFileParsing;
 using OpenLoco.ObjectEditor.Headers;
 using OpenLoco.ObjectEditor.Objects;
 using OpenLoco.ObjectEditor.Types;
+using static System.Formats.Asn1.AsnWriter;
 using Logger = OpenLoco.ObjectEditor.Logging.Logger;
 
 namespace OpenLoco.ObjectEditor.Tests
@@ -531,7 +533,7 @@ void assertFunc(ILocoObject obj, RoadExtraObject struc) => Assert.Multiple(() =>
 				Assert.That(struc.BuildCostFactor, Is.EqualTo(4), nameof(struc.BuildCostFactor));
 				Assert.That(struc.CostIndex, Is.EqualTo(1), nameof(struc.CostIndex));
 				Assert.That(struc.PaintStyle, Is.EqualTo(1), nameof(struc.PaintStyle));
-				Assert.That(struc.RoadPieces, Is.EqualTo(127), nameof(struc.RoadPieces));
+				Assert.That(struc.RoadPieces, Is.EqualTo(RoadObjectPieceFlags.OneWay | RoadObjectPieceFlags.Track | RoadObjectPieceFlags.Slope | RoadObjectPieceFlags.SteepSlope | RoadObjectPieceFlags.Intersection | RoadObjectPieceFlags.OneSided | RoadObjectPieceFlags.Overtake), nameof(struc.RoadPieces));
 				Assert.That(struc.SellCostFactor, Is.EqualTo(-3), nameof(struc.SellCostFactor));
 				Assert.That(struc.var_0E, Is.EqualTo(0), nameof(struc.var_0E));
 			});