Application: gvSIG desktop

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

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.util.Iterator;

import org.gvsig.remoteclient.wfs.WFSFeature;

import org.gvsig.tools.dataTypes.DataType;

import org.gvsig.xmlschema.lib.api.XMLSchemaLocator;

import org.gvsig.xmlschema.lib.api.XMLSchemaManager;

import org.gvsig.xmlschema.lib.api.exceptions.SchemaCreationException;

import org.gvsig.xmlschema.lib.api.som.IXSComplexContent;

import org.gvsig.xmlschema.lib.api.som.IXSComplexTypeDefinition;

import org.gvsig.xmlschema.lib.api.som.IXSContentType;

import org.gvsig.xmlschema.lib.api.som.IXSElementDeclaration;

import org.gvsig.xmlschema.lib.api.som.IXSExtension;

import org.gvsig.xmlschema.lib.api.som.IXSGroup;

import org.gvsig.xmlschema.lib.api.som.IXSRestriction;

import org.gvsig.xmlschema.lib.api.som.IXSSchema;

import org.gvsig.xmlschema.lib.api.som.IXSSimpleContent;

import org.gvsig.xmlschema.lib.api.som.IXSSimpleTypeDefinition;

import org.gvsig.xmlschema.lib.api.som.IXSTypeDefinition;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

/**

 * @author gvSIG Team

 * @version $Id$

 *

 */

public class WFSFeatureFiller {

   private static final Logger LOG = LoggerFactory.getLogger(WFSFeatureFiller.class);

    private XMLSchemaManager xmlSchemaManager = null;

    private static final String XMLSCHEMA_PARSER_PROVIDER_NAME = "xmlschema.providers.kxml";

    private WFSFeature feature;

    public WFSFeatureFiller(WFSFeature feature) {

        super();

        xmlSchemaManager =  XMLSchemaLocator.getXMLSchemaManager();

        this.feature = feature;

    }

    /**

     * @param describeFeatureTypeFile

     * @param feature

     * @throws FileNotFoundException 

     * @throws SchemaCreationException 

     */

    public void fill(File describeFeatureTypeFile) throws SchemaCreationException, FileNotFoundException {

        IXSSchema schema = xmlSchemaManager.parse(XMLSCHEMA_PARSER_PROVIDER_NAME, new FileInputStream(describeFeatureTypeFile));

        IXSElementDeclaration elementDeclaration = null;

        if (feature.getNamespace() == null){

            elementDeclaration = schema.getElementDeclarationByName(null, feature.getName()); 

        }else{

            elementDeclaration = schema.getElementDeclarationByName(feature.getNamespace().getLocation(), feature.getLocalName());

        }

        if (elementDeclaration == null){

            return;

        }

        IXSTypeDefinition typeDefinition = elementDeclaration.getTypeDefinition();

        if (typeDefinition == null){

            return;

        }

        if (typeDefinition instanceof IXSSimpleTypeDefinition){

        }else if (typeDefinition instanceof IXSComplexTypeDefinition){ 

            fill((IXSComplexTypeDefinition)typeDefinition);

        }

        feature.setCompleted(true);

    }

    /**

     * @param typeDefinition

     */

    private void fill(IXSComplexTypeDefinition complexTypeDefinition) {

        IXSContentType contentType = complexTypeDefinition.getContentType();

        if (contentType != null){

            fill((IXSContentType)contentType);

        }

        IXSGroup group = complexTypeDefinition.getGroup();

        if (group != null){

            fill((IXSGroup)contentType);

        }        

    }

    /**

     * @param complexTypeDefinition

     */

    private void fill(IXSGroup group) {

        Iterator it = group.getItems().iterator();

        while (it.hasNext()){

            Object item = it.next();

            if (item instanceof IXSElementDeclaration){

                IXSElementDeclaration elementDeclaration = (IXSElementDeclaration)item;       

                DataType dataType = null;

                if (elementDeclaration.getTypeDefinition() != null){

                    dataType = elementDeclaration.getTypeDefinition().getDataType();

                }

                if (dataType == null || elementDeclaration.getNodeName() == null) {

                    LOG.info("Feature attribute type not recognized: "

                        + elementDeclaration.toString(),

                        new Exception("Feature attribute type not recognized: "

                        + elementDeclaration.toString()));

                } else {

                    feature.addField(elementDeclaration.getNodeName(), elementDeclaration.getTypeName(), dataType);

                }

            }else {

                LOG.info("Feature attribute not recognized: "

                    + (item == null ? "NULL" : item.getClass().getName()),

                        new Exception("Feature attribute not recognized"));

            }               

        }      

    }

    /**

     * @param complexTypeDefinition

     */

    private void fill(IXSContentType contentType) {

        if (contentType instanceof IXSSimpleContent){

           fill((IXSSimpleContent)contentType);

        }else if (contentType instanceof IXSComplexContent){

            fill((IXSComplexContent)contentType);

        }else if (contentType instanceof IXSGroup){

           fill((IXSGroup)contentType);

        }        

    }

    private void fill(IXSSimpleContent simpleContent) {

        IXSRestriction restriction = simpleContent.getRestriction();

        if (restriction != null){

            fill(restriction);

        }

        IXSExtension extension = simpleContent.getExtension();

        if (extension != null){

            fill(extension);

        }

    }

    private void fill(IXSComplexContent complexContent) {

        IXSRestriction restriction = complexContent.getRestriction();

        if (restriction != null){

            fill(restriction);

        }

        IXSExtension extension = complexContent.getExtension();

        if (extension != null){

            fill(extension);

        }

    }

    /**

     * @param extension

     */

    private void fill(IXSExtension extension) {

        IXSGroup group = extension.getGroup();

        if (group != null){

            fill(group);

        }        

    }

    /**

     * @param restriction

     */

    private void fill(IXSRestriction restriction) {

        IXSGroup group = restriction.getGroup();

        if (group != null){

            fill(group);

        }            

    }

}