@orbcharts/plugins-basic 3.0.0-alpha.29 → 3.0.0-alpha.31

package/src/grid/plugins/BarsTriangle.ts

@@ -1,594 +1,31 @@
-import * as d3 from 'd3'
 import {
-  combineLatest,
-  switchMap,
-  map,
-  first,
-  takeUntil,
-  distinctUntilChanged,
   Subject,
   Observable } from 'rxjs'
-// import type { Subject } from 'rxjs'
 import { defineGridPlugin } from '@orbcharts/core'
-import type {
-  ComputedDatumGrid,
-  EventSeries,
-  ChartParams, 
-  DataSeriesDatum,
-  Layout } from '@orbcharts/core'
-import type { BarsTriangleParams } from '../types'
 import { DEFAULT_BARS_TRIANGLE_PLUGIN_PARAMS } from '../defaults'
-import { getD3TransitionEase } from '../../utils/d3Utils'
-import {
-  getClassName,
-  getUniID } from '../../utils/orbchartsUtils'
-
-interface RenderBarParams {
-  selection: d3.Selection<SVGGElement, unknown, any, any>
-  barData: BarDatumGrid[][]
-  zeroY: number
-  groupLabels: string[]
-  barScale: d3.ScalePoint<string>
-  params: BarsTriangleParams
-  chartParams: ChartParams
-  barWidth: number
-  delayGroup: number
-  transitionItem: number
-}
-
-interface BarDatumGrid extends ComputedDatumGrid {
-  linearGradientId: string
-}
-
-type ClipPathDatum = {
-  id: string;
-  // x: number;
-  // y: number;
-  width: number;
-  height: number;
-}
+import { createBaseBarsTriangle } from '../../base/BaseBarsTriangle'
 
 const pluginName = 'BarsTriangle'
