Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.util;

import java.io.File;

import java.io.FileFilter;

import java.io.FilenameFilter;

import java.util.Iterator;

import org.gvsig.tools.observer.Observable;

public interface FolderSet extends Observable {

	public interface FolderEntry {

		public File getFolder();

		public String getLabel() ;

	}

	public void add(File folder);

	public void add(File folder, String description);

	public void clear();

	public void remove(File folder);

	public void set(File folder);

	public Iterator iterator(); // over FolderEntry

	public File[] listFiles();

	public File[] listFiles(FileFilter filter);

	public File[] listFiles(FilenameFilter filter);

}

package org.gvsig.tools.util.impl;

import java.io.File;

import java.io.FileFilter;

import java.io.FilenameFilter;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashSet;

import java.util.Iterator;

import java.util.List;

import java.util.Set;

import org.gvsig.tools.observer.impl.BaseWeakReferencingObservable;

import org.gvsig.tools.util.FolderSet;

public class DefaultFolderSet extends BaseWeakReferencingObservable implements FolderSet {

	public class DefaultFolderEntry implements FolderEntry {

		File folder; 

		String label;

		DefaultFolderEntry(File folder, String label) {

			this.folder = folder;

			this.label = label;

		}

		public boolean equals(Object obj) {

			return this.folder.equals(((FolderEntry)obj).getFolder());

		}

		public int hashCode() {

			return this.folder.hashCode();

		}

		public File getFolder() {

			return this.folder;

		}

		public String getLabel() {

			return this.label;

		}

	}

	private Set folders = null;

	public DefaultFolderSet() {

		folders = new HashSet();

	}

	public void add(File folder) {

		this.add(folder, null);

	}

	public void add(File folder, String description) {

		if( folder == null ) {

			return;

		}

		if( folder.exists() && !folder.isDirectory() ) {

			throw new IllegalArgumentException("folder required");

		}

		this.folders.add(new DefaultFolderEntry(folder, description));

		this.notifyObservers();	

	}

	public void clear() {

		this.folders.clear();

		this.notifyObservers();

	}

	public void remove(File folder) {

		this.folders.remove(new DefaultFolderEntry(folder,null));

		this.notifyObservers();

	}

	public void set(File folder) {

		this.beginComplexNotification();

		this.clear();

		this.add(folder);

		this.endComplexNotification();

	}

	public File[] listFiles() {

		List entries = new ArrayList();

		Iterator it = this.folders.iterator();

		while( it.hasNext() ) {

			FolderEntry folderEntry = (FolderEntry) it.next();

			File[] files = folderEntry.getFolder().listFiles();

			entries.addAll( Arrays.asList(files));

		}

		return  (File[]) entries.toArray();

	}

	public File[] listFiles(FileFilter filter) {

		List entries = new ArrayList();

		Iterator it = this.folders.iterator();

		while( it.hasNext() ) {

			FolderEntry folderEntry = (FolderEntry) it.next();

			File[] files = folderEntry.getFolder().listFiles(filter);

			entries.addAll( Arrays.asList(files));

		}

		return  (File[]) entries.toArray();

	}

	public File[] listFiles(FilenameFilter filter) {

		List entries = new ArrayList();

		Iterator it = this.folders.iterator();

		while( it.hasNext() ) {

			FolderEntry folderEntry = (FolderEntry) it.next();

			File[] files = folderEntry.getFolder().listFiles(filter);

			entries.addAll( Arrays.asList(files));

		}

		return  (File[]) entries.toArray();

	}

	public Iterator iterator() {

		return this.folders.iterator();

	}

}