Source code for simphony_mayavi.plugins.engine_wrappers.openfoam

from .abc_engine_factory import ABCEngineFactory


class OpenFoamFileIOEngineFactory(ABCEngineFactory):

    def create(self):
        from simphony.engine import openfoam_file_io
        try:
            return openfoam_file_io.Wrapper()
        except AttributeError:
            # Handle simphony-openfoam < 0.2.0
            return openfoam_file_io.FoamControlWrapper()


class OpenFoamInternalEngineFactory(ABCEngineFactory):

    def create(self):
        from simphony.engine import openfoam_internal
        try:
            return openfoam_internal.Wrapper()
        except AttributeError:
            # Handle simphony-openfoam < 0.2.0
            return openfoam_internal.FoamInternalWrapper()


ENGINE_REGISTRY = dict(openfoam_file_io=OpenFoamFileIOEngineFactory(),
                       openfoam_internal=OpenFoamInternalEngineFactory())