From 31ec889565d0425ce10ff1bb8bea126ed7ad276f Mon Sep 17 00:00:00 2001 From: btzr-io Date: Wed, 6 Jun 2018 00:06:03 -0600 Subject: [PATCH] add threeViewer component --- package.json | 1 + .../component/common/loading-screen.jsx | 5 +- src/renderer/component/threeViewer/index.jsx | 269 ++++++++++++++++++ .../threeViewer/internal/detector.js | 10 + .../component/threeViewer/internal/loader.js | 35 +++ .../threeViewer/internal/renderer.js | 15 + .../component/threeViewer/internal/scene.js | 57 ++++ .../component/threeViewer/internal/three.js | 10 + yarn.lock | 4 + 9 files changed, 404 insertions(+), 2 deletions(-) create mode 100644 src/renderer/component/threeViewer/index.jsx create mode 100644 src/renderer/component/threeViewer/internal/detector.js create mode 100644 src/renderer/component/threeViewer/internal/loader.js create mode 100644 src/renderer/component/threeViewer/internal/renderer.js create mode 100644 src/renderer/component/threeViewer/internal/scene.js create mode 100644 src/renderer/component/threeViewer/internal/three.js diff --git a/package.json b/package.json index cfcc220bd..117d9114b 100644 --- a/package.json +++ b/package.json @@ -78,6 +78,7 @@ "shapeshift.io": "^1.3.1", "source-map-support": "^0.5.4", "stream-to-blob-url": "^2.1.1", + "three": "^0.93.0", "tree-kill": "^1.1.0", "y18n": "^4.0.0" }, diff --git a/src/renderer/component/common/loading-screen.jsx b/src/renderer/component/common/loading-screen.jsx index c8eb3f3fb..f313b412d 100644 --- a/src/renderer/component/common/loading-screen.jsx +++ b/src/renderer/component/common/loading-screen.jsx @@ -3,8 +3,8 @@ import React from 'react'; import Spinner from 'component/spinner'; type Props = { - status: string, spinner: boolean, + status: string, }; class LoadingScreen extends React.PureComponent { @@ -17,7 +17,8 @@ class LoadingScreen extends React.PureComponent { return (
{spinner && } - {status && {status}} + + {status}
); } diff --git a/src/renderer/component/threeViewer/index.jsx b/src/renderer/component/threeViewer/index.jsx new file mode 100644 index 000000000..28afb5c86 --- /dev/null +++ b/src/renderer/component/threeViewer/index.jsx @@ -0,0 +1,269 @@ +// @flow +import * as React from 'react'; +import * as THREE from './internal/three.js'; +import detectWebGL from './internal/detector.js'; +import ThreeScene from './internal/scene.js'; +import ThreeLoader from './internal/loader.js'; +import ThreeRenderer from './internal/renderer.js'; +import LoadingScreen from 'component/common/loading-screen'; + +type Props = { + theme: string, + autoRotate: boolean, + source: { + fileType: string, + filePath: string, + }, +}; + +class ThreeViewer extends React.PureComponent { + constructor(props: Props) { + super(props); + //Main container + this.viewer = React.createRef(); + // Object colors + this.materialColors = { + red: '#e74c3c', + blue: '#3498db', + green: '#44b098', + orange: '#f39c12', + }; + + this.state = { + error: null, + isReady: false, + isLoading: false, + }; + + this.themes = { + dark: { + gridColor: '#414e5c', + groundColor: '#13233C', + backgroundColor: '#13233C', + centerLineColor: '#7f8c8d', + }, + light: { + gridColor: '#7f8c8d', + groundColor: '#DDD', + backgroundColor: '#EEE', + centerLineColor: '#2F2F2F', + }, + }; + + const { theme } = this.props; + this.theme = this.themes[theme] || this.themes.light; + } + + createOrbitControls(camera, canvas) { + const { autoRotate } = this.props; + const controls = new THREE.OrbitControls(camera, canvas); + // Controls configuration + controls.enableDamping = true; + controls.dampingFactor = 0.75; + controls.enableZoom = true; + controls.minDistance = 1; + controls.maxDistance = 50; + controls.autoRotate = autoRotate; + return controls; + } + + createGeometry(data) { + const geometry = new THREE.Geometry(); + geometry.fromBufferGeometry(data); + geometry.computeBoundingBox(); + geometry.computeVertexNormals(); + geometry.center(); + geometry.rotateX(-Math.PI / 2); + geometry.lookAt(new THREE.Vector3(0, 0, 1)); + return geometry; + } + + createWireFrame(group) { + const wireframe = new THREE.WireframeGeometry(group.geometry); + this.wireframe = new THREE.LineSegments(wireframe); + this.wireframe.material.depthTest = false; + this.wireframe.material.opacity = 0; + this.wireframe.transparent = true; + group.add(this.wireframe); + } + + createMesh(geometry) { + const material = new THREE.MeshPhongMaterial({ + opacity: 1, + transparent: true, + depthWrite: true, + vertexColors: THREE.FaceColors, + // Positive value pushes polygon further away + polygonOffsetFactor: 1, + polygonOffsetUnits: 1, + }); + + // Set material color + material.color.set(this.materialColors.green); + + const mesh = new THREE.Mesh(geometry, material); + mesh.name = 'objectGroup'; + this.scene.add(mesh); + this.fitMeshToCamera(mesh); + this.setControlsTarget(mesh.position); + return mesh; + } + + toggleWireFrame(show = false) { + this.wireframe.opacity = show ? 1 : 0; + this.mesh.material.opacity = show ? 0 : 1; + //this.mesh.material.shading(THREE.FlatShading); + } + + fitMeshToCamera(group) { + let max = { x: 0, y: 0, z: 0 }; + let min = { x: 0, y: 0, z: 0 }; + + group.traverse(child => { + if (child instanceof THREE.Mesh) { + const box = new THREE.Box3().setFromObject(group); + // Max + max.x = box.max.x > max.x ? box.max.x : max.x; + max.y = box.max.y > max.y ? box.max.y : max.y; + max.z = box.max.z > max.z ? box.max.z : max.z; + // Min + min.x = box.min.x < min.x ? box.min.x : min.x; + min.y = box.min.y < min.y ? box.min.y : min.y; + min.z = box.min.z < min.z ? box.min.z : min.z; + } + }); + + const meshY = Math.abs(max.y - min.y); + const meshX = Math.abs(max.x - min.x); + const meshZ = Math.abs(max.z - min.z); + const scaleFactor = 10 / Math.max(meshX, meshY); + + group.scale.set(scaleFactor, scaleFactor, scaleFactor); + group.position.y = meshY / 2 * scaleFactor; + group.position.multiplyScalar(-1); + group.position.y += meshY * scaleFactor; + this.createWireFrame(group); + } + + setControlsTarget(point) { + this.controls.target.fromArray([point.x, point.y, point.z]); + this.controls.update(); + } + + startLoader() { + const { source } = this.props; + source && + ThreeLoader(source, this.renderModel.bind(this), { + onStart: this.handleStart(this), + onLoad: this.handleReady.bind(this), + onError: this.handleError.bind(this), + onProgress: this.handleProgress.bind(this), + }); + } + + handleStart() { + this.setState({ isLoading: true }); + } + + handleReady() { + // Handle load ready + this.setState({ isReady: true, isLoading: false }); + this.toggleWireFrame(); + } + + handleResize = () => { + const { offsetWidth: width, offsetHeight: height } = this.viewer.current; + this.camera.aspect = width / height; + this.camera.updateProjectionMatrix(); + this.controls.update(); + this.renderer.setSize(width, height); + }; + + handleError(url) { + this.setState({ error: "Sorry, looks like we can't load this file" }); + } + + handleProgress(url, currentItem, totalItems) { + /// Handle progress + } + + handleColorChange(color) { + if (!this.mesh) return; + const pickColor = this.materialColors[color] || this.materialColors.green; + this.mesh.material.color.set(pickColor); + this.wireframe.material.color.set(pickColor); + } + + renderModel(fileType, data) { + const geometry = this.createGeometry(data); + this.mesh = this.createMesh(geometry); + } + + renderScene() { + this.renderer = ThreeRenderer({ + antialias: true, + shadowMap: true, + }); + + this.scene = ThreeScene({ + showFog: true, + showGrid: true, + ...this.theme, + }); + + const viewer = this.viewer.current; + const canvas = this.renderer.domElement; + const { offsetWidth: width, offsetHeight: height } = viewer; + // Camera + this.camera = new THREE.PerspectiveCamera(80, width / height, 0.1, 1000); + this.camera.position.set(-9.5, 14, 11); + // Controls + this.controls = this.createOrbitControls(this.camera, canvas); + // Set viewer size + this.renderer.setSize(width, height); + // Load file and render mesh + this.startLoader(); + + const updateScene = () => { + requestAnimationFrame(updateScene); + this.controls.update(); + this.renderer.render(this.scene, this.camera); + }; + + updateScene(); + // Append canvas + viewer.appendChild(canvas); + } + + componentDidMount() { + if (detectWebGL()) { + this.renderScene(); + // Update render on resize window + window.addEventListener('resize', this.handleResize, false); + } else { + // No webgl support, handle Error... + this.state({ error: 'No webgl support!' }); + } + } + + componentWillUnmount() { + window.removeEventListener('resize', this.handleResize, false); + } + + render() { + const { isReady, isLoading, error } = this.state; + const loadingMessage = 'Rendering model.'; + const showViewer = isReady && !error; + const showLoading = isLoading && !error; + + return ( + + {error && } + {showLoading && } +
+ + ); + } +} + +export default ThreeViewer; diff --git a/src/renderer/component/threeViewer/internal/detector.js b/src/renderer/component/threeViewer/internal/detector.js new file mode 100644 index 000000000..f75d46c3f --- /dev/null +++ b/src/renderer/component/threeViewer/internal/detector.js @@ -0,0 +1,10 @@ +const detectWebGL = () => { + // Create canvas element. + const canvas = document.createElement('canvas'); + // Get WebGLRenderingContext from canvas element. + const gl = canvas.getContext('webgl') || canvas.getContext('experimental-webgl'); + // Return the result. + return gl && gl instanceof WebGLRenderingContext; +}; + +export default detectWebGL; diff --git a/src/renderer/component/threeViewer/internal/loader.js b/src/renderer/component/threeViewer/internal/loader.js new file mode 100644 index 000000000..2ba2dc83e --- /dev/null +++ b/src/renderer/component/threeViewer/internal/loader.js @@ -0,0 +1,35 @@ +import { LoadingManager, STLLoader, OBJLoader } from './three.js'; + +const Manager = ({ onLoad, onStart, onProgress, onError }) => { + const manager = new THREE.LoadingManager(); + manager.onLoad = onLoad; + manager.onStart = onStart; + //manager.onProgress = onProgress; + manager.onError = onError; + + return manager; +}; + +const Loader = (fileType, manager) => { + const fileTypes = { + stl: () => new STLLoader(manager), + obj: () => new OBJLoader(manager), + }; + return fileTypes[fileType] ? fileTypes[fileType]() : null; +}; + +const ThreeLoader = ({ fileType, filePath }, renderModel, managerEvents) => { + if (!fileType) return; + + const manager = Manager(managerEvents); + const loader = Loader(fileType, manager); + + // Unsuported loader + if (!loader) return false; + + loader.load(filePath, data => { + renderModel(fileType, data); + }); +}; + +export default ThreeLoader; diff --git a/src/renderer/component/threeViewer/internal/renderer.js b/src/renderer/component/threeViewer/internal/renderer.js new file mode 100644 index 000000000..11b6c0545 --- /dev/null +++ b/src/renderer/component/threeViewer/internal/renderer.js @@ -0,0 +1,15 @@ +import { WebGLRenderer } from './three'; + +const ThreeRenderer = ({ antialias, shadowMap }) => { + const renderer = new WebGLRenderer({ + antialias, + }); + // Renderer configuration + renderer.setPixelRatio(window.devicePixelRatio); + renderer.gammaInput = true; + renderer.gammaOutput = true; + renderer.shadowMap.enabled = shadowMap; + return renderer; +}; + +export default ThreeRenderer; diff --git a/src/renderer/component/threeViewer/internal/scene.js b/src/renderer/component/threeViewer/internal/scene.js new file mode 100644 index 000000000..c6e79c731 --- /dev/null +++ b/src/renderer/component/threeViewer/internal/scene.js @@ -0,0 +1,57 @@ +import * as THREE from './three.js'; + +const addGrid = (scene, { gridColor, centerLineColor, size }) => { + const divisions = size / 2; + const grid = new THREE.GridHelper( + size, + divisions, + new THREE.Color(centerLineColor), + new THREE.Color(gridColor) + ); + grid.material.opacity = 0.4; + grid.material.transparent = true; + scene.add(grid); +}; + +const addLights = (scene, color, groundColor) => { + // Light color + const lightColor = new THREE.Color(color); + // Main light + const light = new THREE.HemisphereLight(lightColor, groundColor, 0.4); + // Shadow light + const shadowLight = new THREE.DirectionalLight(lightColor, 0.4); + shadowLight.position.set(100, 50, 100); + // Back light + const backLight = new THREE.DirectionalLight(lightColor, 0.6); + backLight.position.set(-100, 200, 50); + // Add lights to scene + scene.add(backLight); + scene.add(light); + scene.add(shadowLight); +}; + +const Scene = ({ backgroundColor, groundColor, showFog, showGrid, gridColor, centerLineColor }) => { + // Convert colors + backgroundColor = new THREE.Color(backgroundColor); + groundColor = new THREE.Color(groundColor); + // New scene + const scene = new THREE.Scene(); + // Background color + scene.background = backgroundColor; + // Fog effect + scene.fog = showFog === true ? new THREE.Fog(backgroundColor, 1, 95) : null; + + showGrid && + addGrid(scene, { + size: 100, + gridColor, + centerLineColor, + }); + + // Add basic lights + addLights(scene, '#FFFFFF', groundColor); + // Return new three scene + return scene; +}; + +export default Scene; diff --git a/src/renderer/component/threeViewer/internal/three.js b/src/renderer/component/threeViewer/internal/three.js new file mode 100644 index 000000000..6cb0051d0 --- /dev/null +++ b/src/renderer/component/threeViewer/internal/three.js @@ -0,0 +1,10 @@ +import * as THREE from 'three'; + +// Currently it's not possible to import the files within the "examples/js" directory. +// Fix: https://github.com/mrdoob/three.js/issues/9562#issuecomment-383390251 +global.THREE = THREE; +require('three/examples/js/controls/OrbitControls'); +require('three/examples/js/loaders/OBJLoader'); +require('three/examples/js/loaders/STLLoader'); + +module.exports = global.THREE; diff --git a/yarn.lock b/yarn.lock index b25d9d41d..160424789 100644 --- a/yarn.lock +++ b/yarn.lock @@ -8940,6 +8940,10 @@ text-table@~0.2.0: version "0.2.0" resolved "https://registry.yarnpkg.com/text-table/-/text-table-0.2.0.tgz#7f5ee823ae805207c00af2df4a84ec3fcfa570b4" +three@^0.93.0: + version "0.93.0" + resolved "https://registry.yarnpkg.com/three/-/three-0.93.0.tgz#3fd6c367ef4554abbb6e16ad69936283e895c123" + throttleit@0.0.2: version "0.0.2" resolved "https://registry.yarnpkg.com/throttleit/-/throttleit-0.0.2.tgz#cfedf88e60c00dd9697b61fdd2a8343a9b680eaf"