ZhongMianAdmin/__test__/bim/js/tools/AddObjectTool.js

/*
 * AddObjectTool.js
 *
 * @author realor
 */

import { Tool } from './Tool.js'
import { Cord } from '../core/Cord.js'
import { CordGeometry } from '../core/CordGeometry.js'
import { Profile } from '../core/Profile.js'
import { ProfileGeometry } from '../core/ProfileGeometry.js'
import { Solid } from '../core/Solid.js'
import { SolidGeometry } from '../core/SolidGeometry.js'
import { ObjectBuilder } from '../builders/ObjectBuilder.js'
import { HelicoidBuilder } from '../builders/HelicoidBuilder.js'
import { RectangleBuilder } from '../builders/RectangleBuilder.js'
import { CircleBuilder } from '../builders/CircleBuilder.js'
import { CircularSectorBuilder } from '../builders/CircularSectorBuilder.js'
import { Extruder } from '../builders/Extruder.js'
import { Revolver } from '../builders/Revolver.js'
import { I18N } from '../i18n/I18N.js'
import * as THREE from '../lib/three.module.js'

class AddObjectTool extends Tool {
    constructor(application, options) {
        super(application)
        this.name = 'add_object'
        this.label = 'tool.add_object.label'
        this.help = 'tool.add_object.help'
        this.className = 'add_object'
        this.immediate = true
        this.objectType = null
        this.builderClass = null

        this.setOptions(options)
        this.counter = 0
    }

    execute() {
        const objectType = this.objectType || 'Group'
        let object

        switch (objectType) {
            case 'Group':
                object = new THREE.Group()
                break
            case 'Box':
                object = this.createBox()
                break
            case 'Cylinder':
                object = this.createCylinder()
                break
            case 'Sphere':
                object = this.createSphere()
                break
            case 'Spring':
                object = this.createSpring()
                break
            case 'Logo':
                object = this.createLogo()
                break
            case 'Profile':
                object = this.createProfile(this.builderClass)
                break
            case 'Cord':
                object = this.createCord(this.builderClass)
                break
            case 'Sprite':
                object = this.createSprite()
                break
        }
        if (object) {
            this.counter++
            object.name = objectType + '_' + this.counter

            this.application.addObject(object, null, false, true)
        }
    }

    createBox() {
        const solid = new Solid()
        const profile = new Profile()
        profile.name = 'Rectangle'
        profile.builder = new RectangleBuilder(1, 1)
        solid.add(profile)
        solid.builder = new Extruder(1)
        ObjectBuilder.build(solid)
        return solid
    }

    createCylinder() {
        const solid = new Solid()
        const profile = new Profile()
        profile.name = 'Circle'
        profile.builder = new CircleBuilder(0.5, 32)
        solid.add(profile)
        solid.builder = new Extruder(1)
        solid.builder.smoothAngle = 20
        ObjectBuilder.build(solid)
        return solid
    }

    createSphere() {
        const solid = new Solid()
        const profile = new Profile()
        profile.name = 'CircularSector'
        profile.builder = new CircularSectorBuilder(0.5, 180, 32)
        solid.add(profile)
        solid.builder = new Revolver(360)
        solid.builder.axis.x = 1
        solid.builder.axis.y = 0
        solid.builder.smoothAngle = 20
        solid.builder.optimize = false
        ObjectBuilder.build(solid)
        solid.rotation.y = Math.PI / 2

        return solid
    }

    createSpring() {
        const solid = new Solid()
        let cord = new Cord()
        cord.name = 'Helicoid'
        cord.builder = new HelicoidBuilder()
        ObjectBuilder.build(cord)
        solid.add(cord)
        const profile = new Profile()
        profile.name = 'Circle'
        profile.builder = new CircleBuilder(0.2)
        solid.add(profile)
        solid.builder = new Extruder()
        solid.builder.smoothAngle = 30
        ObjectBuilder.build(solid)
        return solid
    }

    createLogo() {
        const solid = new Solid()
        const shape = new THREE.Shape()
        const size = 0.5
        shape.moveTo(-size, -size)
        shape.lineTo(size, -size)
        shape.lineTo(2 * size, 0)
        shape.lineTo(size, size)
        shape.lineTo(-size, size)
        shape.closePath()
        const hole = new THREE.Path()
        hole.moveTo(-size / 2, -size / 2)
        hole.lineTo(size / 2, -size / 2)
        hole.lineTo(size / 2, size / 2)
        hole.lineTo(-size / 2, size / 2)
        hole.closePath()
        shape.holes.push(hole)
        hole = new THREE.Path()
        hole.moveTo(-size / 4 + 0.5, -size / 4)
        hole.lineTo(size / 4 + 0.5, -size / 4)
        hole.lineTo(size / 4 + 0.5, size / 4)
        hole.lineTo(-size / 4 + 0.5, size / 4)
        hole.closePath()
        shape.holes.push(hole)

        const profile = new Profile(new ProfileGeometry(shape))
        solid.add(profile)
        solid.builder = new Extruder(this.height)
        ObjectBuilder.build(solid)
        return solid
    }

    createProfile(profileBuilderClass) {
        const builder = new profileBuilderClass()
        const profile = new Profile()
        profile.builder = builder
        ObjectBuilder.build(profile)
        return profile
    }

    createCord(cordBuilderClass) {
        const builder = new cordBuilderClass()
        const cord = new Cord()
        cord.builder = builder
        ObjectBuilder.build(cord)
        return cord
    }

    createSprite() {
        const manager = this.application.loadingManager
        const textureLoader = new THREE.TextureLoader(manager)
        const texture = textureLoader.load('textures/sphere.png')

        const material = new THREE.SpriteMaterial({
            name: 'sphere',
            map: texture,
            color: 0xffffff,
            sizeAttenuation: false
        })

        const sprite = new THREE.Sprite(material)
        sprite.scale.set(0.1, 0.1, 0.1)
        return sprite
    }
}

export { AddObjectTool }