From 32c695966ec66a34352c5d8ea78e1384a991c37b Mon Sep 17 00:00:00 2001
From: Dmitry Batrak <Dmitry.Batrak@jetbrains.com>
Date: Fri, 29 Jul 2016 13:34:23 +0300
Subject: [PATCH] EA-78487 - assert: EditorCoordinateMapper.offsetToVisualLine

---
 .../codeInsight/folding/impl/CollapseBlockHandler.java | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/java/java-impl/src/com/intellij/codeInsight/folding/impl/CollapseBlockHandler.java b/java/java-impl/src/com/intellij/codeInsight/folding/impl/CollapseBlockHandler.java
index 4a39b0738fca..8ec485e06248 100644
--- a/java/java-impl/src/com/intellij/codeInsight/folding/impl/CollapseBlockHandler.java
+++ b/java/java-impl/src/com/intellij/codeInsight/folding/impl/CollapseBlockHandler.java
@@ -34,6 +34,7 @@ public class CollapseBlockHandler implements CodeInsightActionHandler {
 
   @Override
   public void invoke(@NotNull final Project project, @NotNull final Editor editor, @NotNull final PsiFile file) {
+    int[] targetCaretOffset = {-1};
     editor.getFoldingModel().runBatchFoldingOperation(() -> {
       final EditorFoldingInfo info = EditorFoldingInfo.get(editor);
       FoldingModelEx model = (FoldingModelEx) editor.getFoldingModel();
@@ -52,7 +53,7 @@ public class CollapseBlockHandler implements CodeInsightActionHandler {
         if (existing != null) {
           if (existing.isExpanded()) {
             existing.setExpanded(false);
-            editor.getCaretModel().moveToOffset(existing.getEndOffset());
+            targetCaretOffset[0] = existing.getEndOffset();
             return;
           }
           previous = existing;
@@ -68,9 +69,7 @@ public class CollapseBlockHandler implements CodeInsightActionHandler {
             info.removeRegion(myPrevious);
             model.removeFoldRegion(myPrevious);
           }
-          int offset = block.getTextRange().getEndOffset() < editor.getCaretModel().getOffset() ?
-              start : end;
-          editor.getCaretModel().moveToOffset(offset);
+          targetCaretOffset[0] = block.getTextRange().getEndOffset() < editor.getCaretModel().getOffset() ? start : end;
           return;
         } else break;
       }
@@ -80,9 +79,10 @@ public class CollapseBlockHandler implements CodeInsightActionHandler {
           info.removeRegion(myPrevious);
           model.removeFoldRegion(myPrevious);
         }
-        editor.getCaretModel().moveToOffset(previous.getEndOffset());
+        targetCaretOffset[0] = previous.getEndOffset();
       }
     });
+    if (targetCaretOffset[0] >= 0) editor.getCaretModel().moveToOffset(targetCaretOffset[0]);
   }
 
   @Override