HYTOS

#!/usr/bin/env/python3

# coding: utf-8

import numpy as np, cv2, subprocess

import os, sys

from xml.dom import minidom

import svg.path

from xml.etree.ElementTree import Element, SubElement, dump, ElementTree

import SymbolBase

from shapely.geometry import LineString

from shapely.ops import linemerge

from LineDetector import LineDetector

# potrace command

POTRACE = os.path.dirname(os.path.realpath(__file__)) + '\\potrace.exe'

ADJUST = 10

def convertImageToSvg(imgFilePath, destFilePath):

    thresh = 127

    image  = cv2.imread(imgFilePath, cv2.IMREAD_GRAYSCALE)

    threshold = cv2.threshold(image, thresh, 255, cv2.THRESH_BINARY)[1]

    kernel = np.ones((3, 3), np.uint8)

    erode = cv2.dilate(threshold, kernel, iterations=1)

    # convert to bmp binary so that potrace can handle it

    retval, buf = cv2.imencode('.bmp', erode)

    if retval == False:

        raise ValueError('Failed to convert into BMP binary data')

    # convert buf from numpy.ndarray to bytes

    binbmp = buf.tobytes()

    args = [

        POTRACE,

        '-', '-o-', '--svg'

    ]

    p = subprocess.Popen(

        args,

        stdin=subprocess.PIPE,

        stdout=subprocess.PIPE,

        stderr=subprocess.PIPE,

        shell=False

    )

    stdout, stderr = p.communicate(input=binbmp)

    if len(stderr) != 0:

        raise RuntimeError('Potrace threw error:\n' + stderr.decode('utf-8'))

    svgData = stdout

    #svgData = stdout.decode("utf-8")

    #svgData = stdout.decode("utf-16")

    #svgData = svgData.replace(u"\u000B", u"")

    #svgData = svgData.encode("utf-8")

    xmlFile = open(destFilePath, "wb")#, encoding = "utf-8")

    xmlFile.write(svgData)

    xmlFile.close()

# extract line from image by using potrace

def passpotrace(imgFile):

    thresh = 127

    image  = cv2.imread(imgFile, cv2.IMREAD_GRAYSCALE)

    threshold = cv2.threshold(image, thresh, 255, cv2.THRESH_BINARY)[1]

    kernel = np.ones((3, 3), np.uint8)

    erode = cv2.dilate(threshold, kernel, iterations=1)

    # convert to bmp binary so that potrace can handle it

    retval, buf = cv2.imencode('.bmp', erode)

    if retval == False:

        raise ValueError('Failed to convert into BMP binary data')

    # convert buf from numpy.ndarray to bytes

    binbmp = buf.tobytes()

    args = [

        POTRACE,

        '-', '-o-', '--svg'

    ]

    p = subprocess.Popen(

        args,

        stdin=subprocess.PIPE,

        stdout=subprocess.PIPE,

        stderr=subprocess.PIPE,

        shell=False

        )

    stdout, stderr = p.communicate(input=binbmp)

    if len(stderr) != 0:

        raise RuntimeError('Potrace threw error:\n' + stderr.decode('utf-8'))

    imgLines = []

    # parse svg data and extract lines

    svgdom = minidom.parseString(stdout)

    transform = svgdom.getElementsByTagName('g')[0].getAttribute('transform')

    translate = transform.split(' ')[0].split('translate')[1].split(',')

    dx = float(translate[0][1:])

    dy = float(translate[1][:-1])

    scale = transform.split(' ')[1].split('scale')[1].split(',')

    sx = float(scale[0][1:])

    sy = float(scale[1][:-1])

    pathstrings = [path.getAttribute('d') for path in svgdom.getElementsByTagName('path')]

    for pathstring in pathstrings:

        pathdata = svg.path.parse_path(pathstring)

        for segment in pathdata:

            if type(segment) is svg.path.Line:

                line = LineString([(segment.start.real / ADJUST, segment.start.imag / ADJUST), (segment.end.real / ADJUST, segment.end.imag / ADJUST)])

                imgLines.append(line)

    # up to here

    return imgLines

if __name__ == '__main__':

    execpath = os.path.dirname(os.path.realpath(__file__))

    path = execpath + '\\res\\UY1-K-2007_P1_300dpi_black.png'

    #lines = linedraw.sketch(path)

    #linedraw.visualize(lines)

#    path = sys.argv[1]  # 1 argument given is a string for the path of drawing

    #imgLines = passpotrace(path)    #execpath + '\\res\\UY1-K-2007_P1_300dpi_black.png')

    image  = cv2.imread(path, cv2.IMREAD_GRAYSCALE)

    lineDetector = LineDetector(image)

    lineDetector.Detect()

    try:

        xmlFilePath = os.path.dirname(path) + '\\' + os.path.basename(path).split('.')[0] + '.xml'

        file = open(xmlFilePath, 'w')

        for line in imgLines:

            file.write(str(line) + '\n')

        file.close()

    except Exception as ex:

        print('에러가 발생했습니다.\n', ex)