plot observables

Author: lfarv

docs/p/api/at.latticetools.response_matrix.rst

@@ -36,8 +36,7 @@ Sub-packages
    howto/multiprocessing
    howto/CavityControl
    howto/Collective
-   Work with MAD-X files <api/at.load.madx>
-   Use response matrices <notebooks/response_matrices>
+   Working with MAD-X files <api/at.load.madx>
 
 .. autosummary::
    :toctree: api

docs/p/index.rst

@@ -936,7 +936,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.11"
+   "version": "3.9.20"
   }
  },
  "nbformat": 4,

docs/p/notebooks/observables.ipynb

@@ -105,32 +105,6 @@ class RefptsCode(Enum):
 End = RefptsCode.End
 
 
-def _chkattr(attrname: str, el):
-    return hasattr(el, attrname)
-
-
-def _chkattrval(attrname: str, attrvalue, el):
-    try:
-        v = getattr(el, attrname)
-    except AttributeError:
-        return False
-    else:
-        return v == attrvalue
-
-
-def _chkpattern(pattern: str, el):
-    return fnmatch(el.FamName, pattern)
-
-
-def _chkregex(pattern: str, el):
-    rgx = re.compile(pattern)
-    return rgx.fullmatch(el.FamName)
-
-
-def _chktype(eltype: type, el):
-    return isinstance(el, eltype)
-
-
 def _type_error(refpts, types):
     if isinstance(refpts, numpy.ndarray):
         tp = refpts.dtype.type
@@ -641,10 +615,14 @@ def checkattr(attrname: str, attrvalue: Optional = None) \
         Returns an iterator over all elements in ring that have a
         :pycode:`K` attribute equal to 0.0
     """
-    if attrvalue is None:
-        return functools.partial(_chkattr, attrname)
-    else:
-        return functools.partial(_chkattrval, attrname, attrvalue)
+    def testf(el):
+        try:
+            v = getattr(el, attrname)
+            return (attrvalue is None) or (v == attrvalue)
+        except AttributeError:
+            return False
+
+    return testf
 
 
 def checktype(eltype: Union[type, Tuple[type, ...]]) -> ElementFilter:
@@ -668,7 +646,7 @@ def checktype(eltype: Union[type, Tuple[type, ...]]) -> ElementFilter:
 
         Returns an iterator over all quadrupoles in ring
     """
-    return functools.partial(_chktype, eltype)
+    return lambda el: isinstance(el, eltype)
 
 
 def checkname(pattern: str, regex: bool = False) -> ElementFilter:
@@ -696,9 +674,10 @@ def checkname(pattern: str, regex: bool = False) -> ElementFilter:
         Returns an iterator over all with name starting with ``QF``.
     """
     if regex:
-        return functools.partial(_chkregex, pattern)
+        rgx = re.compile(pattern)
+        return lambda el: rgx.fullmatch(el.FamName)
     else:
-        return functools.partial(_chkpattern, pattern)
+        return lambda el: fnmatch(el.FamName, pattern)
 
 
 def refpts_iterator(ring: Sequence[Element], refpts: Refpts,

docs/p/notebooks/response_matrices.ipynb

@@ -403,12 +403,6 @@ class VariableList(list):
     appending, insertion or concatenation with the "+" operator.
     """
 
-    def __getitem__(self, index):
-        if isinstance(index, slice):
-            return VariableList(super().__getitem__(index))
-        else:
-            return super().__getitem__(index)
-
     def get(self, ring=None, **kwargs) -> Sequence[float]:
         r"""Get the current values of Variables
 
@@ -462,9 +456,3 @@ def __str__(self) -> str:
     def deltas(self) -> Sequence[Number]:
         """delta values of the variables"""
         return np.array([var.delta for var in self])
-
-    @deltas.setter
-    def deltas(self, value: Number | Sequence[Number]) -> None:
-        deltas = np.broadcast_to(value, len(self))
-        for var, delta in zip(self, deltas):
-            var.delta = delta

pyat/at/lattice/utils.py

@@ -1,6 +1,5 @@
-"""Defines classes for modifying a lattice and observing its parameters."""
+"""Defines classes for modifying a lattice and observing its parameters"""
 
 from .observables import *
 from .observablelist import *
-# from .matching import *
-from .response_matrix import *
+from .matching import *