Agrint.py

from RecebeDados import RecebeDados;

class Agrint:
    
    # obs: AGRINT significa agrupamento de interligacoes
    def __init__(self, recebe_dados, iagrint, nper, nperPos, nPatamares):
        
        # define fonte_dados como o objeto da classe RecebeDados e o index interno do AgrInt
        self.fonte_dados = recebe_dados;
        self.indexAgrintInterno = iagrint;
        self.nper = nper;
        self.nperPos = nperPos;
        self.npat = nPatamares;
        
        # declara os vetores com os dados
        self.fluxos = [];
        self.limites = [[0 for iper in range(0,self.nper)] for ipat in range(0, self.npat)];
        
        # chama o metodo que importa os agrupamentos
        self.importaAgrupamento();
               
        return;
        
    def construirLista(self):
        # inicializa com zeros
        lista = [[0 for isis in range(14)] for jsis in range(14)];
        
        # insere os fluxos
        for (isis,jsis) in self.fluxos:
            lista[isis][jsis] = 1;
            
        return lista;
    
    def importaAgrupamento(self):
        
        # reforca a aba da vez e cria as variaveis auxiliares
        self.fonte_dados.defineAba("AGRINT");
        linhaOffset = 4*self.indexAgrintInterno;
        colunaOffset = 0;
        
        # importa as informacoes de fato
        self.indexAgrintExterno = self.fonte_dados.pegaEscalar("A50", lin_offset=self.indexAgrintInterno);
        self.numInterligacoes = self.fonte_dados.pegaEscalar("B50", lin_offset=self.indexAgrintInterno);
        while (self.fonte_dados.pegaEscalar("C50", lin_offset=self.indexAgrintInterno, col_offset = colunaOffset) is not None):
            vetDePara = [int(self.fonte_dados.pegaEscalar("C50", lin_offset=self.indexAgrintInterno, col_offset = colunaOffset)-1), int(self.fonte_dados.pegaEscalar("C50", lin_offset=self.indexAgrintInterno, col_offset = colunaOffset+1)-1)];
            self.fluxos.append(vetDePara);

            # pega os limites desse agrupamento
            for ipat in range(0, self.npat):
                self.limites[ipat] = self.fonte_dados.pegaVetor("C2", lin_offset = linhaOffset, direcao = "horizontal", tamanho=self.nper);
                linhaOffset += 1;
                # repete o ultimo ano para o periodo pós
                for iper in range(self.nperPos):
                    self.limites[ipat].append(self.limites[ipat][self.nper - 12 + iper%12]);

            # soma-se dois porque tem que pular a coluna do para referente ao agrint em questao
            colunaOffset += 2;
            linhaOffset = 4*self.indexAgrintInterno;
        
        return;