Skip to content

BCDA-APS/guarneri

Repository files navigation

Guarneri Instrument Maker

PyPI Version

A package for creating Ophyd and Ophyd-async devices from configuration files.

Instead of instantiating devices directly in python, Guarneri reads a configuration file and creates/connects the devices for you. This provides the following benefits:

  1. Beamline configuration is in a human-readable configuration file (e.g. TOML).
  2. Other tools can modify the configuration file if needed.
  3. Devices can be connected in parallel (faster).
  4. Missing devices are handled gracefully.
  5. Devices can be simulated/mocked by changing a single value in the config file.

Using Guarneri

Let's say you have some ophyd and ophyd-async devices defined (with type hints) in a file called devices.py. This is not specific to guarneri, just regular Ophyd.

from ophyd_async.epics import epics_signal_rw
from ophyd_async.core import AsyncDevice
from ophyd import Device, Component

from guarneri import Instrument

class MyDevice(Device):
    description = Component(".DESC")


class MyAsyncDevice(AsyncDevice):
    def __init__(self, prefix: str, name: str = ""):
        self.description = epics_signal_rw(str, f"{prefix}.DESC")
        super().__init__(name=name)


def area_detector_factory(hdf: bool=True):
    # Create devices here using the arguments
    area_detector = ...
    return area_detector

Instead of instantiating these in a python startup script, Guarneri will let you create devices from a TOML configuration file. First we create a TOML file (e.g. instrument.toml) with the necessary parameters, these map directly onto the arguments of the device's __init__(), or the arguments of a factory that returns a device.

[[ my_device ]]
name = "device1"
prefix = '255id:'

[[ async_device ]]
name = "device3"
prefix = '255id:'

[[ area_detector ]]
name = "sim_det"
hdf = true

Then in your beamline startup code, create a Guarneri instrument and load the config files.

from io import StringIO

from guarneri import Instrument

from devices import MyDevice, MyAsyncDevice, area_detector_factory

# Prepare the instrument device
instrument = Instrument({
    "my_device": MyDevice,
    "async_device": MyAsyncDevice,
    "area_detector": area_detector_factory,
})

# Create the devices from the TOML configuration file
instrument.load_config_files("instrument.toml")
# Optionally connect all the devices
await instrument.connect()

# Now use the devices for science!
instrument.devices['device_1'].description.get()

The first argument to guarneri.Instrument.__init__() is a mapping of TOML section names to device classes. Guarneri then introspects the device or factory to decide which TOML keys and types are valid. In the above example, the heading [my_device.device1] will create an instance of MyDevice() with the name "device1" and prefix "255id:". This is equivalent to MyDevice(prefix="255id:", name="device1").

Using the Ophyd Registry inside the Guarneri

Guarneri also provides a way to keep track of and retrieve, Ophyd objects. That have been registered using guarneri. This allows for a simple way to keep track of the Ophyd devices that were created in your project.

registry = instrument.devices

# Then elsewhere in your project, use them...
registry['motor'].set(15.3)

Registering Devices without using

There are three ways to have an instrument registry know about a device.

  1. Implicitly capture all Ophyd objects
  2. Register a device class
  3. Register individual objects

By default, a new instrument registry will alert itself to all future devices:

from guarneri import Registry
registry = Registry()

the_device = MyDevice("255id:Dev:", name="my_device")

assert registry.find("my_device") is the_device

This greedy behavior can be suppressed with the auto_register parameter. It can also be turned off after initialization by setting the auto_register attribute to False:

registry = Registry(auto_register=False)

# Make a bunch of devices
...

# Turn if off for this one
registry.auto_register = False
device = MyDevice(...)
registry.auto_register = True

# Register some other devices maybe
...

If auto_register is false, then a device class can be decorated to allow the registry to find it:

from ophyd import Device
from guarneri import Registry

registry = Registry(auto_register=False)

@registry.register
class MyDevice(Device):
    ...

the_device = MyDevice("255id:Dev:", name="my_device")

assert registry.find("my_device") is the_device

Lastly, individual instantions of a device can be explicitly registered.

from ophyd import Device
from guarneri import Registry

registry = Registry(auto_register=False)

class MyDevice(Device):
    ...

the_device = MyDevice("255id:Dev:", name="my_device")
registry.register(the_device)

assert registry.find("my_device") is the_device

Looking Up Registered Devices/Components

Registered objects can be found by name, label, or both. The easist way is to treat the registry like a dictionary: registry['motor1']. This will look for an individual device first by name then by label. It will raise an exception if the number of devices is not 1.

For more sophisticated queries, the Registry.find() method will return a single result, while Registry.findall() returns more than one. By default, findall() will raise an exception if no objects match the criteria, but this can be overridden with the allow_none keyword argument.

