Skip to content

Commit 12cc93a

Browse files
committed
Add setActive function to TeapotTuneAnalysisNode
1 parent e346558 commit 12cc93a

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

py/orbit/diagnostics/TeapotDiagnosticsNode.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,13 +203,20 @@ def __init__(self, name: str = "tuneanalysis no name") -> None:
203203
self.lattlength = 0.0
204204
self.setLength(0.0)
205205
self.position = 0.0
206+
self.active = True
206207

207208
def track(self, paramsDict: dict) -> None:
208209
"""Implementation of the AccNodeBunchTracker class track(probe) method."""
210+
if not self.active:
211+
return
212+
209213
length = self.getLength(self.getActivePartIndex())
210214
bunch = paramsDict["bunch"]
211215
self.bunchtune.analyzeBunch(bunch)
212216

217+
def setActive(self, active: bool) -> None:
218+
self.active = active
219+
213220
def setPosition(self, position: float) -> None:
214221
self.position = position
215222

0 commit comments

Comments
 (0)