exportWebGL.py

#***************************************************************************
#*                                                                         *
#*   Copyright (c) 2013                                                    *
#*   Yorik van Havre <yorik@uncreated.net>                                 *
#*   modified by Daniel Falck <ddfalck@gmail.com>                          *
#*   This program is free software; you can redistribute it and/or modify  *
#*   it under the terms of the GNU Lesser General Public License (LGPL)    *
#*   as published by the Free Software Foundation; either version 2 of     *
#*   the License, or (at your option) any later version.                   *
#*   for detail see the LICENCE text file.                                 *
#*                                                                         *
#*   This program is distributed in the hope that it will be useful,       *
#*   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
#*   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
#*   GNU Library General Public License for more details.                  *
#*                                                                         *
#*   You should have received a copy of the GNU Library General Public     *
#*   License along with this program; if not, write to the Free Software   *
#*   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  *
#*   USA                                                                   *
#*                                                                         *
#***************************************************************************

"FreeCAD webgl exporter"

import FreeCAD,Draft,Part,DraftGeomUtils
if FreeCAD.GuiUp:
    from DraftTools import translate

if FreeCAD.GuiUp:
    import FreeCADGui
else:
    FreeCADGui = None

tab = "                "
addWireframe = False

if open.__module__ == '__builtin__':
    pythonopen = open


def getHTML(objectsList):
    "returns the complete HTML code of a viewer for the given objects"

    # get objects data
    objectsData = ''
    for obj in objectsList:
        objectsData += getObjectData(obj)
    template = getTemplate()
    template = template.replace("$CameraData",getCameraData())
    template = template.replace("$ObjectsData",objectsData)
    return template

def getCameraData():
    "returns the position and direction of the camera as three.js snippet"

    result = ""
    if FreeCADGui:
        # getting camera position
        pos = FreeCADGui.ActiveDocument.ActiveView.viewPosition().Base
        result += "camera.position.set( "
        result += str(pos.x) + ", "
        result += str(pos.y) + ", "
        result += str(pos.z) + " );\n"
    else:
        result += "camera.position.set(200,250,200);\n"
    result += tab+"camera.lookAt( scene.position );\n"+tab
    # print result
    return result

def getObjectData(obj,wireframeMode="multimaterial"):
    """returns the geometry data of an object as three.js snippet. wireframeMode
    can be multimaterial, faceloop or None"""

    result = ""
    wires = []

    if obj.isDerivedFrom("Part::Feature"):
        fcmesh = obj.Shape.tessellate(0.1)
        result = "var geom = new THREE.Geometry();\n"
        # adding vertices data
        for i in range(len(fcmesh[0])):
            v = fcmesh[0][i]
            result += tab+"var v"+str(i)+" = new THREE.Vector3("+str(v.x)+","+str(v.y)+","+str(v.z)+");\n"
        result += tab+"console.log(geom.vertices)\n"
        for i in range(len(fcmesh[0])):
            result += tab+"geom.vertices.push(v"+str(i)+");\n"
        # adding facets data
        for f in fcmesh[1]:
            result += tab+"geom.faces.push( new THREE.Face3"+str(f)+" );\n"
        for f in obj.Shape.Faces:
            for w in f.Wires:
                wo = Part.Wire(DraftGeomUtils.sortEdges(w.Edges))
                p = []
                for v in wo.Vertexes:
                    p.append(v.Point)
                p.append(wo.Vertexes[0].Point)
                wires.append(p)

    elif obj.isDerivedFrom("Part::TopoShape") :
        fcmesh = obj.tessellate(0.1)
        result = "var geom = new THREE.Geometry();\n"
        # adding vertices data
        for i in range(len(fcmesh[0])):
            v = fcmesh[0][i]
            result += tab+"var v"+str(i)+" = new THREE.Vector3("+str(v.x)+","+str(v.y)+","+str(v.z)+");\n"
        result += tab+"console.log(geom.vertices)\n"
        for i in range(len(fcmesh[0])):
            result += tab+"geom.vertices.push(v"+str(i)+");\n"
        # adding facets data
        for f in fcmesh[1]:
            result += tab+"geom.faces.push( new THREE.Face3"+str(f)+" );\n"
        for f in obj.Faces:
            for w in f.Wires:
                wo = Part.Wire(DraftGeomUtils.sortEdges(w.Edges))
                p = []
                for v in wo.Vertexes:
                    p.append(v.Point)
                p.append(wo.Vertexes[0].Point)
                wires.append(p)


    elif obj.isDerivedFrom("Mesh::Feature"):
        mesh = obj.Mesh
        result = "var geom = new THREE.Geometry();\n"
        # adding vertices data
        for p in mesh.Points:
            v = p.Vector
            i = p.Index
            result += tab+"var v"+str(i)+" = new THREE.Vector3("+str(v.x)+","+str(v.y)+","+str(v.z)+");\n"
        result += tab+"console.log(geom.vertices)\n"
        for p in mesh.Points:
            result += tab+"geom.vertices.push(v"+str(p.Index)+");\n"
        # adding facets data
        for f in mesh.Facets:
            result += tab+"geom.faces.push( new THREE.Face3"+str(f.PointIndices)+" );\n"

    if result:
        # adding a base material
        if FreeCADGui:
            col = obj.ViewObject.ShapeColor
            rgb = Draft.getrgb(col,testbw=False)
        else:
