Merge pull request #152 from SeeSharpSoft/fb_clear_cells_index_out_of_bound

[FIX] Index out of bound error on multi line clear cells

Author: SeeSharpSoft

src/main/java/net/seesharpsoft/intellij/plugins/csv/editor/table/CsvTableEditor.java

@@ -392,8 +392,9 @@ public final Object[][] removeRows(int[] indices) {
         currentRows.sort(Collections.reverseOrder());
         TableDataHandler dataHandler = getDataHandler();
         Object[][] currentData = dataHandler.getCurrentState();
+        int offset = getFileEditorState().getFixedHeaders() ? 1 : 0;
         for (int currentRow : currentRows) {
-            currentData = ArrayUtil.remove(currentData, currentRow + (getFileEditorState().getFixedHeaders() ? 1 : 0));
+            currentData = ArrayUtil.remove(currentData, currentRow + offset);
         }
         updateTableComponentData(dataHandler.addState(currentData));
         return currentData;
@@ -414,10 +415,8 @@ public final Object[][] addColumn(int focusedColumnIndex, boolean before) {
     public final Object[][] removeColumns(int[] indices) {
         List<Integer> currentColumns = Ints.asList(indices);
         currentColumns.sort(Collections.reverseOrder());
-
         TableDataHandler dataHandler = getDataHandler();
         Object[][] currentData = dataHandler.getCurrentState();
-
         for (int currentColumn : currentColumns) {
             for (int i = 0; i < currentData.length; ++i) {
                 currentData[i] = ArrayUtil.remove(currentData[i], currentColumn);
@@ -426,4 +425,17 @@ public final Object[][] removeColumns(int[] indices) {
         updateTableComponentData(dataHandler.addState(currentData));
         return currentData;
     }
+
+    public final Object[][] clearCells(int[] columns, int[] rows) {
+        TableDataHandler dataHandler = getDataHandler();
+        Object[][] currentData = dataHandler.getCurrentState();
+        int offset = getFileEditorState().getFixedHeaders() ? 1 : 0;
+        for (int currentColumn : columns) {
+            for (int currentRow : rows) {
+                currentData[currentRow + offset][currentColumn] = "";
+            }
+        }
+        updateTableComponentData(dataHandler.addState(currentData));
+        return currentData;
+    }
 }

src/main/java/net/seesharpsoft/intellij/plugins/csv/editor/table/swing/CsvTableEditorActions.java

@@ -8,7 +8,6 @@
 import net.seesharpsoft.intellij.plugins.csv.editor.CsvFileEditorProvider;
 
 import javax.swing.*;
-import javax.swing.table.DefaultTableModel;
 import java.awt.*;
 import java.awt.event.ActionEvent;
 import java.awt.event.ActionListener;
@@ -184,25 +183,17 @@ public void actionPerformed(ActionEvent e) {
             try {
                 JBTable table = csvTableEditor.getTable();
                 int[] selectedRows = table.getSelectedRows();
-                if (selectedRows == null || selectedRows.length == 0) {
-                    return;
-                }
                 int[] selectedColumns = table.getSelectedColumns();
-                if (selectedColumns == null || selectedColumns.length == 0) {
+
+                if (selectedRows == null || selectedRows.length == 0 ||
+                        selectedColumns == null || selectedColumns.length == 0) {
                     return;
                 }
+
                 int focusedRow = table.getSelectedRow();
                 int focusedColumn = table.getSelectedColumn();
 
-                DefaultTableModel tableModel = csvTableEditor.getTableModel();
-
-                for (int row : selectedRows) {
-                    for (int column : selectedColumns) {
-                        tableModel.setValueAt("", row, column);
-                    }
-                }
-
-                csvTableEditor.syncTableModelWithUI();
+                csvTableEditor.clearCells(selectedColumns, selectedRows);
 
                 selectCell(table, focusedRow, focusedColumn);
             } finally {