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]

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

    ]

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

    svgFile = open(destFilePath, "w")

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

    svgFile.close()

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)