Merge pull request #93 from python-accelerator-middle-layer/add-post-init-hook

Added post_init() hook

Author: JeanLucPons

pyaml/accelerator.py

@@ -74,6 +74,8 @@ def __init__(self, cfg: ConfigModel):
         if cfg.energy is not None:
             self.set_energy(cfg.energy)
 
+        self.post_init()
+
     def set_energy(self, E: float):
         if self._cfg.simulators is not None:
             for s in self._cfg.simulators:
@@ -82,6 +84,17 @@ def set_energy(self, E: float):
             for c in self._cfg.controls:
                 c.set_energy(E)
 
+    def post_init(self):
+        """
+        Method triggered after all initialisations are done
+        """
+        if self._cfg.simulators is not None:
+            for s in self._cfg.simulators:
+                s.post_init()
+        if self._cfg.controls is not None:
+            for c in self._cfg.controls:
+                c.post_init()
+
     @property
     def live(self) -> ControlSystem:
         return self.__live

pyaml/common/element.py

@@ -78,5 +78,11 @@ def get_peer(self) -> str:
             else f"{self._peer.__class__.__name__}:{self._peer.name()}"
         )
 
+    def post_init(self):
+        """
+        Method triggered after all initialisations are done
+        """
+        pass
+
     def __repr__(self):
         return __pyaml_repr__(self)

pyaml/common/element_holder.py

@@ -40,6 +40,13 @@ def __init__(self):
         self.__BPM_ARRAYS: dict = {}
         self.__ELEMENT_ARRAYS: dict = {}
 
+    def post_init(self):
+        """
+        Method triggered after all initialisations are done
+        """
+        for e in self.get_all_elements():
+            e.post_init()
+
     def fill_device(self, elements: list[Element]):
         raise PyAMLException("ElementHolder.fill_device() is not subclassed")
 

pyaml/tuning_tools/tune.py

@@ -179,9 +179,6 @@ def readback(self):
         Return the betatron tune measurement
         """
         self.check_peer()
-        if not self.__tm:
-            # Lazily attach the tune monitor
-            self.__tm = self._peer.get_betatron_tune_monitor(self._cfg.betatron_tune)
         return self.__tm.tune.get()
 
     def set(self, tune: np.array, iter: int = 1, wait_time: float = 0.0):
@@ -211,6 +208,10 @@ def set(self, tune: np.array, iter: int = 1, wait_time: float = 0.0):
     def response(self) -> TuneResponse:
         return self.__tr
 
+    def post_init(self):
+        self.check_peer()
+        self.__tm = self._peer.get_betatron_tune_monitor(self._cfg.betatron_tune)
+
     def attach(
         self,
         peer: "ElementHolder",