wekatreevisualizer.java

一个数据挖掘软件ALPHAMINERR的整个过程的JAVA版源代码

package eti.bi.alphaminer.patch.standard.operation.result.view;

import java.awt.BorderLayout;
import java.awt.Component;
import java.io.File;
import javax.swing.JFileChooser;
import javax.swing.JOptionPane;
import javax.swing.JScrollPane;
import com.prudsys.pdm.Core.MiningException;
import weka.core.Drawable;
import weka.gui.treevisualizer.PlaceNode2;
import weka.gui.treevisualizer.TreeDisplayEvent;
import weka.gui.treevisualizer.TreeDisplayListener;
import weka.gui.treevisualizer.TreeVisualizer;
import eti.bi.alphaminer.operation.result.ResultView;
import eti.bi.alphaminer.operation.result.export.JPEGImageExporter;
import eti.bi.common.Locale.Resource;
import eti.bi.exception.AppException;
import eti.bi.exception.SysException;

public class WekaTreeVisualizer extends ResultView {
	/**
	 * 
	 */
	private static final long serialVersionUID = 5093537243547485504L;
	
	private Drawable drawable;
	private TreeVisualizer treeVisualizer;
	private JScrollPane m_ScrollPane;
	private boolean avaible;
	
	public WekaTreeVisualizer(Drawable wekaDrawable, String viewName) throws SysException, MiningException {
		super(viewName);
		drawable = wekaDrawable;
		m_ScrollPane = new JScrollPane();
		if(drawable!=null) {
			createView();
		}
		m_ScrollPane.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);
		m_ScrollPane.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
		add(m_ScrollPane,BorderLayout.CENTER);
	}

	public void createView() throws SysException, MiningException{
		try {
			treeVisualizer = new TreeVisualizer(new TreeDisplayListener() {

				public void userCommand(TreeDisplayEvent e) {
				// System.out.println("TreeDisplayEvent: "+e);
				}

			}, drawable.graph(), new PlaceNode2());
		} catch (Exception e) {
			e.printStackTrace();
		}
		
		if(treeVisualizer!=null) {
			avaible = true;
			m_ScrollPane.getViewport().add(treeVisualizer,BorderLayout.CENTER);
		}
		m_ViewType = ResultView.TYPE_GRAPH;
	}
	
	public boolean avaible() {
		return avaible;
	}
	
	@Override
	public void export() throws SysException, AppException {
//		 Use user home directory 
			File directory = new File(System.getProperty("user.dir"));
			
			// Create and initialize file chooser for jpg
			JFileChooser chooser = new JFileChooser(directory);
			chooser.setDialogTitle(Resource.srcStr("GraphExport"));
			chooser.setFileFilter(JPEGImageExporter.FILTER);
			chooser.setSelectedFile(JPEGImageExporter.DEFAULT_FILE);
	
			// pop up the file chooser dialog and return the file value
			int returnVal = chooser.showSaveDialog(this);	  
			if (returnVal == JFileChooser.APPROVE_OPTION) 
			{
				File exportFile = chooser.getSelectedFile();	
				
				//<<tyleung 20/4/2005
				if (exportFile.exists()) {
					int option = JOptionPane
							.showConfirmDialog(
									(Component) this,
									"The file  \""
											+ exportFile.getName()
											+ "\""
											+ " already exists. Do you want to replace the existing file?",//						
									"AlphaMiner", JOptionPane.YES_NO_OPTION,
									JOptionPane.QUESTION_MESSAGE);
					if (option != JOptionPane.CANCEL_OPTION) {
						if (option == JOptionPane.YES_OPTION) {
							
							   JPEGImageExporter aExporter = new JPEGImageExporter(treeVisualizer, exportFile, true);
				 			   aExporter.export(); 
							
							
						}else{
							returnVal = chooser.showSaveDialog(this);
						}
					}
				}
				else {
				    JPEGImageExporter aExporter = new JPEGImageExporter(treeVisualizer, exportFile, true);
				    aExporter.export();
				}
			} 
	}
}