Add source and target on style

process-analytics · Oct 4, 2023 · 763f44e · 763f44e
1 parent a8383da
commit 763f44e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/component/mxgraph/renderer/StyleComputer.ts b/src/component/mxgraph/renderer/StyleComputer.ts
@@ -23,9 +23,11 @@ import { MessageVisibleKind, ShapeBpmnCallActivityKind, ShapeBpmnElementKind, Sh
 import { AssociationFlow, SequenceFlow } from '../../../model/bpmn/internal/edge/flows';
 import Shape from '../../../model/bpmn/internal/shape/Shape';
 import {
+  ShapeBpmnIntermediateThrowEvent,
   ShapeBpmnActivity,
   ShapeBpmnBoundaryEvent,
   ShapeBpmnCallActivity,
+  ShapeBpmnIntermediateCatchEvent,
   ShapeBpmnEvent,
   ShapeBpmnEventBasedGateway,
   ShapeBpmnStartEvent,
@@ -101,6 +103,12 @@ export default class StyleComputer {
     if (bpmnElement instanceof ShapeBpmnBoundaryEvent || (bpmnElement instanceof ShapeBpmnStartEvent && bpmnElement.isInterrupting !== undefined)) {
       styleValues.set(BpmnStyleIdentifier.IS_INTERRUPTING, String(bpmnElement.isInterrupting));
     }
+
+    if (bpmnElement instanceof ShapeBpmnIntermediateCatchEvent) {
+      styleValues.set(BpmnStyleIdentifier.LINK_EVENT_SOURCE_IDS, String(bpmnElement.sourceIds));
+    } else if (bpmnElement instanceof ShapeBpmnIntermediateThrowEvent) {
+      styleValues.set(BpmnStyleIdentifier.LINK_EVENT_TARGET_ID, bpmnElement.targetId);
+    }
   }
 
   private static computeActivityShapeStyle(bpmnElement: ShapeBpmnActivity, styleValues: Map<string, string | number>): void {

diff --git a/src/component/mxgraph/style/identifiers.ts b/src/component/mxgraph/style/identifiers.ts
@@ -39,6 +39,10 @@ export const BpmnStyleIdentifier = {
   IS_INSTANTIATING: 'bpmn.isInstantiating',
   IS_INTERRUPTING: 'bpmn.isInterrupting',
 
+  // link event
+  LINK_EVENT_SOURCE_IDS: 'bpmn.linkEventSourceIds',
+  LINK_EVENT_TARGET_ID: 'bpmn.linkEventTargetId',
+
   // other identifiers
   EXTRA_CSS_CLASSES: 'bpmn.extra.css.classes',
   MARKERS: 'bpmn.markers',