Source code for api.simulation.load_cases.load_setup

""" This is the base class for loads setup objects. Particular implementations can be found in the
 :ref:`loads setup implementations documentation <loads>` """

import typing
from abc import ABC, abstractmethod




class LoadSetup(ABC):
    def __init__(self, item):
        self._load = item

    def _get_wrapped_object(self):
        return self._load

    def __enter__(self):
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        pass





class DofSettable(ABC):
    @abstractmethod
    def set_dof_dict(self, dof_values_dict: typing.Dict[str, float]) -> LoadSetup:
        pass





class TranslationSettableMixin(DofSettable):
    @abstractmethod
    def set_u(self, value: float) -> LoadSetup:
        pass

    @abstractmethod
    def set_v(self, value: float) -> LoadSetup:
        pass

    @abstractmethod
    def set_w(self, value: float) -> LoadSetup:
        pass





class RotationSettableMixin(DofSettable):

    @abstractmethod
    def set_ru(self, value: float) -> LoadSetup:
        pass

    @abstractmethod
    def set_rv(self, value: float) -> LoadSetup:
        pass

    @abstractmethod
    def set_rw(self, value: float) -> LoadSetup:
        pass





class SourceTargetSettableMixin(DofSettable):
    @abstractmethod
    def set_target_value(self, value):
        pass

    @abstractmethod
    def set_source_value(self, value):
        pass