@mui/x-charts/esm/ChartsLegend/useAxis.js

The community edition of the Charts components (MUI X).

import * as React from 'react';
import { ZAxisContext } from '../context/ZAxisContextProvider';
import { useCartesianContext } from '../context/CartesianProvider/useCartesianContext';
/**
 * Helper to select an axis definition according to its direction and id.
 */
export function useAxis({
  axisDirection,
  axisId
}) {
  const {
    xAxis,
    xAxisIds,
    yAxis,
    yAxisIds
  } = useCartesianContext();
  const {
    zAxis,
    zAxisIds
  } = React.useContext(ZAxisContext);
  switch (axisDirection) {
    case 'x':
      {
        const id = typeof axisId === 'string' ? axisId : xAxisIds[axisId ?? 0];
        return xAxis[id];
      }
    case 'y':
      {
        const id = typeof axisId === 'string' ? axisId : yAxisIds[axisId ?? 0];
        return yAxis[id];
      }
    case 'z':
    default:
      {
        const id = typeof axisId === 'string' ? axisId : zAxisIds[axisId ?? 0];
        return zAxis[id];
      }
  }
}