mirror of https://github.com/m-labs/artiq.git
grabber: port to NAC3
This commit is contained in:
parent
bf8e188868
commit
2e4233274f
|
@ -1,19 +1,23 @@
|
||||||
from numpy import int32, int64
|
from numpy import int32, int64
|
||||||
|
|
||||||
from artiq.language.core import *
|
from artiq.language.core import *
|
||||||
from artiq.language.types import *
|
|
||||||
from artiq.coredevice.rtio import rtio_output, rtio_input_data
|
from artiq.coredevice.rtio import rtio_output, rtio_input_data
|
||||||
|
from artiq.coredevice.core import Core
|
||||||
|
|
||||||
|
|
||||||
|
@nac3
|
||||||
class OutOfSyncException(Exception):
|
class OutOfSyncException(Exception):
|
||||||
"""Raised when an incorrect number of ROI engine outputs has been
|
"""Raised when an incorrect number of ROI engine outputs has been
|
||||||
retrieved from the RTIO input FIFO."""
|
retrieved from the RTIO input FIFO."""
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@nac3
|
||||||
class Grabber:
|
class Grabber:
|
||||||
"""Driver for the Grabber camera interface."""
|
"""Driver for the Grabber camera interface."""
|
||||||
kernel_invariants = {"core", "channel_base", "sentinel"}
|
core: KernelInvariant[Core]
|
||||||
|
channel_base: KernelInvariant[int32]
|
||||||
|
sentinel: KernelInvariant[int32]
|
||||||
|
|
||||||
def __init__(self, dmgr, channel_base, res_width=12, count_shift=0,
|
def __init__(self, dmgr, channel_base, res_width=12, count_shift=0,
|
||||||
core_device="core"):
|
core_device="core"):
|
||||||
|
@ -26,7 +30,7 @@ class Grabber:
|
||||||
self.sentinel = int32(int64(2**count_width))
|
self.sentinel = int32(int64(2**count_width))
|
||||||
|
|
||||||
@kernel
|
@kernel
|
||||||
def setup_roi(self, n, x0, y0, x1, y1):
|
def setup_roi(self, n: int32, x0: int32, y0: int32, x1: int32, y1: int32):
|
||||||
"""
|
"""
|
||||||
Defines the coordinates of a ROI.
|
Defines the coordinates of a ROI.
|
||||||
|
|
||||||
|
@ -50,7 +54,7 @@ class Grabber:
|
||||||
delay_mu(c)
|
delay_mu(c)
|
||||||
|
|
||||||
@kernel
|
@kernel
|
||||||
def gate_roi(self, mask):
|
def gate_roi(self, mask: int32):
|
||||||
"""
|
"""
|
||||||
Defines which ROI engines produce input events.
|
Defines which ROI engines produce input events.
|
||||||
|
|
||||||
|
@ -70,15 +74,15 @@ class Grabber:
|
||||||
rtio_output((self.channel_base + 1) << 8, mask)
|
rtio_output((self.channel_base + 1) << 8, mask)
|
||||||
|
|
||||||
@kernel
|
@kernel
|
||||||
def gate_roi_pulse(self, mask, dt):
|
def gate_roi_pulse(self, mask: int32, dt: float):
|
||||||
"""Sets a temporary mask for the specified duration (in seconds), before
|
"""Sets a temporary mask for the specified duration (in seconds), before
|
||||||
disabling all ROI engines."""
|
disabling all ROI engines."""
|
||||||
self.gate_roi(mask)
|
self.gate_roi(mask)
|
||||||
delay(dt)
|
self.core.delay(dt)
|
||||||
self.gate_roi(0)
|
self.gate_roi(0)
|
||||||
|
|
||||||
@kernel
|
@kernel
|
||||||
def input_mu(self, data):
|
def input_mu(self, data: list[int32]):
|
||||||
"""
|
"""
|
||||||
Retrieves the accumulated values for one frame from the ROI engines.
|
Retrieves the accumulated values for one frame from the ROI engines.
|
||||||
Blocks until values are available.
|
Blocks until values are available.
|
||||||
|
@ -96,10 +100,10 @@ class Grabber:
|
||||||
|
|
||||||
sentinel = rtio_input_data(channel)
|
sentinel = rtio_input_data(channel)
|
||||||
if sentinel != self.sentinel:
|
if sentinel != self.sentinel:
|
||||||
raise OutOfSyncException
|
raise OutOfSyncException()
|
||||||
|
|
||||||
for i in range(len(data)):
|
for i in range(len(data)):
|
||||||
roi_output = rtio_input_data(channel)
|
roi_output = rtio_input_data(channel)
|
||||||
if roi_output == self.sentinel:
|
if roi_output == self.sentinel:
|
||||||
raise OutOfSyncException
|
raise OutOfSyncException()
|
||||||
data[i] = roi_output
|
data[i] = roi_output
|
||||||
|
|
|
@ -32,6 +32,10 @@
|
||||||
"bank_direction_low": "input",
|
"bank_direction_low": "input",
|
||||||
"bank_direction_high": "output",
|
"bank_direction_high": "output",
|
||||||
"edge_counter": true
|
"edge_counter": true
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"type": "grabber",
|
||||||
|
"ports": [6]
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
|
|
@ -8,6 +8,7 @@ from artiq.coredevice.urukul import CPLD as UrukulCPLD
|
||||||
from artiq.coredevice.ad9912 import AD9912
|
from artiq.coredevice.ad9912 import AD9912
|
||||||
from artiq.coredevice.sampler import Sampler
|
from artiq.coredevice.sampler import Sampler
|
||||||
from artiq.coredevice.edge_counter import EdgeCounter
|
from artiq.coredevice.edge_counter import EdgeCounter
|
||||||
|
from artiq.coredevice.grabber import Grabber
|
||||||
|
|
||||||
|
|
||||||
@nac3
|
@nac3
|
||||||
|
@ -21,6 +22,7 @@ class NAC3Devices(EnvExperiment):
|
||||||
urukul0_ch0: KernelInvariant[AD9912]
|
urukul0_ch0: KernelInvariant[AD9912]
|
||||||
sampler0: KernelInvariant[Sampler]
|
sampler0: KernelInvariant[Sampler]
|
||||||
ttl0_counter: KernelInvariant[EdgeCounter]
|
ttl0_counter: KernelInvariant[EdgeCounter]
|
||||||
|
grabber0: KernelInvariant[Grabber]
|
||||||
|
|
||||||
def build(self):
|
def build(self):
|
||||||
self.setattr_device("core")
|
self.setattr_device("core")
|
||||||
|
@ -32,6 +34,7 @@ class NAC3Devices(EnvExperiment):
|
||||||
self.setattr_device("urukul0_ch0")
|
self.setattr_device("urukul0_ch0")
|
||||||
self.setattr_device("sampler0")
|
self.setattr_device("sampler0")
|
||||||
self.setattr_device("ttl0_counter")
|
self.setattr_device("ttl0_counter")
|
||||||
|
self.setattr_device("grabber0")
|
||||||
|
|
||||||
@kernel
|
@kernel
|
||||||
def run(self):
|
def run(self):
|
||||||
|
|
Loading…
Reference in New Issue