Unverified Commit 530110b8 authored by Daniel Friesel's avatar Daniel Friesel
Browse files

add optional dark mode

parent 58fc39a8

bin/korad-logger

+12 −1
Original line number Diff line number Diff line
@@ -52,6 +52,8 @@ import time 


terminate_measurement = False



matplotlib_theme = "fast"





def running_mean(x: np.ndarray, N: int) -> np.ndarray:

    """
@@ -308,7 +310,7 @@ def measure_data( 
    if live_view:

        import matplotlib.pyplot as plt



        plt.style.use("dark_background")

        plt.style.use(matplotlib_theme)



        timestamps = list()

        voltages = list()
@@ -427,6 +429,8 @@ def measure_data( 
def plot_data(data, mode):

    import matplotlib.pyplot as plt



    plt.style.use(matplotlib_theme)



    if mode == "U":

        (datahandle,) = plt.plot(data[:, 0], data[:, 1], "b-", label="U", markersize=1)

        (meanhandle,) = plt.plot(
@@ -650,6 +654,9 @@ def main(): 
        choices=["U", "I", "P", "UI", "UP", "IU", "IP"],

        help="Plot voltage / current / power over time or voltage vs current / current vs voltage",

    )

    parser.add_argument(

        "--dark-mode", action="store_true", help="Show plots on a dark background"

    )

    parser.add_argument(

        "duration", type=int, nargs="?", help="Measurement duration in seconds"

    )
@@ -660,6 +667,10 @@ def main(): 
        print("Either --load or duration must be specified", file=sys.stderr)

        sys.exit(1)



    if args.dark_mode:

        global matplotlib_theme

        matplotlib_theme = "dark_background"



    current_range = parse_range(args.current_range)

    voltage_range = parse_range(args.voltage_range)