aspen_pysys.model.factory

19class HysysModFactory(ABC):
20    """Interface for HYSYS model factory methods."""
21
22    @staticmethod
23    @abstractmethod
24    def get_process_stream(obj: HysysNamedObject) -> HysysProcessStream:
25        """Get a process stream.
26
27        Args:
28            obj (HysysNamedObject): Named HYSYS object
29
30        Returns:
31            HysysProcessStream: Process stream
32        """
33
34    @staticmethod
35    @abstractmethod
36    def get_material_stream(
37        obj: HysysNamedObject,
38    ) -> HysysMaterialStream:
39        """Get a material stream.
40
41        Args:
42            obj (HysysNamedObject): Named HYSYS object
43
44        Returns:
45            HysysMaterialStream: Material stream
46        """
47
48    @staticmethod
49    @abstractmethod
50    def get_energy_stream(obj: HysysNamedObject) -> HysysEnergyStream:
51        """Get an energy stream.
52
53        Args:
54            obj (HysysNamedObject): Named HYSYS object
55
56        Returns:
57            HysysEnergyStream: Energy stream
58        """
59
60    @staticmethod
61    @abstractmethod
62    def get_unit_operation(obj: HysysNamedObject) -> HysysUnitOperation:
63        """Get a unit operation.
64
65        Args:
66            obj (HysysNamedObject): Named HYSYS object
67
68        Returns:
69            HysysUnitOperation: Unit operation
70        """

20class HysysModManager(HysysModFactory):
21    """Helper class that constructs the right type of HysysObject."""
22
23    @override
24    @staticmethod
25    def get_process_stream(obj: HysysNamedObject) -> HysysProcessStream:
26        return HysysProcessStream.from_obj(obj)
27
28    @override
29    @staticmethod
30    def get_material_stream(obj: HysysNamedObject) -> HysysMaterialStream:
31        return HysysMaterialStream.from_obj(obj)
32
33    @override
34    @staticmethod
35    def get_energy_stream(obj: HysysNamedObject) -> HysysEnergyStream:
36        return HysysEnergyStream.from_obj(obj)
37
38    @override
39    @staticmethod
40    def get_unit_operation(obj: HysysNamedObject) -> HysysUnitOperation:
41        return HysysUnitOperation.from_obj(obj)