android/src/com/android/tools/idea/editors/vmtrace/TraceViewPanel.java - platform/tools/adt/idea - Git at Google

 /*
  * Copyright (C) 2013 The Android Open Source Project
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *      http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 package com.android.tools.idea.editors.vmtrace;

 import com.android.tools.idea.editors.vmtrace.treemodel.VmStatsTreeTableModel;
 import com.android.tools.idea.editors.vmtrace.treemodel.VmStatsTreeUtils;
 import com.android.tools.perflib.vmtrace.ClockType;
 import com.android.tools.perflib.vmtrace.SearchResult;
 import com.android.tools.perflib.vmtrace.ThreadInfo;
 import com.android.tools.perflib.vmtrace.VmTraceData;
 import com.android.tools.perflib.vmtrace.viz.TraceViewCanvas;
 import com.google.common.base.Predicate;
 import com.google.common.collect.Iterables;
 import com.intellij.find.editorHeaderActions.Utils;
 import com.intellij.icons.AllIcons;
 import com.intellij.openapi.actionSystem.DataKey;
 import com.intellij.openapi.actionSystem.DataProvider;
 import com.intellij.openapi.editor.impl.EditorHeaderComponent;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.openapi.wm.IdeFocusManager;
 import com.intellij.ui.ColoredListCellRenderer;
 import com.intellij.ui.DocumentAdapter;
 import com.intellij.ui.JBSplitter;
 import com.intellij.ui.SearchTextField;
 import com.intellij.ui.components.JBLabel;
 import com.intellij.ui.components.JBScrollPane;
 import com.intellij.ui.components.panels.NonOpaquePanel;
 import com.intellij.ui.treeStructure.treetable.TreeTable;
 import icons.AndroidIcons;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;

 import javax.swing.*;
 import javax.swing.event.DocumentEvent;
 import javax.swing.table.TableModel;
 import javax.swing.table.TableRowSorter;
 import javax.swing.text.BadLocationException;
 import java.awt.*;
 import java.awt.event.*;
 import java.util.List;

 public class TraceViewPanel {
   @NonNls public static DataKey<TraceViewPanel> KEY = DataKey.create("android.traceview.panel");

   // The names for the cards used in the card layout.
   // Note that these are duplicated in the layout form.
   @NonNls private static final String CARD_FIND = "FIND";
   @NonNls private static final String CARD_DEFAULT = "DEFAULT";

   /** Default name for main thread in Android apps. */
   @NonNls private static final String MAIN_THREAD_NAME = "main";

   private final Project myProject;

   private JPanel myContainer;

   private JPanel myHeaderPanel;
   private TraceViewCanvas myTraceViewCanvas;
   private TreeTable myTreeTable;

   @SuppressWarnings("UnusedDeclaration") // custom creation only
   private JPanel myDefaultHeaderPanel;
   private JComboBox myThreadCombo;
   private JComboBox myRenderClockSelectorCombo;

   @SuppressWarnings("UnusedDeclaration") // custom creation only
   private JPanel myFindPanel;
   private JPanel myFindFieldWrapper;
   private SearchTextField mySearchField;
   private JLabel myCloseLabel;
   private JBLabel myResultsLabel;
   private JLabel mySearchLabel;
   private JBSplitter mySplitter;
   private JLabel myZoomFitLabel;
   private JCheckBox myUseInclusiveTimeForColoring;

   private static final String[] ourRenderClockOptions = new String[] {
     "Wall Clock Time",
     "Thread Time",
   };

   private static final ClockType[] ourRenderClockTypes = new ClockType[] {
     ClockType.GLOBAL,
     ClockType.THREAD,
   };
   private VmTraceData myTraceData;
   private VmStatsTreeTableModel myVmStatsTreeTableModel;

   public TraceViewPanel(Project project) {
     myProject = project;

     myRenderClockSelectorCombo.setModel(new DefaultComboBoxModel(ourRenderClockOptions));
     myRenderClockSelectorCombo.setSelectedIndex(0);

     ActionListener l = new ActionListener() {
       @Override
       public void actionPerformed(ActionEvent e) {
         if (e.getSource() == myThreadCombo) {
           final ThreadInfo selectedThread = (ThreadInfo)myThreadCombo.getSelectedItem();
           myTraceViewCanvas.displayThread(selectedThread);
           myVmStatsTreeTableModel.setThread(selectedThread);
         } else if (e.getSource() == myRenderClockSelectorCombo) {
           myTraceViewCanvas.setRenderClock(getCurrentRenderClock());
           myVmStatsTreeTableModel.setClockType(getCurrentRenderClock());
         } else if (e.getSource() == myUseInclusiveTimeForColoring) {
           myTraceViewCanvas.setUseInclusiveTimeForColorAssignment(myUseInclusiveTimeForColoring.isSelected());
         }
       }
     };

     myThreadCombo.addActionListener(l);
     myRenderClockSelectorCombo.addActionListener(l);
     myUseInclusiveTimeForColoring.addActionListener(l);
     myUseInclusiveTimeForColoring.setOpaque(false);
   }

   private SearchTextField createSearchField() {
     SearchTextField stf = new SearchTextField(true);
     stf.setOpaque(false);
     stf.setEnabled(true);
     Utils.setSmallerFont(stf);

     stf.addDocumentListener(new DocumentAdapter() {
       @Override
       protected void textChanged(DocumentEvent e) {
         searchTextChanged(getText(e));
       }

       private String getText(DocumentEvent e) {
         try {
           return e.getDocument().getText(0, e.getDocument().getLength());
         }
         catch (BadLocationException e1) {
           return "";
         }
       }
     });

     JTextField editorTextField = stf.getTextEditor();
     editorTextField.setMinimumSize(new Dimension(200, -1));

     editorTextField.registerKeyboardAction(new ActionListener() {
       @Override
       public void actionPerformed(ActionEvent e) {
         closeSearchComponent();
       }
     }, KeyStroke.getKeyStroke(KeyEvent.VK_ESCAPE, 0), JComponent.WHEN_IN_FOCUSED_WINDOW);

     return stf;
   }

   private void searchTextChanged(@Nullable String pattern) {
     if (StringUtil.isEmpty(pattern)) {
       myTraceViewCanvas.setHighlightMethods(null);
       myResultsLabel.setText("");
       return;
     }

     ThreadInfo thread = (ThreadInfo)myThreadCombo.getSelectedItem();
     SearchResult results = myTraceData.searchFor(pattern, thread);
     myTraceViewCanvas.setHighlightMethods(results.getMethods());

     String result = String.format("%1$d %2$s, %3$d %4$s",
                                   results.getMethods().size(),
                                   StringUtil.pluralize("method", results.getMethods().size()),
                                   results.getInstances().size(),
                                   StringUtil.pluralize("instance", results.getInstances().size()));
     myResultsLabel.setText(result);
   }

   public void setTrace(@NotNull VmTraceData trace) {
     myTraceData = trace;

     List<ThreadInfo> threads = trace.getThreads(true);
     if (threads.isEmpty()) {
       return;
     }

     ThreadInfo defaultThread = Iterables.find(threads, new Predicate<ThreadInfo>() {
       @Override
       public boolean apply(ThreadInfo input) {
         return MAIN_THREAD_NAME.equals(input.getName());
       }
     }, threads.get(0));

     myTraceViewCanvas.setTrace(trace, defaultThread, getCurrentRenderClock());
     myThreadCombo.setModel(new DefaultComboBoxModel(threads.toArray()));
     myThreadCombo.setSelectedItem(defaultThread);
     myThreadCombo.setRenderer(new ColoredListCellRenderer() {
       @Override
       protected void customizeCellRenderer(JList list, Object value, int index, boolean selected, boolean hasFocus) {
         String name = value instanceof ThreadInfo ? ((ThreadInfo)value).getName() : value.toString();
         append(name);
       }
     });

     myThreadCombo.setEnabled(true);
     myRenderClockSelectorCombo.setEnabled(true);

     myVmStatsTreeTableModel.setTraceData(trace, defaultThread);
     myVmStatsTreeTableModel.setClockType(getCurrentRenderClock());
     myTreeTable.setModel(myVmStatsTreeTableModel);

     VmStatsTreeUtils.adjustTableColumnWidths(myTreeTable);
     VmStatsTreeUtils.setCellRenderers(myTreeTable);
     VmStatsTreeUtils.setSpeedSearch(myTreeTable);
     VmStatsTreeUtils.enableSorting(myTreeTable, myVmStatsTreeTableModel);
   }

   private ClockType getCurrentRenderClock() {
     return ourRenderClockTypes[myRenderClockSelectorCombo.getSelectedIndex()];
   }

   @NotNull
   public JComponent getComponent() {
     return myContainer;
   }

   private void createUIComponents() {
     MouseAdapter l = new MouseAdapter() {
       @Override
       public void mousePressed(MouseEvent e) {
         if (e.getSource() == myCloseLabel) {
           closeSearchComponent();
         } else if (e.getSource() == mySearchLabel) {
           showSearchComponent();
         } else if (e.getSource() == myZoomFitLabel) {
           myTraceViewCanvas.zoomFit();
         }
       }
     };

     myDefaultHeaderPanel = new EditorHeaderComponent();
     mySearchLabel = new JLabel(AllIcons.Actions.Search);
     mySearchLabel.addMouseListener(l);
     mySearchLabel.setToolTipText("Find (Ctrl + F)");
     myZoomFitLabel = new JLabel(AndroidIcons.ZoomFit);
     myZoomFitLabel.setToolTipText("Zoom Fit");
     myZoomFitLabel.addMouseListener(l);

     myFindPanel = new EditorHeaderComponent();
     myFindFieldWrapper = new NonOpaquePanel(new BorderLayout());
     mySearchField = createSearchField();
     myFindFieldWrapper.add(mySearchField);

     myCloseLabel = new JLabel(AllIcons.Actions.Cross);
     myCloseLabel.addMouseListener(l);

     myVmStatsTreeTableModel = new VmStatsTreeTableModel();
     myTreeTable = new TreeTable(myVmStatsTreeTableModel);
     myTraceViewCanvas = new TraceViewCanvasWrapper();
     JBScrollPane scrollPane = new JBScrollPane(myTreeTable);

     mySplitter = new JBSplitter(true, 0.75f);
     mySplitter.setShowDividerControls(true);
     mySplitter.setShowDividerIcon(true);
     mySplitter.setFirstComponent(myTraceViewCanvas);
     mySplitter.setSecondComponent(scrollPane);
   }

   public void showSearchComponent() {
     CardLayout layout = (CardLayout)myHeaderPanel.getLayout();
     layout.show(myHeaderPanel, CARD_FIND);
     IdeFocusManager.getInstance(myProject).requestFocus(mySearchField, true);
   }

   private void closeSearchComponent() {
     CardLayout layout = (CardLayout)myHeaderPanel.getLayout();
     layout.show(myHeaderPanel, CARD_DEFAULT);
     IdeFocusManager.getInstance(myProject).requestFocus(myTraceViewCanvas, true);
   }

   /**
    * {@link TraceViewCanvasWrapper} is a wrapper around {@link TraceViewCanvas} that also implements the {@link DataProvider} interface.
    * This allows {@link VmTraceEditorSearchAction} to identify the editor from the current context of an event.
    */
   private class TraceViewCanvasWrapper extends TraceViewCanvas implements DataProvider {
     public TraceViewCanvasWrapper() {
       addMouseListener(new MouseAdapter() {
         @Override
         public void mouseReleased(MouseEvent e) {
           IdeFocusManager.getInstance(myProject).requestFocus(TraceViewCanvasWrapper.this, true);
         }
       });
     }

     @Nullable
     @Override
     public Object getData(@NonNls String dataId) {
       return KEY.is(dataId) ? TraceViewPanel.this : null;
     }
   }
 }
	/*
	* Copyright (C) 2013 The Android Open Source Project
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	package com.android.tools.idea.editors.vmtrace;

	import com.android.tools.idea.editors.vmtrace.treemodel.VmStatsTreeTableModel;
	import com.android.tools.idea.editors.vmtrace.treemodel.VmStatsTreeUtils;
	import com.android.tools.perflib.vmtrace.ClockType;
	import com.android.tools.perflib.vmtrace.SearchResult;
	import com.android.tools.perflib.vmtrace.ThreadInfo;
	import com.android.tools.perflib.vmtrace.VmTraceData;
	import com.android.tools.perflib.vmtrace.viz.TraceViewCanvas;
	import com.google.common.base.Predicate;
	import com.google.common.collect.Iterables;
	import com.intellij.find.editorHeaderActions.Utils;
	import com.intellij.icons.AllIcons;
	import com.intellij.openapi.actionSystem.DataKey;
	import com.intellij.openapi.actionSystem.DataProvider;
	import com.intellij.openapi.editor.impl.EditorHeaderComponent;
	import com.intellij.openapi.project.Project;
	import com.intellij.openapi.util.text.StringUtil;
	import com.intellij.openapi.wm.IdeFocusManager;
	import com.intellij.ui.ColoredListCellRenderer;
	import com.intellij.ui.DocumentAdapter;
	import com.intellij.ui.JBSplitter;
	import com.intellij.ui.SearchTextField;
	import com.intellij.ui.components.JBLabel;
	import com.intellij.ui.components.JBScrollPane;
	import com.intellij.ui.components.panels.NonOpaquePanel;
	import com.intellij.ui.treeStructure.treetable.TreeTable;
	import icons.AndroidIcons;
	import org.jetbrains.annotations.NonNls;
	import org.jetbrains.annotations.NotNull;
	import org.jetbrains.annotations.Nullable;

	import javax.swing.*;
	import javax.swing.event.DocumentEvent;
	import javax.swing.table.TableModel;
	import javax.swing.table.TableRowSorter;
	import javax.swing.text.BadLocationException;
	import java.awt.*;
	import java.awt.event.*;
	import java.util.List;

	public class TraceViewPanel {
	@NonNls public static DataKey<TraceViewPanel> KEY = DataKey.create("android.traceview.panel");

	// The names for the cards used in the card layout.
	// Note that these are duplicated in the layout form.
	@NonNls private static final String CARD_FIND = "FIND";
	@NonNls private static final String CARD_DEFAULT = "DEFAULT";

	/** Default name for main thread in Android apps. */
	@NonNls private static final String MAIN_THREAD_NAME = "main";

	private final Project myProject;

	private JPanel myContainer;

	private JPanel myHeaderPanel;
	private TraceViewCanvas myTraceViewCanvas;
	private TreeTable myTreeTable;

	@SuppressWarnings("UnusedDeclaration") // custom creation only
	private JPanel myDefaultHeaderPanel;
	private JComboBox myThreadCombo;
	private JComboBox myRenderClockSelectorCombo;

	@SuppressWarnings("UnusedDeclaration") // custom creation only
	private JPanel myFindPanel;
	private JPanel myFindFieldWrapper;
	private SearchTextField mySearchField;
	private JLabel myCloseLabel;
	private JBLabel myResultsLabel;
	private JLabel mySearchLabel;
	private JBSplitter mySplitter;
	private JLabel myZoomFitLabel;
	private JCheckBox myUseInclusiveTimeForColoring;

	private static final String[] ourRenderClockOptions = new String[] {
	"Wall Clock Time",
	"Thread Time",
	};

	private static final ClockType[] ourRenderClockTypes = new ClockType[] {
	ClockType.GLOBAL,
	ClockType.THREAD,
	};
	private VmTraceData myTraceData;
	private VmStatsTreeTableModel myVmStatsTreeTableModel;

	public TraceViewPanel(Project project) {
	myProject = project;

	myRenderClockSelectorCombo.setModel(new DefaultComboBoxModel(ourRenderClockOptions));
	myRenderClockSelectorCombo.setSelectedIndex(0);

	ActionListener l = new ActionListener() {
	@Override
	public void actionPerformed(ActionEvent e) {
	if (e.getSource() == myThreadCombo) {
	final ThreadInfo selectedThread = (ThreadInfo)myThreadCombo.getSelectedItem();
	myTraceViewCanvas.displayThread(selectedThread);
	myVmStatsTreeTableModel.setThread(selectedThread);
	} else if (e.getSource() == myRenderClockSelectorCombo) {
	myTraceViewCanvas.setRenderClock(getCurrentRenderClock());
	myVmStatsTreeTableModel.setClockType(getCurrentRenderClock());
	} else if (e.getSource() == myUseInclusiveTimeForColoring) {
	myTraceViewCanvas.setUseInclusiveTimeForColorAssignment(myUseInclusiveTimeForColoring.isSelected());
	}
	}
	};

	myThreadCombo.addActionListener(l);
	myRenderClockSelectorCombo.addActionListener(l);
	myUseInclusiveTimeForColoring.addActionListener(l);
	myUseInclusiveTimeForColoring.setOpaque(false);
	}

	private SearchTextField createSearchField() {
	SearchTextField stf = new SearchTextField(true);
	stf.setOpaque(false);
	stf.setEnabled(true);
	Utils.setSmallerFont(stf);

	stf.addDocumentListener(new DocumentAdapter() {
	@Override
	protected void textChanged(DocumentEvent e) {
	searchTextChanged(getText(e));
	}

	private String getText(DocumentEvent e) {
	try {
	return e.getDocument().getText(0, e.getDocument().getLength());
	}
	catch (BadLocationException e1) {
	return "";
	}
	}
	});

	JTextField editorTextField = stf.getTextEditor();
	editorTextField.setMinimumSize(new Dimension(200, -1));

	editorTextField.registerKeyboardAction(new ActionListener() {
	@Override
	public void actionPerformed(ActionEvent e) {
	closeSearchComponent();
	}
	}, KeyStroke.getKeyStroke(KeyEvent.VK_ESCAPE, 0), JComponent.WHEN_IN_FOCUSED_WINDOW);

	return stf;
	}

	private void searchTextChanged(@Nullable String pattern) {
	if (StringUtil.isEmpty(pattern)) {
	myTraceViewCanvas.setHighlightMethods(null);
	myResultsLabel.setText("");
	return;
	}

	ThreadInfo thread = (ThreadInfo)myThreadCombo.getSelectedItem();
	SearchResult results = myTraceData.searchFor(pattern, thread);
	myTraceViewCanvas.setHighlightMethods(results.getMethods());

	String result = String.format("%1$d %2$s, %3$d %4$s",
	results.getMethods().size(),
	StringUtil.pluralize("method", results.getMethods().size()),
	results.getInstances().size(),
	StringUtil.pluralize("instance", results.getInstances().size()));
	myResultsLabel.setText(result);
	}

	public void setTrace(@NotNull VmTraceData trace) {
	myTraceData = trace;

	List<ThreadInfo> threads = trace.getThreads(true);
	if (threads.isEmpty()) {
	return;
	}

	ThreadInfo defaultThread = Iterables.find(threads, new Predicate<ThreadInfo>() {
	@Override
	public boolean apply(ThreadInfo input) {
	return MAIN_THREAD_NAME.equals(input.getName());
	}
	}, threads.get(0));

	myTraceViewCanvas.setTrace(trace, defaultThread, getCurrentRenderClock());
	myThreadCombo.setModel(new DefaultComboBoxModel(threads.toArray()));
	myThreadCombo.setSelectedItem(defaultThread);
	myThreadCombo.setRenderer(new ColoredListCellRenderer() {
	@Override
	protected void customizeCellRenderer(JList list, Object value, int index, boolean selected, boolean hasFocus) {
	String name = value instanceof ThreadInfo ? ((ThreadInfo)value).getName() : value.toString();
	append(name);
	}
	});

	myThreadCombo.setEnabled(true);
	myRenderClockSelectorCombo.setEnabled(true);

	myVmStatsTreeTableModel.setTraceData(trace, defaultThread);
	myVmStatsTreeTableModel.setClockType(getCurrentRenderClock());
	myTreeTable.setModel(myVmStatsTreeTableModel);

	VmStatsTreeUtils.adjustTableColumnWidths(myTreeTable);
	VmStatsTreeUtils.setCellRenderers(myTreeTable);
	VmStatsTreeUtils.setSpeedSearch(myTreeTable);
	VmStatsTreeUtils.enableSorting(myTreeTable, myVmStatsTreeTableModel);
	}

	private ClockType getCurrentRenderClock() {
	return ourRenderClockTypes[myRenderClockSelectorCombo.getSelectedIndex()];
	}

	@NotNull
	public JComponent getComponent() {
	return myContainer;
	}

	private void createUIComponents() {
	MouseAdapter l = new MouseAdapter() {
	@Override
	public void mousePressed(MouseEvent e) {
	if (e.getSource() == myCloseLabel) {
	closeSearchComponent();
	} else if (e.getSource() == mySearchLabel) {
	showSearchComponent();
	} else if (e.getSource() == myZoomFitLabel) {
	myTraceViewCanvas.zoomFit();
	}
	}
	};

	myDefaultHeaderPanel = new EditorHeaderComponent();
	mySearchLabel = new JLabel(AllIcons.Actions.Search);
	mySearchLabel.addMouseListener(l);
	mySearchLabel.setToolTipText("Find (Ctrl + F)");
	myZoomFitLabel = new JLabel(AndroidIcons.ZoomFit);
	myZoomFitLabel.setToolTipText("Zoom Fit");
	myZoomFitLabel.addMouseListener(l);

	myFindPanel = new EditorHeaderComponent();
	myFindFieldWrapper = new NonOpaquePanel(new BorderLayout());
	mySearchField = createSearchField();
	myFindFieldWrapper.add(mySearchField);

	myCloseLabel = new JLabel(AllIcons.Actions.Cross);
	myCloseLabel.addMouseListener(l);

	myVmStatsTreeTableModel = new VmStatsTreeTableModel();
	myTreeTable = new TreeTable(myVmStatsTreeTableModel);
	myTraceViewCanvas = new TraceViewCanvasWrapper();
	JBScrollPane scrollPane = new JBScrollPane(myTreeTable);

	mySplitter = new JBSplitter(true, 0.75f);
	mySplitter.setShowDividerControls(true);
	mySplitter.setShowDividerIcon(true);
	mySplitter.setFirstComponent(myTraceViewCanvas);
	mySplitter.setSecondComponent(scrollPane);
	}

	public void showSearchComponent() {
	CardLayout layout = (CardLayout)myHeaderPanel.getLayout();
	layout.show(myHeaderPanel, CARD_FIND);
	IdeFocusManager.getInstance(myProject).requestFocus(mySearchField, true);
	}

	private void closeSearchComponent() {
	CardLayout layout = (CardLayout)myHeaderPanel.getLayout();
	layout.show(myHeaderPanel, CARD_DEFAULT);
	IdeFocusManager.getInstance(myProject).requestFocus(myTraceViewCanvas, true);
	}

	/**
	* {@link TraceViewCanvasWrapper} is a wrapper around {@link TraceViewCanvas} that also implements the {@link DataProvider} interface.
	* This allows {@link VmTraceEditorSearchAction} to identify the editor from the current context of an event.
	*/
	private class TraceViewCanvasWrapper extends TraceViewCanvas implements DataProvider {
	public TraceViewCanvasWrapper() {
	addMouseListener(new MouseAdapter() {
	@Override
	public void mouseReleased(MouseEvent e) {
	IdeFocusManager.getInstance(myProject).requestFocus(TraceViewCanvasWrapper.this, true);
	}
	});
	}

	@Nullable
	@Override
	public Object getData(@NonNls String dataId) {
	return KEY.is(dataId) ? TraceViewPanel.this : null;
	}
	}
	}