# Question4

P1 = 1.05E5 # [Pa]
T1 = 20 + 273.15 # [K]
P2 = 2E5 # [Pa]
T2 = 100 + 273.15 # [K]
Vdot = 0.3 # [m^3/s]
R = 0.287E3 # [J/kgK]

phi_1 = 0.85

Pv1 = phi_1 * 2.339E3
print("Pv1 = %0.4f [kPa]" % (Pv1/1E3))

Pa1 = P1 - Pv1
print("Pa1 = %0.4f [kPa]" % (Pa1/1E3))

Pa2 = Pa1 * (T2/T1)
print("Pa2 = %0.4f [kPa]" % (Pa2/1E3))

Pv2 = P2 - Pa2
print("Pv2 = %0.4f [kPa]" % (Pv2/1E3))

Pg2 = 101.3E3 # [Pa]
phi_2 = Pv2/Pg2
print("phi_2 = %0.4f [ ]" % (phi_2))

# Compute compressor work

# Start by finding omega1, omega2
omega1 = (0.622*Pv1)/(P1 - Pv1)
omega2 = (0.622*Pv2)/(P2 - Pv2)
print("\nomega1 = %0.4f [ ]" % omega1)
print("omega2 = %0.4f [ ]" % omega2)

# Interpolate for hg1
xVals = np.array([250, 300])
yVals = np.array([461.36, 554.32])
hg1 = np.interp(293.15, xVals, yVals)
print("hg1 = %0.4f [kJ/kg]" % hg1)

# Interpolate for hg2
xVals = np.array([350, 400])
yVals = np.array([647.90, 742.40])
hg2 = np.interp(373.15, xVals, yVals)
print("hg2 = %0.4f [kJ/kg]" % hg2)

# Interpolate for ha1
xVals = np.array([290, 298.15])
yVals = np.array([290.43, 298.62])
ha1 = np.interp(293.15, xVals, yVals)
print("ha1 = %0.4f [kJ/kg]" % ha1)

# Interpolate for ha2
xVals = np.array([360, 380])
yVals = np.array([360.86, 381.06])
ha2 = np.interp(373.15, xVals, yVals)
print("ha2 = %0.4f [kJ/kg]" % ha2)


# Compute h of mixture
h1_mix = ha1 + omega1 * hg1
h2_mix = ha2 + omega2 * hg2
print("h1_mix = %0.4f [kJ/kg_da]" % h1_mix)
print("h2_mix = %0.4f [kJ/kg_da]" % h2_mix)

Wdot = h2_mix - h1_mix
print("Wdot = %0.4f [kJ/kg_da]" % Wdot)

# Compute mass flow to get power
mdot = (P1 * Vdot) / (R * T1)
print("mdot = %0.4f [kg/s]" % mdot)
Power = Wdot * mdot
print("Power = %0.4f [kW]" % Power)