java/java-impl/src/com/intellij/packageDependencies/ui/PackageNode.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.packageDependencies.ui;
  17
  18 import com.intellij.cyclicDependencies.ui.CyclicDependenciesPanel;
  19 import com.intellij.openapi.util.Comparing;
  20 import com.intellij.psi.PsiElement;
  21 import com.intellij.psi.PsiFile;
  22 import com.intellij.psi.PsiJavaFile;
  23 import com.intellij.psi.PsiPackage;
  24 import com.intellij.util.Icons;
  25
  26 import javax.swing.*;
  27 import java.util.Map;
  28 import java.util.Set;
  29
  30 public class PackageNode extends PackageDependenciesNode {
  31
  32   private String myPackageName;
  33   private final String myPackageQName;
  34   private final PsiPackage myPackage;
  35
  36
  37   public PackageNode(PsiPackage aPackage, boolean showFQName) {
  38     myPackage = aPackage;
  39     myPackageName = showFQName ? aPackage.getQualifiedName() : aPackage.getName();
  40     if (myPackageName == null || myPackageName.length() == 0) {
  41       myPackageName = CyclicDependenciesPanel.DEFAULT_PACKAGE_ABBREVIATION;
  42     }
  43     String packageQName = aPackage.getQualifiedName();
  44     if (packageQName.length() == 0) {
  45       packageQName = null;
  46     }
  47     myPackageQName = packageQName;
  48   }
  49
  50   public void fillFiles(Set<PsiFile> set, boolean recursively) {
  51     super.fillFiles(set, recursively);
  52     int count = getChildCount();
  53     for (int i = 0; i < count; i++) {
  54       PackageDependenciesNode child = (PackageDependenciesNode)getChildAt(i);
  55       if (child instanceof FileNode || recursively) {
  56         child.fillFiles(set, true);
  57       }
  58     }
  59   }
  60
  61   public String toString() {
  62     return myPackageName;
  63   }
  64
  65   public void setPackageName(final String packageName) {
  66     myPackageName = packageName;
  67   }
  68
  69   public String getPackageQName() {
  70     return myPackageQName;
  71   }
  72
  73   public PsiElement getPsiElement() {
  74     return myPackage;
  75   }
  76
  77   public int getWeight() {
  78     return 3;
  79   }
  80
  81   public boolean equals(Object o) {
  82     if (isEquals()){
  83       return super.equals(o);
  84     }
  85     if (this == o) return true;
  86     if (!(o instanceof PackageNode)) return false;
  87
  88     final PackageNode packageNode = (PackageNode)o;
  89
  90     if (!myPackageName.equals(packageNode.myPackageName)) return false;
  91     if (myPackageQName != null ? !myPackageQName.equals(packageNode.myPackageQName) : packageNode.myPackageQName != null) return false;
  92
  93     return true;
  94   }
  95
  96   public int hashCode() {
  97     int result;
  98     result = myPackageName.hashCode();
  99     result = 29 * result + (myPackageQName != null ? myPackageQName.hashCode() : 0);
 100     return result;
 101   }
 102
 103   public Icon getOpenIcon() {
 104     return Icons.PACKAGE_OPEN_ICON;
 105   }
 106
 107   public Icon getClosedIcon() {
 108     return Icons.PACKAGE_ICON;
 109   }
 110
 111
 112   public boolean isValid() {
 113     return myPackage != null && myPackage.isValid();
 114   }
 115
 116   @Override
 117   public boolean canSelectInLeftTree(final Map<PsiFile, Set<PsiFile>> deps) {
 118     Set<PsiFile> files = deps.keySet();
 119     String packageName = myPackage.getQualifiedName();
 120     for (PsiFile file : files) {
 121       if (file instanceof PsiJavaFile && Comparing.equal(packageName, ((PsiJavaFile)file).getPackageName())) {
 122         return true;
 123       }
 124     }
 125     return false;
 126   }
 127 }