algorithms-for-computing-line-estimators/LinearRegressionTool/src/main/java/de/wwwu/awolf/view/panels/EvaluationPanel.java

package de.wwwu.awolf.view.panels;

import de.wwwu.awolf.model.Line;
import de.wwwu.awolf.presenter.Presenter;
import de.wwwu.awolf.presenter.algorithms.Algorithm;
import de.wwwu.awolf.presenter.data.DataProvider;
import de.wwwu.awolf.view.MainFrame;
import de.wwwu.awolf.view.custom.ButtonGroupAtLeastTwo;

import javax.swing.*;
import javax.swing.border.TitledBorder;
import javax.swing.filechooser.FileNameExtensionFilter;
import javax.swing.filechooser.FileSystemView;
import javax.swing.table.DefaultTableModel;
import javax.swing.table.JTableHeader;
import java.awt.*;
import java.io.File;
import java.io.Serializable;
import java.util.List;
import java.util.*;

/**
 * Created by
 * Armin Wolf
 * on 02.08.17.
 */
public class EvaluationPanel extends JPanel {

    private final MainFrame view;
    private JTable table;
    private JTableHeader header;
    private JButton start;
    private JButton clearTable;
    private JButton exportData;
    private JRadioButton evalTypeOne; //1: Alg - N: Data
    private JRadioButton evalTypeTwo; //N: Alg - 1: Data
    private ButtonGroup radiobuttonGroup;
    private ButtonGroup checkboxGroup;
    private ButtonGroupAtLeastTwo checkboxAtLeastOne;

    private JCheckBox lms; //1: Alg - N: Data
    private JCheckBox rm; //N: Alg - 1: Data
    private JCheckBox ts; //1: Alg - N: Data

    private JPanel comp;
    private JPanel algorithmPanel;
    private JPanel leftSidePanel;
    private JPanel datasetCount;
    private JComboBox<Integer> datasetCountChoice;
    private JComboBox<String> datasetType;
    private JLabel datasetCountLabel;

    private DefaultTableModel model;
    private int currentRowOfTypes;
    private JPanel buttonPanel;
    private PlotPanel plotPanel;
    private Map<Algorithm.Type, Color> colorMap;
    private String[] selections = {"Approximationsgüte", "Least Median of Squares", "Repeated-Median", "Theil-Sen"};

    /**
     * Konstruktor
     *
     * @param view View
     */
    public EvaluationPanel(MainFrame view) {
        super();
        this.view = view;
        this.setLayout(new BorderLayout());
        this.currentRowOfTypes = 0;
        this.colorMap = new HashMap<>();
        this.colorMap.put(Algorithm.Type.LMS, Color.ORANGE);
        this.colorMap.put(Algorithm.Type.RM, Color.RED);
        this.colorMap.put(Algorithm.Type.TS, Color.MAGENTA);
        this.colorMap.put(Algorithm.Type.NAIV_LMS, Color.BLUE);
        this.colorMap.put(Algorithm.Type.NAIV_RM, Color.GREEN);
        this.colorMap.put(Algorithm.Type.NAIV_TS, Color.CYAN);

        init();
        addListener();
        addComponents();

    }

    /**
     * Initialisiere die Komponenten
     */
    private void init() {
        datasetCountLabel = new JLabel("Größe des Datensatzes");
        Integer[] choice = {50, 100, 200, 500, 1000, 1500};
        datasetCountChoice = new JComboBox(choice);
        String[] datatypes = {"Punktwolke", "Gerade", "Kreis und Gerade", "Import von CSV-Datei"};
        datasetType = new JComboBox<>(datatypes);

        start = new JButton("Start");
        clearTable = new JButton("Löschen");
        exportData = new JButton("Datenexport");
        clearTable.setEnabled(false);
        exportData.setEnabled(false);
        evalTypeOne = new JRadioButton("Algorithmus evaluieren");
        evalTypeTwo = new JRadioButton("Algorithmen vergleichen");
        lms = new JCheckBox("Least Median of Squares");
        rm = new JCheckBox("Repeated Median");
        ts = new JCheckBox("Theil-Sen");
        radiobuttonGroup = new ButtonGroup();
        checkboxGroup = new ButtonGroup();
        checkboxAtLeastOne = new ButtonGroupAtLeastTwo();
        buttonPanel = new JPanel(new FlowLayout());
        leftSidePanel = new JPanel(new BorderLayout());
        comp = new JPanel(new GridLayout(0, 1));
        plotPanel = new PlotPanel();
        model = new DefaultTableModel() {
            @Override
            public boolean isCellEditable(int row, int column) {
                //all cells false
                return false;
            }
        };
        table = new JTable(model);
        UIManager.put("TableHeader.font", new Font("SansSerif", Font.BOLD, 14));
        UIManager.put("TableHeader.foreground", Color.WHITE);
        header = table.getTableHeader();
        header.setBackground(Color.GRAY);

        algorithmPanel = new JPanel(new FlowLayout(FlowLayout.LEFT));
        datasetCount = new JPanel(new FlowLayout(FlowLayout.LEFT));

    }

