updating output

pyrelacs/repros/calbi.py

@@ -1,15 +1,18 @@
 import ctypes
+import faulthandler
+
 import time
 
 import uldaq
 from IPython import embed
-from pyrelacs.repros.repos import MccDac
-from pyrelacs.repros.attcs3310 import Attenuator
-from pyrelacs.util.logging import config_logging
 import numpy as np
 import matplotlib.pyplot as plt
 
+from pyrelacs.repros.mccdac import MccDac
+from pyrelacs.util.logging import config_logging
+
 log = config_logging()
+faulthandler.enable()
 
 
 class Calibration(MccDac):
@@ -22,7 +25,7 @@ class Calibration(MccDac):
         data = AMPLITUDE * np.sin(2 * np.pi * SINFREQ * t)
         # sending stimulus
 
-        stim, ao_device = self.write_analog_dac(
+        self.write_analog(
             data, [0, 0], SAMPLERATE, ScanOption=uldaq.ScanOption.EXTTRIGGER
         )
 
@@ -32,7 +35,7 @@ class Calibration(MccDac):
 
         self.diggital_trigger()
         try:
-            ao_device.scan_wait(uldaq.WaitType.WAIT_UNTIL_DONE, 15)
+            self.ao_device.scan_wait(uldaq.WaitType.WAIT_UNTIL_DONE, 15)
         except uldaq.ul_exception.ULException:
             log.debug("Operation timed out")
             # reset the diggital trigger
@@ -40,11 +43,56 @@ class Calibration(MccDac):
             time.sleep(1)
             self.set_analog_to_zero()
 
+    def check_amplitude(self):
+        # write to ananlog 1
+        t = np.arange(0, DURATION, 1 / SAMPLERATE)
+        data = AMPLITUDE * np.sin(2 * np.pi * SINFREQ * t)
+
+        # sending stimulus
+        log.debug(f"{data}, {data.shape}")
+        embed()
+        exit()
+        # self.set_attenuation_level(db_channel1=0.0, db_channel2=0.0)
+        # self.set_analog_to_zero()
+        # time.sleep(1)
+        log.debug(self.ao_device)
+
+        self.write_analog(
+            data,
+            [1, 1],
+            SAMPLERATE,
+            ScanOption=uldaq.ScanOption.EXTTRIGGER,
+        )
+        # data_channel_one = self.read_analog(
+        #     [0, 0], DURATION, SAMPLERATE, ScanOption=uldaq.ScanOption.EXTTRIGGER
+        # )
+        log.debug(self.ao_device)
+        time.sleep(1)
+
+        self.diggital_trigger()
+
+        try:
+            self.ao_device.scan_wait(uldaq.WaitType.WAIT_UNTIL_DONE, 15)
+            self.write_bit(channel=0, bit=0)
+            time.sleep(1)
+            self.set_analog_to_zero()
+        except uldaq.ul_exception.ULException:
+            log.debug("Operation timed out")
+            # reset the diggital trigger
+            self.write_bit(channel=0, bit=0)
+            time.sleep(1)
+            self.set_analog_to_zero()
+            self.disconnect_dac()
+        embed()
+        exit()
+
 
 if __name__ == "__main__":
     SAMPLERATE = 40_000.0
-    DURATION = 10
+    DURATION = 5
     AMPLITUDE = 1
-    SINFREQ = 100
+    SINFREQ = 10
+
     daq_input = Calibration()
-    daq_input.run_calibration()
+    # daq_input.check_attenuator()
+    daq_input.check_amplitude()