From 5f3e47bcaaecfb984c2df56b6c2655e00c6d884b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ond=C5=99ej=20Hru=C5=A1ka?= <ondra@ondrovo.com>
Date: Mon, 23 Apr 2018 21:57:13 +0200
Subject: [PATCH] lightnings demo

---
 .gitignore             |  2 ++
 demo_adc_lightnings.py | 41 +++++++++++++++++++++++++++++++++++++++++
 show_nparray.py        | 15 +++++++++++++++
 test_adc_trig.py       | 32 --------------------------------
 4 files changed, 58 insertions(+), 32 deletions(-)
 create mode 100644 demo_adc_lightnings.py
 create mode 100755 show_nparray.py
 delete mode 100644 test_adc_trig.py

diff --git a/.gitignore b/.gitignore
index 3e75939..6e712b7 100644
--- a/.gitignore
+++ b/.gitignore
@@ -7,6 +7,8 @@ __javascript__/
 
 *.wav
 .idea/
+*.npy
+*.npz
 
 # C extensions
 *.so
diff --git a/demo_adc_lightnings.py b/demo_adc_lightnings.py
new file mode 100644
index 0000000..58a2f51
--- /dev/null
+++ b/demo_adc_lightnings.py
@@ -0,0 +1,41 @@
+#!/bin/env python3
+import time
+
+import gex
+import numpy as np
+from matplotlib import pyplot as plt
+import datetime
+
+from scipy.io import wavfile
+
+# ADC channel 1 -> 100n -> o -> long wire (antenna)
+#                          |
+#                          '-> 10k -> GND
+
+def capture(tr):
+    now=datetime.datetime.now()
+    now.isoformat()
+    data = tr.data
+    print("Capture! ")
+    print(data)
+    np.save("lightning-%s"%now.isoformat(), data)
+
+with gex.Client(gex.TrxRawUSB()) as client:
+    adc = gex.ADC(client, 'adc')
+
+    adc.on_trigger(capture)
+    adc.setup_trigger(1,
+                      level=500,
+                      count=1000,
+                      edge='rising',
+                      pretrigger=250,
+                      holdoff=500,
+                      auto=True)
+
+    adc.arm()
+
+    sec = 0
+    while True:
+        print('%d s' % sec)
+        sec += 1
+        time.sleep(1)
diff --git a/show_nparray.py b/show_nparray.py
new file mode 100755
index 0000000..d7f4774
--- /dev/null
+++ b/show_nparray.py
@@ -0,0 +1,15 @@
+#!/bin/env python3
+import time
+
+import sys
+
+import gex
+import numpy as np
+from matplotlib import pyplot as plt
+import datetime
+
+from scipy.io import wavfile
+
+data = np.load(sys.argv[1])
+plt.plot(data, 'r-', lw=1)
+plt.show()
diff --git a/test_adc_trig.py b/test_adc_trig.py
deleted file mode 100644
index 45a4793..0000000
--- a/test_adc_trig.py
+++ /dev/null
@@ -1,32 +0,0 @@
-#!/bin/env python3
-import time
-
-import gex
-import numpy as np
-from matplotlib import pyplot as plt
-
-from scipy.io import wavfile
-
-# def show(tr):
-#     data = tr.data
-#     data = np.add(data / 4096, -0.5)
-#     plt.plot(data, 'r-', lw=1)
-#     plt.show()
-
-with gex.Client(gex.TrxRawUSB()) as client:
-    adc = gex.ADC(client, 'adc')
-
-    adc.on_trigger(lambda tr: print(tr.data))
-    adc.setup_trigger(1,
-                      level=3500,
-                      count=500,
-                      edge='rising',
-                      pretrigger=100,
-                      holdoff=100,
-                      auto=True)
-
-    adc.arm()
-
-    while True:
-        print('tick')
-        time.sleep(1)