cleanup class_ems.py

- type hints added
- translated to English
- reduced not needed function calls 

improves optimization time by 50%
This commit is contained in:
Normann 2024-10-05 21:26:31 +02:00 committed by Andreas
parent 3b5fc18665
commit c31a6c457f

View File

@ -1,18 +1,22 @@
from datetime import datetime
from typing import Dict, List, Optional, Union
import numpy as np
def replace_nan_with_none(data):
if isinstance(data, dict):
def replace_nan_with_none(
data: Union[np.ndarray, dict, list, float],
) -> Union[List, dict, float, None]:
if data is None:
return None
if isinstance(data, np.ndarray):
# Use numpy vectorized approach
return np.where(np.isnan(data), None, data).tolist()
elif isinstance(data, dict):
return {key: replace_nan_with_none(value) for key, value in data.items()}
elif isinstance(data, list):
return [replace_nan_with_none(element) for element in data]
elif isinstance(data, np.ndarray):
# Konvertiere das numpy-Array zu einer Liste und rekursiv ersetzen
return replace_nan_with_none(data.tolist())
elif isinstance(data, (float, np.floating)) and np.isnan(data):
# np.floating deckt auch numpy-NaNs ab
return None
else:
return data
@ -21,112 +25,82 @@ def replace_nan_with_none(data):
class EnergieManagementSystem:
def __init__(
self,
pv_prognose_wh=None,
strompreis_euro_pro_wh=None,
einspeiseverguetung_euro_pro_wh=None,
eauto=None,
gesamtlast=None,
haushaltsgeraet=None,
wechselrichter=None,
pv_prognose_wh: Optional[np.ndarray] = None,
strompreis_euro_pro_wh: Optional[np.ndarray] = None,
einspeiseverguetung_euro_pro_wh: Optional[np.ndarray] = None,
eauto: Optional[object] = None,
gesamtlast: Optional[np.ndarray] = None,
haushaltsgeraet: Optional[object] = None,
wechselrichter: Optional[object] = None,
):
self.akku = wechselrichter.akku
# self.lastkurve_wh = lastkurve_wh
self.gesamtlast = gesamtlast
self.pv_prognose_wh = pv_prognose_wh
self.strompreis_euro_pro_wh = (
strompreis_euro_pro_wh # Strompreis in Cent pro Wh
)
self.einspeiseverguetung_euro_pro_wh = (
einspeiseverguetung_euro_pro_wh # Einspeisevergütung in Cent pro Wh
)
self.strompreis_euro_pro_wh = strompreis_euro_pro_wh
self.einspeiseverguetung_euro_pro_wh = einspeiseverguetung_euro_pro_wh
self.eauto = eauto
self.haushaltsgeraet = haushaltsgeraet
self.wechselrichter = wechselrichter
def set_akku_discharge_hours(self, ds):
def set_akku_discharge_hours(self, ds: List[int]) -> None:
self.akku.set_discharge_per_hour(ds)
def set_eauto_charge_hours(self, ds):
def set_eauto_charge_hours(self, ds: List[int]) -> None:
self.eauto.set_charge_per_hour(ds)
def set_haushaltsgeraet_start(self, ds, global_start_hour=0):
def set_haushaltsgeraet_start(
self, ds: List[int], global_start_hour: int = 0
) -> None:
self.haushaltsgeraet.set_startzeitpunkt(ds, global_start_hour=global_start_hour)
def reset(self):
def reset(self) -> None:
self.eauto.reset()
self.akku.reset()
def simuliere_ab_jetzt(self):
def simuliere_ab_jetzt(self) -> dict:
jetzt = datetime.now()
start_stunde = jetzt.hour
# Berechne die Anzahl der Stunden bis zum gleichen Zeitpunkt am nächsten Tag
stunden_bis_ende_tag = 24 - start_stunde
# Füge diese Stunden zum nächsten Tag hinzu
gesamt_stunden = stunden_bis_ende_tag + 24
# Beginne die Simulation ab der aktuellen Stunde und führe sie für die berechnete Dauer aus
return self.simuliere(start_stunde)
def simuliere(self, start_stunde):
def simuliere(self, start_stunde: int) -> dict:
# Ensure arrays have the same length
lastkurve_wh = self.gesamtlast
# Anzahl der Stunden berechnen
assert (
len(lastkurve_wh)
== len(self.pv_prognose_wh)
== len(self.strompreis_euro_pro_wh)
), f"Arraygrößen stimmen nicht überein: Lastkurve = {len(lastkurve_wh)}, PV-Prognose = {len(self.pv_prognose_wh)}, Strompreis = {len(self.strompreis_euro_pro_wh)}"
ende = min(
len(lastkurve_wh),
len(self.pv_prognose_wh),
len(self.strompreis_euro_pro_wh),
)
), f"Array sizes do not match: Load Curve = {len(lastkurve_wh)}, PV Forecast = {len(self.pv_prognose_wh)}, Electricity Price = {len(self.strompreis_euro_pro_wh)}"
# Optimized total hours calculation
ende = len(lastkurve_wh)
total_hours = ende - start_stunde
# Initialisierung der Arrays mit NaN-Werten
last_wh_pro_stunde = np.full(total_hours, np.nan)
netzeinspeisung_wh_pro_stunde = np.full(total_hours, np.nan)
netzbezug_wh_pro_stunde = np.full(total_hours, np.nan)
kosten_euro_pro_stunde = np.full(total_hours, np.nan)
einnahmen_euro_pro_stunde = np.full(total_hours, np.nan)
akku_soc_pro_stunde = np.full(total_hours, np.nan)
eauto_soc_pro_stunde = np.full(total_hours, np.nan)
verluste_wh_pro_stunde = np.full(total_hours, np.nan)
haushaltsgeraet_wh_pro_stunde = np.full(total_hours, np.nan)
# Pre-allocate arrays for the results, optimized for speed
last_wh_pro_stunde = np.zeros(total_hours)
netzeinspeisung_wh_pro_stunde = np.zeros(total_hours)
netzbezug_wh_pro_stunde = np.zeros(total_hours)
kosten_euro_pro_stunde = np.zeros(total_hours)
einnahmen_euro_pro_stunde = np.zeros(total_hours)
akku_soc_pro_stunde = np.zeros(total_hours)
eauto_soc_pro_stunde = np.zeros(total_hours)
verluste_wh_pro_stunde = np.zeros(total_hours)
haushaltsgeraet_wh_pro_stunde = np.zeros(total_hours)
# Setze den initialen Ladezustand für Akku und E-Auto
akku_soc_pro_stunde[start_stunde] = self.akku.ladezustand_in_prozent()
# Set initial state
akku_soc_pro_stunde[0] = self.akku.ladezustand_in_prozent()
if self.eauto:
eauto_soc_pro_stunde[start_stunde] = self.eauto.ladezustand_in_prozent()
eauto_soc_pro_stunde[0] = self.eauto.ladezustand_in_prozent()
for stunde in range(start_stunde + 1, ende):
stunde_since_now = stunde - start_stunde
# print(stunde_since_now)
# Anfangszustände
akku_soc_start = self.akku.ladezustand_in_prozent()
eauto_soc_start = (
self.eauto.ladezustand_in_prozent() if self.eauto else None
)
# Verbrauch und zusätzliche Lasten bestimmen
# Accumulate loads and PV generation
verbrauch = self.gesamtlast[stunde]
haushalts_last = 0
if self.haushaltsgeraet:
verbrauch += self.haushaltsgeraet.get_last_fuer_stunde(stunde)
haushaltsgeraet_wh_pro_stunde[stunde_since_now] = verbrauch
if self.haushaltsgeraet is not None:
haushalts_last = self.haushaltsgeraet.get_last_fuer_stunde(stunde)
verbrauch += haushalts_last
haushaltsgeraet_wh_pro_stunde[stunde_since_now] = haushalts_last
# PV-Erzeugung und Strompreis für die Stunde
erzeugung = self.pv_prognose_wh[stunde]
strompreis = self.strompreis_euro_pro_wh[stunde]
# Verluste initialisieren
verluste_wh_pro_stunde[stunde_since_now] = 0.0
# E-Auto-Verbrauch bestimmen
# E-Auto handling
if self.eauto:
geladene_menge_eauto, verluste_eauto = self.eauto.energie_laden(
None, stunde
@ -137,31 +111,34 @@ class EnergieManagementSystem:
self.eauto.ladezustand_in_prozent()
)
# Wechselrichter-Logik
# Process inverter logic
erzeugung = self.pv_prognose_wh[stunde]
netzeinspeisung, netzbezug, verluste, eigenverbrauch = (
self.wechselrichter.energie_verarbeiten(erzeugung, verbrauch, stunde)
)
# Ergebnisse speichern
netzeinspeisung_wh_pro_stunde[stunde_since_now] = netzeinspeisung
netzbezug_wh_pro_stunde[stunde_since_now] = netzbezug
verluste_wh_pro_stunde[stunde_since_now] += verluste
last_wh_pro_stunde[stunde_since_now] = verbrauch
# Finanzen berechnen
kosten_euro_pro_stunde[stunde_since_now] = netzbezug * strompreis
# Financial calculations
kosten_euro_pro_stunde[stunde_since_now] = (
netzbezug * self.strompreis_euro_pro_wh[stunde]
)
einnahmen_euro_pro_stunde[stunde_since_now] = (
netzeinspeisung * self.einspeiseverguetung_euro_pro_wh[stunde]
)
# Letzter Akkuzustand speichern
# Akku SOC tracking
akku_soc_pro_stunde[stunde_since_now] = self.akku.ladezustand_in_prozent()
# Gesamtkosten berechnen
gesamtkosten_euro = np.nansum(kosten_euro_pro_stunde) - np.nansum(
# Total cost and return
gesamtkosten_euro = np.sum(kosten_euro_pro_stunde) - np.sum(
einnahmen_euro_pro_stunde
)
out = {
# Prepare output dictionary
out: Dict[str, Union[np.ndarray, float]] = {
"Last_Wh_pro_Stunde": last_wh_pro_stunde,
"Netzeinspeisung_Wh_pro_Stunde": netzeinspeisung_wh_pro_stunde,
"Netzbezug_Wh_pro_Stunde": netzbezug_wh_pro_stunde,
@ -170,12 +147,11 @@ class EnergieManagementSystem:
"Einnahmen_Euro_pro_Stunde": einnahmen_euro_pro_stunde,
"Gesamtbilanz_Euro": gesamtkosten_euro,
"E-Auto_SoC_pro_Stunde": eauto_soc_pro_stunde,
"Gesamteinnahmen_Euro": np.nansum(einnahmen_euro_pro_stunde),
"Gesamtkosten_Euro": np.nansum(kosten_euro_pro_stunde),
"Gesamteinnahmen_Euro": np.sum(einnahmen_euro_pro_stunde),
"Gesamtkosten_Euro": np.sum(kosten_euro_pro_stunde),
"Verluste_Pro_Stunde": verluste_wh_pro_stunde,
"Gesamt_Verluste": np.nansum(verluste_wh_pro_stunde),
"Gesamt_Verluste": np.sum(verluste_wh_pro_stunde),
"Haushaltsgeraet_wh_pro_stunde": haushaltsgeraet_wh_pro_stunde,
}
out = replace_nan_with_none(out)
return out
return replace_nan_with_none(out)