@antv/l7-draw/es/source.js

function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }

function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }

function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }

// tslint:disable-next-line:no-submodule-imports
import cloneDeep from 'lodash/cloneDeep';

var DrawSource = /*#__PURE__*/function () {
  function DrawSource(data) {
    _classCallCheck(this, DrawSource);

    this.data = data || this.getDefaultData();
  }

  _createClass(DrawSource, [{
    key: "addFeature",
    value: function addFeature(feature) {
      this.data.features.push(feature);
    }
  }, {
    key: "getData",
    value: function getData() {
      var features = cloneDeep(this.data.features).map(function (feature) {
        var _feature$properties, _feature$properties2;

        feature.properties = {
          id: feature === null || feature === void 0 ? void 0 : (_feature$properties = feature.properties) === null || _feature$properties === void 0 ? void 0 : _feature$properties.id,
          type: feature === null || feature === void 0 ? void 0 : (_feature$properties2 = feature.properties) === null || _feature$properties2 === void 0 ? void 0 : _feature$properties2.type
        };
        return feature;
      });
      return {
        type: 'FeatureCollection',
        features: features
      };
    }
  }, {
    key: "getFeature",
    value: function getFeature(id) {
      var result = this.data.features.find(function (fe) {
        var _fe$properties;

        return (fe === null || fe === void 0 ? void 0 : (_fe$properties = fe.properties) === null || _fe$properties === void 0 ? void 0 : _fe$properties.id) === id;
      });
      return result;
    }
  }, {
    key: "removeAllFeatures",
    value: function removeAllFeatures() {
      this.data = this.getDefaultData();
    }
  }, {
    key: "removeFeature",
    value: function removeFeature(feature) {
      var index = this.getFeatureIndex(feature);

      if (index !== undefined) {
        this.data.features.splice(index, 1);
      }
    }
  }, {
    key: "setFeatureActive",
    value: function setFeatureActive(feature) {
      var _feature$properties3;

      var fe = this.getFeature(feature === null || feature === void 0 ? void 0 : (_feature$properties3 = feature.properties) === null || _feature$properties3 === void 0 ? void 0 : _feature$properties3.id);

      if (fe && fe.properties) {
        fe.properties.active = true;
      }
    }
  }, {
    key: "setFeatureUnActive",
    value: function setFeatureUnActive(feature) {
      var _feature$properties4;

      var fe = this.getFeature(feature === null || feature === void 0 ? void 0 : (_feature$properties4 = feature.properties) === null || _feature$properties4 === void 0 ? void 0 : _feature$properties4.id);

      if (fe && fe.properties) {
        fe.properties.active = false;
      }
    }
  }, {
    key: "clearFeatureActive",
    value: function clearFeatureActive() {
      this.data.features.forEach(function (fe) {
        if (fe && fe.properties) {
          fe.properties.active = false;
        }
      });
    }
  }, {
    key: "updateFeature",
    value: function updateFeature(feature) {
      this.removeFeature(feature);
      this.addFeature(feature);
    }
  }, {
    key: "destroy",
    value: function destroy() {
      this.data = this.getDefaultData();
    }
  }, {
    key: "getDefaultData",
    value: function getDefaultData() {
      return {
        type: 'FeatureCollection',
        features: []
      };
    }
  }, {
    key: "getFeatureIndex",
    value: function getFeatureIndex(feature) {
      return this.data.features.findIndex(function (fe) {
        var _fe$properties2, _feature$properties5;

        return (fe === null || fe === void 0 ? void 0 : (_fe$properties2 = fe.properties) === null || _fe$properties2 === void 0 ? void 0 : _fe$properties2.id) === (feature === null || feature === void 0 ? void 0 : (_feature$properties5 = feature.properties) === null || _feature$properties5 === void 0 ? void 0 : _feature$properties5.id);
      });
    }
  }]);

  return DrawSource;
}();

export { DrawSource as default };