@pie-lib/graphing 4.0.5-next.30 → 4.0.5-next.34

package/src/tools/shared/line/index.jsx

@@ -0,0 +1,487 @@
+import React from 'react';
+import { cloneDeep, isEmpty, isEqual } from 'lodash-es';
+import { BasePoint } from '../point';
+import { gridDraggable, trig, types, utils } from '@pie-lib/plot';
+import PropTypes from 'prop-types';
+import { correct, disabled, disabledSecondary, incorrect, missing } from '../styles';
+import ReactDOM from 'react-dom';
+import MarkLabel from '../../../mark-label';
+import { color } from '@pie-lib/render-ui';
+import { equalPoints, getMiddleOfTwoPoints, sameAxes, stripEmptyLabel } from '../../../utils';
+import { styled } from '@mui/material/styles';
+
+const StyledLineGroup = styled('g')(({ disabled, correctness }) => ({
+  '& line:not(.hit-area)': {
+    fill: 'transparent',
+    stroke: color.defaults.BLACK,
+    strokeWidth: 3,
+    transition: 'stroke 200ms ease-in, stroke-width 200ms ease-in',
+    '&:hover': {
+      strokeWidth: 6,
+      stroke: color.defaults.PRIMARY_DARK,
+    },
+  },
+  ...(disabled && {
+    '& line:not(.hit-area)': {
+      ...disabledSecondary('stroke'),
+      strokeWidth: 2,
+    },
+  }),
+  ...(correctness === 'correct' && {
+    '& line:not(.hit-area)': {
+      ...correct('stroke'),
+    },
+  }),
+  ...(correctness === 'incorrect' && {
+    '& line:not(.hit-area)': {
+      ...incorrect('stroke'),
+    },
+  }),
+  ...(correctness === 'missing' && {
+    '& line:not(.hit-area)': {
+      ...missing('stroke'),
+      strokeWidth: 1,
+      strokeDasharray: '4 3',
+    },
+  }),
+}));
+
+export const lineTool = (type, Component) => () => ({
+  type,
+  Component,
+  addPoint: (point, mark) => {
+    if (mark && equalPoints(mark.root, point)) {
+      return mark;
+    }
+
+    if (!mark) {
+      return {
+        type,
+        building: true,
+        from: point,
+      };
+    }
+
+    if (equalPoints(point, mark.from)) {
+      return { ...mark };
+    }
+
+    return { ...mark, building: false, to: point };
+  },
+});
+
+export const lineToolComponent = (Component) => {
+  return class LineToolComponent extends React.Component {
+    static propTypes = {
+      ...types.ToolPropTypeFields,
+      graphProps: types.GraphPropsType.isRequired,
+      limitLabeling: PropTypes.bool,
+      changeMarkProps: PropTypes.func,
+      disabled: PropTypes.bool,
+      from: types.PointType,
+      to: types.PointType,
+      labelModeEnabled: PropTypes.bool,
+      onClick: PropTypes.func,
+    };
+
+    constructor(props) {
+      super(props);
+      this.state = {};
+    }
+
+    startDrag = () => {
+      const { onDragStart } = this.props;
+      this.setState({ mark: { ...this.props.mark } });
+      if (onDragStart) onDragStart();
+    };
+
+    stopDrag = () => {
+      const { onChange, mark, onDragStop } = this.props;
+      const update = { ...this.state.mark };
+
+      this.setState({ mark: undefined }, () => {
+        const { type } = update;
+        let shouldNotChange =
+          type &&
+          (type === 'parabola' || type === 'sine' || type === 'absolute' || type === 'exponential') &&
+          sameAxes(update.from, update.to);
+        if (!shouldNotChange && type && type === 'exponential' && update.from && update.to) {
+          shouldNotChange = update.from.y === 0 || update.to.y === 0 || update.from.y * update.to.y < 0;
+        }
+        if (!isEqual(mark, update) && !shouldNotChange) {
+          onChange(mark, update);
+        }
+        if (onDragStop) onDragStop();
+      });
+    };
+
+    changeMark = ({ from, to, middle }) => {
+      let mark = { ...this.state.mark, from, to };
+
+      if (middle) {
+        mark = { ...mark, middle };
+      }
+
+      this.setState({ mark });
+    };
+
+    changeMarkProps = ({ from, to, middle }) => {
+      const { onChange, mark } = this.props;
+      let update = { ...mark, ...this.state.mark };
+
+      if (from) {
+        update = { ...update, from };
+      }
+
+      if (to) {
+        update = { ...update, to };
+      }
+
+      if (middle) {
+        update = { ...update, middle };
+      }
+
+      if (!isEqual(mark, update)) {
+        onChange(mark, update);
+      }
+    };
+
+    render() {
+      const { graphProps, onClick, labelNode, labelModeEnabled, coordinatesOnHover, limitLabeling } = this.props;
+      const mark = this.state.mark ? this.state.mark : this.props.mark;
+
+      const from = cloneDeep(mark.from);
+      const to = cloneDeep(mark.to);
+      const middle = cloneDeep(mark.middle);
+
+      // SET DISABLED
+      // if it's a background mark, we need to force disable it
+      if (from && mark.isBackground) {
+        from.disabled = true;
+      }
+
+      if (to && mark.isBackground) {
+        to.disabled = true;
+      }
+
+      if (middle && mark.isBackground) {
+        middle.disabled = true;
+      }
+
+      return (
+        <Component
+          disabled={mark.disabled}
+          coordinatesOnHover={coordinatesOnHover}
+          correctness={mark.correctness}
+          from={from}
+          to={to}
+          middle={middle}
+          graphProps={graphProps}
+          onChange={this.changeMark}
+          changeMarkProps={this.changeMarkProps}
+          onClick={onClick}
+          onDragStart={this.startDrag}
+          onDragStop={this.stopDrag}
+          labelNode={labelNode}
+          labelModeEnabled={labelModeEnabled}
+          limitLabeling={limitLabeling}
+        />
+      );
+    }
+  };
+};
+
+const dragOpts = () => ({
+  bounds: (props, { domain, range }) => {
+    const area = utils.lineToArea(props.from, props.to);
+    return utils.bounds(area, domain, range);
+  },
+  anchorPoint: (props) => {
+    const { from } = props;
+    return from;
+  },
+  fromDelta: (props, delta) => {
+    const { from, to } = props;
+    return {
+      from: utils.point(from).add(utils.point(delta)),
+      to: utils.point(to).add(utils.point(delta)),
+    };
+  },
+});
+
+export const lineBase = (Comp, opts) => {
+  const DraggableComp = gridDraggable(dragOpts())(Comp);
+
+  const FromPoint = opts && opts.from ? opts.from : BasePoint;
+  const ToPoint = opts && opts.to ? opts.to : BasePoint;
+
+  class LineBase extends React.Component {
+    static propTypes = {
+      coordinatesOnHover: PropTypes.bool,
+      graphProps: types.GraphPropsType,
+      from: types.PointType,
+      to: types.PointType,
+      middle: types.PointType,
+      onChange: PropTypes.func,
+      onDragStart: PropTypes.func,
+      onDragStop: PropTypes.func,
+      onClick: PropTypes.func,
+      correctness: PropTypes.string,
+      disabled: PropTypes.bool,
+      limitLabeling: PropTypes.bool,
+      changeMarkProps: PropTypes.func,
+      labelModeEnabled: PropTypes.bool,
+      labelNode: PropTypes.object,
+    };
+
+    onChangePoint = (point) => {
+      const { middle, onChange } = this.props;
+      const { from, to } = point;
+
+      // because point.from.label and point.to.label can be different
+      if (!equalPoints(from, to)) {
+        if (middle) {
+          point.middle = { ...middle, ...getMiddleOfTwoPoints(from, to) };
+        }
+
+        onChange(point);
+      }
+    };
+
+    dragComp = ({ from: draggedFrom, to: draggedTo }) => {
+      const { from, to, onChange, middle } = this.props;
+
+      if (from.label) {
+        draggedFrom.label = from.label;
+      }
+
+      if (to.label) {
+        draggedTo.label = to.label;
+      }
+
+      const updated = { from: draggedFrom, to: draggedTo };
+
+      if (middle) {
+        updated.middle = { ...middle, ...getMiddleOfTwoPoints(draggedFrom, draggedTo) };
+      }
+
+      onChange(updated);
+    };
+
+    dragFrom = (draggedFrom) => {
+      const { from, to } = this.props;
+
+      if (from.label) {
+        draggedFrom.label = from.label;
+      }
+
+      if (!equalPoints(draggedFrom, to)) {
+        this.onChangePoint({ from: draggedFrom, to: to });
+      }
+    };
+
+    dragTo = (draggedTo) => {
+      const { from, to } = this.props;
+
+      if (to.label) {
+        draggedTo.label = to.label;
+      }
+
+      if (!equalPoints(from, draggedTo)) {
+        this.onChangePoint({ from: from, to: draggedTo });
+      }
+    };
+
+    labelChange = (point, type) => {
+      const { changeMarkProps } = this.props;
+      const update = { ...point };
+
+      if (!point.label || isEmpty(point.label)) {
+        delete update.label;
+      }
+
+      changeMarkProps({ [type]: update });
+    };
+
+    clickPoint = (point, type, data) => {
+      const { changeMarkProps, disabled, from, to, middle, labelModeEnabled, limitLabeling, onClick } = this.props;
+
+      if (!labelModeEnabled) {
+        onClick(point || data);
+        return;
+      }
+
+      if (disabled) {
+        return;
+      }
+
+      // limit labeling the points of the line
+      if (limitLabeling) {
+        return;
+      }
+
+      if (type === 'middle' && !point && from && to) {
+        point = { ...point, ...getMiddleOfTwoPoints(from, to) };
+      }
+
+      changeMarkProps({
+        from: stripEmptyLabel(from),
+        to: stripEmptyLabel(to),
+        middle: stripEmptyLabel(middle),
+        [type]: { label: '', ...point },
+      });
+
+      if (this.input[type]) {
+        this.input[type].focus();
+      }
+    };
+
+    // IMPORTANT, do not remove
+    input = {};
+
+    render() {
+      const {
+        coordinatesOnHover,
+        graphProps,
+        onDragStart,
+        onDragStop,
+        from,
+        to,
+        middle,
+        disabled,
+        correctness,
+        onClick,
+        labelNode,
+        labelModeEnabled,
+      } = this.props;
+      const common = { graphProps, onDragStart, onDragStop, disabled, correctness, onClick };
+      const angle = to ? trig.toDegrees(trig.angle(from, to)) : 0;
+
+      let fromLabelNode = null;
+      let toLabelNode = null;
+      let lineLabelNode = null;
+
+      if (labelNode) {
+        if (from && Object.prototype.hasOwnProperty.call(from, 'label')) {
+          fromLabelNode = ReactDOM.createPortal(
+            <MarkLabel
+              inputRef={(r) => (this.input.from = r)}
+              disabled={!labelModeEnabled}
+              mark={from}
+              graphProps={graphProps}
+              onChange={(label) => this.labelChange({ ...from, label }, 'from')}
+            />,
+            labelNode,
+          );
+        }
+
+        if (to && Object.prototype.hasOwnProperty.call(to, 'label')) {
+          toLabelNode = ReactDOM.createPortal(
+            <MarkLabel
+              inputRef={(r) => (this.input.to = r)}
+              disabled={!labelModeEnabled}
+              mark={to}
+              graphProps={graphProps}
+              onChange={(label) => this.labelChange({ ...to, label }, 'to')}
+            />,
+            labelNode,
+          );
+        }
+
+        if (middle && Object.prototype.hasOwnProperty.call(middle, 'label')) {
+          lineLabelNode = ReactDOM.createPortal(
+            <MarkLabel
+              inputRef={(r) => (this.input.middle = r)}
+              disabled={!labelModeEnabled}
+              mark={middle}
+              graphProps={graphProps}
+              onChange={(label) => this.labelChange({ ...middle, label }, 'middle')}
+            />,
+            labelNode,
+          );
+        }
+      }
+
+      return (
+        <StyledLineGroup disabled={disabled} correctness={correctness}>
+          {to && (
+            <DraggableComp
+              from={from}
+              to={to}
+              middle={middle}
+              onDrag={this.dragComp}
+              {...common}
+              onClick={(data) => this.clickPoint(middle, 'middle', data)}
+            />
+          )}
+          {lineLabelNode}
+
+          <FromPoint
+            x={from.x}
+            y={from.y}
+            labelNode={labelNode}
+            coordinatesOnHover={coordinatesOnHover}
+            onDrag={this.dragFrom}
+            {...common}
+            onClick={(data) => this.clickPoint(from, 'from', data)}
+          />
+          {fromLabelNode}
+
+          {to && (
+            <ToPoint
+              x={to.x}
+              y={to.y}
+              angle={angle} //angle + 45}
+              labelNode={labelNode}
+              coordinatesOnHover={coordinatesOnHover}
+              onDrag={this.dragTo}
+              {...common}
+              onClick={(data) => this.clickPoint(to, 'to', data)}
+            />
+          )}
+          {toLabelNode}
+        </StyledLineGroup>
+      );
+    }
+  }
+
+  return LineBase;
+};
+
+export const styles = {
+  line: () => ({
+    fill: 'transparent',
+    stroke: color.defaults.BLACK,
+    strokeWidth: 3,
+    transition: 'stroke 200ms ease-in, stroke-width 200ms ease-in',
+    '&:hover': {
+      strokeWidth: 6,
+      stroke: color.defaults.PRIMARY_DARK,
+    },
+  }),
+  arrow: () => ({
+    fill: color.defaults.BLACK,
+  }),
+  disabledArrow: () => ({
+    ...disabledSecondary(),
+  }),
+  disabledSecondary: () => ({
+    ...disabledSecondary('stroke'),
+    strokeWidth: 2,
+  }),
+  disabled: () => ({
+    ...disabled('stroke'),
+    strokeWidth: 2,
+  }),
+  correct: (theme, key) => ({
+    ...correct(key),
+  }),
+  incorrect: (theme, key) => ({
+    ...incorrect(key),
+  }),
+  missing: (theme, key) => ({
+    ...missing(key),
+    strokeWidth: 1,
+    strokeDasharray: '4 3',
+  }),
+};

