Skip to content
Snippets Groups Projects
Select Git revision
  • colorclipping
  • master default protected
  • support-ragged
  • add_field_tests
  • support36
  • pyramid-master
  • 0.2.1
  • 0.2.0
  • 0.1.0
  • 0.0.1
  • 0.0.0
11 results
Blame Permalink
Forked from empyre / empyre
416 commits behind the upstream repository.
phase_mag_real.pyx 1.02 KiB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33 






import numpy as np
import math
cimport cython
cimport numpy as np


@cython.boundscheck(False)
@cython.wraparound(False)
def phase_mag_real_helper_1(
        unsigned int v_dim, unsigned int u_dim,
        double[:, :] phi_u,
        double[:, :] phi_v,
        double[:, :] u_mag,
        double[:, :] v_mag,
        double[:, :] phase, float threshold):
    cdef unsigned int i, j, ii, jj, iii, jjj
    cdef double u, v
    for j in range(v_dim):
        for i in range(u_dim):
            u = u_mag[j, i]
            v = v_mag[j, i]
            iii = u_dim - 1 - i
            jjj = v_dim - 1 - j
            if abs(u) > threshold:
               for jj in range(phase.shape[0]):
                    for ii in range(phase.shape[1]):
                        phase[jj, ii] += u * phi_u[jjj + jj, iii + ii]
            if abs(v) > threshold:
               for jj in range(phase.shape[0]):
                    for ii in range(phase.shape[1]):
                        phase[jj, ii] -= v * phi_v[jjj + jj, iii + ii]