TEST: add kernel_freq_response in example/, and add some figs in README

Author: Dengda98

README.md

@@ -14,12 +14,13 @@
 [**中文文档**](https://pygrt.readthedocs.io/zh-cn/)
 
 # Overview
-**[PyGRT](https://github.com/Dengda98/PyGRT): An Efficient and Integrated C/Python Package for Computing Synthetic Seismograms, Strain and Stress in a Layered Half-Space Model (Dynamic & Static Cases)**
+**[PyGRT](https://github.com/Dengda98/PyGRT): An Efficient and Integrated C/Python Package for Computing Synthetic Seismograms, Strain, Rotation and Stress Tensor in a Layered Half-Space Model (Dynamic & Static Cases)**
 
  + **PyGRT** now can compute following properties in **both dynamic and static case.**
     - [x] **Displacements and its spatial derivatives**
-    - [x] **Strain**
-    - [x] **Stress**
+    - [x] **Strain Tensor**
+    - [x] **Rotation Tensor**
+    - [x] **Stress Tensor**
 
 
  + **At present, PyGRT can run on**
@@ -138,6 +139,16 @@ Then you can run the command like `grt` in terminal.
   <img alt="static_exp" src="example/static_disp/disp_exp.png" width="200">
 </p>
 
+<p align='center'>
+  <img alt="imag_G" src="example/view_integ_stats/view_stats.png" width="600">
+</p>
+
+<p align='center'>
+  <img alt="imag_G" src="example/kernel_freq_response/imag_G.png" width="500">
+</p>
+
+
+
 # Contact
 If you have any questions or suggestions, feel free to reach out:
 - **Email**: [email protected]

example/kernel_freq_response/imag_G.png

@@ -0,0 +1,5 @@
+0.01    1.5   0.18   1.78   1e4  1e4
+0.01    1.7   0.35   1.85   1e4  1e4
+0.01    1.6   0.25   1.80   1e4  1e4
+0.01    2.0   0.6    1.94   1e4  1e4
+0.0     5.8   3.46   2.72   1e4  1e4      

example/kernel_freq_response/mod1

@@ -0,0 +1,35 @@
+# -----------------------------------------------------------------
+# BEGIN GRN
+import numpy as np
+import matplotlib.pyplot as plt
+from typing import Union
+import pygrt 
+
+modarr = np.loadtxt("mod1")
+
+pymod = pygrt.PyModel1D(modarr, depsrc=0.01, deprcv=0.0)
+
+_ = pymod.compute_grn(distarr=[1], nt=500, dt=0.02, vmin_ref=0.1, Length=20, statsfile="pygrtstats")
+
+vels = np.arange(0.1, 0.6, 0.001)
+
+kerDct = pygrt.utils.read_kernels_freqs("pygrtstats", vels)
+
+
+Type = "HF_v1"
+vels = kerDct['_vels']
+freqs = kerDct['_freqs']
+data = kerDct[Type].copy()
+data[...] = data/np.max(np.abs(data), axis=1)[:,None]
+fig, ax = plt.subplots(1, 1)
+pcm = ax.pcolormesh(freqs, vels, np.abs(np.imag(data)).T, vmin=0, vmax=1, shading='nearest')
+fig.colorbar(pcm)
+
+ax.set_xlim([0.5, 25])
+ax.set_ylim([0.1, 0.6])
+ax.set_xlabel("Frequency (km)")
+ax.set_ylabel("Velocity (m/s)")
+ax.set_title(f"Im[{Type}]")
+
+fig.tight_layout()
+fig.savefig("imag_G.png", dpi=100)

example/kernel_freq_response/run.py

@@ -0,0 +1,3 @@
+#!/bin/bash 
+
+grt -Mmod1 -D0.01/0 -N500/0.02/0.8 -OGRN -R1 -V0.1 -S-1 -L20

example/kernel_freq_response/run.sh

@@ -11,4 +11,6 @@
 
 data1, data2, ptamdata, dist = pygrt.utils.read_statsfile_ptam("stgrtstats/halfspace2_0.1_0/PTAM_0090_*/PTAM")
 fig, ax = pygrt.utils.plot_statsdata_ptam(data1, data2, ptamdata, dist, "DC", "2", "2")
+# fig.tight_layout()
+# fig.savefig("view_stats.png", dpi=100, bbox_inches='tight')
 plt.show()