feat: dot visualization

5 files changed, 358 insertions, 24 deletions

diff --git a/subprojects/frontend/src/editor/EditorStore.ts b/subprojects/frontend/src/editor/EditorStore.ts
index 563725bb..10f01099 100644
--- a/subprojects/frontend/src/editor/EditorStore.ts
+++ b/subprojects/frontend/src/editor/EditorStore.ts
@@ -28,6 +28,7 @@ import { nanoid } from 'nanoid';
 import type PWAStore from '../PWAStore';
 import getLogger from '../utils/getLogger';
 import type XtextClient from '../xtext/XtextClient';
+import type { SemanticsSuccessResult } from '../xtext/xtextServiceResults';
 
 import EditorErrors from './EditorErrors';
 import LintPanelStore from './LintPanelStore';
@@ -65,7 +66,7 @@ export default class EditorStore {
 
   semanticsError: string | undefined;
 
-  semantics: unknown = {};
+  semantics: SemanticsSuccessResult | undefined;
 
   constructor(initialValue: string, pwaStore: PWAStore) {
     this.id = nanoid();
@@ -295,7 +296,7 @@ export default class EditorStore {
     this.semanticsError = semanticsError;
   }
 
-  setSemantics(semantics: unknown) {
+  setSemantics(semantics: SemanticsSuccessResult) {
     this.semanticsError = undefined;
     this.semantics = semantics;
   }
diff --git a/subprojects/frontend/src/graph/GraphArea.tsx b/subprojects/frontend/src/graph/GraphArea.tsx
new file mode 100644
index 00000000..b55245d8
--- /dev/null
+++ b/subprojects/frontend/src/graph/GraphArea.tsx
@@ -0,0 +1,318 @@
+/*
+ * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+
+import Box from '@mui/material/Box';
+import * as d3 from 'd3';
+import { type Graphviz, graphviz } from 'd3-graphviz';
+import type { BaseType, Selection } from 'd3-selection';
+import { reaction, type IReactionDisposer } from 'mobx';
+import { useCallback, useRef, useState } from 'react';
+import { useResizeDetector } from 'react-resize-detector';
+
+import { useRootStore } from '../RootStoreProvider';
+import type { SemanticsSuccessResult } from '../xtext/xtextServiceResults';
+
+function toGraphviz(
+  semantics: SemanticsSuccessResult | undefined,
+): string | undefined {
+  if (semantics === undefined) {
+    return undefined;
+  }
+  const lines = [
+    'digraph {',
+    'graph [bgcolor=transparent];',
+    'node [fontsize=16, shape=plain];',
+    'edge [fontsize=12, color=black];',
+  ];
+  const nodeIds = semantics.nodes.map((name, i) => name ?? `n${i}`);
+  lines.push(
+    ...nodeIds.map(
+      (id, i) =>
+        `n${i} [id="${id}", label=<<table border="1" cellborder="0" cellspacing="0" cellpadding="4" style="rounded" bgcolor="green"><tr><td>${id}</td></tr><hr/><tr><td bgcolor="white">node</td></tr></table>>];`,
+    ),
+  );
+  Object.keys(semantics.partialInterpretation).forEach((relation) => {
+    if (relation === 'builtin::equals' || relation === 'builtin::contains') {
+      return;
+    }
+    const tuples = semantics.partialInterpretation[relation];
+    if (tuples === undefined) {
+      return;
+    }
+    const first = tuples[0];
+    if (first === undefined || first.length !== 3) {
+      return;
+    }
+    const nameFragments = relation.split('::');
+    const simpleName = nameFragments[nameFragments.length - 1] ?? relation;
+    lines.push(
+      ...tuples.map(([from, to, value]) => {
+        if (
+          typeof from !== 'number' ||
+          typeof to !== 'number' ||
+          typeof value !== 'string'
+        ) {
+          return '';
+        }
+        const isUnknown = value === 'UNKNOWN';
+        return `n${from} -> n${to} [
+            id="${nodeIds[from]},${nodeIds[to]},${relation}",
+            xlabel="${simpleName}",
+            style="${isUnknown ? 'dashed' : 'solid'}",
+            class="edge-${value}"
+          ];`;
+      }),
+    );
+  });
+  lines.push('}');
+  return lines.join('\n');
+}
+
+interface Transform {
+  x: number;
+  y: number;
+  k: number;
+}
+
+export default function GraphArea(): JSX.Element {
+  const { editorStore } = useRootStore();
+  const disposerRef = useRef<IReactionDisposer | undefined>();
+  const graphvizRef = useRef<
+    Graphviz<BaseType, unknown, null, undefined> | undefined
+  >();
+  const canvasRef = useRef<HTMLDivElement | undefined>();
+  const zoomRef = useRef<
+    d3.ZoomBehavior<HTMLDivElement, unknown> | undefined
+  >();
+  const [zoom, setZoom] = useState<Transform>({ x: 0, y: 0, k: 1 });
+  const widthRef = useRef<number | undefined>();
+  const heightRef = useRef<number | undefined>();
+
+  const onResize = useCallback(
+    (width: number | undefined, height: number | undefined) => {
+      if (canvasRef.current === undefined || zoomRef.current === undefined) {
+        return;
+      }
+      let moveX = 0;
+      let moveY = 0;
+      if (widthRef.current !== undefined && width !== undefined) {
+        moveX = (width - widthRef.current) / 2;
+      }
+      if (heightRef.current !== undefined && height !== undefined) {
+        moveY = (height - heightRef.current) / 2;
+      }
+      widthRef.current = width;
+      heightRef.current = height;
+      if (moveX === 0 && moveY === 0) {
+        return;
+      }
+      const currentTransform = d3.zoomTransform(canvasRef.current);
+      zoomRef.current.translateBy(
+        d3.select(canvasRef.current),
+        moveX / currentTransform.k - moveX,
+        moveY / currentTransform.k - moveY,
+      );
+    },
+    [],
+  );
+
+  const { ref: setCanvasResize } = useResizeDetector({
+    onResize,
+  });
+
+  const setCanvas = useCallback(
+    (element: HTMLDivElement | null) => {
+      canvasRef.current = element ?? undefined;
+      setCanvasResize(element);
+      if (element === null) {
+        return;
+      }
+      const zoomBehavior = d3.zoom<HTMLDivElement, unknown>();
+      zoomBehavior.on(
+        'zoom',
+        (event: d3.D3ZoomEvent<HTMLDivElement, unknown>) =>
+          setZoom(event.transform),
+      );
+      d3.select(element).call(zoomBehavior);
+      zoomRef.current = zoomBehavior;
+    },
+    [setCanvasResize],
+  );
+
+  const setElement = useCallback(
+    (element: HTMLDivElement | null) => {
+      if (disposerRef.current !== undefined) {
+        disposerRef.current();
+        disposerRef.current = undefined;
+      }
+      if (graphvizRef.current !== undefined) {
+        // `@types/d3-graphviz` does not contain the signature for the `destroy` method.
+        (graphvizRef.current as unknown as { destroy(): void }).destroy();
+        graphvizRef.current = undefined;
+      }
+      if (element !== null) {
+        element.replaceChildren();
+        const renderer = graphviz(element) as Graphviz<
+          BaseType,
+          unknown,
+          null,
+          undefined
+        >;
+        renderer.keyMode('id');
+        renderer.zoom(false);
+        renderer.tweenPrecision('5%');
+        renderer.tweenShapes(false);
+        renderer.convertEqualSidedPolygons(false);
+        const transition = () =>
+          d3.transition().duration(300).ease(d3.easeCubic);
+        /* eslint-disable-next-line @typescript-eslint/no-unsafe-argument,
+          @typescript-eslint/no-explicit-any --
+          Workaround for error in `@types/d3-graphviz`.
+        */
+        renderer.transition(transition as any);
+        renderer.on(
+          'postProcessSVG',
+          // @ts-expect-error Custom `d3-graphviz` hook not covered by typings.
+          (
+            svgSelection: Selection<SVGSVGElement, unknown, BaseType, unknown>,
+          ) => {
+            svgSelection.selectAll('title').remove();
+            const svg = svgSelection.node();
+            if (svg === null) {
+              return;
+            }
+            svg.querySelectorAll('.node').forEach((node) => {
+              node.querySelectorAll('path').forEach((path) => {
+                const d = path.getAttribute('d') ?? '';
+                const points = d.split(/[A-Z ]/);
+                points.shift();
+                const x = points.map((p) => {
+                  return Number(p.split(',')[0] ?? 0);
+                });
+                const y = points.map((p) => {
+                  return Number(p.split(',')[1] ?? 0);
+                });
+                const xmin = Math.min.apply(null, x);
+                const xmax = Math.max.apply(null, x);
+                const ymin = Math.min.apply(null, y);
+                const ymax = Math.max.apply(null, y);
+                const rect = document.createElementNS(
+                  'http://www.w3.org/2000/svg',
+                  'rect',
+                );
+                rect.setAttribute('fill', path.getAttribute('fill') ?? '');
+                rect.setAttribute('stroke', path.getAttribute('stroke') ?? '');
+                rect.setAttribute('x', String(xmin));
+                rect.setAttribute('y', String(ymin));
+                rect.setAttribute('width', String(xmax - xmin));
+                rect.setAttribute('height', String(ymax - ymin));
+                rect.setAttribute('height', String(ymax - ymin));
+                rect.setAttribute('rx', '12');
+                rect.setAttribute('ry', '12');
+                node.replaceChild(rect, path);
+              });
+            });
+          },
+        );
+        disposerRef.current = reaction(
+          () => editorStore?.semantics,
+          (semantics) => {
+            const str = toGraphviz(semantics);
+            if (str !== undefined) {
+              renderer.renderDot(str);
+            }
+          },
+          { fireImmediately: true },
+        );
+        graphvizRef.current = renderer;
+      }
+    },
+    [editorStore],
+  );
+
+  return (
+    <Box
+      sx={(theme) => ({
+        width: '100%',
+        height: '100%',
+        position: 'relative',
+        overflow: 'hidden',
+        '& svg': {
+          userSelect: 'none',
+          '& .node': {
+            '& text': {
+              ...theme.typography.body2,
+              fill: theme.palette.text.primary,
+            },
+            '& [stroke="black"]': {
+              stroke: theme.palette.text.primary,
+            },
+            '& [fill="green"]': {
+              fill:
+                theme.palette.mode === 'dark'
+                  ? theme.palette.primary.dark
+                  : theme.palette.primary.light,
+            },
+            '& [fill="white"]': {
+              fill: theme.palette.background.default,
+              stroke: theme.palette.background.default,
+            },
+          },
+          '& .edge': {
+            '& text': {
+              ...theme.typography.caption,
+              fill: theme.palette.text.primary,
+            },
+            '& [stroke="black"]': {
+              stroke: theme.palette.text.primary,
+            },
+            '& [fill="black"]': {
+              fill: theme.palette.text.primary,
+            },
+          },
+          '& .edge-UNKNOWN': {
+            '& text': {
+              fill: theme.palette.text.secondary,
+            },
+            '& [stroke="black"]': {
+              stroke: theme.palette.text.secondary,
+            },
+            '& [fill="black"]': {
+              fill: theme.palette.text.secondary,
+            },
+          },
+          '& .edge-ERROR': {
+            '& text': {
+              fill: theme.palette.error.main,
+            },
+            '& [stroke="black"]': {
+              stroke: theme.palette.error.main,
+            },
+            '& [fill="black"]': {
+              fill: theme.palette.error.main,
+            },
+          },
+        },
+      })}
+      ref={setCanvas}
+    >
+      <Box
+        sx={{
+          position: 'absolute',
+          top: `${50 * zoom.k}%`,
+          left: `${50 * zoom.k}%`,
+          transform: `
+              translate(${zoom.x}px, ${zoom.y}px)
+              scale(${zoom.k})
+              translate(-50%, -50%)
+            `,
+          transformOrigin: '0 0',
+        }}
+        ref={setElement}
+      />
+    </Box>
+  );
+}
diff --git a/subprojects/frontend/src/graph/GraphPane.tsx b/subprojects/frontend/src/graph/GraphPane.tsx
index f69f52a6..f04b9931 100644
--- a/subprojects/frontend/src/graph/GraphPane.tsx
+++ b/subprojects/frontend/src/graph/GraphPane.tsx
@@ -5,24 +5,24 @@
  */
 
 import Stack from '@mui/material/Stack';
