Application: gvSIG desktop

//    

//  Spatial index "Quadtree" is faster in creation but don't 

//  support of nearest query.

//    

//  Spatial index "RTree" is faster in queries and support 

//  nearest queries.

//  

//  The library expose a default implementation for RTree and QuadTree

//    

    public static final String SPATIALINDEX_DEFAULT_RTREE = "JSIRTree";

    public static final String SPATIALINDEX_DEFAULT_QUADTREE = "JTSQuadtree";

import org.gvsig.fmap.geom.jts.spatialindex.SpatialIndexFactoryJSIRTree;

import org.gvsig.fmap.geom.jts.spatialindex.SpatialIndexFactoryJTSQuadtree;

        this.addServiceFactory(new SpatialIndexFactoryJTSQuadtree());

        this.addServiceFactory(new SpatialIndexFactoryJSIRTree());

        return this.createSpatialIndex(SpatialIndexFactoryJTSQuadtree.NAME, null);

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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.fmap.geom.jts.spatialindex;

import com.infomatiq.jsi.IntProcedure;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import org.gvsig.fmap.geom.SpatialIndex;

import org.gvsig.fmap.geom.SpatialIndexFactory;

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

import org.gvsig.tools.visitor.Visitor;

import com.infomatiq.jsi.Rectangle;

import com.infomatiq.jsi.rtree.RTree;

import java.util.Collection;

import java.util.Properties;

import org.gvsig.fmap.geom.GeometryManager;

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

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.exception.BaseException;

public class SpatialIndexJSIRTree extends AbstractSpatialIndex implements SpatialIndex {

    private RTree index = null;

    class Values implements IntProcedure {

        List elements = new ArrayList();

        public boolean execute(int arg0) {

            elements.add(new Integer(arg0));

            return true;

        }

        public List getElements() {

            return elements;

        }

    }

    public SpatialIndexJSIRTree(GeometryManager geometryManager, SpatialIndexFactory factory, DynObject parameters) {

        super(geometryManager, factory, parameters);

        this.index = null;

        this.open();

    }

    public void open() {

        this.index = new RTree();

        Properties props = new Properties();

        if( this.getParameter("MaxNodeEntries")!=null ) {

            props.setProperty("MaxNodeEntries", ((Integer)this.getParameter("MaxNodeEntries")).toString());

        }

        if( this.getParameter("MinNodeEntries")!=null ) {

            props.setProperty("MinNodeEntries", ((Integer)this.getParameter("MinNodeEntries")).toString());

        }

        this.index.init(props);

    }

    public void close() {

        this.index = null;

    }

    private Rectangle asJSI(Envelope envelope) {

        Point min = envelope.getLowerCorner();

        Point max = envelope.getUpperCorner();

        Rectangle rectangle =

            new Rectangle((float) min.getX(), (float) min.getY(),

                (float) max.getX(), (float) max.getY());

        return rectangle;

    }

    public long size() {

        return this.index.size();

    }

    public void query(Envelope envelope, final Visitor visitor) {

        this.index.intersects(asJSI(envelope), new IntProcedure() {

            public boolean execute(int i) {

                try {

                    visitor.visit(new Integer(i));

                } catch (BaseException ex) {

                    return false;

                }

                return true;

            }

        });

    }

    public Iterator query(Envelope envelope, final long limit) {

        final List results = new ArrayList();

        this.index.intersects(asJSI(envelope), new IntProcedure() {

            public boolean execute(int i) {

                results.add(new Integer(i));

                return results.size()<limit;

            }

        });

        return results.iterator();

    }

    public Iterator queryNearest(Envelope envelope, long limit) {

        Collection results = this.index.nearest(asJSI(envelope), (int)limit);

        return results.iterator();

    }

    public Iterator queryAll() {

        return this.index.iterator();

    }

    public void insert(Envelope envelope, Object data) {

        Integer number = (Integer) this.coerceData(data);

        if( number == null ) {

            throw new IllegalArgumentException("null data are not allowed.");

        }

        this.index.add(asJSI(envelope), number.intValue());

    }

    public boolean remove(Envelope envelope, Object data) {

        Integer number = (Integer) this.coerceData(data);

        if( number == null ) {

            throw new IllegalArgumentException("null data are not allowed.");

        }

        return this.index.delete(asJSI(envelope), number.intValue());

    }

