from artiq.language.units import *
from artiq.language.experiment import *

my_range = range

class CompilerTest(MPO):
	channels = "a b A B"

	def print_done(self):
		print("Done!")

	def set_some_slowdev(self, n):
		print("Slow device setting: {}".format(n))

	@kernel
	def run(self, n, t2):
		for i in my_range(n):
			self.set_some_slowdev(i)
			delay(100*ms)
			with parallel:
				with sequential:
					for j in my_range(3):
						self.a.pulse((j+1)*100*MHz, 20*us)
						self.b.pulse(100*MHz, t2)
				with sequential:
					self.A.pulse(100*MHz, 10*us)
					self.B.pulse(100*MHz, t2)
		self.print_done()

if __name__ == "__main__":
	from artiq.devices import core, dds_core

	coredev = core.Core()
	exp = CompilerTest(
		core=coredev,
		a=dds_core.DDS(coredev, 0, 0),
		b=dds_core.DDS(coredev, 1, 1),
		A=dds_core.DDS(coredev, 2, 2),
		B=dds_core.DDS(coredev, 3, 3)
	)
	exp.run(3, 100*us)