Arch: Added materials support to Arch OBJ import/exporter - fixes #2139

src/Mod/Arch/Init.py

@@ -24,6 +24,7 @@
 # add import/export types
 FreeCAD.addImportType("Industry Foundation Classes (*.ifc)","importIFC")
 FreeCAD.addExportType("Industry Foundation Classes (*.ifc)","importIFC")
+FreeCAD.addImportType("Wavefront OBJ - Arch module (*.obj)","importOBJ")
 FreeCAD.addExportType("Wavefront OBJ - Arch module (*.obj)","importOBJ")
 FreeCAD.addExportType("WebGL file (*.html)","importWebGL")
 FreeCAD.addImportType("Collada (*.dae)","importDAE")

src/Mod/Arch/importDAE.py

@@ -125,13 +125,13 @@ def read(filename):
             if "}" in node.xmlnode.tag:
                 bt = node.xmlnode.tag.split("}")[0]+"}"
                 gnode = node.xmlnode.find(bt+"instance_geometry")
-                if gnode is not None:
+                if gnode != None:
                     bnode = gnode.find(bt+"bind_material")
-                    if bnode is not None:
+                    if bnode != None:
                         tnode = bnode.find(bt+"technique_common")
-                        if tnode is not None:
+                        if tnode != None:
                             mnode = tnode.find(bt+"instance_material")
-                            if mnode is not None:
+                            if mnode != None:
                                 if "target" in mnode.keys():
                                     mname = mnode.get("target").strip("#")
                                     for m in col.materials:

src/Mod/Arch/importOBJ.py

@@ -1,6 +1,6 @@
 #***************************************************************************
 #*                                                                         *
-#*   Copyright (c) 2011 Yorik van Havre <yorik@uncreated.net>              *  
+#*   Copyright (c) 2011 Yorik van Havre <yorik@uncreated.net>              *
 #*                                                                         *
 #*   This program is free software; you can redistribute it and/or modify  *
 #*   it under the terms of the GNU Lesser General Public License (LGPL)    *
@@ -20,7 +20,7 @@
 #*                                                                         *
 #***************************************************************************
 
-import FreeCAD, DraftGeomUtils, Part, Draft, Arch, Mesh
+import FreeCAD, DraftGeomUtils, Part, Draft, Arch, Mesh, os
 if FreeCAD.GuiUp:
     from DraftTools import translate
 else:
@@ -31,11 +31,12 @@ else:
 
 ## @package importOBJ
 #  \ingroup ARCH
-#  \brief OBJ file format importer and exporter
+#  \brief OBJ file format exporter
 #
-#  This module provides tools to import and export OBJ files.
+#  This module provides tools to import & export OBJ files.
-#  It is an alternative tothe standard Mesh OBJ importer/exporter
+#  It is an alternative to the standard Mesh OBJ exporter
-#  and supports exporting faces with more than 3 vertices.
+#  and supports exporting faces with more than 3 vertices
+#  and supports object colors / materials
 
 p = Draft.precision()
 
@@ -119,28 +120,40 @@ def export(exportList,filename):
     offset = 1
     objectslist = Draft.getGroupContents(exportList,walls=True,addgroups=True)
     objectslist = Arch.pruneIncluded(objectslist)
+    filenamemtl = filename[:-4] + ".mtl"
+    materials = []
+    outfile.write("mtllib " + os.path.basename(filenamemtl) + "\n")
     for obj in objectslist:
         if obj.isDerivedFrom("Part::Feature"):
-            mesh = None
+            hires = None
             if FreeCAD.GuiUp:
                 visible = obj.ViewObject.isVisible()
