PyFK is the python port of FK used to calculate the Green's function and the synthetic waveforms for the 1D Earth model.
The main features of this package are:
- Compute the Green's function for the explosion, single force, and double couple source using the frequency-wavenumber method.
- Compute the static displacements and corresponding Green's function.
- Compute the synthetic waveforms by convolving Green's function with the seismic source.
- Use the seismic data format of Obspy, which is easy to perform the signal processing.
And the package is unique as:
all the code is written in pure python, and it's compatible with Unix-like systems including Mac and Linux. The Windows is not supported, as the package uses the complex number in Cython, which uses the C99 standard of "complex.h" that has not been supported by the Visual Studio compiler.
it uses Cython to speed up the computationally expensive part (mainly the wavenumber integration).
The package has also provided three modes:
- Serial mode: the serial version implements the FK algorithm in Python.
- Parallel mode on CPU: the wavenumber integration can be paralleled by MPI.
- Parallel mode on GPU: the wavenumber integration can also be paralleled by CUDA on GPU.
The serial version and the parallel version on GPU can be simply installed using pip:
pip install pyfkOr conda:
conda install -c ziyixi pyfk
It's also easy to install the MPI enabled version:
PYFK_USE_MPI=1 pip install pyfk[mpi]
Extra packages including numba and cupy will be required to enable the GPU mode. For more details about the Installation, you can refer to the Installing part of the document.
The current bugs to fix or features to provide are listed in TODO.
Xi, Z., Li, J., Chen, M. and Wei, S., 2021, December. PyFK: A Fast MPI and CUDA Accelerated Python Package for Calculating Synthetic Seismograms Based on the Frequencywavenumber Method. In AGU Fall Meeting Abstracts (Vol. 2021, pp. S15E-0288).
PyFK is a free software: you can redistribute it and modify it under the terms of the Apache License. A copy of this license is provided in LICENSE.