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.dal.feature.impl.featureset;

import java.util.HashSet;

import java.util.Iterator;

import java.util.Set;

import java.util.logging.Level;

import java.util.logging.Logger;

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

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

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

import org.gvsig.fmap.dal.feature.exception.ConcurrentDataModificationException;

import org.gvsig.fmap.dal.feature.impl.DefaultFeature;

import org.gvsig.fmap.dal.feature.impl.DefaultFeatureStore;

import org.gvsig.fmap.dal.feature.impl.editing.memory.FeatureManager;

import org.gvsig.fmap.dal.feature.impl.featurereference.FeatureReferenceFactory;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.feature.spi.FeatureSetProvider;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.util.ChainedIterator;

/**

 * Iterator for stores in edition mode.

 *

 * @author gvSIG Team

 * @version $Id$

 * @deprecated use the {@link FastEditedIterator} instead

 */

public class EditedIterator extends FilteredIterator {

    private DefaultFeatureStore store;

    protected FeatureManager featureManager;

    private Set<FeatureReference> visitedReferences;

    public EditedIterator(DefaultFeatureSet featureSet) {

        super(featureSet);

        this.store = featureSet.store;

        this.featureManager = this.store.getFeatureManager();

        this.visitedReferences = new HashSet<>();

    }

    public EditedIterator(DefaultFeatureSet featureSet, long index, long elements)

        throws DataException {

        this(featureSet);

        Iterator<EditableFeature> insertedFeatures = this.featureManager.getInsertedFeatures();//Don't use getInserteds because referenceFeature will be missed

        Iterator<EditableFeature> updatedFeatures = this.featureManager.getUpdatedFeatures();

        final Iterator<FeatureProvider> providerIterator;

        FeatureSetProvider provider = featureSet.provider;

        boolean canDelegateInProvider = provider.canFilter() && provider.canIterateFromIndex() && !featureManager.hasDeleteds();

        if (canDelegateInProvider) {

            providerIterator = provider.iterator(index, elements);

        } else {

            providerIterator = provider.iterator();

        }

        Iterator<FeatureProvider> wrappedProviderIterator = new Iterator<FeatureProvider>() {

            @Override

            public boolean hasNext() {

                return providerIterator.hasNext();

            }

            @Override

            public FeatureProvider next() {

                FeatureProvider data = providerIterator.next();

                FeatureReference ref = FeatureReferenceFactory.createFromFeatureProvider(store, data);

                DefaultFeature modified;

                try {

                    modified = featureManager.get(ref, store);

                } catch (DataException ex) {

                    return data;

                }

                if (modified != null) {

                    visitedReferences.add(ref);

                    return modified.getData();

                }

                return data;

            }

            @Override

            public void remove() {

                EditedIterator.super.remove();

            }

        };

        this.iterator = new ChainedIterator(wrappedProviderIterator, insertedFeatures, updatedFeatures);

        if (!canDelegateInProvider) {

            if (index > 0) {

                skypto(index);

            }

        }

    }

    @Override

    public void remove() {

        this.iterator.remove();

    }

    @Override

    public boolean hasNext() {

        if (store.isEditing()) {

            return super.hasNext();

        } else {

            throw new ConcurrentDataModificationException(store.getFullName());

        }

    }

    @Override

    public Object next() {

        if (store.isEditing()) {

            return super.next();

        } else {

            throw new ConcurrentDataModificationException(store.getFullName());

        }

    }

    @Override

    protected ChainedIterator getIterator() {

        return (ChainedIterator) this.iterator;

    }

    @Override

    protected DefaultFeature createFeature(FeatureProvider data) throws DataException {

        DefaultFeature f = this.featureManager.get(data);

//        DefaultFeature f = null;

//        try {

//            FeatureReference ref = new DefaultFeatureReference(store, data);

//            f= (DefaultFeature) this.featureManager.get(ref, store, data.getType());

//        } catch (DataException e) {

//            RuntimeException ex = new RuntimeException();

//            e.initCause(e);

//            throw ex;

//        }

        if (f == null) {

            // La feature no se ha editado.

            f = new DefaultFeature(store, data);

        }

        if (this.fset.transform.isEmpty()) {

            return f;

        } else {

            return (DefaultFeature) this.fset.transform.applyTransform(

                    f, fset.getDefaultFeatureType());

        }

    }

    @Override

    protected boolean skipFeature(FeatureProvider data) {

        try {

            switch (getIterator().getCurrent()) {

                case 0:

                    if (this.featureManager.isDeleted(data)) {

                        return true;

                    }

                    break;

                case 2:

                    FeatureReference ref = FeatureReferenceFactory.createFromFeatureProvider(store, data);

                    if (visitedReferences.contains(ref)){

                        return true;

                    }

                    break;

            }

        } catch (Exception ex) {

            LOGGER.warn("Can't check if must skip feature.", ex);

        }

        return false;

    }

    @Override

    protected void doDispose() throws BaseException {

        super.doDispose();

        this.featureManager = null;

        this.store = null;

    }

}