Application: gvSIG desktop

/**

 * 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.impl;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Iterator;

import java.util.List;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.SpatialIndex;

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

import org.gvsig.fmap.geom.util.Converter;

import org.gvsig.tools.visitor.Visitor;

public class SpatialIndexJTS implements SpatialIndex {

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

                private Visitor visitor = null;

                public JTSVisitorWrapper(Visitor visitor) {

                        this.visitor = visitor;

                }

                public void visitItem(Object arg0) {

                        try {

                                this.visitor.visit(arg0);

                        } catch (Exception e) {

                                throw new RuntimeException();

                        }

                }

        }

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

        public SpatialIndexJTS() {

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

        }

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

                return Converter.geometryToJts(geom);

        }

        public long size() {

                return this.index.size();

        }

        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 = getJTS(envelope.getGeometry()).getEnvelopeInternal();

                this.index.query(env_jts, visitor_jts);

        }

        public void query(org.gvsig.fmap.geom.Geometry geom, Visitor visitor) {

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

                com.vividsolutions.jts.geom.Envelope env_jts = getJTS(geom).getEnvelopeInternal();

                this.index.query(env_jts, visitor_jts);

        }

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

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

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

                return result.iterator();

        }

        public Iterator queryAll() {

                List result = this.index.queryAll();

                return result.iterator();

        }

        public Iterator query(org.gvsig.fmap.geom.Geometry geom) {

                com.vividsolutions.jts.geom.Envelope env_jts = getJTS(geom).getEnvelopeInternal();

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

                return result.iterator();

        }

        public void insert(org.gvsig.fmap.geom.Geometry geom, Object data) {

                com.vividsolutions.jts.geom.Envelope env_jts = getJTS(geom).getEnvelopeInternal();

                index.insert(env_jts, data);

        }

        public void insert(org.gvsig.fmap.geom.Geometry geom) {

                com.vividsolutions.jts.geom.Envelope env_jts = getJTS(geom).getEnvelopeInternal();

                index.insert(env_jts, geom);

        }

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

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

                index.insert(env_jts, data);

        }

        public boolean remove(org.gvsig.fmap.geom.Geometry geom) {

                com.vividsolutions.jts.geom.Envelope env_jts = getJTS(geom).getEnvelopeInternal();

                return index.remove (env_jts, geom);

        }

        public boolean remove(org.gvsig.fmap.geom.Geometry geom, Object data) {

                com.vividsolutions.jts.geom.Envelope env_jts = getJTS(geom).getEnvelopeInternal();

                return index.remove (env_jts, data);

        }

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

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

                return index.remove (env_jts, data);

        }

        public void removeAll() {

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

        }

        public List queryAsList(Envelope envelope) {

                return IteratorToList( query(envelope) ); 

        }

        public List queryAsList(Geometry geom) {

                return IteratorToList( query(geom) ); 

        }

        public List queryAllAsList() {

                return IteratorToList( queryAll() ); 

        }

        private List IteratorToList(Iterator it) {

                List l = new ArrayList();

                while( it.hasNext() ) {

                        l.add(it.next());

                }

                return l;

        }

}