    public void removeAll() {

        this.close();

        this.open();

    }

}

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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.fmap.geom.jts.spatialindex;

import com.vividsolutions.jts.index.quadtree.Quadtree;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.util.Iterator;

import java.util.List;

import org.apache.commons.lang3.SerializationUtils;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.SpatialIndex;

import org.gvsig.fmap.geom.SpatialIndexFactory;

import org.gvsig.fmap.geom.jts.GeometryJTS;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.visitor.Visitor;

public class SpatialIndexJTSQuadtree extends AbstractSpatialIndex implements SpatialIndex {

    private com.vividsolutions.jts.index.quadtree.Quadtree index = null;

    private boolean modified;

    public SpatialIndexJTSQuadtree(GeometryManager geometryManager, SpatialIndexFactory factory, DynObject parameters) {

        super(geometryManager, factory, parameters);

        this.index = null;

        this.modified = false;

        open();

    }

    @Override

    public void open() {

        this.index = new com.vividsolutions.jts.index.quadtree.Quadtree();

        File f = (File) this.getParameter("file");

        if( f!=null && f.exists() ) {

            try {

                this.load(f);

            } catch (FileNotFoundException ex) {

                throw new RuntimeException("Can't load spatial index from '"+f.getAbsolutePath()+"'.");

            }

        }

    }

    @Override

    public void close() {

        File f = (File) this.getParameter("file");

        if( f!=null && modified ) {

            try {

                this.save(f);

            } catch (FileNotFoundException ex) {

                throw new RuntimeException("Can't save spatial index to '"+f.getAbsolutePath()+"'.");

            }

        }

        this.index = null;

    }

    private void load(File fname) throws FileNotFoundException {

        FileInputStream is = new FileInputStream(fname);

        Object x = SerializationUtils.deserialize(is);

        this.index = (Quadtree) x;

        this.modified = false;

    }

    private void save(File fname) throws FileNotFoundException {

        FileOutputStream os = new FileOutputStream(fname);

        SerializationUtils.serialize(this.index, os);

        this.modified = false;

    }

    private com.vividsolutions.jts.geom.Geometry asJTS(org.gvsig.fmap.geom.Geometry geom) {

        return ((GeometryJTS)geom).getJTS();

    }

    @Override

    public long size() {

        return this.index.size();

    }

    @Override

    public void query(org.gvsig.fmap.geom.primitive.Envelope envelope,

            Visitor visitor) {

        com.vividsolutions.jts.index.ItemVisitor visitor_jts = new JTSVisitorWrapper(visitor);

        com.vividsolutions.jts.geom.Envelope env_jts = asJTS(envelope.getGeometry()).getEnvelopeInternal();

        this.index.query(env_jts, visitor_jts);

    }

    @Override

    public Iterator query(org.gvsig.fmap.geom.primitive.Envelope envelope,long limit) {

        if( limit!=0 ) {

            throw new UnsupportedOperationException("Not supported yet.");

        }

        com.vividsolutions.jts.geom.Envelope env_jts = asJTS(envelope.getGeometry()).getEnvelopeInternal();

        List result = this.index.query(env_jts);

        return result.iterator();

    }

    @Override

    public Iterator queryNearest(org.gvsig.fmap.geom.primitive.Envelope envelope, long limit) {

        throw new UnsupportedOperationException("Not supported yet.");

    }

    @Override

    public Iterator queryAll() {

        List result = this.index.queryAll();

        return result.iterator();

    }

    @Override

    public void insert(org.gvsig.fmap.geom.primitive.Envelope envelope, Object data) {

        com.vividsolutions.jts.geom.Envelope env_jts = asJTS(envelope.getGeometry()).getEnvelopeInternal();

        index.insert(env_jts, data);

        this.modified = true;

    }

    @Override

    public boolean remove(org.gvsig.fmap.geom.primitive.Envelope envelope, Object data) {

        com.vividsolutions.jts.geom.Envelope env_jts = asJTS(envelope.getGeometry()).getEnvelopeInternal();

        this.modified = true;

        return index.remove(env_jts, data);

    }

