123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366 |
- /*
- * Copyright (c) 2006-2015 DMDirc Developers
- *
- * Permission is hereby granted, free of charge, to any person obtaining a copy
- * of this software and associated documentation files (the "Software"), to deal
- * in the Software without restriction, including without limitation the rights
- * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- * copies of the Software, and to permit persons to whom the Software is
- * furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in
- * all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- * SOFTWARE.
- */
-
- package com.dmdirc.addons.ui_swing.framemanager.tree;
-
- import com.dmdirc.addons.ui_swing.EdtHandlerInvocation;
- import com.dmdirc.addons.ui_swing.SwingController;
- import com.dmdirc.addons.ui_swing.SwingWindowFactory;
- import com.dmdirc.addons.ui_swing.UIUtilities;
- import com.dmdirc.addons.ui_swing.components.IconManager;
- import com.dmdirc.addons.ui_swing.components.TreeScroller;
- import com.dmdirc.addons.ui_swing.components.frames.TextFrame;
- import com.dmdirc.addons.ui_swing.events.SwingEventBus;
- import com.dmdirc.addons.ui_swing.events.SwingWindowAddedEvent;
- import com.dmdirc.addons.ui_swing.events.SwingWindowDeletedEvent;
- import com.dmdirc.addons.ui_swing.events.SwingWindowSelectedEvent;
- import com.dmdirc.addons.ui_swing.framemanager.FrameManager;
- import com.dmdirc.addons.ui_swing.interfaces.ActiveFrameManager;
- import com.dmdirc.config.GlobalConfig;
- import com.dmdirc.events.FrameIconChangedEvent;
- import com.dmdirc.events.UnreadStatusChangedEvent;
- import com.dmdirc.interfaces.EventBus;
- import com.dmdirc.interfaces.WindowModel;
- import com.dmdirc.interfaces.config.AggregateConfigProvider;
- import com.dmdirc.interfaces.config.ConfigChangeListener;
- import com.dmdirc.plugins.PluginDomain;
- import com.dmdirc.ui.WindowManager;
- import com.dmdirc.ui.messages.ColourManager;
- import com.dmdirc.util.LogUtils;
- import java.awt.Color;
- import java.awt.Rectangle;
- import java.awt.event.MouseEvent;
- import java.io.Serializable;
- import java.util.Collection;
- import java.util.HashMap;
- import java.util.Map;
- import javax.inject.Inject;
- import javax.swing.JComponent;
- import javax.swing.JScrollPane;
- import javax.swing.JTree;
- import javax.swing.ScrollPaneConstants;
- import javax.swing.SwingUtilities;
- import javax.swing.tree.DefaultMutableTreeNode;
- import javax.swing.tree.DefaultTreeModel;
- import javax.swing.tree.MutableTreeNode;
- import javax.swing.tree.TreeNode;
- import javax.swing.tree.TreePath;
- import net.engio.mbassy.listener.Handler;
- import net.engio.mbassy.listener.Invoke;
- import net.miginfocom.swing.MigLayout;
- import org.slf4j.Logger;
- import org.slf4j.LoggerFactory;
-
- /**
- * Manages open windows in the application in a tree style view.
- */
- public class TreeFrameManager implements FrameManager, Serializable, ConfigChangeListener {
-
- private static final Logger LOG = LoggerFactory.getLogger(TreeFrameManager.class);
- /** Serial version UID. */
- private static final long serialVersionUID = 5;
- /** node storage, used for adding and deleting nodes correctly. */
- private final Map<TextFrame, TreeViewNode> nodes;
- /** Configuration manager. */
- private final AggregateConfigProvider config;
- /** Colour manager. */
- private final ColourManager colourManager;
- /** Factory to use to retrieve swing windows. */
- private final SwingWindowFactory windowFactory;
- /** Window manage. */
- private final WindowManager windowManager;
- /** Active frame manager. */
- private final ActiveFrameManager activeFrameManager;
- /** The event bus to post errors to. */
- private final EventBus eventBus;
- /** Swing event bus. */
- private final SwingEventBus swingEventBus;
- /** Icon manager. */
- private final IconManager iconManager;
- /** display tree. */
- private Tree tree;
- /** data model. */
- private TreeViewModel model;
- /** Tree scroller. */
- private TreeScroller scroller;
-
- @Inject
- public TreeFrameManager(final WindowManager windowManager,
- @GlobalConfig final AggregateConfigProvider globalConfig,
- @GlobalConfig final ColourManager colourManager,
- final ActiveFrameManager activeFrameManager,
- final SwingWindowFactory windowFactory,
- @PluginDomain(SwingController.class) final String domain,
- final EventBus eventBus,
- final SwingEventBus swingEventBus,
- final IconManager iconManager) {
- this.windowFactory = windowFactory;
- this.windowManager = windowManager;
- this.nodes = new HashMap<>();
- this.config = globalConfig;
- this.colourManager = colourManager;
- this.activeFrameManager = activeFrameManager;
- this.eventBus = eventBus;
- this.swingEventBus = swingEventBus;
- this.iconManager = iconManager;
-
- UIUtilities.invokeLater(() -> {
- model = new TreeViewModel(config, new TreeViewNode(null, null));
- tree = new Tree(this, model, swingEventBus, globalConfig, domain);
- tree.setCellRenderer(
- new TreeViewTreeCellRenderer(config, colourManager, this));
- tree.setVisible(true);
-
- config.addChangeListener("treeview", this);
- config.addChangeListener("ui", "sortrootwindows", this);
- config.addChangeListener("ui", "sortchildwindows", this);
- config.addChangeListener("ui", "backgroundcolour", this);
- config.addChangeListener("ui", "foregroundcolour", this);
- });
- }
-
- @Override
- public boolean canPositionVertically() {
- return true;
- }
-
- @Override
- public boolean canPositionHorizontally() {
- return false;
- }
-
- @Override
- public void setParent(final JComponent parent) {
- SwingUtilities.invokeLater(() -> {
- final JScrollPane scrollPane = new JScrollPane(tree);
- scrollPane.setAutoscrolls(true);
- scrollPane.setHorizontalScrollBarPolicy(
- ScrollPaneConstants.HORIZONTAL_SCROLLBAR_NEVER);
-
- parent.setVisible(false);
- parent.setLayout(new MigLayout("ins 0, fill"));
- parent.add(scrollPane, "grow");
- parent.setFocusable(false);
- parent.setVisible(true);
-
- setColours();
-
- eventBus.subscribe(this);
- swingEventBus.subscribe(this);
- redoTreeView();
- });
- }
-
- @Handler
- public void doAddWindow(final SwingWindowAddedEvent event) {
- final TextFrame parent = event.getParentWindow().orElse(null);
- final TextFrame window = event.getChildWindow();
- if (nodes.containsKey(window)) {
- return;
- }
- if (parent == null) {
- addWindow(model.getRootNode(), window);
- } else {
- addWindow(nodes.get(parent), window);
- }
- }
-
- @Handler
- public void doDeleteWindow(final SwingWindowDeletedEvent event) {
- final TextFrame window = event.getChildWindow();
- UIUtilities.invokeAndWait(() -> {
- if (nodes.get(window) == null) {
- return;
- }
- final DefaultMutableTreeNode node = nodes.get(window);
- if (node.getLevel() == 0) {
- LOG.warn(LogUtils.USER_ERROR, "delServer triggered for root node {}",
- node, new IllegalArgumentException());
- } else {
- model.removeNodeFromParent(nodes.get(window));
- }
- synchronized (nodes) {
- eventBus.unsubscribe(nodes.get(window).getLabel());
- nodes.remove(window);
- }
- });
- }
-
- /**
- * Adds a window to the frame container.
- *
- * @param parent Parent node
- * @param window Window to add
- */
- public void addWindow(final MutableTreeNode parent, final TextFrame window) {
- UIUtilities.invokeAndWait(() -> {
- final NodeLabel label = new NodeLabel(window, iconManager, getForegroundColour());
- eventBus.subscribe(label);
- swingEventBus.subscribe(label);
- final TreeViewNode node = new TreeViewNode(label, window);
- synchronized (nodes) {
- nodes.put(window, node);
- }
- if (parent == null) {
- model.insertNodeInto(node, model.getRootNode());
- } else {
- model.insertNodeInto(node, parent);
- }
- tree.expandPath(new TreePath(node.getPath()).getParentPath());
- final Rectangle view = tree.getRowBounds(tree.getRowForPath(new TreePath(node.
- getPath())));
- if (view != null) {
- tree.scrollRectToVisible(new Rectangle(0, (int) view.getY(), 0, 0));
- }
-
- refreshNodeLabel(window, node);
- node.getLabel().iconChanged(new FrameIconChangedEvent(window.getContainer(),
- window.getContainer().getIcon()));
- });
- }
-
- /**
- * Returns the tree for this frame manager.
- *
- * @return Tree for the manager
- */
- public JTree getTree() {
- return tree;
- }
-
- /**
- * Checks for and sets a rollover node.
- *
- * @param event event to check
- */
- protected void checkRollover(final MouseEvent event) {
- NodeLabel node = null;
-
- if (event != null && tree.getNodeForLocation(event.getX(), event.getY()) != null) {
- node = tree.getNodeForLocation(event.getX(), event.getY()).getLabel();
- }
-
- synchronized (nodes) {
- for (TreeViewNode treeNode : nodes.values()) {
- final NodeLabel label = treeNode.getLabel();
- label.setRollover(label == node);
- }
- }
- tree.repaint();
- }
-
- /** Sets treeview colours. */
- private void setColours() {
- final Color foregroundColour = getForegroundColour();
- tree.setForeground(foregroundColour);
- tree.setBackground(UIUtilities.convertColour(colourManager.getColourFromString(
- config.getOptionString("treeview", "backgroundcolour", "ui",
- "backgroundcolour"), null)));
- tree.repaint();
-
- nodes.entrySet().forEach(pair -> {
- pair.getValue().getLabel().setDefaultForegroundColour(foregroundColour);
- refreshNodeLabel(pair.getKey(), pair.getValue());
- });
- }
-
- private Color getForegroundColour() {
- return UIUtilities.convertColour(colourManager.getColourFromString(
- config.getOptionString("treeview", "foregroundcolour", "ui",
- "foregroundcolour"), null));
- }
-
- private void refreshNodeLabel(TextFrame window, TreeViewNode node) {
- node.getLabel().unreadStatusChanged(new UnreadStatusChangedEvent(
- window.getContainer(),
- window.getContainer().getUnreadStatusManager(),
- window.getContainer().getUnreadStatusManager().getNotificationColour(),
- window.getContainer().getUnreadStatusManager().getUnreadLines()));
- }
-
- @Override
- public void configChanged(final String domain, final String key) {
- if ("sortrootwindows".equals(key) || "sortchildwindows".equals(key)) {
- redoTreeView();
- } else {
- setColours();
- }
- }
-
- /**
- * Starts the tree from scratch taking into account new sort orders.
- */
- private void redoTreeView() {
- UIUtilities.invokeLater(() -> {
- ((DefaultTreeModel) tree.getModel()).setRoot(null);
- ((DefaultTreeModel) tree.getModel()).setRoot(new TreeViewNode(null, null));
- if (scroller != null) {
- scroller.unregister();
- }
- scroller = new TreeTreeScroller(swingEventBus, tree);
-
- for (WindowModel window : windowManager.getRootWindows()) {
- addWindow(null, windowFactory.getSwingWindow(window));
- final Collection<WindowModel> childWindows = windowManager.getChildren(window);
- for (WindowModel childWindow : childWindows) {
- addWindow(nodes.get(windowFactory.getSwingWindow(window)),
- windowFactory.getSwingWindow(childWindow));
- }
- }
-
- if (activeFrameManager.getActiveFrame() != null) {
- selectionChanged(new SwingWindowSelectedEvent(activeFrameManager.getActiveFrame()));
- }
- });
- }
-
- @Handler(invocation = EdtHandlerInvocation.class)
- public void selectionChanged(final SwingWindowSelectedEvent event) {
- if (event.getWindow().isPresent()) {
- UIUtilities.invokeLater(() -> {
- final TreeNode[] treePath = ((DefaultTreeModel) tree.getModel())
- .getPathToRoot(nodes.get(event.getWindow().get()));
- if (treePath != null && treePath.length > 0) {
- final TreePath path = new TreePath(treePath);
- tree.setTreePath(path);
- tree.scrollPathToVisible(path);
- tree.repaint();
- }
- });
- }
- }
-
- @Handler(invocation = EdtHandlerInvocation.class, delivery = Invoke.Asynchronously)
- public void unreadStatusChanged(final UnreadStatusChangedEvent event) {
- synchronized (nodes) {
- final TreeViewNode node = nodes.get(windowFactory.getSwingWindow(event.getSource()));
- if (node != null) {
- final NodeLabel label = node.getLabel();
- if (label != null) {
- label.unreadStatusChanged(event);
- tree.repaint();
- }
- }
- }
- }
-
- }
|