ophyd-async 0.2.0__py3-none-any.whl → 0.3a2__py3-none-any.whl

ophyd_async/epics/signal/pvi_get.py

@@ -1,22 +0,0 @@
-from typing import Dict, TypedDict
-
-from p4p.client.asyncio import Context
-
-
-class PVIEntry(TypedDict, total=False):
-    d: str
-    r: str
-    rw: str
-    w: str
-    x: str
-
-
-async def pvi_get(pv: str, ctxt: Context, timeout: float = 5.0) -> Dict[str, PVIEntry]:
-    pv_info = ctxt.get(pv, timeout=timeout).get("pvi").todict()
-
-    result = {}
-
-    for attr_name, attr_info in pv_info.items():
-        result[attr_name] = PVIEntry(**attr_info)  # type: ignore
-
-    return result

ophyd_async/panda/panda.py

@@ -1,294 +0,0 @@
-from __future__ import annotations
-
-import atexit
-import re
-from typing import (
-    Callable,
-    Dict,
-    FrozenSet,
-    Optional,
-    Tuple,
-    Type,
-    TypedDict,
-    cast,
-    get_args,
-    get_origin,
-    get_type_hints,
-)
-
-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,
-)
-from ophyd_async.panda.table import SeqTable
-
-
-class PulseBlock(Device):
-    delay: SignalRW[float]
-    width: SignalRW[float]
-
-
-class SeqBlock(Device):
-    table: SignalRW[SeqTable]
-    active: SignalRW[bool]
-
-
-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.2.0.dist-info/RECORD