-import { styled } from '@mui/material/styles';
-import stringify from 'json-stringify-pretty-compact';
-import { observer } from 'mobx-react-lite';
+import { Suspense, lazy } from 'react';
 
-import { useRootStore } from '../RootStoreProvider';
+import Loading from '../Loading';
 
-const StyledCode = styled('code')(({ theme }) => ({
-  ...theme.typography.editor,
-  fontWeight: theme.typography.fontWeightEditorNormal,
-  margin: theme.spacing(2),
-  whiteSpace: 'pre',
-}));
+const GraphArea = lazy(() => import('./GraphArea'));
 
-export default observer(function GraphPane(): JSX.Element {
-  const { editorStore } = useRootStore();
+export default function GraphPane(): JSX.Element {
   return (
-    <Stack direction="column" height="100%" overflow="auto">
-      <StyledCode>{stringify(editorStore?.semantics ?? {})}</StyledCode>
+    <Stack
+      direction="column"
+      height="100%"
+      overflow="auto"
+      alignItems="center"
+      justifyContent="center"
+    >
+      <Suspense fallback={<Loading />}>
+        <GraphArea />
+      </Suspense>
     </Stack>
   );
-});
+}
diff --git a/subprojects/frontend/src/xtext/SemanticsService.ts b/subprojects/frontend/src/xtext/SemanticsService.ts
index 50ec371a..d68b87a9 100644
--- a/subprojects/frontend/src/xtext/SemanticsService.ts
+++ b/subprojects/frontend/src/xtext/SemanticsService.ts
@@ -17,11 +17,15 @@ export default class SemanticsService {
 
   onPush(push: unknown): void {
     const result = SemanticsResult.parse(push);
-    this.validationService.setSemanticsIssues(result.issues ?? []);
-    if (result.error !== undefined) {
-      this.store.setSemanticsError(result.error);
+    if ('issues' in result) {
+      this.validationService.setSemanticsIssues(result.issues);
     } else {
-      this.store.setSemantics(push);
+      this.validationService.setSemanticsIssues([]);
+      if ('error' in result) {
+        this.store.setSemanticsError(result.error);
+      } else {
+        this.store.setSemantics(result);
+      }
     }
     this.store.analysisCompleted();
   }
diff --git a/subprojects/frontend/src/xtext/xtextServiceResults.ts b/subprojects/frontend/src/xtext/xtextServiceResults.ts
index cae95771..12f87b26 100644
--- a/subprojects/frontend/src/xtext/xtextServiceResults.ts
+++ b/subprojects/frontend/src/xtext/xtextServiceResults.ts
@@ -126,9 +126,20 @@ export const FormattingResult = DocumentStateResult.extend({
 
 export type FormattingResult = z.infer<typeof FormattingResult>;
 
-export const SemanticsResult = z.object({
-  error: z.string().optional(),
-  issues: Issue.array().optional(),
+export const SemanticsSuccessResult = z.object({
+  nodes: z.string().nullable().array(),
+  partialInterpretation: z.record(
+    z.string(),
+    z.union([z.number(), z.string()]).array().array(),
+  ),
 });
 
+export type SemanticsSuccessResult = z.infer<typeof SemanticsSuccessResult>;
+
+export const SemanticsResult = z.union([
+  z.object({ error: z.string() }),
+  z.object({ issues: Issue.array() }),
+  SemanticsSuccessResult,
+]);
+
 export type SemanticsResult = z.infer<typeof SemanticsResult>;