PXI 可编程电源曲线控制

#!/usr/bin/env python3
"""Run a CSV curve on a PXI programmable power supply with NI-DCPower.
Generated by the Instrument Data tools. Verify the resource and limits first.
"""
import argparse
import csv
import time
from pathlib import Path

import nidcpower

DEFAULT_RESOURCE = "PXI1Slot2/0"
DEFAULT_MODE = "voltage"
DEFAULT_REPEAT = 1
DEFAULT_VOLTAGE_LIMIT = 6
DEFAULT_CURRENT_LIMIT = 0.1


def load_curve(path: Path):
    with path.open("r", newline="", encoding="utf-8-sig") as handle:
        rows = list(csv.DictReader(handle))
    if len(rows) < 2:
        raise ValueError("CSV must contain at least two data rows")
    values = []
    times = []
    for row in rows:
        value_text = row.get("level") or row.get("voltage_v") or row.get("current_a") or row.get("value")
        if value_text in (None, ""):
            raise ValueError("CSV is missing level/voltage_v/current_a")
        values.append(float(value_text))
        time_text = row.get("time_s") or row.get("time")
        times.append(float(time_text) if time_text not in (None, "") else None)
    if any(value is None for value in times):
        delay = 0.001
        delays = [delay] * len(values)
    else:
        delays = []
        for index in range(len(values)):
            if index + 1 < len(times):
                delays.append(max(0.000001, times[index + 1] - times[index]))
            elif len(times) > 1:
                delays.append(max(0.000001, times[-1] - times[-2]))
            else:
                delays.append(0.001)
    return values, delays, sum(delays)


def main():
    parser = argparse.ArgumentParser(description="Run a CSV voltage/current curve on a PXI programmable power supply")
    parser.add_argument("--resource", default=DEFAULT_RESOURCE)
    parser.add_argument("--csv", type=Path, default=Path("ni_4112_curve.csv"))
    parser.add_argument("--mode", choices=("voltage", "current"), default=DEFAULT_MODE)
    parser.add_argument("--voltage-limit", type=float, default=DEFAULT_VOLTAGE_LIMIT)
    parser.add_argument("--current-limit", type=float, default=DEFAULT_CURRENT_LIMIT)
    parser.add_argument("--repeat", type=int, default=DEFAULT_REPEAT)
    parser.add_argument("--simulate", action="store_true", help="Use the NI driver simulator")
    args = parser.parse_args()
    values, delays, total_time = load_curve(args.csv)
    if args.repeat < 1:
        raise ValueError("--repeat must be at least 1")

    session_args = {"resource_name": args.resource}
    if args.simulate:
        session_args["options"] = "Simulate=1, DriverSetup=Model:4112"

    with nidcpower.Session(**session_args) as session:
        session.output_function = (nidcpower.OutputFunction.DC_VOLTAGE
                                   if args.mode == "voltage"
                                   else nidcpower.OutputFunction.DC_CURRENT)
        session.source_mode = nidcpower.SourceMode.SEQUENCE
        if args.mode == "voltage":
            session.voltage_level_autorange = True
            session.current_limit = args.current_limit
        else:
            session.current_level_autorange = True
            session.voltage_limit = args.voltage_limit
        session.set_sequence(values, delays)
        try:
            session.output_enabled = True
            for cycle in range(args.repeat):
                print(f"Running cycle {cycle + 1}/{args.repeat}: {len(values)} points")
                with session.initiate():
                    time.sleep(total_time)
        finally:
            session.output_enabled = False
            print("Output disabled")


if __name__ == "__main__":
    main()