Pillow/Tests/bench_cffi_access.py

from tester import *

# not running this test by default. No DOS against travis.

from PIL import PyAccess
from PIL import Image

import time

def iterate_get(size, access):
    (w,h) = size
    for x in range(w):
        for y in range(h):
            access[(x,y)]

def iterate_set(size, access):
    (w,h) = size
    for x in range(w):
        for y in range(h):
            access[(x,y)] = (x %256,y%256,0)

def timer(func, label, *args):
    iterations = 5000
    starttime = time.time()
    for x in range(iterations):
        func(*args)
        if time.time()-starttime > 10:
            print ("%s: breaking at %s iterations, %.6f per iteration"%(label, x+1, (time.time()-starttime)/(x+1.0)))
            break
    if x == iterations-1:
        endtime = time.time()
        print ("%s: %.4f s  %.6f per iteration" %(label, endtime-starttime, (endtime-starttime)/(x+1.0)))

def test_direct():
    im = lena()
    im.load()
    #im = Image.new( "RGB", (2000,2000), (1,3,2))
    caccess = im.im.pixel_access(False)
    access = PyAccess.new(im, False)

    assert_equal(caccess[(0,0)], access[(0,0)])

    print ("Size: %sx%s" % im.size)
    timer(iterate_get, 'PyAccess - get', im.size, access)
    timer(iterate_set, 'PyAccess - set', im.size, access)
    timer(iterate_get, 'C-api - get', im.size, caccess)
    timer(iterate_set, 'C-api - set', im.size, caccess)