|
| 1 | +import { Component, CUSTOM_ELEMENTS_SCHEMA, Input } from '@angular/core'; |
| 2 | +import { extend, injectBeforeRender, injectNgtRef, NgtRenderState, NgtSignalStore, type NgtGroup } from 'angular-three'; |
| 3 | +import * as THREE from 'three'; |
| 4 | +import { Group } from 'three'; |
| 5 | + |
| 6 | +extend({ Group }); |
| 7 | + |
| 8 | +export type NgtsFloatState = { |
| 9 | + enabled: boolean; |
| 10 | + speed: number; |
| 11 | + floatIntensity: number; |
| 12 | + rotationIntensity: number; |
| 13 | + floatingRange: [number?, number?]; |
| 14 | +}; |
| 15 | + |
| 16 | +declare global { |
| 17 | + interface HTMLElementTagNameMap { |
| 18 | + 'ngts-float': NgtsFloatState & NgtGroup; |
| 19 | + } |
| 20 | +} |
| 21 | + |
| 22 | +@Component({ |
| 23 | + selector: 'ngts-float', |
| 24 | + standalone: true, |
| 25 | + template: ` |
| 26 | + <ngt-group ngtCompound> |
| 27 | + <ngt-group [ref]="floatRef"> |
| 28 | + <ng-content /> |
| 29 | + </ngt-group> |
| 30 | + </ngt-group> |
| 31 | + `, |
| 32 | + schemas: [CUSTOM_ELEMENTS_SCHEMA], |
| 33 | +}) |
| 34 | +export class NgtsFloat extends NgtSignalStore<NgtsFloatState> { |
| 35 | + readonly #offset = Math.random() * 10000; |
| 36 | + |
| 37 | + @Input() floatRef = injectNgtRef<THREE.Group>(); |
| 38 | + |
| 39 | + @Input() set enabled(enabled: boolean) { |
| 40 | + this.set({ enabled }); |
| 41 | + } |
| 42 | + |
| 43 | + @Input() set speed(speed: number) { |
| 44 | + this.set({ speed }); |
| 45 | + } |
| 46 | + |
| 47 | + @Input() set rotationIntensity(rotationIntensity: number) { |
| 48 | + this.set({ rotationIntensity }); |
| 49 | + } |
| 50 | + |
| 51 | + @Input() set floatIntensity(floatIntensity: number) { |
| 52 | + this.set({ floatIntensity }); |
| 53 | + } |
| 54 | + |
| 55 | + @Input() set floatingRange(floatingRange: [number?, number?]) { |
| 56 | + this.set({ floatingRange }); |
| 57 | + } |
| 58 | + |
| 59 | + constructor() { |
| 60 | + super({ speed: 1, rotationIntensity: 1, floatIntensity: 1, floatingRange: [-0.1, 0.1], enabled: true }); |
| 61 | + injectBeforeRender(this.onBeforeRender.bind(this)); |
| 62 | + } |
| 63 | + |
| 64 | + private onBeforeRender({ clock }: NgtRenderState) { |
| 65 | + if (!this.floatRef.untracked) return; |
| 66 | + const { enabled, speed, floatingRange, floatIntensity, rotationIntensity } = this.get(); |
| 67 | + if (!enabled || speed === 0) return; |
| 68 | + |
| 69 | + const t = this.#offset + clock.getElapsedTime(); |
| 70 | + this.floatRef.untracked.rotation.x = (Math.cos((t / 4) * speed) / 8) * rotationIntensity; |
| 71 | + this.floatRef.untracked.rotation.y = (Math.sin((t / 4) * speed) / 8) * rotationIntensity; |
| 72 | + this.floatRef.untracked.rotation.z = (Math.sin((t / 4) * speed) / 20) * rotationIntensity; |
| 73 | + let yPosition = Math.sin((t / 4) * speed) / 10; |
| 74 | + yPosition = THREE.MathUtils.mapLinear(yPosition, -0.1, 0.1, floatingRange[0] ?? -0.1, floatingRange[1] ?? 0.1); |
| 75 | + this.floatRef.untracked.position.y = yPosition * floatIntensity; |
| 76 | + } |
| 77 | +} |
0 commit comments