inferencecontroller.java

利用Java实现的神经网络工具箱

/*
 * $RCSfile: InferenceController.java,v $
 * $Revision: 1.5 $
 * $Date: 2005/05/01 19:54:17 $
 *
 * NeuralNetworkToolkit
 * Copyright (C) 2004 Universidade de Brasília
 *
 * This file is part of NeuralNetworkToolkit.
 *
 * NeuralNetworkToolkit is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * NeuralNetworkToolkit is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with NeuralNetworkToolkit; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA - 02111-1307 - USA.
 */
package neuralnetworktoolkit.gui.inference;

import java.io.File;
import java.util.ResourceBundle;
import java.util.Vector;

import javax.swing.JFileChooser;
import javax.swing.JOptionPane;

import neuralnetworktoolkit.gui.DefaultFileFilter;
import neuralnetworktoolkit.gui.navigation.NavigationController;
import neuralnetworktoolkit.modelstorage.ModelSerializer;

/**
 * 
 * 
 * @version $Revision: 1.5 $ - $Date: 2005/05/01 19:54:17 $
 * 
 * @author <a href="mailto:[e-mail]">Rodrigo C. M. Coimbra</a>
 * @author <a href="mailto:[e-mail]">Hugo Iver V. Gonçalves</a>
 */
public class InferenceController {
	
	private NavigationController navController;
	private InferenceFrame inferenceFrame;
	private String networkName;
	
	private boolean isManualData = false;
	private Vector manualData;
	private Vector header;
	private int headerSize = 0;
	
	private ResourceBundle resource;

	/**
	 * @param navController
	 */
	public InferenceController(NavigationController navController,
			String networkName) {
		this.navController = navController;
		this.networkName = networkName;
		inferenceFrame = new InferenceFrame(this);
		
		resource = ResourceBundle
				.getBundle("neuralnetworktoolkit.gui.inference.resources.InferenceControllerResource");
		
	} //InferenceController()
	
	/**
	 * 
	 *
	 */
	public void saveNetwork() {
		JFileChooser chooser;
		DefaultFileFilter filter;
		File defaultFile;
		int returnValue;
		
		filter = new DefaultFileFilter(ModelSerializer.modelExtensions,
				resource.getString("fileFilter"));
		chooser = new JFileChooser();
		chooser.setDialogType(JFileChooser.SAVE_DIALOG);
		chooser.addChoosableFileFilter(filter);
		chooser.setApproveButtonText(resource.getString("save"));
		chooser.setDialogTitle(resource.getString("saveNetwork"));
		chooser.setApproveButtonToolTipText(resource.getString("saveNetwork"));
		defaultFile = new File(chooser.getCurrentDirectory().getAbsolutePath()
				+ "/" + networkName + "." + ModelSerializer.modelExtensions[0]);
		chooser.setSelectedFile(defaultFile);
		returnValue = chooser.showSaveDialog(null);
		if(returnValue == JFileChooser.APPROVE_OPTION) {
			navController.saveNetwork(chooser.getSelectedFile()
					.getAbsolutePath());
			
		}
		
	} //saveNetwork()
	
	/**
	 * 
	 *
	 */
	public void loadInferenceDataFromFile() {
		navController.loadInferenceDataFromFile();
		isManualData = false;
		
	} //loadInferenceDataFromFile()
	
	/**
	 * 
	 *
	 */
	public void loadDataManually() {
		isManualData = true;
		header = navController.getInferenceDataHeader(); 
		manualData = new Vector();
		headerSize = header.size();
		manualData.add(new Vector(headerSize));
		inferenceFrame.setManualTable(manualData, header);
		
	} //loadDataManually()
	
	/**
	 * 
	 * @param inferenceData
	 * @param inferenceHeader
	 */
	public void showInferenceData(Vector inferenceData, Vector inferenceHeader) {
		inferenceFrame.showInferenceData(inferenceData, inferenceHeader);
		
	} //showInferenceData()
	
	/**
	 * 
	 * @param inferenceData
	 * @param inferenceHeader
	 */
	public void showInferedData(Vector inferedData, Vector inferedHeader) {
		inferenceFrame.showInferedData(inferedData, inferedHeader);
		
	} //showInferenceData()
	
	/**
	 * @return Returns the inferenceFrame.
	 */
	public InferenceFrame getInferenceFrame() {
		return inferenceFrame;
		
	} //getInferenceFrame()
	
	/**
	 * @return Returns the networkName.
	 */
	public String getNetworkName() {
		return networkName;
		
	} //getNetworkName()


	/**
	 * 
	 */
	public void infereWithNetwork() {
		if(!isManualData) {
			navController.infereWithNetwork();
		
		} else {
			navController.infereWithNetwork(manualData);
			
		}
		
	} //infereWithNetwork()


	/**
	 * 
	 */
	public void saveInferedData() {
		JFileChooser chooser;
		int returnValue;
		
		chooser = new JFileChooser();
		chooser.setDialogType(JFileChooser.SAVE_DIALOG);
		chooser.setApproveButtonText(resource.getString("save"));
		chooser.setApproveButtonToolTipText(resource.getString("saveFile"));
		chooser.setDialogTitle(resource.getString("saveFile"));
		returnValue = chooser.showSaveDialog(null);
		if(returnValue == JFileChooser.APPROVE_OPTION) {
			navController.saveInferedData(chooser.getSelectedFile());
			
		}
		
	} //saveInferedData()

	/**
	 * 
	 */
	public void showInfo() {
		navController.showInfo();
		
	} //showInfo()
	
	/**
	 * 
	 *
	 */
	public void addRow() {
		manualData.add(new Vector(headerSize));
		inferenceFrame.setManualTable(manualData, header);
		
	} //addRow()
	
	/**
	 * 
	 * @param selectedIndex
	 */
	public void removeRow(int selectedIndex) {
		if ((manualData.size() > 0) && (selectedIndex >= 0)
				&& (manualData.size() > selectedIndex)) {
			manualData.remove(selectedIndex);
			
		} else {
			JOptionPane.showMessageDialog(inferenceFrame, resource
					.getString("remove"), resource.getString("nntk"),
					JOptionPane.ERROR_MESSAGE);
			
		}
		
	} //removeRow()
	
} //InferenceController