Salva la produzione di tartaruga come JPEG

Question

Ho un creatore di immagini frattali.Crea un albero frattale casuale come una cosa.Al termine, chiede all'utente di salvare l'albero.Ho salvato come un .svg in questo momento e che funziona, ma voglio che salvi in un tipo di file più conveniente, come JPEG.Qualche idea? Codice:

import turtle
import random
from sys import exit
from time import clock
import canvasvg
turtle.colormode(255)
red = 125
green = 70
blue = 38        
pen = 10
def saveImg():
    print("Done.")
    save = input("Would you like to save this tree? Y/N \n")
    if save.upper() == "Y":
        t.hideturtle()
        name = input("What would you like to name it? \n")
        nameSav = name + ".svg"
        ts = turtle.getscreen().getcanvas()
        canvasvg.saveall(nameSav, ts)
    elif save.upper() == "N":
        def runChk():
            runAgain = input("Would you like to run again? Y/N (N will exit)")
            if runAgain.upper() == "Y":
                print("Running")
                main()
            elif runAgain.upper() == "N":
                print("Exiting...")
                exit()
            else:
                print("Invalid response.")
                runChk()
        runChk()
    else:
        print("Invalid response.")
        saveImg()

def tree(branchLen, t, red, green, blue, pen):
    time = str(round(clock()))
    print("Drawing... " + time)
    if branchLen > 3:
        pen = pen*0.8
        t.pensize(pen)
        if (red > 10 and green < 140):
            red = red - 15
            green = green + 8
    if branchLen > 5:
        angle = random.randrange(18, 55)
        angleTwo = 0.5*angle
        sub = random.randrange(1,16)
        t.color(red, green, blue)
        t.forward(branchLen)
        t.right(angleTwo)
        tree(branchLen-sub,t, red, green, blue, pen)
        t.left(angle)
        tree(branchLen-sub, t, red, green, blue, pen)
        t.right(angleTwo)
        t.backward(branchLen)

def main():
    t = turtle.Turtle()
    myWin = turtle.Screen()
    t.speed(0)
    t.hideturtle()
    t.left(90)
    t.up()
    t.backward(100)
    t.down()
    print("Please wait while I draw...")
    tree(random.randrange(60,95),t,red,green,blue, pen)
    saveImg()
main()

.

Answer 1

Deve essere un JPEG?Png è sufficiente?

Se è così possibile convertire da svg a png usando cairosvg .Sfortunatamente canvasvg.saveall() consente solo di trasmetterlo un nome di file in cui scriverà lo SVG, quindi è necessario utilizzare un file temporaneo per SVG e quindi convertire il file TEMP in PNG utilizzando cairosvg.svg2png().Quindi qualcosa di simile dovrebbe fare il lavoro:

import os
import shutil
import tempfile

import canvasvg

name = raw_input("What would you like to name it? \n")
nameSav = name + ".png"
tmpdir = tempfile.mkdtemp()  # create a temporary directory
tmpfile = os.path.join(tmpdir, 'tmp.svg')  # name of file to save SVG to
ts = turtle.getscreen().getcanvas()
canvasvg.saveall(tmpfile, ts)
with open(tmpfile) as svg_input, open(nameSav, 'wb') as png_output:
    cairosvg.svg2png(bytestring=svg_input.read(), write_to=png_output)
shutil.rmtree(tmpdir)  # clean up temp file(s)

.

Se ti è piaciuto, è possibile scrivere la tua funzione saveall() basata su canvasvg.saveall() (è piuttosto piccolo) che accetta un oggetto simile a un file invece di un nome file e scrive a quell'oggetto.Quindi puoi passare in un oggetto StringIO e non dover disturbare il file temporaneo.O il tuo saveall() potrebbe semplicemente restituire i dati SVG come una stringa di Byte.