TermSelection.java

package org.phenoscape.util;

import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.Transferable;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.io.IOException;

import org.apache.log4j.Logger;
import org.obo.datamodel.OBOClass;

public class TermSelection implements Transferable {

	private final TermTransferObject term;
	private final String termName;
	public static DataFlavor termFlavor = createTermFlavor();

	public TermSelection(OBOClass term) {
		this.term = new TermTransferObject(term);
		this.termName = term.getName();
	}

	@Override
	public Object getTransferData(DataFlavor flavor) throws UnsupportedFlavorException, IOException {
		log().debug("Is flavor supported? " + flavor + ": " + (this.isDataFlavorSupported(flavor)));
		if (this.isDataFlavorSupported(flavor)) {
			log().debug("Returning the term: " + this.term);
			if (flavor instanceof TermFlavor) {
				return this.term;
			} else if (flavor.equals(DataFlavor.stringFlavor)) {
				return this.termName;
			}
		} 
		throw new UnsupportedFlavorException(flavor);
	}

	@Override
	public DataFlavor[] getTransferDataFlavors() {
		return new DataFlavor[] {termFlavor, DataFlavor.stringFlavor};
	}

	@Override
	public boolean isDataFlavorSupported(DataFlavor flavor) {
		if (flavor instanceof TermFlavor) {
			return true;
		} else if (flavor.equals(DataFlavor.stringFlavor)) {
			return true;
		}
		return false;
	}

	private static DataFlavor createTermFlavor() {
		try {
			return new TermFlavor();
		} catch (ClassNotFoundException e) {
			log().error("Term class not found, couldn't create flavor for clipboard", e);
			throw new RuntimeException(e);
		}
	}

	public static class TermFlavor extends DataFlavor {

		public TermFlavor() throws ClassNotFoundException {
			//super(DataFlavor.javaJVMLocalObjectMimeType + ";class=" + OBOClass.class.getName());
			super(TermTransferObject.class, "OBO Term Transfer Object");
		}

	}

	private static Logger log() {
		return Logger.getLogger(TermSelection.class);
	}

}