From 02f965ae6e23db79807cb58528e3a0d46ad4969e Mon Sep 17 00:00:00 2001
From: Soeren Domroes <sdo@informatik.uni-kiel.de>
Date: Wed, 24 Apr 2024 17:55:23 -0700
Subject: [PATCH] Set annotated layout options for modes.

---
 .../java/org/lflang/diagram/synthesis/util/ModeDiagrams.java    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/core/src/main/java/org/lflang/diagram/synthesis/util/ModeDiagrams.java b/core/src/main/java/org/lflang/diagram/synthesis/util/ModeDiagrams.java
index 18c13e2b1f..d6d9826ab2 100644
--- a/core/src/main/java/org/lflang/diagram/synthesis/util/ModeDiagrams.java
+++ b/core/src/main/java/org/lflang/diagram/synthesis/util/ModeDiagrams.java
@@ -129,6 +129,7 @@ public class ModeDiagrams extends AbstractSynthesisExtensions {
   @Inject @Extension private LinguaFrancaStyleExtensions _linguaFrancaStyleExtensions;
   @Inject @Extension private UtilityExtensions _utilityExtensions;
   @Inject @Extension private LayoutPostProcessing _layoutPostProcessing;
+  @Inject @Extension private LinguaFrancaSynthesis synthesis;
 
   @Extension private KRenderingFactory _kRenderingFactory = KRenderingFactory.eINSTANCE;
 
@@ -340,6 +341,7 @@ public void handleModes(List<KNode> nodes, ReactorInstance reactor) {
             modeContainer.getProperty(LayeredOptions.SPACING_NODE_NODE_BETWEEN_LAYERS)
                 + (getBooleanValue(SHOW_TRANSITION_LABELS) ? 6.0 : 10.0));
       }
+      synthesis.setAnnotatedLayoutOptions(reactor.reactorDefinition, modeContainer);
 
       var modeContainerPorts = new HashMap<KPort, KPort>();
       for (var mode : reactor.modes) {