from artiq import *


class SimpleSimulation(AutoContext):
    a = Device("dds")
    b = Device("dds")
    c = Device("dds")
    d = Device("dds")

    @kernel
    def run(self):
        with parallel:
            with sequential:
                self.a.pulse(100*MHz, 20*us)
                self.b.pulse(200*MHz, 20*us)
            with sequential:
                self.c.pulse(300*MHz, 10*us)
                self.d.pulse(400*MHz, 20*us)


def main():
    from artiq.sim import devices as sd
    from artiq.sim import time

    exp = SimpleSimulation(
        core=sd.Core(),
        a=sd.WaveOutput(name="a"),
        b=sd.WaveOutput(name="b"),
        c=sd.WaveOutput(name="c"),
        d=sd.WaveOutput(name="d"),
    )
    exp.run()
    print(time.manager.format_timeline())

if __name__ == "__main__":
    main()