X-Git-Url: http://git.tomasm.cz/imago.git/blobdiff_plain/3790db9ab7a4871faf0b445c8283c7f0b7b30a22..f662aac90eb7fb03ddca08366b83f0fd063b9e86:/imago_pack/linef.py?ds=sidebyside diff --git a/imago_pack/linef.py b/imago_pack/linef.py index eda82ce..06483a3 100644 --- a/imago_pack/linef.py +++ b/imago_pack/linef.py @@ -1,4 +1,4 @@ -"""Go image recognition lines-finding module""" +"""Lines finding module.""" from functools import partial import sys @@ -14,6 +14,8 @@ import filters from hough import Hough def find_lines(image, show_all, do_something, verbose): + """Find lines in the *image*.""" + # TODO refactor into smaller functions if verbose: print >> sys.stderr, "preprocessing" @@ -112,6 +114,7 @@ def find_lines(image, show_all, do_something, verbose): return lines, lines_m[0][0], lines_m[1][0], bounds, im_hough def combine(image1, image2): + """Return a list of points that are present in both images.""" im_l1 = image1.load() im_l2 = image2.load() @@ -124,6 +127,8 @@ def combine(image1, image2): return on_both def line_from_angl_dist((angle, distance), size): + """Take *angle* and *distance* (from the center of the image) of a line and + size of the image. Return the line represented by two points.""" if pi / 4 < angle < 3 * pi / 4: y1 = - size[1] / 2 x1 = int(round((y1 * cos(angle) + distance) / sin(angle))) + size[0] / 2