inewave 1.10.2__tar.gz → 1.10.4__tar.gz

{inewave-1.10.2 → inewave-1.10.4}/CHANGELOG.md

@@ -1,3 +1,14 @@
+# 1.10.4
+
+- Correção na escrita do arquivo `ghmin.dat` em linhas relativas aos anos "POS" [#106](https://github.com/rjmalves/inewave/issues/106) (@joaoCalmon).
+- Melhor processamento de variáveis ausentes ao caso no arquivo MEDIAS-USIH.CSV (@rdlobato)
+- Suporte a processar um número variável de etapas no arquivo newave.tim (@rdlobato)
+- Correção de tipagem estática para mypy
+
+# 1.10.3
+
+- Correção na leitura do arquivo `sistema.dat` em arquivos gerados com espaços adicionais nas linhas dos blocos [#104](https://github.com/rjmalves/inewave/issues/106).
+
 # 1.10.2
 
 - Correção na escrita do arquivo `nwlistop.dat` para suportar o numero de variaveis totais da opção 2 do programa NWLISTOP [#104](https://github.com/rjmalves/inewave/issues/104).

{inewave-1.10.2 → inewave-1.10.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: inewave
-Version: 1.10.2
+Version: 1.10.4
 Summary: inewave
 Project-URL: Documentation, https://rjmalves.github.io/inewave/
 Project-URL: Repository, https://github.com/rjmalves/inewave/

{inewave-1.10.2 → inewave-1.10.4}/inewave/__init__.py

@@ -6,7 +6,7 @@ inewave é um módulo em Python para manipulação dos arquivos de entrada
 e saída do NEWAVE.
 """
 
-__version__ = "1.10.2"
+__version__ = "1.10.4"
 
 from . import newave  # noqa
 from . import nwlistcf  # noqa

{inewave-1.10.2 → inewave-1.10.4}/inewave/_utils/formatacao.py

@@ -166,20 +166,34 @@ def formata_df_meses_para_datas_nwlistop(df: pd.DataFrame) -> pd.DataFrame:
     }
     return mapa_formatacao[colunas_identificacao](df)
 
+def converte_anos(anos: List[str]) -> List[int]:
+    """
+    Converte uma lista de anos representados como strings para inteiros.
+    Se o ano for 'PRE', converte para 1; se for 'POS', converte para 9999.
+    """
+    anos_convertidos = []
+    for a in anos:
+        if a == "PRE":
+            anos_convertidos.append(1)
+        elif a == "POS":
+            anos_convertidos.append(9999)
+        else:
+            anos_convertidos.append(int(a))
+    return anos_convertidos
+
+def prepara_vetor_ano_mes_tabela(anos: List[str], meses: List[int]) -> List[datetime]:
+
+    anos_convertidos = converte_anos(anos)
+    return [
+        datetime(year=int(a), month=int(m), day=1) for a, m in zip(anos_convertidos, meses)
+    ]
+
 
 def prepara_vetor_anos_tabela(anos: List[str]) -> List[datetime]:
     # Se tem pré, substitui por 0001
     # Se tem pós, substitui por 9999
     # Repete os valores existentes 12 vezes
-    anos_convertidos: List[int] = []
-    for a in anos:
-        if a == "PRE":
-            a_convertido = 1
-        elif a == "POS":
-            a_convertido = 9999
-        else:
-            a_convertido = int(a)
-        anos_convertidos.append(a_convertido)
+    anos_convertidos = converte_anos(anos)
 
     anos_array = np.array(anos_convertidos).repeat(len(MESES_DF))
     meses = np.tile(np.arange(1, 13), len(anos))

{inewave-1.10.2 → inewave-1.10.4}/inewave/newave/modelos/ghmin.py

@@ -2,10 +2,14 @@ from cfinterface.components.section import Section
 from cfinterface.components.line import Line
 from cfinterface.components.integerfield import IntegerField
 from cfinterface.components.floatfield import FloatField
-from cfinterface.components.datetimefield import DatetimeField
+from cfinterface.components.literalfield import LiteralField
 from typing import List, IO, Optional
 import pandas as pd  # type: ignore
-from datetime import datetime
+
+from inewave._utils.formatacao import (
+    prepara_valor_ano,
+    prepara_vetor_ano_mes_tabela,
+)
 
 
 class BlocoUHEGhmin(Section):
@@ -22,7 +26,8 @@ class BlocoUHEGhmin(Section):
         self.__linha_uhe = Line(
             [
                 IntegerField(3, 0),
-                DatetimeField(7, 5, format="%m %Y"),
+                LiteralField(2, 5),
+                LiteralField(4, 8),
                 IntegerField(1, 14),
                 FloatField(6, 17, 0),
             ]
@@ -48,7 +53,7 @@ class BlocoUHEGhmin(Section):
         def converte_tabela_em_df():
             df = pd.DataFrame()
             df["codigo_usina"] = codigos
-            df["data"] = datas
+            df["data"] = prepara_vetor_ano_mes_tabela(anos, meses)
             df["patamar"] = patamares
             df["geracao"] = geracoes
             return df
@@ -59,7 +64,8 @@ class BlocoUHEGhmin(Section):
 
         # Variáveis auxiliares
         codigos: List[Optional[int]] = []
-        datas: List[Optional[datetime]] = []
+        meses: List[Optional[str]] = []
+        anos: List[Optional[str]] = []
         patamares: List[Optional[int]] = []
         geracoes: List[Optional[float]] = []
 
@@ -72,9 +78,10 @@ class BlocoUHEGhmin(Section):
                 break
             dados = self.__linha_uhe.read(linha)
             codigos.append(dados[0])
-            datas.append(dados[1])
-            patamares.append(dados[2])
-            geracoes.append(dados[3])
+            meses.append(dados[1])
+            anos.append(dados[2])
+            patamares.append(dados[3])
+            geracoes.append(dados[4])
 
     # Override
     def write(self, file: IO, *args, **kwargs):
@@ -82,8 +89,12 @@ class BlocoUHEGhmin(Section):
             file.write(linha)
         if not isinstance(self.data, pd.DataFrame):
             raise ValueError("Dados do ghmin.dat não foram lidos com sucesso")
+        df_aux = self.data.copy()
+        df_aux['ano'] = df_aux['data'].apply(lambda x: prepara_valor_ano(x.year))
+        df_aux['mes'] = df_aux['data'].apply(lambda x: f"{x.month:2d}")
+        df_aux.drop("data", inplace=True, axis=1)
 
-        for _, linha in self.data.iterrows():
+        for _, linha in df_aux[['codigo_usina','mes','ano','patamar','geracao']].iterrows():
             linha_lida: pd.Series = linha
             dados_linha = linha_lida.tolist()
             file.write(self.__linha_uhe.write(dados_linha))

{inewave-1.10.2 → inewave-1.10.4}/inewave/newave/modelos/newavetim.py

@@ -51,8 +51,10 @@ class BlocoTemposEtapasTim(Block):
         tempos: List[timedelta] = []
 
         # Leitura das etapas
-        for _ in range(5):
+        while True:
             dados = self.__line.read(file.readline())
+            if not dados or not dados[0].strip("- \t\n"):
+                break
             etapas.append(dados[0].split(":")[0])
             h = int(dados[1].split("h")[0])
             min = int(dados[1].split("h")[1].split("min")[0])

{inewave-1.10.2 → inewave-1.10.4}/inewave/newave/modelos/sistema.py

@@ -285,7 +285,7 @@ class BlocoIntercambioSubsistema(Section):
                     self.data = converte_tabela_em_df()
                 break
             # Confere se é uma linha de subsistema ou tabela
-            if len(linha) < 30:
+            if len(linha.strip()) < 30:
                 dados = self.__linha_subsis.read(linha)
                 subsis_de_atual = (
                     subsis_de_atual if dados[0] is None else dados[0]
@@ -575,7 +575,7 @@ class BlocoGeracaoUsinasNaoSimuladas(Section):
                     self.data = converte_tabela_em_df()
                 break
             # Confere se é uma linha de subsistema ou tabela
-            if len(linha) < 30:
+            if len(linha.strip()) < 30:
                 dados = self.__linha_subsis.read(linha)
                 subsis_atual = dados[0]
                 bloco_atual = dados[1]

{inewave-1.10.2 → inewave-1.10.4}/inewave/nwlistop/modelos/mediasusih.py

@@ -44,8 +44,7 @@ class TabelaMediasusih(Section):
             var_name="estagio",
             value_name="valor",
         )
-        df.loc[df["valor"] == "-   ", "valor"] = 0.0
-        df = df.astype({"valor": float})
+        df["valor"] = pd.to_numeric(df["valor"], errors="coerce")
         df["estagio"] = df["estagio"].astype(int)
         df["estagio"] -= df["estagio"].min() - 1
         df = df.pivot_table(