from artiq.experiment import *


class Grabber(EnvExperiment):
    def build(self): 
        self.setattr_device("core")
        self.grabber = self.get_device("grabber")
    
    @kernel 
    def run(self):
        self.core.break_realtime()
        delay(100*us) 
        # setup ROI boundaries 
        grabber.setup_roi(0, 0, 0, 2, 2) 
        grabber.setup_roi(1, 0, 0, 2048, 2048)
        # enable through bitwise mask 
        mask = 0b11 
        grabber.gate_roi(mask)

        # trigger the camera
        
        # retrieves data from enabled ROIs 
        n = [0] * 2 
        grabber.input_mu(n)
        # disable ROIs 
        self.core.break_realtime()
        grabber.gate_roi(0)