package com.knutejohnson.aviation.j3;

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

public class J3WB extends JPanel {
    static final String[] ROW_LABELS =
     {"Aircraft","Front Pax","Rear Pax","Fuel","Baggage","Other","Gross"};
    static final String[] COLUMN_LABELS =
     {"Weight","*","Arm","=","Moment"};
    static final String[] ARMS =
     {"","9.00","36.00","-18.00","49.00","",""};
    static final Color CUB_YELLOW = new Color(0xcf,0xbf,0);

    final JTextField[][] fields = new JTextField[ROW_LABELS.length][3];

    public J3WB() {
        super(new GridBagLayout());

        FocusAdapter transferAdapter = new FocusAdapter() {
            public void focusGained(FocusEvent fe) {
                ((Component)fe.getSource()).transferFocus();
            }
        };

        setBorder(BorderFactory.createLineBorder(CUB_YELLOW,4));

        GridBagConstraints c = new GridBagConstraints();
        c.insets = new Insets(2,2,2,2);

        c.gridx = 1;  c.gridy = 0;
        for (int i=0; i<COLUMN_LABELS.length; i++) {
            add(new JLabel(COLUMN_LABELS[i]),c);
            ++c.gridx;
        }

        c.anchor = GridBagConstraints.EAST;
        for (int i=0; i<ROW_LABELS.length; i++) {
            c.gridx = 0;  ++c.gridy;
            add(new JLabel(ROW_LABELS[i]),c);
            ++c.gridx;
            fields[i][0] = new JTextField(5);
            fields[i][0].setHorizontalAlignment(JTextField.RIGHT);
            add(fields[i][0],c);
            c.gridx += 2;
            fields[i][1] = new JTextField(ARMS[i],5);
            fields[i][1].setHorizontalAlignment(JTextField.RIGHT);
            add(fields[i][1],c);
            c.gridx += 2;
            fields[i][2] = new JTextField(5);
            fields[i][2].setHorizontalAlignment(JTextField.RIGHT);
            fields[i][2].setEditable(false);
            fields[i][2].addFocusListener(transferAdapter);
            add(fields[i][2],c);
            if (i == ROW_LABELS.length - 2) {
                c.gridx = 0;  ++c.gridy;  c.gridwidth = 6;
                c.anchor = GridBagConstraints.CENTER;
                add(new JLabel(
                 "---------------------------------"),c);
                c.gridwidth = 1;  c.anchor = GridBagConstraints.EAST;
            }
        }
        for (int i=0; i<3; i++) {
            fields[ROW_LABELS.length-1][i].setEditable(false);
            fields[ROW_LABELS.length-1][i].addFocusListener(transferAdapter);
        }

        c.gridx = 0;  ++c.gridy;  c.gridwidth = 6;
        c.anchor = GridBagConstraints.CENTER;
        JButton calc = new JButton("Calculate");
        calc.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent ae) {
                double totalWeight = 0.0;
                double totalMoment = 0.0;
                double weight,arm,moment,cg;
                for (int i=0; i<ROW_LABELS.length - 1; i++) {
                    try {
                         weight = Double.parseDouble(fields[i][0].getText());
                    } catch (NumberFormatException nfe) {
                        fields[i][0].setText("");
                        fields[i][2].setText("");
                        continue;
                    }
                    try {
                        arm = Double.parseDouble(fields[i][1].getText());
                        moment = weight * arm;
                        fields[i][2].setText(String.format("%.1f",moment));
                    } catch (NumberFormatException nfe) {
                        fields[i][1].setText("");
                        fields[i][2].setText("");
                        continue;
                    }
                    totalWeight += weight;
                    totalMoment += moment;
                }
                cg = totalMoment / totalWeight;
                fields[ROW_LABELS.length-1][0].setText(String.format("%.1f",
                 totalWeight));
                fields[ROW_LABELS.length-1][1].setText(String.format("%.2f",
                 cg));
                fields[ROW_LABELS.length-1][2].setText(String.format("%.1f",
                 totalMoment));

                if (totalWeight < 1220.0)
                    fields[ROW_LABELS.length-1][0].setForeground(Color.GREEN);
                else
                    fields[ROW_LABELS.length-1][0].setForeground(Color.RED);
                if (cg >= 10.6 && cg <= 22.7)
                    fields[ROW_LABELS.length-1][1].setForeground(Color.GREEN);
                else
                    fields[ROW_LABELS.length-1][1].setForeground(Color.RED);
            }
        });
        add(calc,c);
    }

    public static void main(String[] args) {
        EventQueue.invokeLater(new Runnable() {
            public void run() {
                JFrame f = new JFrame("J3 Weight and Balance");
                f.setLayout(new GridBagLayout());
                f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
                J3WB j3 = new J3WB();
                f.add(j3);
                f.pack();
                f.setVisible(true);
                System.out.println(j3.getWidth()+" "+j3.getHeight());
            }
        });
    }
}