Water-Optimization-Proyect/test.py at master · FVSB/Water-Optimization-Proyect · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import pulp as pl
import os
from Core.Solver import Solve
from Core.Auxiliar_Class import Process, Company
from Input import Read_Excel, Read_Folder
from Core.Output import Output, Output_Numerical
from Numerical.Test import Solve_Numerical


def initial_values(Process_1: Process, Process_2: Process):
    Water_From_2_to_1 = Process_1.How_Much_buy(Process_2)
    Water_From_1_to_2 = Process_2.How_Much_buy(Process_1)
    stateSupply_1 = Process_1.stateSupply()-Water_From_2_to_1
    stateSupply_2 = Process_2.stateSupply()-Water_From_1_to_2

    return Water_From_2_to_1, Water_From_1_to_2, float(stateSupply_1), float(stateSupply_2)


def Numerical_Solution(Process_1: Process, Process_2: Process, Time_of: int, file):

    Water_From_2_to_1, Water_From_1_to_2, stateSupply_1, stateSupply_2 = initial_values(
        Process_1=Process_1, Process_2=Process_2)

    start = [Water_From_1_to_2, Water_From_2_to_1, stateSupply_1,
             stateSupply_2,
             0.5, 0.5, 0.5, 0.5, 0.5, 0.5,
             0.5, 0.5]

    s, i = Solve_Numerical(Consumo_1=Process_1.Cant_Water, Consumo_2=Process_2.Cant_Water, a=Process_1.Sale_Price_State_Supply, b=Process_1.Price_discharge_water,
                           Con_Concen_1=Process_1.C_Max_Out, Con_Concen_2=Process_2.C_Max_Out, d=Process_1.Price_discharge_water, h=Time_of, Array_initial=start)
    Output_Numerical(str(Process_1.Name), s, file)


def Solution(Process_to_analysis: Process, Process_to_buy_Water: Process):
    temp = {}
    Process_1 = Process_to_analysis.Name
    Process_2 = Process_to_buy_Water.Name

    for j in range(2):
        Process_1_ofert = Process_to_analysis.water_Supply_Other(
            Process_to_buy_Water)
        sol = Solve(process_1=Process_to_analysis, process_2=Process_to_buy_Water
                    )
        # Coste optimo
        Total_Cost = pl.value(sol.objective)
        # Añadir valores de las varibles
        xs = sol.variables()
        # Cant de agua que debe proporcionar el lider
        Leader_water_supply = float(xs[0].varValue)
        # Agua que vendo
        Water_From_Hear_To_Other = Process_1_ofert
        # Agua compro
        Water_From_Other_To_Hear = float(xs[1].varValue)
        # Nombre del proceso  [Costo , Agua que debe proporcionar el lider, Agua que va del proceso al otro proceso, Agua que viene del otro proceso]
        temp[Process_to_analysis.Name] = [Total_Cost, Leader_water_supply,
                                          Water_From_Hear_To_Other, Water_From_Other_To_Hear]
        # Cambiar los valores de los procesos dado que se conoce el valor
        # que va del primer proceso al segundo
       # Process_to_buy_Water.Process_To_send[(
        #  Process_to_analysis.Company, Process_to_analysis.Name)][0] = Water_From_Hear_To_Other

        # Swap los procesos
        be = Process_to_analysis
        af = Process_to_buy_Water
        Process_to_analysis = af
        Process_to_buy_Water = be

        # Arreglar el valor de venta real del Primer
        # proceso a analizar pq este asume que el 2do
        # le compra todo la oferta de agua
        if (j > 0):
            t = temp[Process_1][3]
            val = temp[Process_2][3]
            temp[Process_1][2] = val

    return temp, Process_to_analysis, Process_to_buy_Water


def Start(Time_of: int):
    """_summary_

    Args:
        Time_of (int): time of the simulation
    """
    # Leer el archivo de excel
    All_tupples = Read_Folder()
    for t in All_tupples:
        path = os.path.join("Data_Base", t)
        Company_list = Read_Excel(path)
        Output_name = "Output "+t
        Process_to_analysis = Company_list[0].Process_list[0]
        Process_to_buy_Water = Company_list[1].Process_list[0]
        # Solucion numerica
        Numerical_Solution(Process_1=Process_to_analysis,
                           Process_2=Process_to_buy_Water, Time_of=Time_of, file=Output_name)
        # Por la cant de veces a realizar el proceso
        dicc = {int: dict}
        x = 0
        for i in range(Time_of):
            # Respuesta del solver
            temp, Process_to_analysis, Process_to_buy_Water = Solution(
                Process_to_analysis, Process_to_buy_Water)

            # Agregar a diccionario
            dicc[i] = temp
            x += 1
            # Serializar en el excel
        Output(filename=Output_name, dicc=dicc, Process_1_Name=Process_to_analysis.Name,
               Process_2_Name=Process_to_buy_Water.Name, x=x)


def main():
    count = int(input("How many times do you want to run the program? "))
    Start(count)
    print("Done!")


# Iniciar programa
main()