Add UMB model export to the GUI.

Author: davexparker

prism/src/userinterface/model/GUIMultiModel.java

@@ -83,7 +83,7 @@ public class GUIMultiModel extends GUIPlugin implements PrismSettingsListener
 	private AbstractAction computeSS, computeTr, newPRISMModel;
 	private AbstractAction newPEPAModel, loadModel, reloadModel, saveModel, saveAsModel, parseModel, buildModel;
 	private AbstractAction exportStatesPlain, exportStatesMatlab,
-			exportTransPlain, exportTransMatlab, exportTransDot, exportObsPlain, exportObsMatlab, exportStateRewardsPlain, exportStateRewardsMatlab,
+			exportTransPlain, exportTransMatlab, exportTransDot, exportTransUMB, exportObsPlain, exportObsMatlab, exportStateRewardsPlain, exportStateRewardsMatlab,
 			exportTransRewardsPlain, exportTransRewardsMatlab, exportLabelsPlain, exportLabelsMatlab,
 			exportSSPlain, exportSSMatlab, exportTrPlain, exportTrMatlab;
 	private JPopupMenu popup;
@@ -97,6 +97,7 @@ public class GUIMultiModel extends GUIPlugin implements PrismSettingsListener
 	private FileFilter textFilter;
 	private FileFilter matlabFilter;
 	private FileFilter dotFilter;
+	private FileFilter umbFilter;
 	//State
 	private boolean computing = false;
 	private boolean initialised = false;
@@ -190,6 +191,7 @@ public void doEnables()
 		exportTransPlain.setEnabled(!computing);
 		exportTransMatlab.setEnabled(!computing);
 		exportTransDot.setEnabled(!computing);
+		exportTransUMB.setEnabled(!computing);
 		exportObsPlain.setEnabled(!computing);
 		exportObsMatlab.setEnabled(!computing);
 		exportStateRewardsPlain.setEnabled(!computing);
@@ -364,6 +366,9 @@ protected void a_exportBuildAs(ModelExportEntity exportEntity, ModelExportFormat
 			case DOT:
 				res = showSaveFileDialog(dotFilter);
 				break;
+			case UMB:
+				res = showSaveFileDialog(umbFilter);
+				break;
 			case MATLAB:
 				res = showSaveFileDialog(matlabFilter);
 				break;
@@ -678,9 +683,21 @@ public void actionPerformed(ActionEvent e)
 		};
 		exportTransDot.putValue(Action.LONG_DESCRIPTION, "Exports the model to a Dot file");
 		exportTransDot.putValue(Action.MNEMONIC_KEY, Integer.valueOf(KeyEvent.VK_D));
-		exportTransDot.putValue(Action.NAME, "Dot file");
+		exportTransDot.putValue(Action.NAME, "Dot");
 		exportTransDot.putValue(Action.SMALL_ICON, GUIPrism.getIconFromImage("smallFileDot.png"));
 
+		exportTransUMB = new AbstractAction()
+		{
+			public void actionPerformed(ActionEvent e)
+			{
+				a_exportBuildAs(ModelExportEntity.MODEL, ModelExportFormat.UMB);
+			}
+		};
+		exportTransUMB.putValue(Action.LONG_DESCRIPTION, "Exports the model to a UMB file");
+		exportTransUMB.putValue(Action.MNEMONIC_KEY, Integer.valueOf(KeyEvent.VK_D));
+		exportTransUMB.putValue(Action.NAME, "UMB");
+		exportTransUMB.putValue(Action.SMALL_ICON, GUIPrism.getIconFromImage("smallMatrix.png"));
+
 		exportObsPlain = new AbstractAction()
 		{
 			public void actionPerformed(ActionEvent e)
@@ -1044,6 +1061,7 @@ private JMenu initExportMenu()
 		exportPlainMenu.add(exportObsPlain);
 		exportMenu.add(exportPlainMenu);
 		exportMenu.add(exportTransDot);
+		exportMenu.add(exportTransUMB);
 		JMenu exportMatlabMenu = new JMenu("Matlab");
 		exportMatlabMenu.setMnemonic('M');
 		exportMatlabMenu.setIcon(GUIPrism.getIconFromImage("smallFileMatlab.png"));
@@ -1182,6 +1200,7 @@ private void initComponents()
 		textFilter =  new FileNameExtensionFilter("Plain text files (*.txt)", "txt");
 		matlabFilter = new FileNameExtensionFilter("Matlab files (*.m)", "m");
 		dotFilter = new FileNameExtensionFilter("Dot files (*.dot)", "dot");
+		umbFilter = new FileNameExtensionFilter("UMB files (*.umb)", "dot");
 
 		setLayout(new BorderLayout());
 		add(topPanel, BorderLayout.CENTER);