from min_artiq import *


@nac3
class Demo:
    core: KernelInvariant[Core]
    led0: KernelInvariant[TTLOut]
    led1: KernelInvariant[TTLOut]

    def __init__(self):
        self.core = Core()
        self.led0 = TTLOut(self.core, 18)
        self.led1 = TTLOut(self.core, 19)

    @kernel
    def run(self):
        a = np_array([[1., 2.], [3., 4.]])
        b = try_invert_to(a)
        if b:
            # self.core.reset()
            # while True:
            #     with parallel:
            #         self.led0.pulse(100.*ms)
            #         self.led1.pulse(100.*ms)
            #     self.core.delay(100.*ms)
            v = try_invert_to(np_identity(2))
            if v:
                while True:
                    with parallel:
                        self.led0.pulse(100.*ms)
                        self.led1.pulse(100.*ms)
                    self.core.delay(100.*ms)
            


if __name__ == "__main__":
    Demo().run()