summaryrefslogtreecommitdiff
path: root/bin
diff options
context:
space:
mode:
authorDaniel Friesel <derf@finalrewind.org>2021-11-15 22:04:06 +0100
committerDaniel Friesel <derf@finalrewind.org>2021-11-15 22:04:06 +0100
commit74f7db62c27b97483c6b0b4a8b6abd48e4aabce3 (patch)
treeef9d68c8a421bc7c88d354094ec106213bdffa5d /bin
parent273019cac8cb721ad7a04a38a3da9ddb2de227b5 (diff)
add optional live plots
Diffstat (limited to 'bin')
-rwxr-xr-xbin/korad-logger85
1 files changed, 77 insertions, 8 deletions
diff --git a/bin/korad-logger b/bin/korad-logger
index 93a4005..e9d97f5 100755
--- a/bin/korad-logger
+++ b/bin/korad-logger
@@ -163,12 +163,15 @@ class KA320:
self.worker = serial.threaded.ReaderThread(self.ser, self.reader)
self.worker.start()
- def rw(self, cmd, num_chars, exact=False, binary=False):
+ def rw(self, cmd, num_chars, exact=False, binary=False, plot=None):
self.reader.expect(num_chars, binary=binary)
self.ser.write(cmd)
timeout = 20
while not self.reader.read_complete and not timeout == 0:
- time.sleep(0.02)
+ if plot is not None:
+ plot.pause(0.02)
+ else:
+ time.sleep(0.02)
timeout -= 1
if exact:
return self.reader.get_expected_line()
@@ -212,15 +215,15 @@ class KA320:
def get_max_current(self):
return float(self.rw(f"ISET{self.channel:d}?".encode(), 5, True))
- def get_voltage(self):
+ def get_voltage(self, plot=None):
try:
- return float(self.rw(f"VOUT{self.channel:d}?".encode(), 5, True))
+ return float(self.rw(f"VOUT{self.channel:d}?".encode(), 5, True, plot=plot))
except TypeError:
return None
- def get_current(self):
+ def get_current(self, plot=None):
try:
- return float(self.rw(f"IOUT{self.channel:d}?".encode(), 5, True))
+ return float(self.rw(f"IOUT{self.channel:d}?".encode(), 5, True, plot=plot))
except TypeError:
return None
@@ -255,9 +258,13 @@ def measure_data(
on_off=False,
log_voltage=True,
log_current=True,
+ live_view=False,
+ live_history=0,
):
global terminate_measurement
+ plot = None
+
voltage_start, voltage_stop, voltage_step = voltage_range
current_start, current_stop, current_step = current_range
last_range_step = 0
@@ -298,6 +305,33 @@ def measure_data(
print("Enabling over-current protection")
korad.ocp(True)
+ if live_view:
+ import matplotlib.pyplot as plt
+
+ plt.style.use("dark_background")
+
+ timestamps = list()
+ voltages = list()
+ currents = list()
+ max_y = 0
+ plt.ion()
+ if log_current:
+ (currentline,) = plt.plot(
+ timestamps, currents, "r-", label="Current [A]", markersize=1
+ )
+ plt.ylabel("Current [A]")
+ if log_voltage:
+ (voltageline,) = plt.plot(
+ timestamps, voltages, "b-", label="Voltage [V]", markersize=1
+ )
+ plt.ylabel("Voltage [V]")
+ if log_current and log_voltage:
+ plt.legend(handles=[voltageline, currentline])
+ plt.ylabel("")
+ plt.xlabel("Time [s]")
+ plt.show()
+ plot = plt
+
if max_voltage is not None or max_current is not None or on_off:
print("Turning on outputs")
korad.set_output(True)
@@ -314,13 +348,34 @@ def measure_data(
while not terminate_measurement:
ts = time.time()
if log_current:
- current = korad.get_current()
+ current = korad.get_current(plot)
else:
current = None
if log_voltage:
- voltage = korad.get_voltage()
+ voltage = korad.get_voltage(plot)
else:
voltage = None
+ if live_view:
+ timestamps.append(ts - start_ts)
+ voltages.append(voltage)
+ currents.append(current)
+ if live_history:
+ timestamps = timestamps[-live_history:]
+ voltages = voltages[-live_history:]
+ currents = currents[-live_history:]
+ if len(timestamps) > 1:
+ plt.xlim([timestamps[0], timestamps[-1]])
+ if log_voltage:
+ voltageline.set_data([timestamps, voltages])
+ if log_current:
+ currentline.set_data([timestamps, currents])
+ if log_current and current is not None and current > max_y:
+ max_y = current
+ plt.ylim([0, max_y + 0.1])
+ if log_voltage and voltage is not None and voltage > max_y:
+ max_y = voltage
+ plt.ylim([0, max_y + 0.1])
+ plt.show()
if voltage is not None and current is not None:
print(f"{ts:.3f} {voltage:5.2f} {current:5.3f}", file=output_handle)
elif voltage is not None:
@@ -578,6 +633,18 @@ def main():
help="Limit analysis to the first N seconds of data",
)
parser.add_argument(
+ "--live-view",
+ action="store_true",
+ help="Plot live voltage/current data while the measurement is running. May decreases the sample rate.",
+ )
+ parser.add_argument(
+ "--live-history",
+ type=int,
+ metavar="N",
+ default=300,
+ help="Show up to N past samples in the live view. Less history → lower live view overhead → higher sample rate. Set to 0 for unlimited history.",
+ )
+ parser.add_argument(
"--plot",
metavar="UNIT",
choices=["U", "I", "P", "UI", "UP", "IU", "IP"],
@@ -620,6 +687,8 @@ def main():
on_off=args.on_off,
log_voltage=not args.current_only,
log_current=not args.voltage_only,
+ live_view=args.live_view,
+ live_history=args.live_history,
)
data = parse_data(log_data, skip=args.skip, limit=args.limit)