pygal/tests/samples.py

"""
Samples of the various charts. Run this script to generate the reference
samples.
"""
import sys, os

from svg.charts.plot import Plot
from svg.charts import bar
from svg.charts import time_series
from svg.charts import pie
from svg.charts import schedule
from svg.charts import line

def sample_Plot():
	g = Plot({
		'min_x_value': 0,
		'min_y_value': 0,
		'area_fill': True,
		'stagger_x_labels': True,
		'stagger_y_labels': True,
		'show_x_guidelines': True
	   })
	g.add_data({'data': [1, 25, 2, 30, 3, 45], 'title': 'series 1'})
	g.add_data({'data': [1,30, 2, 31, 3, 40], 'title': 'series 2'})
	g.add_data({'data': [.5,35, 1, 20, 3, 10.5], 'title': 'series 3'})
	return g

def sample_TimeSeries():
	g = time_series.Plot({})

	g.timescale_divisions = '4 hours'
	g.stagger_x_labels = True
	g.x_label_format = '%d-%b %H:%M'
	#g.max_y_value = 200

	g.add_data({'data': ['2005-12-21T00:00:00', 20, '2005-12-22T00:00:00', 21], 'title': 'series 1'})

	return g

def generate_samples():
	yield 'Plot', sample_Plot()
	yield 'TimeSeries', sample_TimeSeries()
	yield 'VerticalBar', SampleBar.vertical()
	yield 'HorizontalBar', SampleBar.horizontal()
	yield 'VerticalBarLarge', SampleBar.vertical_large()
	yield 'Pie', sample_Pie()
	yield 'Schedule', sample_Schedule()
	yield 'Line', sample_Line()

class SampleBar:
	fields = ['Internet', 'TV', 'Newspaper', 'Magazine', 'Radio']

	@classmethod
	def vertical(cls):

		g = bar.VerticalBar(cls.fields)

		g.stack = 'side'
		g.scale_integers = True
		g.width, g.height = 640,480
		g.graph_title = 'Question 7'
		g.show_graph_title = True

		g.add_data({'data': [-2, 3, 1, 3, 1], 'title': 'Female'})
		g.add_data({'data': [0, 2, 1, 5, 4], 'title': 'Male'})

		return g

	@classmethod
	def horizontal(cls):
		g = bar.HorizontalBar(cls.fields)

		g.stack = 'side'
		g.scale_integers = True
		g.width, g.height = 640,480
		g.graph_title = 'Question 7'
		g.show_graph_title = True

		g.add_data({'data': [-2, 3, 1, 3, 1], 'title': 'Female'})
		g.add_data({'data': [0, 2, 1, 5, 4], 'title': 'Male'})

		return g

	@classmethod
	def vertical_large(cls):
		g = bar.VerticalBar(cls.fields)
		options = dict(
			scale_integers=True,
			stack='side',
			width=640,
			height=480,
			graph_title='Question 8',
			show_graph_title=True,
			no_css=False,)
		g.__dict__.update(options)

		g.add_data(dict(data=[2,22,98,143,82], title='intermediate'))
		g.add_data(dict(data=[2,26,106,193,105], title='old'))
		return g

def sample_Line():
	g = line.Line()
	options = dict(
		scale_integers = True,
		area_fill = True,
		width = 640,
		height = 480,
		fields = SampleBar.fields,
		graph_title = 'Question 7',
		show_graph_title = True,
		no_css = False,
	)
	g.__dict__.update(options)
	g.add_data({'data': [-2, 3, 1, 3, 1], 'title': 'Female'})
	g.add_data({'data': [0, 2, 1, 5, 4], 'title': 'Male'})
	return g

def sample_Pie():
	g = pie.Pie({})
	options = dict(
		width=640,
		height=480,
		fields=SampleBar.fields,
		graph_title='Question 7',
		expand_greatest = True,
		show_data_labels = True,
	)
	g.__dict__.update(options)
	g.add_data({'data': [-2, 3, 1, 3, 1], 'title': 'Female'})
	g.add_data({'data': [0, 2, 1, 5, 4], 'title': 'Male'})
	return g

def sample_Schedule():
	title = "Billy's Schedule"
	data1 = [
	  "History 107", "5/19/04", "6/30/04",
	  "Algebra 011", "6/2/04", "8/11/04",
	  "Psychology 101", "6/28/04", "8/9/04",
	  "Acting 105", "7/7/04", "8/16/04"
	  ]

	g = schedule.Schedule(dict(
		width = 640,
		height = 480,
		graph_title = title,
		show_graph_title = True,
		key = False,
		scale_x_integers = True,
		scale_y_integers = True,
		show_data_labels = True,
		show_y_guidelines = False,
		show_x_guidelines = True,
		# show_x_title = True, # not yet implemented
		x_title = "Time",
		show_y_title = False,
		rotate_x_labels = True,
		rotate_y_labels = False,
		x_label_format = "%m/%d",
		timescale_divisions = "1 week",
		add_popups = True,
		popup_format = "%m/%d/%y",
		area_fill = True,
		min_y_value = 0,
		))

	g.add_data(dict(data=data1, title="Data"))

	return g

def save_samples():
	root = os.path.dirname(__file__)
	for sample_name, sample in generate_samples():
		res = sample.burn()
		with open(os.path.join(root, sample_name+'.py.svg'), 'w') as f:
			f.write(res)

if __name__ == '__main__':
	save_samples()