nsqdriver 0.10.0__cp312-cp312-macosx_10_13_universal2.whl

nsqdriver/wrapper/ND_NSMCI.py

@@ -0,0 +1,245 @@
+import numpy as np
+from nsqdriver import MCIDriver, QSYNCDriver
+
+
+class _XYChannel:
+    def __init__(self, mci: "DeviceBase", ch=1):
+        self.mci = mci
+        self.ch = ch
+        self.to_zero = np.zeros((16,))
+        self.to_one = np.ones((16,))
+        self.en = True
+        self.mode = 0
+        self.off = 0
+
+    def wave(self, w):
+        self.wavex(w, self.ch)
+
+    def wavex(self, w, idx):
+        if np.max(np.abs(w)) < 1e-30:
+            wr = np.zeros(16)
+        else:
+            wr = w
+        self.mci.mci_driver.set("Waveform", wr, idx)
+
+    def arm(self, k=None):
+        self.mci.mci_driver.set('PushWaveCache')
+
+    def trig_del(self, delay):
+        ...
+
+    def output_del(self, delay):
+        ...
+
+    def __del__(self):
+        pass
+
+    def output(self, b):
+        self.en = bool(b)
+        if not self.en:
+            self.mci.mci_driver.set("Waveform", self.to_zero, self.ch)
+
+    def mode(self, m_):
+        ...
+
+    def offsetx(self, off, idx):
+        self.mci.mci_driver.set('Waveform', off*self.to_one, idx)
+        self.off = off
+
+    def offset(self, off):
+        self.offsetx(off, self.ch)
+
+    W = {
+        "wave": wave,
+        "output": output,
+        "trig_del": trig_del,
+        "output_del": output_del,
+        "mode": mode,
+        "offset": offset,
+    }
+
+    Q = {"arm": arm}
+
+
+class _ZChannel(_XYChannel):
+    pass
+
+
+class _Probe:
+    def __init__(self, mci: "DeviceBase", ch=1):
+        self.mci = mci
+        self.ch = ch
+        self.freqList = []
+        self.SGS = None  # single shot temp cache
+        self.AVG = None  # average temp cache
+        self.depth = 2000
+        self._width = 1000
+        self.start = 500
+        self.demod = 1
+        self.averaged_I = np.zeros((16384, ))  # 直播直采，相当于只有I路数据
+
+    def depth(self, depth_):
+        self.depth = depth_
+        self.mci.mci_driver.set('PointNumber', depth_, self.ch)
+
+    def demodulation_on(self, demod_):
+        self.demod = int(demod_)
+
+    def start(self, start_):
+        self.start = start_
+        self.mci.mci_driver.set('TriggerDelay', start_, self.ch)
+
+    def width(self, width_):
+        self._width = width_/4e9
+        self.mci.mci_driver.set('PointNumber', width_, self.ch)
+
+    def freqs(self, *freqList_):
+        self.freqList = freqList_
+        self.mci.mci_driver.set('FreqList', freqList_, self.ch)
+
+    def shot(self, _shot):
+        self.mci.mci_driver.set('Shot', _shot)
+
+    def measure(self, k=None):
+        self.mci.mci_driver.set('StartCapture')
+        if self.demod:
+            self.SGS = self.mci.mci_driver.get('IQ', self.ch)
+            self.AVG = np.mean(self.SGS, axis=0)
+        else:
+            self.averaged_I = np.mean(self.mci.mci_driver.get('TraceIQ', self.ch), axis=0)
+
+    def single_shot(self, k=None):
+        return self.SGS
+
+    def average(self, k=None):
+        return self.AVG
+
+    def trace_I(self, k=None):
+        return self.averaged_I
+
+    def trace_Q(self, k=None):
+        return self.averaged_I
+
+    def __del__(self):
+        pass
+
+    W = {
+        "demod": demodulation_on,
+        "depth": depth,
+        "width": width,
+        "start": start,
+        "freqs": freqs,
+        "shot": shot,
+    }
+
+    Q = {
+        "measure": measure,
+        "A": average,
+        "S": single_shot,
+        "traceI": trace_I,
+        "traceQ": trace_Q
+    };
+
+
+# one box need one class
+class DeviceBase:
+    def __init__(self):
+        self.mci_driver = MCIDriver('127.0.0.1')
+        self.qsync_driver = QSYNCDriver('127.0.0.1')
+
+
+class NS_MCI(DeviceBase):
+    def __init__(self, addr, srate=10e9, mixmode=2, ref_clk='in'):
+        """!
+        此类涉及到系统同步，放到最后实例化
+        @param addr: 设备ip
+        @param srate: OUT通道采样率
+        @param mixmode: 为2时开启OUT通道混合模式，增强第二奈奎斯特区输出
+        @param ref_clk: 设备参考信号来源，不接外输出100M时都配置为'in'
+        """
+        super(NS_MCI, self).__init__()
+        self.mci_driver = MCIDriver(addr)
+        self.qsync = QSYNCDriver(addr)
+        self.srate = srate
+        self.mixmode = mixmode
+        self.ref_clk = ref_clk
+        self.connect()
+
+    def connect(self):
+        mci_params = {'DArate': self.srate, 'MixMode': self.mixmode}
+        qsync_params = {'RefClock': self.ref_clk}
+
+        self.qsync.open(system_parameter=qsync_params)
+        self.mci_driver.open(system_parameter=mci_params)
+        self.qsync.sync_system()
+
+        self.mci_driver.set('EnableWaveCache', True)
+
+        for _ch in range(22):
+            xy_ch = _ch+1
+            setattr(self, f'OUT{xy_ch}', _XYChannel(self, xy_ch))
+        for _ch in range(2):
+            probe_ch = _ch+1
+            setattr(self, f'IN{probe_ch}', _Probe(self, probe_ch))
+
+    def trig_interval(self, interval):
+        self.interval = interval
+        self.qsync.set('TrigPeriod', int(interval))
+
+    def trig_count(self, count_):
+        self.qsync.set('Shot', int(count_))
+
+    def trig(self):
+        self.qsync.set('GenerateTrig', self.interval)
+
+    def awg_arm(self):
+        self.mci_driver.set('PushWaveCache')
+
+    def __del__(self):
+        pass
+
+    W = {
+        "trig_interval": trig_interval,
+        "trig_count": trig_count,
+        "connect": connect,
+        "trig": trig,
+        "awg_arm": awg_arm,
+    }
+
+    Q = {
+    }
+
+
+class NS_Z(DeviceBase):
+    def __init__(self, addr, mixmode=2):
+        """!
+        此类负责控制24 Z OUT通道的设备，采样率固定为2Gsps，mixmode固定为1
+        @param addr: 设备ip
+        @param mixmode: 为1时关闭OUT通道混合模式
+        """
+        super(NS_Z, self).__init__()
+        self.mci_driver = MCIDriver(addr)
+        self.qsync = QSYNCDriver(addr)
+        self.srate = 2e9
+        self.mixmode = 1
+        self.connect()
+
+    def connect(self):
+        mci_params = {'DArate': self.srate, 'MixMode': self.mixmode}
+        self.mci_driver.open(system_parameter=mci_params)
+
+        self.mci_driver.set('EnableWaveCache', True)
+
+        for _ch in range(24):
+            xy_ch = _ch+1
+            setattr(self, f'OUT{xy_ch}', _ZChannel(self, xy_ch))
+
+    def awg_arm(self):
+        self.mci_driver.set('PushWaveCache')
+
+    W = {
+        "connect": connect,
+    }
+
+    Q = {
+    }

