Overload method AbstractTreeViewer::internalConditionalExpandToLevel

Add function to determine if the widget itself must be expanded.

Author: fedejeanne

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

@@ -1850,6 +1850,19 @@ protected Widget internalGetWidgetToSelect(Object elementOrTreePath) {
 	 */
 	private 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.
+	 * @since 3.36
+	 */
+	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 +1871,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) {