Subversion Repositories svnkaklik

/*

    AVRcamVIEW: A PC application to test out the functionallity of the

     AVRcam real-time image processing engine.

    Copyright (C) 2004    Brent A. Taylor

    This program is free software; you can redistribute it and/or

    modify it under the terms of the GNU General Public

    License as published by the Free Software Foundation; either

    version 2 of the License, or (at your option) any later version.

    This program is distributed in the hope that it will be useful,

    but WITHOUT ANY WARRANTY; without even the implied warranty of

    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    General Public License for more details.

    You should have received a copy of the GNU General Public

    License along with this program; if not, write to the Free Software

    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

   For more information on the AVRcamVIEW, please contact:

   taylorba@comcast.net

   or go to www.jrobot.net for more details regarding the system.

*/

package avr.swing;

import java.awt.*;

import java.awt.image.*;

import java.io.*;

import java.nio.*;

import java.nio.channels.*;

import javax.swing.*;

import avr.lang.*;

public class JCapturePanel extends JPanel {

   // constant variables for the interpolate state machine

   private static final int RED        = 0x01;

   private static final int BLUE       = 0x02;

   private static final int GREEN_EVEN = 0x04;

   private static final int GREEN_ODD  = 0x08;

   private BufferedImage bayerImage;

   private BufferedImage image;

   public JCapturePanel() {

      super(null);

      bayerImage = new BufferedImage(AVRSystem.IMAGE_WIDTH, AVRSystem.IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);

      image = new BufferedImage(AVRSystem.IMAGE_WIDTH, AVRSystem.IMAGE_HEIGHT, BufferedImage.TYPE_INT_RGB);

   }

   public void paintComponent(Graphics g) {

      super.paintComponent(g);

      if(bayerImage != null) {

         Dimension size = getSize();

         Insets insets = getInsets();

         Image offscreenImage = this.createImage(size.width, size.height);

         Graphics2D g2d = (Graphics2D)offscreenImage.getGraphics();

         // scale the images so they fit side by side for either the full

         // width or full height of the window

         double xScale = size.width / (double)(insets.left + 5 + AVRSystem.IMAGE_WIDTH + 10 + AVRSystem.IMAGE_WIDTH + 5 + insets.right);

         double yScale = size.height / (double)(insets.top + 5 + AVRSystem.IMAGE_HEIGHT + 5 + insets.bottom);

         double scale = Math.min(xScale, yScale);

         g2d.scale(scale, scale);

         g2d.drawImage(bayerImage, insets.left + 5, insets.top + 5, null);

         g2d.drawImage(image, insets.left + 5 + AVRSystem.IMAGE_WIDTH + 10, insets.top + 5, null);

         g.drawImage(offscreenImage, 0, 0, null);

      }

   }

   public Dimension getMinimumSize() {

      return getPreferredSize();

   }

   public Dimension getPreferredSize() {

      Insets insets = getInsets();

      return new Dimension(insets.left + 5 + AVRSystem.IMAGE_WIDTH + 10 + AVRSystem.IMAGE_WIDTH + 5 + insets.right,

                           insets.top + 5 + AVRSystem.IMAGE_HEIGHT + 5 + insets.bottom);

   }

   public Dimension getMaximumSize() {

      return getPreferredSize();

   }

   public int getRGB(int x, int y) {

      return image.getRGB(x, y);

   }

   public void openBayer(File file) throws IOException {

      FileInputStream inStream = new FileInputStream(file);

      FileChannel inChannel = inStream.getChannel();

      ByteBuffer[] buffers = new ByteBuffer[bayerImage.getWidth()];

      for(int x = 0; x < buffers.length; x++) {

         buffers[x] = ByteBuffer.allocate(bayerImage.getHeight() * 4);

      }

      inChannel.read(buffers);

      inStream.close();

      inChannel.close();

      int[] pixels = new int[bayerImage.getHeight()];

      for(int x = 0; x < bayerImage.getWidth(); x++) {

         buffers[x].flip();

         buffers[x].asIntBuffer().get(pixels);

         bayerImage.setRGB(x, 0, 1, pixels.length, pixels, 0, 1);

      }

      interpolate();

   }

   public void saveBayer(File file) throws IOException {

      ByteBuffer[] buffers = new ByteBuffer[bayerImage.getWidth()];

      for(int x = 0; x < buffers.length; x++) {

         buffers[x] = ByteBuffer.allocate(bayerImage.getHeight() * 4);

         for(int y = 0; y < bayerImage.getHeight(); y++) {

            buffers[x].putInt(bayerImage.getRGB(x, y));

         }

         buffers[x].flip();

      }

      FileOutputStream outStream = new FileOutputStream(file);

      FileChannel outChannel = outStream.getChannel();

      outChannel.write(buffers);

      outStream.close();

      outChannel.close();

   }