nsqdriver-0.10.0.dist-info/METADATA

@@ -0,0 +1,109 @@
+Metadata-Version: 2.1
+Name: nsqdriver
+Version: 0.10.0
+Summary: Q series measurement and control equipment driver interface
+Home-page: https://g2hoyqcmh4.feishu.cn/wiki/Jgt1wxlKniveYQkqCgbcJHsMnoe
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Scientific/Engineering :: Interface Engine/Protocol Translator
+Classifier: Operating System :: MacOS :: MacOS X
+Classifier: Operating System :: Microsoft :: Windows
+Classifier: Operating System :: POSIX
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: numpy>=1.18
+Requires-Dist: astor>=0.8.1
+Provides-Extra: waveforms
+Requires-Dist: waveforms; extra == "waveforms"
+
+```
+  _   _   ____     ___    ____           _                       
+ | \ | | / ___|   / _ \  |  _ \   _ __  (_) __   __   ___   _ __ 
+ |  \| | \___ \  | | | | | | | | | '__| | | \ \ / /  / _ \ | '__|
+ | |\  |  ___) | | |_| | | |_| | | |    | |  \ V /  |  __/ | |   
+ |_| \_| |____/   \__\_\ |____/  |_|    |_|   \_/    \___| |_|   
+```
+***
+
+### 简介
+NSQDriver为一套测控系列设备的Python通用驱动接口。可使用Python作为编程语言，基于此驱动接口可以实现对测控设备的控制，进而完成各种超导量子芯片的表征实验和高保真门操作。
+
+```
+                              +------------------------+
+                              |          Python        |
+                              |     +------------------+
+                              |     |      NSQDriver   |
+                              +------------------------+
+                                     :     :         :
+                           +----------     :         -----+
+                           :               :              :
+                           V               V              V
+                     +-----------+   +-----------+   +-----------+   
+                     |  Device A |   |  Device B |   |   QSYNC   |    
+                     +-----------+   +-----------+   +-----------+   
+```
+
+NSQDriver接口可用于统一控制Q100/PQ系列测控仪器，可详细控制每台仪器每个OUT、IN通道的播放与采集行为，以及查看设备的运行状态等。在一个多台测控设备组成的多比特测控系统中，可以有效的在软件层面实现对系统中每台设备的详细控制。
+
+***
+### 快速开始
+- 驱动接口的通用流程如下
+```python
+from nsqdriver import MCIDriver, QSYNCDriver
+
+# 实例化设备驱动接口
+driver = MCIDriver('127.0.0.1')
+qsync = QSYNCDriver('127.0.0.1')
+
+# 连接设备
+qsync.open()
+driver.open()
+
+# 初始化系统
+qsync.sync_system()
+
+
+# 相关参数，参数细节相关参考 8.可用参数列表
+driver.set('Shot', 1024)
+```
+
+- NSWave语言
+  - NSWave是一种对于Sequence序列发生技术的配套编程语言，属于python语法的子集，支持与python程序进行混编
+  - Sequence序列发生模式将测控所需的各种门的编辑由主控计算机直接生成波形转化成了设备实时计算
+  - 提供了一套直观的时序编辑接口
+```python
+from nsqdriver import nswave as nw
+from nsqdriver import MCIDriver
+import numpy as np
+
+@nw.kernel
+def program(envelope: np.ndarray):
+    srate: nw.Var = 8e9
+    time_width: nw.Var = 6e-6
+    freq: nw.Var = 4.1e9
+    time_line: np.ndarray = np.linspace(0, 6e-6, int(time_width * srate + 1))
+    wave: np.ndarray = np.cos(2 * np.pi * time_line * freq)
+    frame_0: nw.Frame = nw.ins_frame(0e9, 0*np.pi)
+    frame_1: nw.Frame = nw.ins_frame(4.3e9, 0*np.pi)
+    envelope_0: nw.Envelope = nw.ins_envelope(wave)
+    envelope_1: "nw.Envelope" = nw.evlp_gaussian(2e-6, srate)
+    envelope_2: nw.Envelope = nw.ins_envelope(envelope)
+
+    nw.wait_for_trigger()
+    nw.reset_frame([frame_0, frame_1])
+    nw.wait(3.2e-6)
+    nw.play_wave(envelope_1, frame_1)
+    nw.play_wave(envelope_0, frame_0)
+    nw.inc_phase(frame_1, np.pi)
+    nw.play_wave(envelope_2, frame_1)
+
+    return nw.Kernel()
+
+driver: MCIDriver
+envelope = np.ones((10240, ))
+driver.set("Program", program(envelope), 'S5-O1')
+```

