UNPKG

@deck.gl/geo-layers

Version:

deck.gl layers supporting geospatial use cases and GIS formats

207 lines (174 loc) 8.15 kB
"use strict"; var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault"); Object.defineProperty(exports, "__esModule", { value: true }); exports.default = void 0; var _defineProperty2 = _interopRequireDefault(require("@babel/runtime/helpers/defineProperty")); var _classCallCheck2 = _interopRequireDefault(require("@babel/runtime/helpers/classCallCheck")); var _createClass2 = _interopRequireDefault(require("@babel/runtime/helpers/createClass")); var _get2 = _interopRequireDefault(require("@babel/runtime/helpers/get")); var _inherits2 = _interopRequireDefault(require("@babel/runtime/helpers/inherits")); var _possibleConstructorReturn2 = _interopRequireDefault(require("@babel/runtime/helpers/possibleConstructorReturn")); var _getPrototypeOf2 = _interopRequireDefault(require("@babel/runtime/helpers/getPrototypeOf")); var _experimental = require("@luma.gl/experimental"); var _core = require("@luma.gl/core"); var _meshLayers = require("@deck.gl/mesh-layers"); var _meshLayerVertex = _interopRequireDefault(require("./mesh-layer-vertex.glsl")); var _meshLayerFragment = _interopRequireDefault(require("./mesh-layer-fragment.glsl")); function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); if (enumerableOnly) { symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; }); } keys.push.apply(keys, symbols); } return keys; } function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i] != null ? arguments[i] : {}; if (i % 2) { ownKeys(Object(source), true).forEach(function (key) { (0, _defineProperty2.default)(target, key, source[key]); }); } else if (Object.getOwnPropertyDescriptors) { Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)); } else { ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } } return target; } function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = (0, _getPrototypeOf2.default)(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = (0, _getPrototypeOf2.default)(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return (0, _possibleConstructorReturn2.default)(this, result); }; } function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } } function validateGeometryAttributes(attributes) { var hasColorAttribute = attributes.COLOR_0 || attributes.colors; if (!hasColorAttribute) { attributes.colors = { constant: true, value: new Float32Array([1, 1, 1]) }; } } var defaultProps = { pbrMaterial: { type: 'object', value: null }, featureIds: { type: 'array', value: null, optional: true } }; var _MeshLayer = function (_SimpleMeshLayer) { (0, _inherits2.default)(_MeshLayer, _SimpleMeshLayer); var _super = _createSuper(_MeshLayer); function _MeshLayer() { (0, _classCallCheck2.default)(this, _MeshLayer); return _super.apply(this, arguments); } (0, _createClass2.default)(_MeshLayer, [{ key: "getShaders", value: function getShaders() { var shaders = (0, _get2.default)((0, _getPrototypeOf2.default)(_MeshLayer.prototype), "getShaders", this).call(this); var modules = shaders.modules; modules.push(_core.pbr); return _objectSpread(_objectSpread({}, shaders), {}, { vs: _meshLayerVertex.default, fs: _meshLayerFragment.default }); } }, { key: "initializeState", value: function initializeState() { var featureIds = this.props.featureIds; (0, _get2.default)((0, _getPrototypeOf2.default)(_MeshLayer.prototype), "initializeState", this).call(this); if (featureIds) { this.state.attributeManager.add({ featureIdsPickingColors: { type: 5121, size: 3, noAlloc: true, update: this.calculateFeatureIdsPickingColors } }); } } }, { key: "updateState", value: function updateState(_ref) { var props = _ref.props, oldProps = _ref.oldProps, changeFlags = _ref.changeFlags; (0, _get2.default)((0, _getPrototypeOf2.default)(_MeshLayer.prototype), "updateState", this).call(this, { props: props, oldProps: oldProps, changeFlags: changeFlags }); if (props.pbrMaterial !== oldProps.pbrMaterial) { this.updatePbrMaterialUniforms(props.pbrMaterial); } } }, { key: "draw", value: function draw(opts) { var featureIds = this.props.featureIds; if (!this.state.model) { return; } this.state.model.setUniforms({ u_Camera: this.state.model.getUniforms().project_uCameraPosition, u_pickFeatureIds: Boolean(featureIds) }); (0, _get2.default)((0, _getPrototypeOf2.default)(_MeshLayer.prototype), "draw", this).call(this, opts); } }, { key: "getModel", value: function getModel(mesh) { var _this$props = this.props, id = _this$props.id, pbrMaterial = _this$props.pbrMaterial; var materialParser = this.parseMaterial(pbrMaterial, mesh); var shaders = this.getShaders(); validateGeometryAttributes(mesh.attributes); var model = new _core.Model(this.context.gl, _objectSpread(_objectSpread({}, this.getShaders()), {}, { id: id, geometry: mesh, defines: _objectSpread(_objectSpread(_objectSpread({}, shaders.defines), materialParser === null || materialParser === void 0 ? void 0 : materialParser.defines), {}, { HAS_UV_REGIONS: mesh.attributes.uvRegions }), parameters: materialParser === null || materialParser === void 0 ? void 0 : materialParser.parameters, isInstanced: true })); return model; } }, { key: "updatePbrMaterialUniforms", value: function updatePbrMaterialUniforms(pbrMaterial) { var model = this.state.model; if (model) { var mesh = this.props.mesh; var materialParser = this.parseMaterial(pbrMaterial, mesh); model.setUniforms(materialParser.uniforms); } } }, { key: "parseMaterial", value: function parseMaterial(pbrMaterial, mesh) { var unlit = Boolean(pbrMaterial.pbrMetallicRoughness && pbrMaterial.pbrMetallicRoughness.baseColorTexture); var materialParser = new _experimental.GLTFMaterialParser(this.context.gl, { attributes: { NORMAL: mesh.attributes.normals, TEXCOORD_0: mesh.attributes.texCoords }, material: _objectSpread({ unlit: unlit }, pbrMaterial), pbrDebug: false, imageBasedLightingEnvironment: null, lights: true, useTangents: false }); return materialParser; } }, { key: "calculateFeatureIdsPickingColors", value: function calculateFeatureIdsPickingColors(attribute) { var featureIds = this.props.featureIds; var value = new Uint8ClampedArray(featureIds.length * attribute.size); var pickingColor = []; for (var index = 0; index < featureIds.length; index++) { this.encodePickingColor(featureIds[index], pickingColor); value[index * 3] = pickingColor[0]; value[index * 3 + 1] = pickingColor[1]; value[index * 3 + 2] = pickingColor[2]; } attribute.value = value; } }]); return _MeshLayer; }(_meshLayers.SimpleMeshLayer); exports.default = _MeshLayer; _MeshLayer.layerName = '_MeshLayer'; _MeshLayer.defaultProps = defaultProps; //# sourceMappingURL=mesh-layer.js.map