com.intellij.ui.treeStructure.treetable.TreeTableTree Java Examples

private DBObject getSelectedMongoDocument() {
    TreeTableTree tree = resultTableView.getTree();
    NoSqlTreeNode treeNode = (NoSqlTreeNode) tree.getLastSelectedPathComponent();
    if (treeNode == null) {
        return null;
    }

    NodeDescriptor descriptor = treeNode.getDescriptor();
    if (descriptor instanceof MongoKeyValueDescriptor) {
        MongoKeyValueDescriptor keyValueDescriptor = (MongoKeyValueDescriptor) descriptor;
        if (StringUtils.equals(keyValueDescriptor.getKey(), "_id")) {
            return mongoDocumentOperations.getMongoDocument(keyValueDescriptor.getValue());
        }
    }

    return null;
}
 

public JsonTreeTableView(TreeNode rootNode, ColumnInfo[] columnInfos) {
    super(new ListTreeTableModelOnColumns(rootNode, columnInfos));
    this.columns = columnInfos;

    final TreeTableTree tree = getTree();

    tree.setShowsRootHandles(true);
    tree.setRootVisible(false);
    UIUtil.setLineStyleAngled(tree);
    setTreeCellRenderer(new KeyCellRenderer());

    TreeUtil.expand(tree, 2);

    new TreeTableSpeedSearch(this, new Convertor<TreePath, String>() {
        @Override
        public String convert(final TreePath path) {
            final NoSqlTreeNode node = (NoSqlTreeNode) path.getLastPathComponent();
            NodeDescriptor descriptor = node.getDescriptor();
            return descriptor.getFormattedKey();
        }
    });
}
 

public boolean isSelectedNodeId() {
    if (resultTableView == null) {
        return false;
    }
    TreeTableTree tree = resultTableView.getTree();
    NoSqlTreeNode treeNode = (NoSqlTreeNode) tree.getLastSelectedPathComponent();
    if (treeNode == null) {
        return false;
    }

    NodeDescriptor descriptor = treeNode.getDescriptor();
    if (descriptor instanceof MongoKeyValueDescriptor) {
        MongoKeyValueDescriptor keyValueDescriptor = (MongoKeyValueDescriptor) descriptor;
        return StringUtils.equals(keyValueDescriptor.getKey(), "_id");
    }

    return false;
}
 

/**
 * Highlights refactoring-specific code fragment.
 */
private void highlightCode(InputEvent e) {
    TreeTableTree treeTableTree = treeTable.getTree();
    TreePath selectedPath = treeTableTree.getSelectionModel().getSelectionPath();
    if (selectedPath != null) {
        Object o = selectedPath.getLastPathComponent();
        if (refactoringType.instanceOfCandidateRefactoring(o)) {
            AbstractCandidateRefactoring refactoring = (AbstractCandidateRefactoring) o;
            refactoring.highlightCode();
        } else {
            expandOrCollapsePath(e, treeTableTree, selectedPath);
        }
    }
}
 

/**
 * Collapse if the selected path is extended, otherwise expand.
 */
public static void expandOrCollapsePath(InputEvent e, TreeTableTree treeTableTree, TreePath selectedPath) {
    if (e instanceof KeyEvent) {
        if (treeTableTree.isExpanded(selectedPath)) {
            treeTableTree.collapsePath(selectedPath);
        } else {
            treeTableTree.expandPath(selectedPath);
        }
    }
}
 

/**
 * Opens the definition of appropriate method for the selected suggestion by double-clicking or Enter key pressing.
 */
private void openMethodDefinition(InputEvent e) {
    TreeTableTree treeTableTree = treeTable.getTree();
    TreePath selectedPath = treeTableTree.getSelectionModel().getSelectionPath();
    if (selectedPath != null) {
        Object o = selectedPath.getLastPathComponent();
        if (o instanceof ASTSlice) {
            openDefinition(((ASTSlice) o).getSourceMethodDeclaration(), scope, (ASTSlice) o);
        } else if (o instanceof ExtractMethodCandidateGroup) {
            expandOrCollapsePath(e, treeTableTree, selectedPath);
        }
    }
}
 

