ophyd-async 0.1.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

ophyd_async/panda/panda.py

@@ -1,332 +0,0 @@
-from __future__ import annotations
-
-import atexit
-import re
-from enum import Enum
-from typing import (
-    Callable,
-    Dict,
-    FrozenSet,
-    Optional,
-    Sequence,
-    Tuple,
-    Type,
-    TypedDict,
-    cast,
-    get_args,
-    get_origin,
-    get_type_hints,
-)
-
-import numpy as np
-import numpy.typing as npt
-from p4p.client.thread import Context
-
-from ophyd_async.core import (
-    Device,
-    DeviceVector,
-    Signal,
-    SignalBackend,
-    SignalR,
-    SignalRW,
-    SignalX,
-    SimSignalBackend,
-)
-from ophyd_async.epics.signal import (
-    epics_signal_r,
-    epics_signal_rw,
-    epics_signal_w,
-    epics_signal_x,
-    pvi_get,
-)
-
-
-class PulseBlock(Device):
-    delay: SignalRW[float]
-    width: SignalRW[float]
-
-
-class SeqTrigger(Enum):
-    IMMEDIATE = "Immediate"
-    BITA_0 = "BITA=0"
-    BITA_1 = "BITA=1"
-    BITB_0 = "BITB=0"
-    BITB_1 = "BITB=1"
-    BITC_0 = "BITC=0"
-    BITC_1 = "BITC=1"
-    POSA_GT = "POSA>=POSITION"
-    POSA_LT = "POSA<=POSITION"
-    POSB_GT = "POSB>=POSITION"
-    POSB_LT = "POSB<=POSITION"
-    POSC_GT = "POSC>=POSITION"
-    POSC_LT = "POSC<=POSITION"
-
-
-class SeqTable(TypedDict):
-    repeats: npt.NDArray[np.uint16]
-    trigger: Sequence[SeqTrigger]
-    position: npt.NDArray[np.int32]
-    time1: npt.NDArray[np.uint32]
-    outa1: npt.NDArray[np.bool_]
-    outb1: npt.NDArray[np.bool_]
-    outc1: npt.NDArray[np.bool_]
-    outd1: npt.NDArray[np.bool_]
-    oute1: npt.NDArray[np.bool_]
-    outf1: npt.NDArray[np.bool_]
-    time2: npt.NDArray[np.uint32]
-    outa2: npt.NDArray[np.bool_]
-    outb2: npt.NDArray[np.bool_]
-    outc2: npt.NDArray[np.bool_]
-    outd2: npt.NDArray[np.bool_]
-    oute2: npt.NDArray[np.bool_]
-    outf2: npt.NDArray[np.bool_]
-
-
-class SeqBlock(Device):
-    table: SignalRW[SeqTable]
-
-
-class PcapBlock(Device):
-    active: SignalR[bool]
-
-
-class PVIEntry(TypedDict, total=False):
-    d: str
-    r: str
-    rw: str
-    w: str
-    x: str
-
-
-def _block_name_number(block_name: str) -> Tuple[str, Optional[int]]:
-    """Maps a panda block name to a block and number.
-
-    There are exceptions to this rule; some blocks like pcap do not contain numbers.
-    Other blocks may contain numbers and letters, but no numbers at the end.
-
-    Such block names will only return the block name, and not a number.
-
-    If this function returns both a block name and number, it should be instantiated
-    into a device vector."""
-    m = re.match("^([0-9a-z_-]*)([0-9]+)$", block_name)
-    if m is not None:
-        name, num = m.groups()
-        return name, int(num or 1)  # just to pass type checks.
-
-    return block_name, None
-
-
-def _remove_inconsistent_blocks(pvi_info: Dict[str, PVIEntry]) -> None:
-    """Remove blocks from pvi information.
-
-    This is needed because some pandas have 'pcap' and 'pcap1' blocks, which are
-    inconsistent with the assumption that pandas should only have a 'pcap' block,
-    for example.
-
-    """
-    pvi_keys = set(pvi_info.keys())
-    for k in pvi_keys:
-        kn = re.sub(r"\d*$", "", k)
-        if kn and k != kn and kn in pvi_keys:
-            del pvi_info[k]
-
-
-async def pvi(pv: str, ctxt: Context, timeout: float = 5.0) -> Dict[str, PVIEntry]:
-    result = await pvi_get(pv, ctxt, timeout=timeout)
-    _remove_inconsistent_blocks(result)
-    return result
-
-
-class PandA(Device):
-    _ctxt: Optional[Context] = None
-
-    pulse: DeviceVector[PulseBlock]
-    seq: DeviceVector[SeqBlock]
-    pcap: PcapBlock
-
-    def __init__(self, pv: str) -> None:
-        self._init_prefix = pv
-        self.pvi_mapping: Dict[FrozenSet[str], Callable[..., Signal]] = {
-            frozenset({"r", "w"}): lambda dtype, rpv, wpv: epics_signal_rw(
-                dtype, rpv, wpv
-            ),
-            frozenset({"rw"}): lambda dtype, rpv, wpv: epics_signal_rw(dtype, rpv, wpv),
-            frozenset({"r"}): lambda dtype, rpv, wpv: epics_signal_r(dtype, rpv),
-            frozenset({"w"}): lambda dtype, rpv, wpv: epics_signal_w(dtype, wpv),
-            frozenset({"x"}): lambda dtype, rpv, wpv: epics_signal_x(wpv),
-        }
-
-    @property
-    def ctxt(self) -> Context:
-        if PandA._ctxt is None:
-            PandA._ctxt = Context("pva", nt=False)
-
-            @atexit.register
-            def _del_ctxt():
-                # If we don't do this we get messages like this on close:
-                #   Error in sys.excepthook:
-                #   Original exception was:
-                PandA._ctxt = None
-
-        return PandA._ctxt
-
-    def verify_block(self, name: str, num: Optional[int]):
-        """Given a block name and number, return information about a block."""
-        anno = get_type_hints(self, globalns=globals()).get(name)
-
-        block: Device = Device()
-
-        if anno:
-            type_args = get_args(anno)
-            block = type_args[0]() if type_args else anno()
-
-            if not type_args:
-                assert num is None, f"Only expected one {name} block, got {num}"
-
-        return block
-
-    async def _make_block(
-        self, name: str, num: Optional[int], block_pv: str, sim: bool = False
-    ):
-        """Makes a block given a block name containing relevant signals.
-
-        Loops through the signals in the block (found using type hints), if not in
-        sim mode then does a pvi call, and identifies this signal from the pvi call.
-        """
-        block = self.verify_block(name, num)
-
-        field_annos = get_type_hints(block, globalns=globals())
-        block_pvi = await pvi(block_pv, self.ctxt) if not sim else None
-
-        # finds which fields this class actually has, e.g. delay, width...
-        for sig_name, sig_type in field_annos.items():
-            origin = get_origin(sig_type)
-            args = get_args(sig_type)
-
-            # if not in sim mode,
-            if block_pvi:
-                block_pvi = cast(Dict[str, PVIEntry], block_pvi)
-                # try to get this block in the pvi.
-                entry: Optional[PVIEntry] = block_pvi.get(sig_name)
-                if entry is None:
-                    raise Exception(
-                        f"{self.__class__.__name__} has a {name} block containing a/"
-                        + f"an {sig_name} signal which has not been retrieved by PVI."
-                    )
-
-                signal = self._make_signal(entry, args[0] if len(args) > 0 else None)
-
-            else:
-                backend: SignalBackend = SimSignalBackend(
-                    args[0] if len(args) > 0 else None, block_pv
-                )
-                signal = SignalX(backend) if not origin else origin(backend)
-
-            setattr(block, sig_name, signal)
-
-        # checks for any extra pvi information not contained in this class
-        if block_pvi:
-            for attr, attr_pvi in block_pvi.items():
-                if not hasattr(block, attr):
-                    # makes any extra signals
-                    signal = self._make_signal(attr_pvi)
-                    setattr(block, attr, signal)
-
-        return block
-
-    async def _make_untyped_block(self, block_pv: str):
-        """Populates a block using PVI information.
-
-        This block is not typed as part of the PandA interface but needs to be
-        included dynamically anyway.
-        """
-        block = Device()
-        block_pvi: Dict[str, PVIEntry] = await pvi(block_pv, self.ctxt)
-
-        for signal_name, signal_pvi in block_pvi.items():
-            signal = self._make_signal(signal_pvi)
-            setattr(block, signal_name, signal)
-
-        return block
-
-    def _make_signal(self, signal_pvi: PVIEntry, dtype: Optional[Type] = None):
-        """Make a signal.
-
-        This assumes datatype is None so it can be used to create dynamic signals.
-        """
-        operations = frozenset(signal_pvi.keys())
-        pvs = [signal_pvi[i] for i in operations]  # type: ignore
-        signal_factory = self.pvi_mapping[operations]
-
-        write_pv = pvs[0]
-        read_pv = write_pv if len(pvs) == 1 else pvs[1]
-
-        return signal_factory(dtype, "pva://" + read_pv, "pva://" + write_pv)
-
-    # TODO redo to set_panda_block? confusing name
-    def set_attribute(self, name: str, num: Optional[int], block: Device):
-        """Set a block on the panda.
-
-        Need to be able to set device vectors on the panda as well, e.g. if num is not
-        None, need to be able to make a new device vector and start populating it...
-        """
-        anno = get_type_hints(self, globalns=globals()).get(name)
-
-        # if it's an annotated device vector, or it isn't but we've got a number then
-        # make a DeviceVector on the class
-        if get_origin(anno) == DeviceVector or (not anno and num is not None):
-            self.__dict__.setdefault(name, DeviceVector())[num] = block
-        else:
-            setattr(self, name, block)
-
-    async def connect(self, sim=False) -> None:
-        """Initialises all blocks and connects them.
-
-        First, checks for pvi information. If it exists, make all blocks from this.
-        Then, checks that all required blocks in the PandA have been made.
-
-        If there's no pvi information, that's because we're in sim mode. In that case,
-        makes all required blocks.
-        """
-        pvi_info = await pvi(self._init_prefix + ":PVI", self.ctxt) if not sim else None
-        hints = {
-            attr_name: attr_type
-            for attr_name, attr_type in get_type_hints(self, globalns=globals()).items()
-            if not attr_name.startswith("_")
-        }
-
-        # create all the blocks pvi says it should have,
-        if pvi_info:
-            pvi_info = cast(Dict[str, PVIEntry], pvi_info)
-            for block_name, block_pvi in pvi_info.items():
-                name, num = _block_name_number(block_name)
-
-                if name in hints:
-                    block = await self._make_block(name, num, block_pvi["d"])
-                else:
-                    block = await self._make_untyped_block(block_pvi["d"])
-
-                self.set_attribute(name, num, block)
-
-        # then check if the ones defined in this class are in the pvi info
-        # make them if there is no pvi info, i.e. sim mode.
-        for block_name in hints.keys():
-            if pvi_info is not None:
-                pvi_name = block_name
-
-                if get_origin(hints[block_name]) == DeviceVector:
-                    pvi_name += "1"
-
-                entry: Optional[PVIEntry] = pvi_info.get(pvi_name)
-
-                assert entry, f"Expected PandA to contain {block_name} block."
-                assert list(entry) == [
-                    "d"
-                ], f"Expected PandA to only contain blocks, got {entry}"
-            else:
-                num = 1 if get_origin(hints[block_name]) == DeviceVector else None
-                block = await self._make_block(block_name, num, "sim://", sim=sim)
-                self.set_attribute(block_name, num, block)
-
-        self.set_name(self.name)
-        await super().connect(sim)

