diff options
Diffstat (limited to 'bin')
| -rwxr-xr-x | bin/korad-logger | 85 |
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) |