Arch: Support for per-face colors in OfflineRenderingUtils

src/Mod/Arch/OfflineRenderingUtils.py

@@ -236,6 +236,45 @@ def getGuiData(filename):
 
 
 
+def saveDiffuseColor(colorlist):
+
+    """saveDiffuseColor(colorlist): Saves the given list or tuple of
+    color tuples to a temp file, suitable to include in a DiffuseColor
+    property. Returns the path to the created temp file"""
+
+    def tochr(i):
+        #print("tochr:",i)
+        if sys.version_info.major < 3:
+            return chr(i)
+        else:
+            return bytes((i,))
+    # if too many colors, bail out and use only the first one for now...
+    if len(colorlist) > 254: 
+        colorlist = colorlist[:1]
+        print("debug: too many colors, reducing")
+    output = tochr(len(colorlist))+3*tochr(0)
+    allfloats = True
+    for color in colorlist:
+        for d in color:
+            if d > 1:
+                allfloats = False
+                break
+    for color in colorlist:
+        if len(color) < 4:
+            output += tochr(0)
+        for d in reversed(color):
+            if allfloats:
+                output += tochr(int(d*255))
+            else:
+                output += tochr(int(d))
+    colfile = tempfile.mkstemp(prefix="DiffuseColor")[-1]
+    f = open(colfile,"wb")
+    f.write(output)
+    f.close()
+    return colfile
+
+
+
 def getColors(filename,nodiffuse=False):
 
     """getColors(filename,nodiffuse): Extracts the colors saved in a FreeCAD file
@@ -289,6 +328,7 @@ def getStepData(objects,colors):
     return data
 
 
+
 def render(outputfile,scene=None,camera=None,zoom=False,width=400,height=300,background=(1.0,1.0,1.0),lightdir=None):
 
     """render(outputfile,scene=None,camera=None,zoom=False,width=400,height=300,background=(1.0,1.0,1.0),lightdir=None):
@@ -561,10 +601,13 @@ def save(document,filename=None,guidata=None,colors=None,camera=None):
         guidoc = buildGuiDocumentFromColors(document,colors,camera)
         if guidoc:
             zf = zipfile.ZipFile(filename, mode='a')
-            zf.write(guidoc,'GuiDocument.xml')
+            zf.write(guidoc[0],'GuiDocument.xml')
+            for colorfile in guidoc[1:]:
+                zf.write(colorfile,os.path.basename(colorfile))
             zf.close()
-            # delete the temp file
-            os.remove(guidoc)
+            # delete the temp files
+            for tfile in guidoc:
+                os.remove(tfile)
 
 
 
@@ -599,6 +642,8 @@ def buildGuiDocumentFromColors(document,colors,camera=None):
     guidoc += "-->\n"
     guidoc += "<Document SchemaVersion=\"1\">\n"
 
+    colfiles = []
+
     vps = [obj for obj in document.Objects if obj.Name in colors]
     if not vps:
         return None
@@ -608,12 +653,12 @@ def buildGuiDocumentFromColors(document,colors,camera=None):
         guidoc += "            <Properties Count=\"2\">\n"
         vpcol = colors[vp.Name]
         if isinstance(vpcol[0],tuple):
-            # Diffuse colors not yet supported (need to create extra files...) for now simply use the first color...
-            #guidoc += "                <Property name=\"DiffuseColor\" type=\"App::PropertyColorList\">\n"
-            #guidoc += "                    <ColorList file=\"DiffuseColor\"/>\n"
-            #guidoc += "                </Property>\n"
-            guidoc += "                <Property name=\"ShapeColor\" type=\"App::PropertyColor\">\n"
-            guidoc += "                    <PropertyColor value=\""+getUnsigned(vpcol[0])+"\"/>\n"
+            # distinct diffuse colors
+            colfile = saveDiffuseColor(vpcol)
+            name = os.path.basename(colfile)
+            colfiles.append(colfile)
+            guidoc += "                <Property name=\"DiffuseColor\" type=\"App::PropertyColorList\">\n"
+            guidoc += "                    <ColorList file=\""+name+"\"/>\n"
             guidoc += "                </Property>\n"
         else:
             guidoc += "                <Property name=\"ShapeColor\" type=\"App::PropertyColor\">\n"
@@ -647,7 +692,7 @@ def buildGuiDocumentFromColors(document,colors,camera=None):
     f.write(guidoc)
     f.close()
 
-    return tempxml
+    return [tempxml]+colfiles