nsqdriver 0.12.17__cp313-cp313-manylinux2014_x86_64.manylinux_2_17_x86_64.whl

nsqdriver/nswave/_ir.pyi

@@ -0,0 +1,283 @@
+import abc
+import numpy as np
+from _typeshed import Incomplete
+
+__all__ = ['AssemblyError', 'GenTagMixin', 'IRBase', 'PreIR', 'nsw_config', 'UnfixIR', 'IRFrameRst', 'IRWaitTrig', 'FixIR', 'OverlapIR', 'ReconstitutionIR', 'IRFrame', 'IREnvelope', 'IRNop', 'IRDelay', 'IRProt', 'IRPth', 'IRMov', 'IRJmov', 'IRNsw', 'IRNlw', 'IRPlaceholder', 'IRWaitTrig', 'IRJumpImmediate', 'IRPlayImm', 'IRCapture', 'IRWait', 'IRFrameAdd', 'IRAdd', 'IRPlayDDRImm', 'IRArbWaveform', 'LoadDDRIR', 'IRWaitDDR']
+
+def nsw_config(name, value) -> None: ...
+
+class AssemblyError(RuntimeError): ...
+
+class GenTagMixin:
+    @property
+    def generate_tag(self): ...
+
+class IRBase(GenTagMixin, abc.ABC, metaclass=abc.ABCMeta):
+    tag: Incomplete
+    real_wait_time: int
+    line: int
+    index_map: Incomplete
+    def __init__(self) -> None: ...
+    @property
+    @abc.abstractmethod
+    def overhead(self): ...
+    @property
+    @abc.abstractmethod
+    def wait_time(self): ...
+
+class PreIR(IRBase):
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class UnfixIR(IRBase):
+    ir_sub: Incomplete
+    irhead: int
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+    def fetch_ir(self, ir) -> None: ...
+
+class FixIR(IRBase):
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class LoadDDRIR(IRBase):
+    wait_trig: bool
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class OverlapIR(IRBase):
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class ReconstitutionIR(IRBase):
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class IRFrame(PreIR):
+    freq: Incomplete
+    phase: Incomplete
+    idx: Incomplete
+    real_wait_time: int
+    def __init__(self, freq, phase, idx) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class IRArbWaveform(PreIR):
+    waveform: Incomplete
+    wave_type: Incomplete
+    offset: int
+    step: Incomplete
+    def __init__(self, waveform: np.ndarray) -> None: ...
+
+class IREnvelope(PreIR):
+    envelop_slice: Incomplete
+    def __init__(self, envelope: np.ndarray | complex, amp: int = 1) -> None: ...
+    def __len__(self) -> int: ...
+    def __bytes__(self) -> bytes: ...
+
+class IRNop(PreIR):
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRProt(PreIR):
+    fre: Incomplete
+    rotation_real: Incomplete
+    rotation_imag: Incomplete
+    real_wait_time: int
+    def __init__(self, freq_list, rotation) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class IRPth(PreIR):
+    fre: Incomplete
+    threshold: Incomplete
+    real_wait_time: int
+    def __init__(self, freq_list, threshold) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self, *args): ...
+
+class IRWaitTrig(UnfixIR):
+    irhead: int
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+    @wait_time.setter
+    def wait_time(self, value) -> None: ...
+
+class IRJumpImmediate(UnfixIR):
+    idx: Incomplete
+    def __init__(self, idx: int) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRPlaceholder(UnfixIR):
+    irhead: int
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRJumpConditional(UnfixIR):
+    idx: Incomplete
+    result: Incomplete
+    irhead: int
+    jumpaddr: Incomplete
+    feedback_chnl: Incomplete
+    def __init__(self, register_idx, judge_result, feedback_chnl, jumpaddr: int = 0) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRJumpInside(OverlapIR):
+    irhead: int
+    jumpaddr: int
+    def __init__(self) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRPlayDDRImm(FixIR, LoadDDRIR):
+    wave: Incomplete
+    wait_trig: bool
+    def __init__(self, wave: IRArbWaveform) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRPlayImm(FixIR):
+    frame: Incomplete
+    freq: Incomplete
+    amp: Incomplete
+    bias: Incomplete
+    envelope: Incomplete
+    phase: Incomplete
+    def __init__(self, frame, envelope: IREnvelope, amp, bias, freq, phase) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRCapture(FixIR):
+    fre: Incomplete
+    acq_width: Incomplete
+    delay_width: Incomplete
+    play_width: Incomplete
+    para: Incomplete
+    def __init__(self, freq_list: list, acq_width: float, delay_width: float, play_width: float, raw_data_store: bool = True, iq_data_store: bool = True, judge_data_store: bool = True, double_fre_mode: bool = False) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRWait(FixIR):
+    width: Incomplete
+    def __init__(self, width) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRDelay(FixIR):
+    width: Incomplete
+    def __init__(self, width) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRFrameAdd(OverlapIR):
+    frames: Incomplete
+    frequency: Incomplete
+    phase: Incomplete
+    def __init__(self, frames: list, frequency: float, phase: float) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRFrameRst(OverlapIR):
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRAdd(ReconstitutionIR):
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRMov(PreIR):
+    idx: Incomplete
+    reg_value: Incomplete
+    def __init__(self, register_idx, value) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRNsw(OverlapIR):
+    idx: Incomplete
+    def __init__(self, register_idx) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRNlw(OverlapIR):
+    idx: int
+    feedback_rd_chnl: Incomplete
+    def __init__(self, feedback_rd_chnl) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+    def set_idx(self, register_idx) -> None: ...
+
+class IRJmov(OverlapIR):
+    idx: Incomplete
+    def __init__(self, register_idx) -> None: ...
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...
+
+class IRWaitDDR(OverlapIR):
+    @property
+    def overhead(self): ...
+    @property
+    def wait_time(self): ...

