Merge pull request #2747 from AnsbigetHildElarik/thermodynamics-pyfa

DarkFenX · web-flow · commit cbf9c577d71d · 2026-04-30T23:05:01.000+02:00
Thermodynamics Calculations
diff --git a/gui/builtinViewColumns/heat.py b/gui/builtinViewColumns/heat.py
@@ -0,0 +1,295 @@
+# =============================================================================
+# 2026 Ansbiget Hild Elarik
+#
+# This file is part of pyfa.
+#
+# pyfa is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# pyfa is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
+# =============================================================================
+
+# noinspection PyPackageRequirements
+import wx
+import math
+
+from gui.bitmap_loader import BitmapLoader
+from eos.const import FittingModuleState
+from eos.saveddata.fit import Fit
+from eos.saveddata.module import Module
+from gui.viewColumn import ViewColumn
+from service.fit import Fit
+
+import gui.mainFrame
+
+class Thermodynamics():
+    def __init__(self, fit):
+        self.fit = fit
+        self.hgm = fit.ship.getModifiedItemAttr("heatGenerationMultiplier")
+        self.harm = self.calcHeatAbsorbtionRateModifier()
+        self.slotfactor = self.calcSlotFactor()
+        self.simTime = 120
+
+    def getSlotPos(self, mod): # get rack position of mod, 0-7
+        rack = []
+        for m in self.fit.modules:
+            if m.slot == mod.slot:
+                rack.insert(0, m)
+
+        for i, m in enumerate(rack):
+            if m == mod:
+                return i
+
+    def calcHeatAbsorbtionRateModifier(self):
+        harm = [0,0,0,0] # 0 is a dummy slot, align with mod.slot constants, 1=low, 2=med, 3=hi, 4=rig, ...
+
+        for mod in self.fit.modules:
+            if(mod.state == FittingModuleState.OVERHEATED):
+                harm[mod.slot] += mod.getModifiedItemAttr("heatAbsorbtionRateModifier")
+
+        return harm
+
+        """
+        HANGAR.ShipInfoThermodynamics.prototype.getHARM = function() {
+            var harm = [0,0,0];
+            var rack = ["hs", "ms", "ls"];
+
+            for(var i = 0; i < rack.length; i++) {
+
+                for(var j = 1; j <= 8; j++) {
+                    // if slot and slot is overheated
+                    if(this.shipinfo.ship.slots[rack[i]+j] && this.fitwindow.slots[rack[i]+j].find(" .sloticon").hasClass("heat") ) {
+                        harm[i] += this.shipinfo.ship.slots[rack[i]+j].heatAbsorbtionRateModifier;
+                    }
+                }
+            }
+
+            return harm;
+        };
+        """
+
+    def calcSlotFactor(self):
+        slots = self.fit.ship.getModifiedItemAttr("hiSlots") + self.fit.ship.getModifiedItemAttr("medSlots") + self.fit.ship.getModifiedItemAttr("lowSlots")
+        empty = self.fit.getSlotsFree(3) + self.fit.getSlotsFree(2) + self.fit.getSlotsFree(1) # FittingSlot.HIGH doesn"t work here?
+        rigslots = self.fit.getNumSlots(4)
+
+        return (slots - empty) / (slots + rigslots)
+
+        """
+            HANGAR.ShipInfoThermodynamics.prototype.getSlotFactor = function() {
+            var slots = 0;
+            var emptyslots = 0;
+            for(var i = 1; i <= 8; i++) {
+
+                var hs = this.fitwindow.slots["hs"+i];
+                var ms = this.fitwindow.slots["ms"+i];
+                var ls = this.fitwindow.slots["ls"+i];
+
+                if(hs.hasClass("highslot") ) {
+                    slots++;
+                    if(!hs.hasClass("occupied") || hs.hasClass("offline") ) {
+                        emptyslots++;
+                    }
+                }
+                if(ms.hasClass("midslot") ) {
+                    slots++;
+                    if(!ms.hasClass("occupied") || ms.hasClass("offline") ) {
+                        emptyslots++;
+                    }
+                }
+                if(ls.hasClass("lowslot") ) {
+                    slots++;
+                    if(!ls.hasClass("occupied") || ls.hasClass("offline") ) {
+                        emptyslots++;
+                    }
+                }
+            }
+
+            return (slots-emptyslots)/(slots + this.shipinfo.ship.data.rigSlots);
+        };
+        """
+
+    def calcDamageProbability(self, mod, t): # get chance the module is damaged when overheated at time t
+        keys = ["", "heatAttenuationLow", "heatAttenuationMed", "heatAttenuationHi"]
+        att = self.fit.ship.getModifiedItemAttr(keys[mod.slot], 0.25)
+        rackheat = 1 - pow(math.e, (-t * self.hgm * self.harm[mod.slot]))
+        slotpos = self.getSlotPos(mod)
+
+        probs = []
+        for m in self.fit.modules:
+            if (m == mod): continue
+            if m.slot == mod.slot:
+                if m.state == FittingModuleState.OVERHEATED:
+                    i = self.getSlotPos(m)
+                    pos = abs(i - slotpos) # get rack distance to other overheated module
+                    probs.append(pow(att, pos) * self.slotfactor * rackheat)
+
+        p = 1
+        for i in range(0, len(probs)):
+            p *= (1 - probs[i])
+
+        selfprob = self.slotfactor * rackheat
+        res = selfprob if p == 1 else 1 - p * (1 - selfprob)
+
+        return res
+
+        """
+        HANGAR.ShipInfoThermodynamics.prototype.getDamageProb = function(slot, t) {
+            var rack = slot[0] == "h" ? "hs" : slot[0] == "m" ? "ms" : "ls";
+            var harmNdx = rack === "hs" ? 0 : rack === "ms" ? 1 : 2;
+            var att = rack == "hs" ? this.shipinfo.ship.data.heatAttenuationHi :
+                    rack == "ms" ? this.shipinfo.ship.data.heatAttenuationMed :
+                    this.shipinfo.ship.data.heatAttenuationLow ?
+                    this.shipinfo.ship.data.heatAttenuationLow : 0.25;
+
+            var slotpos = parseInt( slot.substr(2) );
+            var rackheat = 1 + -Math.pow(Math.E, (-t * this.hgm * this.harm[harmNdx]));
+
+            var prob = [];
+            for(var i = 1; i <= 8; i++) {
+                if(rack+i == slot) continue;
+                if(this.shipinfo.ship.slots[ rack+i ] && this.shipinfo.ship.slots[ rack+i ].state === "overload"){
+                    var pos = Math.abs(i - slotpos);
+                    prob.push( Math.pow(att, pos)*this.slotfactor*rackheat );
+                }
+            }
+
+            var p = 1;
+            for(var i = 0; i < prob.length; i++) {
+                p *= (1-prob[i]);
+            }
+
+            var selfprob = this.slotfactor * rackheat;
+            if(p === 1) {
+                return selfprob;
+            } else {
+                return 1 - p*(1-selfprob);
+            }
+        };
+        """
+
+    def calcBurnCycles(self, mod): # estimates the number of cycles a module will OH before it burns out
+        speed = mod.getModifiedItemAttr("speed")
+        duration = mod.getModifiedItemAttr("duration")
+        inc = speed / 1000 if speed else duration / 1000
+        t = inc
+
+        fp = [] # failure probabilities
+        p = lastp = 0
+        while(t < self.simTime):
+            p = self.calcDamageProbability(mod, t)
+            fp.append(p)
+
+            if f"{p:.2f}" == f"{lastp:.2f}":
+                break
+
+            t += inc
+            lastp = p
+
+        E = 0 # expected wait to failure
+        n = math.ceil(mod.getModifiedItemAttr("hp") / mod.getModifiedItemAttr("heatDamage")) # fault tolerance
+        a = [1]
+
+        for i in range(n):
+            a.append(0)
+
+        for t, fp_t in enumerate(fp):
+            E += (t + 1) * fp_t * a[n - 1]
+
+            for k in range(n - 1, 0, -1):
+                a[k] = (1 - fp_t) * a[k] + fp_t * a[k - 1]
+
+            a[0] = (1 - fp_t) * a[0]
+
+        for k in range(n):
+            E += (t + 1 + (n - k) * (1 / fp[t])) * a[k]
+
+        return math.floor(E)
+
+        """
+        HANGAR.ShipInfoThermodynamics.prototype.calcBurnCycles = function(slot) {
+            var fp = [];
+            var p = 0, lastp = 0;
+            var mod = this.shipinfo.ship.slots[slot];
+            var inc = mod.speed ? mod.speed/1000 : mod.duration/1000;
+            var t = inc;
+
+            while(t < this.simTime) {
+                p = this.getDamageProb(slot, t);
+                fp.push(p);
+                if(p.toFixed(2) === lastp.toFixed(2)) break;
+                t += inc;
+                lastp = p;
+            }
+
+            //http://jsfiddle.net/kkspy/86/
+            var E = 0;
+            var n = Math.ceil(mod.hp / mod.heatDamage);
+            var a = [1];
+            for(var i = 1; i < n;i++) { a.push(0); }
+
+            for(var t = 0; t < fp.length; t++) {
+                E += (t+1)*fp[t]*a[n-1];
+                for(var k = n-1; k > 0; k--) {
+                    a[k] = (1-fp[t])*a[k] + fp[t]*a[k-1];
+                }
+                a[0] = (1-fp[t])*a[0];
+            }
+
+            t--;
+            for(var k = 0; k < n; k++) {
+                E += ( t+1 + (n-k)*(1/fp[t]))*a[k];
+            }
+
+            return Math.floor(E);
+        };
+        """
+
+class Heat(ViewColumn):
+    name = "Heat"
+
+    def __init__(self, fittingView, params):
+        ViewColumn.__init__(self, fittingView)
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.resizable = False
+        self.size = 54
+        self.maxsize = self.size * 2
+        self.imageId = fittingView.imageList.GetImageIndex("state_overheated_small", "gui")
+        self.bitmap = BitmapLoader.getBitmap("state_overheated_small", "gui")
+        self.mask = wx.LIST_MASK_IMAGE
+
+    def getText(self, mod):
+        if not isinstance(mod, Module) or mod.state != FittingModuleState.OVERHEATED:
+            return ""
+
+        thermo = Thermodynamics(Fit.getInstance().getFit(self.mainFrame.getActiveFit()))
+        burnCycles = thermo.calcBurnCycles(mod)
+        duration = mod.getModifiedItemAttr("duration") / 1000
+        speed = mod.getModifiedItemAttr("speed") / 1000
+        cycleTime = duration or speed
+
+        t = burnCycles * cycleTime
+        s = t % 60
+        m = (t / 60) % 60
+        h = (t / 3600) % 24
+        out = [f"{int(m):02d}", f"{int(s):02d}"]
+
+        if int(h) > 0: # hours is rarely relevant, only show if it is
+            out.insert(0, f"{int(h):02d}")
+
+        return ":".join(out) # display as 00:00:00 to vertically align across slot cols consistently
+
+    def getToolTip(self, mod):
+        if isinstance(mod, Module) and mod.state == FittingModuleState.OVERHEATED:
+            return "Estimated time til burnout" # TODO localize
+
+
+Heat.register()
diff --git a/gui/builtinViews/fittingView.py b/gui/builtinViews/fittingView.py
@@ -148,6 +148,7 @@ class FittingView(d.Display):
                     "Miscellanea",
                     "Price",
                     "Ammo",
+                    "Heat",
                     ]
 
     def __init__(self, parent):
diff --git a/gui/viewColumn.py b/gui/viewColumn.py
@@ -83,6 +83,7 @@ def delayedText(self, display, colItem):
     graphColor,
     graphLightness,
     graphLineStyle,
+    heat,
     maxRange,
     misc,
     price,

Original file line number	Diff line number	Diff line change
`@@ -148,6 +148,7 @@ class FittingView(d.Display):`
`148`	`148`	`"Miscellanea",`
`149`	`149`	`"Price",`
`150`	`150`	`"Ammo",`
	`151`	`+ "Heat",`
`151`	`152`	`]`
`152`	`153`
`153`	`154`	`def __init__(self, parent):`