| gvSIG desktop 1

package org.gvsig.app.project.documents.table;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

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

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

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

import org.gvsig.tools.dataTypes.DataTypes;

import org.gvsig.utils.swing.objectSelection.ObjectSelectionModel;

import org.gvsig.utils.swing.objectSelection.SelectionException;

/**

 * @author Fernando Gonz?lez Cort?s

 */

public class FieldSelectionModel implements ObjectSelectionModel {

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

        private FeatureStore fs;

        private String msg;

        private int type = DataTypes.INVALID;

        /**

         * Crea un nuevo FirstFieldSelectionModel.

         *

         */

        public FieldSelectionModel(FeatureStore fs, String msg,        int type) {

                this.fs = fs;

                this.msg = msg;

                this.type = type;

        }

        @SuppressWarnings("unchecked")

        public Object[] getObjects() throws SelectionException {

                        List<String> fields = new ArrayList<String>();

                        Iterator<FeatureAttributeDescriptor> iterator=null;

                        try {

                                iterator = fs.getDefaultFeatureType().iterator();

                        } catch (DataException e) {

                                throw new SelectionException("Can't create iterator for the atribute of feature type",e);

                        }

                        while (iterator.hasNext()) {

                                FeatureAttributeDescriptor descriptor = iterator.next();

                                if ( type != DataTypes.INVALID ){

                                        if (descriptor.getDataType() == type) {

                                                fields.add(descriptor.getName());

                                        }

                                }else{

                                        fields.add(descriptor.getName());

                                }

                        }

                        return (String[]) fields.toArray();

        }

        /**

         * @see org.gvsig.utils.swing.objectSelection.ObjectSelectionModel#getMsg()

         */

        public String getMsg() {

                return msg;

        }

}