###############################################################################
# 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/"
#
###############################################################################
"""
This module contains a zero-order representation of a photothermal membrane unit.
"""

from pyomo.environ import Var, Reference, units as pyunits
from idaes.core import declare_process_block_class
from watertap.core import build_sido, ZeroOrderBaseData, pump_electricity

# Some more information about this module
__author__ = "<NAME>"


@declare_process_block_class("PhotothermalMembraneZO")
class PhotothermalMembraneData(ZeroOrderBaseData):
    """
    Zero-Order model for a photothermal membrane unit.
    """

    CONFIG = ZeroOrderBaseData.CONFIG()

    def build(self):
        super().build()

        self._tech_type = "photothermal_membrane"

        build_sido(self)

        # Create water flux variable
        self.water_flux = Var(
            units=pyunits.kg / (pyunits.m**2 * pyunits.hr),
            bounds=(0, None),
            doc="Water flux through membrane",
        )
        self._perf_var_dict["Water Flux"] = self.water_flux
        self._fixed_perf_vars.append(self.water_flux)

        # Create membrane area variable
        self.membrane_area = Var(
            units=pyunits.m**2, bounds=(0, None), doc="Membrane area"
        )
        self._perf_var_dict["Membrane Area"] = self.membrane_area

        @self.Constraint(self.flowsheet().time, doc="Constraint for water flux.")
        def wat_flux(b, t):
            return b.properties_in[t].flow_mass_comp["H2O"] == (
                pyunits.convert(
                    b.water_flux * b.membrane_area, to_units=pyunits.kg / pyunits.s
                )
            )

###############################################################################
# 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/"
#
###############################################################################
"""
This module contains a zero-order representation of a photothermal membrane unit.
"""

from pyomo.environ import Var, Reference, units as pyunits
from idaes.core import declare_process_block_class
from watertap.core import build_sido, ZeroOrderBaseData, pump_electricity

# Some more information about this module
__author__ = "<NAME>"


@declare_process_block_class("PhotothermalMembraneZO")
class PhotothermalMembraneData(ZeroOrderBaseData):
    """
    Zero-Order model for a photothermal membrane unit.
    """

    CONFIG = ZeroOrderBaseData.CONFIG()

    def build(self):
        super().build()

        self._tech_type = "photothermal_membrane"

        build_sido(self)

        # Create water flux variable
        self.water_flux = Var(
            units=pyunits.kg / (pyunits.m**2 * pyunits.hr),
            bounds=(0, None),
            doc="Water flux through membrane",
        )
        self._perf_var_dict["Water Flux"] = self.water_flux
        self._fixed_perf_vars.append(self.water_flux)

        # Create membrane area variable
        self.membrane_area = Var(
            units=pyunits.m**2, bounds=(0, None), doc="Membrane area"
        )
        self._perf_var_dict["Membrane Area"] = self.membrane_area

        @self.Constraint(self.flowsheet().time, doc="Constraint for water flux.")
        def wat_flux(b, t):
            return b.properties_in[t].flow_mass_comp["H2O"] == (
                pyunits.convert(
                    b.water_flux * b.membrane_area, to_units=pyunits.kg / pyunits.s
                )
            )

############################################################################### # 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/" # ############################################################################### This module contains a zero-order representation of a photothermal membrane unit. # Some more information about this module Zero-Order model for a photothermal membrane unit. # Create water flux variable # Create membrane area variable