three.cad/src/Scene.js

311 lines
7.3 KiB
JavaScript

import * as THREE from '../node_modules/three/src/Three';
import { TrackballControls } from './trackball'
import { Sketch } from './Sketch'
import Stats from './stats.module.js';
import { add3DPoint } from './datums'
import { extrude } from './extrude'
import { onHover, onPick } from './mouseEvents';
import { _vec2, _vec3, color, awaitSelection } from './shared'
import {AxesHelper} from './axes'
import CSG from "./three-csg.js"
window.BoolOp = CSG
window.th = THREE
const eq = (a1, a2) => {
if (a1.length != a2.length) return false
for (let i = 0; i < a1.length; i++) {
if (a1[i] != a2[i]) return false
}
return true
}
window.loader = new THREE.ObjectLoader();
window.id = 0
export class Scene {
constructor(store) {
this.store = store;
this.canvas = document.querySelector('#c');
this.rect = this.canvas.getBoundingClientRect().toJSON()
this.renderer = new THREE.WebGLRenderer({ canvas: this.canvas });
const size = 1;
const near = 0;
const far = 100;
this.camera = new THREE.OrthographicCamera(-size, size, size, -size, near, far);
this.camera.zoom = 0.1;
const cameraDist = 50
const xzAngle = 30 * Math.PI / 180
this.camera.position.set(
cameraDist * Math.sin(xzAngle),
cameraDist * Math.tan(30 * Math.PI / 180),
cameraDist * Math.cos(xzAngle)
);
// const controls = new OrbitControls(camera, view1Elem);
const controls = new TrackballControls(this.camera, this.canvas);
controls.target.set(0, 0, 0);
controls.update();
this.obj3d = new THREE.Scene()
this.obj3d.background = new THREE.Color(color.background);
const helpersGroup = new THREE.Group();
helpersGroup.name = "helpersGroup";
this.obj3d.add(helpersGroup);
this.axes = new AxesHelper(this.camera.zoom)
this.axes.visible = false
helpersGroup.add(this.axes);
const planeGeom = new THREE.PlaneGeometry(5, 5)
const pxy = new THREE.Mesh(
planeGeom,
new THREE.MeshBasicMaterial({
color: color.plane,
opacity: 0.05,
side: THREE.DoubleSide,
transparent: true,
depthWrite: false,
toneMapped: false
})
);
pxy.userData.type = 'plane'
pxy.add(
new THREE.LineSegments(
new THREE.EdgesGeometry(planeGeom),
new THREE.LineBasicMaterial({ color: color.planeBorder })
)
)
const pyz = pxy.clone().rotateY(Math.PI / 2);
pyz.material = pyz.material.clone();
const pxz = pxy.clone().rotateX(-Math.PI / 2);
pxz.material = pxz.material.clone();
helpersGroup.add(pxy);
helpersGroup.add(pyz);
helpersGroup.add(pxz);
const intensity = 1;
const light1 = new THREE.DirectionalLight(color.lighting, intensity);
light1.position.set(10, 10, 10);
this.obj3d.add(light1);
const light2 = new THREE.DirectionalLight(color.lighting, intensity);
light2.position.set(-10, -10, -5);
this.obj3d.add(light2);
const ambient = new THREE.AmbientLight(color.lighting, intensity);
this.obj3d.add(ambient);
this.render = render.bind(this);
this.addSketch = addSketch.bind(this);
this.extrude = extrude.bind(this);
this.onHover = onHover.bind(this);
this.onPick = onPick.bind(this);
this.awaitSelection = awaitSelection.bind(this);
this.obj3d.addEventListener('change', this.render);
controls.addEventListener('change', this.render);
controls.addEventListener('start', this.render);
window.addEventListener('resize', this.render);
this.stats = new Stats();
this.stats.showPanel(0); // 0: fps, 1: ms, 2: mb, 3+: custom
document.getElementById('stats').appendChild(this.stats.dom);
this.hovered = [];
this.selected = [];
this.activeSketch = null;
this.render();
}
resizeCanvas(renderer) {
const canvas = renderer.domElement;
const width = canvas.clientWidth;
const height = canvas.clientHeight;
const needResize = canvas.width !== width || canvas.height !== height;
if (needResize) {
renderer.setSize(width, height, false);
}
return needResize;
}
saveState() {
localStorage.setItem(
'sv', JSON.stringify([id, this.store.getState()])
)
}
loadState() { //uglyyy
const [curid, state] = JSON.parse(
localStorage.getItem('sv')
)
window.id = curid
const entries = state.treeEntries.byId
for (let k in entries) {
if (k[0] == 's') {
entries[k].obj3d = loader.parse(entries[k].obj3d)
this.obj3d.add(entries[k].obj3d)
entries[k] = new Sketch(this, state.treeEntries.byId[k])
entries[k].obj3d.addEventListener('change', this.render) // !! took 3 hours to realize
} else if (k[0] == 'm') {
entries[k] = loader.parse(state.treeEntries.byId[k])
console.log(entries[k])
this.obj3d.add(entries[k])
}
}
this.store.dispatch({ type: 'restore-state', state })
}
clearSelection() {
for (let x = 0; x < this.selected.length; x++) {
const obj = this.selected[x]
obj.material.color.set(color[obj.userData.type])
}
for (let x = 0; x < this.hovered.length; x++) {
const obj = this.selected[x]
obj.material.color.set(color[obj.userData.type])
}
this.obj3d.dispatchEvent({ type: 'change' })
this.selected = []
console.log('fireed')
}
}
let idx, x, y, ele, pos, dims, matrix;
function render() {
this.stats.begin();
if (this.resizeCanvas(this.renderer)) {
const canvas = this.renderer.domElement;
this.camera.left = -canvas.clientWidth / canvas.clientHeight;
this.camera.right = canvas.clientWidth / canvas.clientHeight;
this.camera.updateProjectionMatrix();
Object.assign(this.rect, this.canvas.getBoundingClientRect().toJSON())
}
this.axes.resize(this.camera.zoom)
this.renderer.render(this.obj3d, this.camera);
if (this.activeSketch) {
dims = this.activeSketch.obj3d.children[1].children
matrix = this.activeSketch.obj3d.matrix
for (idx = 1; idx < dims.length; idx += 2) {
ele = dims[idx]
pos = _vec3.set(
...ele.geometry.attributes.position.array
).applyMatrix4(matrix).project(this.camera)
x = (pos.x * .5 + .5) * this.canvas.clientWidth + 10 + this.rect.left;
y = (pos.y * -.5 + .5) * this.canvas.clientHeight + this.rect.top;
ele.label.style.transform = `translate(0%, -50%) translate(${x}px,${y}px)`;
}
}
this.stats.end();
}
async function addSketch() {
let sketch;
const references = await this.awaitSelection({ point: 3 }, { plane: 1 });
if (!references) return;
if (references[0].userData.type == 'plane') {
sketch = new Sketch(this)
sketch.obj3d.matrix = references[0].matrix
sketch.plane.applyMatrix4(sketch.obj3d.matrix)
sketch.obj3d.inverse = sketch.obj3d.matrix.clone().invert()
this.obj3d.add(sketch.obj3d)
} else {
sketch = new Sketch(this)
this.obj3d.add(sketch.obj3d)
sketch.align(
...references.map(
el => new THREE.Vector3(...el.geometry.attributes.position.array).applyMatrix4(el.matrixWorld)
)
)
}
this.clearSelection()
sketch.activate()
this.activeSketch = sketch
sketch.obj3d.addEventListener('change', this.render);
this.render()
console.log('render')
this.store.dispatch({ type: 'rx-sketch', obj: sketch })
}
window.sc = new Scene(store)
// sc.loadState()