   public void setRow(int row, ByteBuffer data, boolean finished) {

      int[] pixels = new int[AVRSystem.IMAGE_WIDTH * 2];

      int x = 0;

      while(data.hasRemaining()) {

         byte pixel = data.get();

         if((x & 1) == 0) {

            // green

            pixels[x] = (pixel & 0xF0) << 8;

            // blue

            pixels[AVRSystem.IMAGE_WIDTH + x] = (pixel & 0x0F) << 4;

         } else {

            // green

            pixels[AVRSystem.IMAGE_WIDTH + x] = ((pixel & 0x0F) << 4) << 8;

            // red

            pixels[x] = (pixel & 0xF0) << 16;

         }

         x++;

      }

      bayerImage.setRGB(0, row * 2, AVRSystem.IMAGE_WIDTH, 2, pixels, 0, AVRSystem.IMAGE_WIDTH);

      if(finished) {

         interpolate();

      }

      repaint();

   }

   /* ********************************************************

    * The bayerImage is in the bayer format shown below.

    *

    *      |   |   |   |   |   |   |   |   |   | . | 1 | 1 |

    *      |   |   |   |   |   |   |   |   |   | . | 7 | 7 |

    *      | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | . | 4 | 5 |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    0 | G | R | G | R | G | R | G | R | G | . | G | R |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    1 | B | G | B | G | B | G | B | G | B | . | B | G |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    2 | G | R | G | R | G | R | G | R | G | . | G | R |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    3 | B | G | B | G | B | G | B | G | B | . | B | G |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    4 | G | R | G | R | G | R | G | R | G | . | G | R |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    5 | B | G | B | G | B | G | B | G | B | . | B | G |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *    .   .   .   .   .   .   .   .   .   .   .   .   .

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *  142 | G | R | G | R | G | R | G | R | G | . | G | R |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *  143 | B | G | B | G | B | G | B | G | B | . | B | G |

    * -----+---+---+---+---+---+---+---+---+---+ . +---+---+

    *

    * The corners are calculated, then the edges, then the center.

    *

    */

