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 org.gvsig.fmap.dal.exception.DataException;

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

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

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

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

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

import org.gvsig.tools.exception.BaseException;

/**

 * Iterator implementation which shares the returned Feature object instance.

 * 

 * @author gvSIG Team

 * @version $Id$

 */

public class FastEditedIterator extends EditedIterator {

    private DefaultFeature myFeature;

    public FastEditedIterator(DefaultFeatureSet featureSet, long index)

        throws DataException {

        super(featureSet, index);

    }

    @Override

    protected DefaultFeature createFeature(FeatureProvider data) throws DataException {

        if(myFeature == null){

            myFeature = new DefaultFeature(fset.store);

        }

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

//        DefaultFeature f = null;

//        try {

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

//            f = (DefaultFeature) featureManager.get(ref, fset.store); 

//        } catch (DataException e) {

//            RuntimeException ex = new RuntimeException();

//            e.initCause(e);

//            throw ex;

//        }

        if (f == null) {

            this.myFeature.setData(data);

        } else {

            // TODO Sacamos una copia del data o no???

            this.myFeature.setData(f.getData().getCopy());

        }

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

            return myFeature;

        } else {

            if (f == null) {

                myFeature = (DefaultFeature)

                    this.fset.transform.applyTransform(myFeature,

                        fset.getDefaultFeatureType());

            } else {

                /*

                 * In this case "f != null" the data comes from the featureManager,

                 * so it is data inserted/updated by the user, so we must

                 * overwrite the data after applying the transformations

                 * because the transformations can be "Adding field"

                 * and are necessary in some cases:

                 */

                DefaultFeature saved_feat = (DefaultFeature) myFeature.getCopy();

                myFeature = (DefaultFeature)

                    this.fset.transform.applyTransform(myFeature,

                        fset.getDefaultFeatureType());

                myFeature = overwrite(myFeature, saved_feat);

            }

            return myFeature;

        }       

    }

    private DefaultFeature overwrite(

        DefaultFeature old_feat,

        DefaultFeature new_feat) {

        DefaultEditableFeature resp = (DefaultEditableFeature) old_feat.getEditable();

        FeatureType fty = new_feat.getData().getType();

        FeatureAttributeDescriptor[] atts = fty.getAttributeDescriptors();

        for (int i=0; i<atts.length; i++) {

            try {

                resp.set(atts[i].getName(), new_feat.get(atts[i].getName()));

            } catch (Exception exc) {

                // Field not found

            }

        }

        return (DefaultFeature) resp.getNotEditableCopy();

    }

    @Override

    public void remove() {

        super.remove();

        myFeature = new DefaultFeature(fset.store);

    }

    @Override

    protected void doDispose() throws BaseException {

        super.doDispose();

        myFeature = null;

    }

}