A few days ago Kiran B sent me a challenge via Twitter DM: How can I make a drilldown piechart?
The annoyed emoji is a bit much but I’ll allow it. The challenge was fun to solve. Got it roughly working in a 45min stream 👇
You can also try it out via CodeSandbox. I’m surprised how nice default D3 category colors work with a black border. It’s gorgeous 😍
Here’s how a drilldown piechart works
You can think of a drilldown piechart as a recursive data visualization. Take a tree structure of data and you render a piechart for each level. Click on a slice, drill down into that branch.
Easiest way to get started is to generate that dataset from scratch ✌️
Random number of datapoints,
N, create an array with
d3.range, loop through, generate a datapoint for each step. Random value, some identifier, save the level of recursing, the index, the parent index, a name, and then generate some children.
Generating children calls this same function thus creating a recursive data structure.
We’ll use the level and index and previous index to help us pick slice colors later on.
I like to keep my
<App> component simple so using the data and rendering the piechart looks like this:
An svg element with a
Ok the colors aren’t perfect, the drilling down is confusing without an animation, and you can’t drill up just yet. But it’s a great start!
Three components play together:
- A styled
<Path>component handles colors
<Arc>component renders arcs, deals with mouseovers, and registers clicks
<DrilldownPie>component holds it all together and deals with drilling
<Path> component is a styled SVG path that uses props to decide the color of each slice.
Level 4 is the top level 👉 use a color scheme from D3 and the slice index to pick a color.
With other levels the idea was to take the base color and make it progressively brighter based on the index. Works okay for the first level down, didn’t quite work later on.
Might have to change how we approach this. Maybe colors as part of the dataset itself?
<Arc> component renders a single slice of the piechart and handles mouseover events using hooks to change size. Messing with the radius creates the effect of individual slices popping out.
That’s the whole component. Let’s break it down 🙂
First we’ve got the state management: A single
radiusAdd variable that tells us how much to add or take away from the default radius of our arc.
Second we’ve got the
d3.arc generator. Takes some configuration and produces path definitions using SVG’s crazy path language.
Third we’ve got 2 functions calling the
setRadiusAdd setter to change the radius. That triggers a re-render, the arc generator sees the new value, creates a different path definition 👌
Fourth we render our styled
<Path> component making sure to pass in all the right props. The path definition, the level and index for color, and all the event listeners.
<DrilldownPie> component ties this all together.
Less code even than the Arc component god I love hooks 😱
First we’ve got state. We copy our data into it. That’s so we can do the drilling down. Haven’t tried how this plays with data changing from the outside … trigger a re-render I assume 🤔
Second we’ve got a
d3.pie generator. Takes our data and creates an array of objects perfectly suited for the
d3.arc generator. Angles and values and all.
drilldown function uses our state setter to move data one level down. Take the children, set them as current data, trigger re-render, magic. 🧙♂️
Fourth render it all in a loop.
And that’s how you build a drilldown piechart with React and D3 and hooks.
I’m going to continue this challenge either Friday night (today) or Sunday afternoon. Want to figure out how to make the drilldown transition animated and get drilling up to work.
Drilling up is easy in principle, but gets tricky if you want to support arbitrary levels. That’s because we aren’t using real recursion to drill down 😅
See you at the stream maybe?
Learned something new? Want to improve your skills?
Join over 10,000 engineers just like you already improving their skills!
Here's how it works 👇
PS: You should also follow me on twitter 👉 here.
It's where I go to shoot the shit about programming.