Search code examples
javaswingimage-processinggrayscaleimageicon

Change ImageIcon being displayed in JFrame


I have here this code. I want it to first display an image file as input by the user when calling the main method, then apply changes to it. My first function is grayscale(). I have provided a grayscale button in a JMenuBar such that when clicked, it will create a grayscale version of the current image. It works, however I can't get the new image to display in the JFrame after the method is applied. I tried calling show(); within the method, but that just opens the original image again. I know the grayscale function is doing its job, it just doesn't display the resulting Image after. How can I get it to show image2 after it is created? Thanks for the help.

import java.awt.*;
import java.awt.event.*;
import java.util.*;
import java.io.*;
import javax.swing.*;
import java.awt.image.*;
import javax.imageio.ImageIO;

public class ImageEdit{
    Container content;
    static BufferedImage image;
    BufferedImage  image2;

    public ImageEdit(String filename) {
        File f = new File(filename);
        //assume file is the image file
        try {
            image = ImageIO.read(f);
        } 
        catch (IOException e) {
            System.out.println("Invalid image file: " + filename);
            System.exit(0);
        }
    }

    public void show() {
        final int width = image.getWidth();
        final int height = image.getHeight();

        JFrame frame = new JFrame("Edit Picture");

        //set frame title, set it visible, etc
        content = frame.getContentPane();
        frame.setResizable(false);
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

        //add the image to the frame
        ImageIcon icon = new ImageIcon(image);
        frame.setContentPane(new JLabel(icon));
        frame.pack();

        //add a menubar on the frame with a single option: saving the image
        JMenuBar menuBar = new JMenuBar();
        frame.setJMenuBar(menuBar);
        JMenu fileMenu = new JMenu("File");
        menuBar.add(fileMenu);
        JMenuItem saveAction = new JMenuItem("Save");
        fileMenu.add(saveAction);
        JMenuItem grayScale = new JMenuItem("Grayscale");
        fileMenu.add(grayScale);
        grayScale.addActionListener(new ActionListener() {
                public void actionPerformed(ActionEvent e) {
                    grayscale(width, height);
                }
            });

        //paint the frame
        frame.setVisible(true);
    }

    public void grayscale(int width, int height) {
        // create a grayscale image the same size
        image2 = new BufferedImage(width, height, BufferedImage.TYPE_BYTE_GRAY);

        // convert the original colored image to grayscale
        ColorConvertOp grayScale = new ColorConvertOp(
            image.getColorModel().getColorSpace(),
        image2.getColorModel().getColorSpace(),null);
        grayScale.filter(image,image2);
        show();
    }

    public static void main(String[] args) {
        ImageEdit p = new ImageEdit(args[0]);
        p.show();
    }
}

Solution

  • import java.awt.*;
    import java.awt.image.*;
    import java.awt.event.*;
    import javax.swing.*;
    
    public class ImageEdit{
        Container content;
        BufferedImage image;
        BufferedImage  image2;
        JLabel imageLabel;
    
        public ImageEdit(BufferedImage image) {
            this.image = image;
        }
    
        public void show() {
            final int width = image.getWidth();
            final int height = image.getHeight();
    
            JFrame frame = new JFrame("Edit Picture");
    
            //set frame title, set it visible, etc
            content = frame.getContentPane();
            //frame.setResizable(false);
            frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    
            //add the image to the frame
            ImageIcon icon = new ImageIcon(image);
            imageLabel = new JLabel(icon);
            frame.setContentPane(imageLabel);
    
            //add a menubar on the frame with a single option: saving the image
            JMenuBar menuBar = new JMenuBar();
            frame.setJMenuBar(menuBar);
            JMenu fileMenu = new JMenu("File");
            menuBar.add(fileMenu);
            JMenuItem saveAction = new JMenuItem("Save");
            fileMenu.add(saveAction);
            JMenuItem grayScale = new JMenuItem("Grayscale");
            fileMenu.add(grayScale);
            grayScale.addActionListener(new ActionListener() {
                    public void actionPerformed(ActionEvent e) {
                        grayscale(width, height);
                    }
                });
    
            //paint the frame
            frame.pack();
            frame.setVisible(true);
        }
    
        public void grayscale(int width, int height) {
            // create a grayscale image the same size
            image2 = new BufferedImage(width, height, BufferedImage.TYPE_BYTE_GRAY);
    
            // convert the original colored image to grayscale
            ColorConvertOp grayScale = new ColorConvertOp(
                image.getColorModel().getColorSpace(),
            image2.getColorModel().getColorSpace(),null);
            grayScale.filter(image,image2);
            imageLabel.setIcon(new ImageIcon(image2));
            //show();
        }
    
        public static void main(String[] args) {
            int size = 120;
            int pad = 10;
            BufferedImage bi = new BufferedImage(
                size,
                size,
                BufferedImage.TYPE_INT_RGB);
            Graphics g = bi.createGraphics();
            g.setColor(Color.WHITE);
            g.fillRect(0,0,size,size);
            g.setColor(Color.YELLOW);
            g.fillOval(pad,pad,size-(2*pad),size-(2*pad));
            g.dispose();
    
            ImageEdit p = new ImageEdit(bi);
            p.show();
        }
    }