###############################################################################
# WaterTAP Copyright (c) 2021, The Regents of the University of California,
# through Lawrence Berkeley National Laboratory, Oak Ridge National
# Laboratory, National Renewable Energy Laboratory, and National Energy
# Technology Laboratory (subject to receipt of any required approvals from
# the U.S. Dept. of Energy). All rights reserved.
#
# Please see the files COPYRIGHT.md and LICENSE.md for full copyright and license
# information, respectively. These files are also available online at the URL
# "https://github.com/watertap-org/watertap/"
#
###############################################################################
"""Simple zero order separator examples"""
from pyomo.environ import ConcreteModel, Constraint
from idaes.core import FlowsheetBlock
# from idaes.models.unit_models import Separator # replaced separator
from idaes.models.unit_models.separator import (
SplittingType,
EnergySplittingType,
)
from idaes.core.util.scaling import (
calculate_scaling_factors,
set_scaling_factor,
constraint_scaling_transform,
)
from watertap.examples.flowsheets.full_treatment_train.model_components import (
property_models,
)
from watertap.examples.flowsheets.full_treatment_train.util import (
solve_block,
check_dof,
)
from idaes.models.unit_models import Separator
[docs]def build_SepRO(m, base="TDS"):
"""
Builds RO model based on the IDAES separator.
Requires prop_TDS property package.
"""
prop = property_models.get_prop(m, base=base)
m.fs.RO = Separator(
property_package=prop,
outlet_list=["retentate", "permeate"],
split_basis=SplittingType.componentFlow,
energy_split_basis=EnergySplittingType.equal_temperature,
)
# specify
if base == "TDS":
m.fs.RO.split_fraction[0, "permeate", "H2O"].fix(0.5)
m.fs.RO.split_fraction[0, "permeate", "TDS"].fix(0.01)
else:
raise ValueError(
"Unexpected property base {base} provided to build_SepRO"
"".format(base=base)
)
# scale
set_scaling_factor(
m.fs.RO.split_fraction, 1
) # TODO: IDAES should set these scaling factors by default
constraint_scaling_transform(m.fs.RO.sum_split_frac[0.0, "H2O"], 1)
constraint_scaling_transform(m.fs.RO.sum_split_frac[0.0, "TDS"], 1)
[docs]def build_SepNF(m, base="ion"):
"""
Builds NF model based on the IDAES separator for a specified property base.
Requires prop_ion or prop_salt property package.
"""
prop = property_models.get_prop(m, base=base)
m.fs.NF = Separator(
property_package=prop,
outlet_list=["retentate", "permeate"],
split_basis=SplittingType.componentFlow,
energy_split_basis=EnergySplittingType.equal_temperature,
)
# specify
if base == "ion":
m.fs.NF.split_fraction[0, "permeate", "H2O"].fix(0.9)
m.fs.NF.split_fraction[0, "permeate", "Na"].fix(0.9)
m.fs.NF.split_fraction[0, "permeate", "Ca"].fix(0.1)
m.fs.NF.split_fraction[0, "permeate", "Mg"].fix(0.1)
m.fs.NF.split_fraction[0, "permeate", "SO4"].fix(0.1)
# Cl split fraction determined through electro-neutrality for the retentate
charge_dict = {"Na": 1, "Ca": 2, "Mg": 2, "SO4": -2, "Cl": -1}
m.fs.NF.EN_out = Constraint(
expr=0
== sum(
charge_dict[j]
* m.fs.NF.retentate_state[0].flow_mol_phase_comp["Liq", j]
for j in charge_dict
)
)
constraint_scaling_transform(m.fs.NF.EN_out, 1)
elif base == "salt":
m.fs.NF.split_fraction[0, "permeate", "H2O"].fix(0.9)
m.fs.NF.split_fraction[0, "permeate", "NaCl"].fix(0.9)
m.fs.NF.split_fraction[0, "permeate", "CaSO4"].fix(0.1)
m.fs.NF.split_fraction[0, "permeate", "MgSO4"].fix(0.1)
m.fs.NF.split_fraction[0, "permeate", "MgCl2"].fix(0.2)
# scale
set_scaling_factor(
m.fs.NF.split_fraction, 1
) # TODO: IDAES should set these scaling factors by default
if base == "ion":
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "H2O"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "Na"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "Ca"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "Mg"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "SO4"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "Cl"], 1)
elif base == "salt":
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "H2O"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "NaCl"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "CaSO4"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "MgSO4"], 1)
constraint_scaling_transform(m.fs.NF.sum_split_frac[0.0, "MgCl2"], 1)
def solve_SepRO(base="TDS"):
m = ConcreteModel()
m.fs = FlowsheetBlock(dynamic=False)
property_models.build_prop(m, base=base)
build_SepRO(m, base=base)
property_models.specify_feed(m.fs.RO.mixed_state[0], base=base)
check_dof(m)
calculate_scaling_factors(m)
solve_block(m)
m.fs.RO.inlet.display()
m.fs.RO.permeate.display()
m.fs.RO.retentate.display()
return m
def solve_SepNF(base="ion"):
m = ConcreteModel()
m.fs = FlowsheetBlock(dynamic=False)
property_models.build_prop(m, base=base)
build_SepNF(m, base=base)
property_models.specify_feed(m.fs.NF.mixed_state[0], base=base)
m.fs.NF.mixed_state[0].mass_frac_phase_comp # touching for tests
check_dof(m)
calculate_scaling_factors(m)
solve_block(m)
m.fs.NF.inlet.display()
m.fs.NF.permeate.display()
m.fs.NF.retentate.display()
return m
if __name__ == "__main__":
solve_SepRO(base="TDS")
solve_SepNF(base="ion")
solve_SepNF(base="salt")