   private void interpolate() {

      int red = 0;

      int green = 0;

      int blue = 0;

      int currColor = GREEN_ODD;

      int nextColor = RED;

      int width = AVRSystem.IMAGE_WIDTH;

      int height = AVRSystem.IMAGE_HEIGHT;

      // *** do the corners of the image ***

      // upper left corner

      red   = bayerImage.getRGB(1, 0);

      green = bayerImage.getRGB(0, 0);

      blue  = bayerImage.getRGB(0, 1);

      image.setRGB(0, 0, red | green | blue);

      // upper right corner

      red   = bayerImage.getRGB(width - 1, 0);

      green = (bayerImage.getRGB(width - 2, 0) + bayerImage.getRGB(width - 1, 1)) / 2;

      blue  = bayerImage.getRGB(width - 2, 1);

      image.setRGB(width - 1, 0, red | green | blue);

      // lower left corner

      red   = bayerImage.getRGB(1, height - 2);

      green = (bayerImage.getRGB(0, height - 2) + bayerImage.getRGB(1, height - 1)) / 2;

      blue  = bayerImage.getRGB(0, height - 1);

      image.setRGB(0, height - 1, red | green | blue);

      // lower right corner

      red   = bayerImage.getRGB(width - 1, height - 2);

      green = bayerImage.getRGB(width - 1, height - 1);

      blue  = bayerImage.getRGB(width - 2, height - 1);

      image.setRGB(width - 1, height - 1, red | green | blue);

      // *** do the north edge

      currColor = RED;

      for(int x = 1, y = 0; x < width - 1; x++) {

         switch(currColor) {

            case RED:

               red   = bayerImage.getRGB(x, y);

               green = (bayerImage.getRGB(x + 1, y) + bayerImage.getRGB(x, y + 1) + bayerImage.getRGB(x - 1, y)) / 3;

               blue  = (bayerImage.getRGB(x + 1, y + 1) + bayerImage.getRGB(x - 1, y + 1)) / 2;

               nextColor = GREEN_EVEN;

               break;

            case GREEN_EVEN:

               red   = (bayerImage.getRGB(x - 1, y) + bayerImage.getRGB(x + 1, y)) / 2;

               green = bayerImage.getRGB(x, y);

               blue  = bayerImage.getRGB(x, y + 1);

               nextColor = RED;

               break;

            default:

               AVRSystem.LOG.warning("Invalid color for row start: " + currColor);

         }

         currColor = nextColor;

         image.setRGB(x, y, red | green | blue);

      }

      // *** do the west edge

      currColor = BLUE;

      for(int y = 1, x = 0; y < height - 1; y++) {

         switch(currColor) {

            case BLUE:

               red   = (bayerImage.getRGB(x + 1, y + 1) + bayerImage.getRGB(x + 1, y - 1)) / 2;

               green = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x + 1, y) + bayerImage.getRGB(x, y + 1)) / 3;

               blue  = bayerImage.getRGB(x, y);

               nextColor = GREEN_EVEN;

               break;

            case GREEN_EVEN:

               red   = bayerImage.getRGB(x + 1, y);

               green = bayerImage.getRGB(x, y);

               blue  = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x, y + 1)) / 2;

               nextColor = BLUE;

               break;

         }

         currColor = nextColor;

         image.setRGB(x, y, red | green | blue);

      }

      // *** do the east edge

      currColor = GREEN_ODD;

      for(int y = 1, x = width - 1; y < height - 1; y++) {

         switch(currColor) {

            case RED:

               blue  = (bayerImage.getRGB(x - 1, y + 1) + bayerImage.getRGB(x - 1, y - 1)) / 2;

               green = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x - 1, y) + bayerImage.getRGB(x, y + 1)) / 3;

               red   = bayerImage.getRGB(x, y);

               nextColor = GREEN_EVEN;

               break;

            case GREEN_ODD:

               blue  = bayerImage.getRGB(x - 1, y);

               green = bayerImage.getRGB(x, y);

               red   = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x, y + 1)) / 2;

               nextColor = BLUE;

               break;

         }

         currColor = nextColor;

         image.setRGB(x, y, red | green | blue);

      }

      // *** do the south edge

      currColor = GREEN_ODD;

      for(int x = 1, y = height - 1; x < width - 1; x++) {

         switch(currColor) {

            case GREEN_ODD:

               red   = bayerImage.getRGB(x, y - 1);

               green = bayerImage.getRGB(x, y);

               blue  = (bayerImage.getRGB(x - 1, y) + bayerImage.getRGB(x + 1, y)) / 2;

               nextColor = BLUE;

               break;

            case BLUE:

               red   = (bayerImage.getRGB(x - 1, y - 1) + bayerImage.getRGB(x + 1, y - 1)) / 2;

               green = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x + 1, y) + bayerImage.getRGB(x - 1, y)) / 3;

               blue  = bayerImage.getRGB(x, y);

               nextColor = GREEN_ODD;

               break;

            default:

               AVRSystem.LOG.warning("Invalid color for row start: " + currColor);

         }

         currColor = nextColor;

         image.setRGB(x, y, red | green | blue);

      }

      // *** do the center box ***

      currColor = RED;

      for(int y = 1; y < height - 1; y++) {

         // change the starting color for the row

         switch(currColor) {

            case RED:

               currColor = GREEN_ODD;

               break;

            case GREEN_ODD:

               currColor = RED;

               break;

            default:

               AVRSystem.LOG.warning("Invalid color for row start: " + currColor);

         }

         for(int x = 1; x < width - 1; x++) {

            switch(currColor) {

               case RED:

                  red   = bayerImage.getRGB(x, y);

                  green = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x + 1, y) +

                           bayerImage.getRGB(x, y + 1) + bayerImage.getRGB(x - 1, y)) / 4;

                  blue  = (bayerImage.getRGB(x - 1, y - 1) + bayerImage.getRGB(x + 1, y + 1) +

                           bayerImage.getRGB(x + 1, y - 1) + bayerImage.getRGB(x - 1, y + 1)) / 4;

                  nextColor = GREEN_EVEN;

                  break;

               case GREEN_EVEN:

                  red   = (bayerImage.getRGB(x - 1, y) + bayerImage.getRGB(x + 1, y)) / 2;

                  green = bayerImage.getRGB(x, y);

                  blue  = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x, y + 1)) / 2;

                  nextColor = RED;

                  break;

               case GREEN_ODD:

                  red   = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x, y + 1)) / 2;

                  green = bayerImage.getRGB(x, y);

                  blue  = (bayerImage.getRGB(x - 1, y) + bayerImage.getRGB(x + 1, y)) / 2;

                  nextColor = BLUE;

                  break;

               case BLUE:

                  red   = (bayerImage.getRGB(x - 1, y - 1) + bayerImage.getRGB(x + 1, y + 1) +

                           bayerImage.getRGB(x + 1, y - 1) + bayerImage.getRGB(x - 1, y + 1)) / 4;

                  green = (bayerImage.getRGB(x, y - 1) + bayerImage.getRGB(x + 1, y) +

                           bayerImage.getRGB(x, y + 1) + bayerImage.getRGB(x - 1, y)) / 4;

                  blue  = bayerImage.getRGB(x, y);

                  nextColor = GREEN_ODD;

                  break;

               default:

                  AVRSystem.LOG.warning("Invalid color: " + currColor);

            }

            currColor = nextColor;

            image.setRGB(x, y, red | green | blue);

         }

      }

   }

}