Ulm-IQO / qudi

                    time.sleep(0.25)

        return self.get_analog_level()

    def get_digital_level(self, low=None, high=None):

        """ Retrieve the digital low and high level of the provided channels.

        @param list low: optional, if a specific low value (in Volt) of a

                         channel is desired.

        @param list high: optional, if a specific high value (in Volt) of a

                          channel is desired.

        @return: (dict, dict): tuple of two dicts, with keys being the channel

                               number and items being the values for those

                               channels. Both low and high value of a channel is

                               denoted in (absolute) Voltage.

        Note: Do not return a saved low and/or high value but instead retrieve

              the current low and/or high value directly from the device.

        If no entries provided then the levels of all channels where simply

        returned. If no digital channels provided, return just an empty dict.

        Example of a possible input:

            low = ['d_ch1', 'd_ch4']

        to obtain the low voltage values of digital channel 1 an 4. A possible

        answer might be

            {'d_ch1': -0.5, 'd_ch4': 2.0} {}

        since no high request was performed.

        The major difference to analog signals is that digital signals are

        either ON or OFF, whereas analog channels have a varying amplitude

        range. In contrast to analog output levels, digital output levels are

        defined by a voltage, which corresponds to the ON status and a voltage

        which corresponds to the OFF status (both denoted in (absolute) voltage)

        In general there is no bijective correspondence between

        (amplitude, offset) and (value high, value low)!

        """

        # TODO: Test with multiple channel AWG

        low_val = {}

        high_val = {}

        digital_channels = self._get_all_digital_channels()

        if low is None:

            low = digital_channels

        if high is None:

            high = digital_channels

        # get low marker levels

        for chnl in low:

            if chnl not in digital_channels:

                continue

            d_ch_number = int(chnl.rsplit('_ch', 1)[1])

            a_ch_number = (1 + d_ch_number) // 2

            marker_index = 2 - (d_ch_number % 2)

            low_val[chnl] = float(

                self.query('SOUR{0:d}:MARK{1:d}:VOLT:LOW?'.format(a_ch_number, marker_index)))

        # get high marker levels

        for chnl in high:

            if chnl not in digital_channels:

                continue

            d_ch_number = int(chnl.rsplit('_ch', 1)[1])

            a_ch_number = (1 + d_ch_number) // 2

            marker_index = 2 - (d_ch_number % 2)

            high_val[chnl] = float(

                self.query('SOUR{0:d}:MARK{1:d}:VOLT:HIGH?'.format(a_ch_number, marker_index)))

        return low_val, high_val

    def set_digital_level(self, low=None, high=None):

        """ Set low and/or high value of the provided digital channel.

                    time.sleep(0.1)

        return self.get_analog_level()

    def get_digital_level(self, low=None, high=None):

        """ Retrieve the digital low and high level of the provided/all channels.

        @param list low: optional, if the low value (in Volt) of a specific channel is desired.

        @param list high: optional, if the high value (in Volt) of a specific channel is desired.

        @return: (dict, dict): tuple of two dicts, with keys being the channel descriptor strings

                               (i.e. 'd_ch1', 'd_ch2') and items being the values for those

                               channels. Both low and high value of a channel is denoted in volts.

        Note: Do not return a saved low and/or high value but instead retrieve

              the current low and/or high value directly from the device.

        If nothing (or None) is passed then the levels of all channels are being returned.

        If no digital channels are present, return just an empty dict.

        Example of a possible input:

            low = ['d_ch1', 'd_ch4']

        to obtain the low voltage values of digital channel 1 an 4. A possible answer might be

            {'d_ch1': -0.5, 'd_ch4': 2.0} {'d_ch1': 1.0, 'd_ch2': 1.0, 'd_ch3': 1.0, 'd_ch4': 4.0}

        Since no high request was performed, the high values for ALL channels are returned (here 4).

        """

        low_val = {}

        high_val = {}

        digital_channels = self._get_all_digital_channels()

        if low is None:

            low = digital_channels

        if high is None:

            high = digital_channels

        # get low marker levels

        for chnl in low:

            if chnl not in digital_channels:

                continue

            d_ch_number = int(chnl.rsplit('_ch', 1)[1])

            a_ch_number = (1 + d_ch_number) // 2

            marker_index = 2 - (d_ch_number % 2)

            low_val[chnl] = float(

                self.query('SOUR{0:d}:MARK{1:d}:VOLT:LOW?'.format(a_ch_number, marker_index)))

        # get high marker levels

        for chnl in high:

            if chnl not in digital_channels:

                continue

            d_ch_number = int(chnl.rsplit('_ch', 1)[1])

            a_ch_number = (1 + d_ch_number) // 2

            marker_index = 2 - (d_ch_number % 2)

            high_val[chnl] = float(

                self.query('SOUR{0:d}:MARK{1:d}:VOLT:HIGH?'.format(a_ch_number, marker_index)))

        return low_val, high_val

    def set_digital_level(self, low=None, high=None):

        """ Set low and/or high value of the provided digital channel.