added selections and contact

Author: Teseo Schneider

polyfempy/Problem.py

@@ -1,33 +1,35 @@
 import json
 import types
 
+
 class Problem:
     """Generic problem problem, scalar or tensor depending on the pde. Warning, this problem needs to be used with the `set_pde` function in settings"""
 
-    def __init__(self, rhs=None, exact=None):
+    def __init__(self, rhs=None, exact=None, is_time_dependent=False):
         self.rhs = rhs
+        self.is_time_dependent = is_time_dependent
         self.exact = exact
         self.dirichlet_boundary = []
         self.neumann_boundary = []
         self.pressure_boundary = []
 
-        self.dirichlet_boundary_lambda = []
-        self.neumann_boundary_lambda = []
-        self.pressure_boundary_lambda = []
+        # self.dirichlet_boundary_lambda = []
+        # self.neumann_boundary_lambda = []
+        # self.pressure_boundary_lambda = []
 
-    def set_dirichlet_value(self, id, value, is_dirichlet_dim=None):
+    def set_dirichlet_value(self, id, value, is_dirichlet_dim=None, linear_ramp_to=None):
         """set the Dirichlet value value for the sideset id. Note the value must be a scalar, vector in 2D, or 3D depending on the problem. is_dirichlet_dim is a vector of boolean specifying which dimentions are fixed, only for vector-based problems."""
-        self.add_dirichlet_value(id, value, is_dirichlet_dim)
+        self.add_dirichlet_value(id, value, is_dirichlet_dim, linear_ramp_to)
 
-    def set_neumann_value(self, id, value):
+    def set_neumann_value(self, id, value, linear_ramp_to=None):
         """set the Neumann value value for the sideset id. Note the value must be a scalar, vector in 2D, or 3D depending on the problem"""
-        self.add_neumann_value(id, value)
+        self.add_neumann_value(id, value, linear_ramp_to)
 
-    def set_pressure_value(self, id, value):
+    def set_pressure_value(self, id, value, linear_ramp_to=None):
         """set the Pressure value value for the sideset id. Note the value must be a scalar"""
-        self.add_pressure_value(id, value)
+        self.add_pressure_value(id, value, linear_ramp_to)
 
-    def add_dirichlet_value(self, id, value, is_dirichlet_dim=None):
+    def add_dirichlet_value(self, id, value, is_dirichlet_dim=None, linear_ramp_to=None):
         """set the Dirichlet value value for the sideset id. Note the value must be a scalar, vector in 2D, or 3D depending on the problem. is_dirichlet_dim is a vector of boolean specifying which dimentions are fixed, only for vector-based problems."""
 
         tmp = {}
@@ -38,50 +40,58 @@ def add_dirichlet_value(self, id, value, is_dirichlet_dim=None):
             assert(len(value) == len(is_dirichlet_dim))
             tmp["dimension"] = is_dirichlet_dim
 
+        if linear_ramp_to is not None:
+            tmp["linear_ramp"] = {"to": linear_ramp_to}
+
         if isinstance(value, types.LambdaType) or isinstance(value, types.FunctionType):
             pass
             # self.dirichlet_boundary_lambda.append(tmp)
         else:
             self.dirichlet_boundary.append(tmp)
 
-    def add_neumann_value(self, id, value):
+    def add_neumann_value(self, id, value, linear_ramp_to=None):
         """set the Neumann value value for the sideset id. Note the value must be a scalar, vector in 2D, or 3D depending on the problem"""
 
         tmp = {}
         tmp["id"] = id
         tmp["value"] = value
 
+        if linear_ramp_to is not None:
+            tmp["linear_ramp"] = {"to": linear_ramp_to}
+
         if isinstance(value, types.LambdaType) or isinstance(value, types.FunctionType):
             pass
             # self.neumann_boundary_lambda.append(tmp)
         else:
             self.neumann_boundary.append(tmp)
 
-    def add_pressure_value(self, id, value):
+    def add_pressure_value(self, id, value, linear_ramp_to=None):
         """set the Pressure value value for the sideset id. Note the value must be a scalar"""
 
         tmp = {}
         tmp["id"] = id
         tmp["value"] = value
 
+        if linear_ramp_to is not None:
+            tmp["linear_ramp"] = {"to": linear_ramp_to}
+
         if isinstance(value, types.LambdaType) or isinstance(value, types.FunctionType):
             pass
             # self.pressure_boundary_lambda.append(tmp)
         else:
             self.pressure_boundary.append(tmp)
 
