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.mapcontext.layers;

import java.util.ArrayList;

import java.util.Collections;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.Map;

import org.cresques.cts.IProjection;

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

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

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.GeometryLocator;

import org.gvsig.fmap.geom.SpatialIndex;

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

import org.gvsig.fmap.mapcontext.layers.vectorial.SpatialEvaluatorsFactory;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.DisposableIterator;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.evaluator.Evaluator;

import org.gvsig.tools.i18n.I18nManager;

import org.gvsig.tools.task.AbstractCancellableTask;

import org.gvsig.tools.task.SimpleTaskStatus;

import org.gvsig.tools.visitor.Visitor;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("UseSpecificCatch")

public class SpatialCache {

    private static final Logger logger = LoggerFactory.getLogger(SpatialCache.class);

    private int maxFeatures = 1000; // Por defecto, pero se puede cambiar

    private int fastNumTotalRegs = 0;

    private SpatialIndex index = null;

    private boolean overflown = false;

    private Map<String, Boolean> enabledContexts;

    private boolean enabled = false;

    private AbstractCancellableTask refreshTask;

    public SpatialCache() {

        enabledContexts = new HashMap<>();

        refreshTask = null;

        try {

            this.index = GeometryLocator.getGeometryManager().createDefaultMemorySpatialIndex();

        } catch (Exception e) {

            logger.info("Can't create spatial index", e);

        }

    }

    public int getMaxFeatures() {

        return maxFeatures;

    }

    public void setMaxFeatures(int maxFeatures) {

        this.maxFeatures = maxFeatures;

    }

    public synchronized void insert(Envelope bounds, Geometry geom) {

        if (isEnabled()) {

            if (getMaxFeatures() >= size()) {

                this.index.insert(bounds, geom);

                fastNumTotalRegs++;

            } else {

                overflown = true;

            }

        }

    }

    public synchronized void query(Envelope searchEnv, Visitor visitor) {

        if (!isEnabled()) {

            return;

        }

        this.index.query(searchEnv, visitor);

    }

    public synchronized List query(Envelope searchEnv) {

        if (!isEnabled()) {

            return Collections.EMPTY_LIST;

        }

        List result = new ArrayList();

        Iterator it = index.query(searchEnv);

        while (it.hasNext()) {

            result.add(it.next());

        }

        return result;

    }

    public synchronized void insert(Envelope itemEnv, Object item) {

        if (isEnabled()) {

            if (getMaxFeatures() >= size()) {

                this.index.insert(itemEnv, item);

                fastNumTotalRegs++;

            } else {

                overflown = true;

            }

        }

    }

    public synchronized boolean remove(Envelope itemEnv, Object item) {

        boolean resul = this.index.remove(itemEnv, item);

        if (resul) {

            fastNumTotalRegs--;

        }

        return resul;

    }

    public int size() {

        return fastNumTotalRegs;

    }

    public synchronized void removeAll() {

        if( this.refreshTask!=null ) {

            this.refreshTask.cancelRequest();

        }

        index.removeAll();

        fastNumTotalRegs = 0;

        overflown = false;

    }

    public synchronized void remove(Envelope bounds, Geometry geom) {

        index.remove(bounds, geom);

    }

    public boolean isEnabled() {

        if (enabled) {

            return true;

        }

        for (Boolean value : this.enabledContexts.values()) {

            if (value) {

                return true;

            }

        }

        return false;

    }

    public boolean isEnabled(String contextId) {

        return this.enabledContexts.getOrDefault(contextId, false);

    }

    public void setEnabled(boolean enabled) {

        this.enabled = enabled;

    }

    public void setEnabled(String contextId, boolean enabled) {

        this.enabledContexts.put(contextId, enabled);

    }

    public boolean isOverflown() {

        return overflown;

    }

    public boolean isContextEnabled() {

        for (Boolean value : this.enabledContexts.values()) {

            if (value) {

                return true;

            }

        }

        return false;

    }

    public synchronized void refresh(String name, FeatureStore store, Envelope envelope, IProjection projection) {

        if( store == null  

                || envelope == null || envelope.isEmpty() || envelope.isCollapsed() 

                || projection == null ) {

            return;

        }

        removeAll();

        this.refreshTask = new AbstractCancellableTask("SpatialCacheRefresh_"+name) {

            @Override

            public void run() {

                I18nManager i18n = ToolsLocator.getI18nManager();

                DisposeUtils.bind(store);

                FeatureSet fset = null;

                DisposableIterator<Feature> it = null;

                SimpleTaskStatus status = ToolsLocator.getTaskStatusManager().createDefaultSimpleTaskStatus("SpatialCache "+name);

                try {

                    status.setAutoremove(true);

                    status.add();

                    Evaluator evaluator = SpatialEvaluatorsFactory.getInstance().contains(envelope, projection, store);

                    FeatureQuery query = store.createFeatureQuery();

                    query.addFilter(evaluator);

                    query.setLimit(getMaxFeatures());

                    status.message(i18n.getTranslation("_Searching_geometries"));

                    fset = store.getFeatureSet(query);

                    status.setRangeOfValues(0, fset.size64());

                    it = fset.fastIterator();

                    while( it.hasNext() ) {

                        if( this.isCancellationRequested() || status.isCancellationRequested() ) {

                            status.cancel();

                            return;

                        }

                        Geometry geom = it.next().getDefaultGeometry();

                        insert(geom.getEnvelope(), geom);

                        status.incrementCurrentValue();

                    }

                    status.terminate();

                } catch (Exception ex) {

                    logger.warn("Can't refresh spatial cache of '"+name+"'", ex);

                    status.abort();

                } finally {

                    if( this == refreshTask ) {

                        refreshTask = null;

                    }

                    if( status.isRunning() ) {

                        status.terminate();

                    }

                    DisposeUtils.disposeQuietly(it);

                    DisposeUtils.disposeQuietly(fset);

                    DisposeUtils.disposeQuietly(store);

                }

            }

        };

        this.refreshTask.setPriority((Thread.MAX_PRIORITY+Thread.MIN_PRIORITY)/4);

        this.refreshTask.start();

    }

}