Application: gvSIG desktop » gvSIG legends: base legends

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

  <modelVersion>4.0.0</modelVersion>

  <groupId>org.gvsig</groupId>

  <artifactId>org.gvsig.legend.heatmap.lib.api</artifactId>

  <name>org.gvsig.legend.heatmap.lib.api</name>

  <parent>

    <groupId>org.gvsig</groupId>

    <artifactId>org.gvsig.legend.heatmap.lib</artifactId>

    <version>1.0.0-SNAPSHOT</version>

  </parent>

  <build>

    <plugins>

      <plugin>

        <groupId>org.apache.maven.plugins</groupId>

        <artifactId>maven-jar-plugin</artifactId>

        <configuration>

        </configuration>

        <executions>

          <!-- Generates a jar file only with the test classes -->

          <execution>

            <goals>

              <goal>test-jar</goal>

            </goals>

          </execution>

        </executions>

      </plugin>

    </plugins>

  </build>

  <dependencies>

    <dependency>

      <groupId>org.gvsig</groupId>

      <artifactId>org.gvsig.tools.lib</artifactId>

      <scope>compile</scope>

    </dependency>

    <dependency>

      <groupId>org.gvsig</groupId>

      <artifactId>org.gvsig.fmap.mapcontext.api</artifactId>

      <scope>compile</scope>

    </dependency>

  </dependencies>

</project>

org.gvsig.legend.heatmap.lib.api.HeatmapLegendLibrary

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.legend.heatmap.lib.api;

import org.gvsig.tools.locator.AbstractLocator;

import org.gvsig.tools.locator.Locator;

import org.gvsig.tools.locator.LocatorException;

public class HeatmapLegendLocator extends AbstractLocator {

    /**

     * HeatmapLegend locator name

     */

    private static final String LOCATOR_NAME = "HeatmapLegendLocator";

    /**

     * HeatmapLegend manager name

     */

    public static final String MANAGER_NAME = "HeatmapLegendManager";

    /**

     * HeatmapLegend manager description

     */

    private static final String MANAGER_DESCRIPTION =

        "HeatmapLegend Manager of gvSIG";

    /**

     * Unique instance

     */

    private static final HeatmapLegendLocator instance = new HeatmapLegendLocator();

    @Override

    public String getLocatorName() {

        return LOCATOR_NAME;

    }

    /**

     * Registers the Class implementing the HeatmapLegendManager interface.

     *

     * @param clazz

     *            implementing the HeatmapLegendManager interface

     */

    public static void registerHeatmapLegendManager(Class clazz){

        getInstance().register(MANAGER_NAME, MANAGER_DESCRIPTION, clazz);

    }

    /**

     * Registers the default Class implementing the HeatmapLegendManager interface

     *

     * @param clazz

     *            implementing the HeatmapLegendManager interface

     */

    public static void registerDefaultHeatmapLegendManager(Class clazz){

        getInstance().registerDefault(MANAGER_NAME, MANAGER_DESCRIPTION, clazz);

    }

    /**

     * Return a reference to HeatmapLegendManager.

     *

     * @return a reference to HeatmapLegendManager

     * @throws LocatorException

     *             if there is no access to the class or the class

     *             cannot be instantiated

     * @see Locator#get(String)

     */

    public static HeatmapLegendManager getHeatmapLegendManager() throws LocatorException {

        return (HeatmapLegendManager) getInstance().get(MANAGER_NAME);

    }

    /**

     * @return

     */

    public static Locator getInstance() {

        return instance;

    }

}

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.legend.heatmap.lib.api;

import org.gvsig.tools.library.AbstractLibrary;

import org.gvsig.tools.library.LibraryException;

import org.gvsig.tools.locator.ReferenceNotRegisteredException;

public class HeatmapLegendLibrary extends AbstractLibrary {

    @Override

    protected void doInitialize() throws LibraryException {

        registerAsAPI(HeatmapLegendLibrary.class);

    }

    @Override

