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

Fix bugs in 0.2 (#344)

Browse Source 
* Fix some bugs
This commit is contained in:
FilippoOlivo
2024-09-12 18:12:59 +02:00
committed by Nicola Demo
parent f0d68b34c7
commit 30f865d912
11 changed files with 112 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
examples/problems/stokes.py
View File

@@ -49,11 +49,19 @@ class Stokes(SpatialProblem):
value = 0.0
return output_.extract(['ux', 'uy']) - value
domains = {
'gamma_top': CartesianDomain({'x': [-2, 2], 'y': 1}),
'gamma_bot': CartesianDomain({'x': [-2, 2], 'y': -1}),
'gamma_out': CartesianDomain({'x': 2, 'y': [-1, 1]}),
'gamma_in': CartesianDomain({'x': -2, 'y': [-1, 1]}),
'D': CartesianDomain({'x': [-2, 2], 'y': [-1, 1]})
}
# problem condition statement
conditions = {
'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]))
'gamma_top': Condition(domain='gamma_top', equation=Equation(wall)),
'gamma_bot': Condition(domain='gamma_bot', equation=Equation(wall)),
'gamma_out': Condition(domain='gamma_out', equation=Equation(outlet)),
'gamma_in': Condition(domain='gamma_in', equation=Equation(inlet)),
'D': Condition(domain='D', equation=SystemEquation([momentum, continuity]))
}

4
examples/run_stokes.py
View File

@@ -17,8 +17,8 @@ if __name__ == "__main__":
# create problem and discretise domain
stokes_problem = Stokes()
stokes_problem.discretise_domain(n=1000, locations=['gamma_top', 'gamma_bot', 'gamma_in', 'gamma_out'])
stokes_problem.discretise_domain(n=2000, locations=['D'])
stokes_problem.discretise_domain(n=1000, domains=['gamma_top', 'gamma_bot', 'gamma_in', 'gamma_out'])
stokes_problem.discretise_domain(n=2000, domains=['D'])
# make the model
model = FeedForward(