void update() {
  columns.update();

  reload(getRoot());
  treeTable.updateUI();

  final TreeTableTree tree = treeTable.getTree();

  // Restore/preserve selection state.
  tree.setSelectionRows(treeTable.getSelectedRows());

  // Select and reveal a selection path (e.g., on error) if specified.
  if (pathToSelectAndReveal != null) {
    // TODO(pq): if there are timing issues, consider a delayed uiExec
    // TODO(pq): consider scrolling bounding rectangle to top of scrollpane
    tree.setSelectionPath(pathToSelectAndReveal);
    tree.scrollPathToVisible(pathToSelectAndReveal);
    pathToSelectAndReveal = null;
  }

  if (updateCallback != null) {
    updateCallback.updated();
  }

  if (autoScrollToEnd) {
    uiExec(this::scrollToEnd, 100);
  }
}
 

void update() {
  columns.update();

  reload(getRoot());
  treeTable.updateUI();

  final TreeTableTree tree = treeTable.getTree();

  // Restore/preserve selection state.
  tree.setSelectionRows(treeTable.getSelectedRows());

  // Select and reveal a selection path (e.g., on error) if specified.
  if (pathToSelectAndReveal != null) {
    // TODO(pq): if there are timing issues, consider a delayed uiExec
    // TODO(pq): consider scrolling bounding rectangle to top of scrollpane
    tree.setSelectionPath(pathToSelectAndReveal);
    tree.scrollPathToVisible(pathToSelectAndReveal);
    pathToSelectAndReveal = null;
  }

  if (updateCallback != null) {
    updateCallback.updated();
  }

  if (autoScrollToEnd) {
    uiExec(this::scrollToEnd, 100);
  }
}
 

public AbstractTreeUpdater(@Nonnull AbstractTreeBuilder treeBuilder) {
  myTreeBuilder = treeBuilder;
  final JTree tree = myTreeBuilder.getTree();
  final JComponent component = tree instanceof TreeTableTree ? ((TreeTableTree)tree).getTreeTable() : tree;
  myUpdateQueue = new MergingUpdateQueue("UpdateQueue", 100, component.isShowing(), component);
  myUpdateQueue.setRestartTimerOnAdd(false);

  final UiNotifyConnector uiNotifyConnector = new UiNotifyConnector(component, myUpdateQueue);
  Disposer.register(this, myUpdateQueue);
  Disposer.register(this, uiNotifyConnector);
}
 

@Override
public void setTree(JTree tree) {
  super.setTree(tree);
  treeTable = ((TreeTableTree)tree).getTreeTable();
  columns.init(treeTable);
}
 

@Override
public void setTree(JTree tree) {
  super.setTree(tree);
  treeTable = ((TreeTableTree)tree).getTreeTable();
  columns.init(treeTable);
}
 

void collapseAll() {
    TreeTableTree tree = resultTableView.getTree();
    TreeUtil.collapseAll(tree, 1);
}
 

void collapseAll() {
    TreeTableTree tree = resultTableView.getTree();
    TreeUtil.collapseAll(tree, 1);
}
 

void collapseAll() {
    TreeTableTree tree = resultTableView.getTree();
    TreeUtil.collapseAll(tree, 1);
}
 

@Override
public void setTree(final JTree tree) {
  myTreeTable = ((TreeTableTree)tree).getTreeTable();
}
 

@Override
protected void customizeCellRenderer(JTable table, Object value, boolean selected, boolean hasFocus, int row, int column) {

    TreeTableTree tree = ((TreeTable) table).getTree();
    TreePath pathForRow = tree.getPathForRow(row);

    final NoSqlTreeNode node = (NoSqlTreeNode) pathForRow.getLastPathComponent();

    node.getDescriptor().renderValue(this, tree.isExpanded(pathForRow));
}