nsqdriver/nswave/_ir_pass.pyi

@@ -0,0 +1,7 @@
+from ._checkers import BaseChecker as BaseChecker
+from ._optimizations import BaseOptimizer as BaseOptimizer
+from ._rules import BaseRule as BaseRule
+from ._translate import NormalTranslator as NormalTranslator
+from .kernel import Kernel as Kernel
+
+def ir_pass(kernel: Kernel): ...

nsqdriver/nswave/_optimizations.pyi

@@ -0,0 +1,16 @@
+from _typeshed import Incomplete
+
+__all__ = ['BaseOptimizer', 'JumpOptimizer', 'PlyiAmpOptimizer']
+
+class BaseOptimizer:
+    ir_list: Incomplete
+    def __init__(self) -> None: ...
+    def optimize_ir(self, ir_list: list) -> list: ...
+
+class JumpOptimizer(BaseOptimizer):
+    def __init__(self) -> None: ...
+    def optimize_ir(self, ir_list: list) -> list: ...
+
+class PlyiAmpOptimizer(BaseOptimizer):
+    def __init__(self) -> None: ...
+    def optimize_ir(self, ir_list: list) -> list: ...

nsqdriver/nswave/_rules.pyi

@@ -0,0 +1,56 @@
+from ._ir import *
+from _typeshed import Incomplete
+
+__all__ = ['BaseRule', 'BranchRule', 'TimingRule', 'EnvelopeRule', 'AddJumpRule', 'WitiRule', 'PlaceHolder', 'PreConfigRule', 'WtgAlign', 'PlayDDRRule']
+
+class BaseRule:
+    ir_list: Incomplete
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list: list, **kwargs) -> list: ...
+
+class BranchRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class TimingRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list: list, **kwargs) -> list: ...
+    def insert_ir(self, ir_list_ordered) -> list: ...
+    def append_preir(self, ir): ...
+    def append_unfixir(self, ir): ...
+    def append_fixir(self, ir): ...
+    def append_overlapir(self, ir): ...
+    def append_reconstitutionir(self, ir): ...
+    def order_overlap(self, ir_list_ordered) -> list: ...
+
+class PreConfigRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class EnvelopeRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list: list, **kwargs): ...
+
+class AddJumpRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class JumpRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class WitiRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class WtgAlign(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class PlaceHolder(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...
+
+class PlayDDRRule(BaseRule):
+    def __init__(self) -> None: ...
+    def order_ins(self, ir_list, **kwargs) -> list: ...

nsqdriver/nswave/_translate.pyi

@@ -0,0 +1,12 @@
+import numpy as np
+
+__all__ = ['NormalTranslator', 'PlayDDRTranslator']
+
+class NormalTranslator:
+    def get_envelope(self, ir_list): ...
+    def translate_ir(self, ir_list: list) -> tuple[np.ndarray, str]: ...
+
+class PlayDDRTranslator(NormalTranslator):
+    @staticmethod
+    def get_arb(ir_list): ...
+    def translate_ir(self, ir_list: list) -> tuple[list[str], str]: ...

nsqdriver/nswave/kernel.pyi

@@ -0,0 +1,57 @@
+from ._functions import *
+from ._rules import *
+from ._checkers import *
+from ._optimizations import *
+from ._translate import *
+import ast
+from ._ir import IRBase
+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
+    jail_tag: Incomplete
+    compile_params: Incomplete
+    pass_params: 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_Compare(self, node: ast.Compare): ...
+    def visit_If(self, node: ast.If): ...
+    def visit_For(self, node: ast.For): ...
+    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 with_error(self, error: list): ...
+    def node_line(self, error_ir: IRBase): ...
+
+def kernel(func): ...