Blame | Last modification | View Log | Download
/*
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.event.*;
import java.io.*;
import java.nio.*;
import java.text.*;
import java.util.*;
import javax.swing.*;
import javax.swing.border.*;
import avr.device.event.*;
import avr.lang.*;
import avr.swing.filechooser.LogFileFilter;
public class JCaptureInternalFrame extends JInternalFrame {
private static final Format DATE_FORMAT;
private static final Format DATE_FILE_NAME_FORMAT;
static {
DATE_FORMAT = new SimpleDateFormat("MM/dd/yyyy hh:mm:ss");
DATE_FILE_NAME_FORMAT = new SimpleDateFormat("yyyyMMdd hhmmss");
}
private JMessagePanel messageP;
private JColorMapInterface colorMapP;
private JCapturePanel captureP;
private DataListener dataHandler;
private String filename;
private JLabel redValueL;
private JLabel greenValueL;
private JLabel blueValueL;
private boolean fromCamera;
public JCaptureInternalFrame(JMessagePanel messageP) {
this(messageP, null);
}
public JCaptureInternalFrame(JMessagePanel messageP, JColorMapInterface colorMapP) {
this(messageP, colorMapP, null);
}
public JCaptureInternalFrame(JMessagePanel messageP, JColorMapInterface colorMapP, File file) {
super("Capture Frame" + ((file == null) ? ": " + DATE_FORMAT.format(new Date())
: " " + file.toString()),
true, true, true, true);
this.filename = DATE_FILE_NAME_FORMAT.format(new Date()) + ".byr";
this.fromCamera = file == null;
this.messageP = messageP;
this.colorMapP = colorMapP;
setDefaultCloseOperation(DISPOSE_ON_CLOSE);
setJMenuBar(createMenuBar());
redValueL = new JLabel("");
greenValueL = new JLabel("");
blueValueL = new JLabel("");
captureP = new JCapturePanel();
if(file != null) {
try {
captureP.openBayer(file);
} catch(IOException ioe) {
AVRSystem.LOG.severe(ioe.getMessage());
ioe.printStackTrace();
}
}
captureP.addMouseMotionListener(new MouseMotionHandler());
Box southBox = new Box(BoxLayout.X_AXIS);
southBox.setBorder(new EmptyBorder(5, 5, 5, 5));
southBox.add(Box.createHorizontalGlue());
southBox.add(new JLabel("Red: "));
southBox.add(redValueL);
southBox.add(Box.createHorizontalStrut(5));
southBox.add(new JLabel("Green: "));
southBox.add(greenValueL);
southBox.add(Box.createHorizontalStrut(5));
southBox.add(new JLabel("Blue: "));
southBox.add(blueValueL);
southBox.add(Box.createHorizontalGlue());
getContentPane().add(captureP, BorderLayout.CENTER);
getContentPane().add(southBox, BorderLayout.SOUTH);
if(colorMapP != null) {
captureP.addMouseListener(new MouseHandler());
}
}
private JMenuBar createMenuBar() {
JMenuBar menubar = new JMenuBar();
JMenu fileM = new JMenu("File");
fileM.add(new ProxyAction(this, "save", "Save", 's'));
fileM.add(new ProxyAction(this, "pack", "Reset Size", 'r'));
fileM.addSeparator();
fileM.add(new ProxyAction(this, "dispose", "Exit", 'x'));
menubar.add(fileM);
return menubar;
}
public void save() {
javax.swing.filechooser.FileFilter[] filters = AVRSystem.FILE_CHOOSER.getChoosableFileFilters();
for(int i = 0; i < filters.length; i++) {
AVRSystem.FILE_CHOOSER.removeChoosableFileFilter(filters[i]);
}
AVRSystem.FILE_CHOOSER.addChoosableFileFilter(
new LogFileFilter("Bayer Image File (*." + AVRSystem.BAYER_FILE_EXT + ")",
"." + AVRSystem.BAYER_FILE_EXT));
AVRSystem.FILE_CHOOSER.setSelectedFile(new File(filename));
int option = AVRSystem.FILE_CHOOSER.showSaveDialog(getDesktopPane().getRootPane());
if(option == JFileChooser.APPROVE_OPTION) {
try {
File file = AVRSystem.FILE_CHOOSER.getSelectedFile();
if(!file.getName().toLowerCase().endsWith(AVRSystem.BAYER_FILE_EXT)) {
file = new File(file.getName() + "." + AVRSystem.BAYER_FILE_EXT);
}
captureP.saveBayer(file);
} catch(IOException ioe) {
ioe.printStackTrace();
AVRSystem.LOG.severe(ioe.getMessage());
}
}
}
public void setVisible(boolean visible) {
if(fromCamera) {
if(visible) {
try {
dataHandler = new DumpFrameHandler();
AVRSystem.DEVICE.addDataListener(dataHandler);
getDesktopPane().getRootPane().getGlassPane().setVisible(true);
AVRSystem.DEVICE.sendDumpFrame();
messageP.append("Capture Snapshot");
} catch(IOException ioe) {
AVRSystem.DEVICE.removeDataListener(dataHandler);
AVRSystem.LOG.severe(ioe.getMessage());
messageP.append("Capture not sent");
}
} else {
AVRSystem.DEVICE.removeDataListener(dataHandler);
}
}
super.setVisible(visible);
}
private Point translatePointToImage(Point mouse) {
Dimension size = captureP.getSize();
Insets insets = captureP.getInsets();
Dimension preferredSize = captureP.getPreferredSize();
double scaleX = size.width / (double)preferredSize.width;
double scaleY = size.height / (double)preferredSize.height;
double scale = Math.min(scaleX, scaleY);
int imageX = (int)(insets.left + ((5 + AVRSystem.IMAGE_WIDTH + 10) * scale));
int imageY = (int)(insets.top + (5 * scale));
int imageWidth = (int)(AVRSystem.IMAGE_WIDTH * scale);
int imageHeight = (int)(AVRSystem.IMAGE_HEIGHT * scale);
Point imagePoint = null;
if(((mouse.x >= imageX) && (mouse.x < (imageX + imageWidth))) &&
((mouse.y >= imageY) && (mouse.y < (imageY + imageHeight)))) {
scale = 1 / scale;
int x = (int)((mouse.x * scale) - (insets.left + 5 + AVRSystem.IMAGE_WIDTH + 10));
int y = (int)((mouse.y * scale) - (insets.top + 5));
imagePoint = new Point(x, y);
}
return imagePoint;
}
private final class DumpFrameHandler extends DataAdapter {
private int frameCount;
public DumpFrameHandler() {
frameCount = 0;
}
public void ack() {
frameCount = 0;
}
public void nck() {
getDesktopPane().getRootPane().getGlassPane().setVisible(false);
AVRSystem.DEVICE.removeDataListener(this);
JOptionPane.showMessageDialog(messageP.getRootPane(), "Capture NCK Received", "NCK Received", JOptionPane.ERROR_MESSAGE);
}
public void responseTimerExpired() {
getDesktopPane().getRootPane().getGlassPane().setVisible(false);
AVRSystem.DEVICE.removeDataListener(this);
JOptionPane.showMessageDialog(messageP.getRootPane(), "Response Timer Expired", "Timer Expired", JOptionPane.ERROR_MESSAGE);
}
public void frameData(ByteBuffer data) {
frameCount++;
captureP.setRow(data.get() & 0xFF, data, frameCount == 0x48);
if(frameCount == 0x48) {
AVRSystem.DEVICE.removeDataListener(this);
getDesktopPane().getRootPane().getGlassPane().setVisible(false);
}
}
}
private final class MouseHandler extends MouseAdapter {
private JPopupMenu popupM;
private int color;
public void mouseReleased(MouseEvent me) {
if(SwingUtilities.isRightMouseButton(me)) {
Point imagePoint = translatePointToImage(me.getPoint());
if(imagePoint != null) {
color = captureP.getRGB(imagePoint.x, imagePoint.y);
if(popupM == null) {
popupM = new JPopupMenu();
try {
JMenuItem colorMapMI = new JMenuItem("Add to Color Map");
colorMapMI.addActionListener(new ActionHandler());
popupM.add(colorMapMI);
} catch(Exception e) {
}
}
popupM.show((Component)me.getSource(), me.getX(), me.getY());
}
}
}
public final class ActionHandler implements ActionListener {
private JPanel displayP = null;
private JRadioButton[] colRB = null;
public void actionPerformed(ActionEvent ae) {
int column = 0;
int option = JOptionPane.YES_OPTION;
do {
option = JOptionPane.YES_OPTION;
column = getColorMapColumn();
if(column != -1) {
if(!colorMapP.isColumnClear(column)) {
option = JOptionPane.showConfirmDialog(getRootPane(),
"Index " + column + " is already set. Overwrite current value?",
"Overwrite current index value?",
JOptionPane.
YES_NO_CANCEL_OPTION,
JOptionPane.QUESTION_MESSAGE);
}
} else {
option = JOptionPane.CANCEL_OPTION;
}
} while(option == JOptionPane.NO_OPTION);
if(option == JOptionPane.YES_OPTION) {
SwingUtilities.getWindowAncestor(colorMapP).setVisible(true);
colorMapP.setColor(column, color);
}
}
private int getColorMapColumn() {
if(displayP == null) {
displayP = new JPanel(new BorderLayout());
JPanel selectColP = new JPanel();
ButtonGroup bg = new ButtonGroup();
colRB = new JRadioButton[8];
for(int i = 0; i < colRB.length; i++) {
colRB[i] = new JRadioButton((i + 1) + "");
bg.add(colRB[i]);
selectColP.add(colRB[i]);
}
colRB[0].setSelected(true);
displayP.add(new JLabel("Select Color Map Column:"), BorderLayout.NORTH);
displayP.add(selectColP, BorderLayout.SOUTH);
}
int option = JOptionPane.showConfirmDialog(getRootPane(),
displayP,
"Add to Color Map Column",
JOptionPane.OK_CANCEL_OPTION,
JOptionPane.QUESTION_MESSAGE);
int selected = -1;
if(option == JOptionPane.OK_OPTION) {
for(int i = 0; selected == -1 && i < colRB.length; i++) {
if(colRB[i].isSelected()) {
selected = i;
}
}
}
return selected;
}
}
}
private final class MouseMotionHandler implements MouseMotionListener {
public void mouseMoved(MouseEvent me) {
Point imagePoint = translatePointToImage(me.getPoint());
if(imagePoint != null) {
int color = captureP.getRGB(imagePoint.x, imagePoint.y);
redValueL.setText(((color & 0xFF0000) >> 16) + "");
greenValueL.setText(((color & 0x00FF00) >> 8) + "");
blueValueL.setText(((color & 0x0000FF) >> 0) + "");
} else {
redValueL.setText("");
greenValueL.setText("");
blueValueL.setText("");
}
}
public void mouseDragged(MouseEvent me) {
}
}
}