| gvSIG desktop 1

/* gvSIG. Geographic Information System of the Valencian Government

 *

 * Copyright (C) 2007-2008 Infrastructures and Transports Department

 * of the Valencian Government (CIT)

 * 

 * 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 2

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

 * 

 */

import org.gvsig.tools.dispose.DisposableIterator;

/**

 * Iterator for stores in edition mode.

 * 

 * @author gvSIG Team

 * @version $Id$

 */

    Iterator newsFeatures;

    boolean featureIsNew;

    public EditedIterator(DefaultFeatureSet featureSet) {

        super(featureSet);

    }

    public EditedIterator(DefaultFeatureSet featureSet, long index)

        throws DataException {

        super(featureSet);

        this.newsFeatures = null;

        if (index > 0) {

            if (featureSet.provider.canIterateFromIndex()) {

                try {

                    this.iterator = featureSet.provider.iterator(index);

                } catch (IllegalArgumentException e) {

                    this.iterator = featureSet.provider.iterator();

                    skypto(index);

                } catch (UnsupportedOperationException e) {

                    this.iterator = featureSet.provider.iterator();

                    skypto(index);

                }

            } else {

                this.iterator = featureSet.provider.iterator();

                skypto(index);

            }

        } else {

            this.iterator = featureSet.provider.iterator();

        }

    }

    protected Iterator getIterator() {

        if (this.featureIsNew) {

            return this.newsFeatures;

        } else

            if (this.iterator.hasNext()) {

                featureIsNew = false;

                return this.iterator;

            } else {

                featureIsNew = true;

                this.newsFeatures =

                    this.fset.store.getFeatureManager().getInserted();

                return this.newsFeatures;

            }

    }

    protected DefaultFeature createFeature(FeatureProvider data) {

        DefaultFeature f = null;

        data.setNew(featureIsNew);

        try {

            FeatureReference ref =

                new DefaultFeatureReference(fset.store, data);

            f =

                (DefaultFeature) fset.store.getFeatureManager().get(ref,

                    fset.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(fset.store, data);

        }

        return f;

    }

    protected boolean isDeletedOrHasToSkip(FeatureProvider data) {

        // XXX

        // si recorriendo los originales nos

        // encontramos uno nuevo, no lo devolvemos

        // porque se recorrera mas tarde.

        // Esto es una interaccion con los indices

        // ya que estos contienen todas las features ya

        // sea nuevas o no

        if (data.isNew() && !featureIsNew) {

            return true;

        }

        FeatureReference ref = new DefaultFeatureReference(fset.store, data);

        return fset.store.getFeatureManager().isDeleted(ref);

    }

    protected void doDispose() throws BaseException {

        super.doDispose();

        if (newsFeatures instanceof DisposableIterator) {

            ((DisposableIterator) newsFeatures).dispose();

        }

        newsFeatures = null;

    }