[patch] Improved verstion of ModuleControllerGui

ModuleControllerGui.java (text/plain), 15.28 KB, created by maciej.franek on 2015-02-16 21:01:26 UTC

(hide)

Description: Improved verstion of ModuleControllerGui

Filename:

MIME Type:

Creator: maciej.franek

Created: 2015-02-16 21:01:26 UTC

Size: 15.28 KB

patch

obsolete

>/* > * Licensed to the Apache Software Foundation (ASF) under one or more > * contributor license agreements. See the NOTICE file distributed with > * this work for additional information regarding copyright ownership. > * The ASF licenses this file to You 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 org.apache.jmeter.control.gui; > >import java.awt.Component; >import java.awt.Dimension; >import java.awt.FlowLayout; >import java.awt.event.ActionEvent; >import java.awt.event.ActionListener; >import java.awt.event.MouseAdapter; >import java.awt.event.MouseEvent; >import java.util.Collection; >import java.util.Iterator; > >import javax.swing.Box; >import javax.swing.BoxLayout; >import javax.swing.ImageIcon; >import javax.swing.JButton; >import javax.swing.JLabel; >import javax.swing.JMenu; >import javax.swing.JPanel; >import javax.swing.JPopupMenu; >import javax.swing.JTree; >import javax.swing.tree.DefaultMutableTreeNode; >import javax.swing.tree.DefaultTreeCellRenderer; >import javax.swing.tree.DefaultTreeModel; >import javax.swing.tree.TreeNode; >import javax.swing.tree.TreePath; > >import org.apache.jmeter.control.Controller; >import org.apache.jmeter.control.ModuleController; >import org.apache.jmeter.control.TestFragmentController; >import org.apache.jmeter.gui.GuiPackage; >import org.apache.jmeter.gui.action.ActionNames; >import org.apache.jmeter.gui.tree.JMeterTreeNode; >import org.apache.jmeter.gui.util.MenuFactory; >import org.apache.jmeter.testelement.TestElement; >import org.apache.jmeter.testelement.TestPlan; >import org.apache.jmeter.threads.AbstractThreadGroup; >import org.apache.jmeter.util.JMeterUtils; >import org.apache.jorphan.gui.layout.VerticalLayout; > >/** > * ModuleControllerGui provides UI for configuring ModuleController element. > * It contains filtered copy of test plan tree (called Module to run tree) > * in which target element can be specified by selecting node. > * Allowed types of elements in Module to run tree: > * - TestPlan - cannot be referenced > * - AbstractThreadGroup - cannot be referenced > * - Controller (except ModuleController) > * - TestFragmentController > * > */ >public class ModuleControllerGui extends AbstractControllerGui implements ActionListener >// implements UnsharedComponent >{ > private static final String SEPARATOR = " > "; > > private static final long serialVersionUID = 240L; > > private JMeterTreeNode selected = null; > > /** > * Model of a Module to run tree. It is a copy of a test plan tree. > * User object of each element is set to a correlated JMeterTreeNode element of a test plan tree. > */ > private final DefaultTreeModel m2runTreeModel; > > /** > * Module to run tree. Allows to select a module to be executed. Many ModuleControllers can reference > * the same element. > */ > private final JTree m2runTreeNodes; > > /** > * Used in case if referenced element does not exist in test plan tree (after removing it for example) > */ > private final JLabel warningLabel; > > /** > * Helps navigating test plan > */ > private JButton expandButton; > > /** > * Initializes the gui panel for the ModuleController instance. > */ > public ModuleControllerGui() { > m2runTreeModel = new DefaultTreeModel(new DefaultMutableTreeNode()); > m2runTreeNodes = new JTree(m2runTreeModel); > > //test element name replacer (called after selecting module to run in module controller tree) > MouseAdapter ml = new MouseAdapter() { > public void mouseClicked(MouseEvent e) { > TreePath selPath = m2runTreeNodes.getPathForLocation(e.getX(), > e.getY()); > if (selPath != null) { > Object[] nodesOnPath = selPath.getPath(); > //do not allow to reference test plan > if (nodesOnPath.length > 1) { > JMeterTreeNode tn = (JMeterTreeNode) ((DefaultMutableTreeNode)selPath.getLastPathComponent()).getUserObject(); > //prevent also changing name if thread group is selected > if(!(tn.getTestElement() instanceof AbstractThreadGroup)){ > setElementName(tn.getName()); > } > } > } > } > }; > m2runTreeNodes.addMouseListener(ml); > m2runTreeNodes.setCellRenderer(new ModuleControllerCellRenderer()); > > warningLabel = new JLabel(""); // $NON-NLS-1$ > init(); > } > > private void setElementName(String name) > { > this.namePanel.setName(name); > } > > /** {@inheritDoc}} */ > @Override > public String getLabelResource() { > return "module_controller_title"; // $NON-NLS-1$ > } > /** {@inheritDoc}} */ > @Override > public void configure(TestElement el) { > super.configure(el); > ModuleController controller = (ModuleController) el; > this.selected = controller.getSelectedNode(); > if (selected == null && controller.getNodePath() != null) { > warningLabel.setText(JMeterUtils.getResString("module_controller_warning") // $NON-NLS-1$ > + renderPath(controller.getNodePath())); > } else { > warningLabel.setText(""); // $NON-NLS-1$ > } > reinitialize(); > } > > private String renderPath(Collection<?> path) { > Iterator<?> iter = path.iterator(); > StringBuilder buf = new StringBuilder(); > boolean first = true; > while (iter.hasNext()) { > if (first) { > first = false; > iter.next(); > continue; > } > buf.append(iter.next()); > if (iter.hasNext()) { > buf.append(SEPARATOR); // $NON-NLS-1$ > } > } > return buf.toString(); > } > > /** {@inheritDoc}} */ > @Override > public TestElement createTestElement() { > ModuleController mc = new ModuleController(); > configureTestElement(mc); > if (selected != null) { > mc.setSelectedNode(selected); > } > return mc; > } > > /** {@inheritDoc}} */ > @Override > public void modifyTestElement(TestElement element) { > configureTestElement(element); > JMeterTreeNode tn = null; > DefaultMutableTreeNode lastSelected = (DefaultMutableTreeNode) this.m2runTreeNodes.getLastSelectedPathComponent(); > if(lastSelected != null && lastSelected.getUserObject() instanceof JMeterTreeNode){ > tn = (JMeterTreeNode) lastSelected.getUserObject(); > } > if(tn != null){ > selected = tn; > //prevent from selecting thread group or test plan elements > if (selected != null > && !(selected.getTestElement() instanceof AbstractThreadGroup) > && !(selected.getTestElement() instanceof TestPlan)) > { > ((ModuleController) element).setSelectedNode(selected); > } > } > } > > /** {@inheritDoc}} */ > @Override > public void clearGui() { > super.clearGui(); > > //nodes.setSelectedIndex(-1); > selected = null; > } > > > /** {@inheritDoc}} */ > @Override > public JPopupMenu createPopupMenu() { > JPopupMenu menu = new JPopupMenu(); > JMenu addMenu = MenuFactory.makeMenus( > new String[] { > MenuFactory.CONFIG_ELEMENTS, > MenuFactory.ASSERTIONS, > MenuFactory.TIMERS, > MenuFactory.LISTENERS, > }, > JMeterUtils.getResString("add"), // $NON-NLS-1$ > ActionNames.ADD); > menu.add(addMenu); > MenuFactory.addEditMenu(menu, true); > MenuFactory.addFileMenu(menu); > return menu; > } > > private void init() { > setLayout(new VerticalLayout(5, VerticalLayout.BOTH, VerticalLayout.TOP)); > setBorder(makeBorder()); > add(makeTitlePanel()); > > JPanel modulesPanel = new JPanel(); > > expandButton = new JButton(JMeterUtils.getResString("expand")); //$NON-NLS-1$ > expandButton.addActionListener(this); > modulesPanel.add(expandButton); > modulesPanel.setLayout(new BoxLayout(modulesPanel, BoxLayout.Y_AXIS));; > modulesPanel.add(Box.createRigidArea(new Dimension(0,5))); > > JLabel nodesLabel = new JLabel(JMeterUtils.getResString("module_controller_module_to_run")); // $NON-NLS-1$ > modulesPanel.add(nodesLabel); > modulesPanel.add(warningLabel); > add(modulesPanel); > > JPanel treePanel = new JPanel(); > treePanel.setLayout(new FlowLayout(FlowLayout.LEFT)); > treePanel.add(m2runTreeNodes); > add(treePanel); > } > > /** > * Recursively traverse module to run tree in order to find JMeterTreeNode element given by testPlanPath > * in a DefaultMutableTreeNode tree > * > * @param level - current search level > * @param testPlanPath - path of a test plan tree element > * @param parent - module to run tree parent element > * > * @return path of a found element > */ > private TreeNode[] findPathInTreeModel(int level, TreeNode[] testPlanPath, DefaultMutableTreeNode parent) > { > if(level >= testPlanPath.length) > return null; > int childCount = parent.getChildCount(); > JMeterTreeNode searchedTreeNode = (JMeterTreeNode) testPlanPath[level]; > > for (int i = 0; i < childCount; i++) { > DefaultMutableTreeNode child = (DefaultMutableTreeNode) parent.getChildAt(i); > JMeterTreeNode childUserObj = (JMeterTreeNode) child.getUserObject(); > > if(!childUserObj.equals(searchedTreeNode)){ > continue; > } else { > if(level == (testPlanPath.length - 1)){ > return child.getPath(); > } else { > return findPathInTreeModel(level+1, testPlanPath, child); > } > } > } > return null; > } > > /** > * Expand module to run tree to selected JMeterTreeNode and set selection path to it > * @param selected - referenced module to run > */ > private void focusSelectedOnTree(JMeterTreeNode selected) > { > TreeNode[] path = selected.getPath(); > TreeNode[] filteredPath = new TreeNode[path.length-1]; > > //ignore first element of path - WorkBench, (why WorkBench is appearing in the path ???) > for(int i = 1; i < path.length; i++){ > filteredPath[i-1] = path[i]; > } > > DefaultMutableTreeNode root = (DefaultMutableTreeNode) m2runTreeNodes.getModel().getRoot(); > //treepath of test plan tree and module to run tree cannot be compared directly - m2runTreeModel.getPathToRoot() > //custom method for finding an JMeterTreeNode element in DefaultMutableTreeNode have to be used > TreeNode[] dmtnPath = this.findPathInTreeModel(1, filteredPath, root); > if (dmtnPath != null) { > TreePath treePath = new TreePath(dmtnPath); > m2runTreeNodes.setSelectionPath(treePath); > m2runTreeNodes.scrollPathToVisible(treePath); > } > } > > private void reinitialize() { > ((DefaultMutableTreeNode) m2runTreeModel.getRoot()).removeAllChildren(); > > GuiPackage gp = GuiPackage.getInstance(); > JMeterTreeNode root; > if (gp != null) { > root = (JMeterTreeNode) GuiPackage.getInstance().getTreeModel().getRoot(); > buildTreeNodeModel(root, 0, null); > m2runTreeModel.nodeStructureChanged((TreeNode) m2runTreeModel.getRoot()); > } > if (selected != null) { > //expand Module to run tree to selected node and set selection path to it > this.focusSelectedOnTree(selected); > } > } > > /** > * Recursively build module to run tree. Only 4 types of elements are allowed to be added: > * - All controllers except ModuleController > * - TestPlan > * - TestFragmentController > * - AbstractThreadGroup > * > * @param node - element of test plan tree > * @param level - level of element in a tree > * @param parent > */ > private void buildTreeNodeModel(JMeterTreeNode node, int level, > DefaultMutableTreeNode parent) { > > if (node != null) { > for (int i = 0; i < node.getChildCount(); i++) { > JMeterTreeNode cur = (JMeterTreeNode) node.getChildAt(i); > TestElement te = cur.getTestElement(); > > if (te instanceof Controller > && !(te instanceof ModuleController) && level > 0) { > DefaultMutableTreeNode newNode = new DefaultMutableTreeNode(cur); > parent.add(newNode); > buildTreeNodeModel(cur, level + 1, newNode); > > } else if (te instanceof TestFragmentController) { > DefaultMutableTreeNode newNode = new DefaultMutableTreeNode(cur); > parent.add(newNode); > buildTreeNodeModel(cur, level + 1, newNode); > > } else if (te instanceof AbstractThreadGroup) { > DefaultMutableTreeNode newNode = new DefaultMutableTreeNode(cur); > parent.add(newNode); > buildTreeNodeModel(cur, level + 1, newNode); > } > > else if (te instanceof TestPlan) { > ((DefaultMutableTreeNode) m2runTreeModel.getRoot()) > .setUserObject(cur); > buildTreeNodeModel(cur, level, > (DefaultMutableTreeNode) m2runTreeModel.getRoot()); > } > } > } > } > > /** > * Implementation of Expand button - moves focus to a test plan tree element referenced by > * selected element in Module to run tree > */ > @Override > public void actionPerformed(ActionEvent e) { > if(e.getSource()==expandButton) { > JMeterTreeNode tn = null; > DefaultMutableTreeNode selected = (DefaultMutableTreeNode) this.m2runTreeNodes.getLastSelectedPathComponent(); > if(selected != null && selected.getUserObject() instanceof JMeterTreeNode){ > tn = (JMeterTreeNode) selected.getUserObject(); > } > if(tn != null){ > TreePath treePath = new TreePath(tn.getPath()); > //changing selection in a test plan tree > GuiPackage.getInstance().getTreeListener().getJTree() > .setSelectionPath(treePath); > //expanding tree to make referenced element visible in test plan tree > GuiPackage.getInstance().getTreeListener().getJTree() > .scrollPathToVisible(treePath); > } > } > } > > > /** > * @param selected JMeterTreeNode tree node to expand > */ > protected void expandToSelectNode(JMeterTreeNode selected) { > GuiPackage guiInstance = GuiPackage.getInstance(); > JTree jTree = guiInstance.getMainFrame().getTree(); > jTree.expandPath(new TreePath(selected.getPath())); > selected.setMarkedBySearch(true); > } > > /** > * Renderer class for printing module to run tree > * > */ > private class ModuleControllerCellRenderer extends DefaultTreeCellRenderer { > > private static final long serialVersionUID = 1129098620102526299L; > > @Override > public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, > boolean leaf, int row, boolean p_hasFocus) { > JMeterTreeNode node = (JMeterTreeNode)((DefaultMutableTreeNode) value).getUserObject(); > if(node != null){ > super.getTreeCellRendererComponent(tree, node.getName(), sel, expanded, leaf, row, > p_hasFocus); > //print same icon as in test plan tree > boolean enabled = node.isEnabled(); > ImageIcon ic = node.getIcon(enabled); > if (ic != null) { > if (enabled) { > setIcon(ic); > } else { > setDisabledIcon(ic); > } > } > } > return this; > } > } >} View Attachment As Diff View Attachment As Raw

This is ASF Bugzilla: the Apache Software Foundation bug system. In case of problems with the functioning of ASF Bugzilla, please contact bugzilla-admin@apache.org. Please Note: this e-mail address is only for reporting problems with ASF Bugzilla. Mail about any other subject will be silently ignored.

• • Home
  • | New
  • | Browse
  • | Search
  • |
    [?]
  • | Reports
  • | Help
  • | New Account
  • | Log In
    Remember [x]
  • | Forgot Password
    Login: [x]