from min_artiq import *
from numpy import int32
import string_attribute_issue337
import support_class_attr_issue102
import support_class_attr_issue102 as issue102
import string_attribute_issue337 as issue337

# gvar22: int32 = 22
@nac3
class Test:
    attr1: KernelInvariant[int32] = 2
    attr2: int32 = 4
    attr3: Kernel[int32]

    @kernel
    def __init__(self):
        self.attr3 = 8
    
    @kernel
    def run(self) -> int32:
        # global gvar22
        gvar22 = 10
        self.attr3 = 1
        return self.attr3

@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):
        # global gvar22
        # issue337.Demo().run()
        a = Test()

        a.run()
        Test.attr1
        # g = gvar22
        e = issue337.myTest()
        f = e.run()
        f = f + 1
        # g = gvar22
        b = issue102.fun2smn()
        c = string_attribute_issue337.fn1_dosmn()
        d = issue337.fn1_dosmn()
        
        # self.core.reset()
        # while True:
        #     with parallel:
        #         self.led0.pulse(100.*ms)
        #         self.led1.pulse(100.*ms)
        #     self.core.delay(100.*ms)


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