1 /*******************************************************************************
\r
2 * Copyright (c) 2007, 2012 Association for Decentralized Information Management
\r
3 * in Industry THTH ry.
\r
4 * All rights reserved. This program and the accompanying materials
\r
5 * are made available under the terms of the Eclipse Public License v1.0
\r
6 * which accompanies this distribution, and is available at
\r
7 * http://www.eclipse.org/legal/epl-v10.html
\r
10 * VTT Technical Research Centre of Finland - initial API and implementation
\r
11 *******************************************************************************/
\r
12 package org.simantics.browsing.ui.swt;
\r
14 import java.util.Collections;
\r
15 import java.util.Set;
\r
17 import org.eclipse.jface.resource.FontDescriptor;
\r
18 import org.simantics.browsing.ui.BuiltinKeys;
\r
19 import org.simantics.browsing.ui.NodeContext;
\r
20 import org.simantics.browsing.ui.NodeQueryManager;
\r
21 import org.simantics.browsing.ui.common.ColumnKeys;
\r
22 import org.simantics.browsing.ui.content.LabelDecorator;
\r
23 import org.simantics.browsing.ui.content.PrunedChildrenResult;
\r
26 * A label decorator factory that shows a [children shown/total] suffix for a
\r
27 * node context when it is expanded.
\r
29 * @author Tuukka Lehtonen
\r
31 public class TreeTruncationDecoratorFactory {
\r
33 protected int withStyle;
\r
34 protected Set<String> decoratedColumns;
\r
36 public TreeTruncationDecoratorFactory(int withStyle) {
\r
37 this(withStyle, Collections.singleton(ColumnKeys.SINGLE));
\r
40 public TreeTruncationDecoratorFactory(int withStyle, Set<String> decoratedColumns) {
\r
41 this.withStyle = withStyle;
\r
42 this.decoratedColumns = decoratedColumns;
\r
45 public LabelDecorator create(NodeQueryManager manager, NodeContext context) {
\r
46 boolean expanded = manager.query(context, BuiltinKeys.IS_EXPANDED);
\r
50 PrunedChildrenResult prunedChildren = manager.query(context, BuiltinKeys.PRUNED_CHILDREN);
\r
51 NodeContext[] finalChildren = manager.query(context, BuiltinKeys.FINAL_CHILDREN);
\r
52 final int prunedLength = prunedChildren.getPrunedChildren().length;
\r
53 final int finalLength = finalChildren.length;
\r
54 if (prunedLength == finalLength) {
\r
55 if (finalLength >= 10) {
\r
56 return new LabelDecorator.Stub() {
\r
58 public String decorateLabel(String label, String column, int itemIndex) {
\r
59 if (decoratedColumns.contains(column))
\r
60 return label + " [" + finalLength + "]";
\r
64 public <F> F decorateFont(F font, String column, int itemIndex) {
\r
65 //if (decoratedColumns.contains(column))
\r
66 // return (F) ((FontDescriptor) font).withStyle(withStyle);
\r
74 return new LabelDecorator.Stub() {
\r
76 public String decorateLabel(String label, String column, int itemIndex) {
\r
77 if (decoratedColumns.contains(column))
\r
78 return label + " [showing " + finalLength + "/" + prunedLength + "]";
\r
81 @SuppressWarnings("unchecked")
\r
83 public <F> F decorateFont(F font, String column, int itemIndex) {
\r
84 if (decoratedColumns.contains(column))
\r
85 return (F) ((FontDescriptor) font).withStyle(withStyle);
\r