diff --git a/platform/platform-api/src/com/intellij/ui/treeStructure/actions/CollapseAllAction.java b/platform/platform-api/src/com/intellij/ui/treeStructure/actions/CollapseAllAction.java
index 2d6a1ea1a34e..825651fb9d8b 100644
--- a/platform/platform-api/src/com/intellij/ui/treeStructure/actions/CollapseAllAction.java
+++ b/platform/platform-api/src/com/intellij/ui/treeStructure/actions/CollapseAllAction.java
@@ -28,16 +28,22 @@ import javax.swing.*;
 public class CollapseAllAction extends AnAction implements DumbAware {
 
   protected JTree myTree;
+  protected final int collapseToLevel;
 
   public CollapseAllAction(JTree tree) {
+    this(tree, 0);
+  }
+
+  public CollapseAllAction(JTree tree, int collapseToLevel) {
     super(IdeBundle.messagePointer("action.CollapseAllAction.text.collapse.all"), () -> "", AllIcons.Actions.Collapseall);
+    this.collapseToLevel = collapseToLevel;
     myTree = tree;
   }
 
   @Override
   public void actionPerformed(@NotNull AnActionEvent e) {
     int row = getTree().getRowCount() - 1;
-    while (row >= 0) {
+    while (row >= collapseToLevel) {
       getTree().collapseRow(row);
       row--;
     }
diff --git a/platform/platform-api/src/com/intellij/ui/treeStructure/treetable/TreeTableModelAdapter.java b/platform/platform-api/src/com/intellij/ui/treeStructure/treetable/TreeTableModelAdapter.java
index 2a43d21b7258..46f162adb8c5 100644
--- a/platform/platform-api/src/com/intellij/ui/treeStructure/treetable/TreeTableModelAdapter.java
+++ b/platform/platform-api/src/com/intellij/ui/treeStructure/treetable/TreeTableModelAdapter.java
@@ -90,6 +90,7 @@ public class TreeTableModelAdapter extends AbstractTableModel {
         delayedFireTableDataChanged();
       }
     });
+
   }
 
   // Wrappers, implementing TableModel interface.
@@ -161,4 +162,5 @@ public class TreeTableModelAdapter extends AbstractTableModel {
       }
     }
   }
+
 }