import java.awt.*;     // Needed for window and graphics (explained in COMP1406)
import javax.swing.*;  // Needed for window and graphics (explained in COMP1406)
import java.awt.event.*;

// This application simulates robots attracted to beacons in a window
public class RobotSimulation2 {
    // This variable stores the panel that displays the robots and beacons
    // (This is a topic discussed in COMP1406 course)
    public static RobotEnvironmentPanel2   envPanel;
    
    public static Robot[]                  robots;      // an array to hold the robots
    public static Beacon[]                 beacons;     // an array to hold the beacons
    public static Beacon                   grabbed;     // The beacon that the user just grabbed
    
    public static void startSimulation() {
        while(true) {
            for (int i=0; i<robots.length; i++) 
                move(robots[i]);
            envPanel.repaint();
            try{ Thread.sleep(10); } catch(Exception e){};
        }
    }
    
    // Code for moving a robot towards its beacon
    public static void move(Robot r) {
        float nextX, nextY, crossProduct;
        
        nextX = r.x + (int)(r.speed*Math.cos(r.direction)); 
        nextY = r.y + (int)(r.speed*Math.sin(r.direction));
        
        crossProduct = (nextX - r.x)*(r.beacon.y - r.y)-(nextY - r.y)*(r.beacon.x - r.x);
        r.x = (int)nextX; 
        r.y = (int)nextY;
        if (crossProduct < 0)   // try changing to:  if((crossProduct<0)&&(random(1)<0.6))
            r.direction -= (Robot.TURN_ANGLE + Math.random()*Robot.TURN_ANGLE/4 - (Robot.TURN_ANGLE/8))/180*Math.PI;
        else
            r.direction += (Robot.TURN_ANGLE + Math.random()*Robot.TURN_ANGLE/4 - (Robot.TURN_ANGLE/8))/180*Math.PI;
    }
    
    // Create a window, add beacons and robots, then start simulating
    public static void main(String args[]) {        
        // Make some beacons 
        beacons = new Beacon[5];
        for (int i=0; i<beacons.length; i++) {
            beacons[i] = new Beacon((int)(Math.random()*RobotEnvironmentPanel2.WIDTH), (int)(Math.random()*RobotEnvironmentPanel2.HEIGHT));
        }
        // Make some robots with shared beacons
        robots = new Robot[1000];
        for (int i=0; i<robots.length; i++) {
            robots[i] = new Robot(RobotEnvironmentPanel2.WIDTH/2, RobotEnvironmentPanel2.HEIGHT/2);
            robots[i].beacon = beacons[(int)(Math.random()*beacons.length)];
        }
        
        JFrame frame = new JFrame("Robot Simulation");
        frame.add(envPanel = new RobotEnvironmentPanel2(robots, beacons));      
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.pack(); // Makes size according to panel's preference
        frame.setVisible(true);
        
        startSimulation();
    }
}