import matplotlib.pyplot as plt
import numpy as np

# Datos
parametros = ['Longitud', 'Ancho', 'Diámetro', 'Flujo másico']

# Potencias eléctricas
pot_inicial_elec = [1.212, 1.868, 1.74, 1.542]
pot_final_elec = [2.229, 1.723, 1.522, 1.728]
diferencias_elec = [abs(final - inicial) for inicial, final in zip(pot_inicial_elec, pot_final_elec)]
diferencias_porcentuales_elec = [(dif / inicial) * 100 for dif, inicial in zip(diferencias_elec, pot_inicial_elec)]

# Potencias térmicas
pot_inicial_term = [3.708, 4.859, 5.239, 5.4]
pot_final_term = [6.909, 5.307, 7.109, 5.305]
diferencias_term = [abs(final - inicial) for inicial, final in zip(pot_inicial_term, pot_final_term)]
diferencias_porcentuales_term = [(dif / inicial) * 100 for dif, inicial in zip(diferencias_term, pot_inicial_term)]

# Gráfico de barras para comparar potencias iniciales y finales
x = np.arange(len(parametros))  # Posiciones de las barras
width = 0.35  # Ancho de las barras

fig, ax = plt.subplots(2, 2, figsize=(18, 12))

# Tamaños de texto
titulo_size = 22
label_size = 20
tick_size = 18
legend_size = 18

# Gráfico 1: Potencias eléctricas iniciales y finales
ax[0, 0].bar(x - width/2, pot_inicial_elec, width, label='Potencia Inicial', color='blue', alpha=0.7)
ax[0, 0].bar(x + width/2, pot_final_elec, width, label='Potencia Final', color='orange', alpha=0.7)
ax[0, 0].set_xticks(x)
ax[0, 0].set_xticklabels(parametros, fontsize=tick_size)
ax[0, 0].set_ylabel('Potencia Eléctrica (kW)', fontsize=label_size)
ax[0, 0].set_title('Comparación de Potencias Eléctricas', fontsize=titulo_size)
ax[0, 0].legend(fontsize=legend_size, loc='upper left', frameon=True)
ax[0, 0].grid(axis='y', linestyle='--', alpha=0.7)

# Gráfico 2: Diferencias porcentuales eléctricas
ax[0, 1].bar(parametros, diferencias_porcentuales_elec, color='green', alpha=0.7)
ax[0, 1].set_ylabel('Diferencia (%)', fontsize=label_size)
ax[0, 1].set_title('Diferencias Porcentuales en Potencias Eléctricas', fontsize=titulo_size)
ax[0, 1].tick_params(axis='x', labelsize=tick_size)
ax[0, 1].grid(axis='y', linestyle='--', alpha=0.7)

# Gráfico 3: Potencias térmicas iniciales y finales
ax[1, 0].bar(x - width/2, pot_inicial_term, width, label='Potencia Inicial', color='blue', alpha=0.7)
ax[1, 0].bar(x + width/2, pot_final_term, width, label='Potencia Final', color='orange', alpha=0.7)
ax[1, 0].set_xticks(x)
ax[1, 0].set_xticklabels(parametros, fontsize=tick_size)
ax[1, 0].set_ylabel('Potencia Térmica (kW)', fontsize=label_size)
ax[1, 0].set_title('Comparación de Potencias Térmicas', fontsize=titulo_size)
ax[1, 0].legend(fontsize=legend_size, loc='upper left', frameon=True)
ax[1, 0].grid(axis='y', linestyle='--', alpha=0.7)

# Gráfico 4: Diferencias porcentuales térmicas
ax[1, 1].bar(parametros, diferencias_porcentuales_term, color='green', alpha=0.7)
ax[1, 1].set_ylabel('Diferencia (%)', fontsize=label_size)
ax[1, 1].set_title('Diferencias Porcentuales en Potencias Térmicas', fontsize=titulo_size)
ax[1, 1].tick_params(axis='x', labelsize=tick_size)
ax[1, 1].grid(axis='y', linestyle='--', alpha=0.7)

# Ajustar márgenes
plt.subplots_adjust(hspace=0.5, wspace=0.3)
plt.tight_layout()
plt.show()