| gvSIG desktop 1

package com.iver.cit.gvsig.gui.styling;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileReader;

import java.util.Comparator;

import java.util.TreeSet;

import java.util.Vector;

import org.exolab.castor.xml.MarshalException;

import org.exolab.castor.xml.ValidationException;

import com.iver.andami.messages.NotificationManager;

import com.iver.cit.gvsig.fmap.core.SymbologyFactory;

import com.iver.cit.gvsig.fmap.core.styles.IStyle;

import com.iver.utiles.XMLEntity;

import com.iver.utiles.xmlEntity.generate.XmlTag;

public class StyleSelectorListModel extends SymbolSelectorListModel {

        public static final String STYLE_FILE_EXTENSION = ".style";

        public StyleSelectorListModel(File dir, Object currentElemet, SelectorFilter filter, String fileExtension) {

                super(dir, currentElemet, filter, fileExtension);

                // TODO Auto-generated constructor stub

        }

        public Vector getObjects() {

                if (elements == null) {

                        elements = new Vector();

                        if (currentElement!= null)

                                elements.add(currentElement);

                        File[] ff = dir.listFiles(ffilter);

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

                                XMLEntity xml;

                                try {

                                        xml = new XMLEntity((XmlTag) XmlTag.unmarshal(new FileReader(ff[i])));

                                        IStyle sty = SymbologyFactory.createStyleFromXML(xml, ff[i].getName());

                                        if (sfilter.accepts(sty))

                                                add(sty);

                                } catch (MarshalException e) {

                                        NotificationManager.

                                                addWarning("Error in file ["+ff[i].getAbsolutePath()+"]. " +

                                                                "File corrupted! Skiping it...", e);

                                } catch (ValidationException e) {

                                        NotificationManager.

                                                addWarning("Error validating style file ["+ff[i].getAbsolutePath()+"].", e);

                                } catch (FileNotFoundException e) {

                                        // unreachable code, but anyway...

                                        NotificationManager.

                                                addWarning("File not found: "+ ff[i].getAbsolutePath(), e);

                                }

                        }

                }

                return elements;

        }

        public void add(Object o) {

                TreeSet map = new TreeSet(new Comparator() {

                        public int compare(Object o1, Object o2) {

                                // first will always be the current symbol

                                if (o1.equals(currentElement)) return -1;

                                if (o2.equals(currentElement)) return 1;

                                IStyle sym1 = (IStyle) o1;

                                IStyle sym2 = (IStyle) o2;

                                if (sym1.getDescription() == null && sym2.getDescription() != null) return -1;

                                if (sym1.getDescription() != null && sym2.getDescription() == null) return 1;

                                if (sym1.getDescription() == null && sym2.getDescription() == null) return 1;

                                int result = sym1.getDescription().compareTo(sym2.getDescription());

                                return (result!=0) ? result: 1; /* this will allow adding symbols with

                                                                                                   the same value for description than

                                                                                                   a previous one. */

                        }

                });

                map.addAll(elements);

                map.add(o);

                elements = new Vector(map);

        }

}