import time

import pylink
import pylink.enums


if __name__ == "__main__":
    jl = pylink.JLink()
    jl.open()

    try:
        channels = jl.power_trace_get_channels()
        print(f"{channels=}")

        channel_caps = jl.power_trace_get_channel_capabilities(channels)
        print(f"{channel_caps=}")

        freq = jl.power_trace_configure(channels, channel_caps.max_sample_freq / 10, pylink.enums.JLinkPowerTraceRef.TIME, True)
        print(f"{freq=}")

        jl.power_trace_start()
        time.sleep(0.1)

        num_items = jl.power_trace_get_num_items()
        print(f"{num_items=}")

        items = jl.power_trace_read(num_items)
        print(f"{items=}")

        jl.power_trace_flush()
        jl.power_trace_stop()
    finally:
        jl.close()