    /**
     * Hinzufügen der Komponenten zum ContentPane
     */
    private void addComponents() {
        evalTypeOne.setSelected(true);
        lms.setSelected(true);
        checkboxGroup.add(lms);
        checkboxGroup.add(rm);
        checkboxGroup.add(ts);

        radiobuttonGroup.add(evalTypeOne);
        radiobuttonGroup.add(evalTypeTwo);

        buttonPanel.add(start);
        buttonPanel.add(clearTable);
        buttonPanel.add(exportData);

        algorithmPanel.add(lms);
        algorithmPanel.add(rm);
        algorithmPanel.add(ts);

        datasetCount.add(datasetCountLabel);
        datasetCount.add(datasetCountChoice);
        datasetCount.add(datasetType);

        comp.add(evalTypeOne);
        comp.add(evalTypeTwo);
        comp.add(algorithmPanel);
        comp.add(datasetCount);

        comp.setBorder(new TitledBorder("Konfiguration"));

        //Tabelle
        table.setDragEnabled(true);
        JScrollPane scrollPane = new JScrollPane(table);
        scrollPane.setWheelScrollingEnabled(true);
        scrollPane.setBorder(new TitledBorder("Ergebnisse"));


        //Plot
        plotPanel.createPlot(new LinkedList<>());
        plotPanel.setBorder(new TitledBorder("Plot"));

        leftSidePanel.add(comp, BorderLayout.NORTH);
        leftSidePanel.add(scrollPane, BorderLayout.CENTER);

        this.add(leftSidePanel, BorderLayout.CENTER);
        this.add(buttonPanel, BorderLayout.SOUTH);

    }

    /**
     * Hinzufügen der Listener
     */
    private void addListener() {
        start.addActionListener(e -> {
            int type;
            int alg;
            int n;
            if (radiobuttonGroup.isSelected(evalTypeOne.getModel())) {
                type = 0;
            } else {
                type = 1;
            }
            alg = checkSelection();
            n = (Integer) datasetCountChoice.getSelectedItem();
            String datatyp = (String) datasetType.getSelectedItem();
            if ("Import von CSV-Datei".equals(datatyp)) {
                SwingUtilities.invokeLater(() -> {
                    File file = null;
                    JFileChooser chooser = new JFileChooser(FileSystemView.getFileSystemView().getHomeDirectory());
                    chooser.setPreferredSize(new Dimension(800, 700));
                    chooser.setFileFilter(new FileNameExtensionFilter("Comma-Separated Value, (*.csv)", "csv", "text"));

                    chooser.setMultiSelectionEnabled(false);
                    chooser.setFileSelectionMode(JFileChooser.FILES_ONLY);

                    if (chooser.showOpenDialog(this) == JFileChooser.APPROVE_OPTION) {
                        //Logging.logInfo ("Datei "+chooser.getSelectedFile()+ " ausgewählt.");
                        file = chooser.getSelectedFile();
                        final File input = file;
                        Thread t = new Thread(() -> ((Presenter) view.getPresenter()).startEvaluation(type, alg, input));
                        t.start();
                    }
                });

            } else {
                ((Presenter) view.getPresenter()).startEvaluation(type, n, alg, DataProvider.DataType.values()[datasetType.getSelectedIndex()]);
            }
            clearTable.setEnabled(true);
            exportData.setEnabled(true);
        });

        evalTypeOne.addActionListener(e -> {
            checkboxAtLeastOne.remove(lms);
            checkboxAtLeastOne.remove(rm);
            checkboxAtLeastOne.remove(ts);
            lms.setSelected(true);

            checkboxGroup.add(lms);
            checkboxGroup.add(rm);
            checkboxGroup.add(ts);
        });

        evalTypeTwo.addActionListener(e -> {
            checkboxGroup.remove(lms);
            checkboxGroup.remove(rm);
            checkboxGroup.remove(ts);

            checkboxAtLeastOne.addAll(lms, rm, ts);

            lms.setSelected(true);
            rm.setSelected(true);
        });

        clearTable.addActionListener(e -> {
            SwingUtilities.invokeLater(() -> {
                int n = model.getDataVector().size();
                for (int i = 0; i < n; i++) {
                    model.removeRow(0);
                    currentRowOfTypes--;
                }
                this.revalidate();
                clearTable.setEnabled(false);
                exportData.setEnabled(false);
            });
        });


        exportData.addActionListener(e -> {
            SwingUtilities.invokeLater(() -> {
                SwingUtilities.invokeLater(() -> {
                    File file = null;
                    JFileChooser chooser = new JFileChooser(FileSystemView.getFileSystemView().getHomeDirectory());
                    chooser.setPreferredSize(new Dimension(800, 700));
                    chooser.setFileFilter(new FileNameExtensionFilter("Comma-Separated Value, (*.csv)", "csv", "text"));

                    chooser.setMultiSelectionEnabled(false);
                    chooser.setFileSelectionMode(JFileChooser.FILES_ONLY);


                    if (chooser.showOpenDialog(this) == JFileChooser.APPROVE_OPTION) {
                        file = chooser.getSelectedFile();
                        String filename = file.getAbsolutePath().contains(".csv") ? file.getAbsolutePath() : file.getAbsolutePath().concat(".csv");
                        final File input = new File(filename);
                        Thread t = new Thread(() -> ((Presenter) view.getPresenter()).startDatasetExportEvaluation(input));
                        t.start();
                    }
                });
            });
        });


    }

