import { defineChart } from '@tanstack/charts'
import { angleGrid, polar, radialDot, radialGrid } from '@tanstack/charts/polar'
import { scaleLinear } from 'd3-scale'
import { windDirection, windLatitudeBand, windSpeed } from './transform'
import { tanstackMount } from '../../shared/mount'
import type { ConformanceInput } from '../../types'

const angleDomain = [0, 360] as const
const radiusDomain = [0, 13] as const
const angleGridValues = [0, 45, 90, 135, 180, 225, 270, 315] as const
const radiusGridValues = [3, 6, 9, 12] as const
const dotColor = '#e11d48'
const gridColor = '#94a3b8'

const definition = (input: ConformanceInput) => {
  const rows = windLatitudeBand(input.revision)

  return defineChart({
    marks: [
      polar({
        radiusRatio: 0.72,
        angle: { scale: scaleLinear().domain(angleDomain) },
        radius: { scale: scaleLinear().domain(radiusDomain) },
        guides: [
          radialGrid({
            values: radiusGridValues,
            labels: false,
            stroke: gridColor,
            strokeOpacity: 0.35,
          }),
          angleGrid({
            values: angleGridValues,
            labels: false,
            stroke: gridColor,
            strokeOpacity: 0.35,
          }),
        ],
        marks: [
          radialDot(rows, {
            angle: windDirection,
            radius: windSpeed,
            r: 4.5,
            fill: dotColor,
            stroke: '#ffffff',
            strokeWidth: 1,
          }),
        ],
      }),
    ],
    margin: 0,
  })
}

export const mount = tanstackMount(definition, 'Surface wind polar scatter', {
  format: ({ datum }) =>
    `${datum.latitude}°N, ${datum.longitude}°E · ${windSpeed(datum).toFixed(1)} m/s`,
})

import { wind } from '@charts-poc/demo-data/wind'
import type { WindRow } from '@charts-poc/demo-data/wind'

const selectableLatitudeBands = [48.125, 55.125] as const

export function windLatitudeBand(revision = 0): readonly WindRow[] {
  const latitude =
    selectableLatitudeBands[revision % selectableLatitudeBands.length] ??
    selectableLatitudeBands[0]
  return wind.filter((row) => row.latitude === latitude)
}

export function windDirection(row: WindRow): number {
  return (Math.atan2(row.v, row.u) * (180 / Math.PI) + 360) % 360
}

export function windSpeed(row: WindRow): number {
  return Math.hypot(row.u, row.v)
}