HYTOS

#!/usr/bin/env/python3

# coding: utf-8

import re

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 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.dirname(os.path.realpath(__file__)) + '\\potrace.exe'

ADJUST = 10

def convertImageToSvg(imgFilePath, destFilePath, normalColor, hoverColor):

    thresh = 127

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

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

    # convert to bmp binary so that potrace can handle it

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

    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'

    ]

    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 '''

        svgFile = open(destFilePath, 'w')

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

        svgFile.close()

    except Exception as ex:

        from App import App

        from AppDocData import MessageType

        message = 'error occured({}) 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)