The registry uses the built-in concept of device labels in Ophyd. The registry's find() and findall() methods allows devices to be looked up by label or device name. For example, assuming four devices exist with the label "ion_chambers", then these devices can be retrieved using the registry:

ion_chambers = registry.findall(label="ion_chambers")
assert len(ion_chambers) == 4

Devices can also be found by name:

ion_chambers = registry.find(name="I0")
assert len(ion_chambers) == 4

A set of the root devices, those without a parent, can be retrieved at Registry.root_devices.

Looking Up Sub-Components by Dot-Notation

For simple devices, the full name of the sub-component should be enough to retrieve the device. For example, to find the signal preset_time on the device named "vortex_me4", the following may work fine:

preset_time = haven.registry.find("vortex_me4_preset_time")

However, if the component is lazy and has not been accessed prior to being registered, then it will not be available in the registry. Sub-components can instead be accessed by dot notation. Unlike the full device name, dot-notation names only resolve when the component is requested from the registry, at which point the lazy components can be accessed.

For example, area detectors use many lazy components. If sim_det is an area detector with a camera component sim_det.cam, then the name of the gain channel is "sim_det_cam_gain", however this is a lazy component so is not available. Instead, retrieving the device by haven.registry.find("sim_det.cam.gain") will first find the area detector ("sim_det"), then access the cam attribute, and then cam's gain attribute. This has the side-effect of instantiating the lazy components.

Removing Devices

The OphydRegistry class behaves similarly to a python dictionary.

To remove all devices from the registry, use the clear() method:

registry.clear()

To remove disconnected devices from the registry, use the pop_disconnected() method with an optional timeout:

# Wait 5 seconds to give devices a chance to connect
disconnected_devices = registry.pop_disconnected(timeout=5)

To remove individual objects, use either the del keyword, or the pop() method. These approaches work with either the OphydObject instance itself, or the instance's name:

# Just delete the item and move on
# (by name)
del registry["motor1"]
# (by reference)
motor = registry['motor1']
del registry[motor]

# Remove the item and use it
# (return a simulated motor if "motor1" is not in the registry)
motor = registry.pop("motor1", ophyd.sim.motor)
motor.set(5).wait()

Keeping References

It may be useful to not actually keep a strong reference to the OphydObject s. This means that if all other references to the object are removed, the device may be dropped from the registry.

By default, the registry keeps direct references to the objects that get registered, but if initalized with keep_references=False the Registry will not keep these references. Instead, it is up to you to keep references to the registered objects.

# Create two registers with both referencing behaviors
ref_registry = Registry(keep_references=True)
noref_registry = Registry(keep_references=False)
motor = EpicsMotor(...)

# Check if we can get the motor (should be no problem)
ref_registry[motor.name]  # <- succeeds
noref_registry[motor.name]  # <- succeeds

# Delete the object and try again
del motor
gc.collect()  # <- make sure it's *really* gone

# Check again if we can get the motor (now it gets fun)
ref_registry[motor.name]  # <- succeeds
noref_registry[motor.name]  # <- raises ComponentNotFound

Integrating with Typhos

Typhos includes a PyDM plugin that can directly interact with ophyd devices. It requires ophyd objects to be registered in order to find them. ophyd_registry can automatically register devices with Typhos by simply passing the use_typhos argument when creating the registry:

from guarneri import Registry
registry = Registry(use_typos=True)

or setting the use_typhos attribute on an existing registry:

from guarneri import Registry
registry = Registry()
registry.use_typhos = True

If using the typhos registry, calling the clear() method on the ophyd registry will also clear the Typhos registry.

What About Happi?

Happi has a similar goal to Guarneri, but with a different scope. While Happi is meant for facility-level configuration (e.g. LCLS), Guarneri is aimed at individual beamlines at a synchrotron. Happi uses HappiItem classes with ItemInfo objects to describe the devices definitions, while Guarneri uses the device classes themselves. Happi provides a python client for adding and modifying the devices, while Guarneri uses human-readable configuration files.

Which one is better? Depends on what you're trying to do. If you want a flexible and scalable system that shares devices across a facility, try Happi. If you want a way to get devices running quickly on your beamline before users show up, try Guarneri.

Documentation

Sphinx-generated documentation for this project can be found here: https://spc-group.github.io/guarneri/

Installation

The following will download the package and load it into the python environment.

pip install guarneri

Development

git clone https://github.com/spc-group/guarneri

uv is preferred for managing guarneri. Run the tests (including dependencies) with

uv run --dev pytest

Build wheels with

uv build

Development (uv-free)

First, install the dependencies listed in dependency-groups.dev in pyproject.toml.

Then install an editable guarneri and run the tests with

pip install -e ".[dev]"
pytest

About

No description, website, or topics provided.

Resources

License

Contributing

Stars

Watchers

Forks

Packages

 
 
 

Contributors