Commit 843ea751 authored by Markus Holzer's avatar Markus Holzer Committed by Martin Bauer
Browse files

Dirichlet Boundary Condition for pystencils

parent 940915cf
from pystencils.boundaries.boundaryhandling import BoundaryHandling
from pystencils.boundaries.boundaryconditions import Neumann
from pystencils.boundaries.boundaryconditions import Neumann, Dirichlet
from pystencils.boundaries.inkernel import add_neumann_boundary
__all__ = ['BoundaryHandling', 'Neumann', 'add_neumann_boundary']
__all__ = ['BoundaryHandling', 'Neumann', 'Dirichlet', 'add_neumann_boundary']
from typing import List, Tuple, Any
from pystencils import Assignment
from pystencils.boundaries.boundaryhandling import BoundaryOffsetInfo
from typing import List, Tuple, Any
from pystencils.data_types import create_type
class Boundary:
......@@ -68,3 +69,27 @@ class Neumann(Boundary):
def __eq__(self, other):
return type(other) == Neumann
class Dirichlet(Boundary):
def __init__(self, value, name="Dirchlet"):
self._value = value
def additional_data(self):
if callable(self._value):
return [('value', create_type("double"))]
return []
def additional_data_init_callback(self):
if callable(self._value):
return self._value
def __call__(self, field, direction_symbol, index_field, **kwargs):
if self.additional_data:
return [Assignment(, index_field("value"))]
if field.index_dimensions == 0:
return [Assignment(, self._value)]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment