UI bugfixes in Profile tab

- Row selection instead of column selection
- Columns in tree table are now resized acording to content

Author: nikarh

ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/ColumnDefinitions.java

@@ -58,12 +58,6 @@ public Class<?> getColumnClass() {
                 public String valueOf(DefaultMutableTreeNode o) {
                     return ((GraphQueryPlan) o.getUserObject()).getOperatorType();
                 }
-
-                @Override
-                @NonNls
-                public String getPreferredStringValue() {
-                    return StringUtil.repeatSymbol('m', 25);
-                }
             };
 
     private static final ColumnInfo<DefaultMutableTreeNode, String> ESTIMATED_ROWS_COL =
@@ -86,12 +80,6 @@ public String valueOf(DefaultMutableTreeNode o) {
                             .map(Object::toString)
                             .collect(joining(", "));
                 }
-
-                @Override
-                @NonNls
-                public String getPreferredStringValue() {
-                    return StringUtil.repeatSymbol('m', 15);
-                }
             };
 
     private static final ColumnInfo<DefaultMutableTreeNode, String> IDENTIFIERS_COL =
@@ -102,12 +90,6 @@ public String valueOf(DefaultMutableTreeNode o) {
                     return ((GraphQueryPlan) o.getUserObject()).getIdentifiers().stream()
                             .collect(joining(", "));
                 }
-
-                @Override
-                @NonNls
-                public String getPreferredStringValue() {
-                    return StringUtil.repeatSymbol('m', 12);
-                }
             };
 
     @NotNull
@@ -129,12 +111,6 @@ public String valueOf(DefaultMutableTreeNode o) {
                         })
                         .orElse("0");
             }
-
-            @Override
-            @NonNls
-            public String getPreferredStringValue() {
-                return StringUtil.repeatSymbol('m', 5);
-            }
         };
     }
 }

ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/plan/QueryPlanPanel.java

@@ -15,6 +15,7 @@
 import com.intellij.ui.treeStructure.treetable.TreeTableModel;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryPlan;
 import com.neueda.jetbrains.plugin.graphdb.database.api.query.GraphQueryResult;
+import com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.table.DynamicTreeTable;
 import com.neueda.jetbrains.plugin.graphdb.platform.ShouldNeverHappenException;
 import org.jetbrains.annotations.NotNull;
 
@@ -55,10 +56,11 @@ public void initialize(Container container) {
 
         ListTreeTableModelOnColumns model = createModel(graphQueryPlan, result.isProfilePlan());
 
-        treeTable = new TreeTableView(model);
+        treeTable = new DynamicTreeTable(model);
         treeTable.setAutoCreateColumnsFromModel(true);
         treeTable.setRootVisible(true);
         treeTable.setCellSelectionEnabled(false);
+        treeTable.setRowSelectionAllowed(true);
         treeTable.setAutoResizeMode(TreeTableView.AUTO_RESIZE_OFF);
 
         treeTable.getTree().setShowsRootHandles(true);

ui/jetbrains/src/main/java/com/neueda/jetbrains/plugin/graphdb/jetbrains/ui/console/table/DynamicTreeTable.java

@@ -0,0 +1,46 @@
+package com.neueda.jetbrains.plugin.graphdb.jetbrains.ui.console.table;
+
+import com.intellij.ui.dualView.TreeTableView;
+import com.intellij.ui.treeStructure.treetable.ListTreeTableModelOnColumns;
+import org.jetbrains.annotations.NotNull;
+
+import javax.swing.table.TableCellRenderer;
+import javax.swing.table.TableColumn;
+import java.awt.*;
+
+/**
+ * DynamicTreeTable is a TreeTable which implement automatic column resizing depending
+ * on column content.
+ */
+public class DynamicTreeTable extends TreeTableView {
+
+    private final int minMargin;
+
+    public DynamicTreeTable(ListTreeTableModelOnColumns treeTableModel) {
+        super(treeTableModel);
+
+        minMargin = strWidth("ww");
+    }
+
+    @NotNull
+    @Override
+    public Component prepareRenderer(@NotNull TableCellRenderer renderer, int row, int column) {
+        Component component = super.prepareRenderer(renderer, row, column);
+
+        int rendererWidth = component.getPreferredSize().width;
+        TableColumn tableColumn = getColumnModel().getColumn(column);
+        String header = tableColumn.getHeaderValue().toString();
+
+
+        int width = Math.max(rendererWidth, strWidth(header)) +
+                Math.max(getIntercellSpacing().width, minMargin);
+        width = Math.max(width, tableColumn.getPreferredWidth());
+
+        tableColumn.setPreferredWidth(width);
+        return component;
+    }
+
+    private int strWidth(String str) {
+        return getFontMetrics(getFont()).charsWidth(str.toCharArray(), 0, str.length());
+    }
+}