import { Layer, project32, picking, UNIT } from '@deck.gl/core'; import { Geometry } from '@luma.gl/engine'; import { Model } from '@luma.gl/engine'; import vs from "./text-background-layer-vertex.glsl.js"; import fs from "./text-background-layer-fragment.glsl.js"; const defaultProps = { billboard: true, sizeScale: 1, sizeUnits: 'pixels', sizeMinPixels: 0, sizeMaxPixels: Number.MAX_SAFE_INTEGER, padding: { type: 'array', value: [0, 0, 0, 0] }, getPosition: { type: 'accessor', value: (x) => x.position }, getSize: { type: 'accessor', value: 1 }, getAngle: { type: 'accessor', value: 0 }, getPixelOffset: { type: 'accessor', value: [0, 0] }, getBoundingRect: { type: 'accessor', value: [0, 0, 0, 0] }, getFillColor: { type: 'accessor', value: [0, 0, 0, 255] }, getLineColor: { type: 'accessor', value: [0, 0, 0, 255] }, getLineWidth: { type: 'accessor', value: 1 } }; class TextBackgroundLayer extends Layer { getShaders() { return super.getShaders({ vs, fs, modules: [project32, picking] }); } initializeState() { this.getAttributeManager().addInstanced({ instancePositions: { size: 3, type: 'float64', fp64: this.use64bitPositions(), transition: true, accessor: 'getPosition' }, instanceSizes: { size: 1, transition: true, accessor: 'getSize', defaultValue: 1 }, instanceAngles: { size: 1, transition: true, accessor: 'getAngle' }, instanceRects: { size: 4, accessor: 'getBoundingRect' }, instancePixelOffsets: { size: 2, transition: true, accessor: 'getPixelOffset' }, instanceFillColors: { size: 4, transition: true, type: 'unorm8', accessor: 'getFillColor', defaultValue: [0, 0, 0, 255] }, instanceLineColors: { size: 4, transition: true, type: 'unorm8', accessor: 'getLineColor', defaultValue: [0, 0, 0, 255] }, instanceLineWidths: { size: 1, transition: true, accessor: 'getLineWidth', defaultValue: 1 } }); } updateState(params) { super.updateState(params); const { changeFlags } = params; if (changeFlags.extensionsChanged) { this.state.model?.destroy(); this.state.model = this._getModel(); this.getAttributeManager().invalidateAll(); } } draw({ uniforms }) { const { billboard, sizeScale, sizeUnits, sizeMinPixels, sizeMaxPixels, getLineWidth } = this.props; let { padding } = this.props; if (padding.length < 4) { padding = [padding[0], padding[1], padding[0], padding[1]]; } const model = this.state.model; model.setUniforms(uniforms); model.setUniforms({ billboard, stroked: Boolean(getLineWidth), padding, sizeUnits: UNIT[sizeUnits], sizeScale, sizeMinPixels, sizeMaxPixels }); model.draw(this.context.renderPass); } _getModel() { // a square that minimally cover the unit circle const positions = [0, 0, 1, 0, 1, 1, 0, 1]; return new Model(this.context.device, { ...this.getShaders(), id: this.props.id, bufferLayout: this.getAttributeManager().getBufferLayouts(), geometry: new Geometry({ topology: 'triangle-fan-webgl', vertexCount: 4, attributes: { positions: { size: 2, value: new Float32Array(positions) } } }), isInstanced: true }); } } TextBackgroundLayer.defaultProps = defaultProps; TextBackgroundLayer.layerName = 'TextBackgroundLayer'; export default TextBackgroundLayer;