refactoring/impl/com/intellij/refactoring/ui/MemberSelectionTable.java

   1 /*
   2  * Created by IntelliJ IDEA.
   3  * User: dsl
   4  * Date: 17.06.2002
   5  * Time: 16:35:43
   6  * To change template for new class use
   7  * Code Style | Class Templates options (Tools | IDE Options).
   8  */
   9 package com.intellij.refactoring.ui;
  10
  11 import com.intellij.psi.PsiMember;
  12 import com.intellij.psi.PsiMethod;
  13 import com.intellij.psi.PsiModifier;
  14 import com.intellij.psi.PsiModifierList;
  15 import com.intellij.refactoring.classMembers.MemberInfoModel;
  16 import com.intellij.refactoring.util.classMembers.MemberInfo;
  17 import com.intellij.ui.RowIcon;
  18 import com.intellij.util.IconUtil;
  19 import com.intellij.util.VisibilityIcons;
  20
  21 import javax.swing.*;
  22 import java.util.List;
  23
  24 public class MemberSelectionTable extends AbstractMemberSelectionTable<PsiMember, MemberInfo> {
  25
  26   public MemberSelectionTable(final List<MemberInfo> memberInfos, String abstractColumnHeader) {
  27     this(memberInfos, null, abstractColumnHeader);
  28   }
  29
  30   public MemberSelectionTable(final List<MemberInfo> memberInfos, MemberInfoModel<PsiMember, MemberInfo> memberInfoModel, String abstractColumnHeader) {
  31     super(memberInfos, memberInfoModel, abstractColumnHeader);
  32   }
  33
  34   @Override
  35   protected Object getAbstractColumnValue(MemberInfo memberInfo) {
  36     if (!(memberInfo.getMember() instanceof PsiMethod)) return null;
  37     if (memberInfo.isStatic()) return null;
  38
  39     PsiMethod method = (PsiMethod)memberInfo.getMember();
  40     if (method.hasModifierProperty(PsiModifier.ABSTRACT)) {
  41       final Boolean fixedAbstract = myMemberInfoModel.isFixedAbstract(memberInfo);
  42       if (fixedAbstract != null) return fixedAbstract;
  43     }
  44
  45     if (!myMemberInfoModel.isAbstractEnabled(memberInfo)) {
  46       return myMemberInfoModel.isAbstractWhenDisabled(memberInfo);
  47     }
  48     else {
  49       return memberInfo.isToAbstract() ? Boolean.TRUE : Boolean.FALSE;
  50     }
  51   }
  52
  53   @Override
  54   protected boolean isAbstractColumnEditable(int rowIndex) {
  55     MemberInfo info = myMemberInfos.get(rowIndex);
  56     if (!(info.getMember() instanceof PsiMethod)) return false;
  57
  58     PsiMethod method = (PsiMethod)info.getMember();
  59     if (method.hasModifierProperty(PsiModifier.ABSTRACT)) {
  60       if (myMemberInfoModel.isFixedAbstract(info) != null) {
  61         return false;
  62       }
  63     }
  64
  65     return info.isChecked() && myMemberInfoModel.isAbstractEnabled(info);
  66   }
  67
  68
  69   @Override
  70   protected void setVisibilityIcon(MemberInfo memberInfo, RowIcon icon) {
  71     PsiMember member = memberInfo.getMember();
  72     PsiModifierList modifiers = member != null ? member.getModifierList() : null;
  73     if (modifiers != null) {
  74       VisibilityIcons.setVisibilityIcon(modifiers, icon);
  75     }
  76     else {
  77       icon.setIcon(IconUtil.getEmptyIcon(true), VISIBILITY_ICON_POSITION);
  78     }
  79   }
  80
  81   @Override
  82   protected Icon getOverrideIcon(MemberInfo memberInfo) {
  83     PsiMember member = memberInfo.getMember();
  84     Icon overrideIcon = MemberSelectionTable.EMPTY_OVERRIDE_ICON;
  85     if (member instanceof PsiMethod) {
  86       if (Boolean.TRUE.equals(memberInfo.getOverrides())) {
  87         overrideIcon = MemberSelectionTable.OVERRIDING_METHOD_ICON;
  88       }
  89       else if (Boolean.FALSE.equals(memberInfo.getOverrides())) {
  90         overrideIcon = MemberSelectionTable.IMPLEMENTING_METHOD_ICON;
  91       }
  92       else {
  93         overrideIcon = MemberSelectionTable.EMPTY_OVERRIDE_ICON;
  94       }
  95     }
  96     return overrideIcon;
  97   }
  98 }