nsqdriver 0.12.17__cp313-cp313-manylinux2014_x86_64.manylinux_2_17_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, 12, 17)
+
+__version__ = '.'.join(str(_) for _ in version_pack)
+__all__ = ['MCIDriver', 'QSYNCDriver', 'CSTDriver']

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,9 @@
+from .kernel import *
+from ._functions import *
+from ._ir_pass import ir_pass as ir_pass
+import nsqdriver.nswave._rules as rules
+import nsqdriver.nswave._checkers as checkers
+import nsqdriver.nswave._translate as translator
+
+
+__all__ = ['Kernel', 'ir_pass', 'rules', 'checkers', 'translator', 'form_kernel']

nsqdriver/nswave/_asm.pyi

@@ -0,0 +1,97 @@
+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
+    line: Incomplete
+    def __init__(self, freq, phase, idx, line) -> 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/_checkers.pyi

@@ -0,0 +1,47 @@
+from ._ir import *
+from _typeshed import Incomplete
+
+__all__ = ['BaseChecker', 'OverlapChecker', 'EnvelopeChecker', 'TwoCapiChecker', 'SequenceChecker', 'PlyChecker', 'DrstChecker', 'CapiChecker', 'FmsiChecker', 'IRNumberChecker']
+
+class BaseChecker:
+    warnings_instruction: Incomplete
+    error_instruction: Incomplete
+    IR_list: Incomplete
+    def __init__(self) -> None: ...
+    def check_ir(self, IR_list) -> list: ...
+
+class OverlapChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list: list): ...
+
+class SequenceChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list) -> list: ...
+
+class EnvelopeChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...
+
+class PlyChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...
+
+class DrstChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...
+
+class CapiChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...
+
+class FmsiChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...
+
+class IRNumberChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...
+
+class TwoCapiChecker(BaseChecker):
+    def __init__(self) -> None: ...
+    def check_ir(self, ir_list): ...

nsqdriver/nswave/_errors.pyi

@@ -0,0 +1,24 @@
+from _typeshed import Incomplete
+
+class WaitError(Exception): ...
+class TimingError(Exception): ...
+class PlayError(Exception): ...
+class CaptureError(Exception): ...
+
+class MultipleErrorsException(Exception):
+    errors: Incomplete
+    def __init__(self, errors) -> None: ...
+
+class BaseError:
+    IR_list: Incomplete
+    def __init__(self) -> None: ...
+    def query_ir(self, error_list): ...
+
+class Errors(BaseError):
+    errors: Incomplete
+    def __init__(self) -> None: ...
+    def quire_ir(self, error_list) -> None: ...
+    def check_errors(self) -> None: ...
+
+class Warnings(BaseError):
+    def inquire_ir(self, warning_list): ...

nsqdriver/nswave/_functions.pyi

@@ -0,0 +1,34 @@
+import numpy as np
+from ._ir import IRArbWaveform, IRBase, IREnvelope, IRFrame
+from _typeshed import Incomplete
+from typing import Iterable, TypeVar
+
+__all__ = ['Frame', 'Envelope', 'Int', 'Var', 'set_judge', 'Reg', 'ArbWave', 'receive', 'send', 'init_arbwave', 'play_arb', 'wait_for_trigger', 'ins_envelope', 'init_frame', 'evlp_gaussian', 'evlp_square', 'evlp_cospulse', 'inc_phase', 'inc_frequency', 'reset_frame', 'play_wave', 'wait', 'capture']
+
+Frame = TypeVar('Frame', bound=IRFrame)
+Reg = TypeVar('Reg')
+Envelope = TypeVar('Envelope', bound=IREnvelope)
+ArbWave = TypeVar('ArbWave', bound=IRArbWaveform)
+Placeholder = TypeVar('Placeholder', bound=IRBase)
+Var = TypeVar('Var')
+Int = TypeVar('Int')
+Uint = TypeVar('Uint')
+Float = TypeVar('Float')
+
+def init_frame(freq: float, phase: float, *, idx: int = None) -> IRFrame: ...
+def ins_envelope(envelope: np.ndarray | complex) -> IREnvelope: ...
+def init_arbwave(wlist: dict[str, np.ndarray], arb_name: str) -> IRArbWaveform: ...
+def evlp_gaussian(width: float, srate: float = 8000000000.0, amp: int = 1) -> IREnvelope: ...
+def evlp_cospulse(width: float, srate: float = 8000000000.0, amp: int = 1) -> IREnvelope: ...
+def evlp_square(width: float, srate: float = 8000000000.0, amp: int = 1) -> IREnvelope: ...
+def reset_frame() -> Placeholder: ...
+def wait_for_trigger() -> Placeholder: ...
+def inc_phase(phase: float) -> Placeholder: ...
+def inc_frequency(frame_list: Iterable[Frame], freq: float) -> Placeholder: ...
+def play_wave(envelope: IREnvelope, amp: int = 1, freq: int = 0, phase: int = 0) -> Placeholder: ...
+def play_arb(wave: IRArbWaveform): ...
+def wait(width) -> Placeholder: ...
+def capture(acq_width: float, delay_width: float, play_width: float, freq_list: Incomplete | None = None) -> Int: ...
+def set_judge(rotation: complex, threshold: int, freq_list: list = None) -> tuple[IRBase, IRBase]: ...
+def send(judge_reg) -> IRBase: ...
+def receive(feedback_rd_chnl: int = 0) -> IRBase: ...