update compensate logic

Author: despc

SentisOptimisations/Freezer/FreezeLogic.cs

@@ -145,6 +145,10 @@ private static void CompensateFrozenFrames(MyCubeGrid grid)
                     }
                 }
             }
+            else
+            {
+                LastUpdateFrames.Remove(myCubeBlock.EntityId);
+            }
         }
     }
 
@@ -356,9 +360,13 @@ private static void DoUnfreezePhysics(MyCubeGrid grid)
 
     public static bool NeedToCompensate(MyFunctionalBlock myCubeBlock)
     {
+        if (myCubeBlock == null || !myCubeBlock.IsWorking)
+        {
+            return false;
+        }
+
         var needToCompensate = myCubeBlock is MyProductionBlock
-                               || myCubeBlock is MyGasGenerator
-                               || myCubeBlock is MyFueledPowerProducer;
+                               || myCubeBlock is MyGasTank;
         return needToCompensate;
     }
 

SentisOptimisations/Freezer/FreezerPatches.cs

@@ -1,10 +1,13 @@
 using System;
 using System.Reflection;
+using NLog.Fluent;
 using Sandbox.Engine.Physics;
 using Sandbox.Game;
 using Sandbox.Game.Entities;
+using Sandbox.Game.Entities.Blocks;
 using Sandbox.Game.Entities.Cube;
 using Sandbox.Game.Entities.Inventory;
+using Sandbox.Game.EntityComponents;
 using Torch.Managers.PatchManager;
 using VRage;
 using VRage.Game;
@@ -28,7 +31,7 @@ public static void Patch(PatchContext ctx)
         ctx.GetPattern(MethodAddItems).Prefixes.Add(
             typeof(FreezerPatches).GetMethod(nameof(AddItemsPatched),
                 BindingFlags.Static | BindingFlags.Instance | BindingFlags.NonPublic));
-        
+
         var MethodOnMotionDynamic = typeof(MyPhysicsBody).GetMethod
         ("OnMotionDynamic", BindingFlags.Instance | BindingFlags.Public | BindingFlags.DeclaredOnly);
 
@@ -45,7 +48,7 @@ public static void Patch(PatchContext ctx)
             typeof(FreezerPatches).GetMethod(nameof(RefreshCustomInfoPatched),
                 BindingFlags.Static | BindingFlags.Instance | BindingFlags.NonPublic));
     }
-
+    
     private static bool RefreshCustomInfoPatched(MyTerminalBlock __instance)
     {
         if (!SentisOptimisationsPlugin.Config.FreezerEnabled)