810 lines
43 KiB
Python
810 lines
43 KiB
Python
#***************************************************************************
|
|
#* Copyright (c) 2013 Yorik van Havre <yorik@uncreated.net> *
|
|
#* Copyright (c) 2020 Travis Apple <travisapple@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 *
|
|
#* *
|
|
#***************************************************************************
|
|
#
|
|
# REFS:
|
|
# https://github.com/mrdoob/three.js/blob/master/examples/webgl_interactive_buffergeometry.html
|
|
# https://threejs.org/examples/#webgl_buffergeometry_lines
|
|
# https://forum.freecadweb.org/viewtopic.php?t=51245
|
|
# https://forum.freecadweb.org/viewtopic.php?t=29487
|
|
# https://threejs.org/examples/#webgl_raycast_sprite
|
|
#
|
|
# Params for export()
|
|
# 'colors' is of the form: {'Body': [1,0,0], 'Body001': [1,1,0], 'Body002': [1,0,1] }
|
|
# 'camera' is of the form: "PerspectiveCamera {\n viewportMapping ADJUST_CAMERA\n position 30.242626 -51.772324 85.63475\n orientation -0.4146691 0.088459305 -0.90566254 4.7065201\nnearDistance 53.126431\n farDistance 123.09125\n aspectRatio 1\n focalDistance 104.53851\n heightAngle 0.78539819\n\n}"
|
|
# The 'camera' string for the active document may be generated from: import OfflineRenderingUtils; OfflineRenderingUtils.getCamera(FreeCAD.ActiveDocument.FileName);
|
|
#
|
|
# Development reload oneliner:
|
|
# def re(): from importlib import reload;import importWebGL;reload(importWebGL);o=FreeCAD.getDocument("YourDocName");importWebGL.export([o.getObject("YourBodyName")],u"C:/path/to/your/file.htm");
|
|
|
|
"""FreeCAD WebGL Exporter"""
|
|
|
|
import FreeCAD,Mesh,Draft,Part,DraftGeomUtils,Arch,OfflineRenderingUtils,json,six
|
|
|
|
if FreeCAD.GuiUp:
|
|
import FreeCADGui
|
|
from DraftTools import translate
|
|
else:
|
|
FreeCADGui = None
|
|
def translate(ctxt, txt): return txt
|
|
|
|
if open.__module__ in ['__builtin__','io']: pythonopen = open
|
|
|
|
## @package importWebGL
|
|
# \ingroup ARCH
|
|
# \brief FreeCAD WebGL Exporter
|
|
#
|
|
# This module provides tools to export HTML files containing the
|
|
# exported objects in WebGL format and a simple three.js-based viewer.
|
|
|
|
disableCompression = False # Compress object data before sending to JS
|
|
base = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890!#$%&()*+-:;/=>?@[]^_,.{|}~`' # safe str chars for js in all cases
|
|
baseFloat = ',.-0123456789'
|
|
|
|
def getHTMLTemplate():
|
|
return """<!DOCTYPE html>
|
|
<html lang="en">
|
|
<head>
|
|
<title>$pagetitle</title>
|
|
<meta charset="utf-8" />
|
|
<meta name="viewport" content="width=device-width, user-scalable=no, minimum-scale=1.0, maximum-scale=1.0" />
|
|
<meta name="generator" content="FreeCAD $version" />
|
|
<style>
|
|
* { margin:0; padding:0; }
|
|
body{
|
|
background: #ffffff; /* Old browsers */
|
|
background: -moz-linear-gradient(top, #e3e9fc 0%, #ffffff 70%, #e2dab3 100%); /* FF3.6-15 */
|
|
background: -webkit-linear-gradient(top, #e3e9fc 0%,#ffffff 70%,#e2dab3 100%); /* Chrome10-25, Safari5.1-6 */
|
|
background: linear-gradient(to bottom, #e3e9fc 0%,#ffffff 70%,#e2dab3 100%); /* W3C, IE10+, FF16+, Chrome26+, Opera12+, Safari7+ */
|
|
}
|
|
canvas { display: block; }
|
|
#arrowCanvas {
|
|
position: absolute;
|
|
left: 0px;
|
|
bottom: 0px;
|
|
z-index: 100;
|
|
}
|
|
select { width: 170px; }
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<script type="module">
|
|
|
|
// Direct from mrdoob - r122(10/28/20): https://www.jsdelivr.com/package/npm/three
|
|
import * as THREE from 'https://cdn.jsdelivr.net/npm/three@0.122.0/build/three.module.js';
|
|
import { OrbitControls } from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/controls/OrbitControls.js';
|
|
import Stats from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/libs/stats.module.js';
|
|
import { GUI } from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/libs/dat.gui.module.js';
|
|
import { Line2 } from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/lines/Line2.js';
|
|
import { LineMaterial } from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/lines/LineMaterial.js';
|
|
import { LineGeometry } from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/lines/LineGeometry.js';
|
|
import { EdgeSplitModifier } from 'https://cdn.jsdelivr.net/npm/three@0.122.0/examples/jsm/modifiers/EdgeSplitModifier.js';
|
|
|
|
var disableCompression = $disableCompression;
|
|
var base = '$base';
|
|
var baseFloat = '$float';
|
|
|
|
THREE.Object3D.DefaultUp = new THREE.Vector3(0,0,1); // Z is up for FreeCAD
|
|
|
|
var raycasterObj = []; // list of obj that can mouseover highlight
|
|
|
|
var scene = new THREE.Scene();
|
|
|
|
var renderer = new THREE.WebGLRenderer( { alpha: true, antialias: true } ); // Clear bg so we can set it with css
|
|
renderer.setClearColor( 0x000000, 0 );
|
|
var mainCanvas = document.body.appendChild( renderer.domElement );
|
|
mainCanvas.setAttribute('id', 'mainCanvas');
|
|
|
|
// HemisphereLight gives different colors of light from the top and bottom simulating reflected light from the 'ground' and 'sky'
|
|
scene.add(new THREE.HemisphereLight(0xC7E8FF, 0xFFE3B3, 0.4));
|
|
|
|
var dLight1 = new THREE.DirectionalLight( 0xffffff, 0.4 );
|
|
dLight1.position.set( 5, -2, 3 );
|
|
scene.add( dLight1 );
|
|
var dLight2 = new THREE.DirectionalLight( 0xffffff, 0.4 );
|
|
dLight2.position.set( -5, 2, 3 );
|
|
scene.add( dLight2 );
|
|
|
|
var data = $data;
|
|
|
|
function baseDecode( input ) {
|
|
var baseCt = base.length;
|
|
var output = [];
|
|
var len = parseInt( input[0] ); // num chars of each element
|
|
for ( var i=1; i<input.length; i+=len ) {
|
|
var str = input.substring(i, i+len).trim();
|
|
var val = 0;
|
|
for ( var s=0; s<str.length; s++ ) {
|
|
var ind = base.indexOf(str[s]);
|
|
val += ind * Math.pow(baseCt, s);
|
|
}
|
|
output.push(val);
|
|
}
|
|
return output;
|
|
}
|
|
function floatDecode( input ) {
|
|
var baseCt = base.length;
|
|
var baseFloatCt = baseFloat.length;
|
|
var numString = '';
|
|
for ( var i=0; i<input.length; i+=4 ) {
|
|
var b90chunk = input.substring(i, i+4).trim();
|
|
var quotient = 0;
|
|
for ( var s=0; s<b90chunk.length; s++ ) {
|
|
var ind = base.indexOf(b90chunk[s]);
|
|
quotient += ind * Math.pow(baseCt, s);
|
|
}
|
|
var buffer = '';
|
|
for ( var s=0; s<7; s++ ) {
|
|
buffer = baseFloat[ quotient % baseFloatCt ] + buffer;
|
|
quotient = parseInt(quotient / baseFloatCt);
|
|
}
|
|
numString += buffer;
|
|
}
|
|
var trailingCommas = 0;
|
|
for ( var s=1; s<7; s++ ) {
|
|
if( numString[ numString.length-s ] == baseFloat[0] ) { trailingCommas++; }
|
|
}
|
|
numString = numString.substring(0, numString.length - trailingCommas);
|
|
return numString;
|
|
}
|
|
// Decode from base90 and distribute the floats
|
|
for ( var obj=0; obj<data.objects.length; obj++ ) {
|
|
if (!disableCompression) {
|
|
data.objects[obj].floats = JSON.parse('[' + floatDecode( data.objects[obj].floats ) + ']');
|
|
data.objects[obj].verts = baseDecode( data.objects[obj].verts );
|
|
for ( var v=0; v<data.objects[obj].verts.length; v++ ) {
|
|
data.objects[obj].verts[v] = data.objects[obj].floats[ data.objects[obj].verts[v] ];
|
|
}
|
|
data.objects[obj].facets = baseDecode( data.objects[obj].facets );
|
|
for ( var w=0; w<data.objects[obj].wires.length; w++ ) {
|
|
data.objects[obj].wires[w] = baseDecode( data.objects[obj].wires[w] );
|
|
for ( var wv=0; wv<data.objects[obj].wires[w].length; wv++ ) {
|
|
data.objects[obj].wires[w][wv] = data.objects[obj].floats[ data.objects[obj].wires[w][wv] ];
|
|
}
|
|
}
|
|
for ( var f=0; f<data.objects[obj].facesToFacets.length; f++ ) {
|
|
data.objects[obj].facesToFacets[f] = baseDecode( data.objects[obj].facesToFacets[f] );
|
|
}
|
|
}
|
|
data.objects[obj].floats = false;
|
|
}
|
|
|
|
// Get bounds for global clipping
|
|
for ( var obj=0; obj<data.objects.length; obj++ ) {
|
|
if (obj == 0) {
|
|
var globalMaxMin = [{min:data.objects[obj].verts[0],max:data.objects[obj].verts[0]},
|
|
{min:data.objects[obj].verts[1],max:data.objects[obj].verts[1]},
|
|
{min:data.objects[obj].verts[2],max:data.objects[obj].verts[2]}]
|
|
}
|
|
for ( var v=0; v<data.objects[obj].verts.length; v++ ) {
|
|
if ( data.objects[obj].verts[v] < globalMaxMin[v % 3].min ) { globalMaxMin[v % 3].min = data.objects[obj].verts[v]; }
|
|
if ( data.objects[obj].verts[v] > globalMaxMin[v % 3].max ) { globalMaxMin[v % 3].max = data.objects[obj].verts[v]; }
|
|
}
|
|
}
|
|
var bigrange = 0;
|
|
for ( var i=0; i<globalMaxMin.length; i++ ) { // add a little extra
|
|
var range = globalMaxMin[i].max - globalMaxMin[i].min;
|
|
if ( range > bigrange ) { bigrange = range; }
|
|
globalMaxMin[i].min -= range * 0.01;
|
|
globalMaxMin[i].max += range * 0.01;
|
|
}
|
|
|
|
var camCenter = new THREE.Vector3(
|
|
0.5 * (globalMaxMin[0].max - globalMaxMin[0].min) + globalMaxMin[0].min,
|
|
0.5 * (globalMaxMin[1].max - globalMaxMin[1].min) + globalMaxMin[1].min,
|
|
0.5 * (globalMaxMin[2].max - globalMaxMin[2].min) + globalMaxMin[2].min );
|
|
var viewSize = 1.5 * bigrange; // make the view area a little bigger than the object
|
|
var aspectRatio = window.innerWidth / window.innerHeight;
|
|
var originalAspect = aspectRatio
|
|
function initCam(camera) {
|
|
camera.position.set(
|
|
data.camera.position_x,
|
|
data.camera.position_y,
|
|
data.camera.position_z);
|
|
camera.lookAt( camCenter );
|
|
camera.updateMatrixWorld();
|
|
}
|
|
var cameraType = data.camera.type;
|
|
var persCamera = new THREE.PerspectiveCamera( data.camera.focalDistance, window.innerWidth / window.innerHeight, 1, 10000 );
|
|
initCam(persCamera);
|
|
var orthCamera = new THREE.OrthographicCamera(-aspectRatio * viewSize / 2, aspectRatio * viewSize / 2, viewSize / 2, -viewSize / 2, -10000, 10000);
|
|
initCam(orthCamera);
|
|
|
|
function assignMesh( positions, color, opacity ) {
|
|
|
|
var geometry = new THREE.BufferGeometry();
|
|
geometry.setAttribute( 'position', new THREE.BufferAttribute( positions, 3 ) );
|
|
|
|
// EdgeSplitModifier is used to combine verts so that smoothing normals can be generated WITHOUT removing the hard edges of the design
|
|
// REF: https://threejs.org/examples/?q=edge#webgl_modifier_edgesplit - https://github.com/mrdoob/three.js/pull/20535
|
|
var edgeSplit = new EdgeSplitModifier();
|
|
var cutOffAngle = 20;
|
|
geometry = edgeSplit.modify( geometry, cutOffAngle * Math.PI / 180 );
|
|
geometry.computeVertexNormals();
|
|
geometry.computeBoundingSphere();
|
|
|
|
var transparent = false;
|
|
if (opacity != 1.0) { transparent = true; }
|
|
var material = new THREE.MeshLambertMaterial({
|
|
color: color,
|
|
side: THREE.DoubleSide,
|
|
vertexColors: false,
|
|
flatShading: false,
|
|
opacity: opacity,
|
|
transparent: transparent,
|
|
fog: false
|
|
});
|
|
|
|
var meshobj = new THREE.Mesh( geometry, material );
|
|
meshobj.name = meshobj.uuid;
|
|
faces.push( meshobj.uuid );
|
|
scene.add( meshobj );
|
|
raycasterObj.push(meshobj);
|
|
}
|
|
|
|
var objects = [];
|
|
var positions;
|
|
for ( var obj=0; obj<data.objects.length; obj++ ) { // Loop Objects
|
|
|
|
var faces = []; // Each face gets its own material because they each can have different colors
|
|
if (data.objects[obj].facesToFacets.length > 0) {
|
|
for ( var f=0; f<data.objects[obj].facesToFacets.length; f++ ) {
|
|
var facecolor = data.objects[obj].color;
|
|
if (data.objects[obj].faceColors.length > 0) {
|
|
facecolor = data.objects[obj].faceColors[f];
|
|
}
|
|
positions = new Float32Array( data.objects[obj].facesToFacets[f].length * 9 );
|
|
for ( var a=0; a<data.objects[obj].facesToFacets[f].length; a++ ) {
|
|
for ( var b=0; b<3; b++ ) {
|
|
for ( var c=0; c<3; c++ ) {
|
|
positions[ 9*a + 3*b + c ] = data.objects[obj].verts[ 3*data.objects[obj].facets[ 3*data.objects[obj].facesToFacets[f][a] + b ] + c ];
|
|
}
|
|
}
|
|
}
|
|
assignMesh( positions, facecolor, data.objects[obj].opacity );
|
|
}
|
|
} else { // No facesToFacets means that there was a tessellate() mismatch inside FreeCAD. Use all facets in object create this mesh
|
|
positions = new Float32Array( data.objects[obj].facets.length * 3 );
|
|
for ( var a=0; a<data.objects[obj].facets.length; a++ ) {
|
|
for ( var b=0; b<3; b++ ) {
|
|
positions[ 3*a + b ] = data.objects[obj].verts[ 3*data.objects[obj].facets[a] + b ];
|
|
}
|
|
}
|
|
assignMesh( positions, data.objects[obj].color, data.objects[obj].opacity );
|
|
}
|
|
|
|
// Wires
|
|
// cannot have lines in WebGL that are wider than 1px due to browser limitations so Line2 workaround lib is used
|
|
// REF: https://threejs.org/examples/?q=fat#webgl_lines_fat - https://jsfiddle.net/brLk6aud/1/
|
|
var wirematerial = new LineMaterial( {
|
|
color: new THREE.Color('rgb(0,0,0)'),
|
|
linewidth: 2, // in pixels
|
|
dashed: false, dashSize: 1, gapSize: 1, dashScale: 3
|
|
} );
|
|
wirematerial.resolution.set( window.innerWidth, window.innerHeight );
|
|
|
|
var wires = [];
|
|
for ( var w=0; w<data.objects[obj].wires.length; w++ ) {
|
|
var wiregeometry = new LineGeometry();
|
|
wiregeometry.setPositions( data.objects[obj].wires[w] );
|
|
var wire = new Line2( wiregeometry, wirematerial );
|
|
wire.computeLineDistances();
|
|
wire.scale.set( 1, 1, 1 );
|
|
wire.name = wire.uuid;
|
|
scene.add( wire );
|
|
wires.push( wire.name );
|
|
}
|
|
objects.push( { name: data.objects[obj].name, faces: faces, wires: wires } );
|
|
}
|
|
|
|
// ---- GUI Init ----
|
|
const gui = new GUI( { width: 300 } );
|
|
var guiparams = {
|
|
wiretype: 'Normal',
|
|
wirewidth: wirematerial.linewidth,
|
|
wirecolor: '#'+wirematerial.color.getHexString(),
|
|
clippingx: 100,
|
|
clippingy: 100,
|
|
clippingz: 100,
|
|
cameraType: cameraType,
|
|
navright: function() { navChange( [1,0,0] ); },
|
|
navtop: function() { navChange( [0,1,0] ); },
|
|
navfront: function() { navChange( [0,0,1] ); }
|
|
};
|
|
|
|
// ---- Wires ----
|
|
if (!wirematerial.visible) { guiparams.wiretype = 'None'; }
|
|
if (wirematerial.dashed) { guiparams.wiretype = 'Dashed'; }
|
|
const wiretypes = { Normal: 'Normal', Dashed: 'Dashed', None: 'None' };
|
|
|
|
var wireFolder = gui.addFolder( 'Wire' );
|
|
wireFolder.add( guiparams, 'wiretype', wiretypes ).name('Wire Display').onChange( wireChange );
|
|
wireFolder.add( guiparams, 'wirewidth').min(1).max(5).step(1).name('Wire Width').onChange( wireChange );
|
|
wireFolder.addColor( guiparams, 'wirecolor' ).name('Wire Color').onChange( wireChange );
|
|
wireFolder.close();
|
|
|
|
function wireChange() {
|
|
for ( var obj=0; obj<objects.length; obj++ ) {
|
|
if ( objects[obj].wires.length == 0 ) { continue; }
|
|
var m = scene.getObjectByName( objects[obj].wires[0] ).material; // all wires in obj share mat
|
|
if (m.dashed) {
|
|
if (guiparams.wiretype != 'Dashed') {
|
|
m.dashed = false;
|
|
delete m.defines.USE_DASH;
|
|
}
|
|
} else {
|
|
if (guiparams.wiretype == 'Dashed') {
|
|
m.dashed = true;
|
|
// Dashed lines require this as of r122. delete if not dashed
|
|
m.defines.USE_DASH = ""; // https://discourse.threejs.org/t/dashed-line2-material/10825
|
|
}
|
|
}
|
|
if (guiparams.wiretype == 'None') {
|
|
m.visible = false;
|
|
} else {
|
|
m.visible = true;
|
|
}
|
|
m.linewidth = guiparams.wirewidth;
|
|
m.color = new THREE.Color(guiparams.wirecolor);
|
|
m.needsUpdate = true;
|
|
}
|
|
}
|
|
wireChange();
|
|
|
|
// ---- Clipping ----
|
|
var clippingFolder = gui.addFolder( 'Clipping' );
|
|
clippingFolder.add( guiparams, 'clippingx').min(0).max(100).step(1).name('X-Axis Clipping').onChange( clippingChange );
|
|
clippingFolder.add( guiparams, 'clippingy').min(0).max(100).step(1).name('Y-Axis Clipping').onChange( clippingChange );
|
|
clippingFolder.add( guiparams, 'clippingz').min(0).max(100).step(1).name('Z-Axis Clipping').onChange( clippingChange );
|
|
clippingFolder.close();
|
|
|
|
var clipPlaneX = new THREE.Plane( new THREE.Vector3( -1, 0, 0 ), 0 );
|
|
var clipPlaneY = new THREE.Plane( new THREE.Vector3( 0, -1, 0 ), 0 );
|
|
var clipPlaneZ = new THREE.Plane( new THREE.Vector3( 0, 0, -1 ), 0 );
|
|
function clippingChange() {
|
|
if( guiparams.clippingx < 100 || guiparams.clippingy < 100 || guiparams.clippingz < 100 ) {
|
|
if (renderer.clippingPlanes.length == 0) {
|
|
renderer.clippingPlanes.push( clipPlaneX, clipPlaneY, clipPlaneZ );
|
|
}
|
|
}
|
|
clipPlaneX.constant = (globalMaxMin[0].max - globalMaxMin[0].min) * guiparams.clippingx / 100.0 + globalMaxMin[0].min;
|
|
clipPlaneY.constant = (globalMaxMin[1].max - globalMaxMin[1].min) * guiparams.clippingy / 100.0 + globalMaxMin[1].min;
|
|
clipPlaneZ.constant = (globalMaxMin[2].max - globalMaxMin[2].min) * guiparams.clippingz / 100.0 + globalMaxMin[2].min;
|
|
}
|
|
|
|
// ---- Camera & Navigation ----
|
|
var camFolder = gui.addFolder( 'Camera' );
|
|
const cameraTypes = { Perspective: 'Perspective', Orthographic: 'Orthographic' };
|
|
camFolder.add( guiparams, 'cameraType', cameraTypes ).name('Wire Display').onChange( cameraChange );
|
|
camFolder.add( guiparams, 'navright' ).name('View Right');
|
|
camFolder.add( guiparams, 'navtop' ).name('View Top');
|
|
camFolder.add( guiparams, 'navfront' ).name('View Front');
|
|
function navChange( v ) {
|
|
var t = new THREE.Vector3();
|
|
new THREE.Box3().setFromObject(scene).getSize( t );
|
|
persControls.object.position.set( v[0]*t.x + camCenter.x, v[1]*t.y + camCenter.y, v[2]*t.z + camCenter.z);
|
|
persControls.target = camCenter;
|
|
persControls.update();
|
|
orthControls.object.position.set( v[0]*t.x + camCenter.x, v[1]*t.y + camCenter.y, v[2]*t.z + camCenter.z);
|
|
orthControls.target = camCenter;
|
|
orthControls.update();
|
|
}
|
|
function cameraChange( v ) {
|
|
cameraType = v;
|
|
}
|
|
|
|
var guiObjData = [];
|
|
var guiObjects = gui.addFolder( 'Objects' );
|
|
for ( var obj=0; obj<data.objects.length; obj++ ) {
|
|
guiObjData.push( {index: obj, color: data.objects[obj].color, opacity:data.objects[obj].opacity } );
|
|
var guiObject = guiObjects.addFolder( objects[obj].name );
|
|
guiObject.addColor( guiObjData[ guiObjData.length-1 ], 'color' ).name('Color').onChange( GUIObjectChange );
|
|
guiObject.add( guiObjData[ guiObjData.length-1 ], 'opacity' ).min(0.0).max(1.0).step(0.05).name('Opacity').onChange( GUIObjectChange );
|
|
}
|
|
function GUIObjectChange( v ) {
|
|
for ( var f=0; f<objects[this.object.index].faces.length; f++ ) {
|
|
var m = scene.getObjectByName( objects[this.object.index].faces[f] ).material;
|
|
if (this.property == 'color') { m.color.setStyle( v ); }
|
|
if (this.property == 'opacity') {
|
|
m.opacity = v;
|
|
if ( v == 1.0 ) { m.transparent = false; } else { m.transparent = true; }
|
|
}
|
|
}
|
|
if (this.property == 'opacity' && objects[ this.object.index ].wires.length > 0) {
|
|
var w = scene.getObjectByName( objects[ this.object.index ].wires[0] ).material; // all wires in obj share mat
|
|
w.opacity = v;
|
|
if ( v == 1.0 ) { w.transparent = false; } else { w.transparent = true; }
|
|
}
|
|
}
|
|
|
|
// Make simple orientation arrows and box - REF: http://jsfiddle.net/b97zd1a3/16/
|
|
var arrowCanvasSize = { x: 150, y: 150 }; // in pixels on the lower left
|
|
var arrowRenderer = new THREE.WebGLRenderer( { alpha: true } ); // clear
|
|
arrowRenderer.setClearColor( 0x000000, 0 );
|
|
arrowRenderer.setSize( arrowCanvasSize.x, arrowCanvasSize.y );
|
|
|
|
var arrowCanvas = document.body.appendChild( arrowRenderer.domElement );
|
|
arrowCanvas.setAttribute('id', 'arrowCanvas');
|
|
arrowCanvas.style.width = arrowCanvasSize.x;
|
|
arrowCanvas.style.height = arrowCanvasSize.y;
|
|
|
|
var arrowScene = new THREE.Scene();
|
|
|
|
var arrowCamera = new THREE.PerspectiveCamera( 50, arrowCanvasSize.x / arrowCanvasSize.y, 1, 500 );
|
|
arrowCamera.up = persCamera.up; // important!
|
|
|
|
var arrowPos = new THREE.Vector3( 0,0,0 );
|
|
arrowScene.add( new THREE.ArrowHelper( new THREE.Vector3( 1,0,0 ), arrowPos, 60, 0x7F2020, 20, 10 ) );
|
|
arrowScene.add( new THREE.ArrowHelper( new THREE.Vector3( 0,1,0 ), arrowPos, 60, 0x207F20, 20, 10 ) );
|
|
arrowScene.add( new THREE.ArrowHelper( new THREE.Vector3( 0,0,1 ), arrowPos, 60, 0x20207F, 20, 10 ) );
|
|
arrowScene.add(new THREE.Mesh(
|
|
new THREE.BoxGeometry( 40, 40, 40 ),
|
|
new THREE.MeshLambertMaterial( { color: 0xaaaaaa, flatShading: false } )
|
|
));
|
|
arrowScene.add(new THREE.HemisphereLight(0xC7E8FF, 0xFFE3B3, 1.2));
|
|
|
|
// Controls
|
|
var persControls = new OrbitControls( persCamera, renderer.domElement );
|
|
persControls.target = camCenter; // rotate around center of parts
|
|
persControls.update();
|
|
var orthControls = new OrbitControls( orthCamera, renderer.domElement );
|
|
orthControls.target = camCenter; // rotate around center of parts
|
|
orthControls.update();
|
|
|
|
window.addEventListener( 'resize', onWindowResize, false );
|
|
onWindowResize();
|
|
|
|
var stats = new Stats();
|
|
document.body.appendChild( stats.dom );
|
|
|
|
renderer.domElement.addEventListener( 'mousemove', onMouseMove );
|
|
|
|
var animate = function () {
|
|
requestAnimationFrame( animate );
|
|
|
|
persControls.update();
|
|
if (cameraType == 'Perspective') {
|
|
arrowCamera.position.copy( persCamera.position );
|
|
arrowCamera.position.sub( persControls.target );
|
|
}
|
|
orthControls.update();
|
|
if (cameraType == 'Orthographic') {
|
|
arrowCamera.position.copy( orthCamera.position );
|
|
arrowCamera.position.sub( orthControls.target );
|
|
}
|
|
arrowCamera.lookAt( arrowScene.position );
|
|
arrowCamera.position.setLength( 200 );
|
|
|
|
stats.begin();
|
|
if (cameraType == 'Perspective') { renderer.render( scene, persCamera ); }
|
|
if (cameraType == 'Orthographic') { renderer.render( scene, orthCamera ); }
|
|
arrowRenderer.render( arrowScene, arrowCamera );
|
|
stats.end();
|
|
};
|
|
animate();
|
|
|
|
function onWindowResize() {
|
|
for ( var obj=0; obj<objects.length; obj++ ) {
|
|
if (objects[obj].wires.length > 0) {
|
|
var w = scene.getObjectByName( objects[obj].wires[0] ).material.resolution.set( window.innerWidth, window.innerHeight ); // all wires in obj share mat
|
|
}
|
|
}
|
|
|
|
// Ortho camera needs updating. REF: https://stackoverflow.com/questions/39373113/three-js-resize-window-not-scaling-properly
|
|
var aspect = window.innerWidth / window.innerHeight;
|
|
var change = originalAspect / aspect;
|
|
var newSize = viewSize * change;
|
|
orthCamera.left = -aspect * newSize / 2;
|
|
orthCamera.right = aspect * newSize / 2;
|
|
orthCamera.top = newSize / 2;
|
|
orthCamera.bottom = -newSize / 2;
|
|
orthCamera.updateProjectionMatrix();
|
|
|
|
renderer.setSize(window.innerWidth, window.innerHeight);
|
|
}
|
|
|
|
function onMouseMove( e ) {
|
|
var c = false;
|
|
if (cameraType == 'Orthographic') { c = orthCamera;}
|
|
if (cameraType == 'Perspective') { c = persCamera;}
|
|
if (!c) { return; }
|
|
|
|
var raycaster = new THREE.Raycaster();
|
|
raycaster.setFromCamera( new THREE.Vector2(
|
|
( e.clientX / window.innerWidth ) * 2 - 1,
|
|
-( e.clientY / window.innerHeight ) * 2 + 1
|
|
), c );
|
|
var intersects = raycaster.intersectObjects( raycasterObj );
|
|
|
|
var chosen = '';
|
|
if ( intersects.length > 0 ) {
|
|
for ( var i=0; i<intersects.length; i++ ) {
|
|
var m = intersects[i].object.material;
|
|
if (m.opacity > 0) {
|
|
if (m.emissive.getHex() == 0x000000) {
|
|
m.emissive.setHex( 0x777777 );
|
|
m.needsUpdate = true;
|
|
}
|
|
chosen = intersects[i].object.name;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
for ( var r=0; r<raycasterObj.length; r++ ) {
|
|
if (raycasterObj[r].name == chosen) { continue; }
|
|
if (raycasterObj[r].material.emissive.getHex() != 0x000000) {
|
|
raycasterObj[r].material.emissive.setHex( 0x000000 );
|
|
raycasterObj[r].material.needsUpdate = true;
|
|
}
|
|
}
|
|
}
|
|
data = false; // free up some ram
|
|
</script>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
def export( exportList, filename, colors = None, camera = None ):
|
|
"""Exports objects to an html file"""
|
|
|
|
global disableCompression, base, baseFloat
|
|
|
|
data = { 'camera':{}, 'file':{}, 'objects':[] }
|
|
|
|
if not FreeCADGui and not camera:
|
|
camera = OfflineRenderingUtils.getCamera(FreeCAD.ActiveDocument.FileName)
|
|
|
|
if camera:
|
|
# REF: https://github.com/FreeCAD/FreeCAD/blob/master/src/Mod/Arch/OfflineRenderingUtils.py
|
|
camnode = OfflineRenderingUtils.getCoinCamera(camera)
|
|
cameraPosition = camnode.position.getValue().getValue()
|
|
data['camera']['type'] = 'Orthographic'
|
|
if 'PerspectiveCamera' in camera: data['camera']['type'] = 'Perspective'
|
|
data['camera']['focalDistance'] = camnode.focalDistance.getValue()
|
|
data['camera']['position_x'] = cameraPosition[0]
|
|
data['camera']['position_y'] = cameraPosition[1]
|
|
data['camera']['position_z'] = cameraPosition[2]
|
|
else:
|
|
v = FreeCADGui.ActiveDocument.ActiveView
|
|
data['camera']['type'] = v.getCameraType()
|
|
data['camera']['focalDistance'] = v.getCameraNode().focalDistance.getValue()
|
|
data['camera']['position_x'] = v.viewPosition().Base.x
|
|
data['camera']['position_y'] = v.viewPosition().Base.y
|
|
data['camera']['position_z'] = v.viewPosition().Base.z
|
|
|
|
# Take the objects out of groups
|
|
objectslist = Draft.get_group_contents(exportList, walls=True, addgroups=False)
|
|
# objectslist = Arch.pruneIncluded(objectslist)
|
|
|
|
for obj in objectslist:
|
|
|
|
# Pull all obj data before we dig down the links
|
|
label = obj.Label
|
|
|
|
color = '#cccccc';
|
|
opacity = 1.0
|
|
if FreeCADGui:
|
|
color = Draft.getrgb(obj.ViewObject.ShapeColor, testbw = False)
|
|
opacity = int((100 - obj.ViewObject.Transparency)/5) / 20 # 0>>1 with step of 0.05
|
|
elif colors:
|
|
if label in colors:
|
|
color = Draft.getrgb(colors[label], testbw = False)
|
|
|
|
validObject = False
|
|
if obj.isDerivedFrom('Mesh::Feature'):
|
|
mesh = obj.Mesh
|
|
validObject = True
|
|
if obj.isDerivedFrom('Part::Feature'):
|
|
objShape = obj.Shape
|
|
validObject = True
|
|
if obj.isDerivedFrom('App::Link'):
|
|
linkPlacement = obj.LinkPlacement
|
|
while True: # drill down to get to the actual obj
|
|
if obj.isDerivedFrom("App::Link"):
|
|
if obj.ViewObject.OverrideMaterial: color = Draft.getrgb(obj.ViewObject.ShapeMaterial.DiffuseColor, testbw = False)
|
|
obj = obj.LinkedObject
|
|
if hasattr(obj, "__len__"):
|
|
FreeCAD.Console.PrintMessage(label + ": Sub-Links are Unsupported.\n")
|
|
break
|
|
elif obj.isDerivedFrom('Part::Feature'):
|
|
objShape = obj.Shape.copy(False)
|
|
objShape.Placement = linkPlacement
|
|
validObject = True
|
|
break
|
|
elif obj.isDerivedFrom("Mesh::Feature"):
|
|
mesh = obj.Mesh.copy()
|
|
mesh.Placement = linkPlacement
|
|
validObject = True
|
|
break
|
|
|
|
if not validObject: continue
|
|
|
|
objdata = { 'name': label, 'color': color, 'opacity': opacity, 'verts':'', 'facets':'', 'wires':[], 'faceColors':[], 'facesToFacets':[], 'floats':[] }
|
|
|
|
if obj.isDerivedFrom('Part::Feature'):
|
|
|
|
deviation = 0.5
|
|
if FreeCADGui:
|
|
deviation = obj.ViewObject.Deviation
|
|
|
|
# obj.ViewObject.DiffuseColor is length=1 when all faces are the same color, length=len(faces) for when they're not
|
|
if len(obj.ViewObject.DiffuseColor) == len(objShape.Faces):
|
|
for fc in obj.ViewObject.DiffuseColor:
|
|
objdata['faceColors'].append( Draft.getrgb(fc, testbw = False) )
|
|
|
|
# get verts and facets for ENTIRE object
|
|
shapeData = objShape.tessellate( deviation )
|
|
mesh = Mesh.Mesh(shapeData)
|
|
|
|
if len(objShape.Faces) > 1:
|
|
# Map each Facet created by tessellate() to a Face so that it can be colored correctly using faceColors
|
|
# This is done by matching the results of a tessellate() on EACH FACE to the overall tessellate stored in shapeData
|
|
# if there is any error in matching these two then we display the whole object as one face and forgo the face colors
|
|
for f in objShape.Faces:
|
|
faceData = f.tessellate( deviation )
|
|
found = True
|
|
for fv in range( len(faceData[0]) ): # face verts. List of type Vector()
|
|
found = False
|
|
for sv in range( len(shapeData[0]) ): #shape verts
|
|
if faceData[0][fv] == shapeData[0][sv]: # do not use isEqual() here
|
|
faceData[0][fv] = sv # replace with the index of shapeData[0]
|
|
found = True
|
|
break
|
|
if not found: break
|
|
if not found:
|
|
FreeCAD.Console.PrintMessage("Facet to Face Mismach.\n")
|
|
objdata['facesToFacets'] = []
|
|
break
|
|
|
|
# map each of the face facets to the shape facets and make a list of shape facet indices that belong to this face
|
|
facetList = []
|
|
for ff in faceData[1]: # face facets
|
|
found = False
|
|
for sf in range( len(shapeData[1]) ): #shape facets
|
|
if faceData[0][ff[0]] in shapeData[1][sf] and faceData[0][ff[1]] in shapeData[1][sf] and faceData[0][ff[2]] in shapeData[1][sf]:
|
|
facetList.append(sf)
|
|
found = True
|
|
break
|
|
if not found: break
|
|
if not found:
|
|
FreeCAD.Console.PrintMessage("Facet List Mismach.\n")
|
|
objdata['facesToFacets'] = []
|
|
break
|
|
|
|
objdata['facesToFacets'].append( baseEncode(facetList) )
|
|
|
|
wires = [] # Add wires
|
|
for f in objShape.Faces:
|
|
for w in f.Wires:
|
|
wo = Part.Wire(Part.__sortEdges__(w.Edges))
|
|
wire = []
|
|
for v in wo.discretize(QuasiDeflection = 0.005):
|
|
wire.append( '{:.5f}'.format(v.x) ) # use strings to avoid 0.00001 written as 1e-05
|
|
wire.append( '{:.5f}'.format(v.y) )
|
|
wire.append( '{:.5f}'.format(v.z) )
|
|
wires.append( wire )
|
|
|
|
if not disableCompression:
|
|
for w in range( len(wires) ):
|
|
for wv in range( len(wires[w]) ):
|
|
found = False
|
|
for f in range( len(objdata['floats']) ):
|
|
if objdata['floats'][f] == wires[w][wv]:
|
|
wires[w][wv] = f
|
|
found = True
|
|
break
|
|
if not found:
|
|
objdata['floats'].append( wires[w][wv] )
|
|
wires[w][wv] = len(objdata['floats'])-1
|
|
wires[w] = baseEncode(wires[w])
|
|
objdata['wires'] = wires
|
|
|
|
vIndex = {}
|
|
verts = []
|
|
for p in range( len(mesh.Points) ):
|
|
vIndex[ mesh.Points[p].Index ] = p
|
|
verts.append( '{:.5f}'.format(mesh.Points[p].Vector.x) )
|
|
verts.append( '{:.5f}'.format(mesh.Points[p].Vector.y) )
|
|
verts.append( '{:.5f}'.format(mesh.Points[p].Vector.z) )
|
|
|
|
# create floats list to compress verts and wires being written into the JS
|
|
if not disableCompression:
|
|
for v in range( len(verts) ):
|
|
found = False
|
|
for f in range( len(objdata['floats']) ):
|
|
if objdata['floats'][f] == verts[v]:
|
|
verts[v] = f
|
|
found = True
|
|
break
|
|
if not found:
|
|
objdata['floats'].append( verts[v] )
|
|
verts[v] = len(objdata['floats'])-1
|
|
objdata['verts'] = baseEncode(verts)
|
|
|
|
facets = []
|
|
for f in mesh.Facets:
|
|
for i in f.PointIndices:
|
|
facets.append( vIndex[i] )
|
|
objdata['facets'] = baseEncode(facets)
|
|
|
|
# compress floats
|
|
if not disableCompression:
|
|
# use ratio of 7x base13 to 4x base90 because 13^7 ~ 90^4
|
|
fullstr = json.dumps(objdata['floats'], separators=(',', ':'))
|
|
fullstr = fullstr.replace('[', '').replace(']', '').replace('"', '')
|
|
floatStr = ''
|
|
baseFloatCt = len(baseFloat)
|
|
baseCt = len(base)
|
|
for fs in range( 0, len(fullstr), 7 ): # chunks of 7 chars, skip the first one
|
|
str7 = fullstr[fs:(fs+7)]
|
|
quotient = 0
|
|
for s in range( len(str7) ):
|
|
quotient += baseFloat.find(str7[s]) * pow(baseFloatCt, (6-s))
|
|
for v in range(4):
|
|
floatStr += base[ quotient % baseCt ]
|
|
quotient = int(quotient / baseCt)
|
|
objdata['floats'] = floatStr
|
|
|
|
data['objects'].append( objdata )
|
|
|
|
html = getHTMLTemplate()
|
|
|
|
html = html.replace('$pagetitle',FreeCAD.ActiveDocument.Label)
|
|
version = FreeCAD.Version()
|
|
html = html.replace('$version',version[0] + '.' + version[1] + '.' + version[2])
|
|
|
|
# Remove data compression in JS
|
|
if disableCompression: html = html.replace('$disableCompression','true')
|
|
else: html = html.replace('$disableCompression','false')
|
|
|
|
html = html.replace('$base', base)
|
|
html = html.replace('$float', baseFloat)
|
|
html = html.replace('$data', json.dumps(data, separators=(',', ':')) ) # Shape Data
|
|
|
|
if six.PY2:
|
|
outfile = pythonopen(filename, "wb")
|
|
else:
|
|
outfile = pythonopen(filename, "w")
|
|
outfile.write( html )
|
|
outfile.close()
|
|
FreeCAD.Console.PrintMessage( translate("Arch", "Successfully written") + ' ' + filename + "\n" )
|
|
|
|
def baseEncode( arr ):
|
|
"""Compresses an array of ints into a base90 string"""
|
|
|
|
global disableCompression, base
|
|
if disableCompression: return arr
|
|
if len(arr) == 0: return ''
|
|
|
|
longest = 0
|
|
output = []
|
|
baseCt = len(base)
|
|
for v in range( len(arr) ):
|
|
buffer = ''
|
|
quotient = arr[v]
|
|
while True:
|
|
buffer += base[ quotient % baseCt ]
|
|
quotient = int(quotient / baseCt)
|
|
if quotient == 0: break
|
|
output.append( buffer )
|
|
if len(buffer) > longest: longest = len(buffer)
|
|
output = [('{:>'+str(longest)+'}').format(x) for x in output] # pad each element
|
|
return str(longest) + ('').join(output) |