studio.ui.action.CopyTableSelectionAction Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kdbStudio Show documentation
Studio for kdb+ is a rapid development environment for the ultra-fast database kdb+ from Kx Systems: http://www.kx.com.
There is a newer version: dz4.1
package studio.ui.action;

import studio.kdb.K;
import studio.kdb.KFormatContext;
import studio.ui.StudioOptionPane;
import studio.ui.Util;

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

public class CopyTableSelectionAction implements ActionListener {

    private final static String newline = System.getProperty("line.separator");

    public enum Format {Excel, Html}

    private final JTable table;
    private final Format format;

    private StringBuilder sb;
    private int numcols;
    private int numrows;
    private int[] rowsselected;
    private int[] colsselected;


    public CopyTableSelectionAction(Format format, JTable table) {
        this.table = table;
        this.format = format;
    }

    @Override
    public void actionPerformed(ActionEvent e) {
        sb = new StringBuilder();
        numcols = table.getSelectedColumnCount();
        numrows = table.getSelectedRowCount();
        if (numcols == -1 || numrows == -1) {
            StudioOptionPane.showError("Invalid Copy Selection",
                    "Invalid Copy Selection");
            return;
        }

        rowsselected = table.getSelectedRows();
        colsselected = table.getSelectedColumns();
        if (format == Format.Excel) copyExcelFormat();
        else copyHtmlFormat();
    }

    private void copyExcelFormat() {
        if (table.getSelectedRowCount() == table.getRowCount()) {
            for (int col = 0; col < numcols; col++) {
                sb.append(table.getColumnName(colsselected[col]));
                if (col < numcols - 1)
                    sb.append("\t");
            }
            sb.append(newline);
        }

        for (int row = 0; row < numrows; row++) {
            if (row > 0)
                sb.append(newline);
            for (int col = 0; col < numcols; col++) {
                boolean symColumn = table.getColumnClass(colsselected[col]) == K.KSymbolVector.class;
                if (symColumn)
                    sb.append("\"");

                K.KBase b = (K.KBase) table.getValueAt(rowsselected[row], colsselected[col]);
                if (!b.isNull())
                    sb.append(b.toString(KFormatContext.NO_TYPE));
                if (symColumn)
                    sb.append("\"");
                if (col < numcols - 1)
                    sb.append("\t");
            }
        }
        Util.copyTextToClipboard(sb.toString());
    }

    private void copyHtmlFormat() {
        sb.append("");

        sb.append("");
        for (int col = 0; col < numcols; col++) {
            sb.append("");
        }
        sb.append("").append(newline);

        for (int row = 0; row < numrows; row++) {
            if (row > 0) {
                sb.append(newline);
            }
            sb.append("");
            for (int col = 0; col < numcols; col++) {
                K.KBase b = (K.KBase) table.getValueAt(rowsselected[row], colsselected[col]);
                sb.append("");
            }
            sb.append("");
        }

        sb.append("").append(table.getColumnName(colsselected[col])).append("
");
                if (!b.isNull())
                    sb.append(b.toString(KFormatContext.NO_TYPE));
                sb.append("");
        Util.copyHtmlToClipboard(sb.toString());
    }

}