HYTOS

#!/usr/bin/env/python3

# coding: utf-8

import re

import numpy as np, cv2, subprocess

import os, sys

from xml.dom import minidom

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

import xml.etree.ElementTree as ET

import SymbolBase

from shapely.geometry import LineString

from shapely.ops import linemerge

from LineDetector import LineDetector

# potrace command

POTRACE = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'potrace.exe')

ADJUST = 10

def convertImageToSvg(imgFilePath, destFilePath, normalColor, hoverColor, threshold=200):

    """convert image to svg file by using potrace"""

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

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

    # convert to bmp binary so that potrace can handle it

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

    if not retval:

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

    # convert buf from numpy.ndarray to bytes

    binbmp = buf.tobytes()

    args = [

        POTRACE,

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

    ]

    CREATE_NO_WINDOW = 0x08000000

    p = subprocess.Popen(

        args,

        stdin=subprocess.PIPE,

        stdout=subprocess.PIPE,

        stderr=subprocess.PIPE,

        creationflags=CREATE_NO_WINDOW

    )

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

    if len(stderr) != 0:

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

    svgFile = open(destFilePath, "w")

    svgFile.write(stdout.decode("utf-8"))

    svgFile.close()

    ''' add lineargradient nodes to svg file '''

    try:

        doc = ET.parse(destFilePath)

        svg = doc.getroot()

        matches = [child for child in svg.getchildren() if re.search(re.compile('^\{.+\}g$'), child.tag)]

        if matches:

            matches[0].attrib['fill'] = 'url(#normal)'

            matches[0].attrib['stroke'] = 'url(#normal)'

            pathes = [child for child in matches[0].getchildren() if re.search(re.compile('^\{.+\}path$'), child.tag)]

            for path in pathes:

                path.attrib['stroke-width'] = '50'

        defs = Element('defs')

        gradient = ET.fromstring(normalColor)

        gradient.attrib['id'] = 'normal'

        defs.append(gradient)

        gradient = ET.fromstring(hoverColor)

        gradient.attrib['id'] = 'hover'

        defs.append(gradient)

        svg.append(defs)

        ''' remove ns0 from xml '''

        with open(destFilePath, 'w') as svgFile:

            svgFile.write(ET.tostring(svg, encoding='utf-8').decode('utf-8').replace('ns0:', '').replace(':ns0', ''))

    except Exception as ex:

        from App import App

        from AppDocData import MessageType

        message = 'error occurred({}) in {}:{}'.format(ex, sys.exc_info()[-1].tb_frame.f_code.co_filename, sys.exc_info()[-1].tb_lineno)

        App.mainWnd().addMessage.emit(MessageType.Error, message)

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)