Add name filter to attribute setter on Segment

CHANGELOG.md

@@ -6,6 +6,8 @@
 
 ### 🚀 Features
 
+- Add `filter_name` argument to the attribute setting convenience function of Segment, so it's possible to filter not only by type but also by name. (see #598) (@jank324)
+
 ### 🐛 Bug fixes
 
 ### 🐆 Other

cheetah/accelerator/segment.py

@@ -587,25 +587,32 @@ def get_beam_attrs_along_segment(
     def set_attrs_on_every_element(
         self,
         filter_type: type[Element] | tuple[type[Element]] | None = None,
+        filter_name: str | None = None,
         is_recursive: bool = True,
         **kwargs: dict[str, Any],
     ) -> None:
         """
         Set attributes on every element of a specific type in the segment.
 
         :param filter_type: Type of the elements to set the attributes for.
+        :param filter_name: Names of the elements to set the attributes for.
         :param is_recursive: If `True`, the this method is applied to nested `Segment`s
             as well. If `False`, only the elements directly in the top-level `Segment`
             are considered.
         :param kwargs: Attributes to set and their values.
         """
         for element in self.elements:
-            if filter_type is None or isinstance(element, filter_type):
+            if (filter_type is None or isinstance(element, filter_type)) and (
+                filter_name is None or element.name == filter_name
+            ):
                 for key, value in kwargs.items():
                     setattr(element, key, value)
             elif is_recursive and isinstance(element, Segment):
                 element.set_attrs_on_every_element(
-                    filter_type=filter_type, is_recursive=True, **kwargs
+                    filter_type=filter_type,
+                    filter_name=filter_name,
+                    is_recursive=True,
+                    **kwargs,
                 )
 
     def plot(