processing/java/src/processing/mode/java/pdex/DebugTree.java at master · sunsure/processing · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

package processing.mode.java.pdex;

import java.awt.EventQueue;

import java.awt.Rectangle;

import java.awt.event.ComponentAdapter;

import java.awt.event.ComponentEvent;

import java.util.ArrayDeque;

import java.util.Deque;

import java.util.function.Consumer;

import javax.swing.JDialog;

import javax.swing.JScrollPane;

import javax.swing.JTree;

import javax.swing.WindowConstants;

import javax.swing.tree.DefaultMutableTreeNode;

import javax.swing.tree.DefaultTreeModel;

import org.eclipse.jdt.core.dom.ASTNode;

import org.eclipse.jdt.core.dom.ASTVisitor;

import org.eclipse.jdt.core.dom.CompilationUnit;

import processing.app.Messages;

import processing.app.ui.ZoomTreeCellRenderer;

import processing.mode.java.JavaEditor;

import processing.mode.java.pdex.PreprocessedSketch.SketchInterval;

class DebugTree {

final JDialog window;

final JTree tree;

final Consumer<PreprocessedSketch> updateListener;

DebugTree(JavaEditor editor, PreprocessingService pps) {

updateListener = this::buildAndUpdateTree;

window = new JDialog(editor);

tree = new JTree() {

@Override

public String convertValueToText(Object value, boolean selected,

boolean expanded, boolean leaf,

int row, boolean hasFocus) {

if (value instanceof DefaultMutableTreeNode) {

DefaultMutableTreeNode treeNode = (DefaultMutableTreeNode) value;

Object o = treeNode.getUserObject();

if (o instanceof ASTNode) {

ASTNode node = (ASTNode) o;

return CompletionGenerator.getNodeAsString(node);

}

return super.convertValueToText(value, selected, expanded, leaf, row, hasFocus);

}

};

tree.setCellRenderer(new ZoomTreeCellRenderer(editor.getMode()));

window.addComponentListener(new ComponentAdapter() {

@Override

public void componentHidden(ComponentEvent e) {

pps.unregisterListener(updateListener);

tree.setModel(null);

}

});

window.setDefaultCloseOperation(WindowConstants.HIDE_ON_CLOSE);

window.setBounds(new Rectangle(680, 100, 460, 620));

window.setTitle("AST View - " + editor.getSketch().getName());

JScrollPane sp = new JScrollPane();

sp.setViewportView(tree);

window.add(sp);

pps.whenDone(updateListener);

pps.registerListener(updateListener);

tree.addTreeSelectionListener(e -> {

if (tree.getLastSelectedPathComponent() != null) {

DefaultMutableTreeNode tnode =

(DefaultMutableTreeNode) tree.getLastSelectedPathComponent();

if (tnode.getUserObject() instanceof ASTNode) {

ASTNode node = (ASTNode) tnode.getUserObject();

pps.whenDone(ps -> {

SketchInterval si = ps.mapJavaToSketch(node);

if (!ps.inRange(si)) return;

EventQueue.invokeLater(() -> {

editor.highlight(si.tabIndex, si.startTabOffset, si.stopTabOffset);

});

}

});

}

void dispose() {

if (window != null) {

window.dispose();

}

// Thread: worker

void buildAndUpdateTree(PreprocessedSketch ps) {

CompilationUnit cu = ps.compilationUnit;

if (cu.types().isEmpty()){

Messages.loge("No Type found in CU");

return;

}

Deque<DefaultMutableTreeNode> treeNodeStack = new ArrayDeque<>();

ASTNode type0 = (ASTNode) cu.types().get(0);

type0.accept(new ASTVisitor() {

@Override

public boolean preVisit2(ASTNode node) {

treeNodeStack.push(new DefaultMutableTreeNode(node));

return super.preVisit2(node);

}

@Override

public void postVisit(ASTNode node) {

if (treeNodeStack.size() > 1) {

DefaultMutableTreeNode treeNode = treeNodeStack.pop();

treeNodeStack.peek().add(treeNode);

}

});

DefaultMutableTreeNode codeTree = treeNodeStack.pop();

EventQueue.invokeLater(() -> {

if (tree.hasFocus() || window.hasFocus()) {

return;

}

tree.setModel(new DefaultTreeModel(codeTree));

((DefaultTreeModel) tree.getModel()).reload();

tree.validate();

if (!window.isVisible()) {

window.setVisible(true);

}

});

}

processing/java/src/processing/mode/java/pdex/DebugTree.java at master · sunsure/processing · GitHub

Repository navigation

Expand file tree

Breadcrumbs

DebugTree.java

Latest commit

History

Breadcrumbs

DebugTree.java

File metadata and controls