dhis2-maps-api/src/Choropleth.js

Maps API for DHIS2 based on Leaflet

// Thematic layer

import {GeoJson} from './GeoJson';

export const Choropleth = GeoJson.extend({

    options: {
        style: {
            color: '#333',
            weight: 1,
            fillOpacity: 0.8,
        },
        highlightStyle: {
            weight: 3,
        },
        valueKey: 'value',
        colorKey: 'color',
        radiusKey: 'radius',
    },

    initialize(options = {}) {
        if (!options.pointToLayer) {
            options.pointToLayer = this.pointToLayer.bind(this);
        }
        GeoJson.prototype.initialize.call(this, options);
    },

    addLayer(layer) {
        const feature = layer.feature;
        const prop = feature.properties;
        const color = prop[this.options.colorKey];
        const radius = prop[this.options.radiusKey];

        if (color && layer.setStyle) {
            layer.setStyle({
                fillColor: color,
            });
        }

        if (feature.isSelected) {
            layer.setStyle(this.options.highlightStyle);
        }

        if (radius && layer.setRadius) {
            layer.setRadius(radius);
        }

        GeoJson.prototype.addLayer.call(this, layer);
    },

    // Highlight feature based on id
    highlight(id) {
        const layer = this.findById(id);

        if (layer) {
            return layer.setStyle(this.options.highlightStyle);
        }
    },

    // Remove highlight of feature based on id
    removeHighlight(id) {
        const layer = this.findById(id);

        if (layer) {
            return layer.setStyle(this.options.resetStyle);
        }
    },

    pointToLayer(feature, latlng) {
        return L.circleMarker(latlng, {
            pane: this.options.pane,
        });
    },

});

export default function choropleth(options) {
    return new Choropleth(options);
}