-                if obj.ViewObject.DisplayMode == "Mesh":
+                if obj.ViewObject.DisplayMode == "Hires":
-                    if hasattr(obj,"Mesh"):
+                    # check if high-resolution object is available
-                        if obj.Mesh:
+                    if hasattr(obj,"Hires"):
-                            mesh = obj.Mesh.Mesh.copy()
+                        if obj.Hires:
-                            mesh.Placement = obj.Placement.multiply(obj.Mesh.Mesh.Placement)
+                            if obj.Hires.isDerivedFrom("Mesh::Feature"):
-                    if not mesh:
+                                m = obj.Hires.Mesh
+                            else:
+                                m = obj.Hires.Shape
+                            hires = m.copy()
+                            hires.Placement = obj.Placement.multiply(m.Placement)
+                    if not hires:
                         if hasattr(obj,"CloneOf"):
                             if obj.CloneOf:
-                                if hasattr(obj.CloneOf,"Mesh"):
+                                if hasattr(obj.CloneOf,"Hires"):
-                                    if obj.CloneOf.Mesh:
+                                    if obj.CloneOf.Hires:
-                                        mesh = obj.CloneOf.Mesh.Mesh.copy()
+                                        if obj.CloneOf.Hires.isDerivedFrom("Mesh::Feature"):
-                                        mesh.Placement = obj.Placement.multiply(obj.CloneOf.Placement).multiply(obj.CloneOf.Mesh.Mesh.Placement)
+                                            m = obj.CloneOf.Hires.Mesh
+                                        else:
+                                            m = obj.CloneOf.Hires.Shape
+                                        hires = m.copy()
+                                        hires.Placement = obj.Placement.multiply(obj.CloneOf.Placement).multiply(m.Placement)
             else:
                 visible = True
             if visible:
-                if mesh:
+                if hires:
-                    vlist,elist,flist = getIndices(mesh,offset)
+                    vlist,elist,flist = getIndices(hires,offset)
                 else:
                     vlist,elist,flist = getIndices(obj.Shape,offset)
                 if vlist == None:
@@ -148,6 +161,22 @@ def export(exportList,filename):
                 else:
                     offset += len(vlist)
                     outfile.write("o " + obj.Name + "\n")
+
+                    # write material
+                    m = False
+                    if hasattr(obj,"BaseMaterial"):
+                        if obj.BaseMaterial:
+                            outfile.write("usemtl " + obj.BaseMaterial.Name + "\n")
+                            materials.append(obj.BaseMaterial)
+                            m = True
+                    if not m:
+                        if FreeCAD.GuiUp:
+                            if hasattr(obj.ViewObject,"ShapeColor") and hasattr(obj.ViewObject,"Transparency"):
+                                mn = Draft.getrgb(obj.ViewObject.ShapeColor)[1:]
+                                outfile.write("usemtl color_" + mn + "\n")
+                                materials.append(("color_" + mn,obj.ViewObject.ShapeColor,obj.ViewObject.Transparency))
+
+                    # write geometry
                     for v in vlist:
                         outfile.write("v" + v + "\n")
                     for e in elist:
@@ -156,6 +185,123 @@ def export(exportList,filename):
                         outfile.write("f" + f + "\n")
     outfile.close()
     FreeCAD.Console.PrintMessage(translate("Arch","successfully written ").decode('utf8')+filename+"\n")
