state_initialization Namespace Reference

Functions
	fit_force_law (u_mean_samples, force_samples)
	predict_u_mean (force, linear_coeff, quadratic_coeff)
	scale_solver_state (solver, previous_u_mean, target_u_mean, previous_force, target_force)
	extract_solver_state (solver)
	load_solver_state (solver, state)
	_resample_axis (field, target_coords, axis)
	resample_field_linear (field, target_shape)
	resample_state_linear (state, target_shape)

Detailed Description

Helpers for continuation sweeps and cross-resolution solver initialization.

Function Documentation

fit_force_law()

state_initialization.fit_force_law	(		u_mean_samples,
			force_samples
	)

Fit ``force ~= a*u_mean + b*u_mean**2`` in a least-squares sense.

Definition at line 6 of file state_initialization.py.

predict_u_mean()

state_initialization.predict_u_mean	(		force,
			linear_coeff,
			quadratic_coeff
	)

Invert ``force = a*u_mean + b*u_mean**2`` for the positive root.

Definition at line 15 of file state_initialization.py.

scale_solver_state()

state_initialization.scale_solver_state	(		solver,
			previous_u_mean,
			target_u_mean,
			previous_force,
			target_force
	)

Scale an already converged solver state in place for continuation.

Definition at line 30 of file state_initialization.py.

extract_solver_state()

state_initialization.extract_solver_state

(

solver

)

Copy ``u``, ``v``, ``w`` and ``p`` from a solver into NumPy arrays.

Definition at line 42 of file state_initialization.py.

load_solver_state()

state_initialization.load_solver_state	(		solver,
			state
	)

Load a previously extracted state into a solver.

Definition at line 52 of file state_initialization.py.

_resample_axis()

state_initialization._resample_axis (   field,   target_coords,   axis  )

protected

Periodically resample one axis of a structured field.

Definition at line 62 of file state_initialization.py.

Referenced by resample_field_linear().

resample_field_linear()

state_initialization.resample_field_linear	(		field,
			target_shape
	)

Periodically resample a scalar field to a new ``(nz, ny, nx)`` shape.

Definition at line 77 of file state_initialization.py.

References _resample_axis().

Referenced by resample_state_linear().

resample_state_linear()

state_initialization.resample_state_linear	(		state,
			target_shape
	)

Resample all entries of a state dictionary with linear interpolation.

Definition at line 96 of file state_initialization.py.

References resample_field_linear().