[Upgrades] Add "reinit" flag to upgrades

When set, the object runs its init function whenever it is upgraded or
spawned with an upgrade. Useful when the object caches its tuning value
on creation rather than accessing it directly.

TSOClient/tso.content/Upgrades/Model/ObjectUpgradeConfig.cs

@@ -29,13 +29,19 @@ public class ObjectUpgradeConfig
         /// </summary>
         public bool? Special;
 
+        /// <summary>
+        /// If true, the object's init function will be run again when upgrading to this level.
+        /// </summary>
+        public bool Reinit;
+
         public void Load(int version, BinaryReader reader)
         {
             GUID = reader.ReadString();
             Level = reader.ReadInt32();
             Limit = reader.ReadInt32();
             if (Limit < 0) Limit = null;
             Special = reader.ReadBoolean();
+            Reinit = reader.ReadBoolean();
         }
 
         public void Save(BinaryWriter writer)
@@ -44,6 +50,7 @@ public void Save(BinaryWriter writer)
             writer.Write(Level);
             writer.Write(Limit ?? -1);
             writer.Write(Special ?? false);
+            writer.Write(Reinit);
         }
     }
 }

TSOClient/tso.content/Upgrades/ObjectUpgradeProvider.cs

@@ -119,6 +119,17 @@ public void Init()
             return null;
         }
 
+        public ObjectUpgradeConfig GetUpgradeConfig(string file, uint guid, int level)
+        {
+            RuntimeUpgradeFile upgradeFile = null;
+            if (FileUpgrades.TryGetValue(file, out upgradeFile))
+            {
+                upgradeFile.Load(ContentManager);
+                return upgradeFile.GetConfig(guid);
+            }
+            return null;
+        }
+
         public Dictionary<int, Dictionary<int, short>> GetUpgrade(string file, uint guid, int level)
         {
             // find upgrade table for this file, find entries for this object, and the specified level.

TSOClient/tso.simantics/NetPlay/Model/Commands/VMNetBuyObjectCmd.cs

@@ -120,6 +120,7 @@ private bool TryPlace(VM vm, VMAvatar caller)
                         {
                             state.UpgradeLevel = TargetUpgradeLevel;
                             obj.UpdateTuning(vm);
+                            VMNetUpgradeCmd.TryReinit(obj, vm, TargetUpgradeLevel);
                         }
                     }
                 }

TSOClient/tso.simantics/NetPlay/Model/Commands/VMNetPlaceInventoryCmd.cs

@@ -133,11 +133,14 @@ private bool TryPlace(VM vm, VMAvatar caller)
                 var tsostate = (obj.PlatformState as VMTSOObjectState);
                 if (tsostate != null) {
                     tsostate.OwnerID = caller.PersistID;
+                    bool reinitRequired = false;
                     if (Info.UpgradeLevel > tsostate.UpgradeLevel)
                     {
                         tsostate.UpgradeLevel = Info.UpgradeLevel;
+                        reinitRequired = true;
                     }
                     obj.UpdateTuning(vm);
+                    if (reinitRequired) VMNetUpgradeCmd.TryReinit(obj, vm, tsostate.UpgradeLevel);
                 }
                 obj.PersistID = ObjectPID;
                 ((VMGameObject)obj).DisableIfTSOCategoryWrong(vm.Context);

TSOClient/tso.simantics/NetPlay/Model/Commands/VMNetUpgradeCmd.cs

@@ -18,6 +18,15 @@ public class VMNetUpgradeCmd : VMNetCommandBodyAbstract
         public byte TargetUpgradeLevel;
         public int AddedValue;
 
+        public static void TryReinit(VMEntity obj, VM vm, int level)
+        {
+            var config = Content.Content.Get().Upgrades.GetUpgradeConfig(
+                        obj.Object.Resource.Iff.Filename,
+                        (obj.MasterDefinition ?? obj.Object.OBJ).GUID,
+                        level);
+            if (config?.Reinit == true) obj.ExecuteEntryPoint(0, vm.Context, true);
+        }
+
         public override bool Execute(VM vm, VMAvatar caller)
         {
             var pobj = vm.GetObjectByPersist(ObjectPID);
@@ -33,6 +42,7 @@ public override bool Execute(VM vm, VMAvatar caller)
                     state.Wear = 20 * 4;
                     state.QtrDaysSinceLastRepair = 0;
                     obj.UpdateTuning(vm);
+                    TryReinit(obj, vm, TargetUpgradeLevel);
                 }
             }