Application: gvSIG desktop

 *

 * $Id: RTreeJsi.java 13884 2007-09-19 16:26:04Z jaume $

 * $Log$

 * Revision 1.5  2007-09-19 16:25:39  jaume

 * ReadExpansionFileException removed from this context and removed unnecessary imports

 *

 * Revision 1.4  2007/06/27 20:17:30  azabala

 * new spatial index (rix)

 *

 * Revision 1.3  2007/03/06 17:08:59  caballero

 * Exceptions

 *

 * Revision 1.2  2006/06/05 16:59:08  azabala

 * implementada busqueda de vecino mas proximo a partir de rectangulos

 *

 * Revision 1.1  2006/05/24 21:58:04  azabala

 * *** empty log message ***

 *

 *

 */

import com.infomatiq.jsi.IntProcedure;

import com.infomatiq.jsi.Rectangle;

import com.infomatiq.jsi.rtree.RTree;

 * 

 * 

 * 

 * 

 * 

 * 

 * 

 * 

 * 

public class JSIRTree extends AbstractFeatureIndexProvider implements

    FeatureIndexProvider {

    public static final String NAME = JSIRTree.class.getSimpleName();

    protected RTree rtree;

    public JSIRTree() {

        rtree = new RTree();

    }

    public void initialize() throws InitializeException {

        Properties props = new Properties();

        // props.setProperty("MaxNodeEntries", "500");

        // props.setProperty("MinNodeEntries", "200");

        rtree.init(props);

    }

    class ListIntProcedure implements IntProcedure {

        ArrayList solution = new ArrayList();

        public boolean execute(int arg0) {

            solution.add(new Integer(arg0));

            return true;

        }

        public List getSolution() {

            return solution;

        }

    }

    protected List findNNearest(int numberOfNearest, Point point) {

        com.infomatiq.jsi.Point jsiPoint =

            new com.infomatiq.jsi.Point((float) point.getX(),

                (float) point.getY());

        return (List) rtree.nearest(jsiPoint, numberOfNearest);

    }

    public Iterator iterator() {

        return rtree.iterator();

    }

    public int size() {

        return rtree.size();

    }

    protected Rectangle toJsiRect(Envelope env) {

        Point min = env.getLowerCorner();

        Point max = env.getUpperCorner();

        Rectangle jsiRect =

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

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

        return jsiRect;

    }

    public void insert(Object value, FeatureReferenceProviderServices fref) {

        Envelope env = getEnvelope(value);

        if (env == null) {

            throw new IllegalArgumentException(

                "value is neither Geometry or Envelope");

        }

        Object oid = fref.getOID();

        if (!isCompatibleOID(oid)) {

            throw new IllegalArgumentException("OID type not compatible: "

                + oid.getClass().getName());

        }

        rtree.add(toJsiRect(env), ((Number) oid).intValue());

    }

    public void delete(Object value, FeatureReferenceProviderServices fref) {

        Envelope env = getEnvelope(value);

        if (env == null) {

            throw new IllegalArgumentException(

                "value is neither Geometry or Envelope");

        }

        Object oid = fref.getOID();

        if (!isCompatibleOID(oid)) {

            throw new IllegalArgumentException("OID type not compatible: "

                + oid.getClass().getName());

        }

        rtree.delete(toJsiRect(env), ((Number) oid).intValue());

    }

    public List match(Object value) throws FeatureIndexException {

        Envelope env = getEnvelope(value);

        if (env == null) {

            throw new IllegalArgumentException(

                "value is neither Geometry or Envelope");

        }

        ListIntProcedure solution = new ListIntProcedure();

        rtree.intersects(toJsiRect(env), solution);

        return new LongList(solution.getSolution());

    }

    public List nearest(int count, Object value) {

        if (value == null) {

            throw new IllegalArgumentException("value is null");

        }

        if (value instanceof Point) {

            Point p = (Point) value;

            com.infomatiq.jsi.Point jsiPoint =

                new com.infomatiq.jsi.Point((float) p.getDirectPosition()

                    .getOrdinate(0), (float) p.getDirectPosition().getOrdinate(

                    1));

            return (List) rtree.nearest(jsiPoint, count);

        } else {

            Envelope env = getEnvelope(value);

            if (env == null) {

                throw new IllegalArgumentException(

                    "value is neither Geometry or Envelope");

            }

            return new LongList((List) rtree.nearest(toJsiRect(env), count));

        }

    }

    public boolean isMatchSupported() {

        return true;

    }

    public boolean isNearestSupported() {

        return true;

    }

    public boolean isNearestToleranceSupported() {

        return false;

    }

    public boolean isRangeSupported() {

        return false;

    }

    public List nearest(int count, Object value, Object tolerance)

        throws FeatureIndexException {

        throw new UnsupportedOperationException();

    }

    public List range(Object value1, Object value2)

        throws FeatureIndexException {

        throw new UnsupportedOperationException();

    }

    /**

     * Indicates whether the given OID's type is compatible

     * with this provider

     * 

     * @param oid

     * 

     * @return

     *         true if this index provider supports the given oid type

     */

    private boolean isCompatibleOID(Object oid) {

        if (!(oid instanceof Number)) {

            return false;

        }

        long num = ((Number) oid).longValue();

        if (num > Integer.MAX_VALUE || num < Integer.MIN_VALUE) {

            return false;

        }

        return true;

    }

    protected Envelope getEnvelope(Object value) {

        Envelope env = null;

        if (value instanceof Envelope) {

            env = (Envelope) value;

        } else

            if (value instanceof Geometry) {

                env = ((Geometry) value).getEnvelope();

            }

        return env;

    }