    /**
     * Fügt der Tabelle eine Spalte hinzu
     *
     * @param data    Daten der Spalte
     * @param col     Spalte
     * @param isLabel <code>true</code>, falls es sich um die Approximations Überschriften handelt
     */
    public void addColumn(final List<? extends Serializable> data, final int col, final boolean isLabel) {
        if (isLabel) {
            addBlankRows(data.size());
        }

        for (int i = 0; i < data.size(); i++) {
            int row = currentRowOfTypes - data.size() + i;
            model.setValueAt(data.get(i), row, col);
        }
        this.repaint();
        this.revalidate();
    }

    /**
     * Fügt der Tabelle eine Zeile hinzu
     *
     * @param tableEntries Daten der Tabelle
     */
    public void addRow(Map<Algorithm.Type, Map<String, String>> tableEntries) {

        SwingUtilities.invokeLater(() -> {

            Set<Algorithm.Type> types = tableEntries.keySet();
            List<String> columnlabels = new ArrayList<>();

            types.forEach(type -> {
                columnlabels.add(" ");
            });
            model.setColumnIdentifiers(columnlabels.toArray());
            tableEntries.entrySet().forEach(row -> {
                model.addRow(row.getValue().keySet().toArray());
                model.addRow(row.getValue().values().toArray());
            });

            this.repaint();
            this.revalidate();
        });
    }


    /**
     * <p>
     * Visualisierung der Ausgleichsgerade
     *
     * @param lines the lines
     * @param alg   identifizierung des Alg.
     */
    public void drawLines(final List<Line> lines, final Algorithm.Type alg) {
        for (Line line : lines) {
            plotPanel.addLineToPlot(line.getM(), line.getB(), colorMap.get(alg), alg.name());
        }
    }


    /**
     * Hilfsmethode
     *
     * @param n Anzahl der leeren Zeilen
     */
    private void addBlankRows(int n) {
        for (int i = 0; i < n; i++) {
            String[] tmp = {"", "", "", "",};
            model.addRow(tmp);
        }
    }

    /**
     * Hilfsmethode
     *
     * @param val Anzahl der Zeilen die noch hinzugefügt werden
     */
    public void setCurrentRow(int val) {
        this.currentRowOfTypes += val;
    }

    /**
     * @return Kodierung welche Algorithmen ausgewählt wurden
     */
    private int checkSelection() {
        if (lms.isSelected() && rm.isSelected() && ts.isSelected()) {
            return 6;
        } else if (!lms.isSelected() && rm.isSelected() && ts.isSelected()) {
            return 5;
        } else if (lms.isSelected() && !rm.isSelected() && ts.isSelected()) {
            return 4;
        } else if (lms.isSelected() && rm.isSelected() && !ts.isSelected()) {
            return 3;
        } else if (!lms.isSelected() && !rm.isSelected() && ts.isSelected()) {
            return 2;
        } else if (!lms.isSelected() && rm.isSelected() && !ts.isSelected()) {
            return 1;
        } else if (lms.isSelected() && !rm.isSelected() && !ts.isSelected()) {
            return 0;
        } else {
            throw new IllegalStateException("Mindestens ein Algortihmus muss selektiert werden");
        }
    }
}