From 1d42e910032ccc7c2362655f3f3077fa2b55bccf Mon Sep 17 00:00:00 2001
From: Dmitry Batkovich <dmitry.batkovich@jetbrains.com>
Date: Mon, 24 Nov 2014 13:36:37 +0300
Subject: [PATCH] IDEA-132912 Toggle buttons ignore ''Show Classes'' in
 Override Methods dialog

---
 .../generation/JavaOverrideImplementMemberChooser.java | 10 ++++------
 .../src/com/intellij/ide/util/MemberChooser.java       |  2 +-
 2 files changed, 5 insertions(+), 7 deletions(-)

diff --git a/java/java-impl/src/com/intellij/codeInsight/generation/JavaOverrideImplementMemberChooser.java b/java/java-impl/src/com/intellij/codeInsight/generation/JavaOverrideImplementMemberChooser.java
index f304a06ec96c..cda4ff02cf0f 100644
--- a/java/java-impl/src/com/intellij/codeInsight/generation/JavaOverrideImplementMemberChooser.java
+++ b/java/java-impl/src/com/intellij/codeInsight/generation/JavaOverrideImplementMemberChooser.java
@@ -13,7 +13,6 @@ import com.intellij.openapi.util.NotNullLazyValue;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.infos.CandidateInfo;
 import com.intellij.psi.util.PsiUtil;
-import com.intellij.util.ArrayUtil;
 import com.intellij.util.Function;
 import com.intellij.util.containers.ContainerUtil;
 import org.jetbrains.annotations.NonNls;
@@ -131,10 +130,9 @@ public class JavaOverrideImplementMemberChooser extends MemberChooser<PsiMethodM
 
   @Override
   protected void onAlphabeticalSortingEnabled(final AnActionEvent event) {
+    mySortedByOverriding = false;
     resetElements(myToImplement || myMerge ? myAllElements : myOnlyPrimaryElements, null, true);
-    if (mySortByOverridingAction != null) {
-      mySortByOverridingAction.setSelected(event, false);
-    }
+    restoreTree();
   }
 
   @Override
@@ -221,8 +219,8 @@ public class JavaOverrideImplementMemberChooser extends MemberChooser<PsiMethodM
     @Override
     public void setSelected(AnActionEvent e, boolean state) {
       myMerge = state;
-      if (state && mySortByOverridingAction.isSelected(e)) {
-        mySortByOverridingAction.setSelected(e, false);
+      if (state && mySortedByOverriding) {
+        mySortedByOverriding = false;
       }
       resetElements(state ? myAllElements : myOnlyPrimaryElements, null, true);
       setTitle(getChooserTitle(false, myMerge));
diff --git a/platform/lang-impl/src/com/intellij/ide/util/MemberChooser.java b/platform/lang-impl/src/com/intellij/ide/util/MemberChooser.java
index c8c53db4ef65..157ba0eea4ae 100644
--- a/platform/lang-impl/src/com/intellij/ide/util/MemberChooser.java
+++ b/platform/lang-impl/src/com/intellij/ide/util/MemberChooser.java
@@ -172,7 +172,7 @@ public class MemberChooser<T extends ClassMember> extends DialogWrapper implemen
 
     defaultExpandTree();
 
-    //restoreTree();
+    restoreTree();
 
     if (myOptionControls == null) {
       myCopyJavadocCheckbox = new NonFocusableCheckBox(IdeBundle.message("checkbox.copy.javadoc"));