    protected void doPostInitialize() throws LibraryException {

        // Validate there is any implementation registered.

        HeatmapLegendManager manager = HeatmapLegendLocator.getHeatmapLegendManager();

        if (manager == null) {

            throw new ReferenceNotRegisteredException(

                HeatmapLegendLocator.MANAGER_NAME, HeatmapLegendLocator.getInstance());

        }

    }

}

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 3

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.legend.heatmap.lib.api;

public interface HeatmapLegendManager {

    /**

     * Creates an aggregate legend

     *

     * @return the aggregate legend

     */

    public HeatmapLegend create();

    public Class<? extends HeatmapLegend> getLegendClass();

}

package org.gvsig.legend.heatmap.lib.api;

import org.gvsig.fmap.mapcontext.rendering.legend.IVectorLegend;

public interface HeatmapLegend extends IVectorLegend {

    public int getDistance();

    public void setDistance(int distance);

}

org.gvsig.legend.heatmap.lib.impl.HeatmapLegendLibraryImpl

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.legend.heatmap.lib.impl;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegend;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegendManager;

public class DefaultHeatmapLegendManager implements HeatmapLegendManager {

    @Override

    public HeatmapLegend create() {

        return new DefaultHeatmapLegend();

    }

    @Override

    public Class<? extends HeatmapLegend> getLegendClass() {

        return DefaultHeatmapLegend.class;

    }

}

package org.gvsig.legend.heatmap.lib.impl;

import java.awt.Color;

import java.awt.Graphics2D;

import java.awt.image.BufferedImage;

import java.util.Map;

import org.cresques.cts.ICoordTrans;

import org.gvsig.fmap.dal.exception.DataException;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureQuery;

import org.gvsig.fmap.dal.feature.FeatureSelection;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.primitive.Point;

import org.gvsig.fmap.mapcontext.MapContextException;

import org.gvsig.fmap.mapcontext.ViewPort;

import org.gvsig.fmap.mapcontext.rendering.legend.LegendException;

import org.gvsig.fmap.mapcontext.rendering.symbols.ISymbol;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegend;

import org.gvsig.symbology.fmap.mapcontext.rendering.legend.impl.AbstractVectorialLegend;

import org.gvsig.symbology.fmap.mapcontext.rendering.legend.impl.DefaultFeatureDrawnNotification;

