bundles/org.simantics.browsing.ui.swt/src/org/simantics/browsing/ui/swt/TreeTruncationDecoratorFactory.java

   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
   8  *\r
   9  * Contributors:\r
  10  *     VTT Technical Research Centre of Finland - initial API and implementation\r
  11  *******************************************************************************/\r
  12 package org.simantics.browsing.ui.swt;\r
  13 \r
  14 import java.util.Collections;\r
  15 import java.util.Set;\r
  16 \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
  24 \r
  25 /**\r
  26  * A label decorator factory that shows a [children shown/total] suffix for a\r
  27  * node context when it is expanded.\r
  28  * \r
  29  * @author Tuukka Lehtonen\r
  30  */\r
  31 public class TreeTruncationDecoratorFactory {\r
  32 \r
  33     protected int         withStyle;\r
  34     protected Set<String> decoratedColumns;\r
  35 \r
  36     public TreeTruncationDecoratorFactory(int withStyle) {\r
  37         this(withStyle, Collections.singleton(ColumnKeys.SINGLE));\r
  38     }\r
  39 \r
  40     public TreeTruncationDecoratorFactory(int withStyle, Set<String> decoratedColumns) {\r
  41         this.withStyle = withStyle;\r
  42         this.decoratedColumns = decoratedColumns;\r
  43     }\r
  44 \r
  45     public LabelDecorator create(NodeQueryManager manager, NodeContext context) {\r
  46         boolean expanded = manager.query(context, BuiltinKeys.IS_EXPANDED);\r
  47         if (!expanded)\r
  48             return null;\r
  49 \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
  57                     @Override\r
  58                     public String decorateLabel(String label, String column, int itemIndex) {\r
  59                         if (decoratedColumns.contains(column))\r
  60                             return label + " [" + finalLength + "]";\r
  61                         return null;\r
  62                     }\r
  63                     @Override\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
  67                         return null;\r
  68                     }\r
  69                 };\r
  70             }\r
  71             return null;\r
  72         }\r
  73 \r
  74         return new LabelDecorator.Stub() {\r
  75             @Override\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
  79                 return null;\r
  80             }\r
  81             @SuppressWarnings("unchecked")\r
  82             @Override\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
  86                 return null;\r
  87             }\r
  88         };\r
  89     }\r
  90 \r
  91 }