source/com/intellij/ide/hierarchy/JavaHierarchyUtil.java

   1 package com.intellij.ide.hierarchy;
   2
   3 import com.intellij.ide.util.treeView.AlphaComparator;
   4 import com.intellij.ide.util.treeView.NodeDescriptor;
   5 import com.intellij.ide.util.treeView.SourceComparator;
   6 import com.intellij.openapi.project.Project;
   7 import com.intellij.psi.PsiClass;
   8 import com.intellij.psi.PsiFile;
   9 import com.intellij.psi.PsiJavaFile;
  10 import org.jetbrains.annotations.Nullable;
  11
  12 import java.util.Comparator;
  13
  14 /**
  15  * @author yole
  16  */
  17 public class JavaHierarchyUtil {
  18   private JavaHierarchyUtil() {
  19   }
  20
  21   @Nullable
  22   public static String getPackageName(final PsiClass psiClass) {
  23     final PsiFile file = psiClass.getContainingFile();
  24     if (file instanceof PsiJavaFile){
  25       return ((PsiJavaFile)file).getPackageName();
  26     }
  27     else{
  28       return null;
  29     }
  30   }
  31
  32   public static Comparator<NodeDescriptor> getComparator(Project project) {
  33     if (HierarchyBrowserManager.getInstance(project).getState().SORT_ALPHABETICALLY) {
  34       return AlphaComparator.INSTANCE;
  35     }
  36     else {
  37       return SourceComparator.INSTANCE;
  38     }
  39   }
  40 }