aswi2020vldc-gitlab/src/main/java/vldc/aswi/utils/Converter.java @ ba8b306a

package vldc.aswi.utils;

import vldc.aswi.model.table.TableColumn;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRow;
import vldc.aswi.model.table.contingencyTable.ContingencyTableRowCell;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * Class for converting classes.
 */
public class Converter {

    /**
     * Convert Map of TableColumns into list of ContingencyTableRow.
     * @param tableColumns - Map of TableColumn.
     * @return List of ContingencyTableRow.
     */
    public static List<ContingencyTableRow> convertTableColumnsToContingencyTableRows(Map<String, TableColumn> tableColumns) {
        List<ContingencyTableRow> contingencyTableRows = new ArrayList<>();

        // Create header row with column names.
        ContingencyTableRow contingencyTableRowHeader = new ContingencyTableRow(true, 0);
        for (String columnName : tableColumns.keySet()) {
            contingencyTableRowHeader.addTableRowCell(new ContingencyTableRowCell(columnName, 0));
        }
        contingencyTableRows.add(contingencyTableRowHeader);

        boolean listExpanded = false;

        // Create contingency table row.
        for (TableColumn tableColumn : tableColumns.values()) {
            List<String> values = tableColumn.getValues();

            // If list is not created, then create it.
            if (!listExpanded) {
                for (int j = 0; j < values.size(); j++) {
                    ContingencyTableRow contingencyTableRow = new ContingencyTableRow(false, 0);
                    contingencyTableRows.add(contingencyTableRow);
                }
                listExpanded = true;
            }

            // Iterate through all values and assign one value per one row.
            for (int j = 0; j < values.size(); j++) {
                ContingencyTableRowCell contingencyTableRowCell = new ContingencyTableRowCell(values.get(j), 0);
                contingencyTableRows.get(j + 1).addTableRowCell(contingencyTableRowCell);
            }
        }

        return contingencyTableRows;
    }
}