import java.awt.*;
import javax.swing.*;

// This code is responsible for displaying the Robots in their environment
public class RobotEnvironmentPanel2 extends JPanel { 
    public static final int WIDTH = 800;
    public static final int HEIGHT = 600;
    
    private static Robot[]  robots;      // an array to hold the robots
    private static Beacon[] beacons;     // an array to hold the beacons
    
    public RobotEnvironmentPanel2(Robot[] rArray, Beacon[] bArray) {
        robots = rArray;
        beacons = bArray;
        setPreferredSize(new Dimension(WIDTH, HEIGHT));
    }
    
    // Display the image
    public void paintComponent(Graphics g) {
        super.paintComponent(g);
        
        // Display the beacons first
        for (int i=0; i<beacons.length; i++) {
            g.setColor(beacons[i].col);
            g.fillOval(beacons[i].x-Beacon.RADIUS, 
                       beacons[i].y-Beacon.RADIUS, 
                       Beacon.RADIUS*2, 
                       Beacon.RADIUS*2);
            g.setColor(Color.BLACK);
            g.drawOval(beacons[i].x-Beacon.RADIUS, 
                       beacons[i].y-Beacon.RADIUS, 
                       Beacon.RADIUS*2, 
                       Beacon.RADIUS*2);
        }
        // Now display the robots
        for (int i=0; i<robots.length; i++) {
            g.setColor(robots[i].beacon.col);
            g.fillOval(robots[i].x-Robot.RADIUS, 
                       robots[i].y-Robot.RADIUS, 
                       Robot.RADIUS*2, 
                       Robot.RADIUS*2);
            g.setColor(Color.BLACK);
            g.drawOval(robots[i].x-Robot.RADIUS, 
                       robots[i].y-Robot.RADIUS, 
                       Robot.RADIUS*2, 
                       Robot.RADIUS*2);
        }
    }
}