Merge pull request #17 from Rockdtben/fixImageFileExtensionIssue

Fixes issue #15
tgstation · Jan 30, 2018 · 12dd20d · 12dd20d
2 parents cee33ba + c80b810
commit 12dd20d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/main/java/com/github/monster860/fastdmm/FastDMM.java b/src/main/java/com/github/monster860/fastdmm/FastDMM.java
@@ -502,6 +502,7 @@ public void actionPerformed(ActionEvent e) {
 			}
 		} else if("mapimage".equals(e.getActionCommand())) {
 			JFileChooser fc = new JFileChooser();
+
 			if (fc.getChoosableFileFilters().length > 0)
 				fc.removeChoosableFileFilter(fc.getChoosableFileFilters()[0]);
 			fc.addChoosableFileFilter(new FileNameExtensionFilter("Image (*.png)", "png"));
@@ -512,6 +513,9 @@ public void actionPerformed(ActionEvent e) {
 					mapImage = createMapImage(1);
 				}
 				try {
+					if (!fc.getSelectedFile().getPath().endsWith(".png"))
+						fc.setSelectedFile(new File(fc.getSelectedFile().getPath() + ".png"));					
+
 					ImageIO.write(mapImage, "png", fc.getSelectedFile());
 				} catch (IOException e1) {
 					e1.printStackTrace();