diff --git a/modules/aruco/misc/pattern_generator/MarkerPrinter.py b/modules/aruco/misc/pattern_generator/MarkerPrinter.py new file mode 100644 index 000000000..61314f7a1 --- /dev/null +++ b/modules/aruco/misc/pattern_generator/MarkerPrinter.py @@ -0,0 +1,1288 @@ +#!/usr/bin/env python3 + +# SPDX-License-Identifier: BSD-3-Clause +# +# Copyright (c) 2019, Josh Chien. All rights reserved. + +from argparse import ArgumentParser +import numpy as np +from PIL import Image +import io +import warnings +import os +import cairo +from cairosvg import svg2png +import math +import tempfile + +def SaveArucoDictBytesList(filePath = "arucoDictBytesList.npz"): + import numpy as np + + # cv2 is optional dependency + try: + import cv2 + from cv2 import aruco + + # Name, Flag + dictInfo = \ + [ + ("DICT_4X4_1000", aruco.DICT_4X4_1000), + ("DICT_5X5_1000", aruco.DICT_5X5_1000), + ("DICT_6X6_1000", aruco.DICT_6X6_1000), + ("DICT_7X7_1000", aruco.DICT_7X7_1000), + ("DICT_ARUCO_ORIGINAL", aruco.DICT_ARUCO_ORIGINAL), + ("DICT_APRILTAG_16h5", aruco.DICT_APRILTAG_16h5), + ("DICT_APRILTAG_25h9", aruco.DICT_APRILTAG_25h9), + ("DICT_APRILTAG_36h10", aruco.DICT_APRILTAG_36h10), + ("DICT_APRILTAG_36h11", aruco.DICT_APRILTAG_36h11), + ] + + arucoDictBytesList = {} + for name, flag in dictInfo: + arucoDict = aruco.Dictionary_get(flag) + arucoDictBytesList[name] = arucoDict.bytesList + + np.savez_compressed(filePath, **arucoDictBytesList) + return arucoDictBytesList + + except Exception as e: + warnings.warn(str(e)) + return None + + return None + +class MarkerPrinter: + + debugMode = None # "LINE" "BLOCK" + + # Static Vars + # SVG https://oreillymedia.github.io/Using_SVG/guide/units.html + # for PDF and SVG, 1 pixel = 1/72 inch, 1 cm = 1/2.54 inch, 1pixl = 2.54/72 cm, 1cm = 72/2.54 pixels + ptPerMeter = 72 / 2.54 * 100 + + surface = { + ".SVG": cairo.SVGSurface, + ".PDF": cairo.PDFSurface, + ".PS": cairo.PSSurface } + + if (os.path.isfile("arucoDictBytesList.npz")): + arucoDictBytesList = np.load("arucoDictBytesList.npz") + else: + warnings.warn("Missing build-in arucoDictBytesList.npz, generate it again") + arucoDictBytesList = SaveArucoDictBytesList(filePath = "arucoDictBytesList.npz") + + arucoDictMarkerSize = \ + { + "DICT_4X4_1000": 4, + "DICT_5X5_1000": 5, + "DICT_6X6_1000": 6, + "DICT_7X7_1000": 7, + "DICT_ARUCO_ORIGINAL": 5, + "DICT_APRILTAG_16h5": 4, + "DICT_APRILTAG_25h9": 5, + "DICT_APRILTAG_36h10": 6, + "DICT_APRILTAG_36h11": 6, + } + + def ArucoBits(dictionary, markerID): + bytesList = MarkerPrinter.arucoDictBytesList[dictionary][markerID].ravel() + markerSize = MarkerPrinter.arucoDictMarkerSize[dictionary] + + arucoBits = np.zeros(shape = (markerSize, markerSize), dtype = bool) + base2List = np.array( [128, 64, 32, 16, 8, 4, 2, 1], dtype = np.uint8) + currentByteIdx = 0 + currentByte = bytesList[currentByteIdx] + currentBit = 0 + for row in range(markerSize): + for col in range(markerSize): + if(currentByte >= base2List[currentBit]): + arucoBits[row, col] = True + currentByte -= base2List[currentBit] + currentBit = currentBit + 1 + if(currentBit == 8): + currentByteIdx = currentByteIdx + 1 + currentByte = bytesList[currentByteIdx] + if(8 * (currentByteIdx + 1) > arucoBits.size): + currentBit = 8 * (currentByteIdx + 1) - arucoBits.size + else: + currentBit = 0; + return arucoBits + + def __DrawBlock(context, + dictionary = None, markerLength = None, borderBits = 1, + chessboardSize = (1, 1), squareLength = None, firstMarkerID = 0, + blockX = 0, blockY = 0, originX = 0, originY = 0, pageBorderX = 0, pageBorderY = 0, + mode = "CHESS" ): + + if(squareLength is None): + squareLength = markerLength + + if(markerLength is None): + markerLength = squareLength + + if((squareLength is None) or (markerLength is None)): + raise ValueError("lenght is None") + + if((( blockX % 2 == 0 ) == ( blockY % 2 == 0 )) or mode == "ARUCOGRID"): + if (mode != "CHESS"): + if(dictionary is None): + raise ValueError("dictionary is None") + + if (mode == "CHARUCO"): + originX = (blockX - originX) * squareLength + (squareLength - markerLength)*0.5 + pageBorderX + originY = (blockY - originY) * squareLength + (squareLength - markerLength)*0.5 + pageBorderY + else: + originX = (blockX - originX) * squareLength + pageBorderX + originY = (blockY - originY) * squareLength + pageBorderY + + context.set_source_rgba(0.0, 0.0, 0.0, 1.0) + context.rectangle(originX, originY, markerLength, markerLength) + context.fill() + + # Generate marker + if (mode == "CHARUCO"): + markerID = firstMarkerID + (blockY * chessboardSize[0] + blockX) // 2 + elif (mode == "ARUCO"): + markerID = firstMarkerID + elif (mode == "ARUCOGRID"): + markerID = firstMarkerID + (blockY * chessboardSize[0] + blockX) + + marker = MarkerPrinter.ArucoBits(dictionary, markerID) + markerSize = marker.shape[0] + unitLength = markerLength / (float)(markerSize + borderBits * 2) + + markerBitMap = np.zeros(shape = (markerSize+borderBits*2, markerSize+borderBits*2), dtype = bool) + markerBitMap[borderBits:-borderBits,borderBits:-borderBits] = marker + markerBitMap = np.swapaxes(markerBitMap, 0, 1) + + # Compute edges + hEdges = np.zeros(shape = (markerSize+1,markerSize+1), dtype = bool) + vEdges = np.zeros(shape = (markerSize+1,markerSize+1), dtype = bool) + + for mx in range(markerSize): + for my in range(markerSize+1): + if ( markerBitMap[mx + borderBits, my + borderBits - 1] ^ markerBitMap[mx + borderBits, my + borderBits]): + hEdges[mx, my] = True + + for mx in range(markerSize+1): + for my in range(markerSize): + if ( markerBitMap[mx + borderBits - 1, my + borderBits] ^ markerBitMap[mx + borderBits, my + borderBits]): + vEdges[mx, my] = True + + # Use for debug, check edge or position is correct or not + if(MarkerPrinter.debugMode is not None): + if(MarkerPrinter.debugMode.upper() == "LINE"): + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.set_line_width(unitLength * 0.1) + for mx in range(markerSize+1): + for my in range(markerSize+1): + if(hEdges[mx, my]): + context.move_to(originX + unitLength * (mx + borderBits ), originY + unitLength * (my + borderBits )) + context.line_to(originX + unitLength * (mx + borderBits + 1), originY + unitLength * (my + borderBits )) + context.stroke() + if(vEdges[mx, my]): + context.move_to(originX + unitLength * (mx + borderBits ), originY + unitLength * (my + borderBits )) + context.line_to(originX + unitLength * (mx + borderBits ), originY + unitLength * (my + borderBits + 1)) + context.stroke() + + elif(MarkerPrinter.debugMode.upper() == "BLOCK"): + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + for mx in range(markerSize): + for my in range(markerSize): + if(markerBitMap[mx + borderBits, my + borderBits]): + context.rectangle( + originX + unitLength * (mx + borderBits), + originY + unitLength * (my + borderBits), + unitLength, unitLength) + context.fill() + + else: + while(True): + found = False + + # Find start position + sx = 0 + sy = 0 + for my in range(markerSize): + for mx in range(markerSize): + if(hEdges[mx, my]): + found = True + sx = mx + sy = my + if(markerBitMap[sx + borderBits, sy + borderBits - 1]): + context.set_source_rgba(0.0, 0.0, 0.0, 1.0) + else: + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + break + if(found): + break + + context.move_to (originX + unitLength * (sx + borderBits), originY + unitLength * (sy + borderBits)) + + # Use wall follower maze solving algorithm to draw white part + cx = sx + cy = sy + cd = 3 # 0 right, 1 down, 2 left, 3 up + while(True): + nd = (cd + 1)%4 + moved = False + if(nd == 0): + if(hEdges[cx, cy]): + hEdges[cx, cy] = False + cx = cx + 1 + moved = True + elif(nd == 1): + if(vEdges[cx, cy]): + vEdges[cx, cy] = False + cy = cy + 1 + moved = True + elif(nd == 2): + if(hEdges[cx - 1, cy]): + hEdges[cx - 1, cy] = False + cx = cx - 1 + moved = True + elif(nd == 3): + if(vEdges[cx, cy - 1]): + vEdges[cx, cy - 1] = False + cy = cy - 1 + moved = True + + if((cx == sx) and (cy == sy)): + context.close_path () + break + else: + if(moved): + context.line_to(originX + unitLength * (cx + borderBits), originY + unitLength * (cy + borderBits)) + cd = nd + + if (found): + context.fill() + else: + break + + else: + originX = (blockX - originX) * squareLength + pageBorderX + originY = (blockY - originY) * squareLength + pageBorderY + context.set_source_rgba(0.0, 0.0, 0.0, 1.0) + context.rectangle(originX, originY, squareLength, squareLength) + context.fill() + + def __CheckChessMarkerImage(chessboardSize, squareLength, subSize=None, pageBorder=(0,0)): + if(len(chessboardSize) != 2): + raise ValueError("len(chessboardSize) != 2") + else: + sizeX, sizeY = chessboardSize + + if(len(pageBorder) != 2): + raise ValueError("len(pageBorder) != 2") + else: + pageBorderX, pageBorderY = pageBorder + + if(sizeX <= 1): + raise ValueError("sizeX <= 1") + + if(sizeY <= 1): + raise ValueError("sizeY <= 1") + + if(squareLength <= 0): + raise ValueError("squareLength <= 0") + + if(pageBorderX < 0): + raise ValueError("pageBorderX < 0") + + if(pageBorderY < 0): + raise ValueError("pageBorderY < 0") + + if(subSize is not None): + subSizeX, subSizeY = subSize + + if(subSizeX < 0): + raise ValueError("subSizeX < 0") + + if(subSizeY < 0): + raise ValueError("subSizeY < 0") + + def PreviewChessMarkerImage(chessboardSize, squareLength, pageBorder=(0, 0), dpi=96): + MarkerPrinter.__CheckChessMarkerImage(chessboardSize, squareLength, pageBorder=pageBorder) + + squareLength = squareLength * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + prevImage = None + with tempfile.TemporaryDirectory() as tmpdirname: + with MarkerPrinter.surface[".SVG"] ( + os.path.join(tmpdirname, "tempSVG.svg"), + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + chessboardSize[0] * squareLength, + chessboardSize[1] * squareLength) + context.fill() + + for bx in range(chessboardSize[0]): + for by in range(chessboardSize[1]): + MarkerPrinter.__DrawBlock( + context = context, + chessboardSize = chessboardSize, + squareLength = squareLength, + blockX = bx, + blockY = by, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "CHESS") + + with open(os.path.join(tmpdirname, "tempSVG.svg")) as file: + prevImage = Image.open(io.BytesIO(svg2png(bytestring=file.read(), dpi=dpi))) + + return prevImage + + def GenChessMarkerImage(filePath, chessboardSize, squareLength, subSize=None, pageBorder=(0, 0)): + MarkerPrinter.__CheckChessMarkerImage(chessboardSize, squareLength, subSize=subSize, pageBorder=pageBorder) + + squareLength = squareLength * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + # Check + path, nameExt = os.path.split(filePath) + name, ext = os.path.splitext(nameExt) + + if(len(path) > 0): + if not(os.path.isdir(path)): + os.makedirs(path) + + if((ext.upper() != ".SVG") and (ext.upper() != ".PS") and (ext.upper() != ".PDF")): + raise ValueError("file extention is not supported, should be: svg, ps, pdf") + + # Draw + with MarkerPrinter.surface[ext.upper()] ( + filePath, + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + chessboardSize[0] * squareLength, + chessboardSize[1] * squareLength) + context.fill() + + for bx in range(chessboardSize[0]): + for by in range(chessboardSize[1]): + MarkerPrinter.__DrawBlock( + context = context, + chessboardSize = chessboardSize, + squareLength = squareLength, + blockX = bx, + blockY = by, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "CHESS" ) + + if(subSize is not None): + subDivide = (\ + chessboardSize[0] // subSize[0] + int(chessboardSize[0] % subSize[0] > 0), + chessboardSize[1] // subSize[1] + int(chessboardSize[1] % subSize[1] > 0)) + + subChessboardBlockX = np.clip ( np.arange(0, subSize[0] * subDivide[0] + 1, subSize[0]), 0, chessboardSize[0]) + subChessboardBlockY = np.clip ( np.arange(0, subSize[1] * subDivide[1] + 1, subSize[1]), 0, chessboardSize[1]) + + subChessboardSliceX = subChessboardBlockX.astype(np.float) * squareLength + subChessboardSliceY = subChessboardBlockY.astype(np.float) * squareLength + + for subXID in range(subDivide[0]): + for subYID in range(subDivide[1]): + subName = name + \ + "_X" + str(subChessboardBlockX[subXID]) + "_" + str(subChessboardBlockX[subXID+1]) + \ + "_Y" + str(subChessboardBlockY[subYID]) + "_" + str(subChessboardBlockY[subYID+1]) + + with MarkerPrinter.surface[ext.upper()]( + os.path.join(path, subName + ext), + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID] + pageBorder[0] * 2, + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID] + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID] + pageBorder[0] * 2, + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID] + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID], + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID]) + context.fill() + + for bx in range(subChessboardBlockX[subXID+1] - subChessboardBlockX[subXID]): + for by in range(subChessboardBlockY[subYID+1] - subChessboardBlockY[subYID]): + MarkerPrinter.__DrawBlock( + context = context, + chessboardSize = chessboardSize, + squareLength = squareLength, + blockX = subChessboardBlockX[subXID] + bx, + blockY = subChessboardBlockY[subYID] + by, + originX = subChessboardBlockX[subXID], + originY = subChessboardBlockY[subYID], + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "CHESS" ) + + + def __CheckArucoMarkerImage(dictionary, markerID, markerLength, borderBits=1, pageBorder=(0, 0)): + if(len(pageBorder) != 2): + raise ValueError("len(pageBorder) != 2") + else: + pageBorderX, pageBorderY = pageBorder + + if not (dictionary in MarkerPrinter.arucoDictBytesList): + raise ValueError("dictionary is not support") + + if(MarkerPrinter.arucoDictBytesList[dictionary].shape[0] <= markerID ): + raise ValueError("markerID is not in aruce dictionary") + + if(markerID < 0): + raise ValueError("markerID < 0") + + if(markerLength <= 0): + raise ValueError("markerLength <= 0") + + if(borderBits <= 0): + raise ValueError("borderBits <= 0") + + if(pageBorderX < 0): + raise ValueError("pageBorderX < 0") + + if(pageBorderY < 0): + raise ValueError("pageBorderY < 0") + + def PreviewArucoMarkerImage(dictionary, markerID, markerLength, borderBits=1, pageBorder=(0, 0), dpi=96): + MarkerPrinter.__CheckArucoMarkerImage(dictionary, markerID, markerLength, borderBits=borderBits, pageBorder=pageBorder) + + markerLength = markerLength * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + prevImage = None + with tempfile.TemporaryDirectory() as tmpdirname: + with MarkerPrinter.surface[".SVG"] ( + os.path.join(tmpdirname, "tempSVG.svg"), + markerLength + pageBorder[0] * 2, + markerLength + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + markerLength + pageBorder[0] * 2, + markerLength + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + markerLength, + markerLength) + context.fill() + + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + firstMarkerID = markerID, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "ARUCO") + + with open(os.path.join(tmpdirname, "tempSVG.svg")) as file: + prevImage = Image.open(io.BytesIO(svg2png(bytestring=file.read(), dpi=dpi))) + + return prevImage + + def GenArucoMarkerImage(filePath, dictionary, markerID, markerLength, borderBits=1, pageBorder=(0, 0)): + MarkerPrinter.__CheckArucoMarkerImage(dictionary, markerID, markerLength, borderBits=borderBits, pageBorder=pageBorder) + + markerLength = markerLength * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + # Check + path, nameExt = os.path.split(filePath) + name, ext = os.path.splitext(nameExt) + + if(len(path) > 0): + if not(os.path.isdir(path)): + os.makedirs(path) + + if((ext.upper() != ".SVG") and (ext.upper() != ".PS") and (ext.upper() != ".PDF")): + raise ValueError("file extention is not supported, should be: svg, ps, pdf") + + # Draw + with MarkerPrinter.surface[ext.upper()] ( + filePath, + markerLength + pageBorder[0] * 2, + markerLength + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + markerLength + pageBorder[0] * 2, + markerLength + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + markerLength, + markerLength) + context.fill() + + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + firstMarkerID = markerID, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "ARUCO") + + def __CheckCharucoMarkerImage(dictionary, chessboardSize, squareLength, markerLength, borderBits=1, subSize=None, pageBorder=(0, 0)): + if(len(chessboardSize) != 2): + raise ValueError("len(chessboardSize) != 2") + else: + sizeX, sizeY = chessboardSize + + if(len(pageBorder) != 2): + raise ValueError("len(pageBorder) != 2") + else: + pageBorderX, pageBorderY = pageBorder + + if not (dictionary in MarkerPrinter.arucoDictBytesList): + raise ValueError("dictionary is not support") + + if(MarkerPrinter.arucoDictBytesList[dictionary].shape[0] < (( sizeX * sizeY ) // 2)): + raise ValueError("aruce dictionary is not enough for your board size") + + if(sizeX <= 1): + raise ValueError("sizeX <= 1") + + if(sizeY <= 1): + raise ValueError("sizeY <= 1") + + if(squareLength <= 0): + raise ValueError("squareLength <= 0") + + if(markerLength <= 0): + raise ValueError("markerLength <= 0") + + if(squareLength < markerLength): + raise ValueError("squareLength < markerLength") + + if(borderBits <= 0): + raise ValueError("borderBits <= 0") + + if(pageBorderX < 0): + raise ValueError("pageBorderX < 0") + + if(pageBorderY < 0): + raise ValueError("pageBorderY < 0") + + if(subSize is not None): + subSizeX, subSizeY = subSize + + if(subSizeX < 0): + raise ValueError("subSizeX < 0") + + if(subSizeY < 0): + raise ValueError("subSizeY < 0") + + def PreviewCharucoMarkerImage(dictionary, chessboardSize, squareLength, markerLength, borderBits=1, pageBorder=(0, 0), dpi=96): + MarkerPrinter.__CheckCharucoMarkerImage(dictionary, chessboardSize, squareLength, markerLength, borderBits=borderBits, pageBorder=pageBorder) + + squareLength = squareLength * MarkerPrinter.ptPerMeter + markerLength = markerLength * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + prevImage = None + with tempfile.TemporaryDirectory() as tmpdirname: + with MarkerPrinter.surface[".SVG"] ( + os.path.join(tmpdirname, "tempSVG.svg"), + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + chessboardSize[0] * squareLength, + chessboardSize[1] * squareLength) + context.fill() + + for bx in range(chessboardSize[0]): + for by in range(chessboardSize[1]): + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + chessboardSize = chessboardSize, + squareLength = squareLength, + blockX = bx, + blockY = by, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "CHARUCO") + + with open(os.path.join(tmpdirname, "tempSVG.svg")) as file: + prevImage = Image.open(io.BytesIO(svg2png(bytestring=file.read(), dpi=dpi))) + + return prevImage + + def GenCharucoMarkerImage(filePath, dictionary, chessboardSize, squareLength, markerLength, borderBits=1, subSize=None, pageBorder=(0, 0)): + MarkerPrinter.__CheckCharucoMarkerImage(dictionary, chessboardSize, squareLength, markerLength, borderBits=borderBits, subSize=subSize, pageBorder=pageBorder) + + squareLength = squareLength * MarkerPrinter.ptPerMeter + markerLength = markerLength * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + # Check + path, nameExt = os.path.split(filePath) + name, ext = os.path.splitext(nameExt) + + if(len(path) > 0): + if not(os.path.isdir(path)): + os.makedirs(path) + + if((ext.upper() != ".SVG") and (ext.upper() != ".PS") and (ext.upper() != ".PDF")): + raise ValueError("file extention is not supported, should be: svg, ps, pdf") + + # Draw + with MarkerPrinter.surface[ext.upper()] ( + filePath, + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + chessboardSize[0] * squareLength + pageBorder[0] * 2, + chessboardSize[1] * squareLength + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + chessboardSize[0] * squareLength, + chessboardSize[1] * squareLength) + context.fill() + + for bx in range(chessboardSize[0]): + for by in range(chessboardSize[1]): + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + chessboardSize = chessboardSize, + squareLength = squareLength, + blockX = bx, + blockY = by, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "CHARUCO") + + if(subSize is not None): + subDivide = (\ + chessboardSize[0] // subSize[0] + int(chessboardSize[0] % subSize[0] > 0), + chessboardSize[1] // subSize[1] + int(chessboardSize[1] % subSize[1] > 0)) + + subChessboardBlockX = np.clip ( np.arange(0, subSize[0] * subDivide[0] + 1, subSize[0]), 0, chessboardSize[0]) + subChessboardBlockY = np.clip ( np.arange(0, subSize[1] * subDivide[1] + 1, subSize[1]), 0, chessboardSize[1]) + + subChessboardSliceX = subChessboardBlockX.astype(np.float) * squareLength + subChessboardSliceY = subChessboardBlockY.astype(np.float) * squareLength + + for subXID in range(subDivide[0]): + for subYID in range(subDivide[1]): + subName = name + \ + "_X" + str(subChessboardBlockX[subXID]) + "_" + str(subChessboardBlockX[subXID+1]) + \ + "_Y" + str(subChessboardBlockY[subYID]) + "_" + str(subChessboardBlockY[subYID+1]) + + with MarkerPrinter.surface[ext.upper()]( + os.path.join(path, subName + ext), + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID] + pageBorder[0] * 2, + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID] + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID] + pageBorder[0] * 2, + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID] + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID], + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID]) + context.fill() + + for bx in range(subChessboardBlockX[subXID+1] - subChessboardBlockX[subXID]): + for by in range(subChessboardBlockY[subYID+1] - subChessboardBlockY[subYID]): + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + chessboardSize = chessboardSize, + squareLength = squareLength, + blockX = subChessboardBlockX[subXID] + bx, + blockY = subChessboardBlockY[subYID] + by, + originX = subChessboardBlockX[subXID], + originY = subChessboardBlockY[subYID], + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "CHARUCO") + + def __CheckArucoGridMarkerImage(dictionary, chessboardSize, markerLength, markerSeparation, firstMarker, borderBits=1, subSize=None, pageBorder=(0, 0)): + if(len(chessboardSize) != 2): + raise ValueError("len(chessboardSize) != 2") + else: + sizeX, sizeY = chessboardSize + + if(len(pageBorder) != 2): + raise ValueError("len(pageBorder) != 2") + else: + pageBorderX, pageBorderY = pageBorder + + if not (dictionary in MarkerPrinter.arucoDictBytesList): + raise ValueError("dictionary is not support") + + if(MarkerPrinter.arucoDictBytesList[dictionary].shape[0] < (( sizeX * sizeY ) + firstMarker)): + raise ValueError("aruce dictionary is not enough for your board size and firstMarker") + + if(sizeX <= 1): + raise ValueError("sizeX <= 1") + + if(sizeY <= 1): + raise ValueError("sizeY <= 1") + + if(markerLength <= 0): + raise ValueError("markerLength <= 0") + + if(markerSeparation <= 0): + raise ValueError("markerSeparation <= 0") + + if(borderBits <= 0): + raise ValueError("borderBits <= 0") + + if(pageBorderX < 0): + raise ValueError("pageBorderX < 0") + + if(pageBorderY < 0): + raise ValueError("pageBorderY < 0") + + if(subSize is not None): + subSizeX, subSizeY = subSize + + if(subSizeX < 0): + raise ValueError("subSizeX < 0") + + if(subSizeY < 0): + raise ValueError("subSizeY < 0") + + def PreviewArucoGridMarkerImage(dictionary, chessboardSize, markerLength, markerSeparation, firstMarker, borderBits=1, pageBorder=(0, 0), dpi=96): + MarkerPrinter.__CheckArucoGridMarkerImage(dictionary, chessboardSize, markerLength, markerSeparation, firstMarker, borderBits=borderBits, pageBorder=pageBorder) + + markerLength = markerLength * MarkerPrinter.ptPerMeter + markerSeparation = markerSeparation * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + prevImage = None + with tempfile.TemporaryDirectory() as tmpdirname: + with MarkerPrinter.surface[".SVG"] ( + os.path.join(tmpdirname, "tempSVG.svg"), + chessboardSize[0] * markerLength + (chessboardSize[0] - 1) * markerSeparation + pageBorder[0] * 2, + chessboardSize[1] * markerLength + (chessboardSize[1] - 1) * markerSeparation + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + chessboardSize[0] * markerLength + (chessboardSize[0] - 1) * markerSeparation + pageBorder[0] * 2, + chessboardSize[1] * markerLength + (chessboardSize[1] - 1) * markerSeparation + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + chessboardSize[0] * markerLength + (chessboardSize[0] - 1) * markerSeparation, + chessboardSize[1] * markerLength + (chessboardSize[1] - 1) * markerSeparation) + context.fill() + + for bx in range(chessboardSize[0]): + for by in range(chessboardSize[1]): + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + chessboardSize = chessboardSize, + squareLength = markerLength + markerSeparation, + firstMarkerID = firstMarker, + blockX = bx, + blockY = by, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "ARUCOGRID") + + with open(os.path.join(tmpdirname, "tempSVG.svg")) as file: + prevImage = Image.open(io.BytesIO(svg2png(bytestring=file.read(), dpi=dpi))) + + return prevImage + + def GenArucoGridMarkerImage(filePath, dictionary, chessboardSize, markerLength, markerSeparation, firstMarker, borderBits=1, subSize=None, pageBorder=(0, 0)): + MarkerPrinter.__CheckArucoGridMarkerImage(dictionary, chessboardSize, markerLength, markerSeparation, firstMarker, borderBits=borderBits, subSize=subSize, pageBorder=pageBorder) + + markerLength = markerLength * MarkerPrinter.ptPerMeter + markerSeparation = markerSeparation * MarkerPrinter.ptPerMeter + pageBorder = (pageBorder[0] * MarkerPrinter.ptPerMeter, pageBorder[1] * MarkerPrinter.ptPerMeter) + + # Check + path, nameExt = os.path.split(filePath) + name, ext = os.path.splitext(nameExt) + + if(len(path) > 0): + if not(os.path.isdir(path)): + os.makedirs(path) + + if((ext.upper() != ".SVG") and (ext.upper() != ".PS") and (ext.upper() != ".PDF")): + raise ValueError("file extention is not supported, should be: svg, ps, pdf") + + # Draw + with MarkerPrinter.surface[ext.upper()] ( + filePath, + chessboardSize[0] * markerLength + (chessboardSize[0] - 1) * markerSeparation + pageBorder[0] * 2, + chessboardSize[1] * markerLength + (chessboardSize[1] - 1) * markerSeparation + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + chessboardSize[0] * markerLength + (chessboardSize[0] - 1) * markerSeparation + pageBorder[0] * 2, + chessboardSize[1] * markerLength + (chessboardSize[1] - 1) * markerSeparation + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + chessboardSize[0] * markerLength + (chessboardSize[0] - 1) * markerSeparation, + chessboardSize[1] * markerLength + (chessboardSize[1] - 1) * markerSeparation) + context.fill() + + for bx in range(chessboardSize[0]): + for by in range(chessboardSize[1]): + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + chessboardSize = chessboardSize, + squareLength = markerLength + markerSeparation, + firstMarkerID = firstMarker, + blockX = bx, + blockY = by, + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "ARUCOGRID") + + if(subSize is not None): + subDivide = (\ + chessboardSize[0] // subSize[0] + int(chessboardSize[0] % subSize[0] > 0), + chessboardSize[1] // subSize[1] + int(chessboardSize[1] % subSize[1] > 0)) + + subChessboardBlockX = np.clip ( np.arange(0, subSize[0] * subDivide[0] + 1, subSize[0]), 0, chessboardSize[0]) + subChessboardBlockY = np.clip ( np.arange(0, subSize[1] * subDivide[1] + 1, subSize[1]), 0, chessboardSize[1]) + + subChessboardSliceX = subChessboardBlockX.astype(np.float) * (markerLength + markerSeparation) + subChessboardSliceY = subChessboardBlockY.astype(np.float) * (markerLength + markerSeparation) + + subChessboardSliceX[-1] -= markerSeparation + subChessboardSliceY[-1] -= markerSeparation + + for subXID in range(subDivide[0]): + for subYID in range(subDivide[1]): + subName = name + \ + "_X" + str(subChessboardBlockX[subXID]) + "_" + str(subChessboardBlockX[subXID+1]) + \ + "_Y" + str(subChessboardBlockY[subYID]) + "_" + str(subChessboardBlockY[subYID+1]) + + with MarkerPrinter.surface[ext.upper()]( + os.path.join(path, subName + ext), + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID] + pageBorder[0] * 2, + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID] + pageBorder[1] * 2) as surface: + context = cairo.Context(surface) + + context.set_source_rgba(0.5, 0.5, 0.5, 1.0) + context.rectangle(0, 0, + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID] + pageBorder[0] * 2, + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID] + pageBorder[1] * 2) + context.fill() + + context.set_source_rgba(1.0, 1.0, 1.0, 1.0) + context.rectangle(pageBorder[0], pageBorder[1], + subChessboardSliceX[subXID+1] - subChessboardSliceX[subXID], + subChessboardSliceY[subYID+1] - subChessboardSliceY[subYID]) + context.fill() + + for bx in range(subChessboardBlockX[subXID+1] - subChessboardBlockX[subXID]): + for by in range(subChessboardBlockY[subYID+1] - subChessboardBlockY[subYID]): + MarkerPrinter.__DrawBlock( + context = context, + dictionary = dictionary, + markerLength = markerLength, + borderBits = borderBits, + chessboardSize = chessboardSize, + squareLength = markerLength + markerSeparation, + firstMarkerID = firstMarker, + blockX = subChessboardBlockX[subXID] + bx, + blockY = subChessboardBlockY[subYID] + by, + originX = subChessboardBlockX[subXID], + originY = subChessboardBlockY[subYID], + pageBorderX = pageBorder[0], + pageBorderY = pageBorder[1], + mode = "ARUCOGRID") + +if __name__ == '__main__': + parser = ArgumentParser() + + # Save marker image parameters + chessGroup = parser.add_argument_group('chess', 'Chessboard') + arucoGroup = parser.add_argument_group('aruco', 'ArUco') + arucoGridGroup = parser.add_argument_group('aruco_grid', 'ArUco grid') + charucoGroup = parser.add_argument_group('charuco', 'ChArUco') + exclusiveGroup = parser.add_mutually_exclusive_group() + + exclusiveGroup.add_argument( + "--chess", action='store_true', default=False, + help="Choose to save chessboard marker") + + exclusiveGroup.add_argument( + "--aruco", action='store_true', default=False, + help="Choose to save ArUco marker") + + exclusiveGroup.add_argument( + "--aruco_grid", action='store_true', default=False, + help="Choose to save ArUco grid marker") + + exclusiveGroup.add_argument( + "--charuco", action='store_true', default=False, + help="Choose to save ChArUco marker") + + # Utility functions parameters + exclusiveGroup.add_argument( + "--generate", dest="arucoDataFileName", + help="Generate aruco data to FILE", metavar="FILE") + + exclusiveGroup.add_argument( + "--list_dictionary", action='store_true', default=False, + help="List predefined aruco dictionary") + + # Parameters + # fileName + parser.add_argument( + "--file", dest="fileName", default="./image.pdf", + help="Save marker image to FILE", metavar="FILE") + for group in [chessGroup, arucoGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_file", dest="fileName", + help="Save marker image to FILE", metavar="FILE") + + # dictionary + parser.add_argument( + "--dictionary", dest="dictionary", default="DICT_ARUCO_ORIGINAL", + help="Generate marker via predefined DICTIONARY aruco dictionary", metavar="DICTIONARY") + for group in [arucoGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_dictionary", dest="dictionary", + help="Generate marker via predefined DICTIONARY aruco dictionary", metavar="DICTIONARY") + + # size + parser.add_argument( + "--size_x", dest="sizeX", default="16", + help="Save marker image with N board width", metavar="N") + parser.add_argument( + "--size_y", dest="sizeY", default="9", + help="Save marker image with N board height", metavar="N") + + for group in [chessGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_size_x", dest="sizeX", + help="Save marker image with N board width", metavar="N") + group.add_argument( + "--" + group.title + "_size_y", dest="sizeY", + help="Save marker image with N board height", metavar="N") + + # length + parser.add_argument( + "--square_length", dest="squareLength", default="0.09", + help="Save marker image with L square length (Unit: meter)", metavar="L") + parser.add_argument( + "--marker_length", dest="markerLength", default="0.07", + help="Save marker image with L marker length (Unit: meter)", metavar="L") + parser.add_argument( + "--marker_separation", dest="markerSeparation", default="0.02", + help="Save marker image with L separation length (Unit: meter)", metavar="L") + + for group in [chessGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_square_length", dest="squareLength", + help="Save marker image with L blocks length (Unit: meter)", metavar="L") + + for group in [arucoGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_marker_length", dest="markerLength", + help="Save marker image with L marker length (Unit: meter)", metavar="L") + + for group in [arucoGridGroup]: + group.add_argument( + "--" + group.title + "_marker_separation", dest="markerSeparation", + help="Save marker image with L gap length (Unit: meter)", metavar="L") + + # else + parser.add_argument( + "--marker_id", dest="markerID", default="0", + help="Save marker image with ID marker", metavar="ID") + parser.add_argument( + "--first_marker", dest="firstMarker", default="0", + help="Save marker image that start with ID marker", metavar="ID") + parser.add_argument( + "--border_bits", dest="borderBits", default="1", + help="Save marker image with N border size", metavar="N") + + for group in [arucoGroup]: + group.add_argument( + "--" + group.title + "_marker_id", dest="markerID", + help="Save marker image with ID marker", metavar="ID") + + for group in [arucoGridGroup]: + group.add_argument( + "--" + group.title + "_first_marker", dest="firstMarker", + help="Save marker image that start with ID marker", metavar="ID") + + for group in [arucoGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_border_bits", dest="borderBits", + help="Save marker image with N border size", metavar="N") + + # sub size + parser.add_argument( + "--sub_size_x", dest="subSizeX", default="0", + help="Save marker image with N chuck width", metavar="N") + parser.add_argument( + "--sub_size_y", dest="subSizeY", default="0", + help="Save marker image with N chuck height", metavar="N") + + for group in [chessGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_sub_size_x", dest="subSizeX", + help="Save marker image with N chuck width", metavar="N") + group.add_argument( + "--" + group.title + "_sub_size_y", dest="subSizeY", + help="Save marker image with N chuck height", metavar="N") + + # page border + parser.add_argument( + "--page_border_x", dest="pageBorderX", default="0", + help="Save with page border width L length (Unit: meter)", metavar="L") + parser.add_argument( + "--page_border_y", dest="pageBorderY", default="0", + help="Save with page border height L length (Unit: meter)", metavar="L") + + for group in [chessGroup, arucoGroup, arucoGridGroup, charucoGroup]: + group.add_argument( + "--" + group.title + "_page_border_x", dest="pageBorderX", default="0", + help="Save with page border width L length (Unit: meter)", metavar="L") + group.add_argument( + "--" + group.title + "_page_border_y", dest="pageBorderY", default="0", + help="Save with page border height L length (Unit: meter)", metavar="L") + + # Run + args = parser.parse_args() + + if(args.arucoDataFileName is not None): + print("Generate aruco data to: " + args.arucoDataFileName) + SaveArucoDictBytesList(args.arucoDataFileName) + + elif(args.list_dictionary): + print("List predefined aruco dictionary") + for i in MarkerPrinter.arucoDictBytesList.keys(): + print(i) + + elif(args.chess): + try: + sizeX = int(args.sizeX) + sizeY = int(args.sizeY) + squareLength = float(args.squareLength) + subSizeX = int(args.subSizeX) + subSizeY = int(args.subSizeY) + pageBorderX = float(args.pageBorderX) + pageBorderY = float(args.pageBorderY) + except ValueError as e: + warnings.warn(str(e)) + else: + print("Save chessboard marker with parms: " + \ + str({ \ + "fileName": args.fileName, \ + "sizeX": sizeX, \ + "sizeY": sizeY, \ + "squareLength": squareLength, \ + "subSizeX": subSizeX, \ + "subSizeY": subSizeY, \ + "pageBorderX": pageBorderX, \ + "pageBorderY": pageBorderY, \ + })) + + subSize = None + + if(subSizeX > 0): + if(subSizeY > 0): + subSize = (subSizeX, subSizeY) + else: + subSize = (subSizeX, sizeY) + else: + if(subSizeY > 0): + subSize = (sizeX, subSizeY) + else: + subSize = None + + # Gen + MarkerPrinter.GenChessMarkerImage(args.fileName, (sizeX, sizeY), squareLength, subSize = subSize, pageBorder = (pageBorderX, pageBorderY)) + + elif(args.aruco): + try: + markerLength = float(args.markerLength) + markerID = int(args.markerID) + borderBits = int(args.borderBits) + pageBorderX = float(args.pageBorderX) + pageBorderY = float(args.pageBorderY) + except ValueError as e: + warnings.warn(str(e)) + else: + print("Save ArUco marker with parms: " + \ + str({ \ + "fileName": args.fileName, \ + "dictionary": args.dictionary, \ + "markerLength": markerLength, \ + "markerID": markerID, \ + "borderBits": borderBits, \ + "pageBorderX": pageBorderX, \ + "pageBorderY": pageBorderY, \ + })) + + # Gen + MarkerPrinter.GenArucoMarkerImage(args.fileName, args.dictionary, markerID, markerLength, borderBits=borderBits, pageBorder = (pageBorderX, pageBorderY)) + + elif(args.aruco_grid): + try: + sizeX = int(args.sizeX) + sizeY = int(args.sizeY) + markerLength = float(args.markerLength) + markerSeparation = float(args.markerSeparation) + firstMarker = int(args.firstMarker) + borderBits = int(args.borderBits) + subSizeX = int(args.subSizeX) + subSizeY = int(args.subSizeY) + pageBorderX = float(args.pageBorderX) + pageBorderY = float(args.pageBorderY) + except ValueError as e: + warnings.warn(str(e)) + else: + print("Save ArUco grid marker with parms: " + \ + str({ \ + "fileName": args.fileName, \ + "dictionary": args.dictionary, \ + "sizeX": sizeX, \ + "sizeY": sizeY, \ + "markerLength": markerLength, \ + "markerSeparation": markerSeparation, \ + "firstMarker": firstMarker, \ + "borderBits": borderBits, \ + "subSizeX": subSizeX, \ + "subSizeY": subSizeY, \ + "pageBorderX": pageBorderX, \ + "pageBorderY": pageBorderY, \ + })) + + subSize = None + + if(subSizeX > 0): + if(subSizeY > 0): + subSize = (subSizeX, subSizeY) + else: + subSize = (subSizeX, sizeY) + else: + if(subSizeY > 0): + subSize = (sizeX, subSizeY) + else: + subSize = None + + # Gen + MarkerPrinter.GenArucoGridMarkerImage(args.fileName, args.dictionary, (sizeX, sizeY), markerLength, markerSeparation, firstMarker, borderBits=borderBits, subSize=subSize, pageBorder = (pageBorderX, pageBorderY)) + + elif(args.charuco): + try: + sizeX = int(args.sizeX) + sizeY = int(args.sizeY) + squareLength = float(args.squareLength) + markerLength = float(args.markerLength) + borderBits = int(args.borderBits) + subSizeX = int(args.subSizeX) + subSizeY = int(args.subSizeY) + pageBorderX = float(args.pageBorderX) + pageBorderY = float(args.pageBorderY) + except ValueError as e: + warnings.warn(str(e)) + else: + print("Save ChArUco marker with parms: " + \ + str({ \ + "fileName": args.fileName, \ + "dictionary": args.dictionary, \ + "sizeX": sizeX, \ + "sizeY": sizeY, \ + "squareLength": squareLength, \ + "markerLength": markerLength, \ + "borderBits": borderBits, \ + "subSizeX": subSizeX, \ + "subSizeY": subSizeY, \ + "pageBorderX": pageBorderX, \ + "pageBorderY": pageBorderY, \ + })) + + subSize = None + + if(subSizeX > 0): + if(subSizeY > 0): + subSize = (subSizeX, subSizeY) + else: + subSize = (subSizeX, sizeY) + else: + if(subSizeY > 0): + subSize = (sizeX, subSizeY) + else: + subSize = None + + # Gen + MarkerPrinter.GenCharucoMarkerImage(args.fileName, args.dictionary, (sizeX, sizeY), squareLength, markerLength, borderBits=borderBits, subSize=subSize, pageBorder = (pageBorderX, pageBorderY)) + + else: + parser.print_help() diff --git a/modules/aruco/misc/pattern_generator/MarkerPrinterGUI.py b/modules/aruco/misc/pattern_generator/MarkerPrinterGUI.py new file mode 100644 index 000000000..bdd6d3cb0 --- /dev/null +++ b/modules/aruco/misc/pattern_generator/MarkerPrinterGUI.py @@ -0,0 +1,565 @@ +#!/usr/bin/env python3 + +# SPDX-License-Identifier: BSD-3-Clause +# +# Copyright (c) 2019, Josh Chien. All rights reserved. + +from MarkerPrinter import * + +import tkinter as tk +from tkinter import ttk, filedialog, messagebox + +import time + +import PIL.Image +import PIL.ImageTk + +class MarkerPrinterGUI: + + def VisDPI(self, shape): + scale0 = float(self.displayShape[0]) / float(shape[0]) + scale1 = float(self.displayShape[1]) / float(shape[1]) + if(scale0 > scale1): + return scale1 * 96.0 + else: + return scale0 * 96.0 + + def OnShowingHelpGithub(self): + messagebox.showinfo("Github", + "https://github.com/dogod621/OpenCVMarkerPrinter") + + def OnCloseWindow(self): + if(self.window is not None): + if messagebox.askokcancel("Quit", "Do you want to quit?"): + self.window.destroy() + self.window = None + + def OnSelectCharucoMarkerDictionary(self, pDictName): + self.charucoMarkerDictionaryStr.set(pDictName) + + def __SaveMarker(GenMarkerImageCallback, *args, **kwargs): + + if(kwargs.get("subSize",None) is not None): + subSizeX, subSizeY = kwargs["subSize"] + + kwargs["subSize"] = None + + if(subSizeX > 0): + if(subSizeY > 0): + kwargs["subSize"] = (subSizeX, subSizeY) + else: + kwargs["subSize"] = (subSizeX, sizeY) + else: + if(subSizeY > 0): + kwargs["subSize"] = (sizeX, subSizeY) + else: + kwargs["subSize"] = None + + try: + askFileName = filedialog.asksaveasfilename(initialdir = os.path.abspath("./"), title = "Output", filetypes = (\ + ("scalable vector graphics files","*.svg"), \ + ("portable document format files","*.pdf"), \ + ("post script files","*.ps")), + defaultextension="*.*") + + if (askFileName): + GenMarkerImageCallback(askFileName, *args, **kwargs) + + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Save marker failed") + return + + def OnPreviewOrSaveCharucoMarker(self, askSave = False): + try: + sizeX = int(self.charucoMarkerChessboardSizeXStr.get()) + sizeY = int(self.charucoMarkerChessboardSizeYStr.get()) + squareLength = float(self.charucoMarkerSquareLengthStr.get()) + markerLength = float(self.charucoMarkerMarkerLengthStr.get()) + borderBits = int(self.charucoMarkerBorderBitsStr.get()) + dictionary = self.charucoMarkerDictionaryStr.get() + subSizeX = int(self.charucoMarkerSaveSubSizeXStr.get()) + subSizeY = int(self.charucoMarkerSaveSubSizeYStr.get()) + pageBorderX = float(self.charucoMarkerSavePageBorderXStr.get()) + pageBorderY = float(self.charucoMarkerSavePageBorderYStr.get()) + except ValueError as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Enter invalid parameters") + return + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Fail to get parameters") + return + + # Preview + try: + dpi = self.VisDPI(((sizeY * squareLength + pageBorderY * 2) * MarkerPrinter.ptPerMeter, (sizeX * squareLength + pageBorderX * 2) * MarkerPrinter.ptPerMeter)) + tkImage = PIL.ImageTk.PhotoImage(image = MarkerPrinter.PreviewCharucoMarkerImage(dictionary, (sizeX, sizeY), squareLength, markerLength, borderBits=borderBits, pageBorder = (pageBorderX, pageBorderY), dpi=dpi)) + self.charucoMarkerImageLabel.imgtk = tkImage + self.charucoMarkerImageLabel.config(image=tkImage) + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "create marker failed") + return + + # Save + if(askSave): + MarkerPrinterGUI.__SaveMarker(MarkerPrinter.GenCharucoMarkerImage, \ + dictionary, (sizeX, sizeY), squareLength, markerLength, borderBits=borderBits, subSize = (subSizeX, subSizeY), pageBorder = (pageBorderX, pageBorderY)) + + def OnPreviewCharucoMarker(self): + self.OnPreviewOrSaveCharucoMarker(askSave = False) + + def OnSaveCharucoMarker(self): + self.OnPreviewOrSaveCharucoMarker(askSave = True) + + def InitCharucoMarkerTab(self): + self.charucoMarkerUIFrame = ttk.Frame(self.charucoMarkerTab) + self.charucoMarkerImageTab = ttk.Frame(self.charucoMarkerTab) + self.charucoMarkerUIFrame2 = ttk.Frame(self.charucoMarkerTab) + + self.charucoMarkerUIFrame.grid(row=0, column=0, sticky = tk.NSEW) + self.charucoMarkerImageTab.grid(row=1, column=0, sticky = tk.NSEW) + self.charucoMarkerUIFrame2.grid(row=2, column=0, sticky = tk.NSEW) + + self.charucoMarkerImageLabel = tk.Label(self.charucoMarkerImageTab) + self.charucoMarkerImageLabel.grid(row=0, column=0, sticky = tk.NSEW) + + tk.Label(self.charucoMarkerUIFrame, text="dictionary").grid(row=0, column=0, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame, text="chessboardSizeX").grid(row=0, column=1, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame, text="chessboardSizeY").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame, text="squareLength (Unit: Meter)").grid(row=0, column=3, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame, text="markerLength (Unit: Meter)").grid(row=0, column=4, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame, text="borderBits").grid(row=0, column=5, sticky = tk.NSEW) + + self.charucoMarkerDictionaryStr = tk.StringVar() + self.charucoMarkerChessboardSizeXStr = tk.StringVar() + self.charucoMarkerChessboardSizeXStr.set("16") + self.charucoMarkerChessboardSizeYStr = tk.StringVar() + self.charucoMarkerChessboardSizeYStr.set("9") + self.charucoMarkerSquareLengthStr = tk.StringVar() + self.charucoMarkerSquareLengthStr.set("0.09") + self.charucoMarkerMarkerLengthStr = tk.StringVar() + self.charucoMarkerMarkerLengthStr.set("0.07") + self.charucoMarkerBorderBitsStr = tk.StringVar() + self.charucoMarkerBorderBitsStr.set("1") + + self.charucoMarkerDictionaryMenue = tk.OptionMenu(self.charucoMarkerUIFrame, self.charucoMarkerDictionaryStr, "DICT_ARUCO_ORIGINAL", command = self.OnSelectCharucoMarkerDictionary) + self.charucoMarkerDictionaryMenue.grid(row=1, column=0, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame, textvariable=self.charucoMarkerChessboardSizeXStr).grid(row=1, column=1, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame, textvariable=self.charucoMarkerChessboardSizeYStr).grid(row=1, column=2, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame, textvariable=self.charucoMarkerSquareLengthStr).grid(row=1, column=3, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame, textvariable=self.charucoMarkerMarkerLengthStr).grid(row=1, column=4, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame, textvariable=self.charucoMarkerBorderBitsStr).grid(row=1, column=5, sticky = tk.NSEW) + + tk.Button(self.charucoMarkerUIFrame2, text = "Preview", command = self.OnPreviewCharucoMarker).grid(row=1, column=0, sticky = tk.NSEW) + tk.Button(self.charucoMarkerUIFrame2, text = "Save", command = self.OnSaveCharucoMarker).grid(row=1, column=1, sticky = tk.NSEW) + + tk.Label(self.charucoMarkerUIFrame2, text="Save opetions:").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="(set 0 as disable)").grid(row=1, column=2, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="subSizeX").grid(row=0, column=3, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="subSizeY").grid(row=0, column=4, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="Divide to chunks, chunk sizeX").grid(row=2, column=3, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="Divide to chunks, chunk sizeY").grid(row=2, column=4, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="pageBorderX (Unit: Meter)").grid(row=0, column=5, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="pageBorderY (Unit: Meter)").grid(row=0, column=6, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="Border or page").grid(row=2, column=5, sticky = tk.NSEW) + tk.Label(self.charucoMarkerUIFrame2, text="Border or page").grid(row=2, column=6, sticky = tk.NSEW) + + self.charucoMarkerSaveSubSizeXStr = tk.StringVar() + self.charucoMarkerSaveSubSizeXStr.set("0") + self.charucoMarkerSaveSubSizeYStr = tk.StringVar() + self.charucoMarkerSaveSubSizeYStr.set("0") + self.charucoMarkerSavePageBorderXStr = tk.StringVar() + self.charucoMarkerSavePageBorderXStr.set("0.02") + self.charucoMarkerSavePageBorderYStr = tk.StringVar() + self.charucoMarkerSavePageBorderYStr.set("0.02") + + tk.Entry(self.charucoMarkerUIFrame2, textvariable=self.charucoMarkerSaveSubSizeXStr).grid(row=1, column=3, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame2, textvariable=self.charucoMarkerSaveSubSizeYStr).grid(row=1, column=4, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame2, textvariable=self.charucoMarkerSavePageBorderXStr).grid(row=1, column=5, sticky = tk.NSEW) + tk.Entry(self.charucoMarkerUIFrame2, textvariable=self.charucoMarkerSavePageBorderYStr).grid(row=1, column=6, sticky = tk.NSEW) + + self.charucoMarkerDictionaryMenue['menu'].delete(0, 'end') + for dictName in self.dictList: + self.charucoMarkerDictionaryMenue['menu'].add_command(label=dictName, command=tk._setit(self.charucoMarkerDictionaryStr, dictName, self.OnSelectCharucoMarkerDictionary)) + + self.OnSelectCharucoMarkerDictionary("DICT_ARUCO_ORIGINAL") + + def OnSelectArucoGridMarkerDictionary(self, pDictName): + self.arucoGridMarkerDictionaryStr.set(pDictName) + + def OnPreviewOrSaveArucoGridMarker(self, askSave = False): + try: + markersX = int(self.arucoGridMarkerMarkersXStr.get()) + markersY = int(self.arucoGridMarkerMarkersYStr.get()) + markerLength = float(self.arucoGridMarkerMarkerLengthStr.get()) + markerSeparation = float(self.arucoGridMarkerMarkerSeparationStr.get()) + borderBits = int(self.arucoGridMarkerBorderBitsStr.get()) + firstMarker = int(self.arucoGridMarkerFirstMarkerStr.get()) + dictionary = self.arucoGridMarkerDictionaryStr.get() + subSizeX = int(self.arucoGridMarkerSaveSubSizeXStr.get()) + subSizeY = int(self.arucoGridMarkerSaveSubSizeYStr.get()) + pageBorderX = float(self.arucoGridMarkerSavePageBorderXStr.get()) + pageBorderY = float(self.arucoGridMarkerSavePageBorderYStr.get()) + except ValueError as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Enter invalid parameters") + return + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Fail to get parameters") + return + + # Preview + try: + dpi=self.VisDPI(((markersY * markerLength + (markersY - 1) * markerSeparation + pageBorderY * 2) * MarkerPrinter.ptPerMeter, (markersX * markerLength + (markersX - 1) * markerSeparation + pageBorderX * 2) * MarkerPrinter.ptPerMeter)) + tkImage = PIL.ImageTk.PhotoImage(image = MarkerPrinter.PreviewArucoGridMarkerImage(dictionary, (markersX, markersY), markerLength, markerSeparation, firstMarker, borderBits=borderBits, pageBorder = (pageBorderX, pageBorderY), dpi=dpi)) + self.arucoGridMarkerImageLabel.imgtk = tkImage + self.arucoGridMarkerImageLabel.config(image=tkImage) + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "create marker failed") + return + + # Save + if(askSave): + MarkerPrinterGUI.__SaveMarker(MarkerPrinter.GenArucoGridMarkerImage, \ + dictionary, (markersX, markersY), markerLength, markerSeparation, firstMarker, borderBits=borderBits, subSize = (subSizeX, subSizeY), pageBorder = (pageBorderX, pageBorderY)) + + def OnPreviewArucoGridMarker(self): + self.OnPreviewOrSaveArucoGridMarker(askSave = False) + + def OnSaveArucoGridMarker(self): + self.OnPreviewOrSaveArucoGridMarker(askSave = True) + + def InitArucoGridMarkerTab(self): + self.arucoGridMarkerUIFrame = ttk.Frame(self.arucoGridMarkerTab) + self.arucoGridMarkerImageTab = ttk.Frame(self.arucoGridMarkerTab) + self.arucoGridMarkerUIFrame2 = ttk.Frame(self.arucoGridMarkerTab) + + self.arucoGridMarkerUIFrame.grid(row=0, column=0, sticky = tk.NSEW) + self.arucoGridMarkerImageTab.grid(row=1, column=0, sticky = tk.NSEW) + self.arucoGridMarkerUIFrame2.grid(row=2, column=0, sticky = tk.NSEW) + + self.arucoGridMarkerImageLabel = tk.Label(self.arucoGridMarkerImageTab) + self.arucoGridMarkerImageLabel.grid(row=0, column=0, sticky = tk.NSEW) + + tk.Label(self.arucoGridMarkerUIFrame, text="dictionary").grid(row=0, column=0, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame, text="markersX").grid(row=0, column=1, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame, text="markersY").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame, text="markerLength (Unit: Meter)").grid(row=0, column=3, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame, text="markerSeparation (Unit: Meter)").grid(row=0, column=4, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame, text="firstMarker").grid(row=0, column=5, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame, text="borderBits").grid(row=0, column=6, sticky = tk.NSEW) + + self.arucoGridMarkerDictionaryStr = tk.StringVar() + self.arucoGridMarkerMarkersXStr = tk.StringVar() + self.arucoGridMarkerMarkersXStr.set("16") + self.arucoGridMarkerMarkersYStr = tk.StringVar() + self.arucoGridMarkerMarkersYStr.set("9") + self.arucoGridMarkerMarkerLengthStr = tk.StringVar() + self.arucoGridMarkerMarkerLengthStr.set("0.07") + self.arucoGridMarkerMarkerSeparationStr = tk.StringVar() + self.arucoGridMarkerMarkerSeparationStr.set("0.02") + self.arucoGridMarkerFirstMarkerStr = tk.StringVar() + self.arucoGridMarkerFirstMarkerStr.set("0") + self.arucoGridMarkerBorderBitsStr = tk.StringVar() + self.arucoGridMarkerBorderBitsStr.set("1") + + self.arucoGridMarkerDictionaryMenue = tk.OptionMenu(self.arucoGridMarkerUIFrame, self.arucoGridMarkerDictionaryStr, "DICT_ARUCO_ORIGINAL", command = self.OnSelectArucoGridMarkerDictionary) + self.arucoGridMarkerDictionaryMenue.grid(row=1, column=0, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame, textvariable=self.arucoGridMarkerMarkersXStr).grid(row=1, column=1, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame, textvariable=self.arucoGridMarkerMarkersYStr).grid(row=1, column=2, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame, textvariable=self.arucoGridMarkerMarkerLengthStr).grid(row=1, column=3, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame, textvariable=self.arucoGridMarkerMarkerSeparationStr).grid(row=1, column=4, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame, textvariable=self.arucoGridMarkerFirstMarkerStr).grid(row=1, column=5, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame, textvariable=self.arucoGridMarkerBorderBitsStr).grid(row=1, column=6, sticky = tk.NSEW) + + tk.Button(self.arucoGridMarkerUIFrame2, text = "Preview", command = self.OnPreviewArucoGridMarker).grid(row=1, column=0, sticky = tk.NSEW) + tk.Button(self.arucoGridMarkerUIFrame2, text = "Save", command = self.OnSaveArucoGridMarker).grid(row=1, column=1, sticky = tk.NSEW) + + tk.Label(self.arucoGridMarkerUIFrame2, text="Save opetions:").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="(set 0 as disable)").grid(row=1, column=2, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="subSizeX").grid(row=0, column=3, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="subSizeY").grid(row=0, column=4, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="Divide to chunks, chunk sizeX").grid(row=2, column=3, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="Divide to chunks, chunk sizeY").grid(row=2, column=4, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="pageBorderX (Unit: Meter)").grid(row=0, column=5, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="pageBorderY (Unit: Meter)").grid(row=0, column=6, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="Border or page").grid(row=2, column=5, sticky = tk.NSEW) + tk.Label(self.arucoGridMarkerUIFrame2, text="Border or page").grid(row=2, column=6, sticky = tk.NSEW) + + self.arucoGridMarkerSaveSubSizeXStr = tk.StringVar() + self.arucoGridMarkerSaveSubSizeXStr.set("0") + self.arucoGridMarkerSaveSubSizeYStr = tk.StringVar() + self.arucoGridMarkerSaveSubSizeYStr.set("0") + self.arucoGridMarkerSavePageBorderXStr = tk.StringVar() + self.arucoGridMarkerSavePageBorderXStr.set("0.02") + self.arucoGridMarkerSavePageBorderYStr = tk.StringVar() + self.arucoGridMarkerSavePageBorderYStr.set("0.02") + + tk.Entry(self.arucoGridMarkerUIFrame2, textvariable=self.arucoGridMarkerSaveSubSizeXStr).grid(row=1, column=3, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame2, textvariable=self.arucoGridMarkerSaveSubSizeYStr).grid(row=1, column=4, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame2, textvariable=self.arucoGridMarkerSavePageBorderXStr).grid(row=1, column=5, sticky = tk.NSEW) + tk.Entry(self.arucoGridMarkerUIFrame2, textvariable=self.arucoGridMarkerSavePageBorderYStr).grid(row=1, column=6, sticky = tk.NSEW) + + self.arucoGridMarkerDictionaryMenue['menu'].delete(0, 'end') + for dictName in self.dictList: + self.arucoGridMarkerDictionaryMenue['menu'].add_command(label=dictName, command=tk._setit(self.arucoGridMarkerDictionaryStr, dictName, self.OnSelectArucoGridMarkerDictionary)) + + self.OnSelectArucoGridMarkerDictionary("DICT_ARUCO_ORIGINAL") + + def OnSelectArucoMarkerDictionary(self, pDictName): + self.arucoMarkerDictionaryStr.set(pDictName) + + def OnPreviewOrSaveArucoMarker(self, askSave = False): + try: + markerID = int(self.arucoMarkerMarkerIDStr.get()) + markerLength = float(self.arucoMarkerMarkerLengthStr.get()) + borderBits = int(self.arucoMarkerBorderBitsStr.get()) + dictionary = self.arucoMarkerDictionaryStr.get() + pageBorderX = float(self.arucoMarkerSavePageBorderXStr.get()) + pageBorderY = float(self.arucoMarkerSavePageBorderYStr.get()) + except ValueError as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Enter invalid parameters") + return + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Fail to get parameters") + return + + # Preview + try: + dpi=self.VisDPI(((markerLength + pageBorderY * 2) * MarkerPrinter.ptPerMeter, (markerLength + pageBorderX * 2) * MarkerPrinter.ptPerMeter)) + tkImage = PIL.ImageTk.PhotoImage(image = MarkerPrinter.PreviewArucoMarkerImage(dictionary, markerID, markerLength, borderBits=borderBits, pageBorder = (pageBorderX, pageBorderY), dpi=dpi)) + self.arucoMarkerImageLabel.imgtk = tkImage + self.arucoMarkerImageLabel.config(image=tkImage) + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "create marker failed") + return + + # Save + if(askSave): + MarkerPrinterGUI.__SaveMarker(MarkerPrinter.GenArucoMarkerImage, \ + dictionary, markerID, markerLength, borderBits=borderBits, pageBorder = (pageBorderX, pageBorderY)) + + def OnPreviewArucoMarker(self): + self.OnPreviewOrSaveArucoMarker(askSave = False) + + def OnSaveArucoMarker(self): + self.OnPreviewOrSaveArucoMarker(askSave = True) + + def InitArucoMarkerTab(self): + self.arucoMarkerUIFrame = ttk.Frame(self.arucoMarkerTab) + self.arucoMarkerImageTab = ttk.Frame(self.arucoMarkerTab) + self.arucoMarkerUIFrame2 = ttk.Frame(self.arucoMarkerTab) + + self.arucoMarkerUIFrame.grid(row=0, column=0, sticky = tk.NSEW) + self.arucoMarkerImageTab.grid(row=1, column=0, sticky = tk.NSEW) + self.arucoMarkerUIFrame2.grid(row=2, column=0, sticky = tk.NSEW) + + self.arucoMarkerImageLabel = tk.Label(self.arucoMarkerImageTab) + self.arucoMarkerImageLabel.grid(row=0, column=0, sticky = tk.NSEW) + + tk.Label(self.arucoMarkerUIFrame, text="dictionary").grid(row=0, column=0, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame, text="markerID").grid(row=0, column=1, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame, text="markerLength (Unit: Meter)").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame, text="borderBits").grid(row=0, column=3, sticky = tk.NSEW) + + self.arucoMarkerDictionaryStr = tk.StringVar() + self.arucoMarkerMarkerIDStr = tk.StringVar() + self.arucoMarkerMarkerIDStr.set("0") + self.arucoMarkerMarkerLengthStr = tk.StringVar() + self.arucoMarkerMarkerLengthStr.set("0.07") + self.arucoMarkerBorderBitsStr = tk.StringVar() + self.arucoMarkerBorderBitsStr.set("1") + + self.arucoMarkerDictionaryMenue = tk.OptionMenu(self.arucoMarkerUIFrame, self.arucoMarkerDictionaryStr, "DICT_ARUCO_ORIGINAL", command = self.OnSelectArucoMarkerDictionary) + self.arucoMarkerDictionaryMenue.grid(row=1, column=0, sticky = tk.NSEW) + tk.Entry(self.arucoMarkerUIFrame, textvariable=self.arucoMarkerMarkerIDStr).grid(row=1, column=1, sticky = tk.NSEW) + tk.Entry(self.arucoMarkerUIFrame, textvariable=self.arucoMarkerMarkerLengthStr).grid(row=1, column=2, sticky = tk.NSEW) + tk.Entry(self.arucoMarkerUIFrame, textvariable=self.arucoMarkerBorderBitsStr).grid(row=1, column=3, sticky = tk.NSEW) + + tk.Button(self.arucoMarkerUIFrame2, text = "Preview", command = self.OnPreviewArucoMarker).grid(row=0, column=0, sticky = tk.NSEW) + tk.Button(self.arucoMarkerUIFrame2, text = "Save", command = self.OnSaveArucoMarker).grid(row=0, column=1, sticky = tk.NSEW) + + tk.Label(self.arucoMarkerUIFrame2, text="Save opetions:").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame2, text="(set 0 as disable)").grid(row=1, column=2, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame2, text="pageBorderX (Unit: Meter)").grid(row=0, column=3, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame2, text="pageBorderY (Unit: Meter)").grid(row=0, column=4, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame2, text="Border or page").grid(row=2, column=3, sticky = tk.NSEW) + tk.Label(self.arucoMarkerUIFrame2, text="Border or page").grid(row=2, column=4, sticky = tk.NSEW) + + self.arucoMarkerSavePageBorderXStr = tk.StringVar() + self.arucoMarkerSavePageBorderXStr.set("0.02") + self.arucoMarkerSavePageBorderYStr = tk.StringVar() + self.arucoMarkerSavePageBorderYStr.set("0.02") + + tk.Entry(self.arucoMarkerUIFrame2, textvariable=self.arucoMarkerSavePageBorderXStr).grid(row=1, column=3, sticky = tk.NSEW) + tk.Entry(self.arucoMarkerUIFrame2, textvariable=self.arucoMarkerSavePageBorderYStr).grid(row=1, column=4, sticky = tk.NSEW) + + self.arucoMarkerDictionaryMenue['menu'].delete(0, 'end') + for dictName in self.dictList: + self.arucoMarkerDictionaryMenue['menu'].add_command(label=dictName, command=tk._setit(self.arucoMarkerDictionaryStr, dictName, self.OnSelectArucoMarkerDictionary)) + + self.OnSelectArucoMarkerDictionary("DICT_ARUCO_ORIGINAL") + + def OnPreviewOrSaveChessMarker(self, askSave = False): + try: + sizeX = int(self.chessMarkerChessboardSizeXStr.get()) + sizeY = int(self.chessMarkerChessboardSizeYStr.get()) + squareLength = float(self.chessMarkerSquareLengthStr.get()) + subSizeX = int(self.chessMarkerSaveSubSizeXStr.get()) + subSizeY = int(self.chessMarkerSaveSubSizeYStr.get()) + pageBorderX = float(self.chessMarkerSavePageBorderXStr.get()) + pageBorderY = float(self.chessMarkerSavePageBorderYStr.get()) + except ValueError as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Enter invalid parameters") + return + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "Fail to get parameters") + return + + # Preview + try: + dpi=self.VisDPI(((sizeY * squareLength + pageBorderY * 2) * MarkerPrinter.ptPerMeter, (sizeX * squareLength + pageBorderX * 2) * MarkerPrinter.ptPerMeter)) + tkImage = PIL.ImageTk.PhotoImage(image = MarkerPrinter.PreviewChessMarkerImage((sizeX, sizeY), squareLength, pageBorder = (pageBorderX, pageBorderY), dpi=dpi)) + self.chessMarkerImageLabel.imgtk = tkImage + self.chessMarkerImageLabel.config(image=tkImage) + except Exception as e: + warnings.warn(str(e)) + messagebox.showinfo("Error", "create marker failed") + return + + # Save + if(askSave): + MarkerPrinterGUI.__SaveMarker(MarkerPrinter.GenChessMarkerImage, \ + (sizeX, sizeY), squareLength, subSize = (subSizeX, subSizeY), pageBorder = (pageBorderX, pageBorderY)) + + def OnPreviewChessMarker(self): + self.OnPreviewOrSaveChessMarker(askSave = False) + + def OnSaveChessMarker(self): + self.OnPreviewOrSaveChessMarker(askSave = True) + + def InitChessMarkerTab(self): + self.chessMarkerUIFrame = ttk.Frame(self.chessMarkerTab) + self.chessMarkerImageTab = ttk.Frame(self.chessMarkerTab) + self.chessMarkerUIFrame2 = ttk.Frame(self.chessMarkerTab) + + self.chessMarkerUIFrame.grid(row=0, column=0, sticky = tk.NSEW) + self.chessMarkerImageTab.grid(row=1, column=0, sticky = tk.NSEW) + self.chessMarkerUIFrame2.grid(row=2, column=0, sticky = tk.NSEW) + + self.chessMarkerImageLabel = tk.Label(self.chessMarkerImageTab) + self.chessMarkerImageLabel.grid(row=0, column=0, sticky = tk.NSEW) + + tk.Label(self.chessMarkerUIFrame, text="chessboardSizeX").grid(row=0, column=0, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame, text="chessboardSizeY").grid(row=0, column=1, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame, text="squareLength (Unit: Meter)").grid(row=0, column=2, sticky = tk.NSEW) + + self.chessMarkerChessboardSizeXStr = tk.StringVar() + self.chessMarkerChessboardSizeXStr.set("16") + self.chessMarkerChessboardSizeYStr = tk.StringVar() + self.chessMarkerChessboardSizeYStr.set("9") + self.chessMarkerSquareLengthStr = tk.StringVar() + self.chessMarkerSquareLengthStr.set("0.09") + + tk.Entry(self.chessMarkerUIFrame, textvariable=self.chessMarkerChessboardSizeXStr).grid(row=1, column=0, sticky = tk.NSEW) + tk.Entry(self.chessMarkerUIFrame, textvariable=self.chessMarkerChessboardSizeYStr).grid(row=1, column=1, sticky = tk.NSEW) + tk.Entry(self.chessMarkerUIFrame, textvariable=self.chessMarkerSquareLengthStr).grid(row=1, column=2, sticky = tk.NSEW) + + tk.Button(self.chessMarkerUIFrame2, text = "Preview", command = self.OnPreviewChessMarker).grid(row=1, column=0, sticky = tk.NSEW) + tk.Button(self.chessMarkerUIFrame2, text = "Save", command = self.OnSaveChessMarker).grid(row=1, column=1, sticky = tk.NSEW) + + tk.Label(self.chessMarkerUIFrame2, text="Save opetions:").grid(row=0, column=2, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="(set 0 as disable)").grid(row=1, column=2, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="subSizeX").grid(row=0, column=3, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="subSizeY").grid(row=0, column=4, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="Divide to chunks, chunk sizeX").grid(row=2, column=3, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="Divide to chunks, chunk sizeY").grid(row=2, column=4, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="pageBorderX (Unit: Meter)").grid(row=0, column=5, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="pageBorderY (Unit: Meter)").grid(row=0, column=6, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="Border or page").grid(row=2, column=5, sticky = tk.NSEW) + tk.Label(self.chessMarkerUIFrame2, text="Border or page").grid(row=2, column=6, sticky = tk.NSEW) + + self.chessMarkerSaveSubSizeXStr = tk.StringVar() + self.chessMarkerSaveSubSizeXStr.set("0") + self.chessMarkerSaveSubSizeYStr = tk.StringVar() + self.chessMarkerSaveSubSizeYStr.set("0") + self.chessMarkerSavePageBorderXStr = tk.StringVar() + self.chessMarkerSavePageBorderXStr.set("0.02") + self.chessMarkerSavePageBorderYStr = tk.StringVar() + self.chessMarkerSavePageBorderYStr.set("0.02") + + tk.Entry(self.chessMarkerUIFrame2, textvariable=self.chessMarkerSaveSubSizeXStr).grid(row=1, column=3, sticky = tk.NSEW) + tk.Entry(self.chessMarkerUIFrame2, textvariable=self.chessMarkerSaveSubSizeYStr).grid(row=1, column=4, sticky = tk.NSEW) + tk.Entry(self.chessMarkerUIFrame2, textvariable=self.chessMarkerSavePageBorderXStr).grid(row=1, column=5, sticky = tk.NSEW) + tk.Entry(self.chessMarkerUIFrame2, textvariable=self.chessMarkerSavePageBorderYStr).grid(row=1, column=6, sticky = tk.NSEW) + + def Update(self): + time.sleep(0) + self.window.after(self.delay, self.Update) + + def __init__(self, pDelay=15, pDisplayShape=(int(400), int(1200))): + self.delay = pDelay + self.displayShape = pDisplayShape + + self.dictList = MarkerPrinter.arucoDictBytesList.keys() + + # GUI + self.window = tk.Tk() + self.notebook = ttk.Notebook(self.window) + self.notebook.grid(row=0, column=0, sticky = tk.NSEW) + + self.window.title("MarkerPrinterGUI") + self.window.config(cursor="arrow") + self.window.protocol("WM_DELETE_WINDOW", self.OnCloseWindow) + + # Menues + self.menu = tk.Menu(self.window) + self.helpMenu = tk.Menu(self.menu, tearoff=0) + self.menu.add_cascade(label="Help", menu=self.helpMenu) + self.helpMenu.add_command(label="Github", command=self.OnShowingHelpGithub) + self.helpMenu.add_command(label="DEBUG_LINE_MODE", command=self.On_DEBUG_LINE_MODE) + self.helpMenu.add_command(label="DEBUG_BLOCK_MODE", command=self.On_DEBUG_BLOCK_MODE) + self.helpMenu.add_command(label="CLOSE_DEBUG_MODE", command=self.On_CLOSE_DEBUG_MODE) + self.window.config(menu=self.menu) + + self.charucoMarkerTab = ttk.Frame(self.notebook) + self.arucoMarkerTab = ttk.Frame(self.notebook) + self.arucoGridMarkerTab = ttk.Frame(self.notebook) + self.chessMarkerTab = ttk.Frame(self.notebook) + + self.notebook.add(self.charucoMarkerTab, text='ChArUco Marker') + self.notebook.add(self.arucoMarkerTab, text='ArUco Marker') + self.notebook.add(self.arucoGridMarkerTab, text='ArUcoGrid Marker') + self.notebook.add(self.chessMarkerTab, text='Chessboard Marker') + + self.InitCharucoMarkerTab() + self.InitArucoMarkerTab() + self.InitArucoGridMarkerTab() + self.InitChessMarkerTab() + + self.Update() + self.window.mainloop() + + def On_DEBUG_LINE_MODE(self): + messagebox.showinfo("Note", "You enabled the debug mode: \"LINE\"") + MarkerPrinter.debugMode = "LINE" + + def On_DEBUG_BLOCK_MODE(self): + messagebox.showinfo("Note", "You enabled the debug mode: \"BLOCK\"") + MarkerPrinter.debugMode = "BLOCK" + + def On_CLOSE_DEBUG_MODE(self): + messagebox.showinfo("Note", "You closed the debug mode") + MarkerPrinter.debugMode = None + +if __name__ == '__main__': + MarkerPrinterGUI() diff --git a/modules/aruco/misc/pattern_generator/README.md b/modules/aruco/misc/pattern_generator/README.md new file mode 100644 index 000000000..994dc9311 --- /dev/null +++ b/modules/aruco/misc/pattern_generator/README.md @@ -0,0 +1,68 @@ +# OpenCVMarkerPrinter + +## Description +This small app can save some commonly used opencv markers such as ArUco, ArUcoGrid, Chessboard and ChArUco to vector graphics file. **Supported vector graphics file format: .svg, .pdf and .ps.** + + + +### Dependencies +#### MarkerPrinter + * numpy + * PIL(Pillow, for image processing) + * cairo(for drawing vector graphic) + * cairosvg(for svg to png) + +#### MarkerPrinterGUI + * tkinter(for GUI) + +## Tutorial +#### GUI +``` +python MarkerPrinterGUI.py +``` + +You can switch ArUco, ArUcoGrid, Chessboard and ChArUco mode at the GUI tab, then you can select dictionary from the GUI menu and modify board shape, marker size, border width... etc. at the GUI entry, finally click the preview or save button to show the marker image on the GUI window or save it to file. + +#### Command-Line +##### Print help +``` +python MarkerPrinter.py +``` + +##### Print predefined dictionary list +``` +python MarkerPrinter.py --list_dictionary +``` + +##### Save chessboard +``` +python MarkerPrinter.py --chess --file "./chess.pdf" --size_x 16 --size_y 9 --square_length 0.09 +``` + +##### Save ArUco +``` +python MarkerPrinter.py --aruco --file "./aruco.pdf" --dictionary DICT_ARUCO_ORIGINAL --marker_length 0.07 --marker_id 0 --border_bits 1 +``` + +##### Save ArUco Grid +``` +python MarkerPrinter.py --aruco_grid --file "./aruco_grid.pdf" --dictionary DICT_ARUCO_ORIGINAL --size_x 16 --size_y 9 --marker_length 0.07 --marker_separation 0.02 --first_marker 0 --border_bits 1 +``` + +##### Save ChArUco +``` +python MarkerPrinter.py --charuco --file "./charuco.pdf" --dictionary DICT_ARUCO_ORIGINAL --size_x 16 --size_y 9 --square_length 0.09 --marker_length 0.07 --border_bits 1 +``` + +## Useful Options: +### Divde output to chunks +If you are using consumer level printer, you will suffer from not able printing too large marker, so just set chunks shape at the GUI subSize entry before saving the marker to files, it will divide output marker to chunks. If you are using command-line interface, just add --sub_size_x x --sub_size_y y as parameters. + +### Page border +If you are printing the image directly, you will need add page border to protect the marker, so just set page border at the GUI pageBorder entry before saving the marker to files. If you are using command-line interface, just add --page_border_x x --page_border_y y as parameters. + +### Generate aruco data: +Although there is a built-in aruco dictionary data, but if you want to update the dictionary(If aruco update predefined dictionary list), just install opencv-python and opencv-contrib-python, and than run +``` +python MarkerPrinter.py --generate arucoDictBytesList.npz +``` diff --git a/modules/aruco/misc/pattern_generator/arucoDictBytesList.npz b/modules/aruco/misc/pattern_generator/arucoDictBytesList.npz new file mode 100644 index 000000000..64bcbc96c Binary files /dev/null and b/modules/aruco/misc/pattern_generator/arucoDictBytesList.npz differ diff --git a/modules/aruco/misc/pattern_generator/doc/images/MarkerPrinterGUI.jpg b/modules/aruco/misc/pattern_generator/doc/images/MarkerPrinterGUI.jpg new file mode 100644 index 000000000..4aed27556 Binary files /dev/null and b/modules/aruco/misc/pattern_generator/doc/images/MarkerPrinterGUI.jpg differ diff --git a/modules/aruco/tutorials/table_of_content_aruco.markdown b/modules/aruco/tutorials/table_of_content_aruco.markdown index ff52383c1..7a0536b6b 100644 --- a/modules/aruco/tutorials/table_of_content_aruco.markdown +++ b/modules/aruco/tutorials/table_of_content_aruco.markdown @@ -11,6 +11,11 @@ Also, the ChArUco functionalities combine ArUco markers with traditional chessbo an easy and versatile corner detection. The module also includes the functions to detect ChArUco corners and use them for pose estimation and camera calibration. +If you are going to print out the markers, an useful script/GUI tool is place at +opencv_contrib/modules/aruco/misc/pattern_generator/ that can generate vector graphics +of ArUco, ArUcoGrid and ChArUco boards. It can help you to print out the pattern with real size +and without artifacts. + - @subpage tutorial_aruco_detection *Compatibility:* \> OpenCV 3.0