import org.gvsig.symbology.fmap.mapcontext.rendering.symbol.text.impl.SimpleTextSymbol;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.task.Cancellable;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class DefaultHeatmapLegend extends AbstractVectorialLegend implements HeatmapLegend {

    protected static final Logger LOG = LoggerFactory.getLogger(DefaultHeatmapLegend.class);

    private class DensityAlgorithm {

        private double[][] grid;

        private double[][] kernel;

        private int distance;

        private int height;

        private int with;

        private double maxValue;

        private double minValue;

        public DensityAlgorithm(int distance) {

            this.setDistance(distance);

        }

        public void setDistance(int distance) {

            this.distance = distance;

            this.kernel = new double[2 * this.distance + 1][2 * this.distance + 1];

            for( int y = -this.distance; y < this.distance + 1; y++ ) {

                for( int x = -this.distance; x < this.distance + 1; x++ ) {

                    final double dDist = Math.sqrt(x * x + y * y);

                    if( dDist < this.distance ) {

                        this.kernel[x + this.distance][y + this.distance] = Math.pow(1 - (dDist * dDist) / (this.distance * this.distance), 2);

                    } else {

                        this.kernel[x + this.distance][y + this.distance] = 0;

                    }

                }

            }

        }

        public int getDistance() {

            return this.distance;

        }

        public void init(int with, int height) {

            this.with = with;

            this.height = height;

            this.grid = new double[with][height];

            this.maxValue = 0;

            this.minValue = 0;

        }

        public void add(int px, int py) {

            add(px, py, 1);

        }

        public void add(int px, int py, double value) {

            for( int y = -this.distance; y < this.distance + 1; y++ ) {

                for( int x = -this.distance; x < this.distance + 1; x++ ) {

                    if( this.kernel[x + this.distance][y + this.distance] != 0 ) {

                        addValue(px + x, py + y, value * this.kernel[x + this.distance][y + this.distance]);

                    }

                }

            }

        }

        private void addValue(int px, int py, double value) {

            if( px<0 || py<0 || px>=with || py>=height ) {

                return;

            }

            value += this.grid[px][py];

            this.grid[px][py] = value;

            if( value > this.maxValue ) {

                this.maxValue = value;

            }

            if( value < this.minValue ) {

                this.minValue = value;

            }

        }

        public void draw(BufferedImage img, Color[] colorTable, Cancellable cancel) {

            try {

                int maxColors = colorTable.length;

                for( int x = 0; x < with; x++ ) {

                    for( int y = 0; y < height; y++ ) {

                        if( cancel.isCanceled() ) {

                            return;

                        }

                        double value = this.grid[x][y];

                        if( value > 0 ) {

                            int icolor = (int) (value * maxColors / maxValue);

                            icolor = icolor<0? 0: icolor>=maxColors?maxColors-1:icolor;

                            Color c = colorTable[icolor];

                            img.setRGB(x, y, c.getRGB());

                        }

                    }

                }

            } catch (Exception ex) {

                LOG.warn("Problems drawing heatmap", ex);

            }

        }

    }

    private final ISymbol defaultSymbol;

    private final DensityAlgorithm algorithm;

//    private int distance; // Pixels

    private Color[] colorTable;

    private int opacity;

    public DefaultHeatmapLegend() {

        this.defaultSymbol = new SimpleTextSymbol();

        this.algorithm = new DensityAlgorithm(30);

        this.opacity = 100;

        this.colorTable = createColorTable(255, Color.RED, Color.WHITE);

    }

    @Override

    protected String[] getRequiredFeatureAttributeNames(FeatureStore featureStore) throws DataException {

        return new String[]{

            featureStore.getDefaultFeatureType().getDefaultGeometryAttributeName()};

    }

    @Override

    public ISymbol getDefaultSymbol() {

        return this.defaultSymbol;

    }

    @Override

    public void setDefaultSymbol(ISymbol is) {

    }

    @Override

    public ISymbol getSymbolByFeature(Feature ftr) throws MapContextException {

        return this.defaultSymbol;

    }

    @Override

    public int getShapeType() {

        return Geometry.TYPES.GEOMETRY;

    }

    @Override

    public void setShapeType(int i) {

    }

    @Override

    public boolean isUseDefaultSymbol() {

        return true;

    }

    @Override

    public void useDefaultSymbol(boolean bln) {

    }

    @Override

    public boolean isSuitableForShapeType(int shapeType) {

        return true;

    }

    @Override

    protected void draw(BufferedImage image, Graphics2D g, ViewPort viewPort, Cancellable cancel, double scale, Map queryParameters, ICoordTrans coordTrans, FeatureStore featureStore, FeatureQuery featureQuery, double dpi) throws LegendException {

        this.algorithm.init(image.getWidth(), image.getHeight());

        super.draw(image, g, viewPort, cancel, scale, queryParameters, coordTrans, featureStore, featureQuery, dpi);

        if( !cancel.isCanceled() ) {

            this.algorithm.draw(image, this.colorTable, cancel);

        }

    }

    @Override

    protected void drawFeatures(

        BufferedImage image,

        Graphics2D g,

        final ViewPort viewPort,

        final Cancellable cancel,

        final ICoordTrans coordTrans,

        double dpi,

        DefaultFeatureDrawnNotification drawnNotification,

        FeatureSet featureSet,

        FeatureSelection selection

    ) throws BaseException {

        featureSet.accept(new Visitor() {

            @Override

            public void visit(Object o) throws VisitCanceledException, BaseException {

                if( cancel.isCanceled() ) {

                    throw new VisitCanceledException();

                }

                Feature feature = (Feature) o;

                Geometry geom = feature.getDefaultGeometry();

                if( geom != null ) {

                    Point pointGeo = geom.centroid();

                    if( coordTrans != null ) {

                        pointGeo.reProject(coordTrans);

                    }

                    Point pointPixels = (Point) pointGeo.cloneGeometry();

                    pointPixels.transform(viewPort.getAffineTransform());

                    algorithm.add((int) pointPixels.getX(), (int) pointPixels.getY());

                }

            }

        });

    }

    private Color[] createColorTable(int elements, Color first, Color last) {

        Color[] table = new Color[elements];

        for( int i = 0; i < elements; i++ ) {

            table[i] = new Color(i, 0, 0, i); //, this.opacity);

//            double p = (i * 255.0 ) / elements;

//            int r = (int) (first.getRed() * p + last.getRed() * (1 - p));

//            int g = (int) (first.getGreen()* p + last.getGreen() * (1 - p));

//            int b = (int) (first.getBlue()* p + last.getBlue()* (1 - p));

//            

//            r = r>255?255:r<0?0:r;

//            g = g>255?255:g<0?0:g;

//            b = b>255?255:b<0?0:0;

//            table[i] = new Color(r,g,b, this.opacity);

        }

        return table;

    }

    /**

     * @return the distance

     */

    @Override

    public int getDistance() {

        return this.algorithm.getDistance();

    }

    /**

     * @param distance the distance to set

     */

    @Override

    public void setDistance(int distance) {

        this.algorithm.setDistance(distance);

    }

}

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 gvSIG Association

 *

 * This program is free software; you can redistribute it and/or

 * modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2

 * of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,

 * MA  02110-1301, USA.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.legend.heatmap.lib.impl;

