mirror of
https://github.com/undera/pylgbst.git
synced 2020-11-18 19:37:26 -08:00
115 lines
3.2 KiB
Python
115 lines
3.2 KiB
Python
import logging
|
|
import matplotlib.pyplot as plt
|
|
from threading import Thread
|
|
|
|
import numpy
|
|
from PIL import Image
|
|
|
|
|
|
class Tracer(object):
|
|
def __init__(self, fname):
|
|
super(Tracer, self).__init__()
|
|
self.orig = Image.open(fname)
|
|
self.conv1 = self.orig.convert("1")
|
|
self.src = numpy.asarray(self.conv1)
|
|
self.dst = numpy.copy(self.src)
|
|
self.dst.fill(False)
|
|
self.mark = numpy.copy(self.dst)
|
|
# start in center
|
|
self.height, self.width = self.dst.shape
|
|
self.posy = self.height / 2
|
|
self.posx = self.width / 2
|
|
|
|
def trace(self):
|
|
while self._has_unchecked_pixels():
|
|
# go circles to find a pixel in src
|
|
self._spiral_till_pixel()
|
|
|
|
# move until we find new pixels
|
|
self._move_while_you_can()
|
|
|
|
def _has_unchecked_pixels(self):
|
|
ix, iy = numpy.where(self.mark == False)
|
|
return len(ix) or len(iy)
|
|
|
|
def _spiral_till_pixel(self):
|
|
radius = 1
|
|
direction = 0
|
|
offset = 0
|
|
while self._has_unchecked_pixels():
|
|
in_lower = self.posy < self.height and self.posx < self.width
|
|
in_upper = self.posy >= 0 and self.posx >= 0
|
|
if in_lower and in_upper and not self.mark[self.posy][self.posx]:
|
|
if self.src[self.posy][self.posx]:
|
|
return
|
|
|
|
self.mark[self.posy][self.posx] = True
|
|
|
|
if direction == 0:
|
|
self.posx += 1
|
|
self.posy += 0
|
|
elif direction == 1:
|
|
self.posx += 0
|
|
self.posy += 1
|
|
elif direction == 2:
|
|
self.posx += -1
|
|
self.posy += 0
|
|
elif direction == 3:
|
|
self.posx += 0
|
|
self.posy += -1
|
|
else:
|
|
pass
|
|
|
|
offset += 1
|
|
if offset >= radius:
|
|
# time.sleep(0.01)
|
|
offset = 0
|
|
direction += 1
|
|
if direction > 3:
|
|
direction = 0
|
|
|
|
if direction in (0, 2):
|
|
radius += 1
|
|
|
|
raise KeyboardInterrupt("End of image") # end of image
|
|
|
|
def _move_while_you_can(self):
|
|
logging.debug("%s %s", self.posy, self.posx)
|
|
self.mark[self.posy][self.posx] = True
|
|
self.dst[self.posy][self.posx] = True
|
|
# self.posx += 1
|
|
# self.posy += 1
|
|
|
|
|
|
class TracerVisualizer(object):
|
|
def __init__(self, tracer):
|
|
"""
|
|
:type tracer: Tracer
|
|
"""
|
|
self.tracer = tracer
|
|
|
|
def run(self):
|
|
tracer = self.tracer
|
|
|
|
fig, ((ax1, ax2), (ax3, ax4)) = plt.subplots(nrows=2, ncols=2)
|
|
ax1.imshow(tracer.orig)
|
|
ax2.imshow(tracer.src)
|
|
plt.show(block=False)
|
|
|
|
thr = Thread(target=tracer.trace)
|
|
thr.setDaemon(True)
|
|
thr.start()
|
|
|
|
while plt.get_fignums(): # weird trick to react on close
|
|
ax3.set_title("%s:%s" % (tracer.posx, tracer.posy))
|
|
ax3.imshow(tracer.mark, cmap='gray')
|
|
ax4.imshow(tracer.dst, cmap='gray')
|
|
plt.pause(0.5)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
trc = Tracer("/tmp/truck.png")
|
|
|
|
TracerVisualizer(trc).run()
|