Restricted tower of hanoi implementation with GUI (#561)

Others/RestrictedTowerOfHanoi/Main/Hanoi.java

@@ -0,0 +1,239 @@
+import java.util.*;
+import javax.swing.*;
+import javax.swing.Timer;
+import java.awt.*;
+import java.awt.event.*;
+
+public class Hanoi extends JFrame{
+
+    public static int ONE_SECOND = 1000;
+
+    int number_of_disks=0;
+    int game_counter = 0;
+    int i=0;
+
+    /*  GUI COMPONENTS */
+    public JButton move_button = new JButton();
+    public JButton exit_button = new JButton();
+    public JButton replay_button = new JButton();
+    public JButton auto_button = new JButton();
+
+
+    /* BACKEND COMPONENTS */
+    public ArrayList<String> movements = new ArrayList<String>();
+    public StringBuilder stringBuilder = new StringBuilder();
+
+    public ArrayList<Integer> Stack1 = new ArrayList<Integer>();
+    public ArrayList<Integer> Stack2 = new ArrayList<Integer>();
+    public ArrayList<Integer> Stack3 = new ArrayList<Integer>();
+
+    public void updateStacks() {
+        if(game_counter!=movements.size()) {
+            String temp = movements.get(game_counter);
+            System.out.println(temp);
+            if(temp.charAt(1)=='A') {
+                if(temp.charAt(2)=='B') {
+                    int x = Stack1.get(Stack1.size()-1);
+                    Stack1.remove(Stack1.size()-1);
+                    Stack2.add(x);
+                }
+            }
+            if(temp.charAt(1)=='C') {
+                if(temp.charAt(2)=='B') {
+                    int x = Stack3.get(Stack3.size()-1);
+                    Stack3.remove(Stack3.size()-1);
+                    Stack2.add(x);
+                }
+            }
+
+            if(temp.charAt(1)=='B') {
+                if(temp.charAt(2)=='C') {
+                    int x = Stack2.get(Stack2.size()-1);
+                    Stack2.remove(Stack2.size()-1);
+                    Stack3.add(x);
+                }
+                else if(temp.charAt(2)=='A') {
+                    int x = Stack2.get(Stack2.size()-1);
+                    Stack2.remove(Stack2.size()-1);
+                    Stack1.add(x);
+                }
+            }
+            revalidate();
+            repaint();
+            game_counter++;
+        }
+    }
+
+    public void paint(Graphics canvas) {
+        super.paint(canvas);
+
+        //Drawing pedestels
+        for(int i=0;i<3;i++) {
+            canvas.drawRect(30+i*230,670,200,20);
+            canvas.setColor(new Color(76,174,227)); //Blue Accent
+            canvas.fillRect(30+i*230,670,200,20);
+
+            canvas.fillRect(130+i*230-2,670-170,4,170);
+            canvas.setColor(new Color(150,0,0));   //Arseny
+            canvas.fillRect(130+i*230-2,670-170,4,170);
+        }
+
+        //Disks in stack1
+        for(int i=1;i<=Stack1.size();i++) {
+            canvas.drawRect(130-Stack1.get(i-1)*10,670-i*12,Stack1.get(i-1)*20,10);
+            canvas.setColor(new Color(64,26,0)); //Brown Wolfers
+            canvas.fillRect(130-Stack1.get(i-1)*10,670-i*12,Stack1.get(i-1)*20,10);
+        } 
+
+        //Disks in stack2
+        for(int i=1;i<=Stack2.size();i++) {
+            canvas.drawRect(360-Stack2.get(i-1)*10,670-i*12,Stack2.get(i-1)*20,10);
+            canvas.setColor(new Color(64,26,0)); //Brown Wolfers
+            canvas.fillRect(360-Stack2.get(i-1)*10,670-i*12,Stack2.get(i-1)*20,10);
+        }
+
+        //Disks in stack3
+        for(int i=1;i<=Stack3.size();i++) {
+            canvas.drawRect(590-Stack3.get(i-1)*10,670-i*12,Stack3.get(i-1)*20,10);
+            canvas.setColor(new Color(64,26,0)); //Brown Wolfers
+            canvas.fillRect(590-Stack3.get(i-1)*10,670-i*12,Stack3.get(i-1)*20,10);
+        }
+    }
+
+    // Function to initialize the widget properties and the frame.
+    public void initialize() {
+
+        move_button.setIcon(new ImageIcon("../Resources/rsz_move.png"));
+        move_button.setBounds(130,0,50,50);
+
+        auto_button.setIcon(new ImageIcon("../Resources/rsz_loop.png"));
+        auto_button.setBounds(260,0,50,50);
+
+        replay_button.setIcon(new ImageIcon("../Resources/rsz_replay.jpg"));
+        replay_button.setBounds(390,0,50,50);
+
+        exit_button.setIcon(new ImageIcon("../Resources/rsz_exit.png"));
+        exit_button.setBounds(520,0,50,50);
+
+        add(move_button);
+        add(exit_button);
+        add(replay_button);
+        add(auto_button);
+
+        setLayout(null);
+        setSize(720,720);
+        setVisible(true);
+        setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
+    }
+    // Main cnstructor.
+    Hanoi() {
+        super("restricted tower of hanoi");
+        initialize();
+        
+        //MOVE BUTTON ACTION LISTENER
+        move_button.addActionListener(new ActionListener() {
+            @Override
+            public void actionPerformed(ActionEvent e) {
+                updateStacks();
+            }
+        });
+
+        //EXIT BUTTON ACTION LISTENER
+        exit_button.addActionListener(new ActionListener() {
+            @Override
+            public void actionPerformed(ActionEvent e) {
+                System.exit(0);
+            }
+        });
+
+        //REPLAY BUTTON ACTION LISTENER
+        replay_button.addActionListener(new ActionListener() {
+            @Override
+            public void actionPerformed(ActionEvent e) {
+                startGame();
+                repaint();
+            }
+        });
+
+        //AUTOMATIC PLAY BUTTON ACTION LISTENER
+        auto_button.addActionListener(new ActionListener() {
+            @Override
+            public void actionPerformed(ActionEvent e) {
+                timer.start();
+                if(game_counter == movements.size()) {
+                    timer.stop();
+                }
+            }
+        });
+    }
+
+    Timer timer = new Timer(ONE_SECOND,new ActionListener() {
+        public void actionPerformed(ActionEvent e) {
+            updateStacks();
+        }
+    });
+
+    public void startGame() {
+
+        System.out.println("New Game Started");
+        timer.stop();
+        
+        Stack1 = new ArrayList<Integer>();
+        Stack2 = new ArrayList<Integer>();
+        Stack3 = new ArrayList<Integer>();
+
+        movements = new ArrayList<String>();
+        game_counter = 0;
+
+        for(int i=0;i<number_of_disks;i++) {
+            Stack1.add(number_of_disks-i);
+        }
+
+        towerOfHanoi(number_of_disks,'A','C','B');
+    }
+
+    public static void main(String args[]) {
+        Hanoi tower = new Hanoi();
+        int number = Integer.parseInt(args[0]);
+        tower.number_of_disks = number;
+        tower.startGame();
+        /*for(int i=0;i<tower.movements.size();i++) {
+            System.out.println(tower.movements.get(i));
+            //System.out.println(tower.Stack1.get(i));
+        }*/
+    }
+
+    //Recursive function to formulate restricted tower of hanoi.
+    public void towerOfHanoi(int n, char from_rod, char to_rod, char aux_rod) {
+        if (n == 1) {
+            stringBuilder.setLength(0);
+            stringBuilder.append(Integer.toString(n));
+            stringBuilder.append(from_rod);
+            stringBuilder.append(aux_rod);
+            movements.add(stringBuilder.toString());
+            //System.out.println("Move disk 1 from rod " +  from_rod + " to rod " + aux_rod);
+            stringBuilder.setLength(0);
+            stringBuilder.append(Integer.toString(n));
+            stringBuilder.append(aux_rod);
+            stringBuilder.append(to_rod);
+            movements.add(stringBuilder.toString());
+            //System.out.println("Move disk 1 from rod " +  aux_rod + " to rod " + to_rod);
+            return;
+        }
+        towerOfHanoi(n-1, from_rod, to_rod, aux_rod);
+        stringBuilder.setLength(0);
+        stringBuilder.append(Integer.toString(n));
+        stringBuilder.append(from_rod);
+        stringBuilder.append(aux_rod);
+        movements.add(stringBuilder.toString());
+        //System.out.println("Move disk " + n + " from rod " +  from_rod + " to rod " + aux_rod);
+        towerOfHanoi(n-1, to_rod, from_rod, aux_rod);
+        stringBuilder.setLength(0);
+        stringBuilder.append(Integer.toString(n));
+        stringBuilder.append(aux_rod);
+        stringBuilder.append(to_rod);
+        movements.add(stringBuilder.toString());
+        //System.out.println("Move disk " + n + " from rod " +  aux_rod + " to rod " + to_rod);
+        towerOfHanoi(n-1, from_rod, to_rod, aux_rod);
+    }
+}