-const gClassName = getClassName(pluginName, 'g')
-const gContentClassName = getClassName(pluginName, 'g-content')
-// group的delay在動畫中的佔比（剩餘部份的時間為圖形本身的動畫時間，因為delay時間和最後一個group的動畫時間加總為1）
-const groupDelayProportionOfDuration = 0.3
-
-function calcBarWidth ({ axisWidth, groupAmount, barAmountOfGroup, barPadding = 0, barGroupPadding = 0 }: {
-  axisWidth: number
-  groupAmount: number
-  barAmountOfGroup: number
-  barPadding: number
-  barGroupPadding: number
-}) {
-  const width = (axisWidth / groupAmount - barGroupPadding) / barAmountOfGroup - barPadding
-  return width > 1 ? width : 1
-}
-
-function makeBarScale (barWidth: number, seriesLabels: string[], params: BarsTriangleParams) {
-  const barHalfWidth = barWidth! / 2
-  const barGroupWidth = barWidth * seriesLabels.length + params.barPadding! * seriesLabels.length
-  return d3.scalePoint()
-    .domain(seriesLabels)
-    .range([-barGroupWidth / 2 + barHalfWidth, barGroupWidth / 2 - barHalfWidth])
-}
-
-function calcDelayGroup (barGroupAmount: number, totalDuration: number) {
-  if (barGroupAmount <= 1) {
-    // 一筆內計算會出錯所以不算
-    return 0
-  }
-  return totalDuration / (barGroupAmount - 1) * groupDelayProportionOfDuration // 依group數量計算
-}
-
-function calctransitionItem (barGroupAmount: number, totalDuration: number) {
-  if (barGroupAmount <= 1) {
-    // 一筆內不會有delay
-    return totalDuration
-  }
-  return totalDuration * (1 - groupDelayProportionOfDuration) // delay後剩餘的時間
-}
-
-function renderTriangleBars ({ selection, barData, zeroY, groupLabels, barScale, params, chartParams, barWidth, delayGroup, transitionItem }: RenderBarParams) {
-  // console.log({ selection, data, zeroY, seriesLabels, barScale, params, chartParams, barWidth, delayGroup })
-  // if (barWidth <= 0) {
-  //   return
-  // }
-  const update = selection!
-    .selectAll<SVGGElement, ComputedDatumGrid[]>(`g.${gClassName}`)
-    .data(barData, (d, i) => groupLabels[i])
-  const enter = update.enter()
-    .append('g')
-    .classed(gClassName, true)
-    .attr('cursor', 'pointer')
-  update.exit().remove()
-  const graphicGroupSelection = update.merge(enter)
-  enter
-    .attr('transform', (d, i) => `translate(${d[0] ? d[0].axisX : 0}, ${0})`)
-  update
-    // .transition()
-    // .duration(200)
-    .attr('transform', (d, i) => `translate(${d[0] ? d[0].axisX : 0}, ${0})`)
-
-  const barHalfWidth = barWidth! / 2
-
-  graphicGroupSelection
-    .each((d, i, g) => {
-      const pathUpdate = d3.select(g[i])
-        .selectAll<SVGGElement, ComputedDatumGrid>('g')
-        .data(d, _d => _d.id)
-      const pathEnter = pathUpdate
-        .enter()
-        .append('g')
-        .classed(gContentClassName, true)
-      pathEnter
-        .append('path')
-        // .attr('transform', `translate(${-barHalfWidth}, 0)`)
-        // .attr('x', d => itemScale(d.itemLabel)!)
-        // .attr('y', d => 0)
-        .style('vector-effect', 'non-scaling-stroke')
-        .attr('height', d => 0)
-        .attr('d', (d) => {
-          const x = barScale(d.seriesLabel)!
-          const y1 = zeroY
-          const y2 = y1
-          return `M${x - (barWidth! / 2)},${y1} L${x},${y2} ${x + (barWidth! / 2)},${y1}`
-        })
-      pathUpdate.merge(pathEnter)
-        .select('path')
-        .style('fill', d => `url(#${d.linearGradientId})`)
-        .attr('stroke', d => d.color)
-        .attr('transform', `translate(${-barHalfWidth}, 0)`)
-        .transition()
-        .duration(transitionItem)
-        .ease(getD3TransitionEase(chartParams.transitionEase))
-        .delay((d, i) => d.groupIndex * delayGroup)
-        .attr('transform', `translate(${-barHalfWidth}, 0)`)
-        // .attr('x', d => itemScale(d.itemLabel)!)
-        // .attr('y', d => -d.y)
-        .attr('height', d => Math.abs(d.axisYFromZero))
-        .attr('d', (d) => {
-          const x = barScale(d.seriesLabel)!
-          const y1 = zeroY
-          const y2 = d.axisY
-          return `M${x},${y1} L${x + (barWidth! / 2)},${y2} ${x + barWidth!},${y1}`
-        })
-        // .on('end', () => initHighlight())
-      pathUpdate.exit().remove()
-    })
-
-  const graphicBarSelection: d3.Selection<SVGPathElement, ComputedDatumGrid, any, any> = graphicGroupSelection.selectAll(`g.${gContentClassName}`)
-
-  return graphicBarSelection
-}
-
-function renderLinearGradient ({ defsSelection, barData, params }: {
-  defsSelection: d3.Selection<SVGDefsElement, ComputedDatumGrid, any, any>
-  barData: BarDatumGrid[][]
-  params: BarsTriangleParams
-}) {
-  const linearGradientUpdate = defsSelection!
-      .selectAll<SVGLinearGradientElement, ComputedDatumGrid>('linearGradient')
-      .data(barData[0] ?? [], d => d.seriesLabel)
-    const linearGradientEnter = linearGradientUpdate
-      .enter()
-      .append('linearGradient')
-      .attr('x1', '0%')
-      .attr('x2', '0%')
-      .attr('y1', '100%')
-      .attr('y2', '0%')
-      .attr('spreadMethod', 'pad')
-    linearGradientUpdate.merge(linearGradientEnter)
-      .attr('id', (d, i) => d.linearGradientId)
-      .html((d, i) => `
-        <stop offset="0%"   stop-color="${d.color}" stop-opacity="${params.linearGradientOpacity[0]}"/>
-        <stop offset="100%" stop-color="${d.color}" stop-opacity="${params.linearGradientOpacity[1]}"/>
-      `)
-    linearGradientUpdate.exit().remove()
-}
-
-
-function renderClipPath ({ defsSelection, clipPathData }: {
-  defsSelection: d3.Selection<SVGDefsElement, any, any, any>
-  clipPathData: ClipPathDatum[]
-}) {
-  const update = defsSelection
-    .selectAll<SVGClipPathElement, Layout>('clipPath')
-    .data(clipPathData)
-  const enter = update.enter()
-    .append('clipPath')
-  const cutRect = update.merge(enter)
-    .attr('id', d => d.id)
-  update.exit().remove()
-
-  update.merge(enter).each((d, i, g) => {
-    const updateRect = d3.select(g[i])
-      .selectAll<SVGRectElement, typeof d>('rect')
-      .data([d])
-    const enterRect = updateRect.enter().append('rect')
-    updateRect.merge(enterRect)
-      .attr('x', 0)
-      .attr('y', 0)
-      .attr('width', _d => _d.width)
-      .attr('height', _d => _d.height)
-    updateRect.exit().remove()
-  })
-}
-
-function highlight ({ selection, ids, fullChartParams }: {
-  selection: d3.Selection<SVGPathElement, ComputedDatumGrid, any, any>
-  ids: string[]
-  fullChartParams: ChartParams
-}) {
-  selection.interrupt('highlight')
-
-  const removeHighlight = () => {
-    selection
-      .transition('highlight')
-      .duration(200)
-      .style('opacity', 1)
-  }
-
-  if (!ids.length) {
-    removeHighlight()
-    return
-  }
-  
-  selection
-    .each((d, i, n) => {
-      if (ids.includes(d.id)) {
-        d3.select(n[i])
-          .style('opacity', 1)
-      } else {
-        d3.select(n[i])
-          .style('opacity', fullChartParams.styles.unhighlightedOpacity)
-      }
-    })
-}
 
 export const BarsTriangle = defineGridPlugin(pluginName, DEFAULT_BARS_TRIANGLE_PLUGIN_PARAMS)(({ selection, name, subject, observer }) => {
   const destroy$ = new Subject()
 
-  const clipPathID = getUniID(pluginName, 'clipPath-box')
-
-  const axisSelection: d3.Selection<SVGGElement, any, any, any> = selection
-    .append('g')
-    .attr('clip-path', `url(#${clipPathID})`)
-  const defsSelection: d3.Selection<SVGDefsElement, ComputedDatumGrid, any, any> = axisSelection.append('defs')
-  const graphicGSelection: d3.Selection<SVGGElement, any, any, any> = axisSelection.append('g')
-  let graphicSelection: d3.Selection<SVGGElement, any, any, any> | undefined
-  const pathSelection$: Subject<d3.Selection<SVGPathElement, ComputedDatumGrid, any, any>> = new Subject()
-  // .style('transform', 'translate(0px, 0px) scale(1)')
-
-  observer.gridAxesTransform$
-    .pipe(
-      takeUntil(destroy$),
-      map(d => d.value),
-      distinctUntilChanged()
-    ).subscribe(d => {
-      axisSelection
-        .style('transform', d)
-    })
-
-  observer.gridGraphicTransform$
-    .pipe(
-      takeUntil(destroy$),
-      switchMap(async d => d.value),
-      distinctUntilChanged()
-    ).subscribe(d => {
-      graphicGSelection
-        .transition()
-        .duration(50)
-        .style('transform', d)
-    })
-
-  // const axisSize$ = gridAxisSizeObservable({
-  //   dataFormatter$,
-  //   observer.layout$
-  // })
-
-  const zeroY$ = observer.computedData$.pipe(
-    map(d => d[0] && d[0][0]
-      ? d[0][0].axisY - d[0][0].axisYFromZero
-      : 0),
-    distinctUntilChanged()
-  )
-
-  const barWidth$ = new Observable<number>(subscriber => {
-    combineLatest({
-      computedData: observer.computedData$,
-      params: observer.fullParams$,
-      axisSize: observer.gridAxesSize$
-    }).pipe(
-      switchMap(async d => d)
-    ).subscribe(data => {
-      const barWidth = calcBarWidth({
-        axisWidth: data.axisSize.width,
-        groupAmount: data.computedData[0] ? data.computedData[0].length : 0,
-        barAmountOfGroup: data.computedData.length,
-        barPadding: data.params.barPadding,
-        barGroupPadding: data.params.barGroupPadding
-      })
-      subscriber.next(barWidth)
-    })
-  }).pipe(
-    takeUntil(destroy$),
-    distinctUntilChanged()
-  )
-
-  const seriesLabels$ = observer.computedData$.pipe(
-    takeUntil(destroy$),
-    map(data => data.map((d, i) => d[0] ? d[0].seriesLabel : String(i)))
-  )
-
-  const groupLabels$ = observer.computedData$.pipe(
-    takeUntil(destroy$),
-    map(data => {
-      return data[0] ? data[0].map(d => d.groupLabel) : []
-    })
-  )
-
-  const barScale$: Observable<d3.ScalePoint<string>> = new Observable(subscriber => {
-    combineLatest({
-      seriesLabels: seriesLabels$,
-      barWidth: barWidth$,
-      params: observer.fullParams$,
-    }).pipe(
-      takeUntil(destroy$),
-      switchMap(async d => d)
-    ).subscribe(data => {
-      const barScale = makeBarScale(data.barWidth, data.seriesLabels, data.params)
-      subscriber.next(barScale)
-    })
-  })
-
-  const transitionDuration$ = observer.fullChartParams$.pipe(
-    takeUntil(destroy$),
-    map(d => d.transitionDuration),
-    distinctUntilChanged()
-  )
-
-  const delayGroup$ = new Observable<number>(subscriber => {
-    combineLatest({
-      groupLabels: groupLabels$,
-      transitionDuration: transitionDuration$,
-    }).pipe(
-      switchMap(async d => d)
-    ).subscribe(data => {
-      const delay = calcDelayGroup(data.groupLabels.length, data.transitionDuration)
-      subscriber.next(delay)
-    })
-  }).pipe(
-    takeUntil(destroy$),
-    distinctUntilChanged()
-  )
-
-  const transitionItem$ = new Observable<number>(subscriber => {
-    combineLatest({
-      groupLabels: groupLabels$,
-      transitionDuration: transitionDuration$
-    }).pipe(
-      switchMap(async d => d)
-    ).subscribe(data => {
-      const transition = calctransitionItem(data.groupLabels.length, data.transitionDuration)
-      subscriber.next(transition)
-    })
-  }).pipe(
-    takeUntil(destroy$),
-    distinctUntilChanged()
-  )
-
-  const transposedData$ = observer.computedData$.pipe(
-    takeUntil(destroy$),
-    map(data => {
-      // 取得原始陣列的維度
-      const rows = data.length;
-      const cols = data.reduce((prev, current) => {
-        return Math.max(prev, current.length)
-      }, 0)
-
-      // 初始化轉換後的陣列
-      const transposedArray: typeof data = new Array(cols).fill(null).map(() => new Array(rows).fill(null))
-
-      // 遍歷原始陣列，進行轉換
-      for (let i = 0; i < rows; i++) {
-          for (let j = 0; j < cols; j++) {
-              transposedArray[j][i] = data[i][j]
-          }
-      }
-
-      return transposedArray
-    })
-  )
-
-  const barData$ = transposedData$.pipe(
-    takeUntil(destroy$),
-    map(data => {
-      const linearGradientIds = data.length
-        ? data[0].map(d => getUniID(name, `lineargradient-${d.seriesLabel}`))
-        : []
-      return data.map(d => {
-        return d.map((_d, _i) => {
-          return <BarDatumGrid>{
-            linearGradientId: linearGradientIds[_i],
-            ..._d
-          }
-        })
-      })
-    })
-  )
-
-  observer.gridAxesSize$.pipe(
-    takeUntil(destroy$)
-  ).subscribe(data => {
-    const clipPathData = [{
-      id: clipPathID,
-      width: data.width,
-      height: data.height
-    }]
-    renderClipPath({
-      defsSelection,
-      clipPathData
-    })
+  const unsubscribeBaseBars = createBaseBarsTriangle(pluginName, {
+    selection,
+    computedData$: observer.computedData$,
+    SeriesDataMap$: observer.SeriesDataMap$,
+    GroupDataMap$: observer.GroupDataMap$,
+    fullParams$: observer.fullParams$,
+    fullChartParams$: observer.fullChartParams$,
+    gridAxesTransform$: observer.gridAxesTransform$,
+    gridGraphicTransform$: observer.gridGraphicTransform$,
+    gridAxesSize$: observer.gridAxesSize$,
+    gridHighlight$: observer.gridHighlight$,
+    event$: subject.event$,
   })
 
-  // const SeriesDataMap$ = observer.computedData$.pipe(
-  //   map(d => makeGridSeriesDataMap(d))
-  // )
-
-  // const GroupDataMap$ = observer.computedData$.pipe(
-  //   map(d => makeGridGroupDataMap(d))
-  // )
-
-  const highlightTarget$ = observer.fullChartParams$.pipe(
-    takeUntil(destroy$),
-    map(d => d.highlightTarget),
-    distinctUntilChanged()
-  )
-
-  combineLatest({
-    barData: barData$,
-    computedData: observer.computedData$,
-    zeroY: zeroY$,
-    groupLabels: groupLabels$,
-    barScale: barScale$,
-    params: observer.fullParams$,
-    chartParams: observer.fullChartParams$,
-    highlightTarget: highlightTarget$,
-    barWidth: barWidth$,
-    delayGroup: delayGroup$,
-    transitionItem: transitionItem$,
-    SeriesDataMap: observer.SeriesDataMap$,
-    GroupDataMap: observer.GroupDataMap$
-  }).pipe(
-    takeUntil(destroy$),
-    // 轉換後會退訂前一個未完成的訂閱事件，因此可以取到「同時間」最後一次的訂閱事件
-    switchMap(async (d) => d),
-  ).subscribe(data => {
-    const pathSelection = renderTriangleBars({
-      selection: graphicGSelection,
-      barData: data.barData,
-      zeroY: data.zeroY,
-      groupLabels: data.groupLabels,
-      barScale: data.barScale,
-      params: data.params,
-      chartParams: data.chartParams,
-      barWidth: data.barWidth,
-      delayGroup: data.delayGroup,
-      transitionItem: data.transitionItem
-    })
-    renderLinearGradient({
-      defsSelection,
-      barData: data.barData,
-      params: data.params
-    })
-
-    pathSelection!
-      .on('mouseover', (event, datum) => {
-        event.stopPropagation()
-
-        subject.event$.next({
-          type: 'grid',
-          eventName: 'mouseover',
-          pluginName: name,
-          highlightTarget: data.highlightTarget,
-          datum,
-          series: data.SeriesDataMap.get(datum.seriesLabel)!,
-          seriesIndex: datum.seriesIndex,
-          seriesLabel: datum.seriesLabel,
-          groups: data.GroupDataMap.get(datum.groupLabel)!,
-          groupIndex: datum.groupIndex,
-          groupLabel: datum.groupLabel,
-          event,
-          data: data.computedData
-        })
-      })
-      .on('mousemove', (event, datum) => {
-        event.stopPropagation()
-
-        subject.event$.next({
-          type: 'grid',
-          eventName: 'mousemove',
-          pluginName: name,
-          highlightTarget: data.highlightTarget,
-          datum,
-          series: data.SeriesDataMap.get(datum.seriesLabel)!,
-          seriesIndex: datum.seriesIndex,
-          seriesLabel: datum.seriesLabel,
-          groups: data.GroupDataMap.get(datum.groupLabel)!,
-          groupIndex: datum.groupIndex,
-          groupLabel: datum.groupLabel,
-          event,
-          data: data.computedData
-        })
-      })
-      .on('mouseout', (event, datum) => {
-        event.stopPropagation()
-
-        subject.event$.next({
-          type: 'grid',
-          eventName: 'mouseout',
-          pluginName: name,
-          highlightTarget: data.highlightTarget,
-          datum,
-          series: data.SeriesDataMap.get(datum.seriesLabel)!,
-          seriesIndex: datum.seriesIndex,
-          seriesLabel: datum.seriesLabel,
-          groups: data.GroupDataMap.get(datum.groupLabel)!,
-          groupIndex: datum.groupIndex,
-          groupLabel: datum.groupLabel,
-          event,
-          data: data.computedData
-        })
-      })
-      .on('click', (event, datum) => {
-        event.stopPropagation()
-
-        subject.event$.next({
-          type: 'grid',
-          eventName: 'click',
-          pluginName: name,
-          highlightTarget: data.highlightTarget,
-          datum,
-          series: data.SeriesDataMap.get(datum.seriesLabel)!,
-          seriesIndex: datum.seriesIndex,
-          seriesLabel: datum.seriesLabel,
-          groups: data.GroupDataMap.get(datum.groupLabel)!,
-          groupIndex: datum.groupIndex,
-          groupLabel: datum.groupLabel,
-          event,
-          data: data.computedData
-        })
-      })
-
-    pathSelection$.next(pathSelection)
-  })
-
-  // const datumList$ = observer.computedData$.pipe(
-  //   takeUntil(destroy$),
-  //   map(d => d.flat())
-  // )
-  // const highlight$ = highlightObservable({ datumList$, observer.fullChartParams$, event$: subject.event$ })
-  const highlightSubscription = observer.gridHighlight$.subscribe()
-  
-  combineLatest({
-    pathSelection: pathSelection$,
-    highlight: observer.gridHighlight$,
-    fullChartParams: observer.fullChartParams$
-  }).pipe(
-    takeUntil(destroy$),
-    switchMap(async d => d)
-  ).subscribe(data => {
-    highlight({
-      selection: data.pathSelection,
-      ids: data.highlight,
-      fullChartParams: data.fullChartParams
-    })
-  })
-
-
   return () => {
     destroy$.next(undefined)
-    highlightSubscription.unsubscribe()
+    unsubscribeBaseBars()
   }
 })