// PreTetrisConnect4.java
// Created on Thu Jul 24 09:21:58 EDT 2008

import javax.swing.Timer;
import java.util.*;
import javax.swing.*;
import java.awt.*;
import java.awt.event.*;

public class PreTetrisConnect4 {
   public static void main(String[] args) {
      JFrame frame = new JFrame("PreTetris");
      MyComponent c = new MyComponent();
      frame.getContentPane().add(c);
      frame.setSize(420,360);
      frame.setDefaultCloseOperation(JFrame.DISPOSE_ON_CLOSE);
      frame.setVisible(true);
   }   
}

class MyComponent extends JComponent
                  implements MouseListener, KeyListener, ActionListener {
   // instance variables
   private int pageMargin = 5;
   private int rows = 6;
   private int cols = 7;
   private Color player1    = Color.red;
   private Color player2    = Color.black;
   private Color empty      = Color.white;
   private Color background = Color.blue;
   private Color[][] board = new Color[rows][cols];
   private boolean player1turn = true;
   
   // constructor
   public MyComponent() {
      initBoard();
      setFocusable(true);
      addKeyListener(this);
      addMouseListener(this);
      int delay = 100; //milliseconds
      Timer timer = new Timer(delay, null);
      timer.addActionListener(this);
      timer.start();
  }
  
  private void initBoard() {
     for (int row=0; row<rows; row++)
        for (int col=0; col<cols; col++)
           board[row][col] = empty;
  }

   /////////////////////////
   // Game play methods
   /////////////////////////

   private int getColumn(int x) {
      int width = getWidth();
      int innerWidth = width - 2*this.pageMargin;
      int cellWidth = innerWidth/this.cols;
      int col = (x - this.pageMargin) / cellWidth;
      if (col >= cols) return -1;      
      return col;
   }
   
   private void doMove(int col) {
      int row = lastBlankRow(col);
      if (row < 0) {
          beep();
          return;
      }
      Color playerColor = (player1turn ? player1 : player2);
      board[row][col] = playerColor;
      if (didWin(playerColor)) {
         System.out.println("You win!!!!");
         initBoard();
         player1turn = true;
      }
      else if (boardIsFull()) {
         System.out.println("Tie game!!!!");
         initBoard();
         player1turn = true;         
      }
      else
         player1turn = !player1turn;
   }
   
   private boolean boardIsFull() {
      for (int row=0; row<rows; row++)
         for (int col=0; col<cols; col++)
            if (board[row][col] == empty)
               return false;
      return true;
   }
   
   private boolean didWin(Color color, int row, int col, int drow, int dcol) {
      if ((drow == 0) && (dcol == 0)) return false;
      for (int counter=0; counter<4; counter++) {
         int r = row + counter*drow;
         int c = col + counter*dcol;
         if ((r < 0) || (r >= rows) || (c < 0) || (c >= cols)) return false;
         if (board[r][c] != color) return false;
      }
      return true;
   }
   
   private boolean didWin(Color c) {
      for (int row=0; row<rows; row++)
         for (int col=0; col<cols; col++)
            for (int drow=-1; drow<=1; drow++)
               for (int dcol=-1; dcol<=1; dcol++)
                  if (didWin(c,row,col,drow,dcol))
                     return true;
      return false;
   }
   
   private int lastBlankRow(int col) {
      for (int row=this.rows-1; row>=0; row--)
         if (this.board[row][col] == empty)
            return row;
      return -1;
   }

   /////////////////////////
   // ActionListener methods (used for timer)
   /////////////////////////
   
   public void doTimer() {
   }
   
   public void actionPerformed(ActionEvent e) {
      doTimer();
      repaint();
   }

   /////////////////////////
   // KeyListener methods
   /////////////////////////
   
   public void doUp() {
   }
   
   public void doDown() {
   }

   public void doLeft() {
   }

   public void doRight() {
   }
   
   public void doSpace() {
   }

   public void keyPressed(KeyEvent ke) {
      switch (ke.getKeyCode()) {
         case KeyEvent.VK_UP:    doUp(); break;
         case KeyEvent.VK_DOWN:  doDown(); break;
         case KeyEvent.VK_LEFT:  doLeft(); break;
         case KeyEvent.VK_RIGHT: doRight(); break;
         // default:  beep();
      }
      repaint();
   }
   
   public void keyReleased(KeyEvent ke) { }
   
   public void keyTyped(KeyEvent ke) {
      switch (ke.getKeyChar()) {
         case ' ': doSpace(); break;
         default: beep();
      }
      repaint();
   }
   

   /////////////////////////
   // MouseListener methods
   /////////////////////////
   
   public void mouseClicked(MouseEvent e) {
   }
   
   public void mouseEntered(MouseEvent e) {
   }
   
   public void mouseExited(MouseEvent e) {
   }
   
   public void mousePressed(MouseEvent e) {
      int x = e.getX();
      int y = e.getY();  
      int col = getColumn(x);
      if (col >= 0)
         doMove(col);
      repaint();
   }
   
   public void mouseReleased(MouseEvent e) {
   }
      
   
   private void beep() {
      Toolkit.getDefaultToolkit().beep();
   }
   
   private void paintCell(Graphics page, int row, int col) {
      int width = getWidth();
      int height = getHeight();
      int innerWidth = width - 2*this.pageMargin;
      int innerHeight = height - 2*this.pageMargin;
      int cellWidth = innerWidth/this.cols;
      int cellHeight = innerHeight/this.rows;
      int cellLeft = this.pageMargin + cellWidth*col;
      int cellTop  = this.pageMargin + cellHeight*row;
      int cellMargin = 5;
      page.setColor(this.board[row][col]);
      page.fillOval(cellLeft+cellMargin, cellTop+cellMargin,
                    cellWidth-2*cellMargin, cellHeight-2*cellMargin);
   }

   private void paintBoard(Graphics page) {
      for (int row=0; row<rows; row++)
         for (int col=0; col<cols; col++)
            paintCell(page, row, col);
   }
   
   public void paintComponent(Graphics page) {
      // first paint background
      int width = getWidth();
      int height = getHeight();
      page.setColor(this.background);
      page.fillRect(pageMargin,pageMargin,width-2*pageMargin,height-2*pageMargin);
      // then the board
      paintBoard(page);
   }
}