Flood-It-Bot/main/RobotFrame.java
2012-01-11 16:38:00 +02:00

363 lines
18 KiB
Java

package main;
import java.awt.AWTException;
import java.awt.Point;
import java.awt.image.BufferedImage;
import javax.swing.*;
/**
* Îêíî ïðèëîæåíèÿ.
* @author aNNiMON
*/
public class RobotFrame extends JFrame {
/* Ñòàòóñ ðàáîòû ïðèëîæåíèÿ */
private boolean isRunning;
private Thread robotAction;
/** Creates new form RobotFrame */
public RobotFrame() {
initComponents();
isRunning = false;
setAlwaysOnTop(true);
boardSizeSpinner.setModel(new SpinnerNumberModel(14, 5, 40, 1));
cellSizeSpinner.setModel(new SpinnerNumberModel(24, 5, 100, 1));
}
/** This method is called from within the constructor to
* initialize the form.
* WARNING: Do NOT modify this code. The content of this method is
* always regenerated by the Form Editor.
*/
@SuppressWarnings("unchecked")
// <editor-fold defaultstate="collapsed" desc="Generated Code">//GEN-BEGIN:initComponents
private void initComponents() {
startStop = new javax.swing.JButton();
jLabel1 = new javax.swing.JLabel();
windowPanel = new javax.swing.JPanel();
windowXTextField = new javax.swing.JTextField();
jLabel2 = new javax.swing.JLabel();
windowYTextField = new javax.swing.JTextField();
jLabel3 = new javax.swing.JLabel();
jLabel5 = new javax.swing.JLabel();
jLabel6 = new javax.swing.JLabel();
checkButton = new javax.swing.JButton();
boardSizeSpinner = new javax.swing.JSpinner();
jLabel7 = new javax.swing.JLabel();
cellSizeSpinner = new javax.swing.JSpinner();
detectButton = new javax.swing.JButton();
setDefaultCloseOperation(javax.swing.WindowConstants.EXIT_ON_CLOSE);
setTitle("FloodItBot");
setResizable(false);
startStop.setText("Start");
startStop.setActionCommand("");
startStop.addActionListener(new java.awt.event.ActionListener() {
public void actionPerformed(java.awt.event.ActionEvent evt) {
startStopActionPerformed(evt);
}
});
jLabel1.setHorizontalAlignment(javax.swing.SwingConstants.CENTER);
jLabel1.setText("Bot for FloodIt");
jLabel1.setFocusable(false);
jLabel1.setHorizontalTextPosition(javax.swing.SwingConstants.CENTER);
windowPanel.setBorder(javax.swing.BorderFactory.createTitledBorder("Options"));
windowXTextField.setText("0");
jLabel2.setText("x:");
windowYTextField.setText("0");
jLabel3.setText("y:");
jLabel5.setText("Window:");
jLabel6.setText("Cell size:");
checkButton.setText("Check");
checkButton.setActionCommand("");
checkButton.addActionListener(new java.awt.event.ActionListener() {
public void actionPerformed(java.awt.event.ActionEvent evt) {
checkButtonActionPerformed(evt);
}
});
jLabel7.setFont(new java.awt.Font("Tahoma", 1, 11)); // NOI18N
jLabel7.setText("Field size:");
detectButton.setText("Detect");
detectButton.setActionCommand("");
detectButton.addActionListener(new java.awt.event.ActionListener() {
public void actionPerformed(java.awt.event.ActionEvent evt) {
detectButtonActionPerformed(evt);
}
});
javax.swing.GroupLayout windowPanelLayout = new javax.swing.GroupLayout(windowPanel);
windowPanel.setLayout(windowPanelLayout);
windowPanelLayout.setHorizontalGroup(
windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(windowPanelLayout.createSequentialGroup()
.addContainerGap()
.addGroup(windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addComponent(detectButton, javax.swing.GroupLayout.DEFAULT_SIZE, 140, Short.MAX_VALUE)
.addComponent(checkButton, javax.swing.GroupLayout.Alignment.TRAILING, javax.swing.GroupLayout.DEFAULT_SIZE, 140, Short.MAX_VALUE)
.addGroup(windowPanelLayout.createSequentialGroup()
.addGroup(windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addComponent(jLabel5)
.addGroup(windowPanelLayout.createSequentialGroup()
.addComponent(jLabel2)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(windowXTextField, javax.swing.GroupLayout.PREFERRED_SIZE, 35, javax.swing.GroupLayout.PREFERRED_SIZE)
.addGap(18, 18, 18)
.addComponent(jLabel3)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(windowYTextField, javax.swing.GroupLayout.PREFERRED_SIZE, 35, javax.swing.GroupLayout.PREFERRED_SIZE))
.addGroup(windowPanelLayout.createSequentialGroup()
.addComponent(jLabel7)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(boardSizeSpinner, javax.swing.GroupLayout.PREFERRED_SIZE, 45, javax.swing.GroupLayout.PREFERRED_SIZE))
.addGroup(windowPanelLayout.createSequentialGroup()
.addComponent(jLabel6)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(cellSizeSpinner, javax.swing.GroupLayout.PREFERRED_SIZE, 45, javax.swing.GroupLayout.PREFERRED_SIZE)))
.addGap(0, 0, Short.MAX_VALUE)))
.addContainerGap())
);
windowPanelLayout.setVerticalGroup(
windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(windowPanelLayout.createSequentialGroup()
.addComponent(jLabel5)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addGroup(windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(jLabel2)
.addComponent(windowXTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
.addComponent(jLabel3)
.addComponent(windowYTextField, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
.addGroup(windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(jLabel6)
.addComponent(cellSizeSpinner, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
.addGroup(windowPanelLayout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
.addComponent(jLabel7)
.addComponent(boardSizeSpinner, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE))
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.UNRELATED)
.addComponent(detectButton)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(checkButton))
);
javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
getContentPane().setLayout(layout);
layout.setHorizontalGroup(
layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addContainerGap()
.addComponent(windowPanel, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
.addGroup(layout.createSequentialGroup()
.addContainerGap()
.addComponent(startStop, javax.swing.GroupLayout.DEFAULT_SIZE, 172, Short.MAX_VALUE))
.addGroup(layout.createSequentialGroup()
.addGap(57, 57, 57)
.addComponent(jLabel1)
.addGap(0, 0, Short.MAX_VALUE)))
.addContainerGap())
);
layout.setVerticalGroup(
layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
.addGroup(layout.createSequentialGroup()
.addContainerGap()
.addComponent(jLabel1)
.addGap(18, 18, 18)
.addComponent(windowPanel, javax.swing.GroupLayout.PREFERRED_SIZE, javax.swing.GroupLayout.DEFAULT_SIZE, javax.swing.GroupLayout.PREFERRED_SIZE)
.addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
.addComponent(startStop)
.addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
);
pack();
}// </editor-fold>//GEN-END:initComponents
private void startStopActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_startStopActionPerformed
if (isRunning) {
startStop.setText("Start");
isRunning = false;
} else {
startStop.setText("Stop");
isRunning = true;
robotAction = new Thread(new Runnable() {
@Override
public void run() {
int windowX, windowY, boardSize, cellSize;
BufferedImage detectImage;
RobotUtils robot;
// Ïîëó÷àåì íàñòðîéêè
try {
robot = new RobotUtils();
detectImage = robot.getImage(-1, -1, -1, -1);
windowX = Integer.valueOf(windowXTextField.getText());
windowY = Integer.valueOf(windowYTextField.getText());
boardSize = (Integer) boardSizeSpinner.getValue();
cellSize = (Integer) cellSizeSpinner.getValue();
} catch (Exception ex) {
ex.printStackTrace();
return;
}
ImageUtils iu = new ImageUtils(detectImage, boardSize, cellSize, windowX, windowY);
// Îáðåçàåì êàðòèíêó äî âèäà èãðîâîãî ïîëÿ
detectImage = iu.getBoardImage();
// Ïîëó÷àåì öâåòà èç êàðòèíêè
int[][] table = new int[boardSize][boardSize];
int offset = cellSize / 2;
for (int i = 0; i < boardSize; i++) {
for (int j = 0; j < boardSize; j++) {
table[i][j] = detectImage.getRGB(j*cellSize + offset,
i*cellSize + offset);
}
}
BotFloodIt bfi = new BotFloodIt(table);
// Ïîëó÷àåì ðåçóëüòèðóþùóþ ïîñëåäîâàòåëüíîñòü öâåòîâ
byte[] result = bfi.getFillSequence();
int[] colors = bfi.getColors();
// Ïûòàåìñÿ ïîëó÷èòü êîîðäèíàòû êíîïîê äëÿ àâòîìàòè÷åñêîé èãðû
Point[] buttons = iu.getButtons(colors);
if (buttons == null) {
// Åñëè íå óäàëîñü íàéòè êíîïêè, òî ïðîñòî âûâîäèì ïîñëåäîâàòåëüíîñòü â âèäå êàðòèíêè
BufferedImage out = iu.sequenceToImage(result, colors);
showImageWindow("Result: "+result.length+" steps", out);
} else {
// Çàïóñêàåì àâòîèãðó
robot.autoClick(buttons, result);
}
isRunning = false;
startStop.setText("Start");
}
});
robotAction.start();
}
}//GEN-LAST:event_startStopActionPerformed
private void checkButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_checkButtonActionPerformed
if (isRunning) return; // íåëüçÿ ïðîâåðÿòü íàñòðîéêè âî âðåìÿ ðàáîòû
int windowX, windowY, boardSize, cellSize;
BufferedImage detectImage;
RobotUtils robot;
try {
robot = new RobotUtils();
detectImage = robot.getImage(-1, -1, -1, -1);
windowX = Integer.valueOf(windowXTextField.getText());
windowY = Integer.valueOf(windowYTextField.getText());
boardSize = (Integer) boardSizeSpinner.getValue();
cellSize = (Integer) cellSizeSpinner.getValue();
} catch (Exception ex) {
ex.printStackTrace();
return;
}
ImageUtils iu = new ImageUtils(detectImage, boardSize, cellSize, windowX, windowY);
showImageWindow("Checking", iu.getBoardImage());
}//GEN-LAST:event_checkButtonActionPerformed
private void detectButtonActionPerformed(java.awt.event.ActionEvent evt) {//GEN-FIRST:event_detectButtonActionPerformed
if (isRunning) return; // íåëüçÿ îïðåäåëÿòü íàñòðîéêè âî âðåìÿ ðàáîòû
RobotUtils robot;
try {
robot = new RobotUtils();
BufferedImage detectImage = robot.getImage(-1, -1, -1, -1);
int boardSize = (Integer) boardSizeSpinner.getValue();
ImageUtils iu = new ImageUtils(detectImage, boardSize);
Point pt = iu.getBoardParameters();
int cellSize = iu.getCellSize();
windowXTextField.setText(String.valueOf(pt.x));
windowYTextField.setText(String.valueOf(pt.y));
cellSizeSpinner.setValue(cellSize);
} catch (AWTException ex) {
ex.printStackTrace();
}
}//GEN-LAST:event_detectButtonActionPerformed
/**
* Ïîêàçàòü ìîäàëüíîå îêíî ñ èçîáðàæåíèåì
* @param title çàãîëîâîê îêíà
* @param image èçîáðàæåíèå
* @throws SecurityException
*/
private void showImageWindow(String title, BufferedImage image) throws SecurityException {
ImageIcon icon = new ImageIcon(image);
JLabel backlabel = new JLabel(icon);
getLayeredPane().add(backlabel, new Integer(Integer.MIN_VALUE));
backlabel.setBounds(0, 0, icon.getIconWidth(), icon.getIconHeight());
JDialog dialog = new JDialog(this);
dialog.setAlwaysOnTop(true);
dialog.setLocationByPlatform(true);
dialog.setTitle(title);
dialog.setResizable(true);
dialog.add(backlabel);
dialog.pack();
dialog.setVisible(true);
}
/**
* @param args the command line arguments
*/
public static void main(String args[]) {
/* Set the Nimbus look and feel */
//<editor-fold defaultstate="collapsed" desc=" Look and feel setting code (optional) ">
/* If Nimbus (introduced in Java SE 6) is not available, stay with the default look and feel.
* For details see http://download.oracle.com/javase/tutorial/uiswing/lookandfeel/plaf.html
*/
try {
for (javax.swing.UIManager.LookAndFeelInfo info : javax.swing.UIManager.getInstalledLookAndFeels()) {
if ("Nimbus".equals(info.getName())) {
javax.swing.UIManager.setLookAndFeel(info.getClassName());
break;
}
}
} catch (ClassNotFoundException ex) {
java.util.logging.Logger.getLogger(RobotFrame.class.getName()).log(java.util.logging.Level.SEVERE, null, ex);
} catch (InstantiationException ex) {
java.util.logging.Logger.getLogger(RobotFrame.class.getName()).log(java.util.logging.Level.SEVERE, null, ex);
} catch (IllegalAccessException ex) {
java.util.logging.Logger.getLogger(RobotFrame.class.getName()).log(java.util.logging.Level.SEVERE, null, ex);
} catch (javax.swing.UnsupportedLookAndFeelException ex) {
java.util.logging.Logger.getLogger(RobotFrame.class.getName()).log(java.util.logging.Level.SEVERE, null, ex);
}
//</editor-fold>
/* Create and display the form */
java.awt.EventQueue.invokeLater(new Runnable() {
@Override
public void run() {
new RobotFrame().setVisible(true);
}
});
}
// Variables declaration - do not modify//GEN-BEGIN:variables
private javax.swing.JSpinner boardSizeSpinner;
private javax.swing.JSpinner cellSizeSpinner;
private javax.swing.JButton checkButton;
private javax.swing.JButton detectButton;
private javax.swing.JLabel jLabel1;
private javax.swing.JLabel jLabel2;
private javax.swing.JLabel jLabel3;
private javax.swing.JLabel jLabel5;
private javax.swing.JLabel jLabel6;
private javax.swing.JLabel jLabel7;
private javax.swing.JButton startStop;
private javax.swing.JPanel windowPanel;
private javax.swing.JTextField windowXTextField;
private javax.swing.JTextField windowYTextField;
// End of variables declaration//GEN-END:variables
}