Roadmap: Update road cost automatically

Greavesy1899 · Mar 29, 2024 · 0254e0e · 0254e0e
1 parent cf1fb6d
commit 0254e0e
Show file tree

Hide file tree

Showing 4 changed files with 88 additions and 0 deletions.
diff --git a/Mafia2Libs/Core/IO/FileRoadMap.cs b/Mafia2Libs/Core/IO/FileRoadMap.cs
@@ -105,6 +105,19 @@ public override void Save()
                     Spline.CalculateLength();
                 }
 
+                foreach(ICostMapEntry CostEntry in NewRoadmap.CostMap)
+                {
+                    // TODO: Make it work for crossroad/junctions, not been determined
+                    if(CostEntry.RoadGraphEdgeType == RoadGraphEdgeType.Road)
+                    {
+                        IRoadDefinition Road = NewRoadmap.Roads[CostEntry.RoadGraphEdgeLink];
+                        if(Road != null && Road.RoadType == RoadType.Road) // Only for roads for now.
+                        {
+                            CostEntry.Cost = NewRoadmap.CalculateRoadCost(Road);
+                        }
+                    }
+                }
+
                 // now save
                 using (FileStream FStream = File.Open(file.FullName, FileMode.Open))
                 {

diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Navigation/Roadmap/Roadmap.cs b/Mafia2Libs/ResourceTypes/FileTypes/Navigation/Roadmap/Roadmap.cs
@@ -161,6 +161,7 @@ public interface IRoadmap : IRoadmapSerializable
         ushort GetRoadIndexForRoadGraphEdge(int edgeIndex);
         void AddRoadEdge(ushort roadIndex);
         void AddCrossroadJunctionEdge(ushort targetRoadIndex);
+        ushort CalculateRoadCost(IRoadDefinition road);
     }
 
 }
diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Navigation/Roadmap/RoadmapCe.cs b/Mafia2Libs/ResourceTypes/FileTypes/Navigation/Roadmap/RoadmapCe.cs
@@ -861,5 +861,42 @@ public void AddCrossroadJunctionEdge(ushort targetRoadIndex)
         {
             RoadGraphEdges.Add((ushort) (targetRoadIndex * 2));
         }
+
+        public ushort CalculateRoadCost(IRoadDefinition road)
+        {
+            float splineLength = Splines[road.RoadSplineIndex].Length;
+
+            bool hasHighwayFlag = false;
+            bool hasMainRoadLaneType = false;
+
+            foreach (ILaneDefinition lane in road.Lanes)
+            {
+                if (lane.LaneFlags.HasFlag(LaneFlags.Highway))
+                {
+                    hasHighwayFlag = true;
+                }
+
+                if (lane.LaneType == LaneType.MainRoad)
+                {
+                    hasMainRoadLaneType = true;
+                }
+            }
+
+            float multCoef = 1.0f;
+            if (hasHighwayFlag)
+            {
+                multCoef = 0.5f;
+            }
+            else if (hasMainRoadLaneType)
+            {
+                multCoef = 0.75f;
+            }
+            else
+            {
+                multCoef = 1.5f;
+            }
+
+            return (ushort)(splineLength * multCoef);
+        }
     }
 }
diff --git a/Mafia2Libs/ResourceTypes/FileTypes/Navigation/Roadmap/RoadmapDe.cs b/Mafia2Libs/ResourceTypes/FileTypes/Navigation/Roadmap/RoadmapDe.cs
@@ -564,5 +564,42 @@ private ushort GetLastRoadEdgeIndex()
 
             return 0;
         }
+
+        public ushort CalculateRoadCost(IRoadDefinition road)
+        {
+            float splineLength = Splines[road.RoadSplineIndex].Length;
+
+            bool hasHighwayFlag = false;
+            bool hasMainRoadLaneType = false;
+
+            foreach (ILaneDefinition lane in road.Lanes)
+            {
+                if (lane.LaneFlags.HasFlag(LaneFlags.Highway))
+                {
+                    hasHighwayFlag = true;
+                }
+
+                if (lane.LaneType == LaneType.MainRoad)
+                {
+                    hasMainRoadLaneType = true;
+                }
+            }
+
+            float multCoef = 1.0f;
+            if (hasHighwayFlag)
+            {
+                multCoef = 0.5f;
+            }
+            else if (hasMainRoadLaneType)
+            {
+                multCoef = 0.75f;
+            }
+            else
+            {
+                multCoef = 1.5f;
+            }
+
+            return (ushort)(splineLength * multCoef);
+        }
     }
 }