/*
* 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.gui.action;
import java.awt.event.ActionEvent;
import java.util.Enumeration;
import java.util.HashSet;
import java.util.Set;
import javax.swing.JTree;
import javax.swing.tree.TreeNode;
import javax.swing.tree.TreePath;
import org.apache.jmeter.gui.GuiPackage;
import org.apache.jmeter.gui.tree.JMeterTreeListener;
import org.apache.jmeter.gui.tree.JMeterTreeNode;
/**
* Processes the collapse and expand of a tree branch
*/
public class CollapseExpandTreeBranch extends AbstractAction {
private static final Set<String> commands = new HashSet<>();
static {
commands.add(ActionNames.COLLAPSE);
commands.add(ActionNames.EXPAND);
}
/**
* Constructor
*/
public CollapseExpandTreeBranch() {}
/**
* Gets the ActionNames attribute of the CollapseExpandTreeBranch object.
*
* @return the ActionNames value
*/
@Override
public Set<String> getActionNames() {
return commands;
}
/**
* This method performs the actual command processing.
*
* @param e the generic UI action event
*/
@Override
public void doAction(ActionEvent e) {
JMeterTreeListener treeListener = GuiPackage.getInstance().getTreeListener();
JTree jTree = GuiPackage.getInstance().getMainFrame().getTree();
JMeterTreeNode[] selectedNodes = treeListener.getSelectedNodes();
for (JMeterTreeNode currentNode : selectedNodes) {
if(!currentNode.isLeaf()) {
TreeNode[] nodes = GuiPackage.getInstance().getTreeModel().getPathToRoot(currentNode);
TreePath path = new TreePath(nodes);
boolean collapse = ActionNames.COLLAPSE.equals(e.getActionCommand());
expandCollapseNode(jTree, path, collapse);
}
}
}
private void expandCollapseNode(JTree jTree, TreePath parent, boolean collapse) {
TreeNode node = (TreeNode) parent.getLastPathComponent();
if (node.isLeaf()) {
return;
}
Enumeration<?> children = node.children();
while(children.hasMoreElements()) {
TreeNode child = (TreeNode) children.nextElement();
TreePath path = parent.pathByAddingChild(child);
expandCollapseNode(jTree, path, collapse);
}
if(collapse) {
jTree.collapsePath(parent);
}
else {
jTree.expandPath(parent);
}
}
}