package org.ntlab.traceDebugger; import java.util.ArrayList; import java.util.HashMap; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Set; import org.eclipse.core.resources.IMarker; import org.eclipse.core.runtime.CoreException; import org.eclipse.jface.action.Action; import org.eclipse.jface.action.IAction; import org.eclipse.jface.action.IMenuListener; import org.eclipse.jface.action.IMenuManager; import org.eclipse.jface.action.IToolBarManager; import org.eclipse.jface.action.MenuManager; import org.eclipse.jface.action.Separator; import org.eclipse.jface.dialogs.InputDialog; import org.eclipse.jface.viewers.ISelectionChangedListener; import org.eclipse.jface.viewers.IStructuredSelection; import org.eclipse.jface.viewers.ITreeViewerListener; import org.eclipse.jface.viewers.SelectionChangedEvent; import org.eclipse.jface.viewers.TreeExpansionEvent; import org.eclipse.jface.viewers.TreeNode; import org.eclipse.jface.viewers.TreeViewer; import org.eclipse.swt.SWT; import org.eclipse.swt.widgets.Composite; import org.eclipse.swt.widgets.Menu; import org.eclipse.swt.widgets.Tree; import org.eclipse.swt.widgets.TreeColumn; import org.eclipse.swt.widgets.TreeItem; import org.eclipse.ui.IWorkbenchActionConstants; import org.eclipse.ui.IWorkbenchPage; import org.eclipse.ui.part.ViewPart; import org.ntlab.traceAnalysisPlatform.tracer.trace.MethodExecution; import org.ntlab.traceAnalysisPlatform.tracer.trace.TracePoint; import org.ntlab.traceDebugger.analyzerProvider.Alias; import org.ntlab.traceDebugger.analyzerProvider.DeltaMarkerManager; public class VariableView extends ViewPart { private TreeViewer viewer; private IAction jumpAction; private IAction deltaAction; private IAction deltaActionForCollection; private IAction deltaActionForContainerToComponent; private IAction deltaActionForThisToAnother; private Variable selectedVariable; private Variables variables = Variables.getInstance(); public static final String ID = "org.ntlab.traceDebugger.variableView"; public VariableView() { // TODO Auto-generated constructor stub System.out.println("VariableViewクラスが生成されたよ!"); } @Override public void createPartControl(Composite parent) { // TODO Auto-generated method stub System.out.println("VariableView#createPartControl(Composite)が呼ばれたよ!"); viewer = new TreeViewer(parent, SWT.BORDER | SWT.H_SCROLL | SWT.V_SCROLL | SWT.FULL_SELECTION); Tree tree = viewer.getTree(); tree.setHeaderVisible(true); tree.setLinesVisible(true); String[] treeColumnTexts = {"Name", "Value"}; int[] treeColumnWidth = {100, 200}; TreeColumn[] treeColumns = new TreeColumn[treeColumnTexts.length]; for (int i = 0; i < treeColumns.length; i++) { treeColumns[i] = new TreeColumn(tree, SWT.NULL); treeColumns[i].setText(treeColumnTexts[i]); treeColumns[i].setWidth(treeColumnWidth[i]); } viewer.setContentProvider(new MyTreeNodeContentProvider()); viewer.setLabelProvider(new VariableLabelProvider()); viewer.addSelectionChangedListener(new ISelectionChangedListener() { @Override public void selectionChanged(SelectionChangedEvent event) { IStructuredSelection sel = (IStructuredSelection)event.getSelection(); Object element = sel.getFirstElement(); if (element instanceof TreeNode) { Object value = ((TreeNode)element).getValue(); if (value instanceof Variable) { selectedVariable = (Variable)value; } } } }); viewer.addTreeListener(new ITreeViewerListener() { @Override public void treeExpanded(TreeExpansionEvent event) { // ツリーを開いた後に実行される。 ここでは開いたノードから3つ先のノードを生成して追加する。 Object element = event.getElement(); if (!(element instanceof MyTreeNode)) return; MyTreeNode expandedNode = (MyTreeNode)element; Object value = expandedNode.getValue(); if (!(value instanceof Variable)) return; List<MyTreeNode> childNodes = expandedNode.getChildList(); if (childNodes == null) return; for (MyTreeNode childNode : childNodes) { List<MyTreeNode> grandChildNodes = childNode.getChildList(); if (grandChildNodes == null) continue; for (MyTreeNode grandChildNode : grandChildNodes) { Variable grandChildVariable = (Variable)grandChildNode.getValue(); grandChildVariable.createNextHierarchyState(); List<Variable> list = grandChildVariable.getChildren(); List<MyTreeNode> nodes = new ArrayList<>(); for (int i = 0; i < list.size(); i++) { nodes.add(i, new MyTreeNode(list.get(i))); } grandChildNode.setChildList(nodes); } } viewer.refresh(); } @Override public void treeCollapsed(TreeExpansionEvent event) {} }); createActions(); createToolBar(); createMenuBar(); createPopupMenu(); TraceDebuggerPlugin.setActiveView(ID, this); } @Override public void setFocus() { // TODO Auto-generated method stub TraceDebuggerPlugin.setActiveView(ID, this); viewer.getControl().setFocus(); } private void createActions() { jumpAction = new Action() { public void run() { TracePoint tp = selectedVariable.getLastUpdatePoint(); if (tp == null) return; DebuggingController controller = DebuggingController.getInstance(); controller.jumpToTheTracePoint(tp, false); controller.stepOverAction(); } }; jumpAction.setText("Jump to Related Point"); jumpAction.setToolTipText("Jump to Related Point"); deltaAction = new Action() { @Override public void run() { DeltaMarkerView newDeltaMarkerView = (DeltaMarkerView)TraceDebuggerPlugin.createNewView(DeltaMarkerView.ID, IWorkbenchPage.VIEW_ACTIVATE); newDeltaMarkerView.extractDelta(selectedVariable, true); } }; deltaAction.setText("Extract Delta"); deltaAction.setToolTipText("Extract Delta"); deltaActionForContainerToComponent = new Action() { @Override public void run() { DeltaMarkerView newDeltaMarkerView = (DeltaMarkerView)TraceDebuggerPlugin.createNewView(DeltaMarkerView.ID, IWorkbenchPage.VIEW_ACTIVATE); newDeltaMarkerView.extractDelta(selectedVariable, true); } }; deltaActionForContainerToComponent.setText("Extract Delta"); deltaActionForContainerToComponent.setToolTipText("Extract Delta"); deltaActionForThisToAnother = new Action() { @Override public void run() { DeltaMarkerView newDeltaMarkerView = (DeltaMarkerView)TraceDebuggerPlugin.createNewView(DeltaMarkerView.ID, IWorkbenchPage.VIEW_ACTIVATE); newDeltaMarkerView.extractDelta(selectedVariable, false); } }; deltaActionForThisToAnother.setText("Extract Delta"); deltaActionForThisToAnother.setToolTipText("Extract Delta"); deltaActionForCollection = new Action() { @Override public void run() { // InputDialog inputContainerIdDialog = new InputDialog(null, "Extract Delta for Collection", "Input cotainer id", "87478208", null); InputDialog inputContainerIdDialog = new InputDialog(null, "Extract Delta for Collection", "Input cotainer id", "155140910", null); if (inputContainerIdDialog.open() != InputDialog.OK) return; String containerId = inputContainerIdDialog.getValue(); // InputDialog inputContainerTypeDialog = new InputDialog(null, "Extract Delta for Collection", "Input cotainer type", "java.util.LinkedHashSet", null); InputDialog inputContainerTypeDialog = new InputDialog(null, "Extract Delta for Collection", "Input cotainer type", "java.util.ArrayList", null); if (inputContainerTypeDialog.open() != InputDialog.OK) return; String containerType = inputContainerTypeDialog.getValue(); String valueId = selectedVariable.getValueId(); String valueType = selectedVariable.getValueClassName(); TracePoint tp = DebuggingController.getInstance().getCurrentTp(); Variable variable = new Variable("tmp", containerType, containerId, valueType, valueId, tp, false); DeltaMarkerView newDeltaMarkerView = (DeltaMarkerView)TraceDebuggerPlugin.createNewView(DeltaMarkerView.ID, IWorkbenchPage.VIEW_ACTIVATE); newDeltaMarkerView.extractDelta(variable, true); } }; deltaActionForCollection.setText("Extract Delta for Collection"); deltaActionForCollection.setToolTipText("Extract Delta for Collection"); } private void createToolBar() { IToolBarManager mgr = getViewSite().getActionBars().getToolBarManager(); } private void createMenuBar() { IMenuManager mgr = getViewSite().getActionBars().getMenuManager(); } private void createPopupMenu() { MenuManager menuMgr = new MenuManager("#PopupMenu"); menuMgr.setRemoveAllWhenShown(true); menuMgr.addMenuListener(new IMenuListener() { @Override public void menuAboutToShow(IMenuManager manager) { // 右クリックする度に呼び出される manager.add(jumpAction); updateDeltaActionsTexts(selectedVariable); if (selectedVariable.getVariableName().equals(Variables.RETURN_VARIABLE_NAME)) { manager.add(deltaActionForContainerToComponent); manager.add(deltaActionForThisToAnother); } else { manager.add(deltaAction); // manager.add(deltaActionForCollection); } // manager.add(deltaActionForCollection); manager.add(new Separator(IWorkbenchActionConstants.MB_ADDITIONS)); } }); Menu menu = menuMgr.createContextMenu(viewer.getControl()); viewer.getControl().setMenu(menu); getSite().registerContextMenu(menuMgr, viewer); } public void reset() { variables.resetData(); viewer.setInput(variables.getVariablesTreeNodesList()); viewer.refresh(); } private void updateDeltaActionsTexts(Variable variable) { String valueId = selectedVariable.getValueId(); String valueClassName = selectedVariable.getValueClassName(); valueClassName = valueClassName.substring(valueClassName.lastIndexOf(".") + 1); String containerId = selectedVariable.getContainerId(); String containerClassName = selectedVariable.getContainerClassName(); if (containerId != null && containerClassName != null) { containerClassName = containerClassName.substring(containerClassName.lastIndexOf(".") + 1); String textForContainerToComponent = String.format("Extract Delta (%s: %s → %s: %s)", containerId, containerClassName, valueId, valueClassName); deltaActionForContainerToComponent.setText(textForContainerToComponent); deltaActionForContainerToComponent.setToolTipText(textForContainerToComponent); deltaAction.setText(textForContainerToComponent); deltaAction.setToolTipText(textForContainerToComponent); } else { deltaAction.setText("Extract Delta"); deltaAction.setToolTipText("Extract Delta"); } TracePoint before = selectedVariable.getBeforeTracePoint(); String thisId = before.getMethodExecution().getThisObjId(); String thisClassName = before.getMethodExecution().getThisClassName(); if (thisId != null && thisClassName != null) { thisClassName = thisClassName.substring(thisClassName.lastIndexOf(".") + 1); String textForThisToAnother = String.format("Extract Delta (%s: %s → %s: %s)", thisId, thisClassName, valueId, valueClassName); deltaActionForThisToAnother.setText(textForThisToAnother); deltaActionForThisToAnother.setToolTipText(textForThisToAnother); } } public void updateVariablesByTracePoint(TracePoint tp, boolean isReturned) { updateVariablesByTracePoint(null, tp, isReturned); } public void updateVariablesByTracePoint(TracePoint from, TracePoint to, boolean isReturned) { variables.updateAllObjectDataByTracePoint(from, to, isReturned); viewer.setInput(variables.getVariablesTreeNodesList()); } public void updateVariablesForDifferential(TracePoint from, TracePoint to, boolean isReturned) { variables.updateForDifferentialAndReturnValue(from, to, isReturned); // viewer.setInput(variables.getVariablesTreeNodes()); viewer.refresh(); } public void markAndExpandVariablesByDeltaMarkers(Map<String, List<IMarker>> markers) { List<IMarker> srcSideDeltaMarkers = markers.get(DeltaMarkerManager.SRC_SIDE_DELTA_MARKER); List<IMarker> dstSideDeltaMarkers = markers.get(DeltaMarkerManager.DST_SIDE_DELTA_MARKER); List<IMarker> coordinatorMarker = markers.get(DeltaMarkerManager.COORDINATOR_DELTA_MARKER); if (srcSideDeltaMarkers != null) { markVariables(DeltaMarkerManager.SRC_SIDE_DELTA_MARKER, srcSideDeltaMarkers); } if (dstSideDeltaMarkers != null) { markVariables(DeltaMarkerManager.DST_SIDE_DELTA_MARKER, dstSideDeltaMarkers); } if (coordinatorMarker != null) { markVariables(DeltaMarkerManager.COORDINATOR_DELTA_MARKER, coordinatorMarker); } viewer.refresh(); expandAllMarkedNodes(); } private void markVariables(String markerId, List<IMarker> markerList) { Set<String> idSet = new HashSet<>(); Map<String, Object> additionalAttributesForVariables = new HashMap<>(); additionalAttributesForVariables.put("markerId", markerId); for (IMarker marker : markerList) { try { Object data = marker.getAttribute(DeltaMarkerManager.DELTA_MARKER_ATR_DATA); if (data instanceof Alias) { idSet.add(((Alias)data).getObjectId()); } else if (data instanceof MethodExecution) { idSet.add(((MethodExecution)data).getThisObjId()); } } catch (CoreException e) { e.printStackTrace(); } } variables.addAdditionalAttributes(idSet, additionalAttributesForVariables); } private void expandAllMarkedNodes() { Set<TreeNode> expandedNodes = new HashSet<>(); for (TreeItem item : viewer.getTree().getItems()) { Object obj = item.getData(); if (!(obj instanceof TreeNode)) continue; collectNodes((TreeNode)obj, expandedNodes); } viewer.setExpandedElements(expandedNodes.toArray(new Object[expandedNodes.size()])); } private void collectNodes(TreeNode node, final Set<TreeNode> expandedNodes) { Object value = node.getValue(); if (!(value instanceof Variable)) return; Variable variable = (Variable)value; if (variable.getAdditionalAttribute("markerId") != null) { TreeNode parent = node.getParent(); if (parent != null) { expandedNodes.add(parent); } } TreeNode[] children = node.getChildren(); if (children == null) return; for (TreeNode child : children) { collectNodes(child, expandedNodes); } } }