Application: gvSIG desktop » gvSIG data provider for WFS

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

 *

 */

package org.gvsig.fmap.dal.store.wfs;

import java.io.ByteArrayOutputStream;

import java.io.OutputStream;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.geom.operation.GeometryOperationContext;

import org.gvsig.fmap.geom.operation.GeometryOperationException;

import org.gvsig.fmap.geom.operation.GeometryOperationNotSupportedException;

import org.gvsig.gpe.lib.api.GPELocator;

import org.gvsig.gpe.lib.api.GPEManager;

import org.gvsig.gpe.lib.api.exceptions.WriterHandlerCreationException;

import org.gvsig.gpe.lib.api.writer.IGPEWriterHandler;

import org.gvsig.remoteclient.wfs.WFSClient;

import org.gvsig.remoteclient.wfs.WFSStatus;

import org.gvsig.remoteclient.wfs.edition.WFSTAttributesOperation;

import org.gvsig.remoteclient.wfs.edition.WFSTInsertOperation;

import org.gvsig.remoteclient.wfs.edition.WFSTTransaction;

import org.gvsig.remoteclient.wfs.edition.WFSTUpdateOperation;

/**

 * @author gvSIG Team

 * @version $Id$

 *

 */

public class WFSTTransactionBuilder {

    private WFSStatus wfsStatus;

    private WFSClient wfsClient;

    private static final GPEManager gpeManager = GPELocator.getGPEManager();

    private WFSTTransaction transaction = null;

    public WFSTTransactionBuilder(WFSStatus wfsStatus, WFSClient wfsClient) {

        super();

        this.wfsStatus = wfsStatus;

        this.wfsClient = wfsClient;

        transaction = wfsStatus.createNewTransaction();

    }  

    private void addFeatureAttributes(WFSTAttributesOperation attributesOperation, FeatureProvider featureProvider) throws WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException{

        for (int i=0 ; i<featureProvider.getType().size() ; i++){

            FeatureAttributeDescriptor featureAttributeDescriptor = featureProvider.getType().getAttributeDescriptor(i);

            String attributeValue = "";

            if (featureAttributeDescriptor.getType() != DataTypes.GEOMETRY){

                if (featureProvider.get(i) != null){

                    attributeValue = featureProvider.get(i).toString();

                }

            }else{

                Geometry geometry = (Geometry)featureProvider.get(i);

                attributeValue = toGML(geometry, featureAttributeDescriptor.getSRS().getAbrev());               

            }  

            attributesOperation.addAttribute(featureAttributeDescriptor.getName(), attributeValue);

        }

    }

    private String toGML(Geometry geometry, String srs) throws GeometryOperationNotSupportedException, GeometryOperationException, WriterHandlerCreationException{

        OutputStream os = new ByteArrayOutputStream();

        IGPEWriterHandler writerHandler = gpeManager.createWriterByMimeType("text/xml; subtype=gml/3.1.2");  

        writerHandler.setOutputStream(os);

        writerHandler.initialize();       

        GeometryOperationContext geometryOperationContext = new GeometryOperationContext();

        geometryOperationContext.setAttribute("writerHandler", writerHandler);

        geometryOperationContext.setAttribute("srs", srs);

        geometryOperationContext.setAttribute("id", null);

        geometry.invokeOperation("writeGml2", geometryOperationContext);

        writerHandler.close(); 

        //Remove the xml header

        String gml = os.toString();

        return gml.substring(38, gml.length());

    }

    public void addInsertFeature(FeatureProvider featureProvider) throws WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException{

        WFSTInsertOperation insertOperation = transaction.createInsertOperation();

        addFeatureAttributes(insertOperation, featureProvider);

    } 

    public void addUpdateFeature(FeatureProvider featureProvider) throws WriterHandlerCreationException, GeometryOperationNotSupportedException, GeometryOperationException{

        WFSTUpdateOperation updateOperation = transaction.createUpdateOperation(featureProvider.getOID().toString());

        addFeatureAttributes(updateOperation, featureProvider);

    }

    public void addDeleteFeature(String id) throws DataException{

        transaction.createDeleteOperation(id);      

    }

}