folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Examples update for v0.1 (#206)

Browse Source 
* modify examples/problems
* modify tutorials

---------

Co-authored-by: Dario Coscia <dariocoscia@dhcp-235.eduroam.sissa.it>
Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
This commit is contained in:
Dario Coscia
2023-11-14 18:24:07 +01:00
committed by Nicola Demo
parent 0d38de5afe
commit ee39b39805
19 changed files with 605 additions and 613 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
examples/problems/stokes.py
View File

@@ -1,9 +1,11 @@
import numpy as np
import torch
""" Navier Stokes Problem """
import torch
from pina.problem import SpatialProblem
from pina.operators import laplacian, grad, div
from pina import Condition, Span, LabelTensor
from pina import Condition, LabelTensor
from pina.geometry import CartesianDomain
from pina.equation import SystemEquation, Equation
# ===================================================== #
# #
@@ -21,7 +23,7 @@ class Stokes(SpatialProblem):
# assign output/ spatial variables
output_variables = ['ux', 'uy', 'p']
spatial_domain = Span({'x': [-2, 2], 'y': [-1, 1]})
spatial_domain = CartesianDomain({'x': [-2, 2], 'y': [-1, 1]})
# define the momentum equation
def momentum(input_, output_):
@@ -49,17 +51,9 @@ class Stokes(SpatialProblem):
# problem condition statement
conditions = {
'gamma_top': Condition(location=Span({'x': [-2, 2], 'y': 1}), function=wall),
'gamma_bot': Condition(location=Span({'x': [-2, 2], 'y': -1}), function=wall),
'gamma_out': Condition(location=Span({'x': 2, 'y': [-1, 1]}), function=outlet),
'gamma_in': Condition(location=Span({'x': -2, 'y': [-1, 1]}), function=inlet),
'D1': Condition(location=Span({'x': [-2, 2], 'y': [-1, 1]}), function=momentum),
'D2': Condition(location=Span({'x': [-2, 2], 'y': [-1, 1]}), function=continuity),
'gamma_top': Condition(location=CartesianDomain({'x': [-2, 2], 'y': 1}), equation=Equation(wall)),
'gamma_bot': Condition(location=CartesianDomain({'x': [-2, 2], 'y': -1}), equation=Equation(wall)),
'gamma_out': Condition(location=CartesianDomain({'x': 2, 'y': [-1, 1]}), equation=Equation(outlet)),
'gamma_in': Condition(location=CartesianDomain({'x': -2, 'y': [-1, 1]}), equation=Equation(inlet)),
'D': Condition(location=CartesianDomain({'x': [-2, 2], 'y': [-1, 1]}), equation=SystemEquation([momentum, continuity]))
}
# conditions = {
# 'gamma_top': Condition(location=Span({'x': [-2, 2], 'y': 1}), function=wall),
# 'gamma_bot': Condition(location=Span({'x': [-2, 2], 'y': -1}), function=wall),
# 'gamma_out': Condition(location=Span({'x': 2, 'y': [-1, 1]}), function=outlet),
# 'gamma_in': Condition(location=Span({'x': -2, 'y': [-1, 1]}), function=inlet),
# 'D': Condition(location=Span({'x': [-2, 2], 'y': [-1, 1]}), function=[momentum, continuity]),
# }