Skip to main content
summaryrefslogtreecommitdiffstats
blob: d7c102e441c4a98e34941d1f656e26ce9b12fe9f (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
/*******************************************************************************
 * Copyright (c) 2004, 2007 Boeing.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Boeing - initial API and implementation
 *******************************************************************************/
package org.eclipse.osee.framework.ui.plugin.util;

import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.jface.viewers.Viewer;
import org.eclipse.osee.framework.plugin.core.util.Jobs;
import org.eclipse.osee.framework.ui.swt.IContentProviderRunnable;
import org.eclipse.osee.framework.ui.swt.ITreeNode;
import org.eclipse.osee.framework.ui.swt.TreeNode;

/**
 * @author Robert A. Fisher
 */
public class JobbedNode extends TreeNode {
   private static final long serialVersionUID = 2285210369260889786L;
   private static final JobbedNode LOADING_NODE = new JobbedNode("Loading...");
   private static final JobbedNode CANCELLED_NODE = new JobbedNode("Cancelled");
   private static final Object[] LOADING = new Object[] {LOADING_NODE};
   private static final Object[] CANCELLED = new Object[] {CANCELLED_NODE};
   private final Viewer viewer;
   private final IContentProviderRunnable providerRunnable;
   private boolean expand = false;

   private JobbedNode(String title) {
      super(null, null, title);
      this.viewer = null;
      this.providerRunnable = null;
   }

   public JobbedNode(Object backingData, Viewer viewer, IContentProviderRunnable providerRunnable) {
      this(backingData, viewer, providerRunnable, null);
   }

   public JobbedNode(Object backingData, Viewer viewer, IContentProviderRunnable providerRunnable, JobbedNode parent) {
      super(parent, null, backingData);
      this.viewer = viewer;
      this.providerRunnable = providerRunnable;
   }

   @Override
   public Object[] getChildren() {
      if (children == null && providerRunnable != null) {
         children = LOADING;
         Jobs.startJob(new GetChildrenJob(null, viewer, this, providerRunnable));
      }
      return children;
   }

   public void cancelled() {
      cancelled(CANCELLED);
   }

   public void cancelled(Exception ex) {
      cancelled(new Object[] {ex});
   }

   private void cancelled(Object[] children) {
      this.children = children;
      Displays.ensureInDisplayThread(new Runnable() {
         public void run() {
            viewer.refresh();
         }
      });
   }

   @Override
   public void setChildren(Object[] objChildren) {
      super.setChildren(objChildren);

      if (expand) {
         if (viewer instanceof TreeViewer) {
            final TreeViewer treeViewer = (TreeViewer) viewer;

            Displays.ensureInDisplayThread(new Runnable() {

               public void run() {
                  treeViewer.refresh();
                  treeViewer.expandToLevel(2);
               }
            });
         }
      }
   }

   public void refresh() {
      refresh(false);
   }

   public void refresh(boolean expand) {
      this.expand = expand;

      if (children != null) for (Object obj : children)
         if (obj instanceof JobbedNode) ((JobbedNode) obj).refresh(true);

      children = null;
   }

   @Override
   protected ITreeNode getChild(Object backingData) {
      return new JobbedNode(backingData, viewer, providerRunnable, this);
   }

}

Back to the top