blocks.py

# This should run in python 3. I hope. It does for me.
# SCRIPT BY WNP78. Post at https://www.simpleplanes.com/Forums/View/597333/Pixel-art-script
# This is to show people how to generate crafts with scripts, in the hope that you can use it for something better. It is *NOT* just for people to download, and upload random pictures with!


import xml.etree.ElementTree as et
import tkinter,sys,os,time
class XMLCraft:
    def __init__(self, name, theme=None):
        self.root = et.Element("Aircraft",{"name":name,"url":rlu,"theme":"Custom","xmlVersion":"4"})
        self.Assembly = et.SubElement(self.root,"Assembly")
        if theme == None:
            self.Theme = et.fromstring("""<Theme name="Custom"> <Material color="FFFFFF" r="0.15" m="0.5" s="0.7" /> </Theme>""")
        else:
            self.Theme = et.SubElement(self.root, "Theme", {"name":"Custom"})
            for col in theme:
                et.SubElement(self.Theme, "Material", {"color":col, "r":"0", "m":"0.65", "s":"0.8"})
        self.root.append(self.Theme)
        self.Parts = et.SubElement(self.Assembly,"Parts")
        self.Bodies = et.SubElement(self.Assembly, "Bodies")
        self.Connections = et.SubElement(self.Assembly, "Connections")
        self.body = et.SubElement(self.Bodies,"Body",{"position":"0,0,0", "rotation":"0,0,0", "velocity":"0,0,0", "angularVelocity":"0,0,0",})
        self.rparts = {"1":et.fromstring("""<Part id="1" partType="Cockpit-1" position="0,0,1" rotation="0,0,0" drag="0,0,0,0,0,0" materials="7,0"><Cockpit.State primaryCockpit="True" /></Part>""")}
        self._cid = 1
        self.dps = []
    def cid(self):
        self._cid += 1
        return self._cid
    def addBlock(self,x,y,z,mat=1,connectIds=[[1,0,0]]):
        if (x,y,z) in self.dps: return
        self.dps.append((x,y,z))
        mid = str(self.cid())
        self.rparts[mid] = et.XML('<Part id="'+mid+'" partType="Block-2" position="'+str(x) + "," + str(y) + "," + str(z)+'" rotation="0,0,0" drag="0.008,0.008,0.254,0.254,0.254,0.254" materials="'+str(mat)+'" scale="1,1,1" />')
        for cp in connectIds:
            et.SubElement(self.Connections, "Connection", {
            "partA":str(cp[0]),
            "partB":mid,
            "attachPointsA":str(cp[2]),
            "attachPointsB":str(cp[1])
            })
        return mid
    def save(self,path):
        for p in self.rparts: self.Parts.append(self.rparts[p])
        self.body.set("partIds", ",".join(self.rparts.keys()))
        with open(path, "wb") as f:et.ElementTree(self.root).write(f)
class Utils:
    @staticmethod
    def readimg(path):
        global rlu
        rlu = "vR9" + "ERl"
        t = tkinter.Tk()
        img = tkinter.PhotoImage(None, {"file":path}, t)
        data = []
        for y in range(img.height()):
            c = []
            for x in range(img.width()):
                c.append(list(img.get(x,y)))
            data.append(c)
        t.destroy()
        return data
    @staticmethod
    def hexcol(col):
        res = ""
        for i in col:
            r = hex(i)[2:].upper()
            if len(r)==1: r = "0"+r
            res += r
        return res
    @staticmethod
    def tryFindSpDir():
        if sys.platform.startswith("win"):
            p = os.path.abspath(os.path.join(os.getenv("APPDATA"),"..\\LocalLow\Jundroo\SimplePlanes\AircraftDesigns\Image.xml"))
            if os.path.isdir("\\".join(  p.split("\\")[:-1] )):
                return p, "Saved to simpleplanes files as \"Image\""
            return "", "SP save directory not found, move file \"Image.xml\" from the folder this script is in to your simpleplanes aircraftdesigns folder."
        return "", "Non-windows platform detected. Move file \"Image.xml\" from the folder this script is in to your simpleplanes aircraftdesigns folder."

print("Image to SimplePlane v1.0 by WNP78. Please note that if the image has too many different colours in it (>250? roughly.) the game can take it badly and glitch out. If this happens, use the posterise function on an image editing suite such as paint.net (free), until it works. You'll probably also want to decrease the resolution a bit, which you can also do in image editing software.")
inPath = input("Enter the file path to an image (png)> ")

time.clock()
print("Reading image...")
pixels = Utils.readimg(inPath)
print("Processing image...")
theme = []
mats = []
for row in pixels:
    tr = []
    for p in row:
        px = Utils.hexcol(p)
        if px in theme:
            tr.append(theme.index(px))
        else:
            tr.append(len(theme))
            theme.append(px)
    mats.append(tr)
print("Different colours:",len(theme))
print("Part count:",(len(mats) * len(mats[0]))+1)
print("Constructing craft...")
craft = XMLCraft("Image", theme)
lId = 1
for y,row in enumerate(mats):
    for x,px in enumerate(row):
        connectIds = []
        if x != 0: connectIds = [[lId, 3, 2]]
        if y != 0: connectIds.append([lId - (len(row)-1), 5, 4])
        lId = int(craft.addBlock((len(row)-x)/2,(len(mats)-y)/2,0,px, connectIds=connectIds))

print("Saving craft...")
path,msg = Utils.tryFindSpDir()
path = os.path.abspath(path)
craft.save(path)
print(msg)
endT = time.time()
print("Done in",round(time.clock()),"second(s).")
print("You may need to attach your cockpit to the blocks, as they are probably detached.")
input("Press enter to exit")