A small package for fem
Project description
SIMple Finite Element Methods in PYthon
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
import pathlib, sys
simfempypath = str(pathlib.Path(__file__).parent.parent.parent)
sys.path.insert(0,simfempypath)
from simfempy.models.elliptic import Elliptic
from simfempy.applications.application import Application
#--------------------------------------------------------------
class HeatExample(Application):
# nomen est omen
def defineProblemData(self, problemdata):
# boundary conditions
problemdata.bdrycond.set(type="Dirichlet", colors=[1000, 3000])
problemdata.bdrycond.set(type="Neumann", colors=[1001, 1002, 1003])
problemdata.bdrycond.fct[1000] = lambda x, y, z: 200
problemdata.bdrycond.fct[3000] = lambda x, y, z: 320
# postprocess
problemdata.postproc.set(name='bdrymean_right', type='bdry_mean', colors=1001)
problemdata.postproc.set(name='bdrymean_left', type='bdry_mean', colors=1003)
problemdata.postproc.set(name='bdrymean_up', type='bdry_mean', colors=1002)
problemdata.postproc.set(name='bdrynflux', type='bdry_nflux', colors=[3000])
# paramaters in equation
problemdata.params.set_scal_cells("kheat", [100], 0.001)
problemdata.params.set_scal_cells("kheat", [200], 10.0)
# nomen est omen
def defineGeometry(self, geom, h):
h=0.2
holes = []
rectangle = geom.add_rectangle(xmin=-1.5, xmax=-0.5, ymin=-1.5, ymax=-0.5, z=0, mesh_size=h)
geom.add_physical(rectangle.surface, label="200")
geom.add_physical(rectangle.lines, label="20") # required for correct boundary labels (!?)
holes.append(rectangle)
circle = geom.add_circle(x0=[0, 0], radius=0.5, mesh_size=h, num_sections=6, make_surface=False)
geom.add_physical(circle.curve_loop.curves, label="3000")
holes.append(circle)
p = geom.add_rectangle(xmin=-2, xmax=2, ymin=-2, ymax=2, z=0, mesh_size=h, holes=holes)
geom.add_physical(p.surface, label="100")
for i in range(len(p.lines)): geom.add_physical(p.lines[i], label=f"{1000 + i}")
#--------------------------------------------------------------
heat = Elliptic(application=HeatExample(), fem='cr1')
result, u = heat.static()
print(f"{result=}")
fig = plt.figure(figsize=(10, 8))
fig.suptitle(f"{heat.application.__class__.__name__} (static)", fontsize=16)
outer = gridspec.GridSpec(1, 2, wspace=0.2, hspace=0.2)
heat.mesh.plot(fig=fig, outer=outer[0], bdry=True)
data = u.tovisudata()
data.update({'cell': {'k': heat.kheatcell}})
heat.mesh.plot(data=data, alpha=0.5, fig=fig, outer=outer[1])
plt.show()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
simfempy-2.2.9.tar.gz
(81.0 kB
view hashes)