Use new AI for computer players

git-svn-id: svn://sunsvr01.isp.uni-luebeck.de/swproj13/trunk@448 72836036-5685-4462-b002-a69064685172

1 files changed, 197 insertions, 0 deletions

diff --git a/src/jrummikub/control/turn/AIControl.java b/src/jrummikub/control/turn/AIControl.java
new file mode 100644
index 0000000..8ab0aa5
--- /dev/null
+++ b/src/jrummikub/control/turn/AIControl.java
@@ -0,0 +1,197 @@
+package jrummikub.control.turn;
+
+import java.awt.event.ActionEvent;
+import java.awt.event.ActionListener;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+import java.util.TreeMap;
+
+import javax.swing.SwingUtilities;
+import javax.swing.Timer;
+
+import jrummikub.ai.TurnLogic;
+import jrummikub.control.AIUtil;
+import jrummikub.model.Position;
+import jrummikub.model.Stone;
+import jrummikub.model.StoneColor;
+import jrummikub.model.StoneSet;
+import jrummikub.util.Pair;
+
+/**
+ * Base class for AI players
+ * 
+ */
+public class AIControl extends AbstractTurnControl {
+	private TurnLogic logic;
+	private boolean turnDone = false;
+	boolean useBackgroundThread = true;
+	long startTime;
+
+	@Override
+	public void startTurn() {
+		timer.startTimer();
+		startTime = System.currentTimeMillis();
+		compute();
+	}
+
+	protected void timeOut() {
+		executeTurn();
+	}
+
+	@Override
+	protected void cleanUp() {
+		if (logic != null)
+			logic.abort();
+		turnDone = true;
+		super.cleanUp();
+	}
+
+	private void emitEndOfTurn() {
+		turnDone = true;
+		long turnLength = System.currentTimeMillis() - startTime;
+
+		if (useBackgroundThread) {
+			Timer timer = new Timer(Math.max(0, (int) (1000 + Math.random() * 2000 - turnLength)),
+					new ActionListener() {
+						@Override
+						public void actionPerformed(ActionEvent event) {
+							cleanUp();
+							endOfTurnEvent.emit();
+						}
+					});
+			timer.setRepeats(false);
+			timer.start();
+		} else {
+			cleanUp();
+			endOfTurnEvent.emit();
+		}
+
+	}
+
+	private void compute() {
+		switch (turnInfo.getTurnMode()) {
+		case MAY_REDEAL:
+			emitEndOfTurn();
+			break;
+		case INSPECT_ONLY:
+			emitEndOfTurn();
+			break;
+		case NORMAL_TURN:
+			turn();
+			break;
+		}
+	}
+
+	private void turn() {
+		List<Stone> tableStones = new ArrayList<Stone>();
+		List<Stone> handStones = new ArrayList<Stone>();
+
+		for (Pair<Stone, Position> entry : turnInfo.getHand()) {
+			handStones.add(entry.getFirst());
+		}
+
+		if (turnInfo.getLaidOut()) {
+			for (Pair<StoneSet, Position> entry : turnInfo.getTable()) {
+				for (Stone stone : entry.getFirst()) {
+					tableStones.add(stone);
+				}
+			}
+		}
+
+		logic = new TurnLogic(settings, tableStones, handStones);
+
+		if (!turnInfo.getLaidOut()) {
+			logic.needIntialMeldThreshold();
+		}
+		if (useBackgroundThread) {
+			
+			Timer timer = new Timer(10000, new ActionListener() {
+				@Override
+				public void actionPerformed(ActionEvent e) {
+					logic.autoAbort();
+				}
+			});
+			timer.setRepeats(false);
+			timer.start();
+			
+			Thread computeThread = new Thread(new Runnable() {
+				@Override
+				public void run() {
+					logic.optimize();
+					SwingUtilities.invokeLater(new Runnable() {
+						@Override
+						public void run() {
+							executeTurn();
+						}
+					});
+				}
+			});
+			computeThread.start();
+		} else {
+			logic.optimize();
+			executeTurn();
+		}
+
+	}
+
+	private void executeTurn() {
+		if (turnDone) {
+			return;
+		}
+
+		List<StoneSet> result = logic.getResult();
+
+		if (result != null) {
+
+			if (turnInfo.getLaidOut()) {
+				outerLoop: for (Iterator<Pair<StoneSet, Position>> it = turnInfo
+						.getTable().iterator(); it.hasNext();) {
+					Pair<StoneSet, Position> pair = it.next();
+					setSearch: for (Iterator<StoneSet> it2 = result.iterator(); it2
+							.hasNext();) {
+						StoneSet set = it2.next();
+						if (set.getSize() != pair.getFirst().getSize()) {
+							continue;
+						}
+						for (int i = 0; i < set.getSize(); i++) {
+							if (set.get(i) != pair.getFirst().get(i)) {
+								continue setSearch;
+							}
+						}
+						it2.remove();
+						continue outerLoop;
+					}
+					it.remove();
+				}
+			}
+
+			for (StoneSet set : result) {
+				turnInfo.getTable().drop(
+						set,
+						new Position(10 * (Math.random() * 2 - 1), 5 * (Math
+								.random() * 2 - 1)));
+				for (Stone stone : set) {
+					turnInfo.getHand().pickUp(stone);
+				}
+			}
+		}
+
+		emitEndOfTurn();
+	}
+
+	/**
+	 * Get the factory for the base AI control
+	 * 
+	 * @return the factory
+	 */
+	static public TurnControlFactory getFactory() {
+		return new TurnControlFactory() {
+			@Override
+			public ITurnControl create() {
+				return new AIControl();
+			}
+		};
+	}
+
+}