nsqdriver 0.5.7__cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

nsqdriver/__init__.py

@@ -0,0 +1,10 @@
+from .NS_MCI import Driver as MCIDriver
+from .NS_QSYNC import Driver as QSYNCDriver
+from .NS_CST import Driver as CSTDriver
+from .compiler.ns_wave import InsChannel
+from .compiler.py_wave_asm import nsw_config, AssemblyError
+
+version_pack = (0, 5, 7)
+
+__version__ = '.'.join(str(_) for _ in version_pack)
+__all__ = ['MCIDriver', 'QSYNCDriver', 'CSTDriver', 'InsChannel']

nsqdriver/common.py

@@ -0,0 +1,20 @@
+class Quantity(object):
+    def __init__(self, name: str, value=None, ch: int = 1, unit: str = ''):
+        self.name = name
+        self.default = dict(value=value, ch=ch, unit=unit)
+
+
+class QInteger:
+    def __init__(self, name, value=None, unit='', ch=None,
+                 get_cmd='', set_cmd='',):
+        self.name = name
+
+
+class BaseDriver:
+    def __init__(self, addr, timeout, **kw):
+        self.addr = addr
+        self.timeout = timeout
+
+
+def get_coef(*args):
+    return '', '', '', ''

nsqdriver/compiler/ns_wave.pyi

@@ -0,0 +1,151 @@
+import numpy as np
+import waveforms as wf
+
+from enum import IntEnum
+from typing import Union, List, Dict
+from nsqdriver.compiler.py_wave_asm import *
+
+
+class QInsPlaceholder(NSQCommand):
+    ...
+
+
+class Wave(GenTagMixin):
+    class Tag(IntEnum): ...
+
+    def __init__(self, ins_obj: "InstructionQ", ins_id: dict): ...
+
+class Frame(Wave):
+    def __init__(self, ins_obj: "InstructionQ", ins_id: int, freq: float): ...
+
+    @property
+    def freq(self) -> float: ...
+
+    @freq.setter
+    def freq(self, value: float): ...
+
+    def __mul__(self, other: Wave) -> Signal: ...
+
+    def format(self) -> "NSQCommand": ...
+
+
+class Envelope(Wave):
+    def __init__(self, ins_obj: "InstructionQ", ins_id: int, content: "Union[wf.Waveform, np.ndarray]"): ...
+
+    @property
+    def content(self) -> "Union[wf.Waveform, np.ndarray]": ...
+
+    @content.setter
+    def content(self, value: "Union[wf.Waveform, np.ndarray]"): ...
+
+    def __mul__(self, other: Wave) -> Signal: ...
+
+    def format(self) -> "NSQCommand": ...
+
+
+class Signal(Wave):
+    def __init__(self, ins_obj: "InstructionQ", ins_id: dict): ...
+
+
+class InstructionQ(GenTagMixin):
+    """!
+    包含各种具体的指令
+    """
+
+    def __init__(self, freqs=None, envelopes=None):
+        self.i_set: "List[Union[NSQCommand, InstructionQ]]" = []
+        self.f_set: "Dict[int, Frame]" = {}
+        self.e_set: "Dict[int, Envelope]" = {}
+        self.symbol_set: Dict[str, int] = {}
+        self.symbol_idx: int = -1
+        self.is_first_trig: bool = True
+        self.last_ins: NSQCommand
+
+    def clear(self) -> None: ...
+
+    @property
+    def length(self) -> int: ...
+
+    def ins_frame(self, freq, idx=None) -> Frame: ...
+
+    def ins_envelope(self, envelope: "Union[np.ndarray, wf.Waveform, str]", idx=None) -> Envelope: ...
+
+    def evlp_gaussian(self, width: float) -> Envelope: ...
+
+    def evlp_cospulse(self, width: float) -> Envelope: ...
+
+    def evlp_square(self, width: float) -> Envelope: ...
+
+    def _append_ins(self, cmd: "Union[NSQCommand, InstructionQ]"): ...
+
+    def _map_var(self, var: str) -> int: ...
+
+    def wait_for_trigger(self): ...
+
+    def ins_variable(self, reg: str, value: int): ...
+
+    def ins_add(self, reg: str, value: int): ...
+
+    def ins_reset_frame(self, flag: str, frame: "Frame"): ...
+
+    def inc_phase(self, frame: "Frame", phase: float): ...
+
+    def play_wave(self, wave: Signal, amp=1, freq=0, phase=0): ...
+
+    def play_zero(self, width: float): ...
+
+    def wait(self): ...
+
+    def end(self): ...
+
+    def capture(self, width: float, delay=0.): ...
+
+
+class InsChannel(InstructionQ):
+    def __init__(self, freqs=None, envelopes=None):
+        self.if_stack: "List[InsIF]" = []
+        self.looping = False
+
+    def ins_if(self, formula: str) -> "InsIF": ...
+
+    def ins_else(self) -> "InsElse": ...
+
+    def ins_loop(self, times: int) -> "InsLoop": ...
+
+
+class InsIF(InsChannel):
+    ch_judge_name = {f'FREQ_{i}': 1<<i for i in range(6)}
+
+    def __init__(self, freqs=None, envelopes=None):
+        self.channel: "InsChannel" = None
+        self.key_ins: "NSQCommand" = None
+        self.ins_else = None
+
+    @classmethod
+    def from_channel(cls, channel :InsChannel) -> InsIF: ...
+
+    @property
+    def formula(self): ...
+
+    @formula.setter
+    def formula(self, formula: str): ...
+
+
+class InsElse(InsChannel):
+    def __init__(self, freqs=None, envelopes=None): ...
+
+    @classmethod
+    def from_channel(cls, channel: InsChannel) -> InsElse: ...
+
+    def capture(self, width: float, delay=0): ...
+
+
+class InsLoop(InsChannel):
+    def __init__(self, freqs: float=None, envelopes: "Union[wf.Waveform, np.ndarray]"=None): ...
+
+    @classmethod
+    def from_channel(cls, channel: InsChannel) -> InsLoop: ...
+
+    def capture(self, width: float, delay=0): ...
+
+    def _compile(self) -> "List[NSQCommand]": ...

nsqdriver/compiler/py_wave_asm.pyi

@@ -0,0 +1,29 @@
+from typing import Any
+
+import numpy as np
+
+
+global_config = {
+    'play_zero_step': 4e-9,
+    'OUTSrate': 8e9,
+    'envelope_dtype': np.int16,  # 描述包络每个点的数据类型
+    'envelope_step': 64,  # 包络步进粒度，单位为bytes
+    'envelope_quant': 16383,  # 包络量化范围
+    'envelope_cache': 204800,  # 包络缓存大小，单位bytes
+    'envelope_head': np.array([2, 0, 0, 4096, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], dtype=np.int16),   # 包络更新包头
+}
+
+
+def nsw_config(name: str, value: Any) -> None: ...
+
+
+class AssemblyError(RuntimeError): ...
+
+
+class GenTagMixin: ...
+
+
+class NSQCommand: ...
+
+
+class Assembler: ...