Colony

import java.awt.BorderLayout;

import java.awt.Color;

import java.awt.Component;

import java.awt.Dimension;

import java.awt.GridLayout;

import java.awt.LayoutManager;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.Enumeration;

import javax.swing.AbstractButton;

import javax.swing.BorderFactory;

import javax.swing.Box;

import javax.swing.BoxLayout;

import javax.swing.ButtonGroup;

import javax.swing.JButton;

import javax.swing.JFrame;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JRadioButton;

import javax.swing.JScrollPane;

import javax.swing.JTextArea;

import javax.swing.SwingUtilities;

import javax.swing.border.Border;

import org.jfree.chart.ChartFactory;

import org.jfree.chart.ChartPanel;

import org.jfree.chart.JFreeChart;

import org.jfree.chart.plot.PlotOrientation;

import org.jfree.data.xy.XYSeries;

import org.jfree.data.xy.XYSeriesCollection;

public class ServerGUI implements ActionListener{

        private static final long serialVersionUID = 1L;

        // whether or not to use the console

        private static final boolean USE_CONSOLE = true;

        private int currentTest = -1;

        private int testNum = -1;

        // the area that shows messages

        private JTextArea console;

        // the DiagnosticsServer that does all the processing

        private DiagnosticsServer server;

        // a list of buttons

        private JButton[] buttons = new JButton[DiagnosticsServer.NUM_TESTS];

        // the stop button

        JButton stop;

        ButtonGroup numSelectGroup;

        private JRadioButton[] radios;

        private JFrame numSelectWindow, frame;

        private JPanel graphPanel, top;

        private LayoutManager layout;

        // Constructor - sets up the GUI

        public ServerGUI(DiagnosticsServer server){

                frame = new JFrame("Diagnostic Station Control Panel");

                this.server = server;

                frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

                frame.setLayout(new BoxLayout(frame.getContentPane(), BoxLayout.Y_AXIS));

                top = new JPanel();

                layout = new BoxLayout(top, BoxLayout.X_AXIS);

                top.setLayout(layout);

                top.add(buttonPanel());

                top.add(Box.createGlue());

                graphPanel = graphPanel();

                top.add(graphPanel);

                frame.add(top);

                if(USE_CONSOLE)

                        frame.add(console());

                frame.pack();

                frame.setVisible(true);

        }

        // returns the Panel containing the graph

        private JPanel graphPanel(){

                if(currentTest == -1 || currentTest == DiagnosticsServer.ALL_TESTS)

                        return blankPanel();

                String[] labels = server.getLabels(currentTest, testNum);

                ArrayList[] data = server.getData(currentTest, testNum);

                int numSeries = labels.length - 3;

                XYSeries series[] = new XYSeries[numSeries];

                for(int i=0; i<numSeries; i++){

                        series[i] = new XYSeries(labels[i+3]);

                        ArrayList x = data[2*i];

                        ArrayList y = data[2*i+1];

                        for(int j=0; j<x.size(); j++){

                                series[i].add((Integer)x.get(j), (Integer)y.get(j));

                        }

                }

                XYSeriesCollection xyDataset = new XYSeriesCollection(series[0]);

                for(int i=1; i<numSeries; i++)

                        xyDataset.addSeries(series[i]);

                JFreeChart chart = ChartFactory.createXYLineChart(labels[0],

                                labels[1], labels[2], xyDataset, PlotOrientation.VERTICAL, true,

                                false, false);

                ChartPanel panel = new ChartPanel(chart);

                JPanel graphPanel = new JPanel();

                graphPanel.add(panel);

                return graphPanel;

        }

        // returns a blank panel

        private JPanel blankPanel(){

                JPanel blank = new JPanel();

                blank.setBorder(BorderFactory.createMatteBorder(20, 20, 20, 20, new Color(238, 238, 238)));

                blank.setBackground(Color.WHITE);

                blank.add(Box.createRigidArea(new Dimension(680, 420)));

                return blank;

        }

        // returns the console

        private JScrollPane console(){

                console = new JTextArea(10, 50);

                JScrollPane pane = new JScrollPane(console); 

                console.setEditable(false);

                console.setLineWrap(false);

                Border inner = BorderFactory.createLineBorder(Color.BLACK);

                Border outer = BorderFactory.createEmptyBorder(5, 15, 5, 15);

                Border border = BorderFactory.createCompoundBorder(outer, inner);

                pane.setBorder(border);

                return pane;

        }

        // adds the given String to the console

        public void addTextToConsole(final String s){

                SwingUtilities.invokeLater(new Runnable (){

                        public void run() {

                                console.append(s);

                        }

                });

        }