@@ -1,53 +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=H-qsvrxCpdhaQzyddR-yajEqI71hPxLa4KxzpP3uS1g,411
-ophyd_async/core/__init__.py,sha256=Us5DMr4ys-P4_zKWs6rN-J8QipWpOjAsmF5Cu0BY9k8,2188
-ophyd_async/core/_providers.py,sha256=sgEv6MabSKlrFwB8TEoTU7qKiQ6D1YsO33q2E3oKK_M,940
-ophyd_async/core/async_status.py,sha256=j3sE2zCvfg0GZWVY1NCEhdkdrqGhwtoDqwX_SVnFhbA,2765
-ophyd_async/core/detector.py,sha256=seeokcIRBj-maUuODUOKxesXRJjZLuEhOh6BzDmdIQI,6375
-ophyd_async/core/device.py,sha256=4PCvf7MU91CiWvJf0ERWEWE5wu7XPNrUND1ScxzMOXY,5958
-ophyd_async/core/device_save_loader.py,sha256=oAxWleSSOUD5TrnZWjOOmsVEfk2dE0cOvdwLTkEifvQ,6935
-ophyd_async/core/flyer.py,sha256=n3lqcvfXJiZooXXJocvaN6Wzi3wsqIPKzG4jD5veO7o,9970
-ophyd_async/core/signal.py,sha256=2szVlaxdZwc5z8gtnH8Yaol1jQIMrNMwhNWlvdZtwwI,11589
-ophyd_async/core/signal_backend.py,sha256=brD1MTug_AkNj3hDkhz6CxG5ygE1LF4AS9YVWCMFI0M,1312
-ophyd_async/core/sim_signal_backend.py,sha256=XZ2NCgoLnVPJyqwhRRhuUjY4TRXrVl6zH-5UAVLJGvY,5539
-ophyd_async/core/standard_readable.py,sha256=9cBetIYttAho-7wOB3T1YgSJy0iWEFTAdn9jZkrGvLA,2560
-ophyd_async/core/utils.py,sha256=uxmKSbLsRizna482Zsl2q14H_jiM3JksqHCFkp_QYIM,3193
-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=izEUbxHKRLalsbcfybmMKUHw7NvDOvST4lqzIRxlOwg,9033
-ophyd_async/epics/_backend/_p4p.py,sha256=vrCjX2W6mkghZfyCFI_i5VLhG3380nyW1-HiupzRjKQ,11001
-ophyd_async/epics/areadetector/__init__.py,sha256=oEOzL7gEVVSULeT2WkFrva8lUKeX0zmQD0tmwFz6L24,325
-ophyd_async/epics/areadetector/single_trigger_det.py,sha256=q5mG-OUVagIjvXLb28lsrGj4eUSoH2pNW2rT4rQR8fA,1206
-ophyd_async/epics/areadetector/utils.py,sha256=dez54oElIkGMnhSM9qghToUB1opSqjdWTV2vhHCgRMA,3133
-ophyd_async/epics/areadetector/controllers/__init__.py,sha256=UG2-M5d2ykp2T8isQJCbAsGZF1aH0BtC_OPlzzPTjnA,149
-ophyd_async/epics/areadetector/controllers/ad_sim_controller.py,sha256=Q5GT7nzXWL3MxJ25b6rab_qft-XQtwGEfrGG9dNKJcA,1591
-ophyd_async/epics/areadetector/controllers/pilatus_controller.py,sha256=n9Gkq2RtPnTzdoqpKArvod92Nu1Jo7XHJBGUkZr-N2s,1556
-ophyd_async/epics/areadetector/drivers/__init__.py,sha256=AOpIEYfoBhG9Nc4-SId99v4PpyEh4_RBXfNaqiXlwUI,315
-ophyd_async/epics/areadetector/drivers/ad_base.py,sha256=ikfyNcZwJa5ah52DckjrBzkMMT_eDY1smM4XWfb6A6E,3689
-ophyd_async/epics/areadetector/drivers/pilatus_driver.py,sha256=cn5WNz913UOnOttw2bssjV2Bo3p9SuJma3ckRyCdvw8,442
-ophyd_async/epics/areadetector/writers/__init__.py,sha256=tpPcrYd1hs8WS7C0gmCnR2EBwjE5RzCljI7WwZ2V_LM,191
-ophyd_async/epics/areadetector/writers/_hdfdataset.py,sha256=E0C9VgsPyY35h7k0mvcIhjsIVNavApLxizqNWlM388w,167
-ophyd_async/epics/areadetector/writers/_hdffile.py,sha256=bFVDlVIMXANtE146DCm4uVbnYaibQM8Qzoa32XcwMJg,1373
-ophyd_async/epics/areadetector/writers/hdf_writer.py,sha256=_JAU6CSOigxi7SW6Wbu1jX7__8U-03HkjaPLF5VfHfQ,4666
-ophyd_async/epics/areadetector/writers/nd_file_hdf.py,sha256=rutCstILCGGwhP5pH_2lWM2QUcZ88-uxx5dTZIJUMWQ,1562
-ophyd_async/epics/areadetector/writers/nd_plugin.py,sha256=l0yBBEazviyFsWJv_4_sfGn_YM_Iyd0_SlMdAmUlXDU,871
-ophyd_async/epics/demo/__init__.py,sha256=DbVO4ufJWjQnZteilW8SBs5A8DN-Xajn0YibM0q8UkE,5500
-ophyd_async/epics/demo/demo_ad_sim_detector.py,sha256=06y65yvaqXvL2rDocjYyLz9kTVzuwV-LeuPhEfExdOA,944
-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=lQcA3PSPxA6XOnk8y5VuJGnKTq2VjkcO6ldCrssGr3M,3392
-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=7GnGa4CgFgTwyHeh4wYOJ2GEWwdXsC6vHD3z9LsaHaM,2543
-ophyd_async/panda/__init__.py,sha256=1fyb-sP56K5zDmhKrn97YzMe7hL7LscHGgogqkGEobU,420
-ophyd_async/panda/panda.py,sha256=ThNzYarMlmyLqrUsDo-vjd-IxYLxnqKDIwvcBKWzED0,9901
-ophyd_async/panda/table.py,sha256=dLoRP4zYNOkD_s0Vkp2wVYAwkjVG8nNdf8-FaXOTfPo,5655
-ophyd_async-0.2.0.dist-info/LICENSE,sha256=pU5shZcsvWgz701EbT7yjFZ8rMvZcWgRH54CRt8ld_c,1517
-ophyd_async-0.2.0.dist-info/METADATA,sha256=hF1Gj4qjAmUc2eeaX3RA8awW41LYrkSieBJ5y36pqsI,7197
-ophyd_async-0.2.0.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-ophyd_async-0.2.0.dist-info/entry_points.txt,sha256=O0YNJTEufO0w9BozXi-JurTy2U1_o0ypeCgJLQ727Jk,58
-ophyd_async-0.2.0.dist-info/top_level.txt,sha256=-hjorMsv5Rmjo3qrgqhjpal1N6kW5vMxZO3lD4iEaXs,12
-ophyd_async-0.2.0.dist-info/RECORD,,