#            rgb = "#f85500" # test color
            rgb = "#cccccc"
        result += tab+"var basematerial = new THREE.MeshBasicMaterial( { color: 0x"+str(rgb)[1:]+" } );\n"
        #result += tab+"var basematerial = new THREE.MeshLambertMaterial( { color: 0x"+str(rgb)[1:]+" } );\n"
        #result += tab+"var basematerial = new THREE.MeshPhongMaterial( { color: 0x"+str(rgb)[1:]+" } );\n"

        if wireframeMode == "faceloop":
            # adding the mesh to the scene with a wireframe copy
            result += tab+"var mesh = new THREE.Mesh( geom, basematerial );\n"
            result += tab+"scene.add( mesh );\n"
            result += tab+"var linematerial = new THREE.LineBasicMaterial({color: 0x000000,});\n"
            for w in wires:
                result += tab+"var wire = new THREE.Geometry();\n"
                for p in w:
                    result += tab+"wire.vertices.push(new THREE.Vector3("
                    result += str(p.x)+", "+str(p.y)+", "+str(p.z)+"));\n"

                result += tab+"var line = new THREE.Line(wire, linematerial);\n"
                result += tab+"scene.add(line);\n"

        elif wireframeMode == "multimaterial":
            # adding a wireframe material
            result += tab+"var wireframe = new THREE.MeshBasicMaterial( { color: "
            result += "0x555555, wireframe: true, transparent: true } );\n"
            result += tab+"var material = [ basematerial, wireframe ];\n"
            result += tab+"var mesh = new THREE.SceneUtils.createMultiMaterialObject( geom, material );\n"
            result += tab+"scene.add( mesh );\n"+tab

        elif wireframeMode == "edges":
            # adding the mesh to the scene with a wireframe copy
#            result += tab+"var mesh = new THREE.Mesh( geom, basematerial );\n"
#            result += tab+"scene.add( mesh );\n"
#            result += tab+"var linematerial = new THREE.LineBasicMaterial({color: 0x000000,});\n"
            for f in obj.Faces:
                for w in f.Wires:
                    for edge in w.Edges:
                        if isinstance(edge.Curve,Part.Line):
#                            result += tab+"wire.vertices.push(new THREE.Vector3("
                            print "wire.vertices.push(new THREE.Vector3("
#                            result += str(edge.Vertexes[1].X)+", "+str(edge.Vertexes[1].Y)+", "+str(edge.Vertexes[1].Z)+"));\n"
                            print str(edge.Vertexes[1].X)+", "+str(edge.Vertexes[1].Y)+", "+str(edge.Vertexes[1].Z)+"));\n"
#                        elif isinstance(edge.Curve,Part.Circle):
#                            pass

        else:
            # adding the mesh to the scene with simple material
            result += tab+"var mesh = new THREE.Mesh( geom, basematerial );\n"
            result += tab+"scene.add( mesh );\n"+tab

    return result

def getTemplate():
    "returns a html template"

    result = """<!DOCTYPE html>
        <html>
        <head>
            <title>FreeCAD model</title>
            <script type="text/javascript" src="http://cdnjs.cloudflare.com/ajax/libs/three.js/r50/three.min.js"></script>

            <script>

            var camera, controls, scene, renderer;

            window.onload = function() {

                var SCREEN_WIDTH = window.innerWidth, SCREEN_HEIGHT = window.innerHeight;
                var VIEW_ANGLE = 35, ASPECT = SCREEN_WIDTH / SCREEN_HEIGHT, NEAR = 0.1, FAR = 20000;
                //var ASPECT = SCREEN_WIDTH / SCREEN_HEIGHT, NEAR = 1, FAR = 1000;
                renderer = new THREE.WebGLRenderer();
                //renderer.setSize( SCREEN_WIDTH, SCREEN_HEIGHT );
        renderer.setSize(SCREEN_WIDTH/2.0 , SCREEN_HEIGHT/2.0);
                // Set the background color of the renderer to black, with full opacity
                renderer.setClearColorHex(0x000000, 1);
                //renderer.setClearColorHex(0x9797AA, 1);

                document.body.appendChild( renderer.domElement );

                scene = new THREE.Scene();

                camera = new THREE.PerspectiveCamera(
                    VIEW_ANGLE,      // Field of view
                    ASPECT,          // Aspect ratio
                    NEAR,            // Near plane
                    FAR              // Far plane
                );

                //var camera = new THREE.OrthographicCamera( -SCREEN_WIDTH / 2, SCREEN_WIDTH / 2,   SCREEN_HEIGHT / 2, -SCREEN_HEIGHT / 2, NEAR, FAR );
                $CameraData // placeholder for the FreeCAD camera

                controls = new THREE.TrackballControls( camera );
                controls.rotateSpeed = 2.0;
                controls.zoomSpeed = 1.2;
                controls.panSpeed = 0.8;
                controls.noZoom = false;
                controls.noPan = false;
                controls.staticMoving = true;
                controls.dynamicDampingFactor = 0.3;
                controls.keys = [ 65, 83, 68 ];

                $ObjectsData // placeholder for the FreeCAD objects

                //var light = new THREE.PointLight( 0xFFFF00 );
                //light.position.set( -10000, -10000, 10000 );
                //scene.add( light );
                var directionalLight = new THREE.DirectionalLight(0xffffff);
                directionalLight.position.set(1, 1, 1).normalize();
                scene.add(directionalLight);
                renderer.render( scene, camera );

                animate();
            };

            function animate(){
                requestAnimationFrame( animate );
                render();
            };

            function render(){
                controls.update();
                renderer.render( scene, camera );
            };
            </script>
        </head>
        <body></body>
        </html>"""

    return result