artiq/examples/coredev_test.py

from artiq.language.experiment import MPO, kernel
from artiq.devices import corecom_serial, runtime, core, gpio_core

class CompilerTest(MPO):
	channels = "led"

	def output(self, n):
		print("Received: "+str(n))

	def get_max(self):
		return int(input("Maximum: "))

	@kernel
	def run(self):
		self.led.set(1)
		x = 1
		m = self.get_max()
		while x < m:
			d = 2
			prime = 1
			while d*d <= x:
				if x % d == 0:
					prime = 0
				d = d + 1
			if prime == 1:
				self.output(x)
			x = x + 1
		self.led.set(0)

if __name__ == "__main__":
	with corecom_serial.CoreCom() as com:
		coredev = core.Core(runtime.Environment(), com)
		exp = CompilerTest(
			core=coredev,
			led=gpio_core.GPIOOut(coredev, 0)
		)
		exp.run()