forked from pressel/pycles
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPressureFFTSerial.pxd
More file actions
20 lines (18 loc) · 809 Bytes
/
PressureFFTSerial.pxd
File metadata and controls
20 lines (18 loc) · 809 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
cimport ParallelMPI
cimport Grid
cimport ReferenceState
cimport SparseSolvers
cimport DiagnosticVariables
cdef class PressureFFTSerial:
cdef:
double [:] a
double [:] b
double [:] c
double [:] kx2
double [:] ky2
inline void compute_diagonal(self, Grid.Grid Gr, ReferenceState.ReferenceState RS, Py_ssize_t i, Py_ssize_t j) nogil
cdef SparseSolvers.TDMA TDMA_Solver
cpdef initialize(self, Grid.Grid Gr, ReferenceState.ReferenceState RS, ParallelMPI.ParallelMPI Pa)
cpdef compute_modified_wave_numbers(self,Grid.Grid Gr)
cpdef compute_off_diagonals(self,Grid.Grid Gr, ReferenceState.ReferenceState RS)
cpdef solve(self,Grid.Grid Gr, ReferenceState.ReferenceState RS,DiagnosticVariables.DiagnosticVariables DV, ParallelMPI.ParallelMPI PM)