package jmri.jmrix.cmri.serial.assignment;

import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.IOException;

import javax.swing.*;
import javax.swing.border.Border;
import javax.swing.table.*;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import jmri.jmrix.cmri.CMRISystemConnectionMemo;
import jmri.jmrix.cmri.serial.SerialNode;
import jmri.util.davidflanagan.HardcopyWriter;

/**
 * Frame for running CMRI assignment list.
 *
 * @author Dave Duchamp Copyright (C) 2006
 * @author Chuck Catania Copyright (C) 2016, 2017
 */
public class ListFrame extends jmri.util.JmriJFrame {

    // configured node information
    protected int numConfigNodes = 0;
    protected SerialNode[] configNodes = new SerialNode[128];
    protected int[] configNodeAddresses = new int[128];
    protected boolean inputSelected = false;  // true if displaying input assignments, false for output
    protected SerialNode selNode = null;
    protected String selNodeID = "x"; // text address of selected Node
    public int selNodeNum = 0;  // Address (ua) of selected Node
    public int numBits = 48;  // number of bits in assignment table
    public int numInputBits = 24;  // number of input bits for selected node
    public int numOutputBits = 48; // number of output bits for selected node

    // node select pane items
    JLabel nodeLabel = new JLabel(Bundle.getMessage("NodeBoxLabel") + " ");
    JComboBox<String> nodeSelBox = new JComboBox<>();
    ButtonGroup bitTypeGroup = new ButtonGroup();
    JRadioButton inputBits = new JRadioButton(Bundle.getMessage("ShowInputButton"), false);
    JRadioButton outputBits = new JRadioButton(Bundle.getMessage("ShowOutputButton"), true);
    JLabel nodeInfoText = new JLabel(Bundle.getMessage("NodeInfoCol"));

    // assignment pane items
    protected JPanel assignmentPanel = null;
    protected Border inputBorder = BorderFactory.createEtchedBorder();
    protected Border inputBorderTitled = BorderFactory.createTitledBorder(inputBorder,
            Bundle.getMessage("AssignmentPanelInputName"));
    protected Border outputBorder = BorderFactory.createEtchedBorder();
    protected Border outputBorderTitled = BorderFactory.createTitledBorder(outputBorder,
            Bundle.getMessage("AssignmentPanelOutputName"));
    protected JTable assignmentTable = null;
    protected TableModel assignmentListModel = null;

    // button pane items
    JButton printButton = new JButton(Bundle.getMessage("ButtonPrint"));

    ListFrame curFrame;

    private CMRISystemConnectionMemo _memo = null;

