nsqdriver 0.2.6__py3-none-any.whl → 0.3.0__py3-none-any.whl

nsqdriver/wrapper/BD_NSMCI.py

@@ -0,0 +1,310 @@
+import time
+from typing import Dict, Any, List
+
+import numpy as np
+
+from nsqdriver import MCIDriver, QSYNCDriver
+
+
+try:
+    import Instrument
+except ImportError as e:
+    class Instrument(object):
+        ...
+
+
+class NSInstrument(Instrument):
+    """ Instrument for GuoDun devices """
+    ParameterMC = {
+        'OUTSrate': 8e9,      # 配置OUT通道采样率
+        'OUTMix': 2,          # 设置OUT通道增强第二奈奎斯特区信号
+        'INZone': 'zone3'     # 设置IN通道输入为Zone4，即要IN通道要采集的信号范围为6~8G
+    }
+    ParameterZ = {
+        'OUTSrate': 2e9,      # 配置OUT通道采样率
+        'OUTMix': 1,          # 设置OUT通道增强第一奈奎斯特区信号
+        'KeepAmp': 2          # 开启Z通道的保持电平模式，2为保持drive信号电平
+    }
+    ParameterQSYNC = {
+        'RefClock': 'in'      # MC112的100M参考信号来源，机箱上的100M IN没有接的情况下，选择为in，反之选out
+    }
+
+    def __init__(self):
+        """ Initialization """
+        super(NSInstrument, self).__init__()
+
+        # 设备驱动
+        self.driver_mc: MCIDriver = MCIDriver('192.168.111.152')
+        self.driver_z: MCIDriver = MCIDriver('192.168.111.132')
+        self.qsync: QSYNCDriver = QSYNCDriver('192.168.111.152')
+
+        # 参数
+        self.waves = dict()
+        self.roFreq = dict()
+        self.roQubits = dict()
+        self.roLength = dict()
+        self.roDADelay = dict()
+        self.zpdc = dict()
+        self.group = dict()
+        self.roLines = dict()
+        self.connectDict = dict()
+        self.deviceDict = dict()
+        self.qubitROChannels = dict()
+        self.integrationDelay = dict()
+        self.inputRange = dict()
+        self.outputRange = dict()
+        self.devices = {
+            'awg': self.driver_mc,
+            'dc': self.driver_z,
+            'qa': self.driver_mc,
+        }  # type: Dict[str, Any]
+        self.shots = 1000
+        self.devicesList = {
+            'awg': [],
+            'dc': [],
+            'qa': []
+        }  # type: Dict[str, Any]
+
+    def connectDevices(self, *args, **kwargs):
+        """ Connect and initialize devices """
+        if 'devices' not in kwargs.keys():
+            raise ValueError('Parameter `devices` does not exist in connectDevices().')
+        if 'connect' not in kwargs.keys():
+            raise ValueError('Parameter `connect` does not exist in connectDevices().')
+        if 'roLines' not in kwargs.keys():
+            raise ValueError('Parameter `roLines` does not exist in connectDevices().')
+
+        self.roLines = kwargs['roLines']
+        """
+        The readout line dict. Example:
+        {"r1":["q01","q02","q03","q04","q05"], "r2": ["q06","q07","q08","q09","q10"], ...}
+        """
+
+        self.deviceDict = kwargs['devices']
+        """
+        The dict for all devices. Example:
+        {
+            "XY":["mc_1", ...],
+            "Z":["z_1", ...],
+            "ZDC":["z_1", ...],
+            "RO":["out01", "out09", ...],
+            "QA":["in01", "in09", ...]
+        }
+        """
+
+        self.connectDict = kwargs['connect']
+        """
+        The mapping between device channels and logical channels. Example:
+        {
+            "QA": {"r1QA": ["in01", "range(20)"], "r2QA": ["in09", "range(20)"], ...},
+            "RO": {"r1RO": ["out01", "range(20)"], "r2RO": ["out09", "range(20)"], ...},
+            "RO_qubits": {"q01RO": "r1RO", "q02RO": "r1RO", "q03RO": "r1RO", ...},
+            "RO_trig": {"r1RO": "AWG5M1", "r2RO": "AWG5M1", ...},
+            "XY": {"q03XY": ["mc_1", "out03"], "q04XY": ["mc_1", "out04"], ...},
+            "Z": {"q03Z": ["z_1", "out01"], "q04Z": ["z_1", "out02"]}
+        }
+
+        """
+
+        # Parse readout channels
+        for _line, _qubits in self.roLines.items():
+            ro_channels = [i for i in eval(self.connectDict['RO'][f'{_line}RO'][1])]
+            if len(ro_channels) != len(_qubits):
+                raise RuntimeError(f'Qubit number in {_line} does not equal to '
+                                   f'the number of channels ({ro_channels}).')
+            for _q, _c in zip(self.roLines[_line], ro_channels):
+                self.qubitROChannels[f'{_q}RO'] = _c
+
+        # Connect device
+        # 在kw参数中寻找系统初始化相关参数
+        qs_sp = self.ParameterQSYNC.copy()
+        qs_sp.update(kwargs.get('qs_param', {}))
+        mc_sp = self.ParameterMC.copy()
+        mc_sp.update(kwargs.get('mc_param', {}))
+        z_sp = self.ParameterZ.copy()
+        z_sp.update(kwargs.get('z_param', {}))
+
+        self.qsync.open(system_parameter=qs_sp)
+        self.driver_mc.open(system_parameter=mc_sp)
+        self.driver_z.open(system_parameter=z_sp)
+        self.qsync.sync_system()
+
+    def _rline2chnl(self, rline):
+        return int(self.connectDict['QA'][f'{rline}QA'][0][-2:])
+
+    def _xy2chnl(self, _key):
+        return int(self.connectDict['XY'][_key][1][-2:])
+
+    def _z2chnl(self, _key):
+        return int(self.connectDict['Z'][_key][1][-2:])
+
+    def _qubit2rline(self, qubit):
+        return self.connectDict['RO_qubits'][f'{qubit}RO'][:2]
+
+    def disconnectDevices(self, *args, **kwargs):
+        """ Initialize devices """
+        ...
+
+    # def expendBatchChannels(self):
+
+    def stopOutput(self, *args, **kwargs):
+        """ Stop the output of waveforms """
+        self.qsync.set('ResetTrig')
+
+    def startOutput(self, *args, **kwargs):
+        """ Stop the output of waveforms """
+        self.qsync.set('GenerateTrig')
+
+    def uploadWaveforms(self, *args, **kwargs):
+        """ Upload waveforms """
+        if 'waves' not in kwargs.keys():
+            raise ValueError('Parameter `waves` does not exist in uploadWaveforms().')
+        if 'zpdc' not in kwargs.keys():
+            raise ValueError('Parameter `zpdc` does not exist in uploadWaveforms().')
+
+        self.setINDemod(**kwargs)
+        self.setRODADelay(**kwargs)
+        self.zpdc = kwargs['zpdc']
+
+        waves = kwargs['waves']
+        waves_map = {}
+        for _key, _wave in waves.items():
+            _devChannel = None
+            if _key in self.connectDict['XY'].keys():
+                _devChannel = self._xy2chnl(_key)
+                self.driver_mc.set('Waveform', _wave, _devChannel)
+            elif _key in self.connectDict['Z'].keys():
+                _devChannel = self._z2chnl(_key)
+                self.driver_mc.set('Waveform', _wave, _devChannel)
+            elif _key in self.qubitROChannels.keys():
+                _devChannel = self._qubit2rline(_key)
+                wave_list = waves_map.get(_devChannel, [])
+                wave_list.append(_wave)
+            if _devChannel is None:
+                raise RuntimeError(f'Channel {_key} does not found.')
+            print(_key, _devChannel, len(_wave))
+
+        # 合并下发probe da波形
+        for _line, wave_list in waves_map.items():
+            max_points = max(_w.size() for _w in wave_list)
+            wave_array = np.zeros((len(self.roLines[_line]), max_points))
+            for idx, _w in enumerate(wave_list):
+                wave_array[idx][:_w.size()] = _w
+            self.driver_mc.set('Waveform', wave_array.mean(axis=0), self._rline2chnl(_line))
+
+        # import matplotlib.pyplot as plt
+        # for _key, _seq in self.waves.items():
+        #     plt.plot(_seq, label=_key)
+        # plt.legend()
+        # plt.show()
+
+    def downloadIQ(self, *args, **kwargs):
+        """ Download IQ data """
+        self.setIntegrationDelay(**kwargs)
+        self.startOutput()
+        st_time = time.time()
+        roChannels = {}
+        print('self.roQubits', self.roQubits)
+        for _r in self.roQubits.keys():
+            roChannels[_r] = [self.qubitROChannels[f'{_q}RO'] for _q in self.roQubits[_r]]
+        print('roChannels', roChannels)
+        full = {
+            _r: self.driver_mc.get('IQ', channel=self._rline2chnl(_r))
+            for _r in self.roQubits.keys()
+        }
+        print('get iq data timing:', time.time() - st_time)
+
+        iqDict = dict()
+        for qubit, iqData in zip(self.roQubits.values(), full.values()):
+            for _q in qubit:
+                iqDict[_q] = iqData[self.qubitROChannels[f'{_q}RO']]
+        return iqDict
+
+    def setINDemod(self, *args, **kwargs):
+        """!
+        配置MC设备的解模系数
+        @param args:
+        @param kwargs:
+        @return:
+        """
+        if 'roQubits' not in kwargs.keys():
+            raise ValueError('Parameter `roQubits` does not exist in uploadWaveforms().')
+        if 'roLength' not in kwargs.keys():
+            raise ValueError('Parameter `roLength` does not exist in uploadWaveforms().')
+        if 'roFreq' not in kwargs.keys():
+            raise ValueError('Parameter `roFreq` does not exist in uploadWaveforms().')
+        self.roQubits = kwargs['roQubits']
+        self.roLength = {_line: [length / 1e9 for length in lengths] for _line, lengths in kwargs['roLength'].items()}
+        self.roFreq = {_line: [freq / 1e9 for freq in freqs] for _line, freqs in kwargs['roFreq'].items()}
+
+        for _line, qubits in self.roQubits.items():
+            freqs = {_q: 1e9 for _q in self.roLines[_line]}
+            lengths = {_q: 512e-9 for _q in self.roLines[_line]}
+            for _q, _freq, _len in zip(qubits, self.roFreq[_line], self.roLength[_line]):
+                freqs[_q] = _freq
+                lengths[_q] = _len
+            print(f'The configured demodulation frequency point is ({freqs})')
+            self.driver_mc.set('TimeWidth', list(lengths.values()), self._rline2chnl(_line))
+            self.driver_mc.set('FreqList', max(freqs.values()), self._rline2chnl(_line))
+
+    def setRODADelay(self, *args, **kwargs):
+        """!
+        配置probe da相对触发的延迟
+        配置粒度为16ns的整倍数
+        @param args:
+        @param kwargs:
+        @return:
+        """
+        if 'roDADelay' in kwargs.keys():
+            for _line, delay in kwargs['roDADelay'].items():
+                self.roDADelay[_line] = delay/1e9
+                self.driver_mc.set('OUTDelay', self.roDADelay[_line], self._rline2chnl(_line))
+        else:
+            raise ValueError('Parameter `roDADelay` does not exist in uploadWaveforms().')
+
+    def setShots(self, *args, **kwargs):
+        """ Set shots """
+        if 'shots' not in kwargs.keys():
+            raise ValueError('Parameter `shots` does not exist in setShots().')
+        shots = kwargs['shots']
+        self.shots = shots
+        self.qsync.set('Shot', shots)
+        self.driver_mc.set('Shot', shots)
+        print("set shots:", shots)
+
+    def setHoldoffTime(self, *args, **kwargs):
+        """ Set hold-off time """
+        if 'interval' not in kwargs.keys():
+            raise ValueError('Parameter `interval` does not exist in setHoldoffTime().')
+        interval = kwargs['interval']
+        self.qsync.set('TrigPeriod', interval / 1e9)  # 单位：s
+
+    def setLO(self, *args, **kwargs):
+        """ Set LO """
+        ...
+
+    def setIntegrationDelay(self, *args, **kwargs):
+        """ Set readout integration delay. """
+        if 'integrationDelay' in kwargs.keys():
+            for _channel, delay in kwargs['integrationDelay'].items():
+                self.integrationDelay[_channel] = delay
+                self.driver_mc.set('INDelay', delay / 1e9, self._rline2chnl(_channel))
+        else:
+            raise ValueError('Parameter `integrationDelay` does not exist in setIntegrationDelay().')
+
+    def setInputRange(self, *args, **kwargs):
+        """ Set input range for output channels """
+        if 'inputRange' in kwargs.keys():
+            for _channel, _value in kwargs['inputRange'].items():
+                devChannel = self.connectDict['QA'][_channel][1]
+                self.inputRange[devChannel] = _value
+        else:
+            raise ValueError('Parameter `inputRange` does not exist in setInputRange().')
+
+    def setOutputRange(self, *args, **kwargs):
+        """ Set output range for output channels """
+        if 'outputRange' in kwargs.keys():
+            for _channel, delay in kwargs['outputRange'].items():
+                self.outputRange[_channel] = delay
+        else:
+            raise ValueError('Parameter `outputRange` does not exist in setOutputRange().')

