Application: gvSIG desktop » gvSIG geoprocess

/*

 * gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2010 Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 */

package org.gvsig.sextante.app.algorithm.base.core;

import java.util.Iterator;

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

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

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

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

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

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

import org.gvsig.fmap.geom.exception.CreateGeometryException;

import org.gvsig.fmap.geom.util.Converter;

import org.gvsig.sextante.app.extension.core.gvVectorLayer;

import com.vividsolutions.jts.geom.Geometry;

import es.unex.sextante.core.Sextante;

import es.unex.sextante.dataObjects.IVectorLayer;

import es.unex.sextante.exceptions.UnsupportedOutputChannelException;

/**

 * Writes features in a FeatureStore

 * @author Nacho Brodin (nachobrodin@gmail.com)

 */

public class DALFeaturePersister {

        private FeatureStore              store             = null;

        private String[]                  fieldNames        = null;

        /**

         * Sets the output FeatureType

         * @param out

         * @throws DataException 

         */

        public DALFeaturePersister(FeatureStore out, String[] fieldNames) {

                this.store = out;

                this.fieldNames = fieldNames;

                try {

                        store.edit();

                } catch (DataException e) {

                }

        }

        /**

         * Adds a JTS feature to the FeatureStore

         * @param entry

         * @param newGeom

         * @throws CreateGeometryException

         * @throws DataException

         */

        public EditableFeature addFeature(Feature feature, Geometry newGeom) throws CreateGeometryException, DataException {

                org.gvsig.fmap.geom.Geometry newDalGeom = Converter.jtsToGeometry(newGeom);

                EditableFeature feat = store.createNewFeature(store.getDefaultFeatureType(), feature);

                feat.setGeometry("GEOMETRY", newDalGeom);

                store.insert(feat);

                return feat;

        }

        /**

         * Adds a DAL feature to the FeatureStore

         * @param entry

         * @param newGeom

         * @throws CreateGeometryException

         * @throws DataException

         */

        public EditableFeature addFeature(Feature feature, org.gvsig.fmap.geom.Geometry newGeom) throws CreateGeometryException, DataException {

                EditableFeature feat = store.createNewFeature(store.getDefaultFeatureType(), feature);

                feat.setGeometry(fieldNames.length, newGeom);

                store.insert(feat);

                return feat;

        }

        /**

         * Adds a DAL feature to the FeatureStore

         * @param entry

         * @param newGeom

         * @throws CreateGeometryException

         * @throws DataException

         */

        public EditableFeature addGeometryToExistingFeature(EditableFeature feature, org.gvsig.fmap.geom.Geometry newGeom) throws CreateGeometryException, DataException {

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

                        feature.set(fieldNames[i], feature.get(i));

                feature.setGeometry(fieldNames.length, newGeom);

                store.update(feature);

                return feature;

        }

        /**

         * Ends the edition and closes the FeatureStore

         */

        public void end() {

                try {

                        store.finishEditing();

                } catch (DataException e) {

                        Sextante.addErrorToLog(e);

                }

                store.dispose();

        }

}