diff --git a/src/main/java/org/academiadecodigo/simplegraphics/graphics/Canvas.java b/src/main/java/org/academiadecodigo/simplegraphics/graphics/Canvas.java index 68d348f..bffcd55 100755 --- a/src/main/java/org/academiadecodigo/simplegraphics/graphics/Canvas.java +++ b/src/main/java/org/academiadecodigo/simplegraphics/graphics/Canvas.java @@ -28,6 +28,7 @@ private Canvas() { component = new CanvasComponent(); frame = new JFrame(); + frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE); frame.add(component); frame.pack(); frame.setLocation(LOCATION_OFFSET, LOCATION_OFFSET); @@ -171,4 +172,19 @@ public Dimension getPreferredSize() { return new Dimension(maxx + MARGIN, maxy + MARGIN); } } + + public void setTitle(String title){ + frame.setTitle(title); + } + + public void setIcon(String imagePath){ + try { + File pathToFile = new File(imagePath); + Image image = ImageIO.read(pathToFile); + + frame.setIconImage(image); + } catch (IOException e) { + System.err.println("Was unable to Find Image."); + } + } }