private static void replaceChildren(
final DefaultMutableTreeNode node, final Collection<? extends ElementNode> arrayList) {
node.removeAllChildren();
for (ElementNode child : arrayList) {
node.add(child);
}
}
@Override
public String convertValueToText(
Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) {
String s = super.convertValueToText(value, selected, expanded, leaf, row, hasFocus);
String newProp = jEdit.getProperty(HIGHLIGHT_PROP);
if (newProp == null || newProp.isEmpty()) return s;
DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
while (node != null && !(node.getUserObject() instanceof HyperSearchOperationNode)) {
node = (DefaultMutableTreeNode) node.getParent();
}
if (node == null) return s;
if (!newProp.equals(prop)) {
prop = newProp;
Font f = (resultTree != null) ? resultTree.getFont() : UIManager.getFont("Tree.font");
styleTag = HtmlUtilities.style2html(prop, f);
}
SearchMatcher matcher = ((HyperSearchOperationNode) node.getUserObject()).getSearchMatcher();
int i = s.indexOf(": ");
if (i > 0) i += 2;
else i = 0;
Match m;
List<Integer> matches = new ArrayList<Integer>();
while ((m = matcher.nextMatch(s.substring(i), true, true, true, false)) != null) {
matches.add(i + m.start);
matches.add(i + m.end);
i += m.end;
}
return HtmlUtilities.highlightString(s, styleTag, matches);
}
/**
* Recursively rebuild the tree nodes.
*
* @param root The full abstract path to the root saved layout directory.
* @param local The current directory relative to root (null if none).
* @param tnode The current parent tree node.
*/
private void rebuildTreeModel(Path root, Path local, DefaultMutableTreeNode tnode) {
TreeSet<Path> subdirs = new TreeSet<Path>();
TreeSet<String> layouts = new TreeSet<String>();
{
Path current = new Path(root, local);
File files[] = current.toFile().listFiles();
if (files != null) {
int wk;
for (wk = 0; wk < files.length; wk++) {
String name = files[wk].getName();
if (files[wk].isDirectory()) subdirs.add(new Path(local, name));
else if (files[wk].isFile()) layouts.add(name);
}
}
}
for (Path subdir : subdirs) {
TreeData data = new TreeData(subdir);
DefaultMutableTreeNode child = new DefaultMutableTreeNode(data, true);
tnode.add(child);
rebuildTreeModel(root, subdir, child);
}
for (String lname : layouts) {
TreeData data = new TreeData(new Path(local, lname), lname);
DefaultMutableTreeNode child = new DefaultMutableTreeNode(data, false);
tnode.add(child);
}
}
/**
* Adds nodes to the specified parent node recurrsively.
*
* @param parent_node the parent node.
* @param list the list of child node names.
*/
protected void addNode(DefaultMutableTreeNode parent_node, Vector list) {
SortableArray array = null;
if (mode == DATE_ORIENTED && parent_node.getLevel() <= 2) {
array = new Array(list.size());
for (int i = 0; i < list.size(); i++)
((Array) array).set(i, Integer.parseInt((String) list.elementAt(i)));
} else {
array = new StringArray(list.size());
for (int i = 0; i < list.size(); i++)
((StringArray) array).set(i, (String) list.elementAt(i));
}
ArrayIndex index = array.sortAscendant();
for (int i = 0; i < array.getArraySize(); i++) {
String name = (String) list.elementAt(index.get(i));
// Converts 1...12 to January...December.
if (mode == DATE_ORIENTED && parent_node.getLevel() == 1) {
int month = Integer.parseInt(name);
name = JulianDay.getFullSpellMonthString(month);
}
DefaultMutableTreeNode node = new DefaultMutableTreeNode(name);
parent_node.add(node);
}
}
@Nullable
private static String getActionId(DefaultMutableTreeNode node) {
return (String)
(node.getUserObject() instanceof String
? node.getUserObject()
: node.getUserObject() instanceof Pair ? ((Pair) node.getUserObject()).first : null);
}
public MsnTreeTest() {
String[] tab = {"hello", "test", "blabla"};
container = getContentPane();
container.setLayout(null);
eleve = new DefaultMutableTreeNode("MSN");
worker = new DefaultMutableTreeNode("Worker");
prof = new DefaultMutableTreeNode("Profs");
for (int i = 0; i < tab.length; i++) {
worker.add(new DefaultMutableTreeNode(tab[i]));
prof.add(new DefaultMutableTreeNode(tab[i]));
}
// worker.add(new DefaultMutableTreeNode("hello world2"));
eleve.add(worker);
eleve.add(prof);
tree = new JTree(eleve);
scroll = new JScrollPane(tree);
scroll.setBounds(10, 10, 100, 100);
container.add(scroll);
setSize(300, 300);
setLocation(200, 200);
setDefaultCloseOperation(EXIT_ON_CLOSE);
setVisible(true);
// worker.add(n);
}
private static boolean isToolbarAction(DefaultMutableTreeNode node) {
return node.getParent() != null
&& ((DefaultMutableTreeNode) node.getParent()).getUserObject() instanceof Group
&& ((Group) ((DefaultMutableTreeNode) node.getParent()).getUserObject())
.getName()
.equals(ActionsTreeUtil.MAIN_TOOLBAR);
}
private void goToSelectedNode(int mode) {
TreePath path = resultTree.getSelectionPath();
if (path == null) return;
DefaultMutableTreeNode node = (DefaultMutableTreeNode) path.getLastPathComponent();
Object value = node.getUserObject();
// do nothing if clicked "foo (showing n occurrences in m files)"
if (node.getParent() != resultTreeRoot && value instanceof HyperSearchNode) {
HyperSearchNode n = (HyperSearchNode) value;
Buffer buffer = n.getBuffer(view);
if (buffer == null) return;
EditPane pane;
switch (mode) {
case M_OPEN:
pane = view.goToBuffer(buffer);
break;
case M_OPEN_NEW_VIEW:
pane = jEdit.newView(view, buffer, false).getEditPane();
break;
case M_OPEN_NEW_PLAIN_VIEW:
pane = jEdit.newView(view, buffer, true).getEditPane();
break;
case M_OPEN_NEW_SPLIT:
pane = view.splitHorizontally();
break;
default:
throw new IllegalArgumentException("Bad mode: " + mode);
}
n.goTo(pane);
}
} // }}}
@Override
public void actionPerformed(ActionEvent evt) {
JCheckBoxMenuItem menuItem = (JCheckBoxMenuItem) evt.getSource();
boolean curState = menuItem.isSelected();
TreePath path = resultTree.getSelectionPath();
DefaultMutableTreeNode operNode = (DefaultMutableTreeNode) path.getLastPathComponent();
HyperSearchOperationNode operNodeObj = (HyperSearchOperationNode) operNode.getUserObject();
if (curState) operNodeObj.cacheResultNodes(operNode);
operNode.removeAllChildren();
if (curState) {
Exception excp = null;
try {
operNodeObj.insertTreeNodes(resultTree, operNode);
} catch (Exception ex) {
operNodeObj.restoreFlatNodes(resultTree, operNode);
menuItem.setSelected(false);
excp = ex;
} finally {
((DefaultTreeModel) resultTree.getModel()).nodeStructureChanged(operNode);
expandAllNodes(operNode);
resultTree.scrollPathToVisible(new TreePath(operNode.getPath()));
}
if (excp != null) throw new RuntimeException(excp);
} else operNodeObj.restoreFlatNodes(resultTree, operNode);
operNodeObj.setTreeViewDisplayed(menuItem.isSelected());
}
/**
* This gets called whenever one of the ConfigElements we are editing adds a new property value.
*/
public void propertyValueAdded(ConfigElementEvent evt) {
ConfigElement src = (ConfigElement) evt.getSource();
int idx = evt.getIndex();
PropertyDefinition prop_def = src.getDefinition().getPropertyDefinition(evt.getProperty());
DefaultMutableTreeNode elt_node = getNodeFor(src);
// Get the node containing the property description under the source
// ConfigElement node
for (Enumeration e = elt_node.children(); e.hasMoreElements(); ) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) e.nextElement();
if (node.getUserObject().equals(prop_def)) {
// The newly inserted property value must be added as a child to
// this node
if (prop_def.getType() != ConfigElement.class) {
DefaultMutableTreeNode new_child = new DefaultMutableTreeNode(evt.getValue());
insertNodeInto(new_child, node, idx);
} else {
// Embedded elements are handled specially in that all of their
// respective child properties and such also need to be added to
// the tree at this time.
addEmbeddedElement(node, (ConfigElement) evt.getValue(), idx);
}
}
}
}
private void doSave() {
ObjectOutputStream objectStream = getObjectOutputStream();
if (objectStream != null) {
try {
System.out.println("Saving " + selectedChildrenPaths.size() + " Selected Generations...");
for (int i = 0; i < selectedChildrenPaths.size(); i++) {
// Get the userObject at the supplied path
Object selectedPath =
((TreePath) selectedChildrenPaths.elementAt(i)).getLastPathComponent();
DefaultMutableTreeNode selectedNode = (DefaultMutableTreeNode) selectedPath;
objectStream.writeObject(selectedNode.getUserObject());
}
objectStream.close();
System.out.println("Save completed successfully.");
} catch (IOException e) {
System.err.println(e);
}
} else {
System.out.println("Save Selected Files has been aborted!");
}
}
/** Gets the value for the given node at the given column. */
public Object getValueAt(Object node, int col) {
DefaultMutableTreeNode tree_node = (DefaultMutableTreeNode) node;
Object value = tree_node.getUserObject();
switch (col) {
// Name
case 0:
return value;
// Value
case 1:
if (value instanceof ConfigElement) {
return null;
} else if (value instanceof PropertyDefinition) {
// Only provide comma delimited editing for simple types
if (((PropertyDefinition) value).getType() != ConfigElement.class) {
StringBuffer buffer = new StringBuffer();
for (Enumeration e = tree_node.children(); e.hasMoreElements(); ) {
DefaultMutableTreeNode child_node = (DefaultMutableTreeNode) e.nextElement();
buffer.append(child_node.getUserObject());
if (e.hasMoreElements()) {
buffer.append(", ");
}
}
return buffer.toString();
}
return null;
}
return value;
}
return null;
}
/**
* Called when the selection changed in the tree. Loads the selected certificate.
*
* @param e the event
*/
private void valueChangedPerformed(TreeSelectionEvent e) {
Object o = e.getNewLeadSelectionPath().getLastPathComponent();
if (o instanceof DefaultMutableTreeNode) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) o;
infoTextPane.setText(toString(node.getUserObject()));
}
}
public DefaultMutableTreeNode findNode(DefaultMutableTreeNode parent, String match) {
if (parent == null) return null;
// Commented by Balan on 15/03/03
// String treename = (String)parent.getUserObject ();
// Comment Ends
// Added by Balan on 15/03/03
String treename = "" + ((Hashtable) parent.getUserObject()).get("TREE-NAME");
// Add Ends
if (treename != null) {
if (treename.equals(match)) return parent;
}
if (frame.model.isLeaf(parent)) return null;
Enumeration en = parent.children();
if ((en == null) || (!en.hasMoreElements())) return null;
for (; en.hasMoreElements(); ) {
DefaultMutableTreeNode child = (DefaultMutableTreeNode) en.nextElement();
DefaultMutableTreeNode returnNode = findNode(child, match);
if (returnNode != null) return returnNode;
}
return null;
}
/** This gets called whenever one of the ConfigElements we are modeling has changed its name. */
public void nameChanged(ConfigElementEvent evt) {
ConfigElement src = (ConfigElement) evt.getSource();
DefaultMutableTreeNode root = (DefaultMutableTreeNode) getRoot();
DefaultMutableTreeNode name_node = (DefaultMutableTreeNode) root.getChildAt(0);
name_node.setUserObject(src.getName());
fireTreeNodesChanged(
this, new Object[] {getPathToRoot(root)}, new int[] {0}, new Object[] {name_node});
}
public static void sort(final DefaultMutableTreeNode node, final Comparator comparator) {
final List<TreeNode> children = childrenToArray(node);
Collections.sort(children, comparator);
node.removeAllChildren();
addChildrenTo(node, children);
for (int i = 0; i < node.getChildCount(); i++) {
sort((DefaultMutableTreeNode) node.getChildAt(i), comparator);
}
}
@Nullable
public static TreeNode findNodeWithObject(
final Object object, final TreeModel model, final Object parent) {
for (int i = 0; i < model.getChildCount(parent); i++) {
final DefaultMutableTreeNode childNode = (DefaultMutableTreeNode) model.getChild(parent, i);
if (childNode.getUserObject().equals(object)) return childNode;
}
return null;
}
private MutableTreeNode populatePlot(Plot p) throws SQLException {
DefaultMutableTreeNode tree = new DefaultMutableTreeNode(p);
// tree.add(populateAttributes(p));
Statement stmt = db.statement();
for (Tree t : p.loadTrees(stmt)) {
tree.add(populateTree(t));
}
stmt.close();
return tree;
}
private MutableTreeNode populateTree(Tree t) throws SQLException {
DefaultMutableTreeNode tree = new DefaultMutableTreeNode(t);
// tree.add(populateAttributes(t));
Statement stmt = db.statement();
for (Cavity c : t.loadCavities(stmt)) {
tree.add(populateCavity(c));
}
stmt.close();
return tree;
}
private MutableTreeNode populateNest(Nest n) throws SQLException {
DefaultMutableTreeNode tree = new DefaultMutableTreeNode(n);
// tree.add(populateAttributes(n));
Statement stmt = db.statement();
for (Visit v : n.loadVisits(stmt)) {
tree.add(populateVisit(v));
}
stmt.close();
return tree;
}
private MutableTreeNode populateCavity(Cavity c) throws SQLException {
DefaultMutableTreeNode tree = new DefaultMutableTreeNode(c);
// tree.add(populateAttributes(c));
Statement stmt = db.statement();
for (Nest n : c.loadNests(stmt)) {
tree.add(populateNest(n));
}
stmt.close();
return tree;
}
@Override
public void actionPerformed(ActionEvent evt) {
TreePath path = resultTree.getSelectionPath();
DefaultMutableTreeNode operNode = (DefaultMutableTreeNode) path.getLastPathComponent();
for (Enumeration e = operNode.children(); e.hasMoreElements(); ) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) e.nextElement();
resultTree.collapsePath(new TreePath(node.getPath()));
}
resultTree.scrollPathToVisible(new TreePath(operNode.getPath()));
}
/**
* This gets called whenever one of the ConfigElements we are editing removes a property value.
*/
public void propertyValueRemoved(ConfigElementEvent evt) {
ConfigElement src = (ConfigElement) evt.getSource();
int idx = evt.getIndex();
PropertyDefinition prop_def = src.getDefinition().getPropertyDefinition(evt.getProperty());
DefaultMutableTreeNode elt_node = getNodeFor(src);
// Get the node containing the property description under the source
// ConfigElement node
for (Enumeration e = elt_node.children(); e.hasMoreElements(); ) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) e.nextElement();
if (node.getUserObject().equals(prop_def)) {
// The newly removed property value must be a child to this node
System.out.println("Removing child " + idx + " from node: " + node.getUserObject());
DefaultMutableTreeNode child = (DefaultMutableTreeNode) node.getChildAt(idx);
// If the child is an embedded element, stop listening to it
if (child.getUserObject() instanceof ConfigElement) {
ConfigElement removed_elt = (ConfigElement) child.getUserObject();
removed_elt.removeConfigElementListener(this);
}
// Physically remove the child from the tree
removeNodeFromParent(child);
}
}
}
@Nullable
private static TreePath findNodePath(MavenArchetype object, TreeModel model, Object parent) {
for (int i = 0; i < model.getChildCount(parent); i++) {
DefaultMutableTreeNode each = (DefaultMutableTreeNode) model.getChild(parent, i);
if (each.getUserObject().equals(object)) return new TreePath(each.getPath());
TreePath result = findNodePath(object, model, each);
if (result != null) return result;
}
return null;
}
public void editSelected() {
TreePath selected = tree.getSelectionPath();
if (selected != null) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) selected.getLastPathComponent();
Object obj = node.getUserObject();
if (obj instanceof CavityDBObject) {
CavityDBObject dbObj = (CavityDBObject) obj;
new ObjectEditingFrame(new ObjectEditingPanel(dbObj));
}
}
}
/** Clears the data in the model. */
private void clear() {
// Stop listening to all config elements in the tree
List nodes = getNodesOfClass(ConfigElement.class);
for (Iterator itr = nodes.iterator(); itr.hasNext(); ) {
DefaultMutableTreeNode node = (DefaultMutableTreeNode) itr.next();
ConfigElement elt = (ConfigElement) node.getUserObject();
elt.removeConfigElementListener(mElementListener);
}
// Clear out all the old nodes from the tree.
((DefaultMutableTreeNode) getRoot()).removeAllChildren();
}
@Override
public void actionPerformed(ActionEvent evt) {
TreePath path = resultTree.getSelectionPath();
DefaultMutableTreeNode operNode = (DefaultMutableTreeNode) path.getLastPathComponent();
HyperSearchFolderNode nodeObj = (HyperSearchFolderNode) operNode.getUserObject();
String glob = "*";
SearchFileSet dirList = SearchAndReplace.getSearchFileSet();
if (dirList instanceof DirectoryListSet) glob = ((DirectoryListSet) dirList).getFileFilter();
SearchAndReplace.setSearchFileSet(
new DirectoryListSet(nodeObj.getNodeFile().getAbsolutePath(), glob, true));
SearchDialog.showSearchDialog(view, null, SearchDialog.DIRECTORY);
}
protected void restoreTree() {
Pair<ElementNode, List<ElementNode>> selection = storeSelection();
DefaultMutableTreeNode root = getRootNode();
if (!myShowClasses || myContainerNodes.isEmpty()) {
List<ParentNode> otherObjects = new ArrayList<ParentNode>();
Enumeration<ParentNode> children = getRootNodeChildren();
ParentNode newRoot =
new ParentNode(
null, new MemberChooserObjectBase(getAllContainersNodeName()), new Ref<Integer>(0));
while (children.hasMoreElements()) {
final ParentNode nextElement = children.nextElement();
if (nextElement instanceof ContainerNode) {
final ContainerNode containerNode = (ContainerNode) nextElement;
Enumeration<MemberNode> memberNodes = containerNode.children();
List<MemberNode> memberNodesList = new ArrayList<MemberNode>();
while (memberNodes.hasMoreElements()) {
memberNodesList.add(memberNodes.nextElement());
}
for (MemberNode memberNode : memberNodesList) {
newRoot.add(memberNode);
}
} else {
otherObjects.add(nextElement);
}
}
replaceChildren(root, otherObjects);
sortNode(newRoot, myComparator);
if (newRoot.children().hasMoreElements()) root.add(newRoot);
} else {
Enumeration<ParentNode> children = getRootNodeChildren();
while (children.hasMoreElements()) {
ParentNode allClassesNode = children.nextElement();
Enumeration<MemberNode> memberNodes = allClassesNode.children();
ArrayList<MemberNode> arrayList = new ArrayList<MemberNode>();
while (memberNodes.hasMoreElements()) {
arrayList.add(memberNodes.nextElement());
}
Collections.sort(arrayList, myComparator);
for (MemberNode memberNode : arrayList) {
myNodeToParentMap.get(memberNode).add(memberNode);
}
}
replaceChildren(root, myContainerNodes);
}
myTreeModel.nodeStructureChanged(root);
defaultExpandTree();
restoreSelection(selection);
}
protected boolean doSetIcon(
DefaultMutableTreeNode node, @Nullable String path, Component component) {
if (StringUtil.isNotEmpty(path) && !new File(path).isFile()) {
Messages.showErrorDialog(
component,
IdeBundle.message("error.file.not.found.message", path),
IdeBundle.message("title.choose.action.icon"));
return false;
}
String actionId = getActionId(node);
if (actionId == null) return false;
final AnAction action = ActionManager.getInstance().getAction(actionId);
if (action != null && action.getTemplatePresentation() != null) {
if (StringUtil.isNotEmpty(path)) {
Image image = null;
try {
image =
ImageLoader.loadFromStream(
VfsUtil.convertToURL(VfsUtil.pathToUrl(path.replace(File.separatorChar, '/')))
.openStream());
} catch (IOException e) {
LOG.debug(e);
}
Icon icon = new File(path).exists() ? IconLoader.getIcon(image) : null;
if (icon != null) {
if (icon.getIconWidth() > EmptyIcon.ICON_18.getIconWidth()
|| icon.getIconHeight() > EmptyIcon.ICON_18.getIconHeight()) {
Messages.showErrorDialog(
component,
IdeBundle.message("custom.icon.validation.message"),
IdeBundle.message("title.choose.action.icon"));
return false;
}
node.setUserObject(Pair.create(actionId, icon));
mySelectedSchema.addIconCustomization(actionId, path);
}
} else {
node.setUserObject(Pair.create(actionId, null));
mySelectedSchema.removeIconCustomization(actionId);
final DefaultMutableTreeNode nodeOnToolbar = findNodeOnToolbar(actionId);
if (nodeOnToolbar != null) {
editToolbarIcon(actionId, nodeOnToolbar);
node.setUserObject(nodeOnToolbar.getUserObject());
}
}
return true;
}
return false;
}
public void fillOptions(@NotNull ColorAndFontOptions options) {
DefaultMutableTreeNode root = new DefaultMutableTreeNode();
for (EditorSchemeAttributeDescriptor description : getOrderedDescriptors(options)) {
if (!description.getGroup().equals(myCategoryName)) continue;
List<String> path = extractPath(description);
if (path != null && path.size() > 1) {
MyTreeNode groupNode = ensureGroup(root, path, 0);
groupNode.add(new MyTreeNode(description, path.get(path.size() - 1)));
} else {
root.add(new MyTreeNode(description));
}
}
myTreeModel.setRoot(root);
}
