nsqdriver 0.10.0__cp312-cp312-macosx_10_13_universal2.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, 10, 0)
+
+__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: ...

nsqdriver/nswave/__init__.py

@@ -0,0 +1,3 @@
+from .kernel import *
+from ._functions import *
+from ._ir_pass import ir_pass as ir_pass

nsqdriver/nswave/_asm.pyi

@@ -0,0 +1,96 @@
+import numpy as np
+from _typeshed import Incomplete
+
+__all__ = ['AssemblyError', 'GenTagMixin', 'NSQCommand', 'QInsFrame', 'nsw_config', 'QInsEnvelope', 'QInsWait', 'QInsNop', 'QInsJumpImmediate', 'QInsPlayImm', 'QInsWaitTrig', 'QInsFrameRst', 'QInsWaitTrig', 'QInsFrameAdd', 'QInsCapture']
+
+def nsw_config(name, value) -> None: ...
+
+class QNumber:
+    value: int
+    def __init__(self) -> None: ...
+    def __int__(self) -> int: ...
+    def __float__(self) -> float: ...
+    def __bool__(self) -> bool: ...
+
+class AssemblyError(RuntimeError): ...
+
+class GenTagMixin:
+    @property
+    def generate_tag(self): ...
+
+class NSQCommand(GenTagMixin):
+    tag: Incomplete
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @classmethod
+    def compile(cls, inst_list: list[Self]) -> tuple[np.ndarray, str]: ...
+
+class QInsFrame(NSQCommand):
+    freq: Incomplete
+    phase: Incomplete
+    idx: Incomplete
+    def __init__(self, freq, phase, idx) -> None: ...
+    @property
+    def overhead(self): ...
+
+class QInsEnvelope(NSQCommand):
+    envelope: Incomplete
+    envelop_slice: Incomplete
+    def __init__(self, envelope: np.ndarray) -> None: ...
+    def __len__(self) -> int: ...
+    def __bytes__(self) -> bytes: ...
+
+class QInsFrameRst(NSQCommand):
+    @property
+    def overhead(self): ...
+
+class QInsFrameAdd(NSQCommand):
+    frames: Incomplete
+    frequency: Incomplete
+    phase: Incomplete
+    def __init__(self, frames: list[QInsFrame], frequency: float, phase: float) -> None: ...
+    @property
+    def overhead(self): ...
+
+class QInsWait(NSQCommand):
+    width: Incomplete
+    def __init__(self, width) -> None: ...
+    @property
+    def overhead(self): ...
+
+class QInsWaitTrig(NSQCommand):
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+
+class QInsNop(NSQCommand):
+    @property
+    def overhead(self): ...
+
+class QInsJumpImmediate(NSQCommand):
+    idx: Incomplete
+    def __init__(self, idx: int) -> None: ...
+    @property
+    def overhead(self): ...
+
+class QInsCapture(NSQCommand):
+    fre: Incomplete
+    acq_width: Incomplete
+    delay_width: Incomplete
+    play_width: Incomplete
+    para: Incomplete
+    def __init__(self, fre: list[int], acq_width: float, delay_width: float, play_width: float, raw_data_store: bool, iq_data_store: bool, judge_data_store: bool, double_fre_mode: bool) -> None: ...
+    @property
+    def overhead(self): ...
+
+class QInsPlayImm(NSQCommand):
+    frame: Incomplete
+    freq: Incomplete
+    amp: Incomplete
+    bias: Incomplete
+    envelope: Incomplete
+    phase: Incomplete
+    def __init__(self, frame: QInsFrame, envelope: QInsEnvelope, amp, bias, freq, phase) -> None: ...
+    @property
+    def overhead(self): ...

nsqdriver/nswave/_functions.pyi

