From f8eb1c66d38b354878e50350a453b557e9b9abd5 Mon Sep 17 00:00:00 2001
From: RealSilverMoon <31100241+RealSilverMoon@users.noreply.github.com>
Date: Sun, 30 Jun 2024 03:59:11 +0800
Subject: Several fixes for Tesla Cover (#2671)

* Fix tesla cover memory leak

* Fix tesla crash when destroying tesla cover

* Add max output display
---
 src/main/java/gregtech/api/util/GT_CoverBehavior.java  | 18 ++++++++++++++++++
 .../java/gregtech/api/util/GT_CoverBehaviorBase.java   | 11 +++++++++++
 2 files changed, 29 insertions(+)

(limited to 'src/main/java/gregtech/api/util')

diff --git a/src/main/java/gregtech/api/util/GT_CoverBehavior.java b/src/main/java/gregtech/api/util/GT_CoverBehavior.java
index 34fc151b9a..3c3a8607fe 100644
--- a/src/main/java/gregtech/api/util/GT_CoverBehavior.java
+++ b/src/main/java/gregtech/api/util/GT_CoverBehavior.java
@@ -99,6 +99,18 @@ public abstract class GT_CoverBehavior extends GT_CoverBehaviorBase<ISerializabl
         return onCoverRemoval(side, aCoverID, convert(aCoverVariable), aTileEntity, aForced);
     }
 
+    @Override
+    protected void onBaseTEDestroyedImpl(ForgeDirection side, int aCoverID,
+        ISerializableObject.LegacyCoverData aCoverVariable, ICoverable aTileEntity) {
+        onBaseTEDestroyed(side, aCoverID, convert(aCoverVariable), aTileEntity);
+    }
+
+    @Override
+    protected void onCoverUnloadImpl(ForgeDirection side, int aCoverID, ISerializableObject aCoverVariable,
+        ICoverable aTileEntity) {
+        onCoverUnload(aTileEntity);
+    }
+
     @Override
     protected String getDescriptionImpl(ForgeDirection side, int aCoverID,
         ISerializableObject.LegacyCoverData aCoverVariable, ICoverable aTileEntity) {
@@ -275,6 +287,12 @@ public abstract class GT_CoverBehavior extends GT_CoverBehaviorBase<ISerializabl
         return true;
     }
 
+    public void onCoverUnload(ICoverable aTileEntity) {
+
+    }
+
+    public void onBaseTEDestroyed(ForgeDirection side, int aCoverID, int aCoverVariable, ICoverable aTileEntity) {}
+
     /**
      * Gives a small Text for the status of the Cover.
      */
diff --git a/src/main/java/gregtech/api/util/GT_CoverBehaviorBase.java b/src/main/java/gregtech/api/util/GT_CoverBehaviorBase.java
index be9492ebba..76408e59c0 100644
--- a/src/main/java/gregtech/api/util/GT_CoverBehaviorBase.java
+++ b/src/main/java/gregtech/api/util/GT_CoverBehaviorBase.java
@@ -255,6 +255,14 @@ public abstract class GT_CoverBehaviorBase<T extends ISerializableObject> {
         return getDescriptionImpl(side, aCoverID, forceCast(aCoverVariable), aTileEntity);
     }
 
+    /**
+     * Called when Base TE being unloaded.
+     */
+    public void onCoverUnload(ForgeDirection side, int aCoverID, ISerializableObject aCoverVariable,
+        ICoverable aTileEntity) {
+        onCoverUnloadImpl(side, aCoverID, forceCast(aCoverVariable), aTileEntity);
+    }
+
     /**
      * How Blast Proof the Cover is. 30 is normal.
      */
@@ -581,6 +589,9 @@ public abstract class GT_CoverBehaviorBase<T extends ISerializableObject> {
 
     protected void onBaseTEDestroyedImpl(ForgeDirection side, int aCoverID, T aCoverVariable, ICoverable aTileEntity) {}
 
+    protected void onCoverUnloadImpl(ForgeDirection side, int aCoverID, ISerializableObject aCoverVariable,
+        ICoverable aTileEntity) {}
+
     protected boolean isRedstoneSensitiveImpl(ForgeDirection side, int aCoverID, T aCoverVariable,
         ICoverable aTileEntity, long aTimer) {
         return false;
-- 
cgit