source: trunk/anuga_validation/convergence_study/tilted.py @ 8067

Last change on this file since 8067 was 8067, checked in by steve, 13 years ago

Pulling together basic validate tests for anuga

File size: 3.9 KB
Line 
1"""Simple water flow example using ANUGA
2
3Will Powers example of a simple sinusoidal wave which showed diffusive effects of
4thefirst order and standard second order method. Problem resolved if "rk2" timestepping
5and higher beta = 2 limiter used. Also new edge limiter with rk2 resolves problem
6"""
7
8#------------------------------------------------------------------------------
9# Import necessary modules
10#------------------------------------------------------------------------------
11
12import sys
13from anuga import rectangular_cross
14from anuga import Domain
15from anuga import Reflective_boundary
16from anuga import Dirichlet_boundary
17from anuga import Time_boundary
18from anuga import Transmissive_boundary
19from anuga import Transmissive_momentum_set_stage_boundary
20#from anuga.data_manager import start_screen_catcher, copy_code_files
21
22from math import cos
23import numpy as num
24from time import localtime, strftime, gmtime
25from os import sep
26
27
28
29#-------------------------------------------------------------------------------
30# Copy scripts to time stamped output directory and capture screen
31# output to file
32#-------------------------------------------------------------------------------
33model_name = 'tilted'
34
35time = strftime('%Y%m%d_%H%M%S',localtime())
36output_dir = model_name + '_'+time
37output_file = model_name
38#copy_code_files(output_dir,__file__)
39#start_screen_catcher(output_dir+sep)
40
41#------------------------------------------------------------------------------
42# Setup domain
43#------------------------------------------------------------------------------
44dx = 6.0*0.01
45dy = dx
46L = 600*0.01
47W = 10*dx
48
49# structured mesh
50points, vertices, boundary = rectangular_cross(int(L/dx), int(W/dy), L, W, (0.0, -W/2))
51
52domain = Domain(points, vertices, boundary) 
53
54domain.set_name(output_file)               
55domain.set_datadir(output_dir) 
56
57#------------------------------------------------------------------------------
58# Setup Algorithm
59#------------------------------------------------------------------------------
60domain.set_timestepping_method('rk2')
61domain.set_default_order(2)
62
63print domain.get_timestepping_method()
64
65#domain.use_edge_limiter = True
66#domain.tight_slope_limiters = True
67
68domain.CFL = 1.0
69
70domain.beta_w      = 0.6
71domain.beta_w_dry  = 0.0
72domain.beta_uh     = 0.6
73domain.beta_uh_dry = 0.0
74domain.beta_vh     = 0.6
75domain.beta_vh_dry = 0.0
76
77
78interactive_visualisation = True
79
80
81#------------------------------------------------------------------------------
82# Setup initial conditions
83#------------------------------------------------------------------------------
84domain.set_quantity('elevation',0.0)
85domain.set_quantity('friction', 0.00)
86domain.set_quantity('stage', lambda x,y : 14.0*0.01 + x/(600.0*0.01)*14.0*0.01 )           
87
88#-----------------------------------------------------------------------------
89# Setup boundary conditions
90#------------------------------------------------------------------------------
91from math import sin, pi, exp
92Br = Reflective_boundary(domain)      # Solid reflective wall
93
94# Associate boundary tags with boundary objects
95domain.set_boundary({'left': Br, 'right': Br, 'top': Br, 'bottom': Br})
96
97
98#===============================================================================
99if interactive_visualisation:
100    from anuga.visualiser import RealtimeVisualiser
101    vis = RealtimeVisualiser(domain)
102    vis.render_quantity_height("stage", zScale =1, dynamic=True)
103    vis.colour_height_quantity('stage', (1.0, 0.5, 0.5))
104    vis.start()
105#===============================================================================
106
107
108#------------------------------------------------------------------------------
109# Evolve system through time
110#------------------------------------------------------------------------------
111
112for t in domain.evolve(yieldstep = 0.5, finaltime = 60.0):
113    domain.write_time()
114    if interactive_visualisation:
115        vis.update()
116
117if interactive_visualisation:
118    vis.evolveFinished()
119
Note: See TracBrowser for help on using the repository browser.