SaveYourDrawingToFile.java Source code

Introduction

Here is the source code for SaveYourDrawingToFile.java

Source

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Graphics;
import java.awt.Panel;
import java.awt.Point;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Vector;

import javax.swing.JButton;
import javax.swing.JFrame;

public class SaveYourDrawingToFile extends JFrame implements MouseListener, ActionListener {
    List<Point> displayList = new ArrayList<Point>();
    String pathname = "data.dat";
    JButton clearBtn = new JButton("Clear");
    JButton saveBtn = new JButton("Save");
    JButton restoreBtn = new JButton("Restore");
    JButton quitBtn = new JButton("Quit");

    public static void main(String args[]) {
        SaveYourDrawingToFile that = new SaveYourDrawingToFile();
        that.setVisible(true);
    }

    public SaveYourDrawingToFile() {
        addMouseListener(this);

        setLayout(new BorderLayout());
        Panel pan = new Panel();
        clearBtn.addActionListener(this);
        pan.add(clearBtn);
        saveBtn.addActionListener(this);
        pan.add(saveBtn);
        restoreBtn.addActionListener(this);
        pan.add(restoreBtn);
        quitBtn.addActionListener(this);
        pan.add(quitBtn);
        add("North", pan);
        setSize(350, 200);
    }

    public void paint(Graphics g) {
        g.setColor(Color.white);
        g.fillRect(0, 0, getSize().width, getSize().height);

        g.setColor(Color.black);
        int i = 0;
        while (i < displayList.size()) {
            Point p0 = (Point) (displayList.get(i++));
            Point p1 = (Point) (displayList.get(i++));
            int x = Math.min(p0.x, p1.x);
            int y = Math.min(p0.y, p1.y);
            int w = Math.abs(p0.x - p1.x);
            int h = Math.abs(p0.y - p1.y);
            g.drawRect(x, y, w, h);
        }
    }

    public void mousePressed(MouseEvent e) {
        Point p = new Point(e.getX(), e.getY());
        displayList.add(p);
    }

    public void mouseReleased(MouseEvent e) {
        Point p = new Point(e.getX(), e.getY());
        displayList.add(p);
        repaint();
    }

    public void mouseClicked(MouseEvent e) {
    }

    public void mouseEntered(MouseEvent e) {
    }

    public void mouseExited(MouseEvent e) {
    }

    public void actionPerformed(ActionEvent e) {
        if (e.getSource() == clearBtn) {
            displayList = new Vector();
            repaint();
        } else if (e.getSource() == saveBtn) {
            try {
                FileOutputStream fos = new FileOutputStream(pathname);
                ObjectOutputStream oos = new ObjectOutputStream(fos);
                oos.writeObject(displayList);
                oos.flush();
                oos.close();
                fos.close();
            } catch (Exception ex) {
                System.out.println("Trouble writing display list vector");
            }
        } else if (e.getSource() == restoreBtn) {
            try {
                FileInputStream fis = new FileInputStream(pathname);
                ObjectInputStream ois = new ObjectInputStream(fis);
                displayList = (Vector) (ois.readObject());
                ois.close();
                fis.close();
                repaint();
            } catch (Exception ex) {
                System.out.println("Trouble reading display list vector");
            }
        } else if (e.getSource() == quitBtn) {
            setVisible(false);
            dispose();
            System.exit(0);
        }
    }
}