Multi Level Tree Map

pythondemo\multileveltreemap.py
#!/usr/bin/python

# The ChartDirector for Python module is assumed to be in "../lib"
import sys, os
sys.path.insert(0, os.path.join(os.path.abspath(sys.path[0]), "..", "lib"))

from pychartdir import *

# The first level nodes of the tree map. There are 5 nodes.
animals = ["Fish", "Amphibian", "Reptile", "Bird", "Mammal"]

# In this example, the colors are based on the first level nodes.
colors = [0xff5555, 0xff9933, 0xffff44, 0x66ff66, 0x44ccff]

# Data for the second level nodes in "Fish"
fish_names = ["Shark", "Ray", "Swordfish", "Sawfish", "Eel", "Lionfish", "Seahorse"]
fish_data = [170, 144, 109, 115, 75, 45, 54]

# Data for the second level nodes in "Bird"
bird_names = ["Swan", "Ostrich", "Eagle", "Penguin", "Kiwi", "Flamingo", "Owl", "Peacock"]
bird_data = [89, 64, 94, 106, 68, 81, 40, 73]

# Data for the second level nodes in "Amphibian"
amphibian_names = ["Toad", "Salamander", "Frog", "Caecilian"]
amphibian_data = [67, 47, 58, 36]

# Data for the second level nodes in "Reptile"
reptile_names = ["Turtle", "Crocodile", "Lizard", "Snake"]
reptile_data = [58, 154, 97, 41]

# Data for the second level nodes in "Mammal"
mammal_names = ["Big Cat", "Primate", "Panda", "Elephant", "Hippo", "Rhino", "Giraffe"]
mammal_data = [266, 207, 160, 194, 168, 149, 202]

# Create a Tree Map object of size 600 x 600 pixels
c = TreeMapChart(600, 600)

# Add a title to the chart
c.addTitle("Animal Kingdom Census", "Arial Bold Italic", 18)

# Set the plotarea at (30, 30) and of size 540 x 540 pixels
c.setPlotArea(30, 30, 540, 540)

# Obtain the root of the tree map, which is the entire plot area
root = c.getRootNode()

# Add first level nodes to the root. We do not need to provide data as they will be computed as the
# sum of the second level nodes.
root.setData(None, animals, colors)

# Add second level nodes to each of the first level node
root.getNode(0).setData(fish_data, fish_names)
root.getNode(1).setData(amphibian_data, amphibian_names)
root.getNode(2).setData(reptile_data, reptile_names)
root.getNode(3).setData(bird_data, bird_names)
root.getNode(4).setData(mammal_data, mammal_names)

# Get the prototype (template) for the first level nodes.
nodeConfig = c.getLevelPrototype(1)

# Set the label format for the nodes to show the label with 8pt Arial Bold font in semi-transparent
# black color (0x77000000). Put the text at the top left corner of the cell.
nodeConfig.setLabelFormat("{label}", "Times New Roman Bold Italic", 18, 0x77000000, TopLeft)

# Set the border color to white (ffffff)
nodeConfig.setColors(-1, 0xffffff)

# Get the prototype (template) for the second level nodes.
nodeConfig2 = c.getLevelPrototype(2)

# Set the label format for the nodes to show the label and value with 8pt Arial Bold font. Put the
# text at the center of the cell.
nodeConfig2.setLabelFormat("{label}<*br*>{value}", "Arial Bold", 8, TextColor, Center)

# Set the border color to white (ffffff)
nodeConfig2.setColors(-1, 0xffffff)

# Output the chart
c.makeChart("multileveltreemap.png")