package/src/tools/shared/line/line-path.jsx

@@ -0,0 +1,80 @@
+import React from 'react';
+import PropTypes from 'prop-types';
+import { styled } from '@mui/material/styles';
+import { types } from '@pie-lib/plot';
+import { correct, disabled, disabledSecondary, incorrect, missing } from '../styles';
+import * as vx from '@visx/shape';
+import { color } from '@pie-lib/render-ui';
+
+const dragging = () => ({
+  strokeWidth: 4,
+  stroke: color.defaults.BLACK,
+});
+
+const StyledDrawLine = styled(vx.LinePath)(({ disabled: isDisabled, correctness }) => ({
+  fill: 'none',
+  strokeWidth: 3,
+  stroke: color.black(),
+  ...(isDisabled && {
+    ...disabledSecondary('stroke'),
+    strokeWidth: 2,
+  }),
+  ...(correctness === 'correct' && correct('stroke')),
+  ...(correctness === 'incorrect' && incorrect('stroke')),
+  ...(correctness === 'missing' && {
+    ...missing('stroke'),
+    strokeWidth: 1,
+    strokeDasharray: '4 3',
+  }),
+}));
+
+const StyledLine = styled(vx.LinePath)(({ theme, disabled: isDisabled, correctness, isDragging }) => ({
+  strokeWidth: 7,
+  fill: 'none',
+  transition: 'stroke-width 200ms ease-in, stroke 200ms ease-in',
+  stroke: 'transparent',
+  pointerEvents: 'stroke',
+  '&:hover': dragging(theme),
+  ...(isDragging && dragging(theme)),
+  ...((isDisabled || correctness) && { pointerEvents: 'none' }),
+}));
+
+export class RawLinePath extends React.Component {
+  static propTypes = {
+    className: PropTypes.string,
+    data: PropTypes.arrayOf(PropTypes.arrayOf(PropTypes.number)),
+    graphProps: types.GraphPropsType.isRequired,
+    disabled: PropTypes.bool,
+    correctness: PropTypes.string,
+    from: types.PointType,
+    to: types.PointType,
+    isDragging: PropTypes.bool,
+  };
+
+  static defaultProps = {
+    from: {},
+    to: {},
+  };
+
+  render() {
+    /* eslint-disable no-unused-vars */
+    const { data, className, disabled, correctness, from, to, graphProps, isDragging, ...rest } = this.props;
+    /* eslint-enable */
+
+    return (
+      <React.Fragment>
+        <StyledDrawLine data={data} className={className} disabled={disabled} correctness={correctness} {...rest} />
+        <StyledLine
+          data={data}
+          className={className}
+          isDragging={isDragging}
+          disabled={disabled}
+          correctness={correctness}
+          {...rest}
+        />
+      </React.Fragment>
+    );
+  }
+}
+
+export const LinePath = RawLinePath;