    public ListFrame(CMRISystemConnectionMemo memo) {
        super();
        curFrame = this;
        _memo = memo;
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public void initComponents() {

        // set the frame's initial state
        setTitle(Bundle.getMessage("WindowTitle") + Bundle.getMessage("WindowConnectionMemo") + _memo.getUserName()); // NOI18N
        setSize(500, 300);
        Container contentPane = getContentPane();
        contentPane.setLayout(new BoxLayout(contentPane, BoxLayout.Y_AXIS));

        // Set up the node selection panel
        initializeNodes();
        nodeSelBox.setEditable(false);
        if (numConfigNodes > 0) {
            nodeSelBox.addActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent event) {
                    displayNodeInfo((String) nodeSelBox.getSelectedItem());
                }
            });
            inputBits.addActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent event) {
                    if (inputSelected == false) {
                        inputSelected = true;
                        displayNodeInfo(selNodeID);
                    }
                }
            });
            outputBits.addActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent event) {
                    if (inputSelected == true) {
                        inputSelected = false;
                        displayNodeInfo(selNodeID);
                    }
                }
            });
        } else {
            nodeInfoText.setText(Bundle.getMessage("NoNodesError"));
        }
        nodeSelBox.setToolTipText(Bundle.getMessage("NodeBoxTip"));
        inputBits.setToolTipText(Bundle.getMessage("ShowInputTip"));
        outputBits.setToolTipText(Bundle.getMessage("ShowOutputTip"));

        JPanel panel1 = new JPanel();
        panel1.setLayout(new BoxLayout(panel1, BoxLayout.Y_AXIS));
        JPanel panel11 = new JPanel();
        panel11.add(nodeLabel);
        panel11.add(nodeSelBox);
        panel11.add(new JLabel(Bundle.getMessage("Show")));
        bitTypeGroup.add(outputBits);
        bitTypeGroup.add(inputBits);
        panel11.add(inputBits);
        panel11.add(outputBits);
        JPanel panel12 = new JPanel();
        panel12.add(nodeInfoText);
        panel1.add(panel11);
        panel1.add(panel12);
        Border panel1Border = BorderFactory.createEtchedBorder();
        Border panel1Titled = BorderFactory.createTitledBorder(panel1Border," ");
        panel1.setBorder(panel1Titled);
        contentPane.add(panel1);

        if (numConfigNodes > 0) {
            // Set up the assignment panel
            assignmentPanel = new JPanel();
            assignmentPanel.setLayout(new BoxLayout(assignmentPanel, BoxLayout.Y_AXIS));
            assignmentListModel = new AssignmentTableModel();
            assignmentTable = new JTable(assignmentListModel);
            assignmentTable.setRowSelectionAllowed(false);
            assignmentTable.setPreferredScrollableViewportSize(new java.awt.Dimension(300, 350));
            TableColumnModel assignmentColumnModel = assignmentTable.getColumnModel();
            TableColumn bitColumn = assignmentColumnModel.getColumn(AssignmentTableModel.BIT_COLUMN);
            bitColumn.setMinWidth(20);
            bitColumn.setMaxWidth(40);
            bitColumn.setResizable(true);
            TableColumn addressColumn = assignmentColumnModel.getColumn(AssignmentTableModel.ADDRESS_COLUMN);
            addressColumn.setMinWidth(40);
            addressColumn.setMaxWidth(85);
            addressColumn.setResizable(true);
            TableColumn sysColumn = assignmentColumnModel.getColumn(AssignmentTableModel.SYSNAME_COLUMN);
            sysColumn.setMinWidth(75);
            sysColumn.setMaxWidth(100);
            sysColumn.setResizable(true);
            TableColumn userColumn = assignmentColumnModel.getColumn(AssignmentTableModel.USERNAME_COLUMN);
            userColumn.setMinWidth(90);
            userColumn.setMaxWidth(450);
            userColumn.setResizable(true);
            TableColumn commentColumn = assignmentColumnModel.getColumn(AssignmentTableModel.COMMENT_COLUMN);
            commentColumn.setMinWidth(90);
            commentColumn.setMaxWidth(250);
            commentColumn.setResizable(true);
            JScrollPane assignmentScrollPane = new JScrollPane(assignmentTable);
            assignmentPanel.add(assignmentScrollPane, BorderLayout.CENTER);
            if (inputSelected) {
                assignmentPanel.setBorder(inputBorderTitled);
            } else {
                assignmentPanel.setBorder(outputBorderTitled);
            }
            contentPane.add(assignmentPanel);
        }

        // Set up Print button
        JPanel panel3 = new JPanel();
        panel3.setLayout(new FlowLayout());
        printButton.setVisible(true);
        printButton.setToolTipText(Bundle.getMessage("PrintButtonTip"));
        if (numConfigNodes > 0) {
            printButton.addActionListener(new java.awt.event.ActionListener() {
                @Override
                public void actionPerformed(java.awt.event.ActionEvent e) {
                    printButtonActionPerformed(e);
                }
            });
        }
        panel3.add(printButton);
        contentPane.add(panel3);

        if (numConfigNodes > 0) {
            // initialize for the first time
            displayNodeInfo((String) nodeSelBox.getSelectedItem());
        }

        addHelpMenu("package.jmri.jmrix.cmri.serial.assignment.ListFrame", true);

        // pack for display
        pack();
    }

    /**
     * Initialize configured nodes and set up the node select combo box.
     */
    public void initializeNodes() {
        String str = "";
        // clear the arrays
        for (int i = 0; i < 128; i++) {
            configNodeAddresses[i] = -1;
            configNodes[i] = null;
        }
        // get all configured nodes
        SerialNode node = (SerialNode) _memo.getTrafficController().getNode(0);
        int index = 1;
        while (node != null) {
            configNodes[numConfigNodes] = node;
            configNodeAddresses[numConfigNodes] = node.getNodeAddress();
            str = Integer.toString(configNodeAddresses[numConfigNodes]);
            nodeSelBox.addItem(str);
            if (index == 1) {
                selNode = node;
                selNodeNum = configNodeAddresses[numConfigNodes];
                selNodeID = "y";  // to force first time initialization
            }
            numConfigNodes++;
            // go to next node
            node = (SerialNode) _memo.getTrafficController().getNode(index);
            index++;
        }
    }

    /**
     * Method to handle selection of a Node for info display.
     * @param nodeID node ID string.
     */
    public void displayNodeInfo(String nodeID) {
        if (!nodeID.equals(selNodeID)) {
            // The selected node is changing - initialize it
            int nAdd = Integer.parseInt(nodeID);
            SerialNode s = null;
            for (int k = 0; k < numConfigNodes; k++) {
                if (nAdd == configNodeAddresses[k]) {
                    s = configNodes[k];
                }
            }
            if (s == null) {
                // serious trouble, log error and ignore
                log.error("Cannot find Node {} in list of configured Nodes.", nodeID);
                return;
            }
            // have node, initialize for new node
            selNodeID = nodeID;
            selNode = s;
            selNodeNum = nAdd;
            // prepare the information line
            int type = selNode.getNodeType();
            if (type == SerialNode.SMINI) {
                nodeInfoText.setText("SMINI - 24 " + Bundle.getMessage("InputBitsAnd") + " 48 "
                        + Bundle.getMessage("OutputBits"));
                numInputBits = 24;
                numOutputBits = 48;
            } else if (type == SerialNode.USIC_SUSIC) {
                int bitsPerCard = selNode.getNumBitsPerCard();
                int numInputCards = selNode.numInputCards();
                int numOutputCards = selNode.numOutputCards();
                numInputBits = bitsPerCard * numInputCards;
                numOutputBits = bitsPerCard * numOutputCards;
                nodeInfoText.setText("USIC_SUSIC - " + bitsPerCard + Bundle.getMessage("BitsPerCard")
                        + ", " + numInputBits + " " + Bundle.getMessage("InputBitsAnd") + " "
                        + numOutputBits + " " + Bundle.getMessage("OutputBits"));
            } else if (type == SerialNode.CPNODE) {
                int bitsPerCard = selNode.getNumBitsPerCard();
                int numInputCards = selNode.numInputCards();
                int numOutputCards = selNode.numOutputCards();
                numInputBits = bitsPerCard * numInputCards;
                numOutputBits = bitsPerCard * numOutputCards;
                nodeInfoText.setText("CPNODE - " + bitsPerCard + " " + Bundle.getMessage("BitsPerCard")
                        + ", " + numInputBits + " " + Bundle.getMessage("InputBitsAnd") + " "
                        + numOutputBits + " " + Bundle.getMessage("OutputBits"));
            } else if (type == SerialNode.CPMEGA) {
                int bitsPerCard = selNode.getNumBitsPerCard();
                int numInputCards = selNode.numInputCards();
                int numOutputCards = selNode.numOutputCards();
                numInputBits = bitsPerCard * numInputCards;
                numOutputBits = bitsPerCard * numOutputCards;
                nodeInfoText.setText("CPMEGA - " + bitsPerCard + " " + Bundle.getMessage("BitsPerCard")
                        + ", " + numInputBits + " " + Bundle.getMessage("InputBitsAnd") + " "
                        + numOutputBits + " " + Bundle.getMessage("OutputBits"));
            }

// here insert code for new types of C/MRI nodes
        }
        // initialize for input or output assignments
        if (inputSelected) {
            numBits = numInputBits;
            assignmentPanel.setBorder(inputBorderTitled);
        } else {
            numBits = numOutputBits;
            assignmentPanel.setBorder(outputBorderTitled);
        }
        ((AssignmentTableModel) assignmentListModel).fireTableDataChanged();
    }

    /**
     * Handle print button in List Frame.
     * @param e unused.
     */
    public void printButtonActionPerformed(java.awt.event.ActionEvent e) {
        int[] colWidth = new int[AssignmentTableModel.MAX_COLS];
        // initialize column widths
        TableColumnModel assignmentColumnModel = assignmentTable.getColumnModel();
        colWidth[0] = assignmentColumnModel.getColumn(AssignmentTableModel.BIT_COLUMN).getWidth();
        colWidth[1] = assignmentColumnModel.getColumn(AssignmentTableModel.ADDRESS_COLUMN).getWidth();
        colWidth[2] = assignmentColumnModel.getColumn(AssignmentTableModel.SYSNAME_COLUMN).getWidth();
        colWidth[3] = assignmentColumnModel.getColumn(AssignmentTableModel.USERNAME_COLUMN).getWidth();
        colWidth[4] = assignmentColumnModel.getColumn(AssignmentTableModel.COMMENT_COLUMN).getWidth();

        // set up a page title
        String head;
        if (inputSelected) {
            head = Bundle.getMessage("Connection") +" "+ _memo.getUserName() + "  "+ Bundle.getMessage("AssignmentPanelInputName") + " "
                    + Bundle.getMessage("NodeBoxLabel") + " " + selNodeID + "  ";
        } else {
            head = Bundle.getMessage("Connection") +" "+ _memo.getUserName() + "  " + Bundle.getMessage("AssignmentPanelOutputName") + " "
                    + Bundle.getMessage("NodeBoxLabel") + " " + selNodeID + "  ";
        }
        // initialize a printer writer
        HardcopyWriter writer = null;
        try {
            writer = new HardcopyWriter(curFrame, head, 10, .8, .5, .5, .5, false);
        } catch (HardcopyWriter.PrintCanceledException ex) {
            //log.debug("Print cancelled");
            return;
        }
        writer.increaseLineSpacing(20);
        // print the assignments
        ((AssignmentTableModel) assignmentListModel).printTable(writer, colWidth);
    }

    /**
     * Set up table for displaying bit assignments
     */
    public class AssignmentTableModel extends AbstractTableModel {

        private String free = Bundle.getMessage("AssignmentFree");
        private int curRow = -1;
        private String curRowSysName = "";

        @Override
        public String getColumnName(int c) {
            return assignmentTableColumnNames[c];
        }

        @Override
        public Class<?> getColumnClass(int c) {
            return String.class;
        }

        @Override
        public boolean isCellEditable(int r, int c) {
            return false;
        }

        @Override
        public int getColumnCount() {
            return MAX_COLS;
        }

        @Override
        public int getRowCount() {
            return numBits;
        }

        @Override
        public Object getValueAt(int r, int c) {
            if (c == BIT_COLUMN) {
                return Integer.toString(r + 1);
            } else if (c == ADDRESS_COLUMN) {
                return Integer.toString((selNodeNum * 1000) + r + 1);
            } else if (c == SYSNAME_COLUMN) {
                String sName = null;
                if (curRow != r) {
                    if (inputSelected) {
                        sName = _memo.isInputBitFree(selNodeNum, (r + 1));
                    } else {
                        sName = _memo.isOutputBitFree(selNodeNum, (r + 1));
                    }
                    curRow = r;
                    curRowSysName = sName;
                } else {
                    sName = curRowSysName;
                }
                if (sName == null) {
                    return (free);
                } else {
                    return sName;
                }

            } else if (c == USERNAME_COLUMN) {
                String sName = null;
                if (curRow != r) {
                    if (inputSelected) {
                        sName = _memo.isInputBitFree(selNodeNum, (r + 1));
                    } else {
                        sName = _memo.isOutputBitFree(selNodeNum, (r + 1));
                    }
                    curRow = r;
                    curRowSysName = sName;
                } else {
                    sName = curRowSysName;
                }
                if (sName == null) {
                    return ("");
                } else {
                    return (_memo.getUserNameFromSystemName(sName));
                }


            } else if (c == COMMENT_COLUMN) {
                String sName = null;
                if (curRow != r) {
                    if (inputSelected) {
                        sName = _memo.isInputBitFree(selNodeNum, (r + 1));
                    } else {
                        sName = _memo.isOutputBitFree(selNodeNum, (r + 1));
                    }
                    curRow = r;
                    curRowSysName = sName;
                } else {
                    sName = curRowSysName;
                }
                if (sName == null) {
                    return ("");
                }

                if (inputSelected) {
                    jmri.Sensor s = null;
                    s = jmri.InstanceManager.sensorManagerInstance().getBySystemName(sName);
                    if (s != null) {
                        return s.getComment();
                    }
                } else {
                    jmri.Turnout t = null;
                    t = jmri.InstanceManager.turnoutManagerInstance().getBySystemName(sName);
                    if (t != null) {
                        return t.getComment();
                    }
                }

            }

            return "";
        }

        @Override
        public void setValueAt(Object type, int r, int c) {
            // nothing is stored here
        }

        public static final int BIT_COLUMN = 0;
        public static final int ADDRESS_COLUMN = 1;
        public static final int SYSNAME_COLUMN = 2;
        public static final int USERNAME_COLUMN = 3;
        public static final int COMMENT_COLUMN = 4;
        public static final int MAX_COLS = COMMENT_COLUMN + 1;

        /**
         * Print or print preview the assignment table. Printed in
         * proportionately sized columns across the page with headings and
         * vertical lines between each column. Data is word wrapped within a
         * column. Can only handle 4 columns of data as strings. Adapted from
         * routines in BeanTableDataModel.java by Bob Jacobsen and Dennis Miller
         * @param w hard copy writer instance.
         * @param colWidth column width array.
         */
        public void printTable(HardcopyWriter w, int colWidth[]) {
            // determine the column sizes - proportionately sized, with space between for lines
            int[] columnSize = new int[MAX_COLS];
            int charPerLine = w.getCharactersPerLine();
            int tableLineWidth = 0;  // table line width in characters
            int totalColWidth = 0;
            for (int j = 0; j < MAX_COLS; j++) {
                totalColWidth += colWidth[j];
            }
            float ratio = ((float) charPerLine) / ((float) totalColWidth);
            for (int j = 0; j < MAX_COLS; j++) {
                columnSize[j] = ((int) (colWidth[j] * ratio)) - 1;
                tableLineWidth += (columnSize[j] + 1);
            }

            // Draw horizontal dividing line
            w.write(w.getCurrentLineNumber(), 0, w.getCurrentLineNumber(),
                    tableLineWidth);

            // print the column header labels
            String[] columnStrings = new String[AssignmentTableModel.MAX_COLS];
            // Put each column header in the array
            for (int i = 0; i < MAX_COLS; i++) {
                columnStrings[i] = this.getColumnName(i);
            }
            w.setFontStyle(Font.BOLD);
            printColumns(w, columnStrings, columnSize);
            w.setFontStyle(Font.PLAIN);
            // draw horizontal line
            w.write(w.getCurrentLineNumber(), 0, w.getCurrentLineNumber(),
                    tableLineWidth);

            // now print each row of data
            String[] spaces = new String[MAX_COLS];
            // create base strings the width of each of the columns
            for (int k = 0; k < MAX_COLS; k++) {
                spaces[k] = "";
                for (int i = 0; i < columnSize[k]; i++) {
                    spaces[k] = spaces[k] + " ";
                }
            }
            for (int i = 0; i < this.getRowCount(); i++) {
                for (int j = 0; j < MAX_COLS; j++) {
                    //check for special, null contents
                    if (this.getValueAt(i, j) == null) {
                        columnStrings[j] = spaces[j];
                    } else {
                        columnStrings[j] = (String) this.getValueAt(i, j);
                    }
                }
                printColumns(w, columnStrings, columnSize);
                // draw horizontal line
                w.write(w.getCurrentLineNumber(), 0, w.getCurrentLineNumber(),
                        tableLineWidth);
            }
            w.close();
        }

        protected void printColumns(HardcopyWriter w, String columnStrings[], int columnSize[]) {
            String columnString = "";
            StringBuilder lineString = new StringBuilder();
            StringBuilder[] spaces = new StringBuilder[MAX_COLS];
            // create base strings the width of each of the columns
            for (int k = 0; k < MAX_COLS; k++) {
                spaces[k] = new StringBuilder();
                for (int i = 0; i < columnSize[k]; i++) {
                    spaces[k].append(" ");
                }
            }
            // loop through each column
            boolean complete = false;
            while (!complete) {
                complete = true;
                for (int i = 0; i < MAX_COLS; i++) {
                    // if the column string is too wide cut it at word boundary (valid delimiters are space, - and _)
                    // use the initial part of the text,pad it with spaces and place the remainder back in the array
                    // for further processing on next line
                    // if column string isn't too wide, pad it to column width with spaces if needed
                    if (columnStrings[i].length() > columnSize[i]) {
                        // this column string will not fit on one line
                        boolean noWord = true;
                        for (int k = columnSize[i]; k >= 1; k--) {
                            if (columnStrings[i].substring(k - 1, k).equals(" ")
                                    || columnStrings[i].substring(k - 1, k).equals("-")
                                    || columnStrings[i].substring(k - 1, k).equals("_")) {
                                columnString = columnStrings[i].substring(0, k)
                                        + spaces[i].substring(columnStrings[i].substring(0, k).length());
                                columnStrings[i] = columnStrings[i].substring(k);
                                noWord = false;
                                complete = false;
                                break;
                            }
                        }
                        if (noWord) {
                            columnString = columnStrings[i].substring(0, columnSize[i]);
                            columnStrings[i] = columnStrings[i].substring(columnSize[i]);
                            complete = false;
                        }
                    } else {
                        // this column string will fit on one line
                        columnString = columnStrings[i] + spaces[i].substring(columnStrings[i].length());
                        columnStrings[i] = "";
                    }
                    lineString.append(columnString).append(" ");
                }
                try {
                    w.write(lineString.toString());
                    //write vertical dividing lines
                    int iLine = w.getCurrentLineNumber();
                    for (int i = 0, k = 0; i < w.getCharactersPerLine(); k++) {
                        w.write(iLine, i, iLine + 1, i);
                        if (k < MAX_COLS) {
                            i = i + columnSize[k] + 1;
                        } else {
                            i = w.getCharactersPerLine();
                        }
                    }
                    w.write("\n");
                    lineString = new StringBuilder();
                } catch (IOException e) {
                    log.warn("error during printing", e);
                }
            }
        }
    }
    private String[] assignmentTableColumnNames = {
        Bundle.getMessage("HeadingBit"),
        Bundle.getMessage("HeadingAddress"),
        Bundle.getMessage("HeadingSystemName"),
        Bundle.getMessage("HeadingUserName"),
        Bundle.getMessage("HeadingComment")
    };

    private final static Logger log = LoggerFactory.getLogger(ListFrame.class);

}