From ca1069c3e8c87117e153d14547a1843965fac8d3 Mon Sep 17 00:00:00 2001
From: Julia Beliaeva <Julia.Beliaeva@jetbrains.com>
Date: Wed, 7 Sep 2016 23:40:56 +0300
Subject: [PATCH] [vcs-log] fix branch filter

---
 .../vcs/log/graph/impl/facade/PermanentGraphImpl.java      | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/platform/vcs-log/graph/src/com/intellij/vcs/log/graph/impl/facade/PermanentGraphImpl.java b/platform/vcs-log/graph/src/com/intellij/vcs/log/graph/impl/facade/PermanentGraphImpl.java
index e533ef838857..26f6fee0c55c 100644
--- a/platform/vcs-log/graph/src/com/intellij/vcs/log/graph/impl/facade/PermanentGraphImpl.java
+++ b/platform/vcs-log/graph/src/com/intellij/vcs/log/graph/impl/facade/PermanentGraphImpl.java
@@ -108,9 +108,14 @@ public class PermanentGraphImpl<CommitId> implements PermanentGraph<CommitId>, P
       baseController = new BekBaseController(this, myBekIntMap.get());
     }
 
-    LinearGraphController controller;
+    // TODO this code is unclear and obviously needs some refactoring
+    // I'll leave it for later to reorganize, and add some duplication for now, in order just to fix stuff
+    CascadeController controller;
     if (matchingCommits != null) {
       controller = new FilteredController(baseController, this, myPermanentCommitsInfo.convertToNodeIds(matchingCommits));
+      if (visibleHeads != null) {
+        controller = new BranchFilterController(controller, this, myPermanentCommitsInfo.convertToNodeIds(visibleHeads, true));
+      }
     }
     else if (sortType == SortType.LinearBek) {
       if (visibleHeads != null) {