Maze Progress Bar - Main

import java.awt.image.*;
import javax.imageio.*;
import java.io.*;
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
import java.beans.*;
import javax.swing.filechooser.*;
import java.lang.Math;

public class Main extends JPanel implements ActionListener, PropertyChangeListener {

  static int w = 500;
  static int h = 500;
  static int baseCol = -5708545;

  private JLabel iterLabel, pic;
  private JTextField iterField;
  private JFileChooser fc;
  private JProgressBar progressBar;
  private JButton startButton;
  private Task task;

  class Task extends SwingWorker<Void, Void> {
    /*
     * Main task. Executed in background thread.
     */
    @Override
    public Void doInBackground() {
      RecursiveGen rg = new RecursiveGen(w/2, h/2);
      setProgress(0);
      BufferedImage bi = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);

      /*for (int y = 0; y < h; y++) {
        for (int x = 0; x < w; x++) {
          bi.setRGB(x, y, (int)(Math.random()*-16777217));
        }
      }*/

      //for (int n = 0; n < 10; n++) {
      //int iterations = Integer.parseInt(iterField.getText());
      /*for (int i = 0; i < iterations; i++) {
       bi = smooth(bi);
       pic.setIcon(new ImageIcon(bi));
       setProgress((int)(i*100f/iterations+.5f));
       }*/

      while (!rg.checkCompleted()) {
        try {
          rg.step();
          /*bi = formatImg(rg.step());
          pic.setIcon(new ImageIcon(bi));*/
          setProgress(rg.getProgress());
          try {
            Thread.sleep(1);
          } catch (InterruptedException ignore) {
          }
        } catch (Exception ex) {
          //System.err.println(ex);
        }
      }
      trySave(formatImg(rg.getMaze()));
      //}
      setProgress(100);
      return null;
    }

