mieki256's diary

お絵かきアプリでよくある、閉じた範囲の中でクリックすると指定色で塗り潰す処理を、Python + PySide (+ PIL(Pillow)) を使って試したい。どうやら、その手の処理を、 _Flood fill と呼ぶらしいけど。

Ruby + DXRuby を使った版なら、昔書いたことがあったようで。 _2015/12/24の日記 にメモしてあった。Flood fill の一つ、Scanline Seed Fill を試していた模様。

コレを Python + PySide に移植してもいいのだけど、もしかしてこの手の処理は既に用意されているんじゃないか、と思ってググっていたら、どうやら PIL (Pillow) に、そのものズバリの floodfill というメソッドがあるようで。ただ、文書化はされていない模様。

_python - How to flood-fill part of a bitmap enclosed by a black border with my chosen color? - Stack Overflow

>>> from PIL import Image, ImageDraw
>>> help(ImageDraw.floodfill)
Help on function floodfill in module PIL.ImageDraw:

floodfill(image, xy, value, border=None)
    (experimental) Fills a bounded region with a given color.
    
    :param image: Target image.
    :param xy: Seed position (a 2-item coordinate tuple).
    :param value: Fill color.
    :param border: Optional border value.  If given, the region consists of
        pixels with a color different from the border color.  If not given,
        the region consists of pixels having the same color as the seed
        pixel.

たしかに入っているっぽい。だったら、コレを使わせてもらおうか…。

_floodfill_test.py

u"""
PIL(Pillow)のfloodfillを試す.

PIL(Pillow)には、文書化されていないが floodfill という関数があって、
塗りつぶし処理ができるらしい。

動作確認環境 : Windows10 x64 + Python 2.7.12 + Pillow 3.4.2
"""

from PIL import Image
from PIL import ImageDraw
import time

im = Image.open("./tmp_image.png")

start = time.time()
ImageDraw.floodfill(im, (64, 64), (255, 0, 0, 255), border=None)
ImageDraw.floodfill(im, (200, 240), (0, 255, 0, 255), border=None)
ImageDraw.floodfill(im, (480, 400), (0, 0, 255, 255), border=None)
etime = time.time() - start
print("%f" % etime)

im.save("./tmp_image_result.png")

以下の画像を渡してみる。
以下の結果が得られた。
塗りつぶしができてる。

3回塗り潰してみたけど、手元の環境で時間を測ってみたら 0.433000秒だった。そんなに遅くはない、かな…。

ググってみたら、以下のページが。

_python-imaging 1.1.7 - namespace_p_i_l_1_1_image_draw.html

この floodfill() は Python で書かれてる、ということかな…。上下左右を1ドットずつ見ていく、という比較的シンプルなアルゴリズムっぽい。

◎ QImageからPIL Imageへの変換。 :