Forked from
empyre / empyre
402 commits behind the upstream repository.
-
Jan Caron authoredphasemapper: added Kernel helper class with methods to calculate the jacobi matrix or to multiply a vector with the (transposed) jacobi matrix without explicitly calculating the full matrix magcreator: added Shape.ellipsoid numcore: added (experimental) speed-up for jacobi-matrix (not faster) scripts: modified get_jacobi to test the new Kernel class, tests: test_compliance now also searches and creates a list of TODOs
Jan Caron authoredphasemapper: added Kernel helper class with methods to calculate the jacobi matrix or to multiply a vector with the (transposed) jacobi matrix without explicitly calculating the full matrix magcreator: added Shape.ellipsoid numcore: added (experimental) speed-up for jacobi-matrix (not faster) scripts: modified get_jacobi to test the new Kernel class, tests: test_compliance now also searches and creates a list of TODOs
phase_mag_real.pyx 3.86 KiB
# -*- coding: utf-8 -*-
"""Numerical core routines for the phase calculation using the real space approach.
Provides a helper function to speed up :func:`~pyramid.phasemapper.phase_mag_real` of module
:mod:`~pyramid.phasemapper`, by using C-speed for the for-loops and by omitting boundary and
wraparound checks.
"""
import numpy as np
import math
cimport cython
cimport numpy as np
@cython.boundscheck(False)
@cython.wraparound(False)
def phase_mag_real_core(
unsigned int v_dim, unsigned int u_dim,
double[:, :] v_phi, double[:, :] u_phi,
double[:, :] v_mag, double[:, :] u_mag,
double[:, :] phase, float threshold):
'''Numerical core routine for the phase calculation using the real space approach.
Parameters
----------
v_dim, u_dim : int
Dimensions of the projection along the two major axes.
v_phi, u_phi : :class:`~numpy.ndarray` (N=2)
Lookup tables for the pixel fields oriented in `u`- and `v`-direction.
v_mag, u_mag : :class:`~numpy.ndarray` (N=2)
Magnetization components in `u`- and `v`-direction.
phase : :class:`~numpy.ndarray` (N=2)
Matrix in which the resulting magnetic phase map should be stored.
threshold : float
The `threshold` determines which pixels contribute to the magnetic phase.
Returns
-------
None
'''
cdef unsigned int i, j, p, q, p_c, q_c
cdef double u_m, v_m
for j in range(v_dim):
for i in range(u_dim):
u_m = u_mag[j, i]
v_m = v_mag[j, i]
p_c = u_dim - 1 - i
q_c = v_dim - 1 - j
if abs(u_m) > threshold:
for q in range(v_dim):
for p in range(u_dim):
phase[q, p] += u_m * u_phi[q_c + q, p_c + p]
if abs(v_m) > threshold:
for q in range(v_dim):
for p in range(u_dim):
phase[q, p] -= v_m * v_phi[q_c + q, p_c + p]
@cython.boundscheck(False)
@cython.wraparound(False)
def get_jacobi_core(
unsigned int v_dim, unsigned int u_dim,
double[:, :] v_phi, double[:, :] u_phi,
double[:, :] jacobi):
'''DOCSTRING!'''
# TODO: Docstring!!!
cdef unsigned int i, j, p, q, p_min, p_max, q_min, q_max, u_column, v_column, row
for j in range(v_dim):
for i in range(u_dim):
q_min = (v_dim-1) - j
q_max = (2*v_dim-1) - j
p_min = (u_dim-1) - i
p_max = (2*u_dim-1) - i
v_column = i + u_dim*j + u_dim*v_dim
u_column = i + u_dim*j
for q in range(q_min, q_max):
for p in range(p_min, p_max):
q_c = q - (v_dim-1-j) # start at zero for jacobi column
p_c = p - (u_dim-1-i) # start at zero for jacobi column
row = p_c + u_dim*q_c
# u-component:
jacobi[row, u_column] = u_phi[q, p]
# v-component (note the minus!):
jacobi[row, v_column] = -v_phi[q, p]
@cython.boundscheck(False)
@cython.wraparound(False)
def get_jacobi_core(
unsigned int v_dim, unsigned int u_dim,
np.ndarray v_phi, np.ndarray u_phi,
np.ndarray jacobi):
'''DOCSTRING!'''
# TODO: Docstring!!!
cdef unsigned int i, j, p, q, p_min, p_max, q_min, q_max, u_column, v_column, row
for j in range(v_dim):
for i in range(u_dim):
# v_dim*u_dim columns for the u-component:
jacobi[:, i+u_dim*j] = \
np.reshape(u_phi[v_dim-1-j:(2*v_dim-1)-j, u_dim-1-i:(2*u_dim-1)-i], -1)
# v_dim*u_dim columns for the v-component (note the minus!):
jacobi[:, u_dim*v_dim+i+u_dim*j] = \
-np.reshape(v_phi[v_dim-1-j:(2*v_dim-1)-j, u_dim-1-i:(2*u_dim-1)-i], -1)