Retrieve JetClassObject element by stub in ResolveSession#getClassDescriptor()

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetObjectDeclaration.java

@@ -101,6 +101,14 @@ public class JetObjectDeclaration extends JetNamedDeclarationStub<PsiJetObjectSt
         return parent != null && parent.getNode().getElementType().equals(JetNodeTypes.CLASS_OBJECT);
     }
 
+    @Nullable
+    public JetClassObject getClassObjectElement() {
+        if (!isClassObject()) {
+            return null;
+        }
+        return (JetClassObject) getParentByStub();
+    }
+
     @Override
     public boolean hasModifier(JetModifierKeywordToken modifier) {
         JetModifierList modifierList = getModifierList();

compiler/frontend/src/org/jetbrains/jet/lang/resolve/lazy/ResolveSession.java

@@ -256,8 +256,12 @@ public class ResolveSession implements KotlinCodeAnalyzer {
     @Override
     @NotNull
     public ClassDescriptor getClassDescriptor(@NotNull JetClassOrObject classOrObject) {
-        if (classOrObject.getParent() instanceof JetClassObject) {
+        if (classOrObject instanceof JetObjectDeclaration) {
-            return getClassObjectDescriptor((JetClassObject) classOrObject.getParent());
+            JetObjectDeclaration objectDeclaration = (JetObjectDeclaration) classOrObject;
+            JetClassObject classObjectElement = objectDeclaration.getClassObjectElement();
+            if (classObjectElement != null) {
+                return getClassObjectDescriptor(classObjectElement);
+            }
         }
         JetScope resolutionScope = getScopeProvider().getResolutionScopeForDeclaration(classOrObject);
         Name name = safeNameForLazyResolve(classOrObject.getNameAsName());