Application: gvSIG desktop » gvSIG Linear Reference System

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2015 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 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.

 *

 * 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.lrs.swing.impl;

import java.util.ArrayList;

import java.util.List;

import javax.swing.AbstractListModel;

import javax.swing.ComboBoxModel;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

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

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.DataTypes;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.operations.LayerCollection;

import org.gvsig.fmap.mapcontext.layers.operations.LayersVisitor;

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.visitor.VisitCanceledException;

/**

 * @author dmartinez

 *

 */

public class PointLayersComboBoxModel extends AbstractListModel<FLayer> implements ComboBoxModel<FLayer> {

    /**

     *

     */

    private static final long serialVersionUID = -1271707498886332136L;

    private static final Logger logger = LoggerFactory.getLogger(PointLayersComboBoxModel.class);

    private LayerCollection layers;

    private List<FLyrVect> vectLayers;

    private FLayer selectedLayer;

    /**

     *

     */

    public PointLayersComboBoxModel(LayerCollection layers) {

        this.layers = layers;

        vectLayers = new ArrayList<FLyrVect>();

        try {

            layers.accept(new LayersVisitor() {

                public void visit(Object obj) throws VisitCanceledException, BaseException {

                    // TODO Auto-generated method stub

                }

                public void visit(FLayer layer) throws BaseException {

                    if (layer instanceof FLyrVect) {

                        FLyrVect lyrVect = (FLyrVect) layer;

                        if (lyrVect.getGeometryType().isTypeOf(Geometry.TYPES.POINT)

                            || lyrVect.getGeometryType().isTypeOf(Geometry.TYPES.MULTIPOINT)) {

                            FeatureStore featureStore = lyrVect.getFeatureStore();

                            try {

                                FeatureType type = featureStore.getDefaultFeatureType();

                                FeatureAttributeDescriptor[] attributeDescriptors = type.getAttributeDescriptors();

                                Boolean nonGeometricDataTypeFound=false;

                                Boolean numericDataTypeFound=false;

                                Boolean dataTypesFound=false;

                                Boolean validateNext=true;

                                for (int i = 0; i < attributeDescriptors.length&&!dataTypesFound; i++) {

                                    FeatureAttributeDescriptor featureAttributeDescriptor = attributeDescriptors[i];

                                    validateNext=true;

                                    if (featureAttributeDescriptor.getDataType().isNumeric()&&!numericDataTypeFound) {

                                        numericDataTypeFound = true;

                                        validateNext=false;

                                    }

                                    //This is done to avoid to validate twice the same numeric field

                                    if (validateNext){

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

                                            nonGeometricDataTypeFound = true;

                                        }

                                    }

                                    dataTypesFound=nonGeometricDataTypeFound&&numericDataTypeFound;

                                }

                                if(dataTypesFound){

                                    logger.debug("Correct point layer found");

                                    vectLayers.add(lyrVect);

                                    if(lyrVect.isActive()){

                                        selectedLayer = lyrVect;

                                    }

                                }

                            } catch (DataException e1) {

                                logger.warn(

                                    new StringBuilder().append("Error getting default feature type from layer ").append(lyrVect.getName())

                                        .toString(), e1);

                            }

                        }

                    }

                }

            });

        } catch (BaseException e1) {

            logger.warn("Can't get linear vector layers", e1);

        }

    }

    /*

     * (non-Javadoc)

     *

     * @see javax.swing.ListModel#getSize()

     */

    public int getSize() {

        return vectLayers.size(); //.getLayersCount();

    }

    /*

     * (non-Javadoc)

     *

     * @see javax.swing.ListModel#getElementAt(int)

     */

    public FLayer getElementAt(int index) {

        return vectLayers.get(index); //.getLayer(index);

    }

    /*

     * (non-Javadoc)

     *

     * @see javax.swing.ComboBoxModel#setSelectedItem(java.lang.Object)

     */

    public void setSelectedItem(final Object anItem) {

        selectedLayer = (FLayer) anItem;

        fireContentsChanged(this, -1, -1);

    }

    /*

     * (non-Javadoc)

     *

     * @see javax.swing.ComboBoxModel#getSelectedItem()

     */

    public Object getSelectedItem() {

        return selectedLayer;

    }

}