Container weight: remove two code duplications

Author: Csantucci

Source/Orts.Simulation/Simulation/Container.cs

@@ -279,6 +279,24 @@ public void ComputeWorldPosition (FreightAnimationDiscrete freightAnimDiscrete)
             var invWagonMatrix = Matrix.Invert(freightAnimDiscrete.Wagon.WorldPosition.XNAMatrix);
             RelativeContainerMatrix = Matrix.Multiply(WorldPosition.XNAMatrix, invWagonMatrix);
         }
+
+        public void ComputeLoadWeight(LoadState loadState)
+        {
+            switch (loadState)
+            {
+                case LoadState.Empty:
+                    MassKG = EmptyMassKG;
+                    break;
+                case LoadState.Loaded:
+                    MassKG = MaxMassWhenLoadedKG;
+                    break;
+                case LoadState.Random:
+                    var loadPercent = Simulator.Random.Next(101);
+                    if (loadPercent < 30) MassKG = EmptyMassKG;
+                    else MassKG = MaxMassWhenLoadedKG * loadPercent / 100f;
+                    break;
+            }
+        }
     }
 
     public class ContainerManager
@@ -542,20 +560,8 @@ public void Preload(string loadFilePath, int stackLocationIndex, LoadState loadS
                 container.LoadFromContainerFile(loadFilePath, Simulator.BasePath + @"\trains\trainset\");
                 ContainerManager.LoadedContainers.Add(loadFilePath, container);
             }
-            switch (loadState)
-            {
-                case LoadState.Empty:
-                    container.MassKG = container.EmptyMassKG;
-                    break;
-                case LoadState.Loaded:
-                    container.MassKG = container.MaxMassWhenLoadedKG;
-                    break;
-                case LoadState.Random:
-                    var loadPercent = Simulator.Random.Next(101);
-                    if (loadPercent < 30) container.MassKG = container.EmptyMassKG;
-                    else container.MassKG = container.MaxMassWhenLoadedKG * loadPercent / 100f;
-                    break;
-            }
+            container.ComputeLoadWeight(loadState);
+
             var stackLocation = StackLocations[stackLocationIndex];
             if (stackLocation.Containers != null && stackLocation.Containers.Count >= stackLocation.MaxStackedContainers)
                 Trace.TraceWarning("Stack Location {0} is full, can't lay down container", stackLocationIndex);

Source/Orts.Simulation/Simulation/RollingStocks/MSTSWagon.cs

@@ -861,7 +861,7 @@ public virtual void LoadFromWagFile(string wagFilePath)
                                 totalContainerMassKG += discreteAnim.Container.MassKG;
                             }
                     }
-                    MassKG = FreightAnimations.WagonEmptyWeight + FreightAnimations.FreightWeight + FreightAnimations.StaticFreightWeight + totalContainerMassKG;
+                    CalculateTotalMass(totalContainerMassKG);
 
                     if (FreightAnimations.StaticFreightAnimationsPresent) // If it is static freight animation, set wagon physics to full wagon value
                     {
@@ -950,6 +950,12 @@ public virtual void LoadFromWagFile(string wagFilePath)
                 BrakeSystem = MSTSBrakeSystem.Create(CarBrakeSystemType, this);
         }
 
+        // Compute total mass of wagon including freight animations and variable loads like containers
+        public void CalculateTotalMass(float totalContainerMassKG)
+        {
+            MassKG = FreightAnimations.WagonEmptyWeight + FreightAnimations.FreightWeight + FreightAnimations.StaticFreightWeight + totalContainerMassKG;
+        }
+
         public void GetMeasurementUnits()
         {
             IsMetric = Simulator.Settings.Units == "Metric" || (Simulator.Settings.Units == "Automatic" && System.Globalization.RegionInfo.CurrentRegion.IsMetric) ||
@@ -1978,7 +1984,7 @@ public override void Update(float elapsedClockSeconds)
                 // Updates the mass of the wagon considering all types of loads
                 if (FreightAnimations != null && FreightAnimations.WagonEmptyWeight != -1)
                 {
-                    MassKG = FreightAnimations.WagonEmptyWeight + FreightAnimations.FreightWeight + FreightAnimations.StaticFreightWeight + totalContainerMassKG;
+                    CalculateTotalMass(totalContainerMassKG);
                 }
             }
         }

Source/Orts.Simulation/Simulation/RollingStocks/SubSystems/FreightAnimations.cs

@@ -309,20 +309,7 @@ public void Load(string loadFilePath, LoadPosition loadPosition, LoadState loadS
                     container.LoadFromContainerFile(loadFilePath, Wagon.Simulator.BasePath +@"\trains\trainset\");
                     ContainerManager.LoadedContainers.Add(loadFilePath, container);
                 }
-                switch (loadState)
-                {
-                    case LoadState.Empty:
-                        container.MassKG = container.EmptyMassKG;
-                        break;
-                    case LoadState.Loaded:
-                        container.MassKG = container.MaxMassWhenLoadedKG;
-                        break;
-                    case LoadState.Random:
-                        var loadPercent = Simulator.Random.Next(101);
-                        if (loadPercent < 30) container.MassKG = container.EmptyMassKG;
-                        else container.MassKG = container.MaxMassWhenLoadedKG * loadPercent / 100f;
-                        break;
-                }
+                container.ComputeLoadWeight(loadState);
 
                 Vector3 offset = new Vector3(0, 0, 0);
                 var validity = Validity(Wagon, container, loadPosition, Offset, LoadingAreaLength, out offset);