def run_strat_sim(set_ICs, name, params):
snapshots_dir = SNAPSHOTS_DIR % name
logger = logging.getLogger(name)
solver, domain = get_solver(params)
# Initial conditions
set_ICs(solver, domain, params)
# filename = '{s}/{s}_s1.h5'.format(s=snapshots_dir)
# write, dt = solver.load_state(filename, -30)
# print("=====strat_helper.py=====", "loaded", write)
cfl = CFL(solver,
initial_dt=params['DT'],
cadence=10,
max_dt=5,
min_dt=0.01,
safety=0.5,
threshold=0.10)
cfl.add_velocities(('ux', 'uz'))
cfl.add_frequency(params['DT'])
snapshots = solver.evaluator.add_file_handler(snapshots_dir,
sim_dt=params['T_F'] /
params['NUM_SNAPSHOTS'])
snapshots.add_system(solver.state)
# Flow properties
flow = GlobalFlowProperty(solver, cadence=10)
flow.add_property("sqrt(ux*ux + uz*uz) / NU", name='Re')
# Main loop
logger.info('Starting sim...')
slices = domain.dist.coeff_layout.slices(scales=1)
(len_x, len_z) = domain.dist.coeff_layout.global_shape(scales=1)
# mask kicks in 1/3 of the way, gaussian decay
len_x_damp = len_x - (len_x // 3)
len_z_damp = len_z - (len_z // 3)
x_mask = np.concatenate(
(np.ones(len_x // 3),
np.exp(-16 * np.arange(len_x_damp)**2 / len_x_damp**2)))
z_mask = np.concatenate(
(np.ones(len_z // 3),
np.exp(-16 * np.arange(len_z_damp)**2 / len_z_damp**2)))
mask = np.outer(x_mask, z_mask)
while solver.ok:
cfl_dt = cfl.compute_dt() if params.get('USE_CFL') else params['DT']
solver.step(cfl_dt)
curr_iter = solver.iteration
for field in solver.state.fields:
field['c'] *= mask[slices]
if curr_iter % int(
(params['T_F'] / params['DT']) / params['NUM_SNAPSHOTS']) == 0:
logger.info('Reached time %f out of %f, timestep %f vs max %f',
solver.sim_time, solver.stop_sim_time, cfl_dt,
params['DT'])
logger.info('Max Re = %f' % flow.max('Re'))
def run_strat_sim(set_ICs, name, params):
snapshots_dir = SNAPSHOTS_DIR % name
filename = '{s}/{s}_s1.h5'.format(s=snapshots_dir)
logger = logging.getLogger(name)
solver, domain = get_solver(params)
# Initial conditions
dt = params['DT']
_, dt = set_ICs(name, solver, domain, params)
cfl = CFL(solver,
initial_dt=dt,
cadence=10,
max_dt=params['DT'],
min_dt=0.01,
safety=0.5,
threshold=0.10)
cfl.add_velocities(('ux', 'uz'))
cfl.add_frequency(params['DT'])
snapshots = solver.evaluator.add_file_handler(snapshots_dir,
sim_dt=params['T_F'] /
params['NUM_SNAPSHOTS'])
snapshots.add_system(solver.state)
# Flow properties
flow = GlobalFlowProperty(solver, cadence=10)
flow.add_property('sqrt((ux / NU)**2 + (uz / NU)**2)', name='Re')
# Main loop
logger.info('Starting sim...')
while solver.ok:
cfl_dt = cfl.compute_dt() if params.get('USE_CFL') else params['DT']
# if cfl_dt < params['DT'] / 8: # small step sizes if strongly cfl limited
# cfl_dt /= 2
solver.step(cfl_dt)
curr_iter = solver.iteration
if curr_iter % int(
(params['T_F'] / params['DT']) / params['NUM_SNAPSHOTS']) == 0:
logger.info('Reached time %f out of %f, timestep %f vs max %f',
solver.sim_time, solver.stop_sim_time, cfl_dt,
params['DT'])
logger.info('Max Re = %f' % flow.max('Re'))
def run_strat_sim(name, params):
populate_globals(params)
snapshots_dir = SNAPSHOTS_DIR % name
solver, domain = get_solver(params)
# Initial conditions
_, dt = set_ic(name, solver, domain, params)
cfl = CFL(solver,
initial_dt=dt,
cadence=5,
max_dt=DT,
min_dt=0.007,
safety=0.5,
threshold=0.10)
cfl.add_velocities(('ux', 'uz'))
cfl.add_frequency(DT)
snapshots = solver.evaluator.add_file_handler(snapshots_dir,
sim_dt=T_F / NUM_SNAPSHOTS)
snapshots.add_system(solver.state)
snapshots.add_task("integ(RHO0 * ux * uz, 'x') / XMAX", name='S_{px}')
# Flow properties
flow = GlobalFlowProperty(solver, cadence=10)
flow.add_property('sqrt(ux**2 + ux**2)', name='u')
# Main loop
logger.info('Starting sim...')
while solver.ok:
cfl_dt = cfl.compute_dt() if NL else DT
solver.step(cfl_dt)
curr_iter = solver.iteration
if curr_iter % int((T_F / DT) / NUM_SNAPSHOTS) == 0:
logger.info('Reached time %f out of %f, timestep %f vs max %f',
solver.sim_time, solver.stop_sim_time, cfl_dt, DT)
logger.info('Max u = %e' % flow.max('u'))
print('Attempting to load snapshots in', filename)
_, dt = solver.load_state(filename, -1)
print('Loaded snapshots')
else:
print('No snapshots found')
dt = DT
cfl = CFL(solver,
initial_dt=dt,
cadence=5,
max_dt=DT,
min_dt=0.007,
safety=0.5,
threshold=0.10)
cfl.add_velocities(('ux', 'uz'))
cfl.add_frequency(DT)
snapshots = solver.evaluator.add_file_handler(snapshots_dir,
sim_dt=T_F / NUM_SNAPSHOTS)
snapshots.add_system(solver.state)
# Flow properties
flow = GlobalFlowProperty(solver, cadence=10)
flow.add_property('sqrt(ux**2 + ux**2)', name='u')
# Main loop
logger.info('Starting sim...')
while solver.ok:
cfl_dt = cfl.compute_dt()
solver.step(cfl_dt)
curr_iter = solver.iteration