@orbcharts/plugins-basic 3.0.0-beta.15 → 3.0.0-beta.16

package/src/base/BaseRacingLabels.ts

@@ -0,0 +1,392 @@
+import * as d3 from 'd3'
+import {
+  Observable,
+  Subject,
+  combineLatest,
+  takeUntil,
+  map,
+  distinctUntilChanged,
+  of,
+  iif,
+  switchMap,
+  shareReplay
+} from 'rxjs'
+import type {
+  ChartParams,
+  ComputedDatumMultiValue,
+  ComputedDataMultiValue,
+  ComputedXYDatumMultiValue,
+  ContainerSize,
+  ContainerPositionScaled,
+  DataFormatterMultiValue,
+  Layout
+} from '../../lib/core-types'
+import type { BaseRacingLabelsParams } from '../../lib/plugins-basic-types'
+import type { BasePluginFn } from './types'
+import { getColor, getClassName, getUniID } from '../utils/orbchartsUtils'
+import { multiValueContainerSelectionsObservable } from '../multiValue/multiValueObservables'
+
+interface BaseRacingAxisContext {
+  selection: d3.Selection<any, unknown, any, unknown>
+  computedData$: Observable<ComputedDataMultiValue>
+  visibleComputedRankingData$: Observable<ComputedDatumMultiValue[][]>
+  rankingScaleList$: Observable<Array<d3.ScalePoint<string>>>
+  xScale$: Observable<(n: number) => number>
+  fullParams$: Observable<BaseRacingLabelsParams>
+  fullDataFormatter$: Observable<DataFormatterMultiValue>
+  fullChartParams$: Observable<ChartParams>
+  // layout$: Observable<Layout>
+  containerPosition$: Observable<ContainerPositionScaled[]>
+  containerSize$: Observable<ContainerSize>
+  isCategorySeprate$: Observable<boolean>
+  // xyValueIndex$: Observable<[number, number]>
+}
+
+type ClipPathDatum = {
+  id: string;
+  // x: number;
+  // y: number;
+  width: number;
+  height: number;
+}
+
+// const pluginName = 'RacingAxis'
+
+const yTickTextAnchor = 'end'
+const yTickDominantBaseline = 'middle'
+const yAxisLabelAnchor = 'end'
+const yAxisLabelDominantBaseline = 'auto'
+// const textClassName = getClassName(pluginName, 'yLabel')
+
+function renderRacingAxisLabel ({ selection, textClassName, fullParams, containerSize, fullDataFormatter, fullChartParams, textReverseTransform }: {
+  selection: d3.Selection<SVGGElement, any, any, any>,
+  textClassName: string
+  fullParams: BaseRacingLabelsParams
+  // axisLabelAlign: TextAlign
+  containerSize: ContainerSize
+  fullDataFormatter: DataFormatterMultiValue,
+  fullChartParams: ChartParams
+  textReverseTransform: string,
+}) {
+  const offsetX = fullParams.barLabel.padding - fullParams.axisLabel.offset[0]
+  const offsetY = - fullParams.barLabel.padding - fullParams.axisLabel.offset[1]
+  let labelX = - offsetX
+  let labelY = - offsetY
+
+  selection
+    .attr('transform', d => `translate(0, ${containerSize.height})`)
+    .selectAll<SVGTextElement, BaseRacingLabelsParams>(`text`)
+    .data([fullParams])
+    .join(
+      enter => {
+        return enter
+          .append('text')
+          .style('font-weight', 'bold')
+      },
+      update => update,
+      exit => exit.remove()
+    )
+    .attr('text-anchor', yAxisLabelAnchor)
+    .attr('dominant-baseline', yAxisLabelDominantBaseline)
+    .attr('font-size', fullChartParams.styles.textSize)
+    .style('fill', getColor(fullParams.axisLabel.colorType, fullChartParams))
+    // .style('transform', textReverseTransform)
+    // 偏移使用 x, y 而非 transform 才不會受到外層 scale 變形影響
+    .attr('x', labelX)
+    .attr('y', labelY)
+    .text(d => fullDataFormatter.yAxis.label)
+}
+
+function renderRacingLabels ({ selection, fullParams, fullChartParams, rankingScale, valueScale, categoryData }: {
+  selection: d3.Selection<SVGGElement, any, any, any>,
+  fullParams: BaseRacingLabelsParams
+  fullChartParams: ChartParams
+  rankingScale: d3.ScalePoint<string>
+  valueScale: ((n: number) => number)
+  categoryData: ComputedDatumMultiValue[]
+  // textReverseTransformWithRotate: string,
+  // xyValueIndex: [number, number]
+}) {
+  const labelData = fullParams.barLabel.position === 'none' ? [] : categoryData
+
+  const labelSelection = selection
+    .selectAll<SVGGElement, string>(`text`)
+    .data(labelData, (d: ComputedDatumMultiValue) => d.id)
+    // .join('g')
+    // .classed(boxClassName, true)
+    .join(
+      enter => {
+        return enter
+          .append('text')
+          .style('font-weight', 'bold')
+          .attr('x', d => valueScale(d.value[d.xValueIndex] ?? 0) - fullParams.barLabel.padding)
+          .attr('y', d => rankingScale(d.label)!)
+      },
+      update => {
+        return update
+          .transition()
+          .duration(fullChartParams.transitionDuration)
+          .ease(d3.easeLinear)
+          // 偏移使用 x, y 而非 transform 才不會受到外層 scale 變形影響
+          .attr('x', d => valueScale(d.value[d.xValueIndex] ?? 0) - fullParams.barLabel.padding)
+          .attr('y', d => rankingScale(d.label)!)
+      },
+      exit => exit.remove()
+    )
+    .attr('text-anchor', yTickTextAnchor)
+    .attr('dominant-baseline', yTickDominantBaseline)
+    .attr('font-size', fullChartParams.styles.textSize)
+    .style('fill', getColor(fullParams.barLabel.colorType, fullChartParams))
+    // .style('transform', textReverseTransformWithRotate)
+    .text(d => d.label)
+
+  return labelSelection
+}
+
+
+function renderClipPath ({ defsSelection, clipPathData }: {
+  defsSelection: d3.Selection<SVGDefsElement, any, any, any>
+  clipPathData: ClipPathDatum[]
+  // textReverseTransform: string
+}) {
+  const clipPath = defsSelection
+    .selectAll<SVGClipPathElement, Layout>('clipPath')
+    .data(clipPathData)
+    .join(
+      enter => {
+        return enter
+          .append('clipPath')
+      },
+      update => update,
+      exit => exit.remove()
+    )
+    .attr('id', d => d.id)
+    // .attr('transform', textReverseTransform)
+    .each((d, i, g) => {
+      const rect = d3.select(g[i])
+        .selectAll<SVGRectElement, typeof d>('rect')
+        .data([d])
+        .join(
+          enter => {
+            return enter
+              .append('rect')
+          },
+          update => update,
+          exit => exit.remove()
+        )
+        .attr('x', _d => - _d.width)
+        .attr('y', 0)
+        .attr('width', _d => _d.width * 2)
+        .attr('height', _d => _d.height)
+    })
+}
+
+export const createBaseRacingLabels: BasePluginFn<BaseRacingAxisContext> = (pluginName: string, {
+  selection,
+  computedData$,
+  visibleComputedRankingData$,
+  rankingScaleList$,
+  xScale$,
+  fullParams$,
+  fullDataFormatter$,
+  fullChartParams$,
+  // layout$,
+  containerPosition$,
+  containerSize$,
+  isCategorySeprate$,
+  // xyValueIndex$
+}) => {
+
+  const destroy$ = new Subject()
+  
+  // const containerClassName = getClassName(pluginName, 'container')
+  const boxClassName = getClassName(pluginName, 'box')
+  const textClassName = getClassName(pluginName, 'text')
+  const clipPathID = getUniID(pluginName, 'clipPath-box')
+
+  const containerSelection$ = multiValueContainerSelectionsObservable({
+    selection,
+    pluginName,
+    clipPathID: null,
+    computedData$,
+    containerPosition$,
+    isCategorySeprate$,
+  }).pipe(
+    takeUntil(destroy$),
+  )
+
+  // const containerSelection$ = combineLatest({
+  //   computedData: computedData$.pipe(
+  //     distinctUntilChanged((a, b) => {
+  //       // 只有當series的數量改變時，才重新計算
+  //       return a.length === b.length
+  //     }),
+  //   ),
+  //   isCategorySeprate: isCategorySeprate$
+  // }).pipe(
+  //   takeUntil(destroy$),
+  //   switchMap(async (d) => d),
+  //   map(data => {
+  //     return data.isCategorySeprate
+  //       // category分開的時候顯示各別axis
+  //       ? data.computedData
+  //       // category合併的時候只顯示第一個axis
+  //       : [data.computedData[0]]
+  //   }),
+  //   map((computedData, i) => {
+  //     return selection
+  //       .selectAll<SVGGElement, ComputedDatumMultiValue[]>(`g.${containerClassName}`)
+  //       .data(computedData, d => d[0] ? d[0].categoryIndex : i)
+  //       .join('g')
+  //       .classed(containerClassName, true)
+  //   })
+  // )
+
+  // combineLatest({
+  //   containerSelection: containerSelection$,
+  //   containerPosition: containerPosition$
+  // }).pipe(
+  //   takeUntil(destroy$),
+  //   switchMap(async d => d)
+  // ).subscribe(data => {
+  //   data.containerSelection
+  //     .attr('transform', (d, i) => {
+  //       const containerPosition = data.containerPosition[i] ?? data.containerPosition[0]
+  //       const translate = containerPosition.translate
+  //       const scale = containerPosition.scale
+  //       // return `translate(${translate[0]}, ${translate[1]}) scale(${scale[0]}, ${scale[1]})`
+  //       return `translate(${translate[0]}, ${translate[1]})`
+  //     })
+  //     // .attr('opacity', 0)
+  //     // .transition()
+  //     // .attr('opacity', 1)
+  // })
+
+  const textReverseTransform$ = containerPosition$.pipe(
+    takeUntil(destroy$),
+    switchMap(async (d) => d),
+    map(containerPosition => {
+      // const axesRotateXYReverseValue = `rotateX(${data.gridAxesReverseTransform.rotateX}deg) rotateY(${data.gridAxesReverseTransform.rotateY}deg)`
+      // const axesRotateReverseValue = `rotate(${data.gridAxesReverseTransform.rotate}deg)`
+      const containerScaleReverseValue = `scale(${1 / containerPosition[0].scale[0]}, ${1 / containerPosition[0].scale[1]})`
+      // 抵消最外層scale
+      return `${containerScaleReverseValue}`
+    }),
+    distinctUntilChanged()
+  )
+
+  // const textReverseTransformWithRotate$ = combineLatest({
+  //   textReverseTransform: textReverseTransform$,
+  //   fullParams: fullParams$,
+  // }).pipe(
+  //   takeUntil(destroy$),
+  //   switchMap(async (d) => d),
+  //   map(data => {
+  //     // 必須按照順序（先抵消外層rotate，再抵消最外層scale，最後再做本身的rotate）
+  //     return `${data.textReverseTransform} rotate(${data.fullParams.barLabel.rotate}deg)`
+  //   })
+  // )
+
+  // const rankingLabelList$ = visibleComputedRankingData$.pipe(
+  //   takeUntil(destroy$),
+  //   map(data => {
+  //     return data.map(categoryData => categoryData.map(d => d.label))
+  //   })
+  // )
+
+  const valueScale$: Observable<((n: number) => number)> = fullParams$.pipe(
+    map(fullParams => fullParams.barLabel.position),
+    distinctUntilChanged(),
+    switchMap(position => {
+      return iif(
+        () => position === 'inside',
+        xScale$,
+        of(() => 0)
+      )
+    })
+  )
+
+  containerSize$.subscribe(data => {
+    const defsSelection = selection.selectAll<SVGDefsElement, any>('defs')
+      .data([clipPathID])
+      .join('defs')
+    const clipPathData = [{
+      id: clipPathID,
+      width: data.width,
+      height: data.height
+    }]
+    renderClipPath({
+      defsSelection: defsSelection,
+      clipPathData,
+      // textReverseTransform: data.textReverseTransform
+    })
+  })
+
+  combineLatest({
+    containerSelection: containerSelection$,
+    fullParams: fullParams$,
+    // layout: layout$,
+    containerSize: containerSize$,
+    fullDataFormatter: fullDataFormatter$,
+    fullChartParams: fullChartParams$,
+    visibleComputedRankingData: visibleComputedRankingData$,
+    // rankingLabelList: rankingLabelList$,
+    rankingScaleList: rankingScaleList$,
+    valueScale: valueScale$,
+    textReverseTransform: textReverseTransform$,
+    // textReverseTransformWithRotate: textReverseTransformWithRotate$,
+    // xyValueIndex: xyValueIndex$
+  }).pipe(
+    takeUntil(destroy$),
+    switchMap(async (d) => d),
+  ).subscribe(data => {
+
+    data.containerSelection.each((d, i, g) => {
+      const _containerSelection = d3.select(g[i])
+      // const rankingLabels = data.rankingLabelList[i]
+      const rankingScale = data.rankingScaleList[i]
+      if (!rankingScale) {
+        return
+      }
+      
+      // const containerClipPathID = `${clipPathID}-${i}`  
+      const axisSelection = _containerSelection
+        .selectAll<SVGGElement, any>(`g.${boxClassName}`)
+        .data([i])
+        .join('g')
+        .attr('class', boxClassName)
+        .attr('clip-path', `url(#${clipPathID})`)
+      const axisLabelSelection = _containerSelection
+        .selectAll<SVGGElement, BaseRacingLabelsParams>(`g.${textClassName}`)
+        .data([data.fullParams])
+        .join('g')
+        .classed(textClassName, true)
+
+      renderRacingLabels({
+        selection: axisSelection,
+        fullParams: data.fullParams,
+        fullChartParams: data.fullChartParams,
+        rankingScale: rankingScale,
+        categoryData: data.visibleComputedRankingData[i],
+        // textReverseTransformWithRotate: data.textReverseTransformWithRotate,
+        valueScale: data.valueScale,
+        // xyValueIndex: data.xyValueIndex,
+      })
+  
+      renderRacingAxisLabel({
+        selection: axisLabelSelection,
+        textClassName,
+        fullParams: data.fullParams,
+        containerSize: data.containerSize,
+        fullDataFormatter: data.fullDataFormatter,
+        fullChartParams: data.fullChartParams,
+        textReverseTransform: data.textReverseTransform,
+      })
+    })
+
+  })
+
+  return () => {
+    destroy$.next(undefined)
+  }
+}