From c2b53ecb4328805e9d0da02f04a99004dfeeb02d Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Wed, 13 Dec 2023 14:27:48 +0800
Subject: [PATCH] core: add option to trigger analyzer proxy at run end

---
 artiq/coredevice/core.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/artiq/coredevice/core.py b/artiq/coredevice/core.py
index 10af8e3dd..a29e6018b 100644
--- a/artiq/coredevice/core.py
+++ b/artiq/coredevice/core.py
@@ -86,7 +86,7 @@ class Core:
 
     def __init__(self, dmgr,
                  host, ref_period,
-                 analyzer_proxy=None,
+                 analyzer_proxy=None, analyze_at_run_end=False,
                  ref_multiplier=8,
                  target="rv32g", satellite_cpu_targets={}):
         self.ref_period = ref_period
@@ -99,6 +99,7 @@ class Core:
         else:
             self.comm = CommKernel(host)
         self.analyzer_proxy_name = analyzer_proxy
+        self.analyze_at_run_end = analyze_at_run_end
 
         self.first_run = True
         self.dmgr = dmgr
@@ -106,6 +107,10 @@ class Core:
         self.comm.core = self
         self.analyzer_proxy = None
 
+    def notify_run_end(self):
+        if self.analyze_at_run_end:
+            self.trigger_analyzer_proxy()
+
     def close(self):
         self.comm.close()