@@ -0,0 +1,26 @@
+import numpy as np
+from ._asm import NSQCommand, QInsEnvelope, QInsFrame
+from typing import Iterable, TypeVar
+
+__all__ = ['Frame', 'Envelope', 'Int', 'Var', 'wait_for_trigger', 'ins_envelope', 'ins_frame', 'evlp_gaussian', 'evlp_square', 'evlp_cospulse', 'inc_phase', 'inc_frequency', 'reset_frame', 'play_wave', 'wait', 'capture']
+
+Frame = TypeVar('Frame', bound=QInsFrame)
+Envelope = TypeVar('Envelope', bound=QInsEnvelope)
+Placeholder = TypeVar('Placeholder', bound=NSQCommand)
+Var = TypeVar('Var')
+Int = TypeVar('Int')
+Uint = TypeVar('Uint')
+Float = TypeVar('Float')
+
+def ins_frame(freq: float, phase: float, *, idx: int = None) -> QInsFrame: ...
+def ins_envelope(envelope: np.ndarray) -> QInsEnvelope: ...
+def evlp_gaussian(width: float, srate: float = 8000000000.0) -> QInsEnvelope: ...
+def evlp_cospulse(width: float, srate: float = 8000000000.0) -> QInsEnvelope: ...
+def evlp_square(width: float, srate: float = 8000000000.0) -> QInsEnvelope: ...
+def reset_frame(frame_list: Iterable[Frame] = ...) -> Placeholder: ...
+def wait_for_trigger() -> Placeholder: ...
+def inc_phase(frame_list: Iterable[Frame], phase: float) -> Placeholder: ...
+def inc_frequency(frame_list: Iterable[Frame], freq: float) -> Placeholder: ...
+def play_wave(envelope: QInsEnvelope, frame: QInsFrame, amp: int = 1, freq: int = 0, phase: int = 0) -> Placeholder: ...
+def wait(width) -> Placeholder: ...
+def capture(freq_list, acq_width: float, delay_width: float, play_width: float, raw_data_store: bool, iq_data_store: bool, judge_data_store: bool, double_fre_mode: bool) -> Int: ...

nsqdriver/nswave/kernel.pyi

@@ -0,0 +1,45 @@
+from ._functions import *
+import ast
+from _typeshed import Incomplete
+
+__all__ = ['kernel', 'Kernel']
+
+class KernelParseError(Exception): ...
+
+class NSQCKernel(type):
+    def __init__(cls, name, bases, attrs) -> None: ...
+
+class Kernel(ast.NodeVisitor, metaclass=NSQCKernel):
+    max_frame: int
+    name: Incomplete
+    func_args: Incomplete
+    func_kwargs: Incomplete
+    top_src: Incomplete
+    top_node: Incomplete
+    var_symbol: Incomplete
+    frame_symbol: Incomplete
+    frame_idx: int
+    envelope_symbol: Incomplete
+    register_symbol: Incomplete
+    register_idx: int
+    instruction_list: Incomplete
+    def __init__(self, func: Incomplete | None = None, args: Incomplete | None = None, kwargs: Incomplete | None = None) -> None: ...
+    def clear(self) -> None: ...
+    def parse(self): ...
+    def show(self) -> None: ...
+    def __call__(self, *args, **kwargs): ...
+    def visit_arg(self, node: ast.arg): ...
+    def visit_Assign(self, node: ast.Assign): ...
+    def visit_AugAssign(self, node: ast.AugAssign): ...
+    def visit_AnnAssign(self, node: ast.AnnAssign): ...
+    def visit_Call(self, node: ast.Call) -> object: ...
+    def visit_BinOp(self, node: ast.BinOp): ...
+    def visit_Constant(self, node): ...
+    def visit_Expr(self, node: ast.Expr): ...
+    def visit_If(self, node: ast.If): ...
+    def visit_Return(self, node) -> None: ...
+    def visit_FunctionDef(self, node: ast.FunctionDef): ...
+    def visit_arguments(self, node: ast.arguments): ...
+    def visit_Module(self, node: ast.Module): ...
+
+def kernel(func): ...