wekabayesenetvisualizer.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 weka.core.Drawable;
import weka.gui.graphvisualizer.BIFFormatException;
import weka.gui.graphvisualizer.GraphVisualizer;
import com.prudsys.pdm.Core.MiningException;
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 WekaBayeseNetVisualizer extends ResultView {
	/**
	 * 
	 */
	private static final long serialVersionUID = -4878317421225794761L;
	private Drawable drawable;
	private GraphVisualizer bayesVisualizer;
	private JScrollPane m_ScrollPane;
	private boolean avaible;

	public WekaBayeseNetVisualizer(Drawable wekaDrawable, String viewName) throws SysException, MiningException {
		super(viewName);
		drawable = wekaDrawable;
		m_ScrollPane = new JScrollPane();
		if (drawable != null) {
			
			String graph = null;
			try {
				graph = wekaDrawable.graph();
			}
			catch(Exception e) {
				e.printStackTrace();
				throw new SysException("Can't load graph description!");
			}
			
			createView(graph);
		}
		m_ScrollPane.setHorizontalScrollBarPolicy(JScrollPane.HORIZONTAL_SCROLLBAR_AS_NEEDED);
		m_ScrollPane.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
		add(m_ScrollPane, BorderLayout.CENTER);
	}

	public void createView(String XMLBIF) throws SysException, MiningException {
		try {
			bayesVisualizer = new GraphVisualizer();
			try {
				bayesVisualizer.readBIF(XMLBIF);
				bayesVisualizer.layoutGraph();
			} catch (BIFFormatException be) {
				be.printStackTrace();
				throw new SysException("unable to visualize BayesNet");
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
		if (bayesVisualizer != null) {
			avaible = true;
			m_ScrollPane.getViewport().add(bayesVisualizer, 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(bayesVisualizer, exportFile, true);
						aExporter.export();
					} else {
						returnVal = chooser.showSaveDialog(this);
					}
				}
			} else {
				JPEGImageExporter aExporter = new JPEGImageExporter(bayesVisualizer, exportFile, true);
				aExporter.export();
			}
		}
	}
}