Source code for bodies

from base_types import *



class base_state:
    """mechanical state of a body"""
    position: Vector3 = Vector3(0,0,0)
    """**[mandatory]** position [:math:`L`]"""
    velocity: Vector3 = Vector3(0,0,0)
    """**[mandatory]** translational velocity [:math:`L T^{-1}`]"""
    orientation: Quaternion = Quaternion(0,0,0,1)
    """**[mandatory]** orientation [:math:`-`]"""
    angular_velocity: Vector3 = Vector3(0,0,0)
    """**[mandatory]** angular velocity [:math:`T^{-1}`]"""
    mass: float = None
    """**[mandatory]** body's mass [:math:`M`]"""
    inertia: Matrix3 = Matrix3.zero()
    """**[mandatory]** tensor of inertia [:math:`M L^{2}`]"""
    volume: float = None
    """**[mandatory]** volume [:math:`L^{3}`]"""




class thermal(base_state):
    """mechanical + thermal state"""
    temperature: float = None
    """*[optional]* temperature [:math:`\Theta`]"""




class liquid_film(base_state):
    """mechanical + liquid film state"""
    liquid_film_volume: float = None
    """*[optional]* liquid film volume [L^3]"""




class base_shape:
    """shape of a body"""
    color: Vector3 = Vector3(1,1,1)
    """*[optional]* RGB color values [:math:`-`]"""




class sphere(base_shape):
    """sphere"""
    radius: float = None
    """**[mandatory]** radius"""




class box(base_shape):
    """cuboid"""
    dimensions: Vector3 = Vector3(0,0,0)
    """**[mandatory]** length in each direction of space"""




class polyhedron(base_shape):
    """A generic polyhedron"""
    vertices: list = []
    """**[mandatory]** list of positions of the vertices"""





class base_body:
    """class defining a body"""
    material_id: int = -1
    """**[mandatory]** material id [:math:`-`]"""
    clump_id: int = -1
    """**[mandatory]** clump id [:math:`-`]"""
    body_id: int = None
    """**[mandatory]** body id [:math:`-`]"""
    body_state: type(base_state) = base_state()
    """**[mandatory]** body state [:math:`-`]"""
    body_shape: type(base_shape) = base_shape()
    """**[mandatory]** body shape [:math:`-`]"""



#class body(SomeStateClass,SomeShapeClass,SomeMaterialClass):
#    """a DEM body"""
#    state: type(SomeStateClass) = SomeStateClass()
#    """The state of the body, instance of a concrete state class"""
#    shape: type(SomeShapeClass) = SomeShapeClass()
#    """The shape of the body, instance of a concrete shape class"""
#    material: type(SomeMaterialClass) = SomeMaterialClass()
#    """The material of the body, instance of a concrete material class"""