ptyrad.utils.physics

ptyrad.utils.physics#

Physics-related functions of probes, propagators, and constants, etc.

Functions

complex_object_interp3d(complex_object, ...)

Interpolate a 3D complex object while preserving multiscattering behavior.

complex_object_z_resample_torch(obj, dz_now, ...)

Resample a complex 3D object along the depth (z) axis while conserving amplitude product, phase sum, and total thickness.

get_EM_constants(acceleration_voltage, ...)

get_default_probe_simu_params(init_params)

Get default probe simulation parameters based on the init_params dict

hermite_like(fundam, M, N)

infer_dx_from_params([dx, dk, kMax, da, ...])

Infer the real-space pixel size (dx) based on available unit-related parameters.

make_fzp_probe(probe_params[, verbose])

Generates a Fresnel zone plate probe with internal Fresnel propagation for x-ray ptychography simulations.

make_mixed_probe(probe, pmodes, pmode_init_pows)

Make a mixed state probe from a single state probe

make_stem_probe(probe_params[, verbose])

near_field_evolution(Npix_shape, dx, dz, lambd)

Fresnel propagator

near_field_evolution_torch(Npix_shape, dx, ...)

Fresnel propagator

orthogonalize_modes_vec_np(modes[, sort])

Orthogonalize the modes using SVD-like procedure via eigen decomposition.

sort_by_mode_int_np(modes)