NN2022100801/addons/admin/node_modules/@antv/f2/lib/plugin/scroll-bar.js

"use strict";

exports.__esModule = true;
exports.init = init;
exports.clear = clear;
exports.changeData = changeData;
exports.clearInner = clearInner;
exports.afterGeomDraw = afterGeomDraw;
exports["default"] = void 0;

var _helper = require("../interaction/helper");

var _common = require("../util/common");

var DEFAULT_CFG = {
  mode: 'x',
  xStyle: {
    backgroundColor: 'rgba(202, 215, 239, .2)',
    fillerColor: 'rgba(202, 215, 239, .5)',
    size: 4,
    lineCap: 'round',
    offsetX: 0,
    offsetY: 8
  },
  yStyle: {
    backgroundColor: 'rgba(202, 215, 239, .2)',
    fillerColor: 'rgba(202, 215, 239, .5)',
    size: 4,
    lineCap: 'round',
    offsetX: 8,
    offsetY: 0
  }
};

function init(chart) {
  chart.set('_limitRange', {});

  chart.scrollBar = function (cfg) {
    if (cfg === true) {
      cfg = DEFAULT_CFG;
    } else if ((0, _common.isObject)(cfg)) {
      cfg = (0, _common.deepMix)({}, DEFAULT_CFG, cfg);
    }

    this.set('_scrollBarCfg', cfg);
  };
}

function clear(chart) {
  chart.set('_limitRange', {});
}

function changeData(chart) {
  chart.set('_limitRange', {});
}

function clearInner(chart) {
  var hBar = chart.get('_horizontalBar');
  var vBar = chart.get('_verticalBar');
  hBar && hBar.remove(true);
  vBar && vBar.remove(true);
  chart.set('_horizontalBar', null);
  chart.set('_verticalBar', null);
}

function afterGeomDraw(chart) {
  var scrollBarCfg = chart.get('_scrollBarCfg');
  if (!scrollBarCfg) return;
  var data = chart.get('data');
  var plotRange = chart.get('plotRange');
  var backPlot = chart.get('backPlot');
  var canvas = chart.get('canvas');
  var canvasHeight = canvas.get('height');
  var limitRange = chart.get('_limitRange');
  var mode = scrollBarCfg.mode;

  if ((0, _common.directionEnabled)(mode, 'x')) {
    var _scrollBarCfg$xStyle = scrollBarCfg.xStyle,
        offsetX = _scrollBarCfg$xStyle.offsetX,
        offsetY = _scrollBarCfg$xStyle.offsetY,
        lineCap = _scrollBarCfg$xStyle.lineCap,
        backgroundColor = _scrollBarCfg$xStyle.backgroundColor,
        fillerColor = _scrollBarCfg$xStyle.fillerColor,
        size = _scrollBarCfg$xStyle.size;
    var xScale = chart.getXScale();
    var xLimitRange = limitRange[xScale.field];

    if (!xLimitRange) {
      xLimitRange = (0, _helper.getLimitRange)(data, xScale);
      limitRange[xScale.field] = xLimitRange;
    }

    var currentRange = (0, _helper.getFieldRange)(xScale, xLimitRange, xScale.type);
    var horizontalBar = chart.get('_horizontalBar');
    var yPos = canvasHeight - size / 2 + offsetY;

    if (horizontalBar) {
      var progressLine = horizontalBar.get('children')[1];
      progressLine.attr({
        x1: Math.max(plotRange.bl.x + plotRange.width * currentRange[0] + offsetX, plotRange.bl.x),
        x2: Math.min(plotRange.bl.x + plotRange.width * currentRange[1] + offsetX, plotRange.br.x)
      });
    } else {
      horizontalBar = backPlot.addGroup({
        className: 'horizontalBar'
      });
      horizontalBar.addShape('line', {
        attrs: {
          x1: plotRange.bl.x + offsetX,
          y1: yPos,
          x2: plotRange.br.x + offsetX,
          y2: yPos,
          lineWidth: size,
          stroke: backgroundColor,
          lineCap: lineCap
        }
      });
      horizontalBar.addShape('line', {
        attrs: {
          x1: Math.max(plotRange.bl.x + plotRange.width * currentRange[0] + offsetX, plotRange.bl.x),
          y1: yPos,
          x2: Math.min(plotRange.bl.x + plotRange.width * currentRange[1] + offsetX, plotRange.br.x),
          y2: yPos,
          lineWidth: size,
          stroke: fillerColor,
          lineCap: lineCap
        }
      });
      chart.set('_horizontalBar', horizontalBar);
    }
  }

  if ((0, _common.directionEnabled)(mode, 'y')) {
    var _scrollBarCfg$yStyle = scrollBarCfg.yStyle,
        _offsetX = _scrollBarCfg$yStyle.offsetX,
        _offsetY = _scrollBarCfg$yStyle.offsetY,
        _lineCap = _scrollBarCfg$yStyle.lineCap,
        _backgroundColor = _scrollBarCfg$yStyle.backgroundColor,
        _fillerColor = _scrollBarCfg$yStyle.fillerColor,
        _size = _scrollBarCfg$yStyle.size;
    var yScale = chart.getYScales()[0];
    var yLimitRange = limitRange[yScale.field];

    if (!yLimitRange) {
      yLimitRange = (0, _helper.getLimitRange)(data, yScale);
      limitRange[yScale.field] = yLimitRange;
    }

    var _currentRange = (0, _helper.getFieldRange)(yScale, yLimitRange, yScale.type);

    var verticalBar = chart.get('_verticalBar');
    var xPos = _size / 2 + _offsetX;

    if (verticalBar) {
      var _progressLine = verticalBar.get('children')[1];

      _progressLine.attr({
        y1: Math.max(plotRange.tl.y + plotRange.height * _currentRange[0] + _offsetY, plotRange.tl.y),
        y2: Math.min(plotRange.tl.y + plotRange.height * _currentRange[1] + _offsetY, plotRange.bl.y)
      });
    } else {
      verticalBar = backPlot.addGroup({
        className: 'verticalBar'
      });
      verticalBar.addShape('line', {
        attrs: {
          x1: xPos,
          y1: plotRange.tl.y + _offsetY,
          x2: xPos,
          y2: plotRange.bl.y + _offsetY,
          lineWidth: _size,
          stroke: _backgroundColor,
          lineCap: _lineCap
        }
      });
      verticalBar.addShape('line', {
        attrs: {
          x1: xPos,
          y1: Math.max(plotRange.tl.y + plotRange.height * _currentRange[0] + _offsetY, plotRange.tl.y),
          x2: xPos,
          y2: Math.min(plotRange.tl.y + plotRange.height * _currentRange[1] + _offsetY, plotRange.bl.y),
          lineWidth: _size,
          stroke: _fillerColor,
          lineCap: _lineCap
        }
      });
      chart.set('_verticalBar', verticalBar);
    }
  }
}

var _default = {
  init: init,
  clear: clear,
  changeData: changeData,
  clearInner: clearInner,
  afterGeomDraw: afterGeomDraw
};
exports["default"] = _default;