ui-space-time: zoom behavior mouse + slider

Signed-off-by: Valentin Chanas <[email protected]>

Author: anisometropie

ui-spacetimechart/src/stories/horizontal-zoom.stories.tsx

@@ -0,0 +1,157 @@
+import React, { useEffect, useState } from 'react';
+
+import { Slider } from '@osrd-project/ui-core';
+import type { Meta } from '@storybook/react';
+
+import '@osrd-project/ui-core/dist/theme.css';
+
+import { OPERATIONAL_POINTS, PATHS } from './lib/paths';
+import { PathLayer } from '../components/PathLayer';
+import { SpaceTimeChart } from '../components/SpaceTimeChart';
+import { type Point, type PathData, type OperationalPoint } from '../lib/types';
+import { getDiff } from '../utils/vectors';
+
+const DEFAULT_WIDTH = 1200;
+const DEFAULT_HEIGHT = 550;
+
+const MIN_ZOOM = 0;
+const MAX_ZOOM = 100;
+const MIN_ZOOM_MS_PER_PX = 600000;
+const MAX_ZOOM_MS_PER_PX = 625;
+type SpaceTimeHorizontalZoomWrapperProps = {
+  offset: number;
+  operationalPoints: OperationalPoint[];
+  paths: (PathData & { color: string })[];
+};
+
+const sliderToTimeScale = (slider: number) =>
+  MIN_ZOOM_MS_PER_PX * Math.pow(MAX_ZOOM_MS_PER_PX / MIN_ZOOM_MS_PER_PX, Number(slider) / 100);
+
+const SpaceTimeHorizontalZoomWrapper = ({
+  offset,
+  operationalPoints = [],
+  paths = [],
+}: SpaceTimeHorizontalZoomWrapperProps) => {
+  const [state, setState] = useState<{
+    zoomValue: number;
+    xOffset: number;
+    yOffset: number;
+    panning: null | { initialOffset: Point };
+  }>({
+    zoomValue: 36.18, // 7500 ms/px
+    xOffset: offset, // px
+    yOffset: 0,
+    panning: null,
+  });
+  useEffect(() => {
+    setState((prev) => ({ ...prev, xOffset: offset }));
+  }, [offset]);
+  const handleZoom = (zoomValue: number, position = DEFAULT_WIDTH / 2) => {
+    if (MIN_ZOOM <= zoomValue && zoomValue <= MAX_ZOOM) {
+      const oldTimeScale = sliderToTimeScale(state.zoomValue);
+      const newTimeScale = sliderToTimeScale(zoomValue);
+      // by default zooming expands the graph around timeOrigin
+      // changing the offset to timeOriginShift alone keeps the left border in place.
+      // subtracting centerShift keeps the center in place
+      // (xOffset = how many px the timeOrigin is shifted to the right)
+      const timeOriginShift = (state.xOffset * oldTimeScale) / newTimeScale;
+      const centerShift = (position * oldTimeScale - position * newTimeScale) / newTimeScale;
+      const newOffset = timeOriginShift - centerShift;
+      setState((prev) => ({ ...prev, zoomValue, xOffset: newOffset }));
+    }
+  };
+  const simpleOperationalPoints = operationalPoints.map(({ id, position }) => ({
+    id,
+    label: id,
+    position,
+  }));
+  const spaceScale = [
+    {
+      from: 0,
+      to: 75000,
+      coefficient: 300,
+    },
+  ];
+  return (
+    <div
+      className="space-time-horizontal-zoom-wrapper"
+      style={{
+        height: `${DEFAULT_HEIGHT}px`,
+        width: `${DEFAULT_WIDTH}px`,
+      }}
+    >
+      <SpaceTimeChart
+        className="inset-0 absolute h-full"
+        spaceOrigin={0}
+        xOffset={state.xOffset}
+        yOffset={state.yOffset}
+        timeOrigin={+new Date('2024/04/02')}
+        operationalPoints={simpleOperationalPoints}
+        timeScale={sliderToTimeScale(state.zoomValue)}
+        spaceScales={spaceScale}
+        onZoom={({ delta, position: { x } }) => {
+          handleZoom(state.zoomValue + delta, x);
+        }}
+        onPan={({ initialPosition, position, isPanning }) => {
+          const diff = getDiff(initialPosition, position);
+          setState((s) => {
+            // Stop panning:
+            if (!isPanning) {
+              return { ...s, panning: null };
+            }
+            // Start panning:
+            else if (!s.panning) {
+              return {
+                ...s,
+                panning: {
+                  initialOffset: {
+                    x: s.xOffset,
+                    y: s.yOffset,
+                  },
+                },
+              };
+            }
+            // Keep panning:
+            else {
+              const { initialOffset } = s.panning;
+              return {
+                ...s,
+                xOffset: initialOffset.x + diff.x,
+                yOffset: initialOffset.y + diff.y,
+              };
+            }
+          });
+        }}
+      >
+        {paths.map((path) => (
+          <PathLayer key={path.id} path={path} color={path.color} />
+        ))}
+      </SpaceTimeChart>
+      <div style={{ display: 'flex', flexDirection: 'column', gap: '10px' }}>
+        <Slider
+          min={0}
+          max={100}
+          value={state.zoomValue}
+          onChange={(e) => {
+            handleZoom(Number(e.target.value));
+          }}
+        />
+        <div>offset: {state.xOffset}</div>
+        <div>timescale: {sliderToTimeScale(state.zoomValue)} ms/px</div>
+      </div>
+    </div>
+  );
+};
+
+export default {
+  title: 'SpaceTimeChart/Horiontal Zoom',
+  component: SpaceTimeHorizontalZoomWrapper,
+} as Meta<typeof SpaceTimeHorizontalZoomWrapper>;
+
+export const Default = {
+  args: {
+    offset: 0,
+    operationalPoints: OPERATIONAL_POINTS,
+    paths: PATHS.slice(2, 4),
+  },
+};