PES-Embroidery/src/main.py

import argparse
import sys
from svgFunctions import *
import pyembroidery
from utilities import *


parser = argparse.ArgumentParser(description="Converts an SVG file into a PES embroidery file.")

parser.add_argument("-i", dest="inputFile", type=str, action='store', required=True)
parser.add_argument("-o", dest="outputFile", type=str, action='store', default="output.PES", required=False)
parser.add_argument("-d", dest="debug", action='store_true')

args = parser.parse_args()

# Load the SVG file from disk
svg = loadVectorGraphic(args.inputFile)
paths, attributes = svg
if paths is None or attributes is None:
    sys.exit(0)

# Enumerate the shapes in the SVG to find where stitches should go.
for shape in paths:
    stitchLines = makeStitchLines(shape)
    print(shape)

if args.debug:
    pes = pyembroidery.read(args.outputFile)
    if pes is not None:
        print("Generating debug image.")
        pyembroidery.write_png(pes, replaceFilenameAndExtensionFromPath(args.inputFile, "debugPicture", "png"))
    else:
        print("Couldn't find output file.")