File: agg_memleak.py

package info (click to toggle)
matplotlib 2.0.0%2Bdfsg1-2
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 91,640 kB
  • ctags: 29,525
  • sloc: python: 122,697; cpp: 60,806; ansic: 30,799; objc: 2,830; makefile: 224; sh: 85
file content (62 lines) | stat: -rw-r--r-- 1,795 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
"""
And another broken test...
"""

from __future__ import print_function
import os
from matplotlib.ft2font import FT2Font
from numpy.random import rand
from matplotlib.backend_bases import GraphicsContextBase
from matplotlib.backends._backend_agg import RendererAgg


def report_memory(i):
    pid = os.getpid()
    a2 = os.popen('ps -p %d -o rss,sz' % pid).readlines()
    print(i, '  ', a2[1], end='')
    return int(a2[1].split()[0])

fname = '/usr/local/share/matplotlib/Vera.ttf'

N = 200
for i in range(N):
    gc = GraphicsContextBase()
    gc.set_clip_rectangle([20, 20, 20, 20])
    o = RendererAgg(400, 400, 72)

    for j in range(50):
        xs = [400 * int(rand()) for k in range(8)]
        ys = [400 * int(rand()) for k in range(8)]
        rgb = (1, 0, 0)
        pnts = zip(xs, ys)
        o.draw_polygon(gc, rgb, pnts)   # no such method??
        o.draw_polygon(gc, None, pnts)

    for j in range(50):
        x = [400 * int(rand()) for k in range(4)]
        y = [400 * int(rand()) for k in range(4)]
        o.draw_lines(gc, x, y)

    for j in range(50):
        args = [400 * int(rand()) for k in range(4)]
        rgb = (1, 0, 0)
        o.draw_rectangle(gc, rgb, *args)

    if 1:  # add text
        font = FT2Font(fname)
        font.clear()
        font.set_text('hi mom', 60)
        font.set_size(12, 72)
        o.draw_text_image(font.get_image(), 30, 40, gc)

    o.write_png('aggtest%d.png' % i)
    val = report_memory(i)
    if i == 1:
        start = val

end = val
print('Average memory consumed per loop: %1.4f\n' % ((end - start) / float(N)))

# w/o text and w/o write_png: Average memory consumed per loop: 0.02
# w/o text and w/ write_png : Average memory consumed per loop: 0.3400
# w/ text and w/ write_png  : Average memory consumed per loop: 0.32