1 files changed, 85 insertions, 0 deletions
diff --git a/python/src/com/jetbrains/python/hierarchy/PyHierarchyNodeDescriptor.java b/python/src/com/jetbrains/python/hierarchy/PyHierarchyNodeDescriptor.java
new file mode 100644
index 000000000000..c9720fb01af3
--- /dev/null
+++ b/python/src/com/jetbrains/python/hierarchy/PyHierarchyNodeDescriptor.java
@@ -0,0 +1,85 @@
+/*
+ * Copyright 2000-2013 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.jetbrains.python.hierarchy;
+
+import com.intellij.ide.IdeBundle;
+import com.intellij.ide.hierarchy.HierarchyNodeDescriptor;
+import com.intellij.ide.util.treeView.NodeDescriptor;
+import com.intellij.navigation.ItemPresentation;
+import com.intellij.openapi.roots.ui.util.CompositeAppearance;
+import com.intellij.openapi.util.Comparing;
+import com.intellij.psi.NavigatablePsiElement;
+import com.intellij.psi.PsiElement;
+import com.jetbrains.python.psi.PyClass;
+import com.jetbrains.python.psi.PyFunction;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+/**
+ * Created by IntelliJ IDEA.
+ * User: Alexey.Ivanov
+ * Date: Jul 31, 2009
+ * Time: 6:26:37 PM
+ */
+public class PyHierarchyNodeDescriptor extends HierarchyNodeDescriptor {
+  public PyHierarchyNodeDescriptor(final NodeDescriptor parentDescriptor, @NotNull final PsiElement element, final boolean isBase) {
+    super(element.getProject(), parentDescriptor, element, isBase);
+  }
+
+  @Nullable
+  public PsiElement getPsiElement() {
+    return myElement;
+  }
+
+  public boolean isValid() {
+    return myElement != null && myElement.isValid();
+  }
+
+  @Override
+  public boolean update() {
+    boolean changes = super.update();
+    final CompositeAppearance oldText = myHighlightedText;
+
+    myHighlightedText = new CompositeAppearance();
+
+    NavigatablePsiElement element = (NavigatablePsiElement)myElement;
+    if (element == null) {
+      final String invalidPrefix = IdeBundle.message("node.hierarchy.invalid");
+      if (!myHighlightedText.getText().startsWith(invalidPrefix)) {
+        myHighlightedText.getBeginning().addText(invalidPrefix, HierarchyNodeDescriptor.getInvalidPrefixAttributes());
+      }
+      return true;
+    }
+
+    final ItemPresentation presentation = element.getPresentation();
+    if (presentation != null) {
+      if (element instanceof PyFunction) {
+        final PyClass cls = ((PyFunction)element).getContainingClass();
+        if (cls != null) {
+          myHighlightedText.getEnding().addText(cls.getName() + ".");
+        }
+      }
+      myHighlightedText.getEnding().addText(presentation.getPresentableText());
+      myHighlightedText.getEnding().addText(" " + presentation.getLocationString(), HierarchyNodeDescriptor.getPackageNameAttributes());
+    }
+    myName = myHighlightedText.getText();
+
+    if (!Comparing.equal(myHighlightedText, oldText)) {
+      changes = true;
+    }
+    return changes;
+  }
+}