| 0,0 → 1,354 |
|---|
| /* |
| 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.nio.channels.*; |
| import java.util.*; |
| import javax.swing.*; |
| import javax.swing.border.*; |
| import javax.swing.event.*; |
| |
| import avr.lang.*; |
| import avr.connection.event.ConnectionListener; |
| import avr.connection.event.ConnectionEvent; |
| |
| public class JTrackingInternalFrame extends JInternalFrame { |
| |
| private static final int BLOB_LENGTH = 5; |
| |
| private JTrackingPanel trackingP; |
| |
| private AbstractButton recordB; |
| private AbstractButton sendCameraDataB; |
| |
| private Action recordAction; |
| private Action playAction; |
| private Action pauseAction; |
| private Action stopAction; |
| |
| private FileInputStream inStream; |
| private FileChannel inChannel; |
| |
| private Map indexMap; |
| private int numTracked; |
| |
| private boolean paused; |
| |
| private javax.swing.Timer timer; |
| |
| private JSlider playbackS; |
| |
| private ConnectionListener connectionHandler; |
| |
| // common initializer for both constructors |
| { |
| trackingP = new JTrackingPanel(); |
| |
| createActions(); |
| |
| setJMenuBar(createMenuBar()); |
| getContentPane().add(trackingP, BorderLayout.CENTER); |
| } |
| |
| public JTrackingInternalFrame() { |
| super("Tracking", true, false, true, false); |
| getContentPane().add(createToolBar(), BorderLayout.NORTH); |
| } |
| |
| public JTrackingInternalFrame(File trackingData) throws FileNotFoundException, IOException { |
| super("Tracking: " + trackingData.toString(), true, true, true, true); |
| |
| inStream = new FileInputStream(trackingData); |
| inChannel = inStream.getChannel(); |
| |
| indexMap = new HashMap(); |
| numTracked = -1; |
| |
| paused = false; |
| |
| // build an index map to map an index to a position in a file |
| indexMap.put(new Integer(0), new Integer(0)); |
| |
| ByteBuffer numTrackedBuffer = ByteBuffer.allocate(1); |
| int currentIndex = 0; |
| |
| while(inChannel.read(numTrackedBuffer) != -1) { |
| |
| numTrackedBuffer.flip(); |
| numTracked = numTrackedBuffer.get() & 0xFF; |
| |
| currentIndex++; |
| int position = (int)(inChannel.position() + (numTracked * BLOB_LENGTH)); |
| |
| indexMap.put(new Integer(currentIndex), new Integer(position)); |
| |
| inChannel.position(position); |
| numTrackedBuffer.clear(); |
| |
| } |
| |
| sendCameraDataB = new JCheckBox(new ProxyAction(this, "sendCameraData", "Send To Serial Port")); |
| sendCameraDataB.setEnabled(AVRSystem.DEVICE.isConnected()); |
| |
| playbackS = new JSlider(JSlider.HORIZONTAL, 0, currentIndex - 1, 0); |
| playbackS.setMinorTickSpacing(1); |
| playbackS.setMajorTickSpacing(20); |
| playbackS.setPaintTicks(true); |
| playbackS.setPaintTrack(true); |
| playbackS.setPaintLabels(true); |
| playbackS.setSnapToTicks(true); |
| playbackS.addChangeListener(new PlaybackHandler()); |
| |
| JPanel southP = new JPanel(new BorderLayout()); |
| southP.setBorder(new EmptyBorder(5, 0, 0, 0)); |
| |
| // JPanel controlP = new JPanel(new BorderLayout()); |
| JPanel controlP = new JPanel(); |
| |
| controlP.add(new JButton(playAction)); |
| controlP.add(new JButton(pauseAction)); |
| controlP.add(new JButton(stopAction)); |
| |
| // |
| // JPanel controlNorthP = new JPanel(); |
| // controlNorthP.add(new JButton(playAction)); |
| // controlNorthP.add(new JButton(pauseAction)); |
| // controlNorthP.add(new JButton(stopAction)); |
| // |
| // JPanel controlSouthP = new JPanel(); |
| // controlSouthP.add(sendCameraDataB); |
| // |
| // controlP.add(controlNorthP, BorderLayout.NORTH); |
| // controlP.add(controlSouthP, BorderLayout.SOUTH); |
| |
| southP.add(playbackS, BorderLayout.NORTH); |
| southP.add(controlP, BorderLayout.SOUTH); |
| |
| getContentPane().add(southP, BorderLayout.SOUTH); |
| |
| trackingP.setTrackingData(read(0)); |
| |
| connectionHandler = new ConnectionHandler(); |
| AVRSystem.DEVICE.addConnectionListener(connectionHandler); |
| |
| } |
| |
| private void createActions() { |
| recordAction = new ProxyAction(this, "record", "Start Recording", 's'); |
| playAction = new ProxyAction(this, "play", "Play", 'p'); |
| stopAction = new ProxyAction(this, "stop", "Stop", 's'); |
| pauseAction = new ProxyAction(this, "pause", "Pause"); |
| |
| stopAction.setEnabled(false); |
| pauseAction.setEnabled(false); |
| |
| } |
| |
| private JMenuBar createMenuBar() { |
| |
| JMenuBar menubar = new JMenuBar(); |
| |
| JMenu fileM = new JMenu("File"); |
| |
| fileM.add(new ProxyAction(this, "pack", "Reset Size", 'R')); |
| fileM.addSeparator(); |
| fileM.add(new ProxyAction(this, "dispose", "Exit", 'X')); |
| |
| menubar.add(fileM); |
| |
| return menubar; |
| |
| } |
| |
| private JToolBar createToolBar() { |
| |
| JToolBar toolbar = new JToolBar(); |
| toolbar.setFloatable(false); |
| |
| recordB = new JToggleButton(recordAction); |
| |
| toolbar.add(recordB); |
| |
| return toolbar; |
| |
| } |
| |
| public void dispose() { |
| if(inStream != null) { |
| if(timer != null) { |
| stop(); |
| } |
| try { |
| inStream.close(); |
| inChannel.close(); |
| } catch(Exception e) { |
| e.printStackTrace(); |
| } |
| } |
| AVRSystem.DEVICE.removeConnectionListener(connectionHandler); |
| super.dispose(); |
| } |
| |
| public void play() { |
| if(timer == null) { |
| // set the timer to fire for 30 frames per second |
| timer = new javax.swing.Timer((int)(1000 / 30), new UpdateSliderHandler()); |
| } |
| timer.start(); |
| playAction.setEnabled(false); |
| stopAction.setEnabled(true); |
| pauseAction.setEnabled(true); |
| } |
| |
| public void pause() { |
| if(!paused) { |
| timer.stop(); |
| playAction.setEnabled(false); |
| stopAction.setEnabled(false); |
| pauseAction.putValue(Action.NAME, "Resume"); |
| paused = true; |
| } else { |
| timer.start(); |
| playAction.setEnabled(false); |
| stopAction.setEnabled(true); |
| pauseAction.putValue(Action.NAME, "Pause"); |
| paused = false; |
| } |
| } |
| |
| public void stop() { |
| timer.stop(); |
| playAction.setEnabled(true); |
| stopAction.setEnabled(false); |
| pauseAction.setEnabled(false); |
| playbackS.setValue(0); |
| } |
| |
| public void sendCameraData() { |
| if(sendCameraDataB.isSelected()) { |
| trackingP.startSendingCameraData(); |
| } else { |
| trackingP.stopSendingCameraData(); |
| } |
| } |
| |
| private ByteBuffer read(int position) throws IOException { |
| |
| inChannel.position(position); |
| |
| ByteBuffer numTrackedBuffer = ByteBuffer.allocate(1); |
| inChannel.read(numTrackedBuffer); |
| numTrackedBuffer.flip(); |
| |
| int numTracked = numTrackedBuffer.get() & 0xFF; |
| |
| ByteBuffer blobBuffer = ByteBuffer.allocate(1 + (numTracked * BLOB_LENGTH)); |
| |
| inChannel.position(inChannel.position() - 1); |
| inChannel.read(blobBuffer); |
| blobBuffer.flip(); |
| |
| return blobBuffer; |
| |
| } |
| |
| public void record() { |
| if(recordB.isSelected()) { |
| try { |
| trackingP.startRecording(); |
| recordB.setText("Stop Recording"); |
| } catch(Exception e) { |
| recordB.setSelected(false); |
| } |
| } else { |
| stopRecording(); |
| } |
| } |
| |
| private void stopRecording() { |
| try { |
| trackingP.stopRecording(); |
| recordB.setText("Start Recording"); |
| } catch(Exception e) { |
| recordB.setSelected(true); |
| } |
| } |
| |
| public void startTracking() { |
| trackingP.startTracking(); |
| } |
| |
| public void stopTracking() { |
| if(recordB.isSelected()) { |
| stopRecording(); |
| } |
| trackingP.stopTracking(); |
| } |
| |
| private final class PlaybackHandler implements ChangeListener { |
| public void stateChanged(ChangeEvent ce) { |
| try { |
| trackingP.setTrackingData(read(((Integer)indexMap.get(new Integer(playbackS.getValue()))).intValue())); |
| } catch(IOException ioe) { |
| ioe.printStackTrace(); |
| AVRSystem.LOG.severe(ioe.getMessage()); |
| } |
| } |
| } |
| |
| private final class UpdateSliderHandler implements ActionListener { |
| public void actionPerformed(ActionEvent ae) { |
| if(playbackS.getValue() == playbackS.getMaximum()) { |
| stop(); |
| } else { |
| try { |
| playbackS.setValue(playbackS.getValue() + 1); |
| } catch(Exception e) { |
| e.printStackTrace(); |
| } |
| } |
| } |
| } |
| |
| private final class ConnectionHandler implements ConnectionListener { |
| public void connected(ConnectionEvent ce) { |
| sendCameraDataB.setEnabled(true); |
| } |
| |
| public void disconnected(ConnectionEvent ce) { |
| trackingP.stopSendingCameraData(); |
| sendCameraDataB.setSelected(false); |
| sendCameraDataB.setEnabled(false); |
| } |
| |
| } |
| |
| } |