初期コミット。

kinmojr · Feb 15, 2017 · 85f52df · 85f52df
1 parent d92e054
commit 85f52df
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/main/java/to/kishimo/minist/ImageViewer.java b/src/main/java/to/kishimo/minist/ImageViewer.java
@@ -2,6 +2,7 @@
 
 import javax.imageio.ImageIO;
 import javax.swing.*;
+import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
@@ -66,7 +67,11 @@ public void showImageAsText(int index) {
     public void showImage(int index) {
         BufferedImage image = makeImage(index);
         Icon icon = new ImageIcon(image);
-        JOptionPane.showMessageDialog(null, labels[index], "MnistImageViewer", JOptionPane.PLAIN_MESSAGE, icon);
+        try {
+            JOptionPane.showMessageDialog(null, labels[index], "MnistImageViewer", JOptionPane.PLAIN_MESSAGE, icon);
+        } catch (HeadlessException e) {
+            System.out.println("Image dialog can't be displayed on CUI environment.");
+        }
     }
 
     /**