def _draw_finder_pattern(image, transform, fp):
center = Point(transform.x, transform.y)
angle = transform.rot
rotated = image.rotate(transform.rot, center)
c1 = _rotate_around_point(fp.c1, -angle, center)
c2 = _rotate_around_point(fp.c2, -angle, center)
c3 = _rotate_around_point(fp.c3, -angle, center)
img = rotated.to_alpha()
img.draw_line(c1, c2, Color.Green(), 1)
img.draw_line(c1, c3, Color.Green(), 1)
return img
def _DEBUG_SQUARE_LOCATOR(self, slot_img, fp, slot_num):
if not self.DEBUG:
return
color = slot_img.to_alpha()
fp.draw_to_image(color, Color.Green())
self._DEBUG_SAVE_IMAGE(color, "SQUARE_FP", slot_num)
def __init__(self, file):
Config.__init__(self, file)
add = self.add
if IS_BUNDLED:
default_store = "./store/"
else:
default_store = "../store/"
self.color_ok = add(ColorConfigItem, "Read Color", Color.Green())
self.color_unreadable = add(ColorConfigItem, "Not Read Color",
Color.Red())
self.color_empty = add(ColorConfigItem, "Empty Color", Color.Grey())
self.camera_number = add(IntConfigItem, "Camera Number", default=0)
self.camera_width = add(IntConfigItem, "Camera Width", default=1920)
self.camera_height = add(IntConfigItem, "Camera Height", default=1080)
self.plate_type = add(EnumConfigItem,
"Sample Plate Type",
default=Geometry.UNIPUCK,
extra_arg=Geometry.TYPES)
self.barcode_size = add(EnumConfigItem,
"Datamatrix Size",
default=DataMatrix.DEFAULT_SIZE,
extra_arg=DatamatrixSizeTable.valid_sizes())
self.scan_beep = add(BoolConfigItem,
"Beep While Scanning",
default=True)
self.scan_clipboard = add(BoolConfigItem,
"Results to Clipboard",
default=True)
self.image_puck = add(BoolConfigItem, "Draw Puck", default=True)
self.image_pins = add(BoolConfigItem,
"Draw Slot Highlights",
default=True)
self.image_crop = add(BoolConfigItem, "Crop to Puck", default=True)
self.store_directory = add(DirectoryConfigItem,
"Store Directory",
default=default_store)
self.store_capacity = add(IntConfigItem,
"Results History Size",
default=50)
self.console_frame = add(BoolConfigItem,
"Print Frame Summary",
default=False)
self.slot_images = add(BoolConfigItem,
"Save Debug Images",
default=False)
self.slot_image_directory = add(DirectoryConfigItem,
"Debug Directory",
default="../debug-output/")
self.initialize_from_file()
def edges_image(edge_sets):
blank = Image.blank(image_original.width, image_original.height, 3, 255)
for shape in edge_sets:
for edge in shape:
print(edge[1])
blank.draw_line(Point.from_array(edge[0]), Point.from_array(edge[1]), Color.Green(), 1)
return blank
def _draw_square_and_writing(image, transform, size):
marked_img = _draw_square(image, transform, size)
txt_height = SquareLocator.TXT_HEIGHT * size
txt_width = SquareLocator.TXT_WIDTH * size
center = transform.trans
x1 = center.x - txt_width/2
x2 = x1 + txt_width
y_offset = SquareLocator.TXT_OFFSET*size
y1 = center.y + y_offset - txt_height/2
y2 = y1 + txt_height
marked_img.draw_rectangle((x1, y1, x2, y2), Color.Green(), 1)
y1 = center.y - y_offset - txt_height/2
y2 = y1 + txt_height
marked_img.draw_rectangle((x1, y1, x2, y2), Color.Green(), 1)
return marked_img
def _draw_square(image, transform, size):
radius = size/2
center = Point(transform.x, transform.y)
rotated = image.rotate(transform.rot, center)
x1, y1 = center.x-radius, center.y-radius
x2, y2 = x1 + size, y1 + size
roi = (x1, y1, x2, y2)
marked_img = rotated.to_alpha()
marked_img.draw_rectangle(roi, Color.Green(), 1)
return marked_img
def TRIANGLE_DEMO():
""" Draw a set of axes and a triangle. Perform a series of random transformations
on the triangle and display the results.
"""
A = Point(143, 52)
B = Point(17, 96.5)
C = Point(0, 0)
for i in range(10):
# Create random transformation
angle = random.random() * 2 * math.pi
scale = random.random() * 3
delX = (random.random() - 0.5) * 200
delY = (random.random() - 0.5) * 200
translate = Point(delX, delY)
transform = Transform(translate, angle, scale)
# Transform the triangle
A_ = transform.transform(A)
B_ = transform.transform(B)
C_ = transform.transform(C)
# From the line A-B and the transformed line A'-B', determine what the transformation was
# This should be the same as the original transformation
trans_calc = Transform.line_mapping(A, B, A_, B_)
print("Angle: {:.2f}; {:.2f}".format(rad_to_deg(angle), rad_to_deg(trans_calc.rot)))
print("Trans: ({}); ({})".format(translate, trans_calc.trans))
print("Zoom: {:.2f}; {:.2f}".format(scale, trans_calc.zoom))
# Display on image
image = Image.blank(1000, 800)
image.draw_offset = IMG_CENTER
draw_axes(image, 300)
# Draw original triangle
image.draw_line(A, B, Color.Red(), 5)
image.draw_line(C, B, Color.Red(), 5)
image.draw_line(A, C, Color.Red(), 5)
#Draw transformed triangle
image.draw_line(A_, B_, Color.Green())
image.draw_line(A_, C_, Color.Green())
image.draw_line(C_, B_, Color.Green())
# Check that the reverse transformation works properly
A__ = transform.reverse(A_)
B__ = transform.reverse(B_)
C__ = transform.reverse(C_)
# Draw the reverse transformation - this should overlap the origianl triangle
image.draw_line(A__, B__, Color.Green(), 1)
image.draw_line(A__, C__, Color.Green(), 1)
image.draw_line(C__, B__, Color.Green(), 1)
# Write the transformation on the image
image.draw_text(transform.__str__(), Point(-450, 350), Color.White(), centered=False, scale=0.5, thickness=1)
# Show the image
image.popup()
def _DEBUG_WIGGLES_READ(self, barcode, locate_type, side_length):
if not self.DEBUG:
return
if barcode.is_valid():
print("DEBUG - WIGGLES SUCCESSFUL - " + locate_type)
result = "_SUCCESS"
else:
result = "_FAIL"
slot_img = Image(barcode._image).to_alpha()
self._DEBUG_SAVE_IMAGE(slot_img, locate_type + result, side_length - 1)
fp = barcode._finder_pattern
fp.draw_to_image(slot_img, Color.Green())
self._DEBUG_SAVE_IMAGE(slot_img, locate_type + result, side_length - 1)
def __init__(self, file, file_manager):
Config.__init__(self, file, file_manager)
add = self.add
if IS_BUNDLED:
default_store = "./store/"
else:
default_store = "../store/"
self.color_ok = add(ColorConfigItem, "Pin/Puck Read", Color.Green())
self.color_accept = add(ColorConfigItem, "Puck Partially Read",
Color.Yellow())
self.color_unreadable = add(ColorConfigItem, "Pin/Puck Not Read",
Color.Red())
self.color_empty = add(ColorConfigItem, "Pin Empty", Color.Grey())
self.plate_type = add(EnumConfigItem,
"Sample Plate Type",
default=Geometry.UNIPUCK,
extra_arg=Geometry.TYPES)
self.top_barcode_size = add(
EnumConfigItem,
"Datamatrix Size",
default=DataMatrix.DEFAULT_SIZE,
extra_arg=DatamatrixSizeTable.valid_sizes())
self.top_camera_timeout = add(IntConfigItem,
"Scan Timeout",
default=15,
extra_arg="s")
self.scan_beep = add(BoolConfigItem,
"Beep While Scanning",
default=True)
self.scan_clipboard = add(BoolConfigItem,
"Results to Clipboard",
default=True)
self.image_puck = add(BoolConfigItem, "Puck Highlight", default=True)
self.image_pins = add(BoolConfigItem, "Slots Highlight", default=True)
self.image_crop = add(BoolConfigItem, "Crop to Puck", default=True)
self.store_directory = add(DirectoryConfigItem,
"Store Directory",
default=default_store)
self.store_capacity = add(IntConfigItem,
"Results History Size",
default=50)
self.console_frame = add(BoolConfigItem,
"Print Frame Summary",
default=False)
self.slot_images = add(BoolConfigItem,
"Save Debug Images",
default=False)
self.slot_image_directory = add(DirectoryConfigItem,
"Debug Directory",
default="../debug-output/")
self.top_camera_number = add(IntConfigItem,
"Top Camera Number",
default=1)
self.top_camera_width = add(IntConfigItem,
"Top Camera Width",
default=1600)
self.top_camera_height = add(IntConfigItem,
"Top Camera Height",
default=1200)
self.side_camera_number = add(IntConfigItem,
"Side Camera Number",
default=2)
self.side_camera_width = add(IntConfigItem,
"Side Camera Width",
default=1600)
self.side_camera_height = add(IntConfigItem,
"Side Camera Height",
default=1200)
self.initialize_from_file()
def _scanner_worker(task_queue, overlay_queue, result_queue, options):
""" Function used as the main loop of a worker process. Scan images for barcodes,
combining partial scans until a full puck is reached.
Keep the record of the last scan which was at least partially successful (aligned geometry
and some barcodes scanned). For each new frame, we can attempt to merge the results with
this previous plates so that we don't have to re-read any of the previously captured barcodes
(because this is a relatively expensive operation).
"""
last_plate_time = time.time()
SlotScanner.DEBUG = options.slot_images.value()
SlotScanner.DEBUG_DIR = options.slot_image_directory.value()
plate_type = options.plate_type.value()
barcode_size = options.barcode_size.value()
if plate_type == "None":
scanner = OpenScanner(barcode_size)
else:
scanner = GeometryScanner(plate_type, barcode_size)
while True:
# Get next image from queue (terminate if a queue contains a 'None' sentinel)
frame = task_queue.get(True)
if frame is None:
break
# Make grayscale version of image
image = Image(frame)
gray_image = image.to_grayscale()
# If we have an existing partial plate, merge the new plate with it and only try to read the
# barcodes which haven't already been read. This significantly increases efficiency because
# barcode read is expensive.
scan_result = scanner.scan_next_frame(gray_image)
if options.console_frame.value():
scan_result.print_summary()
if scan_result.success():
# Record the time so we can see how long its been since we last saw a plate
last_plate_time = time.time()
plate = scan_result.plate()
if scan_result.already_scanned():
overlay_queue.put(TextOverlay(SCANNED_TAG, Color.Green()))
elif scan_result.any_valid_barcodes():
overlay_queue.put(PlateOverlay(plate, options))
_plate_beep(plate, options)
if scan_result.any_new_barcodes():
result_queue.put((plate, image))
else:
time_since_plate = time.time() - last_plate_time
if time_since_plate > NO_PUCK_TIME:
overlay_queue.put(TextOverlay(scan_result.error(),
Color.Red()))
def draw_to_image(self, image, color=None):
if color is None:
color = Color.Green()
image.draw_line(self.c1, self.c2, color, 1)
image.draw_line(self.c3, self.c1, color, 1)
