pygrt-kit 0.2.0__py3-none-win_amd64.whl

pygrt/C_extension/Makefile

@@ -0,0 +1,137 @@
+
+SRC_DIR := src
+INC_DIR := include
+BUILD_DIR := build
+BIN_DIR := bin
+LIB_DIR := lib
+LNAME := grt
+LIBS_NAME := lib$(LNAME).a
+LIBD_NAME := lib$(LNAME).so
+
+VERSION := v$(shell cat version )
+CURRENT_DIR := $(shell pwd)
+
+# use static library, make life easier...
+# RPATH_FLAGS :=  -L$(LIB_DIR) -l:$(LIBS_NAME)  # ':libXYZ.a' not worked on MacOS
+RPATH_FLAGS := $(LIB_DIR)/$(LIBS_NAME)
+
+# link options for FFTW
+# NEED EXPLICITY SET LIBRARY PATH on MacOS
+LFFT_FLAGS := -l:libfftw3.a  -l:libfftw3f.a
+
+CC := gcc
+
+
+# add -static to enforce gcc link the static library, no matter standard-lib or custom-lib,
+# -l like "-l:libXYZ.a" should be saved for conditional compile.
+FOMPFLAGS := -fopenmp 
+
+# link static library on Windows
+LINK_STATIC := 
+LDFLAGS := $(FOMPFLAGS)
+
+# expand stack memory for Windows
+STACK_MEM := 
+
+ifeq ($(OS),Windows_NT)  # link static oenpmp on windows
+	STACK_MEM := -Wl,-stack,0x1000000
+    LINK_STATIC := -static
+	LDFLAGS := $(LINK_STATIC) $(FOMPFLAGS)
+endif
+
+# change architecture for macOS, from make command
+ARCH = 
+
+# However, Maybe -static is not working for standard-lib on MacOS, 
+# at least for now, it's not a big problem.
+CFLAGS :=  $(LINK_STATIC) $(LFFT_FLAGS) -lm -O3 -g \
+		-fPIC -Wall -Wextra $(STACK_MEM) -I$(INC_DIR) \
+		  -DGRT_VERSION="\"$(VERSION)\""  $(ARCH)  $(FOMPFLAGS)
+#  -fdump-tree-all  -g -ffast-math -O3 -fno-associative-math  -march=native -mtune=native 
+
+
+INCS := $(wildcard $(INC_DIR)/*.h)
+SRCS := $(wildcard $(SRC_DIR)/*.c)
+SRCS_NOMAIN := $(SRCS)
+
+# remove main functions
+SRCS_NOMAIN := $(filter-out $(SRC_DIR)/grt_main.c, $(SRCS_NOMAIN))
+SRCS_NOMAIN := $(filter-out $(SRC_DIR)/grt_syn.c, $(SRCS_NOMAIN))
+SRCS_NOMAIN := $(filter-out $(SRC_DIR)/grt_travt.c, $(SRCS_NOMAIN))
+OBJS_NOMAIN := $(patsubst $(SRC_DIR)/%.c, $(BUILD_DIR)/%.o, $(SRCS_NOMAIN))
+OBJS := $(patsubst $(SRC_DIR)/%.c, $(BUILD_DIR)/%.o, $(SRCS))
+
+
+.PHONY: all clean cleanbuild
+
+# library and several programs
+TARGET = libgrt \
+         grt \
+		 grt.syn \
+		 grt.travt
+
+all: $(TARGET) 
+
+
+libgrt:      $(LIB_DIR)/$(LIBS_NAME)  $(LIB_DIR)/$(LIBD_NAME)
+grt:         $(BIN_DIR)/grt 
+grt.syn:     $(BIN_DIR)/grt.syn 
+grt.travt:	 $(BIN_DIR)/grt.travt
+
+
+# generate executable files
+$(BIN_DIR)/grt: $(BUILD_DIR)/grt_main.o  libgrt
+	@echo $(shell ls $(LIB_DIR)/*)
+	@mkdir -p $(BIN_DIR)
+	$(CC) -o $@ $<  $(RPATH_FLAGS) $(CFLAGS)
+
+$(BIN_DIR)/grt.syn: $(BUILD_DIR)/grt_syn.o  libgrt
+	@echo $(shell ls $(LIB_DIR)/*)
+	@mkdir -p $(BIN_DIR)
+	$(CC) -o $@ $<  $(RPATH_FLAGS) $(CFLAGS)
+
+$(BIN_DIR)/grt.travt: $(BUILD_DIR)/grt_travt.o  libgrt
+	@echo $(shell ls $(LIB_DIR)/*)
+	@mkdir -p $(BIN_DIR)
+	$(CC) -o $@ $<  $(RPATH_FLAGS) $(CFLAGS)
+
+
+# build dynamic library
+# The order in the compiled statement is critical, otherwise the library will not be linked
+$(LIB_DIR)/$(LIBD_NAME): $(OBJS_NOMAIN)
+	@mkdir -p $(BUILD_DIR)
+	@mkdir -p $(LIB_DIR)
+	$(CC) -shared -o $@ $^ $(LDFLAGS) 
+
+# build static library
+$(LIB_DIR)/$(LIBS_NAME): $(OBJS_NOMAIN)
+	@mkdir -p $(BUILD_DIR)
+	@mkdir -p $(LIB_DIR)
+	ar rcs $@ $^ 
+	
+
+# Compile object file
+$(BUILD_DIR)/%.o: $(SRC_DIR)/%.c
+	@mkdir -p $(BUILD_DIR)
+	$(CC)  -o $@ -c $< $(CFLAGS)
+
+
+# Automatically generate dependencies
+DEPENDS := $(OBJS:.o=.d)
+-include $(DEPENDS)
+
+$(BUILD_DIR)/%.d: $(SRC_DIR)/%.c
+	@mkdir -p $(BUILD_DIR)
+	@$(CC) $(CFLAGS) -MM $< > $@.$$$$; \
+	sed 's,\($*\)\.o[ :]*,$(BUILD_DIR)/\1.o $@ : ,g' < $@.$$$$ > $@; \
+	rm -f $@.$$$$
+
+# remove all generated files
+clean:
+	rm -rf $(BIN_DIR)
+	rm -rf $(BUILD_DIR)
+	rm -rf $(LIB_DIR) 
+
+# remove object files
+cleanbuild:
+	rm -rf $(BUILD_DIR)

pygrt/C_extension/include/attenuation.h

@@ -0,0 +1,24 @@
+/**
+ * @file   attenuation.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ * 
+ */
+
+
+#pragma once 
+
+#include "const.h"
+
+/**
+ *  品质因子Q 对 波速的影响, Futterman causal Q, 参考Aki&Richards, 2009, Chapter 5.5.1
+ * \f[
+ *  c(\omega) = c(2\pi)\times (1 + \frac{1}{\pi Q} \log(\frac{\omega}{2\pi}) + \frac{i}{2Q}) 
+ * \f] 
+ * 
+ * @param Qinv     (in) 1/Q
+ * @param omega    (in)复数频率\f$ \tilde{\omega} =\omega - i\zeta \f$ 
+ * @return atncoef 系数因子，作用在 \f$ k=\omega / c(\omega)\f$的计算
+ */
+MYCOMPLEX attenuation_law(MYREAL Qinv, MYCOMPLEX omega);

pygrt/C_extension/include/bessel.h

@@ -0,0 +1,22 @@
+/**
+ * @file   bessel.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ *                   
+ */
+
+#pragma once
+
+#include "const.h"
+
+/**
+ * 计算Bessel函数 \f$ J_m(x), m=0,1,2 \f$ 
+ * 
+ * @param x          自变量 
+ * @param bj0  (out) \f$ J_0(x) \f$
+ * @param bj1  (out) \f$ J_1(x) \f$
+ * @param bj2  (out) \f$ J_2(x) \f$
+ * 
+ */
+void bessel012(MYREAL x, MYREAL *bj0, MYREAL *bj1, MYREAL *bj2);

pygrt/C_extension/include/colorstr.h

@@ -0,0 +1,19 @@
+/**
+ * @file   colorstr.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-12
+ * 
+ * 定义一些字符串宏，以打印有颜色的字符串
+ *                   
+ */
+
+
+#pragma once 
+
+#define DEFAULT_RESTORE  "\033[0m" 
+#define REGULAR_RED      "\033[0;31m"
+#define BOLD_RED         "\033[1;31m"
+#define REGULAR_GREEN    "\033[0;32m"
+#define BOLD_GREEN       "\033[1;32m"
+#define REGULAR_BLUE     "\033[0;35m"
+#define BOLD_BLUE        "\033[1;35m"

pygrt/C_extension/include/const.h

@@ -0,0 +1,115 @@
+/**
+ * @file   const.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ *                   
+ */
+
+#pragma once
+
+#include <complex.h> 
+#include <math.h>
+// #include <tgmath.h>
+
+// CMPLX macro not exist on MacOS
+#ifndef CMPLX
+#define CMPLX(real, imag) ((double)(real) + (double)(imag) * I)  ///< 复数扩展宏，添加此指令以适配MacOS
+#endif
+
+
+#if defined(_WIN32) || defined(__WIN32__)
+#define _TEST_WHETHER_WIN32_ 1
+#else
+#define _TEST_WHETHER_WIN32_ 0  ///< 测试是否是windows系统
+#endif
+
+#if _TEST_WHETHER_WIN32_
+#include <direct.h> /* _mkdir */
+#define mkdir(x, y) _mkdir(x)  ///< 为windows系统修改mkdir函数
+#endif
+
+
+
+// #define GRT_USE_FLOAT  ///< 是否使用单精度浮点数
+
+typedef int MYINT;  ///< 整数 
+
+
+#ifdef GRT_USE_FLOAT 
+    typedef float _Complex  MYCOMPLEX;   ///< 复数
+    typedef float MYREAL;   ///< 浮点数
+    // 数学函数
+    #define EXP(x)    (expf(x))      ///< \f$ \exp(x) \f$
+    #define CEXP(x)   (cexpf(x))      ///< 复数域 \f$ \exp(x) \f$
+    #define CLOG(x)   (clogf(x))      ///< 复数域 \f$ \ln(x) \f$
+    #define CSQRT(x)  (csqrtf(x))     ///< 复数域 \f$ x^{1/2} \f$
+    #define SQRT(x)   (sqrtf(x))     ///< \f$ \sqrt{x} \f$
+    #define SIN(x)    (sinf(x))      ///< \f$ \sin(x) \f$
+    #define COS(x)    (cosf(x))      ///< \f$ \cos(x) \f$
+    #define FABS(x)   (fabsf(x))     ///< \f$ |x| \f$
+    #define CABS(x)   (cabsf(x))     ///< 复数域 \f$ |x| \f$
+    #define J0(x)     (j0f(x))       ///< \f$ J_0(x) \f$
+    #define J1(x)     (j1f(x))       ///< \f$ J_1(x) \f$
+    #define JN(n,x)   (jnf((n),(x))) ///< \f$ J_n(x) \f$
+
+    #define CREAL(x)  (crealf(x))    ///< 复数实部 \f$ \Re[x] \f$
+    #define CIMAG(x)  (cimagf(x))    ///< 复数虚部 \f$ \Im[x] \f$
+
+    // 常数
+    #define RZERO 0.0f            ///< 0.0
+    #define RQUART 0.25f          ///< 0.25
+    #define RHALF 0.5f            ///< 0.5
+    #define RONE 1.0f            ///< 1.0
+    #define RTWO 2.0f            ///< 2.0
+    #define RTHREE 3.0f            ///< 3.0
+    #define RFOUR 4.0f            ///< 4.0
+    #define PI  3.1415926f         ///< \f$ \pi \f$
+    #define PI2 6.2831853f        ///< \f$ 2\pi \f$
+    #define HALFPI 1.5707963f   ///< \f$ \frac{\pi}{2} \f$
+    #define QUARTERPI 0.78539816f   ///< \f$ \frac{\pi}{4} \f$
+    #define INV_SQRT_TWO 0.70710678f   ///< \f$ \frac{1}{\sqrt{2}} \f$ 
+
+#else 
+    typedef double _Complex MYCOMPLEX;
+    typedef double MYREAL;
+    // 数学函数
+    #define EXP(x)    (exp(x))      ///< \f$ \exp(x) \f$
+    #define CEXP(x)   (cexp(x))      ///< 复数域 \f$ \exp(x) \f$
+    #define CLOG(x)   (clog(x))      ///< 复数域 \f$ \ln(x) \f$
+    #define CSQRT(x)  (csqrt(x))     ///< 复数域 \f$ x^{1/2} \f$
+    #define SQRT(x)   (sqrt(x))     ///< \f$ \sqrt{x} \f$
+    #define SIN(x)    (sin(x))      ///< \f$ \sin(x) \f$
+    #define COS(x)    (cos(x))      ///< \f$ \cos(x) \f$
+    #define FABS(x)   (fabs(x))     ///< \f$ |x| \f$
+    #define CABS(x)   (cabs(x))     ///< 复数域 \f$ |x| \f$
+    #define J0(x)     (j0(x))       ///< \f$ J_0(x) \f$
+    #define J1(x)     (j1(x))       ///< \f$ J_1(x) \f$
+    #define JN(n,x)   (jn((n),(x))) ///< \f$ J_n(x) \f$
+
+    #define CREAL(x)  (creal(x))    ///< 复数实部 \f$ \Re[x] \f$
+    #define CIMAG(x)  (cimag(x))    ///< 复数虚部 \f$ \Im[x] \f$
+
+    // 常数
+    #define RZERO 0.0            ///< 0.0
+    #define RQUART 0.25          ///< 0.25
+    #define RHALF 0.5             ///< 0.5
+    #define RONE 1.0            ///< 1.0
+    #define RTWO 2.0            ///< 2.0
+    #define RTHREE 3.0            ///< 3.0
+    #define RFOUR 4.0            ///< 4.0
+    #define PI  3.141592653589793        ///< \f$ \pi \f$
+    #define PI2 6.283185307179586        ///< \f$ 2\pi \f$
+    #define HALFPI 1.5707963267948966   ///< \f$ \frac{\pi}{2} \f$
+    #define QUARTERPI 0.7853981633974483   ///< \f$ \frac{\pi}{4} \f$
+    #define INV_SQRT_TWO 0.7071067811865475   ///< \f$ \frac{1}{\sqrt{2}} \f$ 
+
+#endif
+
+#define CZERO CMPLX(RZERO, RZERO)  ///< 0.0 + j0.0
+#define CONE  CMPLX(RONE,  RZERO)  ///< 1.0 + j0.0
+#define INIT_C_ZERO_2x2_MATRIX {{CZERO, CZERO}, {CZERO, CZERO}}   ///< 初始化复数0矩阵
+#define INIT_C_IDENTITY_2x2_MATRIX {{CONE, CZERO}, {CZERO, CONE}}  ///< 初始化复数单位阵
+
+#define MIN_DEPTH_GAP_SRC_RCV  1.0  ///< 震源和台站的最小深度差（不做绝对限制，仅用于参考波数积分上限）
+#define GCC_ALWAYS_INLINE __attribute__((always_inline))  ///< gcc编译器不改动内联函数

pygrt/C_extension/include/dwm.h

@@ -0,0 +1,44 @@
+/**
+ * @file   dwm.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ * 以下代码实现的是 使用离散波数法求积分，参考：
+ * 
+ *         1. 姚振兴, 谢小碧. 2022/03. 理论地震图及其应用（初稿）.  
+ *         2. Yao Z. X. and D. G. Harkrider. 1983. A generalized refelection-transmission coefficient 
+ *               matrix and discrete wavenumber method for synthetic seismograms. BSSA. 73(6). 1685-1699
+ * 
+ */
+
+#pragma once 
+
+#include "model.h"
+
+
+/**
+ * 传统的离散波数积分，结果以三维数组的形式返回，形状为[nr][3][4], 分别代表震中距、阶数(m=0,1,2)
+ * 和4种积分类型(p=0,1,2,3)
+ * 
+ * @param  mod1d     (in)`MODEL1D` 结构体指针
+ * @param  dk        (in)波数积分间隔
+ * @param  kmax      (in)波数积分的上限
+ * @param  keps      (in)波数积分的收敛条件，要求在某震中距下所有格林函数都收敛，为负数代表不提前判断收敛，按照波数积分上限进行积分
+ * @param  omega     (in)复数频率
+ * @param  nr        (in)震中距数量
+ * @param  rs        (in)震中距数组
+ * 
+ * @param  sum_EXP_J[nr][3][4]  (out)爆炸源
+ * @param  sum_VF_J[nr][3][4]   (out)垂直力源
+ * @param  sum_HF_J[nr][3][4]   (out)水平力源
+ * @param  sum_DC_J[nr][3][4]   (out)双力偶源
+ * @param  fstats[nr])          (out)不同震中距的格林函数积分过程文件
+ * 
+ * @return  k        积分截至时的波数
+ */
+MYREAL discrete_integ(
+    const MODEL1D *mod1d, MYREAL dk, MYREAL kmax, MYREAL keps, MYCOMPLEX omega, 
+    MYINT nr, MYREAL *rs,
+    MYCOMPLEX sum_EXP_J[nr][3][4], MYCOMPLEX sum_VF_J[nr][3][4],  
+    MYCOMPLEX sum_HF_J[nr][3][4],  MYCOMPLEX sum_DC_J[nr][3][4],  
+    FILE *(fstats[nr]));

pygrt/C_extension/include/fim.h

@@ -0,0 +1,78 @@
+/**
+ * @file   fim.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ * 以下代码实现的是基于线性插值的Filon积分，参考：
+ * 
+ *         1. 姚振兴, 谢小碧. 2022/03. 理论地震图及其应用（初稿）.   
+ *         2. 纪晨, 姚振兴. 1995. 区域地震范围的宽频带理论地震图算法研究. 地球物理学报. 38(4)    
+ *               
+ */
+
+#pragma once 
+
+#include "const.h"
+#include "model.h"
+
+
+
+/**
+ * 基于线性插值的Filon积分(5.9.6-11), 在大震中距下对Bessel函数取零阶近似，得
+ * \f[
+ * J_m(x) \approx \sqrt{\frac{2}{\pi x}} \cos(x - \frac{m \pi}{2} - \frac{\pi}{4})
+ * \f]
+ * 其中\f$x=kr\f$. 结果以三维数组的形式返回，形状为[nr][3][4], 分别代表震中距、阶数(m=0,1,2)
+ * 和4种积分类型(p=0,1,2,3)
+ * 
+ * 
+ * @param  mod1d     (in)`MODEL1D` 结构体指针
+ * @param  dk        (in)波数积分间隔
+ * @param  kmax      (in)波数积分的上限
+ * @param  keps      (in)波数积分的收敛条件，要求在某震中距下所有格林函数都收敛
+ * @param  omega     (in)复数频率
+ * @param  nr        (in)震中距数量
+ * @param  rs        (in)震中距数组
+ *
+ * @param  sum_EXP_J[nr][3][4]  (out)爆炸源
+ * @param  sum_VF_J[nr][3][4]   (out)垂直力源
+ * @param  sum_HF_J[nr][3][4]   (out)水平力源
+ * @param  sum_DC_J[nr][3][4]   (out)双力偶源
+ * @param  fstats[nr]           (out)不同震中距的格林函数积分过程文件
+ * 
+ * @return  k        积分截至时的波数
+ */
+MYREAL linear_filon_integ(
+    const MODEL1D *mod1d, MYREAL dk, MYREAL kmax, MYREAL keps, MYCOMPLEX omega, 
+    MYINT nr, MYREAL *rs,
+    MYCOMPLEX sum_EXP_J[nr][3][4], MYCOMPLEX sum_VF_J[nr][3][4],  
+    MYCOMPLEX sum_HF_J[nr][3][4],  MYCOMPLEX sum_DC_J[nr][3][4],  
+    FILE *(fstats[nr]));
+
+
+
+/**
+ *  和int_Pk函数类似，不过是计算核函数和渐近Bessel函数的乘积，其中涉及两种数组形状：
+ *    + [3][3]. 存储的是核函数，第一个维度3代表阶数m=0,1,2，第二个维度3代表三类系数qm,wm,vm  
+ *    + [3][4]. 存储的是该dk区间内的积分值，维度3代表阶数m=0,1,2，维度4代表4种类型的F(k,w)Jm(kr)k的类型
+ * 
+ * 
+ * @param     k     (in)波数  
+ * @param     r     (in)震中距 
+ * @param    EXP_qwv[3][3]    (in)爆炸源核函数
+ * @param    VF_qwv[3][3]     (in)垂直力源核函数
+ * @param    HF_qwv[3][3]     (in)水平力源核函数
+ * @param    DC_qwv[3][3]     (in)双力偶源核函数 
+ * @param    EXP_J[3][4]      (out)爆炸源，该dk区间内的积分值，下同
+ * @param    VF_J[3][4]       (out)垂直力源
+ * @param    HF_J[3][4]       (out)水平力源
+ * @param    DC_J[3][4]       (out)双力偶源
+ *  
+ */
+void int_Pk_filon(
+    MYREAL k, MYREAL r, 
+    const MYCOMPLEX EXP_qwv[3][3], const MYCOMPLEX VF_qwv[3][3], 
+    const MYCOMPLEX HF_qwv[3][3],  const MYCOMPLEX DC_qwv[3][3], 
+    MYCOMPLEX EXP_J[3][4], MYCOMPLEX VF_J[3][4], 
+    MYCOMPLEX HF_J[3][4],  MYCOMPLEX DC_J[3][4] );
+

pygrt/C_extension/include/grt.h

@@ -0,0 +1,128 @@
+/**
+ * @file   grt.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ * 以下代码实现的是 广义反射透射系数矩阵+离散波数法 计算理论地震图，参考：
+ * 
+ *         1. 姚振兴, 谢小碧. 2022/03. 理论地震图及其应用（初稿）.  
+ *         2. Yao Z. X. and D. G. Harkrider. 1983. A generalized refelection-transmission coefficient 
+ *               matrix and discrete wavenumber method for synthetic seismograms. BSSA. 73(6). 1685-1699
+ *   
+ *                 
+ */
+
+#pragma once 
+
+#include "model.h"
+
+/**
+ * 设置OpenMP多线程数
+ * 
+ * @param   num_threads    (in)线程数
+ */
+void set_num_threads(int num_threads);
+
+
+/**
+ * 积分计算Z, R, T三个分量格林函数的频谱的核心函数（被C函数调用）
+ * 
+ * @param      pymod1d      (in)`PYMODEL1D` 结构体指针 
+ * @param      nf1          (in)开始计算频谱的频率索引值, 总范围在[nf1, nf2]
+ * @param      nf2          (in)结束计算频谱的频率索引值, 总范围在[nf1, nf2]
+ * @param      nf           (in)所有频点个数
+ * @param      freqs        (in)所有频点的频率值（包括未计算的）
+ * @param      nr           (in)震中距数量
+ * @param      rs           (in)震中距数组 
+ * @param      wI           (in)虚频率, \f$ \tilde{\omega} =\omega - i\wI \f$ 
+ * @param      vmin_ref     (in)参考最小速度，用于定义波数积分的上限
+ * @param      keps         (in)波数积分的收敛条件，要求在某震中距下所有格林函数都收敛，为负数代表不提前判断收敛，按照波数积分上限进行积分 
+ * @param      ampk         (in)影响波数k积分上限的系数，见下方
+ * @param      iwk0         (in)k0是否取随频率变化的线性关系，即 \f$ k_{max} = k_{max} * f/f_{max} \f$ 
+ * @param      k0           (in)波数积分的上限 \f$ \tilde{k_{max}}=\sqrt{(k_{0}*\pi/hs)^2 + (ampk*w/vmin_{ref})^2} \f$ ，k循环必须退出, hs=max(震源和台站深度差,1.0) 
+ * @param      Length       (in)波数k积分间隔 \f$ dk=2\pi/(fabs(L)*r_{max}) \f$ , 如果为负数，则使用线性Filon积分
+ * @param      print_progressbar    (in)是否打印进度条
+ * @param      EXPgrn[nr][2]      (out)复数数组，爆炸源的Z、R分量频谱结果
+ * @param      VFgrn[nr][2]       (out)复数数组，垂直力源的Z、R分量频谱结果
+ * @param      HFgrn[nr][3]       (out)复数数组，水平力源的Z、R、T分量频谱结果
+ * @param      DDgrn[nr][2]       (out)复数数组，45度倾滑的Z、R分量频谱结果
+ * @param      DSgrn[nr][3]       (out)复数数组，90度倾滑的Z、R、T分量频谱结果
+ * @param      SSgrn[nr][3]       (out)复数数组，90度走滑的Z、R、T分量频谱结果
+ * 
+ * @param      statsstr          (in) 积分结果输出目录
+ * @param      nstatsidxs        (in) 输出积分结果的特定频点的个数
+ * @param      statsidxs         (in) 特定频点的索引值
+ * 
+ */ 
+void integ_grn_spec_in_C(
+    PYMODEL1D *pymod1d, MYINT nf1, MYINT nf2, MYINT nf, MYREAL *freqs,  
+    MYINT nr, MYREAL *rs, MYREAL wI, 
+    MYREAL vmin_ref, MYREAL keps, MYREAL ampk, bool iwk0, MYREAL k0, MYREAL Length,       
+    bool print_progressbar, 
+
+    // 返回值，维度2代表Z、R分量，维度3代表Z、R、T分量
+    MYCOMPLEX *EXPcplx[nr][2], // EXZ, EXR 的实部和虚部
+    MYCOMPLEX *VFcplx[nr][2],  // VFZ, VFR 的实部和虚部
+    MYCOMPLEX *HFcplx[nr][3],  // HFZ, HFR, HFT 的实部和虚部
+    MYCOMPLEX *DDcplx[nr][2],  // DDZ, DDR 的实部和虚部      [DD: 45-dip slip]
+    MYCOMPLEX *DScplx[nr][3],  // DSZ, DSR, DST 的实部和虚部 [DS: 90-dip slip]
+    MYCOMPLEX *SScplx[nr][3],  // SSZ, SSR, SST 的实部和虚部 [SS: strike slip]
+
+    const char *statsstr, // 积分结果输出
+    MYINT  nstatsidxs, // 仅输出特定频点
+    MYINT *statsidxs
+);
+
+
+/**
+ * 积分计算Z, R, T三个分量格林函数的频谱的核心函数（被Python调用）  
+ * 
+ * @param      pymod1d      (in)`PYMODEL1D` 结构体指针 
+ * @param      nf1          (in)开始计算频谱的频率索引值, 总范围在[nf1, nf2]
+ * @param      nf2          (in)结束计算频谱的频率索引值, 总范围在[nf1, nf2]
+ * @param      nf           (in)所有频点个数
+ * @param      freqs        (in)所有频点的频率值（包括未计算的）
+ * @param      nr           (in)震中距数量
+ * @param      rs           (in)震中距数组 
+ * @param      wI           (in)虚频率, \f$ \tilde{\omega} =\omega - i\wI \f$ 
+ * @param      vmin_ref     (in)参考最小速度，用于定义波数积分的上限
+ * @param      keps         (in)波数积分的收敛条件，要求在某震中距下所有格林函数都收敛，为负数代表不提前判断收敛，按照波数积分上限进行积分 
+ * @param      ampk         (in)影响波数k积分上限的系数，见下方
+ * @param      iwk0         (in)k0是否取随频率变化的线性关系，即 \f$ k_{max} = k_{max} * f/f_{max} \f$ 
+ * @param      k0           (in)波数积分的上限 \f$ \tilde{k_{max}}=\sqrt{(k_{0}*\pi/hs)^2 + (ampk*w/vmin_{ref})^2} \f$ ，k循环必须退出, hs=max(震源和台站深度差,1.0) 
+ * @param      Length       (in)波数k积分间隔 \f$ dk=2\pi/(fabs(L)*r_{max}) \f$ , 如果为负数，则使用线性Filon积分
+ * @param      print_progressbar    (in)是否打印进度条
+ * @param      EXPgrn[nr][2]      (out)`GRN` 结构体指针，爆炸源的Z、R分量频谱结果
+ * @param      VFgrn[nr][2]       (out)`GRN` 结构体指针，垂直力源的Z、R分量频谱结果
+ * @param      HFgrn[nr][3]       (out)`GRN` 结构体指针，水平力源的Z、R、T分量频谱结果
+ * @param      DDgrn[nr][2]       (out)`GRN` 结构体指针，45度倾滑的Z、R分量频谱结果
+ * @param      DSgrn[nr][3]       (out)`GRN` 结构体指针，90度倾滑的Z、R、T分量频谱结果
+ * @param      SSgrn[nr][3]       (out)`GRN` 结构体指针，90度走滑的Z、R、T分量频谱结果
+ * 
+ * @param      statsstr          (in) 积分结果输出目录
+ * @param      nstatsidxs        (in) 输出积分结果的特定频点的个数
+ * @param      statsidxs         (in) 特定频点的索引值
+ * 
+ */ 
+void integ_grn_spec(
+    PYMODEL1D *pymod1d, MYINT nf1, MYINT nf2, MYINT nf, MYREAL *freqs,  
+    MYINT nr, MYREAL *rs, MYREAL wI, 
+    MYREAL vmin_ref, MYREAL keps, MYREAL ampk, bool iwk0, MYREAL k0, MYREAL Length,       
+    bool print_progressbar, 
+
+    // 返回值，维度2代表Z、R分量，维度3代表Z、R、T分量
+    GRN *EXPgrn[nr][2], // EXZ, EXR 的实部和虚部
+    GRN *VFgrn[nr][2],  // VFZ, VFR 的实部和虚部
+    GRN *HFgrn[nr][3],  // HFZ, HFR, HFT 的实部和虚部
+    GRN *DDgrn[nr][2],  // DDZ, DDR 的实部和虚部      [DD: 45-dip slip]
+    GRN *DSgrn[nr][3],  // DSZ, DSR, DST 的实部和虚部 [DS: 90-dip slip]
+    GRN *SSgrn[nr][3],  // SSZ, SSR, SST 的实部和虚部 [SS: strike slip]
+
+    const char *statsstr, // 积分结果输出
+    MYINT  nstatsidxs, // 仅输出特定频点
+    MYINT *statsidxs
+);
+
+
+
+

pygrt/C_extension/include/iostats.h

@@ -0,0 +1,72 @@
+/**
+ * @file   iostats.h
+ * @author Zhu Dengda (zhudengda@mail.iggcas.ac.cn)
+ * @date   2024-07-24
+ * 
+ * 对积分过程中的核函数和积分值进行记录                  
+ */
+
+#pragma once 
+
+#include <stdbool.h>
+#include <stdio.h>
+
+#include "const.h"
+
+
+
+/**
+ * 对离散波数法以及Filon积分的中间结果进行记录，其中涉及两种数组形状：
+ *    + [3][3]. 存储的是核函数，第一个维度3代表阶数m=0,1,2，第二个维度3代表三类系数qm,wm,vm  
+ *    + [3][4]. 存储的是该dk区间内的积分值，维度3代表阶数m=0,1,2，维度4代表4种类型的F(k,w)Jm(kr)k的类型
+ * 
+ * @param    f0     (out)二进制文件指针 
+ * @param    k       (in)波数 
+ * @param    EXP_qwv[3][3]    (in)爆炸源核函数
+ * @param    VF_qwv[3][3]     (in)垂直力源核函数
+ * @param    HF_qwv[3][3]     (in)水平力源核函数
+ * @param    DC_qwv[3][3]     (in)双力偶源核函数
+ * @param    EXP_J[3][4]      (in)爆炸源，该dk区间内的积分值，下同
+ * @param    VF_J[3][4]       (in)垂直力源
+ * @param    HF_J[3][4]       (in)水平力源
+ * @param    DC_J[3][4]       (in)双力偶源
+ * 
+ * 
+ * @note     文件记录的值均为波数积分的中间结果，与最终的结果还差一系列的系数，
+ *           记录其值主要用于参考其变化趋势。
+ */
+void write_stats(
+    FILE *f0, MYREAL k, 
+    const MYCOMPLEX EXP_qwv[3][3], const MYCOMPLEX VF_qwv[3][3], 
+    const MYCOMPLEX HF_qwv[3][3],  const MYCOMPLEX DC_qwv[3][3], 
+    const MYCOMPLEX EXP_J[3][4], const MYCOMPLEX VF_J[3][4], 
+    const MYCOMPLEX HF_J[3][4],  const MYCOMPLEX DC_J[3][4]);
+
+
+
+/**
+ * 对峰谷平均法的中间结果进行记录，其中[3][4]的数组形状代表存储的
+ * 是最终的积分值，维度3代表阶数m=0,1,2，维度4代表4种类型的F(k,w)Jm(kr)k的类型
+ * 
+ * @param    f0     (out)二进制文件指针 
+ * @param    k       (in)波数 
+ * @param    maxNpt  (in)波峰+波谷的数量(本质是计算中提前预设的量，见ptam.c文件中PTA_method函数)
+ * @param    EXPpt[3][4][maxNpt]    (in)爆炸源，最终收敛积分值使用的波峰波谷值，下同
+ * @param    VFpt[3][4][maxNpt]     (in)垂直力源
+ * @param    HFpt[3][4][maxNpt]     (in)水平力源
+ * @param    DCpt[3][4][maxNpt]     (in)双力偶源
+ * @param    kEXPpt[3][4][maxNpt]    (in)爆炸源，最终收敛积分值使用的波峰波谷值对应的波数k值，下同
+ * @param    kVFpt[3][4][maxNpt]     (in)垂直力源
+ * @param    kHFpt[3][4][maxNpt]     (in)水平力源
+ * @param    kDCpt[3][4][maxNpt]     (in)双力偶源
+ * 
+ * @note     文件记录的积分值与最终的结果还差一系列的系数，
+ *           记录其值主要用于参考其变化趋势。
+ * 
+ */
+void write_stats_ptam(
+    FILE *f0, MYREAL k, MYINT maxNpt, 
+    const MYCOMPLEX EXPpt[3][4][maxNpt], const MYCOMPLEX VFpt[3][4][maxNpt],
+    const MYCOMPLEX HFpt[3][4][maxNpt],  const MYCOMPLEX DCpt[3][4][maxNpt],
+    const MYREAL kEXPpt[3][4][maxNpt], const MYREAL kVFpt[3][4][maxNpt],
+    const MYREAL kHFpt[3][4][maxNpt],  const MYREAL kDCpt[3][4][maxNpt]);