    /*
     * Executed in event dispatching thread
     */
    @Override
    public void done() {
      Toolkit.getDefaultToolkit().beep();
      startButton.setEnabled(true);
      setCursor(null); //turn off the wait cursor
    }
  }

  public BufferedImage formatImg(boolean[][][] m) {
    BufferedImage b = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);
    for (int y = 0; y < m[0].length; y++) {
      for (int x = 0; x < m.length; x++) {
        b.setRGB(x*2+1, y*2+1, Color.red.getRGB());
        if (!m[x][y][1]) {
          b.setRGB(x*2+2, y*2+1, Color.red.getRGB());
        }
        if (!m[x][y][2]) {
          b.setRGB(x*2+1, y*2+2, Color.red.getRGB());
        }
      }
    }
    return b;
  }

  public void actionPerformed(ActionEvent evt) {
    if (evt.getSource() == startButton) {
      startButton.setEnabled(false);
      setCursor(Cursor.getPredefinedCursor(Cursor.WAIT_CURSOR));
      //Instances of javax.swing.SwingWorker are not reusuable, so
      //we create new instances as needed.
      task = new Task();
      task.addPropertyChangeListener(this);
      task.execute();
      //debug();
    }
  }

  public void propertyChange(PropertyChangeEvent evt) {
    if ("progress" == evt.getPropertyName()) {
      int progress = (Integer) evt.getNewValue();
      progressBar.setValue(progress);
    }
  }

  public static BufferedImage smooth(BufferedImage bi) {
    BufferedImage img = new BufferedImage(bi.getWidth(), bi.getHeight(), bi.getType());
    for (int y = 0; y < h; y++) {
      for (int x = 0; x < w; x++) {
        img.setRGB(x, y, getAverageColor(x, y, bi));
      }
    }
    return img;
  }

  public static int getAverageRGB(int x, int y, BufferedImage template) {
    int n = 0;
    int d = 0;
    if (x > 0) {
      n+=template.getRGB(x-1, y);
      d++;
    }
    if (y > 0) {
      n+=template.getRGB(x, y-1);
      d++;
    }
    if (x < template.getWidth()-1) {
      n+=template.getRGB(x+1, y);
      d++;
    }
    if (y < template.getHeight()-1) {
      n+=template.getRGB(x, y+1);
      d++;
    }
    if (x > 0 && y > 0) {
      n+=template.getRGB(x-1, y-1);
      d++;
    }
    if (y > 0 && x < template.getWidth()-1) {
      n+=template.getRGB(x+1, y-1);
      d++;
    }
    if (x < template.getWidth()-1 && y < template.getHeight()-1) {
      n+=template.getRGB(x+1, y+1);
      d++;
    }
    if (y < template.getHeight()-1 && x > 0) {
      n+=template.getRGB(x-1, y+1);
      d++;
    }
    return (int)((n*1f/d)+.5f);
  }

  public static int getAverageColor(int x, int y, BufferedImage template) {
    int nRed = 0;
    int nGreen = 0;
    int nBlue = 0;
    int d = 0;
    if (x > 0) {
      Color col = new Color(template.getRGB(x-1, y));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (y > 0) {
      Color col = new Color(template.getRGB(x, y-1));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (x < template.getWidth()-1) {
      Color col = new Color(template.getRGB(x+1, y));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (y < template.getHeight()-1) {
      Color col = new Color(template.getRGB(x, y+1));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (x > 0 && y > 0) {
      Color col = new Color(template.getRGB(x-1, y-1));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (y > 0 && x < template.getWidth()-1) {
      Color col = new Color(template.getRGB(x+1, y-1));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (x < template.getWidth()-1 && y < template.getHeight()-1) {
      Color col = new Color(template.getRGB(x+1, y+1));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    if (y < template.getHeight()-1 && x > 0) {
      Color col = new Color(template.getRGB(x-1, y+1));
      nRed += col.getRed();
      nGreen += col.getGreen();
      nBlue += col.getBlue();
      d++;
    }
    return new Color((int)((nRed*1f/d)+.5f), (int)((nGreen*1f/d)+.5f), (int)((nBlue*1f/d)+.5f)).getRGB();
  }

  public Main() {
    super(new BorderLayout());

    Dimension screen = Toolkit.getDefaultToolkit().getScreenSize();
    if (w == 0 && h == 0) {
      w = (int)screen.getWidth();
      h = (int)screen.getHeight();
    }
    System.out.println(w + ":" + h);

    fc = new JFileChooser();
    FileNameExtensionFilter filter = new FileNameExtensionFilter("Image files", "png", "jpg", "jpeg", "bmp");
    fc.setFileFilter(filter);
    //Create the UI.

    pic = new JLabel();

    startButton = new JButton("Start");
    startButton.setActionCommand("start");
    startButton.addActionListener(this);

    progressBar = new JProgressBar(0, 100);
    progressBar.setValue(0);
    progressBar.setStringPainted(true);

    iterLabel = new JLabel("Iterations:");
    iterField = new JTextField("100");
    iterField.addActionListener(this);

    JPanel panel = new JPanel();
    panel.add(startButton);
    panel.add(progressBar);
    panel.add(iterLabel);
    panel.add(iterField);

    panel.add(pic);

    add(panel, BorderLayout.PAGE_START);
    setBorder(BorderFactory.createEmptyBorder(20, 20, 20, 20));

  }

  private static void createAndShowGUI() {
    //Create and set up the window.
    JFrame frame = new JFrame("Smooth Image");
    frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

    //Create and set up the content pane.
    JComponent newContentPane = new Main();
    newContentPane.setOpaque(true); //content panes must be opaque
    frame.setContentPane(newContentPane);

    //Display the window.
    frame.pack();
    frame.setVisible(true);
  }

  public void trySave(BufferedImage bi) {
    int returnVal = fc.showSaveDialog(this);
    if (returnVal == JFileChooser.APPROVE_OPTION) {
      File currentFile = fc.getSelectedFile();
      String filePath = currentFile.getPath();
      if (filePath.lastIndexOf(".") != -1) {
        filePath = currentFile.getPath().substring(0, filePath.lastIndexOf("."));
      }
      filePath = filePath + ".png";

      try {
        ImageIO.write(bi, "png", new File(filePath));
      } catch(IOException e) {
      }
    }
  }


  public void debug() {
    RecursiveGen rg = new RecursiveGen(w/2, h/2);
      BufferedImage bi = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);

      for (int y = 0; y < h; y++) {
        for (int x = 0; x < w; x++) {
          bi.setRGB(x, y, (int)(Math.random()*-16777217));
        }
      }

      while (!rg.checkCompleted()) {
        bi = formatImg(rg.step());
        pic.setIcon(new ImageIcon(bi));
      }
  }


  public static void main(String[] args) {
    //Schedule a job for the event-dispatching thread:
    //creating and showing this application's GUI.

    javax.swing.SwingUtilities.invokeLater(new Runnable() {
      public void run() {
        createAndShowGUI();
      }
    });
  }

}