def test_hide_axis():
"""Test with hidden axis"""
cs_1 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_1.add(elements.Axis(horizontal=True, hide=True))
cs_1.add(elements.Axis(vertical=True))
cs_1.add(
charts.Function(lambda x: x * math.sin(x * x),
start=-4,
end=5,
step=0.02,
color=(0, 0, 255)))
cs_2 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_2.add(elements.Axis(horizontal=True))
cs_2.add(elements.Axis(vertical=True, hide=True))
cs_2.add(
charts.Function(lambda x: x * math.sin(x * x),
start=-4,
end=5,
step=0.02,
color=(0, 0, 255)))
return cs_1.draw(150, 150), cs_2.draw(150, 150), cs_1.draw(
150, 150, antialiasing=True), cs_2.draw(150, 150, antialiasing=True)
def test_barchart_with_generator():
""" Bar charts with data given as generator function """
import cartesius.colors as colors
coordinate_system = cartesius.CoordinateSystem()
bottom_collor = (0, 0, 255)
top_color = (255, 255, 255)
# In case you don't want to keep all your data in memory, you can give the data as a
# generator function that will lazy-load your data. This works for all charts:
def data_generator():
for x in range(25):
key = x / 2.
value = 2 * math.sin(x / 4.)
color = colors.get_color_between(bottom_collor, top_color,
(value + 2) / 4.)
yield charts.data(key, value, fill_color=color)
barchart_data_generator = data_generator
barchart = charts.BarChart(data=barchart_data_generator,
vertical=True,
width=0.5)
coordinate_system.add(barchart)
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
def test_barchart_1():
""" Bar charts with same column width. One with default and the other with custom colors """
coordinate_system = cartesius.CoordinateSystem()
barchart_data_1 = (charts.data(-1, -.5), charts.data(0, .7),
charts.data(1, 2), charts.data(2,
2.7), charts.data(3, 4),
charts.data(4, 3.1), charts.data(5, 2.1),
charts.data(6, 1), charts.data(7, -.3))
barchart_1 = charts.BarChart(vertical=True,
data=barchart_data_1,
width=0.95)
coordinate_system.add(barchart_1)
custom_color = (100, 100, 200)
barchart_data_2 = (charts.data(-1, -.25, fill_color=custom_color),
charts.data(0, .35, fill_color=custom_color),
charts.data(1, 1, fill_color=custom_color),
charts.data(2, 1.35, fill_color=custom_color),
charts.data(3, 2, fill_color=custom_color),
charts.data(4, 1.65, fill_color=custom_color),
charts.data(5, 1, fill_color=custom_color),
charts.data(6, .5, fill_color=custom_color),
charts.data(7, -.6, fill_color=custom_color))
barchart_2 = charts.BarChart(vertical=True,
data=barchart_data_2,
width=0.75,
color=(0, 0, 0))
coordinate_system.add(barchart_2)
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
def test_circles_2():
""" Another example with circles """
# The colors package contain e few utility functions for colors:
import cartesius.colors as colors
coordinate_system = cartesius.CoordinateSystem()
from_color = (255, 0, 255)
to_color = (0, 255, 0)
iterations = 20
for i in range(1, iterations):
x = i / 2.
y = math.sin(x)
center = (i / 1, math.sin(x))
color = colors.get_color_between(from_color, to_color,
i / float(iterations))
coordinate_system.add(
elements.Circle(center,
radius=math.sqrt(x),
transparency_mask=50,
fill_color=color,
color=(0, 0, 0)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_function_with_custom_bounds():
"""Same function, but with custom coordinate system bounds"""
coordinate_system = cartesius.CoordinateSystem(bounds=(-32, 20, -3, 3))
f = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=(0, 0, 255)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_function():
""" Function math.sin from -4 to 5"""
coordinate_system = cartesius.CoordinateSystem()
f = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=0x0000ff))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_detached_axes():
""" Detached axes"""
coordinate_system = cartesius.CoordinateSystem(bounds=(-10, 10, -10, 10))
coordinate_system.add(
charts.Function(lambda x: x * math.sin(x * x),
start=-4,
end=5,
step=0.02,
color=(0, 0, 255)))
# Standard axes:
coordinate_system.add(elements.Axis(horizontal=True, points=2))
coordinate_system.add(elements.Axis(vertical=True, labels=2))
# You can have only one horizontal and one vertical standard axis and, if you add
# more -- the newer will overwrite the older.
# But, you can make as many as you want *detached axes*. These are just like normal
# ones, but their center is not (0,0).
# Detached:
detached_axes_center = (-5, 4)
coordinate_system.add(
elements.Axis(horizontal=True,
points=2,
labels=2,
detached_center=detached_axes_center,
color=(255, 0, 0)))
coordinate_system.add(
elements.Axis(vertical=True,
points=2,
labels=2,
detached_center=detached_axes_center,
color=(0, 0, 255)))
# Another pair of detached axes with hidden negative/positive halfs:
detached_axes_center = (4, -5)
coordinate_system.add(
elements.Axis(horizontal=True,
points=2,
labels=2,
detached_center=detached_axes_center,
hide_negative=True))
coordinate_system.add(
elements.Axis(vertical=True,
points=2,
labels=2,
detached_center=detached_axes_center,
hide_positive=True))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_lines():
"""Lines different colors"""
coordinate_system = cartesius.CoordinateSystem()
coordinate_system.add(elements.Line((0, 0), (-.7, -.7)))
coordinate_system.add(
elements.Line((.5, -.5), (-.5, .5), color=(0, 255, 0)))
coordinate_system.add(elements.Line((0, 0), (7, 3), color=(0, 0, 255)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_with_two_horizontal_grids():
"""Two horizontal grids"""
coordinate_system = cartesius.CoordinateSystem()
coordinate_system.add(elements.Grid(0.25, None, color=(200, 200, 200)))
coordinate_system.add(elements.Grid(1, None, color=(250, 50, 50)))
f = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=(0, 0, 255)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_points():
"""Test points of different styles with/without label and different label positions"""
coordinate_system = cartesius.CoordinateSystem(bounds=(-.5, 5, -.5, 5))
# Without labels:
coordinate_system.add(elements.Point((1, 4), style='.'))
coordinate_system.add(elements.Point((2, 4), style='+'))
coordinate_system.add(elements.Point((3, 4), style='x'))
coordinate_system.add(elements.Point((4, 4), style='o'))
# With labels:
coordinate_system.add(elements.Point((1, 3), style='.', label='A'))
coordinate_system.add(elements.Point((2, 3), style='+', label='B'))
coordinate_system.add(elements.Point((3, 3), style='x', label='C'))
coordinate_system.add(elements.Point((4, 3), style='o', label='D'))
# With labels and custom colors:
coordinate_system.add(
elements.Point((1, 2), style='.', label='A', color=(255, 0, 0)))
coordinate_system.add(
elements.Point((2, 2), style='+', label='B', color=(0, 255, 0)))
coordinate_system.add(
elements.Point((3, 2), style='x', label='C', color=(0, 0, 255)))
coordinate_system.add(
elements.Point((4, 2), style='o', label='D', color=(150, 150, 150)))
# With labels on custom positions:
coordinate_system.add(
elements.Point((1, 1),
style='.',
label='A',
label_position=cartesius.RIGHT_CENTER))
coordinate_system.add(
elements.Point((2, 1),
style='+',
label='B',
label_position=cartesius.LEFT_CENTER))
coordinate_system.add(
elements.Point((3, 1),
style='x',
label='C',
label_position=cartesius.CENTER_UP))
coordinate_system.add(
elements.Point((4, 1),
style='o',
label='D',
label_position=cartesius.CENTER_DOWN))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_hide_axis_positive_or_negative_parts():
"""Hide positive and/or negative parts of axes"""
cs_1 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_1.add(
charts.Function(lambda x: x * math.sin(x * x),
start=-4,
end=5,
step=0.02,
color=(0, 0, 255)))
cs_1.add(elements.Axis(horizontal=True, hide_positive=True))
cs_1.add(elements.Axis(vertical=True, hide_positive=True))
cs_2 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_2.add(
charts.Function(lambda x: x * math.sin(x * x),
start=-4,
end=5,
step=0.02,
color=(0, 0, 255)))
cs_2.add(elements.Axis(horizontal=True, hide_negative=True))
cs_2.add(elements.Axis(vertical=True, hide_negative=True))
return cs_1.draw(150, 150), cs_2.draw(150, 150), cs_1.draw(
150, 150, antialiasing=True), cs_2.draw(150, 150, antialiasing=True)
def test_line_charts():
""" Line charts"""
import cartesius.colors as colors
coordinate_system = cartesius.CoordinateSystem()
# linechart (that mimic function chart) with generator and colors computed on the fly:
def f():
from_color = (255, 255, 0)
to_color = (100, 100, 255)
i = -2
while i < 6:
value = 2 * math.sin(i)
color = colors.get_color_between(from_color, to_color,
(value + 2) / 4.)
yield charts.data(i, value, fill_color=color, color=(0, 0, 0))
i += .05
coordinate_system.add(charts.LineChart(data=f, transparency_mask=0))
# filled line chart with labels
coordinate_system.add(
charts.LineChart(data=(charts.data(-2, 1, label='aaa'),
charts.data(0, -1, label='bbb'),
charts.data(3, 1.2), charts.data(7, 1.2)),
fill_color=(50, 50, 50),
transparency_mask=50))
# normal line chart with labels:
coordinate_system.add(
charts.LineChart(
data=(charts.data(0, 0), charts.data(1, -3),
charts.data(4,
3,
label='aaa',
label_position=cartesius.CENTER_DOWN),
charts.data(5,
-2,
label='custom label',
label_position=cartesius.RIGHT_CENTER),
charts.data(7, 0)),
color=(255, 0, 0),
transparency_mask=150))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_filled_function():
""" Line function and normal function but with filled graph"""
coordinate_system = cartesius.CoordinateSystem()
f = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=0x0000ff))
g = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(g,
start=1,
end=4,
step=0.02,
fill_color=(200, 255, 200)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_axis_with_custom_labels():
"""Axis with custom label positions and decorations:"""
coordinate_system = cartesius.CoordinateSystem()
coordinate_system.add(elements.Axis(horizontal=True, labels=1,
points=0.25))
coordinate_system.add(
elements.Axis(vertical=True,
labels=2,
labels_decorator=lambda x: 'speed=%sm/s' % x,
points=1))
f = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=(0, 0, 255)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_axis_custom_colors():
"""Axis with custom colors"""
coordinate_system = cartesius.CoordinateSystem()
coordinate_system.add(
elements.Axis(horizontal=True,
color=(255, 0, 0),
labels=1,
points=0.25))
coordinate_system.add(
elements.Axis(vertical=True, color=(0, 255, 0), labels=2, points=1))
f = lambda x: x * math.sin(x * x)
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=(0, 0, 255)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_circles_5():
""" Circles with vertical grid every 0.5 """
coordinate_system = cartesius.CoordinateSystem()
coordinate_system.add(elements.Grid(None, 0.5, transparency_mask=200))
for i in range(1, 20):
x = i / 2.
y = math.sin(x)
center = (i / 1, math.sin(x))
coordinate_system.add(
elements.Circle(center,
radius=math.sqrt(x),
transparency_mask=50,
fill_color=(i * 10, 2 * 10, i * 10),
color=(0, 0, 0)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_circles():
"""50 random circles"""
coordinate_system = cartesius.CoordinateSystem()
for i in range(50):
center = (random.randint(-20, 20), random.randint(0, 20))
coordinate_system.add(
elements.Circle(center,
radius=random.randint(1, 5),
transparency_mask=random.randint(0, 255),
fill_color=(random.randint(0, 255),
random.randint(0, 255),
random.randint(0, 255)),
color=(random.randint(0,
255), random.randint(0, 255),
random.randint(0, 255))))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
def test_axis_with_custom_labels_2():
"""Axis with custom labels II"""
coordinate_system = cartesius.CoordinateSystem(bounds=(-1500, 1500, -1500,
1500))
# Labels with suffixes 'm':
coordinate_system.add(
elements.Axis(horizontal=True, labels='500m', points=100))
# Custom labels on custom positions:
coordinate_system.add(
elements.Axis(vertical=True,
labels={
1000: 'one km',
500: 'half km'
},
points=100))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_piechart_2():
""" PieChart with custom colors """
coordinate_system = cartesius.CoordinateSystem()
piechart_data = (
charts.data('abc', 1, fill_color=(255, 200, 200)),
charts.data('cde', 2, fill_color=(200, 255, 200)),
charts.data('efg', 5, fill_color=(200, 200, 255)),
charts.data('ijk', 3, fill_color=(255, 255, 255)),
)
piechart = charts.PieChart(data=piechart_data, color=(0, 0, 0))
coordinate_system.add(piechart)
# No need for axes:
coordinate_system.add(elements.Axis(horizontal=True, hide=True))
coordinate_system.add(elements.Axis(vertical=True, hide=True))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
def test_barchart_2():
""" BarChart with different column widths """
coordinate_system = cartesius.CoordinateSystem()
barchart_data = (
charts.data(-5, -0, -.5, label='Negative value'),
charts.data(0, 1, size=.7),
charts.data(1, 3, size=2),
charts.data(3, 4, size=4, label='#1'),
charts.data(4, 5.5, size=3.1),
charts.data(6, 7, size=2.1),
charts.data(7, 9, size=1),
)
barchart = charts.BarChart(vertical=True,
data=barchart_data,
color=(0, 0, 0))
coordinate_system.add(barchart)
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
def test_filled_function():
""" Previous example with grid behind graphs """
coordinate_system = cartesius.CoordinateSystem()
# Grid:
coordinate_system.add(elements.Grid(1, 1))
f = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(f, start=-4, end=5, step=0.02, color=(0, 0, 255)))
g = lambda x: math.sin(x) * 2
coordinate_system.add(
charts.Function(g,
start=1,
end=4,
step=0.02,
fill_color=(200, 255, 200)))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_filled_transparent_graphs():
""" Two functions """
coordinate_system = cartesius.CoordinateSystem()
coordinate_system.add(
charts.Function(math.sin,
start=-4,
end=5,
step=0.02,
fill_color=(0, 0, 255),
transparency_mask=100))
coordinate_system.add(
charts.Function(math.cos,
start=-4,
end=5,
step=0.02,
fill_color=(200, 255, 200),
transparency_mask=100))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True)
def test_piechart_1():
""" PieChart with default colors """
coordinate_system = cartesius.CoordinateSystem()
# list or tuple of two-element tuples (value, label):
piechart_data = (
charts.data('abc', 1),
charts.data('cde', 2),
charts.data('efg', 4),
charts.data('ijk', 1),
charts.data('lmn', 5),
charts.data('opq', 5),
charts.data('xyz', 3),
)
piechart = charts.PieChart(data=piechart_data, color=(0, 0, 0))
coordinate_system.add(piechart)
# No need for axes:
coordinate_system.add(elements.Axis(horizontal=True, hide=True))
coordinate_system.add(elements.Axis(vertical=True, hide=True))
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
def test_barchart_horizontal():
""" Horizontal bar charts """
coordinate_system = cartesius.CoordinateSystem()
barchart_data_1 = (charts.data(-1, -.5,
label='a'), charts.data(0, .7, label='b'),
charts.data(1, 2,
label='c'), charts.data(2, 2.7, label='d'),
charts.data(3, 4, label='Maximum value'),
charts.data(4, 3.1,
label='f'), charts.data(5, 2.1, label='g'),
charts.data(6, 1, label='h'),
charts.data(7, -.3, label='The end!'))
barchart_1 = charts.BarChart(horizontal=True,
data=barchart_data_1,
width=0.95)
coordinate_system.add(barchart_1)
custom_color = (100, 100, 200)
barchart_data_2 = (charts.data(-1, -.25, fill_color=custom_color),
charts.data(0, .35, fill_color=custom_color),
charts.data(1, 1, fill_color=custom_color),
charts.data(2, 1.35, fill_color=custom_color),
charts.data(3, 2, fill_color=custom_color),
charts.data(4, 1.65, fill_color=custom_color),
charts.data(5, 1, fill_color=custom_color),
charts.data(6, .5, fill_color=custom_color),
charts.data(7, -.6, fill_color=custom_color))
barchart_2 = charts.BarChart(horizontal=True,
data=barchart_data_2,
width=0.75,
color=(0, 0, 0))
coordinate_system.add(barchart_2)
return coordinate_system.draw(300, 200), coordinate_system.draw(
300, 200, antialiasing=True),
for point in gpx.walk(only_points=True):
if previous_point:
length += previous_point.distance_2d(point)
previous_point = point
yield mod_charts.data(length, point.elevation)
return mod_charts.LineChart(data=f,
color=color,
transparency_mask=transparency_mask)
def sample_gpx():
return mod_gpxpy.parse(open('sample_files/setnjica-kod-karojbe.gpx'))
coordinate_system = mod_cartesius.CoordinateSystem(bounds=(-300, 6800, -40,
480))
coordinate_system.add(mod_elements.Grid(20, 100))
gpx = sample_gpx()
coordinate_system.add(get_line(gpx, color=(0, 0, 0)))
data = mod_srtm.get_data()
gpx = sample_gpx()
data.add_elevations(gpx)
coordinate_system.add(get_line(gpx, color=(0, 0, 255), transparency_mask=150))
gpx = sample_gpx()
data.add_elevations(gpx, smooth=True)
coordinate_system.add(get_line(gpx, color=(255, 0, 0)))
y = [(math.sin(x * 0.02) + 1) * 0.5 * 5 for x in range(int(5 / 0.02))]
t = time()
f = lambda x: y[int(x)]
def data_generator():
for x in range(int(5 / 0.02)):
key = x
value = f(x)
color = (0, 0, 255)
yield charts.data(key, value, color=color)
coordinate_system = cartesius.CoordinateSystem(bounds=(
-5, len(y), 0,
5)) #left=bounds[0], right=bounds[1], bottom=bounds[2], top=bounds[3]
coordinate_system.add(elements.Axis(vertical=True, labels=1, points=1))
#coordinate_system.add(elements.Grid(0.5, None, color=(100, 100, 100)))
coordinate_system.add(elements.Grid(1, None, color=(200, 200, 200)))
#coordinate_system.add(charts.Function(f, start=0, end=len(y), step=1, color=(0, 0, 255)))
coordinate_system.add(charts.LineChart(data=data_generator))
print('crt', time() - t)
img1 = coordinate_system.draw(w, h // 2, antialiasing=True)
#img2 = coordinate_system.draw(w, h//2, antialiasing=True)
img2 = img1
print('drw', time() - t)
new_im = Image.new('RGBA', (w, h))
new_im.paste(img1, (0, 0))
def test_labels_positions():
"""Labels on different positions"""
cs_1 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_1.add(
elements.Axis(
horizontal=True,
points=1,
labels=1,
label_position=cartesius.LEFT_UP,
))
cs_1.add(
elements.Axis(
vertical=True,
points=1,
labels=1,
label_position=cartesius.LEFT_CENTER,
))
cs_2 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_2.add(
elements.Axis(
horizontal=True,
points=1,
labels=1,
label_position=cartesius.LEFT_DOWN,
))
cs_2.add(
elements.Axis(
vertical=True,
points=1,
labels=1,
label_position=cartesius.CENTER_UP,
))
cs_3 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_3.add(
elements.Axis(
horizontal=True,
points=1,
labels=1,
label_position=cartesius.CENTER,
))
cs_3.add(
elements.Axis(
vertical=True,
points=1,
labels=1,
label_position=cartesius.CENTER_DOWN,
))
cs_4 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_4.add(
elements.Axis(
horizontal=True,
points=1,
labels=1,
label_position=cartesius.RIGHT_UP,
))
cs_4.add(
elements.Axis(
vertical=True,
points=1,
labels=1,
label_position=cartesius.RIGHT_CENTER,
))
cs_5 = cartesius.CoordinateSystem(bounds=(-2.5, 2.5, -2.5, 2.5))
cs_5.add(
elements.Axis(
horizontal=True,
points=1,
labels=1,
label_position=cartesius.RIGHT_DOWN,
))
cs_5.add(
elements.Axis(
vertical=True,
points=1,
labels=1,
label_position=cartesius.RIGHT_DOWN,
))
return cs_1.draw(150, 150), cs_2.draw(150,
150), cs_3.draw(150, 150), cs_4.draw(
150, 150), cs_5.draw(150, 150)