{nsqdriver-0.2.6.dist-info → nsqdriver-0.3.0.dist-info}/METADATA

@@ -1,16 +1,14 @@
 Metadata-Version: 2.1
 Name: nsqdriver
-Version: 0.2.6
+Version: 0.3.0
 Summary: Naishu Q series quantum measurement and control equipment driver interface
 Home-page: https://g2hoyqcmh4.feishu.cn/wiki/wikcnzvyMd82DLZUe2NsI6HxsFc
 Author: Naishu Technology
 Author-email: jilianyi@naishu.tech
 Classifier: Intended Audience :: Developers
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Requires-Python: >=3.8
+Requires-Python: >=3.10
 Requires-Dist: numpy (>=1.18)
 Requires-Dist: waveforms
 

nsqdriver-0.3.0.dist-info/RECORD

@@ -0,0 +1,17 @@
+nsqdriver/NS_CST.py,sha256=_QYCjg1YglIJ4YVmxvISqvy5rExWGlMhyrK9zYIRbRY,7877
+nsqdriver/NS_MCI.py,sha256=ZRf_waF7Q3VnClZnvjogfhVkqW7ltmwZaXXCRU8qA0o,19542
+nsqdriver/NS_QSYNC.py,sha256=68qFZ04KVXb0MSaeKIZVZYqR0UV_tkJYWz7KSz4Dvg0,23281
+nsqdriver/__init__.py,sha256=BMKtTehGZlv0CgT1vJAbyZW1KF670NZwQJ9wWQBYzC0,326
+nsqdriver/common.py,sha256=IHfShqgNN3THbe7KD7ICdfFY7JKC_kQj0_KJk-449gQ,499
+nsqdriver/setup.py,sha256=4ajKQtY4hr2240f5sA81f66_92Yp_wG6gPM-zsOWq9M,390
+nsqdriver/compiler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nsqdriver/compiler/ns_wave.py,sha256=Z-O--9OtmLiILGx-4imSJuAcdgjP1PNknQlDrSo2CIw,15411
+nsqdriver/compiler/py_wave_asm.py,sha256=aXMr3CqMC0vDP4mtTdyVfnwpvlrE14hGgdthWAF5S6g,17407
+nsqdriver/wrapper/AWG_ADC.py,sha256=8K5xLEa_3CKnStWcoJeTj8ufEQfrx6o4l4XGNuOP3Ug,19211
+nsqdriver/wrapper/BD_NSMCI.py,sha256=SMnDbS6iXDiPBhoR9QVJB_lJXhIXRQIVLJRSBEihLzw,12245
+nsqdriver/wrapper/ND_NSMCI.py,sha256=gTyrDObhbggl35V_q-kpDoPsfttIEnDFc6f0-pk_P0k,6288
+nsqdriver/wrapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nsqdriver-0.3.0.dist-info/METADATA,sha256=htBmUOmmEPWvuvX1fmTGsoFlXiZPqysgxrwFnjDDwp0,497
+nsqdriver-0.3.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+nsqdriver-0.3.0.dist-info/top_level.txt,sha256=o7EbQoFO6BoaG3KGbS9Lg_aRheZSY5KYzoYuI9vx-AI,10
+nsqdriver-0.3.0.dist-info/RECORD,,

