Source code for siriushla.as_ps_control.control_widget.TrimAllControlWidget

"""Widgets to control Trim Quadrupoles magnets."""
from .BasePSControlWidget import BasePSControlWidget




class SITrimAllControlWidget(BasePSControlWidget):
    """Class to control a Fam Quad from the Sirius section."""

    def _getFilter(self, subsection=None):
        filt = {"sec": "SI", "sub": "\w{4}", "dev": "Q(D|F|[1-4]).*"}
        if subsection:
            filt.update({'sub': subsection})
        return filt

    def _hasTrimButton(self):
        return True

    def _hasScrollArea(self):
        return True

    def _getGroups(self):
        return [('Focusing Quadrupoles Trims', "-QF"),
                ('Defocusing Quadrupoles Trims', "-QD"),
                ('Dispersive Quadrupoles Trims', "-Q[1-4]")]