test

Author: juleluj

examples/1-import-data.ipynb

@@ -34,18 +34,19 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [
     {
-     "data": {
-      "text/plain": [
-       "<lightguide.blast.Blast at 0x7fde7e1681c0>"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
+     "ename": "ModuleNotFoundError",
+     "evalue": "No module named 'lightguide.blast'",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[1], line 2\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39mpathlib\u001b[39;00m \u001b[39mimport\u001b[39;00m Path\n\u001b[0;32m----> 2\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39mlightguide\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mblast\u001b[39;00m \u001b[39mimport\u001b[39;00m Blast\n\u001b[1;32m      3\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39mlightguide\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mutils\u001b[39;00m \u001b[39mimport\u001b[39;00m download_http\n\u001b[1;32m      5\u001b[0m file \u001b[39m=\u001b[39m Path(\u001b[39m\"\u001b[39m\u001b[39m/tmp/vsp-data.mseed\u001b[39m\u001b[39m\"\u001b[39m)\n",
+      "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'lightguide.blast'"
+     ]
     }
    ],
    "source": [
@@ -80,7 +81,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.2"
+   "version": "3.8.10"
   },
   "orig_nbformat": 4
  },

examples/2-data-processing.ipynb

@@ -132,7 +132,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.2"
+   "version": "3.8.10"
   },
   "orig_nbformat": 4,
   "vscode": {

examples/3-event-analysis.ipynb

@@ -0,0 +1,35 @@
+# %config InlineBackend.figure_format = 'jpg'
+import matplotlib.pyplot as plt
+from datetime import timedelta
+from lightguide.client import ExampleData
+from lightguide.blast import Blast
+
+
+# # This will download example data
+# blast = ExampleData.vsp_shot()
+
+# # Pre-Processing
+# blast = blast.trim_channels(begin=120, end=550)
+# blast = blast.trim_time(begin=1.5, end=2.5)
+# blast.lowpass(70.0)
+# blast.afk_filter()
+
+
+# pick_time = blast.start_time + timedelta(seconds=0.573)
+
+# pick_channel = 350
+
+# # Follow the phase from initial pick
+# pick_channels, pick_times, pick_correlation = blast.follow_phase(
+#     pick_time, pick_channel, window_size=10, max_shift=100, threshold=1e-2
+# )
+
+# fig = plt.figure(figsize=(12, 6))
+# ax = fig.gca()
+
+# blast.plot(axes=ax, cmap="binary", show_date=True)
+# ax.scatter(pick_channels, pick_times, s=4, c="fuchsia", alpha=0.6)
+
+# # Plot initial pick
+# ax.scatter(pick_channel, pick_time, marker="x", c="limegreen")
+# plt.show()

examples/my_test.py

@@ -0,0 +1,47 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "#from lightguide.utils import download_numpy, ExampleData\n",
+    "# from lightguide.blast\n",
+    "#from lightguide.blast import Blast\n",
+    "#import lightguide.blast\n",
+    "\n",
+    "from collections import deque"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.8.10"
+  },
+  "orig_nbformat": 4
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

examples/my_tests.ipynb

@@ -330,6 +330,7 @@ def follow_phase(
         window_size: int | tuple[int, int] = 50,
         threshold: float = 5e-1,
         max_shift: int = 20,
+        stack_traces: int = 1,
     ) -> tuple[np.ndarray, list[datetime], np.ndarray]:
         """Follow a phase pick through a Blast.
 
@@ -375,6 +376,10 @@ def follow_phase(
         def prepare_template(data: np.ndarray) -> np.ndarray:
             return data * template_taper
 
+        # def stack_n_straces(data: np.ndarray,stack_traces) -> np.ndarray:
+
+        #     return stacked_data
+
         def correlate(data: np.ndarray, direction: Literal[1, -1] = 1) -> None:
             template = root_template.copy()
             index = root_idx
@@ -416,6 +421,7 @@ def correlate(data: np.ndarray, direction: Literal[1, -1] = 1) -> None:
 
         pick_channels = np.array(pick_channels) + self.start_channel
         pick_correlations = np.array(pick_correlations)
+        print("aa")
 
         return pick_channels, pick_times, pick_correlations