/+junk/pygooglechart-py3k

#!/usr/bin/env python

"""

Copyright Gerald Kaszuba 2008

This program is free software: you can redistribute it and/or modify

it under the terms of the GNU General Public License as published by

the Free Software Foundation, either version 3 of the License, or

(at your option) any later version.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

GNU General Public License for more details.

You should have received a copy of the GNU General Public License

along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""

import os

import sys

import math

ROOT = os.path.dirname(os.path.abspath(__file__))

sys.path.insert(0, os.path.join(ROOT, '..'))

from pygooglechart import Chart

from pygooglechart import SimpleLineChart

from pygooglechart import XYLineChart

from pygooglechart import SparkLineChart

from pygooglechart import Axis

import settings

import helper

def simple_random():

    chart = SimpleLineChart(settings.width, settings.height, y_range=(0, 100))

    chart.add_data(helper.random_data())

    chart.download('line-simple-random.png')

def xy_random():

    chart = XYLineChart(settings.width, settings.height,

                        x_range=(0, 100), y_range=(0, 100))

    chart.add_data(helper.random_data())

    chart.add_data(helper.random_data())

    chart.download('line-xy-random.png')

def xy_rect():

    chart = XYLineChart(settings.width, settings.height,

                        x_range=(0, 100), y_range=(0, 100))

    chart.add_data([10, 90, 90, 10, 10])

    chart.add_data([10, 10, 90, 90, 10])

    chart.download('line-xy-rect.png')

def xy_circle():

    chart = XYLineChart(settings.width, settings.height,

                        x_range=(0, 100), y_range=(0, 100))

    steps = 40

    xradius = 25

    yradius = 45

    xmid = 50

    ymid = 50

    xlist = []

    ylist = []

    for angle in xrange(0, steps + 1):

        angle = float(angle) / steps * math.pi * 2

        xlist.append(math.cos(angle) * xradius + xmid)

        ylist.append(math.sin(angle) * yradius + ymid)

    chart.add_data(xlist)

    chart.add_data(ylist)

    chart.download('line-xy-circle.png')

def sparklines():

    chart = SparkLineChart(settings.width, settings.height)

    chart.add_data(helper.random_data())

    chart.download('line-sparkline.png')

def fill():

    # Set the vertical range from 0 to 50

    max_y = 50

    chart = SimpleLineChart(200, 125, y_range=[0, max_y])

    # First value is the highest Y value. Two of them are needed to be

    # plottable.

    chart.add_data([max_y] * 2)

    # 3 sets of real data

    chart.add_data([28, 30, 31, 33, 35, 36, 42, 48, 43, 37, 32, 24, 28])

    chart.add_data([16, 18, 18, 21, 23, 23, 29, 36, 31, 25, 20, 12, 17])

    chart.add_data([7, 9, 9, 12, 14, 14, 20, 27, 21, 15, 10, 3, 7])

    # Last value is the lowest in the Y axis.

    chart.add_data([0] * 2)

    # Black lines

    chart.set_colours(['000000'] * 5)

    # Filled colours

    # from the top to the first real data

    chart.add_fill_range('76A4FB', 0, 1)

    # Between the 3 data values

    chart.add_fill_range('224499', 1, 2)

    chart.add_fill_range('FF0000', 2, 3)

    # from the last real data to the

    chart.add_fill_range('80C65A', 3, 4)

    # Some axis data

    chart.set_axis_labels(Axis.LEFT, ['', max_y / 2, max_y])

    chart.set_axis_labels(Axis.BOTTOM, ['Sep', 'Oct', 'Nov', 'Dec'])

    chart.download('line-fill.png')

def stripes():

    # Set the vertical range from 0 to 100

    max_y = 100

    # Chart size of 200x125 pixels and specifying the range for the Y axis

    chart = SimpleLineChart(200, 125, y_range=[0, max_y])

    # Add the chart data

    data = [

        32, 34, 34, 32, 34, 34, 32, 32, 32, 34, 34, 32, 29, 29, 34, 34, 34, 37,

        37, 39, 42, 47, 50, 54, 57, 60, 60, 60, 60, 60, 60, 60, 62, 62, 60, 55,

        55, 52, 47, 44, 44, 40, 40, 37, 34, 34, 32, 32, 32, 31, 32

    ]

    chart.add_data(data)

    # Set the line colour to blue

    chart.set_colours(['0000FF'])

    # Set the vertical stripes

    chart.fill_linear_stripes(Chart.CHART, 0, 'CCCCCC', 0.2, 'FFFFFF', 0.2)

    # Set the horizontal dotted lines

    chart.set_grid(0, 25, 5, 5)

    # The Y axis labels contains 0 to 100 skipping every 25, but remove the

    # first number because it's obvious and gets in the way of the first X

    # label.

    left_axis = range(0, max_y + 1, 25)

    left_axis[0] = ''

    chart.set_axis_labels(Axis.LEFT, left_axis)

    # X axis labels

    chart.set_axis_labels(Axis.BOTTOM, \

        ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun'])

    chart.download('line-stripes.png')

def main():

    simple_random()

    xy_random()

    xy_rect()

    xy_circle()

    sparklines()

    fill()

    stripes()

if __name__ == '__main__':

    main()