All files / components/hooks/lineChart useLineChartXscale.ts

0% Statements 0/23
0% Branches 0/24
0% Functions 0/14
0% Lines 0/15

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60                                                                                                                       
import { scaleLinear, scaleTime, min, max } from "d3";
import { useMemo } from "react";
import { LineChartDataItem } from "src/types/data";
 
interface Props {
  filteredDataSet: LineChartDataItem[];
  width: number;
  margin: { left: number; right: number };
  xAxisDataType: "number" | "date_annual" | "date_monthly";
}
 
const useLineChartXscale = (
  filteredDataSet: LineChartDataItem[],
  width: number,
  margin: { left: number; right: number },
  xAxisDataType: "number" | "date_annual" | "date_monthly"
) => {
  return useMemo(() => {
    if (xAxisDataType === "number") {
      return scaleLinear()
        .domain([
          min(filteredDataSet.flatMap(item => item.series.map(d => Number(d.date)))) || 0,
          max(filteredDataSet.flatMap(item => item.series.map(d => Number(d.date)))) || 1,
        ])
        .range([margin.left, width - margin.right])
        .clamp(true)
        .nice();
    }
 
    if (xAxisDataType === "date_annual") {
      // sometimes the first tick is missing, so do a hack here
      const minDate = min(
        filteredDataSet.flatMap(item => item.series.map(d => new Date(`${d.date}-01-01`)))
      );
      const maxDate = max(
        filteredDataSet.flatMap(item => item.series.map(d => new Date(`${d.date}`)))
      );
 
      return scaleTime()
        .domain([
          minDate instanceof Date ? minDate : new Date(minDate || 0),
          maxDate instanceof Date ? maxDate : new Date(maxDate || 1),
        ])
        .range([margin.left, width - margin.right]);
    }
 
    const minDate = min(filteredDataSet.flatMap(item => item.series.map(d => new Date(d.date))));
    const maxDate = max(filteredDataSet.flatMap(item => item.series.map(d => new Date(d.date))));
 
    return scaleTime()
      .domain([
        minDate instanceof Date ? minDate : new Date(minDate || 0),
        maxDate instanceof Date ? maxDate : new Date(maxDate || 1),
      ])
      .range([margin.left, width - margin.right]);
  }, [filteredDataSet, width, margin, xAxisDataType]);
};
 
export default useLineChartXscale;