-            
+    if materials:
-            
+        outfile = pythonopen(filenamemtl,"wb")
-            
+        outfile.write("# FreeCAD v" + ver[0] + "." + ver[1] + " build" + ver[2] + " Arch module\n")
+        outfile.write("# http://www.freecadweb.org\n")
+        kinds = {"AmbientColor":"Ka ","DiffuseColor":"Kd ","SpecularColor":"Ks ","EmissiveColor":"Ke ","Transparency":"d "}
+        done = [] # store names to avoid duplicates
+        for mat in materials:
+            if isinstance(mat,tuple):
+                if not mat[0] in done:
+                    outfile.write("newmtl " + mat[0] + "\n")
+                    outfile.write("Kd " + str(mat[1][0]) + " " + str(mat[1][1]) + " " + str(mat[1][2]) + "\n")
+                    outfile.write("d " + str(mat[2]) + "\n")
+                    done.append(mat[0])
+            else:
+                if not mat.Name in done:
+                    outfile.write("newmtl " + mat.Name + "\n")
+                    for prop in kinds:
+                        if prop in mat.Material:
+                            outfile.write(kinds[prop] + mat.Material[prop].strip("()").replace(',',' ') + "\n")
+                    done.append(mat.Name)
+        outfile.write("# Material Count: " + str(len(materials)))
+        outfile.close()
+        FreeCAD.Console.PrintMessage(translate("Arch","successfully written ") + filenamemtl + "\n")
+
+
+def decode(name):
+    "decodes encoded strings"
+    try:
+        decodedName = (name.decode("utf8"))
+    except UnicodeDecodeError:
+        try:
+            decodedName = (name.decode("latin1"))
+        except UnicodeDecodeError:
+            FreeCAD.Console.PrintError(translate("Arch","Error: Couldn't determine character encoding"))
+            decodedName = name
+    return decodedName
+
+def open(filename):
+    "called when freecad wants to open a file"
+    docname = (os.path.splitext(os.path.basename(filename))[0]).encode("utf8")
+    doc = FreeCAD.newDocument(docname)
+    doc.Label = decode(docname)
+    return insert(filename,doc.Name)
+
+def insert(filename,docname):
+    "called when freecad wants to import a file"
+    try:
+        doc = FreeCAD.getDocument(docname)
+    except NameError:
+        doc = FreeCAD.newDocument(docname)
+    FreeCAD.ActiveDocument = doc
+
+    with pythonopen(filename,"rb") as infile:
+        verts = []
+        facets = []
+        activeobject = None
+        material = None
+        colortable = {}
+        for line in infile:
+            line = line.strip()
+            if line[:7] == "mtllib ":
+                matlib = os.path.join(os.path.dirname(filename),line[7:])
+                if os.path.exists(matlib):
+                    with pythonopen(matlib,"rb") as matfile:
+                        mname = None
+                        color = None
+                        trans = None
+                        for mline in matfile:
+                            mline = mline.strip()
+                            if mline[:7] == "newmtl ":
+                                if mname and color:
+                                    colortable[mname] = [color,trans]
+                                color = None
+                                trans = None
+                                mname = mline[7:]
+                            elif mline[:3] == "Kd ":
+                                color = tuple([float(i) for i in mline[3:].split()])
+                            elif mline[:2] == "d ":
+                                trans = int(float(mline[2:])*100)
+                        if mname and color:
+                            colortable[mname] = [color,trans]
+            elif line[:2] == "o ":
+                if activeobject:
+                    makeMesh(doc,activeobject,verts,facets,material,colortable)
+                material = None
+                facets = []
+                activeobject = line[2:]
+            elif line[:2] == "v ":
+                verts.append([float(i) for i in line[2:].split()])
+            elif line[:2] == "f ":
+                facets.append([int(i) for i in line[2:].split()])
+            elif line[:7] == "usemtl ":
+                material = line[7:]
+        if activeobject:
+            makeMesh(doc,activeobject,verts,facets,material,colortable)
+    FreeCAD.Console.PrintMessage(translate("Arch","successfully imported ")+filename+"\n")
+    return doc
+
+def makeMesh(doc,activeobject,verts,facets,material,colortable):
+    mfacets = []
+    if facets:
+        for facet in facets:
+            if len(facet) > 3:
+                vecs = [FreeCAD.Vector(*verts[i-1]) for i in facet]
+                vecs.append(vecs[0])
+                pol = Part.makePolygon(vecs)
+                face = Part.Face(pol)
+                tris = face.tessellate(1)
+                for tri in tris[1]:
+                    mfacet.append([tris[0][i] for i in tri])
+            else:
+                mfacets.append([verts[i-1] for i in facet])
+    if mfacets:
+        mobj = doc.addObject("Mesh::Feature",activeobject)
+        mobj.Mesh = Mesh.Mesh(mfacets)
+        if material and FreeCAD.GuiUp:
+            if material in colortable:
+                mobj.ViewObject.ShapeColor = colortable[material][0]
+                if colortable[material][1] != None:
+                    mobj.ViewObject.Transparency = colortable[material][1]