Skip to content

Commit

Permalink
Support numpy floating and integer dtypes in SimConverter
Browse files Browse the repository at this point in the history
  • Loading branch information
jsouter authored and root committed Apr 19, 2024
1 parent f78780d commit fd480a7
Show file tree
Hide file tree
Showing 5 changed files with 26 additions and 27 deletions.
22 changes: 10 additions & 12 deletions src/ophyd_async/core/signal.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,9 +60,7 @@ def set_name(self, name: str = ""):

async def connect(self, sim=False, timeout=DEFAULT_TIMEOUT):
if sim:
self._backend = SimSignalBackend(
datatype=self._init_backend.datatype
)
self._backend = SimSignalBackend(datatype=self._init_backend.datatype)
_sim_backends[self] = self._backend
else:
self._backend = self._init_backend
Expand Down Expand Up @@ -254,30 +252,30 @@ def set_sim_callback(signal: Signal[T], callback: ReadingValueCallback[T]) -> No


def soft_signal_rw(
datatype: Optional[Type[T]],
name: str,
datatype: Optional[Type[T]] = None,
initial_value: Optional[T] = None,
name: Optional[str] = None,
) -> SignalRW[T]:
"""Creates a read-writable Signal with a SimSignalBackend"""
signal = SignalRW(
SimSignalBackend(datatype, initial_value)
)
signal.set_name(name)
signal = SignalRW(SimSignalBackend(datatype, initial_value))
if name is not None:
signal.set_name(name)
return signal


def soft_signal_r_and_backend(
datatype: Optional[Type[T]],
name: str,
datatype: Optional[Type[T]] = None,
initial_value: Optional[T] = None,
name: Optional[str] = None,
) -> Tuple[SignalR[T], SimSignalBackend]:
"""Returns a tuple of a read-only Signal and its SimSignalBackend through
which the signal can be internally modified within the device. Use
soft_signal_rw if you want a device that is externally modifiable
"""
backend = SimSignalBackend(datatype, initial_value)
signal = SignalR(backend)
signal.set_name(name)
if name is not None:
signal.set_name(name)
return (signal, backend)


Expand Down
12 changes: 9 additions & 3 deletions src/ophyd_async/core/sim_signal_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
from enum import Enum
from typing import Any, Dict, Generic, Optional, Type, Union, cast, get_origin

import numpy as np
from bluesky.protocols import Descriptor, Dtype, Reading

from .signal_backend import SignalBackend
Expand Down Expand Up @@ -36,11 +37,16 @@ def reading(self, value: T, timestamp: float, severity: int) -> Reading:
)

def descriptor(self, source: str, value) -> Descriptor:
dtype = type(value)
if np.issubdtype(dtype, np.integer):
dtype = int
elif np.issubdtype(dtype, np.floating):
dtype = float
assert (
type(value) in primitive_dtypes
dtype in primitive_dtypes
), f"invalid converter for value of type {type(value)}"
dtype = primitive_dtypes[type(value)]
return {"source": source, "dtype": dtype, "shape": []}
dtype_name = primitive_dtypes[dtype]
return {"source": source, "dtype": dtype_name, "shape": []}

def make_initial_value(self, datatype: Optional[Type[T]]) -> T:
if datatype is None:
Expand Down
4 changes: 2 additions & 2 deletions tests/core/test_signal.py
Original file line number Diff line number Diff line change
Expand Up @@ -132,9 +132,9 @@ async def test_create_soft_signal(signal_method, signal_class):
SIGNAL_NAME = "TEST-PREFIX:SIGNAL"
INITIAL_VALUE = "INITIAL"
if signal_method == soft_signal_r_and_backend:
signal, backend = signal_method(str, SIGNAL_NAME, INITIAL_VALUE)
signal, backend = signal_method(str, INITIAL_VALUE, SIGNAL_NAME)
elif signal_method == soft_signal_rw:
signal = signal_method(str, SIGNAL_NAME, INITIAL_VALUE)
signal = signal_method(str, INITIAL_VALUE, SIGNAL_NAME)
backend = signal._backend
assert signal.source == f"soft://{SIGNAL_NAME}"
assert isinstance(signal, signal_class)
Expand Down
7 changes: 3 additions & 4 deletions tests/core/test_sim.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,10 +100,9 @@ async def test_backend_get_put_monitor(
try:
# Check descriptor
source = "soft://test"
assert (
dict(source=source, **descriptor(initial_value))
== await backend.get_descriptor(source)
)
assert dict(
source=source, **descriptor(initial_value)
) == await backend.get_descriptor(source)
# Check initial value
await q.assert_updates(
pytest.approx(initial_value) if initial_value != "" else initial_value
Expand Down
8 changes: 2 additions & 6 deletions tests/panda/test_writer.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,12 +69,8 @@ async def sim_writer(tmp_path, sim_panda) -> PandaHDFWriter:
async def test_get_capture_signals_gets_all_signals(sim_panda):
async with DeviceCollector(sim=True):
sim_panda.test_seq = Device("seq")
sim_panda.test_seq.seq1_capture = SignalR(
backend=SimSignalBackend(str)
)
sim_panda.test_seq.seq2_capture = SignalR(
backend=SimSignalBackend(str)
)
sim_panda.test_seq.seq1_capture = SignalR(backend=SimSignalBackend(str))
sim_panda.test_seq.seq2_capture = SignalR(backend=SimSignalBackend(str))
await asyncio.gather(
sim_panda.test_seq.connect(),
sim_panda.test_seq.seq1_capture.connect(),
Expand Down

0 comments on commit fd480a7

Please sign in to comment.