        // the panel with the control buttons

        private JPanel buttonPanel(){

                int maxButtonWidth = 200;

                int maxButtonHeight = 30;

                JPanel buttonPanel = new JPanel();

                buttonPanel.setBorder(BorderFactory.createEmptyBorder(10, 15, 5, 15));

                buttonPanel.setLayout(new BoxLayout(buttonPanel, BoxLayout.Y_AXIS));

                buttonPanel.setMinimumSize(new Dimension(maxButtonWidth, 25 + ((maxButtonHeight + 5) * DiagnosticsServer.NUM_TESTS)));

                buttonPanel.setAlignmentX(Component.LEFT_ALIGNMENT);

                buttonPanel.setAlignmentY(Component.BOTTOM_ALIGNMENT);

                buttons[DiagnosticsServer.ALL_TESTS] = new JButton("Run all tests");

                buttons[DiagnosticsServer.RANGEFINDER] = new JButton("Run rangefinder test");

                buttons[DiagnosticsServer.ENCODER] = new JButton("Run encoder test");

                buttons[DiagnosticsServer.MOTOR] = new JButton("Run motor test");

                buttons[DiagnosticsServer.BOM] = new JButton("Run bom test");

                buttons[DiagnosticsServer.BOM_EMITTER] = new JButton("Run bom emitter test");

                buttons[DiagnosticsServer.BOM_DETECTOR] = new JButton("Run bom detector test");

                for(int i=0; i<DiagnosticsServer.NUM_TESTS; i++){

                        buttons[i].setVerticalTextPosition(AbstractButton.CENTER);

                        buttons[i].setActionCommand("frame " + i);

                        buttons[i].addActionListener(this);

                        buttons[i].setAlignmentX(Component.LEFT_ALIGNMENT);

                        buttons[i].setAlignmentY(Component.TOP_ALIGNMENT);

                        buttons[i].setMaximumSize(new Dimension(maxButtonWidth, maxButtonHeight));

                        if(i != 0)

                                buttonPanel.add(Box.createRigidArea(new Dimension(0, 5)));

                        buttonPanel.add(buttons[i]);                        

                }

                stop = new JButton("Stop testing");

                stop.setVerticalTextPosition(AbstractButton.CENTER);

                stop.setActionCommand("stop");

                stop.addActionListener(this);

                stop.setAlignmentX(Component.LEFT_ALIGNMENT);

                stop.setAlignmentY(Component.TOP_ALIGNMENT);

                stop.setMaximumSize(new Dimension(maxButtonWidth, maxButtonHeight));

                //stop.setBackground(Color.RED);

                stop.setEnabled(false);

                buttonPanel.add(Box.createRigidArea(new Dimension(0, 5)));

                buttonPanel.add(stop);        

                buttonPanel.add(Box.createGlue());

                return buttonPanel;

        }

        // shows the dialog asking the user which specific parts to test

        private void getNumDialog(int testNum){

                String instr = "Choose which test to run";

                if(testNum == DiagnosticsServer.BOM){

                        instr = "Choose which bom to test";

                }

                else if(testNum == DiagnosticsServer.BOM_DETECTOR){

                        instr = "Choose which detector to test";

                }

                else if(testNum == DiagnosticsServer.BOM_EMITTER){

                        instr = "Choose which emitter to test";

                }

                else if(testNum == DiagnosticsServer.RANGEFINDER){

                        instr = "Choose which rangefinder to test";

                }

                else if(testNum == DiagnosticsServer.MOTOR){

                        instr = "Choose which motor to test";

                }

                else if(testNum == DiagnosticsServer.ENCODER){

                        instr = "Choose which encoder to test";

                }

        JLabel label = new JLabel(instr);

                radios = radios(testNum);

        JPanel grid = new JPanel(new GridLayout(0, 1));

        for (int i = 0; i < radios.length; i++) {

            grid.add(radios[i]);

        }

        JPanel box = new JPanel();

        box.setBorder(BorderFactory.createEmptyBorder(5, 15, 5, 15));

        box.setLayout(new BoxLayout(box, BoxLayout.PAGE_AXIS));

        label.setAlignmentX(Component.LEFT_ALIGNMENT);

        box.add(label);

        box.add(grid);

        grid.setAlignmentX(Component.LEFT_ALIGNMENT);

        JPanel buttons = new JPanel();

        buttons.setLayout(new BoxLayout(buttons, BoxLayout.Y_AXIS));

        box.add(buttons);

        JButton run = new JButton("Run test");

        run.addActionListener(this);

        run.setActionCommand("run");

        run.setMaximumSize(new Dimension(200, 30));

        buttons.add(run);

        buttons.add(Box.createRigidArea(new Dimension(0, 5)));

                JButton cancel = new JButton("Cancel");

                cancel.addActionListener(this);

                cancel.setActionCommand("cancel");

                cancel.setMaximumSize(new Dimension(200, 30));

                buttons.add(cancel);

        numSelectWindow = new JFrame("Select test");

        numSelectWindow.setLayout(new BoxLayout(numSelectWindow.getContentPane(), BoxLayout.Y_AXIS));

        numSelectWindow.add(box, BorderLayout.PAGE_START);

                numSelectWindow.pack();

                numSelectWindow.setVisible(true);

        }