import org.gvsig.fmap.mapcontext.MapContextLibrary;

import org.gvsig.fmap.mapcontext.MapContextLocator;

import org.gvsig.fmap.mapcontext.MapContextManager;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegendLibrary;

import org.gvsig.legend.heatmap.lib.api.HeatmapLegendLocator;

import org.gvsig.tools.library.AbstractLibrary;

import org.gvsig.tools.library.LibraryException;

public class HeatmapLegendLibraryImpl extends AbstractLibrary {

    @Override

    public void doRegistration() {

        registerAsImplementationOf(HeatmapLegendLibrary.class);

        this.require(MapContextLibrary.class);

    }

    @Override

    protected void doInitialize() throws LibraryException {

        HeatmapLegendLocator.registerHeatmapLegendManager(DefaultHeatmapLegendManager.class);

        MapContextManager mcmanager = MapContextLocator.getMapContextManager();

        mcmanager.registerLegend("HeatmapLegend", DefaultHeatmapLegend.class);

    }

    @Override

    protected void doPostInitialize() throws LibraryException {

    }

}

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

  <modelVersion>4.0.0</modelVersion>

  <artifactId>org.gvsig.legend.heatmap.lib.impl</artifactId>

  <name>org.gvsig.legend.heatmap.lib.impl</name>

  <parent>

    <groupId>org.gvsig</groupId>

    <artifactId>org.gvsig.legend.heatmap.lib</artifactId>

    <version>1.0.0-SNAPSHOT</version>

  </parent>

  <groupId>org.gvsig</groupId>

  <dependencies>

    <dependency>

      <groupId>org.gvsig</groupId>

      <artifactId>org.gvsig.legend.heatmap.lib.api</artifactId>

      <scope>compile</scope>

    </dependency>

    <dependency>

      <groupId>org.gvsig</groupId>

      <artifactId>org.gvsig.fmap.dal.api</artifactId>

      <scope>compile</scope>

    </dependency>

    <dependency>

      <groupId>org.gvsig</groupId>

      <artifactId>org.gvsig.symbology.lib.impl</artifactId>

      <version>2.0.197-SNAPSHOT</version>

      <type>jar</type>

    </dependency>

  </dependencies>

</project>

<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">

  <modelVersion>4.0.0</modelVersion>

  <parent>

    <groupId>org.gvsig</groupId>

    <artifactId>org.gvsig.legend.heatmap</artifactId>

    <version>1.0.0-SNAPSHOT</version>

  </parent>

  <groupId>org.gvsig</groupId>

  <artifactId>org.gvsig.legend.heatmap.lib</artifactId>

  <packaging>pom</packaging>

  <modules>

    <module>org.gvsig.legend.heatmap.lib.api</module>

    <module>org.gvsig.legend.heatmap.lib.impl</module>

  </modules>

</project>