-
-    def set_velocity(self, id, value, is_dim_fixed=None):
+    def set_velocity(self, id, value, is_dim_fixed=None, linear_ramp_to=None):
         """set the velocity value for the sideset id. Note the value must be a vector in 2D or 3D depending on the problem"""
-        self.add_dirichlet_value(id, value, is_dim_fixed)
+        self.add_dirichlet_value(id, value, is_dim_fixed, linear_ramp_to)
 
-    def set_displacement(self, id, value, is_dim_fixed=None):
+    def set_displacement(self, id, value, is_dim_fixed=None, linear_ramp_to=None):
         """set the displacement value for the sideset id. Note the value must be a vector in 2D or 3D depending on the problem"""
-        self.add_dirichlet_value(id, value, is_dim_fixed)
+        self.add_dirichlet_value(id, value, is_dim_fixed, linear_ramp_to)
 
-    def set_force(self, id, value):
+    def set_force(self, id, value, linear_ramp_to=None):
         """set the force value for the sideset id. Note the value must be a vector in 2D or 3D depending on the problem"""
-        self.add_neumann_value(id, value)
+        self.add_neumann_value(id, value, linear_ramp_to)
 
     def set_x_symmetric(self, id):
         """x coorinate is fixed, y is allowed to move (Neumann)"""
@@ -108,8 +118,8 @@ def params(self):
         tmp = dict(
             (key, value)
             for (key, value) in self.__dict__.items())
-        tmp.pop('dirichlet_boundary_lambda', None)
-        tmp.pop('neumann_boundary_lambda', None)
-        tmp.pop('pressure_boundary_lambda', None)
+        # tmp.pop('dirichlet_boundary_lambda', None)
+        # tmp.pop('neumann_boundary_lambda', None)
+        # tmp.pop('pressure_boundary_lambda', None)
 
         return tmp

polyfempy/Selection.py

@@ -0,0 +1,53 @@
+import json
+
+
+class Selection:
+    """Object used to select sideset and bodies"""
+
+    def __init__(self):
+        self.body_ids = []
+        self.boundary_sidesets = []
+
+    def select_body_with_sphere(self, id, center, radius):
+        """Select a body using a sphere"""
+        self.body_ids.append({"id": id, "center": center, "radius": radius})
+
+    def select_body_with_box(self, id, box_min, box_max):
+        """Select a body using an axis-aligned box"""
+        self.body_ids.append({"id": id, "box": [box_min, box_max]})
+
+    def select_body_with_axis_plane(self, id, axis, position):
+        """Select a body using an axis-aligned plane at position position, axis 1, 2, 3 is x, y, and z respectively. Use negative to flip axis (e.g., -1 is negative x axis)"""
+        self.body_ids.append({"id": id, "position": position, "axis": axis})
+
+    def select_body_with_plane(self, id, normal, offset):
+        """Select a body using a generic plane with normal normal, the point on the plane is defined by normal*offset"""
+        self.body_ids.append({"id": id, "normal": normal, "offset": offset})
+
+    def select_sideset_with_sphere(self, id, center, radius):
+        """Select a boundary sideset using a sphere"""
+        self.boundary_sidesets.append(
+            {"id": id, "center": center, "radius": radius})
+
+    def select_sideset_with_box(self, id, box_min, box_max):
+        """Select a boundary sideset using an axis-aligned box"""
+        self.boundary_sidesets.append({"id": id, "box": [box_min, box_max]})
+
+    def select_sideset_with_axis_plane(self, id, axis, position):
+        """Select a boundary sideset using an axis-aligned plane at position position, axis 1, 2, 3 is x, y, and z respectively. Use negative to flip axis (e.g., -1 is negative x axis)"""
+        self.boundary_sidesets.append(
+            {"id": id, "position": position, "axis": axis})
+
+    def select_sideset_with_plane(self, id, normal, offset):
+        """Select a boundary sideset using a generic plane with normal normal, the point on the plane is defined by normal*offset"""
+        self.boundary_sidesets.append(
+            {"id": id, "normal": normal, "offset": offset})
+
+    def __str__(self):
+        """stringygied json description of this class, used to run the solver"""
+
+        tmp = dict(
+            (key, value)
+            for (key, value) in self.__dict__.items())
+
+        return json.dumps(tmp, sort_keys=True, indent=4)