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)