nsqdriver-0.10.0.dist-info/RECORD

@@ -0,0 +1,31 @@
+nsqdriver/NS_CST.py,sha256=_QYCjg1YglIJ4YVmxvISqvy5rExWGlMhyrK9zYIRbRY,7877
+nsqdriver/NS_MCI.py,sha256=zbAA70_g_02mF76LqcIzVeqRRNrVscTL60iHpFsgmZ4,21088
+nsqdriver/NS_QSYNC.py,sha256=ZSOznM5PMlOm89rWAMiqVyZLWn3y0lTQp3d1z6LnkV0,27121
+nsqdriver/__init__.py,sha256=qGxm26SCYU2CZT2nD597nxHmvSHkC6bV3Sjw63hceMs,372
+nsqdriver/common.py,sha256=IHfShqgNN3THbe7KD7ICdfFY7JKC_kQj0_KJk-449gQ,499
+nsqdriver/compiler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nsqdriver/compiler/assembler.cpython-312-darwin.so,sha256=WqBTLhcSH2V_MkNX_CD8nPxJ2g9R38-b0V7s_8Ue3Ow,1033872
+nsqdriver/compiler/ns_wave.cpython-312-darwin.so,sha256=dCpWfXbT5jwmw8c31sZSsQDYctjKYGu20eICu5IXJd8,640752
+nsqdriver/compiler/ns_wave.pyi,sha256=JQiE_9HqCzm00bS6c2JaaIXwt7xN_x9XXax-BBrE7-o,3824
+nsqdriver/compiler/py_wave_asm.cpython-312-darwin.so,sha256=PogSEvF5br5wwXp1jILB8Dxajmu2hj7lM_je6igSbFI,838840
+nsqdriver/compiler/py_wave_asm.pyi,sha256=w2PWlBrns4tfDYjTLyOBfNgjjjVA_xzMJZYYyDPKce4,656
+nsqdriver/nswave/__init__.py,sha256=BOSBAccspBvBK0eX49b3731HYaMuSFrtGHsUKu8u2z8,88
+nsqdriver/nswave/_asm.pyi,sha256=3jDUjyfmn7iBgmQCviSzhvGEXoWkyQlEGfgmGJAyaH8,2722
+nsqdriver/nswave/_checkers.cpython-312-darwin.so,sha256=QecJhssHODT6QX4O0PtSNNEPou5vJGwM4X2Vrnw6wTY,473120
+nsqdriver/nswave/_errors.cpython-312-darwin.so,sha256=j7zi1CZGmTxd_eUyGyQkTRMHqBWB_wev56RHWjs3Z-Y,238976
+nsqdriver/nswave/_functions.cpython-312-darwin.so,sha256=tNXhN8uKj4yfMtraWZ37OkVBseFhKLcm9B1_9Wjqzm4,269128
+nsqdriver/nswave/_functions.pyi,sha256=GpLk_DBPy_rgO6pQ2iFtz24sdpJ4HmDDlrSTZsHEMi4,1565
+nsqdriver/nswave/_ir.cpython-312-darwin.so,sha256=QGfHftUTmVc929To1JWVrmJ4YaeW-23synyHf_6Jih4,727456
+nsqdriver/nswave/_ir_pass.cpython-312-darwin.so,sha256=ZWJCC-TzPAYgtBfYuX9SOqL-L_c2UJEAOzJKEpp-NlU,182944
+nsqdriver/nswave/_optimizations.cpython-312-darwin.so,sha256=vqCyeaAyvW0JZEmmQe_06o-Cnx9UPHaOnnZReYJecFY,183896
+nsqdriver/nswave/_rules.cpython-312-darwin.so,sha256=E4M5MU50Hs4bQzU6KD_gz4k7yY7_0WLkZBDYyxIUVlI,461584
+nsqdriver/nswave/_translate.cpython-312-darwin.so,sha256=AML1pAzFWTO-C6G6U2PugVDAecjm5AD0_2nRJHaaVAQ,272296
+nsqdriver/nswave/kernel.cpython-312-darwin.so,sha256=xtfKU4AHB23JzP567v28ddQKYlWHnzH9XQFoWqSsVxc,634560
+nsqdriver/nswave/kernel.pyi,sha256=83P6dTTARLmXC3B_fJn_ZqtyOWZ3-scfP7uN45xJAlY,1555
+nsqdriver/wrapper/AWG_ADC.py,sha256=ktQkORc8VZUFkNg6huC7DKIkvZyHYXRolQ2SAA89JnM,19199
+nsqdriver/wrapper/ND_NSMCI.py,sha256=gTyrDObhbggl35V_q-kpDoPsfttIEnDFc6f0-pk_P0k,6288
+nsqdriver/wrapper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nsqdriver-0.10.0.dist-info/METADATA,sha256=UpCBt0PQXapdlNa_R4Kqun3D6Ce2RNmy6449-JqW37s,4275
+nsqdriver-0.10.0.dist-info/WHEEL,sha256=JjfPN3tuNVktXeVx2CdYSsvuQYIjI_7f2R0S3yE0Gks,115
+nsqdriver-0.10.0.dist-info/top_level.txt,sha256=o7EbQoFO6BoaG3KGbS9Lg_aRheZSY5KYzoYuI9vx-AI,10
+nsqdriver-0.10.0.dist-info/RECORD,,

nsqdriver-0.10.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.1.0)
+Root-Is-Purelib: false
+Tag: cp312-cp312-macosx_10_13_universal2
+

nsqdriver-0.10.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+nsqdriver