nsqdriver-0.2.6.dist-info/RECORD

@@ -1,13 +0,0 @@
-nsqdriver/NS_CST.py,sha256=_QYCjg1YglIJ4YVmxvISqvy5rExWGlMhyrK9zYIRbRY,7877
-nsqdriver/NS_MCI.py,sha256=nyCQji_UxVFfyaE8QHJ_laE0Ga8ybSHbe0-d5k65QjY,19114
-nsqdriver/NS_QSYNC.py,sha256=P5BDiTz7hu6omwFnjGfGXfnzAoBfrvGYAYLTT45RoOc,22747
-nsqdriver/__init__.py,sha256=bUQSdaBfww7fv1v64SORPYrqp6cjXiLklnqdSUc24h8,258
-nsqdriver/common.py,sha256=IHfShqgNN3THbe7KD7ICdfFY7JKC_kQj0_KJk-449gQ,499
-nsqdriver/setup.py,sha256=4ajKQtY4hr2240f5sA81f66_92Yp_wG6gPM-zsOWq9M,390
-nsqdriver/wrapper/AWG_ADC.py,sha256=8K5xLEa_3CKnStWcoJeTj8ufEQfrx6o4l4XGNuOP3Ug,19211
-nsqdriver/wrapper/ND_NSMCI.py,sha256=gTyrDObhbggl35V_q-kpDoPsfttIEnDFc6f0-pk_P0k,6288
-nsqdriver/wrapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nsqdriver-0.2.6.dist-info/METADATA,sha256=zVTz9ysW-HrG8YuzIRo9MEue4PqJFb28tFMKvV9ORdQ,596
-nsqdriver-0.2.6.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-nsqdriver-0.2.6.dist-info/top_level.txt,sha256=o7EbQoFO6BoaG3KGbS9Lg_aRheZSY5KYzoYuI9vx-AI,10
-nsqdriver-0.2.6.dist-info/RECORD,,