package/src/tools/shared/line/with-root-edge.jsx

@@ -0,0 +1,97 @@
+import React from 'react';
+import PropTypes from 'prop-types';
+import { types } from '@pie-lib/plot';
+import { LinePath } from './line-path';
+import { curveMonotoneX } from '@visx/curve';
+import { lineBase, lineToolComponent } from './index';
+
+const toRootEdge = (m) => {
+  const out = { ...m };
+  out.root = { ...m.from };
+  out.edge = m.to ? { ...m.to } : undefined;
+  delete out.from;
+  delete out.to;
+  return out;
+};
+
+const toFromTo = (m) => {
+  const out = { ...m };
+  out.from = { ...m.root };
+  out.to = m.edge ? { ...m.edge } : undefined;
+  delete out.root;
+  delete out.edge;
+  return out;
+};
+
+export const rootEdgeToFromToWrapper = (BaseComp) => {
+  const Wrapper = (props) => {
+    const m = toFromTo(props.mark);
+
+    const onChange = (current, next) => {
+      props.onChange(toRootEdge(current), toRootEdge(next));
+    };
+
+    return <BaseComp {...props} mark={m} onChange={onChange} />;
+  };
+
+  Wrapper.propTypes = {
+    onChange: PropTypes.func,
+    mark: PropTypes.object,
+  };
+
+  return Wrapper;
+};
+
+export const rootEdgeComponent = (RootEdgeComp) => {
+  const BaseComponent = lineToolComponent(RootEdgeComp);
+  return rootEdgeToFromToWrapper(BaseComponent);
+};
+
+const withPointsGenerationLinePath = (getPoints) => {
+  const LinePathComponent = (props) => {
+    const { graphProps, from, to, onClick, onDragStart, onDragStop, onChange, disabled, correctness, ...rest } = props;
+
+    const { dataPoints, enableCurve = true } = getPoints({
+      graphProps: props.graphProps,
+      root: from,
+      edge: to,
+    });
+
+    const raw = dataPoints.map((d) => [graphProps.scale.x(d.x), graphProps.scale.y(d.y)]);
+
+    const common = {
+      onClick,
+      graphProps,
+      onDragStart,
+      onDragStop,
+      onChange,
+      disabled,
+      correctness,
+    };
+
+    if (!enableCurve) {
+      return <LinePath data={raw} from={from} to={to} {...common} {...rest} />;
+    } else {
+      return <LinePath data={raw} from={from} to={to} curve={curveMonotoneX} {...common} {...rest} />;
+    }
+  };
+
+  LinePathComponent.propTypes = {
+    graphProps: types.GraphPropsType.isRequired,
+    from: types.PointType.isRequired,
+    to: types.PointType,
+    onClick: PropTypes.func,
+    onDragStart: PropTypes.func,
+    onDragStop: PropTypes.func,
+    onChange: PropTypes.func,
+    disabled: PropTypes.bool,
+    correctness: PropTypes.string,
+  };
+
+  return LinePathComponent;
+};
+
+export const withRootEdge = (getPoints) => {
+  const LinePathComp = withPointsGenerationLinePath(getPoints);
+  return lineBase(LinePathComp);
+};