        // returns the radio buttons for the getNumDialog

        private JRadioButton[] radios(int testNum){

                int numTests = 0;

                int startIndex = 0;

                String allText = "All tests";

                if(testNum == DiagnosticsServer.BOM || testNum == DiagnosticsServer.BOM_DETECTOR || testNum == DiagnosticsServer.BOM_EMITTER){

                        numTests = DiagnosticsServer.NUM_BOMS;

                        startIndex = DiagnosticsServer.BOM_OFFSET;

                        if(testNum == DiagnosticsServer.BOM)

                                allText = "All boms";

                        else if(testNum == DiagnosticsServer.BOM_DETECTOR)

                                allText = "All detectors";

                        else if(testNum == DiagnosticsServer.BOM_EMITTER)

                                allText = "All emitters";

                }

                else if(testNum == DiagnosticsServer.RANGEFINDER){

                        numTests = DiagnosticsServer.NUM_RANGEFINDERS;

                        startIndex = DiagnosticsServer.RANGEFINDER_OFFSET;

                        allText = "All rangefinders";

                }

                else if(testNum == DiagnosticsServer.MOTOR){

                        numTests = DiagnosticsServer.NUM_MOTORS;

                        startIndex = DiagnosticsServer.MOTOR_OFFSET;

                        allText = "Both motors";

                }

                else if(testNum == DiagnosticsServer.ENCODER){

                        numTests = DiagnosticsServer.NUM_ENCODERS;

                        startIndex = DiagnosticsServer.ENCODER_OFFSET;

                        allText = "Both encoders";

                }

                JRadioButton radios[] = new JRadioButton[numTests+1];

                numSelectGroup = new ButtonGroup();

                for(int i=0; i<numTests+1; i++){

                        if(i==0){

                                radios[i] = new JRadioButton(allText);

                                radios[i].setSelected(true);

                        }

                        else{

                                radios[i] = new JRadioButton((i+startIndex-1) + "");

                                radios[i].setSelected(false);

                        }

                        radios[i].addActionListener(this);

                        radios[i].setActionCommand(testNum + " " + (i-1));

                        numSelectGroup.add(radios[i]);

                }

                return radios;

        }

        // the Action handler

        public void actionPerformed(ActionEvent event) {

                String actionCommand = event.getActionCommand();

                try{

                        if(actionCommand.equals("stop")){

                                server.stopTest();

                                stop.setBackground(null);

                                stop.setEnabled(false);

                        }

                        else if(actionCommand.equals("cancel")){

                                numSelectWindow.dispose();

                                return;

                        }

                        else if(actionCommand.equals("run")){

                                Enumeration<AbstractButton> buttons = numSelectGroup.getElements();

                                while(buttons.hasMoreElements()){

                                        AbstractButton button = buttons.nextElement();

                                        if(button.isSelected()){

                                                String command = button.getActionCommand();

                                                int spaceIndex = command.indexOf(" ");

                                                Integer testNum = Integer.parseInt(command.substring(0, spaceIndex));

                                                Integer num = Integer.parseInt(command.substring(spaceIndex+1));

                                                numSelectWindow.dispose();

                                                currentTest = testNum;

                                                this.testNum = num;

                                                stop.setBackground(Color.RED);

                                                stop.setEnabled(true);

                                                frame.repaint();

                                                server.startTest(testNum, num);

                                                stop.setBackground(null);

                                                stop.setEnabled(false);

                                                top.remove(graphPanel);

                                                graphPanel = graphPanel();

                                                top.add(graphPanel);

                                                top.validate();

                                                break;

                                        }

                                }

                        }

                        else if(actionCommand.startsWith("frame ")){

                                Integer num = Integer.parseInt(actionCommand.substring(6));

                                getNumDialog(num);

                        }

                }

                catch(Exception e){

                        System.out.println(e.getMessage());

                }

        }

}