    @Override

    public void removeAll() {

        index = new com.vividsolutions.jts.index.quadtree.Quadtree();

        this.modified = true;

    }

    private class JTSVisitorWrapper implements com.vividsolutions.jts.index.ItemVisitor {

        private Visitor visitor = null;

        public JTSVisitorWrapper(Visitor visitor) {

            this.visitor = visitor;

        }

        @Override

        public void visitItem(Object arg0) {

            try {

                this.visitor.visit(arg0);

            } catch (Exception e) {

                throw new RuntimeException();

            }

        }

    }

}

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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.fmap.geom.jts.spatialindex;

import org.gvsig.fmap.geom.DataTypes;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.SpatialIndexFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.DynObjectManager;

import org.gvsig.tools.service.Service;

import org.gvsig.tools.service.ServiceException;

import org.gvsig.tools.service.spi.ServiceManager;

public class SpatialIndexFactoryJSIRTree extends AbstractSpatialIndexFactory implements SpatialIndexFactory {

    public static final String NAME = GeometryManager.SPATIALINDEX_DEFAULT_RTREE;

    //

    // https://github.com/aled/jsi/issues/31

    // RTree only supports 2 dimensions 

    //

    public String getName() {

        return NAME;

    }

    public boolean isNearestQuerySupported() {

        return true;

    }

    public boolean canLimitResults() {

        return true;

    }

    public int getDataTypeSupported() {

        return DataTypes.INT;

    }

    public Service create(DynObject parameters, ServiceManager serviceManager) throws ServiceException {

        return new SpatialIndexJSIRTree((GeometryManager) serviceManager, this, parameters);

    }

    public DynObject createParameters() {

        DynObjectManager manager = ToolsLocator.getDynObjectManager();

        return manager.createDynObject(this.parametersDefinition);

    }

    public void initialize() {

        super.initialize(); 

        if( this.parametersDefinition == null ) {

            this.parametersDefinition = this.getBaseParametersDefinition();

            this.parametersDefinition.addDynFieldInt("MaxNodeEntries")

                    .setDefaultFieldValue(null)

                    .setMandatory(false);

            this.parametersDefinition.addDynFieldInt("MinNodeEntries")

                    .setDefaultFieldValue(null)

                    .setMandatory(false);

        }

    }

}

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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.fmap.geom.jts.spatialindex;

import java.io.File;

import org.gvsig.fmap.geom.GeometryManager;

import org.gvsig.fmap.geom.SpatialIndexFactory;

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.tools.dynobject.DynObject;

import org.gvsig.tools.dynobject.DynStruct;

import org.gvsig.tools.dynobject.impl.DefaultDynObject;

import org.gvsig.tools.service.Service;

import org.gvsig.tools.service.ServiceException;

import org.gvsig.tools.service.spi.ServiceManager;

import org.gvsig.tools.util.HasAFile;

public class SpatialIndexFactoryJTSQuadtree extends AbstractSpatialIndexFactory implements SpatialIndexFactory  {

    public static final String NAME = GeometryManager.SPATIALINDEX_DEFAULT_QUADTREE;

    public static class SpatialIndexJTSParameters extends DefaultDynObject implements HasAFile {

        public SpatialIndexJTSParameters(DynStruct dynClass) {

            super(dynClass);

        }

        @Override

        public File getFile() {

            return (File) this.getDynValue("file");

        }

        @Override

        public void setFile(File file) {

            this.setDynValue("file", file);

        }

    }

    @Override

    public String getName() {

        return NAME;

    }

    @Override

    public int getDataTypeSupported() {

        return DataTypes.OBJECT;

    }

    @Override

    public Service create(DynObject parameters, ServiceManager serviceManager) throws ServiceException {

        return new SpatialIndexJTSQuadtree((GeometryManager) serviceManager, this, parameters);

    }

    @Override

    public DynObject createParameters() {

        return new SpatialIndexJTSParameters(parametersDefinition);

    }

    @Override

    public void initialize() {

        super.initialize(); 

        if( this.parametersDefinition == null ) {

            this.parametersDefinition = this.getBaseParametersDefinition();

            this.parametersDefinition.addDynFieldFile("file").setMandatory(false);

        }

    }

}