Examples with BlockTreeBuilder com.intellij.internal.psiView.formattingblocks.BlockTreeBuilder used on opensource projects

Example 1 with BlockTreeBuilder

use of com.intellij.internal.psiView.formattingblocks.BlockTreeBuilder in project intellij-community by JetBrains.

the class PsiViewerDialog method doOKAction.

@Override
protected void doOKAction() {
    if (myBlockTreeBuilder != null) {
        Disposer.dispose(myBlockTreeBuilder);
    }
    final String text = myEditor.getDocument().getText();
    myEditor.getSelectionModel().removeSelection();
    myLastParsedText = text;
    myLastParsedTextHashCode = text.hashCode();
    myNewDocumentHashCode = myLastParsedTextHashCode;
    PsiElement rootElement = parseText(text);
    focusTree();
    ViewerTreeStructure structure = getTreeStructure();
    structure.setRootPsiElement(rootElement);
    myPsiTreeBuilder.queueUpdate();
    myPsiTree.setRootVisible(true);
    myPsiTree.expandRow(0);
    myPsiTree.setRootVisible(false);
    if (!myShowBlocksCheckBox.isSelected()) {
        return;
    }
    Block rootBlock = rootElement == null ? null : buildBlocks(rootElement);
    if (rootBlock == null) {
        myBlockTreeBuilder = null;
        myBlockTree.setRootVisible(false);
        myBlockTree.setVisible(false);
        return;
    }
    myBlockTree.setVisible(true);
    BlockTreeStructure blockTreeStructure = new BlockTreeStructure();
    BlockTreeNode rootNode = new BlockTreeNode(rootBlock, null);
    blockTreeStructure.setRoot(rootNode);
    myBlockTreeBuilder = new BlockTreeBuilder(myBlockTree, blockTreeStructure);
    myPsiToBlockMap = new HashMap<>();
    final PsiElement psiFile = (getTreeStructure()).getRootPsiElement();
    initMap(rootNode, psiFile);
    PsiElement rootPsi = rootNode.getBlock() instanceof ASTBlock ? ((ASTBlock) rootNode.getBlock()).getNode().getPsi() : rootElement;
    BlockTreeNode blockNode = myPsiToBlockMap.get(rootPsi);
    if (blockNode == null) {
        LOG.error(LogMessageEx.createEvent("PsiViewer: rootNode not found", "Current language: " + rootElement.getContainingFile().getLanguage(), AttachmentFactory.createAttachment(rootElement.getContainingFile().getOriginalFile().getVirtualFile())));
        blockNode = findBlockNode(rootPsi);
    }
    blockTreeStructure.setRoot(blockNode);
    myBlockTree.addTreeSelectionListener(new MyBlockTreeSelectionListener());
    myBlockTree.setRootVisible(true);
    myBlockTree.expandRow(0);
    myBlockTreeBuilder.queueUpdate();
}

Example 2 with BlockTreeBuilder

use of com.intellij.internal.psiView.formattingblocks.BlockTreeBuilder in project intellij-community by JetBrains.

the class PsiViewerDialog method findBlockNode.

@Nullable
private BlockTreeNode findBlockNode(TextRange range, boolean selectParentIfNotFound) {
    final BlockTreeBuilder builder = myBlockTreeBuilder;
    if (builder == null || !myBlockStructurePanel.isVisible()) {
        return null;
    }
    AbstractTreeStructure treeStructure = builder.getTreeStructure();
    if (treeStructure == null)
        return null;
    BlockTreeNode node = (BlockTreeNode) treeStructure.getRootElement();
    main_loop: while (true) {
        if (node.getBlock().getTextRange().equals(range)) {
            return node;
        }
        for (BlockTreeNode child : node.getChildren()) {
            if (child.getBlock().getTextRange().contains(range)) {
                node = child;
                continue main_loop;
            }
        }
        return selectParentIfNotFound ? node : null;
    }
}