ophyd_async-0.1.0.dist-info/RECORD

@@ -1,45 +0,0 @@
-ophyd_async/__init__.py,sha256=WJoRU7gO-hRzyf7a-C952zF6-7zwPjP7qZ1Qu5GTUC8,124
-ophyd_async/__main__.py,sha256=G-Zcv_G9zK7Nhx6o5L5w-wyhMxdl_WgyMELu8IMFqAE,328
-ophyd_async/_version.py,sha256=IMl2Pr_Sy4LVRKy_Sm4CdwUl1Gryous6ncL96EMYsnM,411
-ophyd_async/core/__init__.py,sha256=8uqw8y6JNIjtrxo5dz1FPiFKJC_8yaY4opoGIAWNYyU,1353
-ophyd_async/core/async_status.py,sha256=oiqu57rti-nGNBaYn7bhXbCK3NyXgn5UNKie0jWHKxk,2735
-ophyd_async/core/utils.py,sha256=QxaBgmJLNGVlLYnaYq_Ia7SGWzfUHK6KISQs5BbnkUY,3086
-ophyd_async/core/_device/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ophyd_async/core/_device/device.py,sha256=9ikcXve_CxW45XSkvj0W1S3ZdhqrIGyeAmMV2y36xKo,1643
-ophyd_async/core/_device/device_collector.py,sha256=0HJrn1yRW4gP4YBHABKRiyiny749mEo_cev9KRRD4Gc,4143
-ophyd_async/core/_device/device_vector.py,sha256=5JUkMvYc6t477WxWzvRGCthtU2Y1ex-4sN5BOSAafqU,431
-ophyd_async/core/_device/standard_readable.py,sha256=CkJD5mE10Vl9JLiK2ANMLKu6YMBZ-Z_AhHoEtea3ZD4,2570
-ophyd_async/core/_device/_backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ophyd_async/core/_device/_backend/signal_backend.py,sha256=AK2QcROEyaMBpAQE5sRZspiZGKG0kUifjyQP1faCOlc,1190
-ophyd_async/core/_device/_backend/sim_signal_backend.py,sha256=R2Lcg6DrQsq49kdFY8Rd79jGS7zYZ9YlQOq66qfjDoM,5377
-ophyd_async/core/_device/_signal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ophyd_async/core/_device/_signal/signal.py,sha256=2kPPeJYp-sMiC6uNzpmoQgN0Ey5bkqou8Zf0uCJRvvA,10897
-ophyd_async/epics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ophyd_async/epics/_backend/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ophyd_async/epics/_backend/_aioca.py,sha256=cyB8G7smbF04VZJjVay69wij7x0FqXN2h4d672C4RhY,8320
-ophyd_async/epics/_backend/_p4p.py,sha256=ffphwOVXCFnTOUhK21l1rlqTRQP-PiwGw4dMhZJmrs8,10844
-ophyd_async/epics/areadetector/__init__.py,sha256=qBoVqsTArNiTSDeiKJslgFCch6dPTnjIhOKRfAWNi5A,584
-ophyd_async/epics/areadetector/ad_driver.py,sha256=7qaP0_T-NgFLi1ItVfMhTIhp40hKdUh7PBc0SdVChRA,776
-ophyd_async/epics/areadetector/directory_provider.py,sha256=61sxLihCdqGw6ppPgsgVBbXcrcSLC33A4pH-KnpzEeQ,419
-ophyd_async/epics/areadetector/hdf_streamer_det.py,sha256=DRDIv0fcycES-W56UAmfE82ovxzSlL2sPCnAYebmMmQ,6214
-ophyd_async/epics/areadetector/nd_file_hdf.py,sha256=Lo4mHnvcfYTARNNrhuXDVzq1KSAukzfb1dYPvVgwlmo,1033
-ophyd_async/epics/areadetector/nd_plugin.py,sha256=v3Ou4ay9FW1bW-Cwvp2W0o1nVF0jSiXP2PlFobx53S8,260
-ophyd_async/epics/areadetector/single_trigger_det.py,sha256=dA3GtYv36Zg1nmIZeZQptLNjSp8RTrcI4PhMoaFZCVo,1188
-ophyd_async/epics/areadetector/utils.py,sha256=ZsQF8f5evvpY6ubsTy1fib1hhtxbLoGS_AAsGXK5lZY,596
-ophyd_async/epics/demo/__init__.py,sha256=02gv5i1YyIKa6BVHUxeJdD90GZctezvjpIjsSXSJpEc,5471
-ophyd_async/epics/demo/mover.db,sha256=RFz0rxZue689Wh1sWTZwWeFMUrH04ttPq2u5xJH_Fp4,998
-ophyd_async/epics/demo/sensor.db,sha256=AVtiydrdtwAz2EFurO2Ult9SSRtre3r0akOBbL98LT0,554
-ophyd_async/epics/motion/__init__.py,sha256=tnmVRIwKa9PdN_xonJdAUD04UpEceh-hoD7XI62yDB0,46
-ophyd_async/epics/motion/motor.py,sha256=1xsVB5v1ydMReg9xuVmI-N0FbkcI26Ve4qUAurtdrpA,3368
-ophyd_async/epics/signal/__init__.py,sha256=M1dz74L_SM5Qf3GRuQVKn-1Pvs-CEOCkoxyg1J-etHM,232
-ophyd_async/epics/signal/_epics_transport.py,sha256=DEIL0iYUAWssysVEgWGu1fHSM1l-ATV2kjUgPtDN9LY,858
-ophyd_async/epics/signal/pvi_get.py,sha256=YTNQ9nXi1v8Rig54goc4_nFTxoH_wmF3yl75FzkdXy4,479
-ophyd_async/epics/signal/signal.py,sha256=XAf4LCokAVxO4caIO1fmN3EbchOcSM0jRG0ijz8ahnI,2543
-ophyd_async/panda/__init__.py,sha256=oP7hSaVNgFLIf3dkvJhrLxjw7yv791FBBYVxTZCRsiM,272
-ophyd_async/panda/panda.py,sha256=-n4nxE_-d3GXh8N4Z7Z7oX_clf0JYO_If8PbYHlQK64,10882
-ophyd_async-0.1.0.dist-info/LICENSE,sha256=pU5shZcsvWgz701EbT7yjFZ8rMvZcWgRH54CRt8ld_c,1517
-ophyd_async-0.1.0.dist-info/METADATA,sha256=lFGBmQ9p4j2iI-j2HoGLzo-SxlHrttb0C23dLb0zVdc,6989
-ophyd_async-0.1.0.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-ophyd_async-0.1.0.dist-info/entry_points.txt,sha256=O0YNJTEufO0w9BozXi-JurTy2U1_o0ypeCgJLQ727Jk,58
-ophyd_async-0.1.0.dist-info/top_level.txt,sha256=-hjorMsv5Rmjo3qrgqhjpal1N6kW5vMxZO3lD4iEaXs,12
-ophyd_async-0.1.0.dist-info/RECORD,,