@@ -7,6 +7,7 @@ import { MakeOptional } from '@mui/x-internals/types';
77import { interpolateRgbBasis } from '@mui/x-charts-vendor/d3-interpolate' ;
88import { ChartsAxis , ChartsAxisProps } from '@mui/x-charts/ChartsAxis' ;
99import { ChartsTooltipProps } from '@mui/x-charts/ChartsTooltip' ;
10+ import { ChartsSurface } from '@mui/x-charts/ChartsSurface' ;
1011import {
1112 ChartsAxisSlots ,
1213 ChartsAxisSlotProps ,
@@ -22,13 +23,14 @@ import {
2223 ChartsOverlaySlots ,
2324} from '@mui/x-charts/ChartsOverlay' ;
2425import { DEFAULT_X_AXIS_KEY , DEFAULT_Y_AXIS_KEY } from '@mui/x-charts/constants' ;
25- import { ChartContainerPro , ChartContainerProProps } from '../ChartContainerPro' ;
26+ import { ChartContainerProProps } from '../ChartContainerPro' ;
2627import { HeatmapSeriesType } from '../models/seriesType/heatmap' ;
2728import { HeatmapPlot } from './HeatmapPlot' ;
2829import { seriesConfig as heatmapSeriesConfig } from './seriesConfig' ;
2930import { HeatmapTooltip , HeatmapTooltipProps } from './HeatmapTooltip/HeatmapTooltip' ;
3031import { HeatmapItemSlotProps , HeatmapItemSlots } from './HeatmapItem' ;
3132import { HEATMAP_PLUGINS , HeatmapPluginsSignatures } from './Heatmap.plugins' ;
33+ import { ChartDataProviderPro } from '../ChartDataProviderPro' ;
3234
3335export interface HeatmapSlots extends ChartsAxisSlots , ChartsOverlaySlots , HeatmapItemSlots {
3436 /**
@@ -181,8 +183,7 @@ const Heatmap = React.forwardRef(function Heatmap(
181183 const Tooltip = props . slots ?. tooltip ?? HeatmapTooltip ;
182184
183185 return (
184- < ChartContainerPro < 'heatmap' , HeatmapPluginsSignatures >
185- ref = { ref }
186+ < ChartDataProviderPro < 'heatmap' , HeatmapPluginsSignatures >
186187 seriesConfig = { seriesConfig }
187188 series = { series . map ( ( s ) => ( {
188189 type : 'heatmap' ,
@@ -196,23 +197,24 @@ const Heatmap = React.forwardRef(function Heatmap(
196197 zAxis = { zAxisWithDefault }
197198 colors = { colors }
198199 dataset = { dataset }
199- sx = { sx }
200200 disableAxisListener
201201 highlightedItem = { highlightedItem }
202202 onHighlightChange = { onHighlightChange }
203203 onAxisClick = { onAxisClick }
204204 plugins = { HEATMAP_PLUGINS }
205205 >
206- < g clipPath = { `url(# ${ clipPathId } )` } >
207- < HeatmapPlot slots = { slots } slotProps = { slotProps } / >
208- < ChartsOverlay loading = { loading } slots = { slots } slotProps = { slotProps } />
209- </ g >
210- < ChartsAxis slots = { slots } slotProps = { slotProps } / >
211- { ! loading && < Tooltip { ... slotProps ?. tooltip } /> }
206+ < ChartsSurface ref = { ref } sx = { sx } >
207+ < g clipPath = { `url(# ${ clipPathId } )` } >
208+ < HeatmapPlot slots = { slots } slotProps = { slotProps } />
209+ < ChartsOverlay loading = { loading } slots = { slots } slotProps = { slotProps } / >
210+ </ g >
211+ < ChartsAxis slots = { slots } slotProps = { slotProps } />
212212
213- < ChartsClipPath id = { clipPathId } />
214- { children }
215- </ ChartContainerPro >
213+ < ChartsClipPath id = { clipPathId } />
214+ { children }
215+ </ ChartsSurface >
216+ { ! loading && < Tooltip { ...slotProps ?. tooltip } /> }
217+ </ ChartDataProviderPro >
216218 ) ;
217219} ) ;
218220
0 commit comments