| gvSIG desktop 1

package org.gvsig.tools.swing.icontheme.impl;

import java.io.File;

import java.io.FilenameFilter;

import java.net.MalformedURLException;

import java.util.HashMap;

import org.gvsig.installer.lib.api.InstallerLocator;

import org.gvsig.installer.lib.api.InstallerManager;

import org.gvsig.installer.lib.api.PackageInfo;

import org.gvsig.tools.swing.icontheme.IconTheme;

/**

 * This class extends AbstractIconTheme and implements the abstract methods of this

 * class. This methods are <code>loadIcon</code> and <code>load</code>. This methods

 * allows load one icon or all icons in the resource.

 */

public class FolderIconTheme extends BaseIconTheme{

        protected File resource = null; 

        protected PackageInfo packageInfo = null;

        /**

         * Constructor.Constructs an Icon Theme with a default one.

         * @param def. The default icon theme

         */

        public FolderIconTheme(IconTheme def) {

                super(def);

        }

        public void load(Object resource) throws IllegalArgumentException {

                if( !(resource instanceof File) ) {

                        throw new IllegalArgumentException();

                }

                File res = (File) resource;

                if( !res.isDirectory() ) {

                        throw new IllegalArgumentException();

                }

                this.resource = res;

                this.iconList = null;

                this.loadPackageInfo();

                this.setName(this.packageInfo.getName());

                this.setDescription(this.packageInfo.getDescription());

        }

        private void loadPackageInfo() {

                File pkgfile = new File(this.resource,"package.info"); 

                InstallerManager manager = InstallerLocator.getInstallerManager();

                try {

                        this.packageInfo = manager.createPackageInfo(pkgfile.toURI().toURL().openStream());

                } catch (Exception e) {

                        throw new IllegalArgumentException(e);

                }

        }

        protected void deferredLoad() {

                if( this.iconList != null ) {

                        return;

                }

                iconList = new HashMap<String, IconTheme.Icon>();

                load(resource, "");

                File[] files = resource.listFiles();

                for (int i=files.length-1; i>=0; i--) {

                        if( files[i].isDirectory() ) {

                                load(files[i],files[i].getName());

                        }

                }

        }

        private void load(File folder, String group) {

                File[] imagefiles = folder.listFiles(new ImageFileFilter());

                String name;

                for (int i=imagefiles.length-1; i>=0; i--) {

                        name = computeKey(imagefiles[i].getName());

                        try {

                                this.register(null, group, name, null, imagefiles[i].toURI().toURL());

                        } catch (MalformedURLException e) {

                                logger.info("Can't load icon", e);

                        }

                }

        }

        /**

         * This class allows filter the images in the directory. Allows to load the images

         * with a apropiate extension.

         */

        private class ImageFileFilter implements FilenameFilter {

                public boolean accept(File dir, String fileName) {

                        String extension = "";

                        int pointPos = fileName.lastIndexOf(".");

                        if (pointPos>0 && pointPos < (fileName.length()-1) )

                                extension = fileName.substring(pointPos+1).toLowerCase();

                        if ( extension.equals("jpg") || extension.equals("jpeg")

                                        || extension.equals("png")

                                        || extension.equals("gif")

                                        )

                                return true;

                        else

                                return false;

                }

        }

        /**

         * Returns the key of the image without extension

         * @param fileName

         * @return string

         */

        private String computeKey(String fileName) {

                int pointPos = fileName.lastIndexOf(".");

                if (pointPos!=-1)

                        return fileName.substring(0, pointPos);

                else

                        return fileName;

        }

}