idecomp 1.2.0__tar.gz → 1.4.0__tar.gz

{idecomp-1.2.0 → idecomp-1.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: idecomp
-Version: 1.2.0
+Version: 1.4.0
 Summary: Interface para arquivos do DECOMP
 Home-page: https://github.com/rjmalves/idecomp
 Author: Rogerio Alves
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE.md
 Requires-Dist: numpy
 Requires-Dist: pandas
-Requires-Dist: cfinterface>=1.5.3
+Requires-Dist: cfinterface>=1.7.0
 
 # idecomp
 

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/__init__.py

@@ -6,6 +6,6 @@ idecomp é um módulo em Python para manipulação dos arquivos de entrada
 e saída do DECOMP.
 """
 
-__version__ = "1.2.0"
+__version__ = "1.4.0"
 
 from . import decomp  # noqa

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/config.py

@@ -5,7 +5,7 @@ MAX_SUBSISTEMAS = 10
 MAX_UTES = 300
 MAX_UHES = 300
 MAX_REES = 20
-MAX_CENARIOS = 999
+MAX_CENARIOS = 9999
 MESES = [
     "JANEIRO",
     "FEVEREIRO",
@@ -48,3 +48,4 @@ MESES_ABREV = [
     "NOV",
     "DEZ",
 ]
+FORMATO_CAMPOS_DATA_LIBS = ["%Y/%m/%d %H:%M", "%Y/%m/%d", "%Y/%m"]

idecomp-1.4.0/idecomp/decomp/avl_cortesfpha_dec.py

@@ -0,0 +1,35 @@
+from idecomp.decomp.modelos.blocos.versaomodelo import VersaoModelo
+from idecomp.decomp.modelos.avl_cortesfpha_dec import TabelaCortesFpha
+
+from idecomp.decomp.modelos.arquivoscsv.arquivocsv import ArquivoCSV
+from typing import Optional
+import pandas as pd  # type: ignore
+
+
+class AvlCortesFpha(ArquivoCSV):
+    """
+    Arquivo com os cortes da função de produção hidráulica aproximada do DECOMP.
+    """
+
+    BLOCKS = [VersaoModelo, TabelaCortesFpha]
+
+    @property
+    def tabela(self) -> Optional[pd.DataFrame]:
+        """
+        A tabela de dados que está contida no arquivo.
+
+        - codigo_usina (`int`)
+        - estagio (`int`)
+        - nome_usina (`str`)
+        - segmento_fpha (`int`)
+        - fator_correcao (`float`)
+        - rhs (`float`)
+        - coeficiente_volume_util (`float`)
+        - coeficiente_vazao_turbinada (`float`)
+        - coeficiente_vazao_vertida (`float`)
+        - coeficiente_vazao_lateral (`float`)
+
+        :return: A tabela como um dataframe
+        :rtype: pd.DataFrame | None
+        """
+        return self._tabela()

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dadger.py

@@ -84,6 +84,7 @@ from idecomp.decomp.modelos.dadger import (
     VL,
     VA,
     VU,
+    DA,
 )
 
 import pandas  # type: ignore
@@ -211,6 +212,7 @@ class Dadger(RegisterFile):
         VL,
         VA,
         VU,
+        DA,
     ]
 
     def __init__(self, data=...) -> None:
@@ -1000,6 +1002,35 @@ class Dadger(RegisterFile):
         """
         return self.__registros_ou_df(TI, codigo_usina=codigo_usina, df=df)
 
+    def da(
+        self,
+        codigo_usina_retirada: Optional[int] = None,
+        codigo_usina_retorno: Optional[int] = None,
+        df: bool = False,
+    ) -> Optional[Union[DA, List[DA], pandas.DataFrame]]:
+        """
+        Obtém um registro que especifica as retiradas de água para
+        outros usos (desvios de água) por usina (UHE) existente no
+        estudo especificado no :class:`Dadger`
+
+        :param codigo_usina_retirada: Código da UHE a montante da qual será feita a retirada
+        :type codigo_usina_retirada: int | None
+        :param codigo_usina_retorno: Código da UHE a montante da qual se derá o retorno.
+        :type codigo_usina_retorno: int | None
+        :param df: ignorar os filtros e retornar
+            todos os dados de registros como um DataFrame
+        :type df: bool
+
+        :return: Um ou mais registros, se existirem.
+        :rtype: :class:`DA` | list[:class:`DA`] | :class:`pandas.DataFrame` | None
+        """
+        return self.__registros_ou_df(
+            DA,
+            codigo_usina_retirada=codigo_usina_retirada,
+            codigo_usina_retorno=codigo_usina_retorno,
+            df=df,
+        )
+
     def mp(
         self,
         codigo_usina: Optional[int] = None,

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_avl_evap.py

@@ -18,7 +18,7 @@ class DecAvlEvap(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - codigo_usina (`int`)
         - nome_usina (`str`)
         - codigo_submercado (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_cortes_evap.py

@@ -18,7 +18,7 @@ class DecCortesEvap(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - codigo_usina (`int`)
         - nome_usina (`str`)
         - codigo_submercado (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_desvfpha.py

@@ -19,7 +19,7 @@ class DecDesvFpha(ArquivoCSV):
         A tabela de dados que está contida no arquivo.
 
         - codigo_usina (`int`)
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - patamar (`int`)
         - nome_usina (`str`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_eco_discr.py

@@ -19,7 +19,7 @@ class DecEcoDiscr(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - patamar (`int`)
         - duracao (`float`)
         - numero_patamares (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_gnl.py

@@ -23,7 +23,7 @@ class DecOperGnl(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - patamar (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_interc.py

@@ -25,7 +25,7 @@ class DecOperInterc(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - patamar (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_ree.py

@@ -22,7 +22,7 @@ class DecOperRee(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - codigo_ree (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_sist.py

@@ -25,7 +25,7 @@ class DecOperSist(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - patamar (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_usie.py

@@ -25,7 +25,7 @@ class DecOperUsie(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - patamar (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_usih.py

@@ -25,7 +25,7 @@ class DecOperUsih(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - patamar (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/dec_oper_usit.py

@@ -25,7 +25,7 @@ class DecOperUsit(ArquivoCSV):
         """
         A tabela de dados que está contida no arquivo.
 
-        - periodo (`int`)
+        - estagio (`int`)
         - no (`int`)
         - cenario (`int`)
         - patamar (`int`)

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/modelos/arquivos.py

@@ -11,6 +11,8 @@ class BlocoNomesArquivos(Section):
     arquivos de entrada do DECOMP.
     """
 
+    __slots__ = ["__linha"]
+
     def __init__(self, previous=None, next=None, data=None) -> None:
         super().__init__(previous, next, data)
         self.__linha = Line([LiteralField(80, 0)])

idecomp-1.4.0/idecomp/decomp/modelos/avl_cortesfpha_dec.py

@@ -0,0 +1,46 @@
+# Imports de módulos externos
+from cfinterface.components.line import Line
+from cfinterface.components.integerfield import IntegerField
+from cfinterface.components.literalfield import LiteralField
+from cfinterface.components.floatfield import FloatField
+
+
+from idecomp.decomp.modelos.blocos.tabelacsv import TabelaCSV
+
+
+class TabelaCortesFpha(TabelaCSV):
+    """
+    Bloco com as informações dos cortes da função de produção hidráulica.
+    """
+
+    __slots__ = []
+
+    BEGIN_PATTERN = "-----;-----;--------------;"
+    LINE_MODEL = Line(
+        [
+            IntegerField(size=5),
+            IntegerField(size=5),
+            LiteralField(size=14),
+            IntegerField(size=7),
+            FloatField(size=10, decimal_digits=6),
+            FloatField(size=16, decimal_digits=8),
+            FloatField(size=16, decimal_digits=8),
+            FloatField(size=16, decimal_digits=8),
+            FloatField(size=16, decimal_digits=8),
+            FloatField(size=16, decimal_digits=8),
+        ],
+        delimiter=";",
+    )
+    COLUMN_NAMES = [
+        "codigo_usina",
+        "estagio",
+        "nome_usina",
+        "segmento_fpha",
+        "fator_correcao",
+        "rhs",
+        "coeficiente_volume_util",
+        "coeficiente_vazao_turbinada",
+        "coeficiente_vazao_vertida",
+        "coeficiente_vazao_lateral",
+    ]
+    END_PATTERN = ""

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/modelos/blocos/tabelacsv.py

@@ -11,6 +11,8 @@ class TabelaCSV(Block):
     a partir de um modelo de linha, para arquivos de saída do DECOMP.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = "-----;------;"
     LINE_MODEL = Line([])
     COLUMN_NAMES: List[str] = []
@@ -66,6 +68,8 @@ class TabelaCSVLibs(Block):
     a partir de um modelo de linha, para arquivos de saída do DECOMP.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = "&IIIIIII;IIIIIII;"
     LINE_MODEL = Line([])
     COLUMN_NAMES: List[str] = []

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/modelos/blocos/versaomodelo.py

@@ -11,6 +11,8 @@ class VersaoModelo(Block):
     título do arquivo.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = r"CEPEL: DECOMP"
     END_PATTERN = ""
 
@@ -34,6 +36,8 @@ class VersaoModeloLibs(Block):
     título do arquivo.
     """
 
+    __slots__ = []
+
     BEGIN_PATTERN = r"Nome do Modelo: DECOMP Versão:"
     END_PATTERN = ""
 

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/modelos/caso.py

@@ -10,6 +10,8 @@ class NomeCaso(Section):
     entrada do DECOMP `caso.dat`.
     """
 
+    __slots__ = ["__linha"]
+
     def __init__(self, previous=None, next=None, data=None) -> None:
         super().__init__(previous, next, data)
         self.__linha = Line([LiteralField(80, 0)])

{idecomp-1.2.0 → idecomp-1.4.0}/idecomp/decomp/modelos/custos.py

@@ -13,12 +13,20 @@ from typing import IO, List
 class BlocoRelatorioCustos(Block):
     """ """
 
+    __slots__ = [
+        "__scenario_line",
+        "__dual_variables_line",
+        "__fcf_line",
+        "dados_cenario",
+        "data",
+    ]
+
     BEGIN_PATTERN = r" RELATORIO DAS VARIAVEIS DUAIS NO ESTAGIO"
     END_PATTERN = "X----X-"
 
     def __init__(self, previous=None, next=None, data=None) -> None:
         super().__init__(previous, next, data)
-        self.__scenario_line = Line([IntegerField(2, 44), IntegerField(3, 55)])
+        self.__scenario_line = Line([IntegerField(2, 44), IntegerField(4, 54)])
         self.__dual_variables_line = Line(
             [
                 LiteralField(14, 4),