Overload method AbstractTreeViewer::internalConditionalExpandToLevel

Author: fedejeanne

bundles/org.eclipse.jface/src/org/eclipse/jface/viewers/AbstractTreeViewer.java

@@ -1846,10 +1846,22 @@ protected Widget internalGetWidgetToSelect(Object elementOrTreePath) {
 	 *                             expand all levels of the tree
 	 * @param shouldChildrenExpand predicate that defines for a given widget if it
 	 *                             should be expanded.
-	 * @since 3.32
+	 * @since 3.36
 	 */
-	private void internalConditionalExpandToLevel(Widget widget, int level,
+	protected void internalConditionalExpandToLevel(Widget widget, int level,
 			Function<Widget, Boolean> shouldChildrenExpand) {
+		internalConditionalExpandToLevel(widget, level, __ -> Boolean.TRUE, shouldChildrenExpand);
+	}
+
+	/**
+	 * Same as
+	 * {@link #internalConditionalExpandToLevel(Widget, int, Function, Function)}i
+	 *
+	 * @param shouldWidgetExpand evaluates if the widget itself needs to be
+	 *                           expanded.
+	 */
+	protected void internalConditionalExpandToLevel(Widget widget, int level,
+			Function<Widget, Boolean> shouldWidgetExpand, Function<Widget, Boolean> shouldChildrenExpand) {
 		if (level == ALL_LEVELS || level > 0) {
 			Object data = widget.getData();
 			if (widget instanceof Item it && data != null && !isExpandable(it, null, data)) {
@@ -1858,6 +1870,10 @@ private void internalConditionalExpandToLevel(Widget widget, int level,
 			createChildren(widget, false);
 			// XXX for performance widget should be expanded after expanding children:
 			if (widget instanceof Item it) {
+				if (!shouldWidgetExpand.apply(widget).booleanValue()) {
+					return;
+				}
+
 				setExpanded(it, true);
 			}
 			if (level == ALL_LEVELS || level > 1) {