Examples with KtElement org.jetbrains.kotlin.psi.KtElement used on opensource projects

Example 1 with KtElement

use of org.jetbrains.kotlin.psi.KtElement in project kotlin by JetBrains.

the class KotlinComponentUnwrapper method doUnwrap.

@Override
protected void doUnwrap(PsiElement element, Context context) throws IncorrectOperationException {
    KtElement targetElement = (KtElement) element;
    KtExpression expressionToUnwrap = getExpressionToUnwrap(targetElement);
    assert expressionToUnwrap != null;
    KtElement enclosingElement = getEnclosingElement(targetElement);
    context.extractFromExpression(expressionToUnwrap, enclosingElement);
    context.delete(enclosingElement);
}

Example 2 with KtElement

use of org.jetbrains.kotlin.psi.KtElement in project kotlin by JetBrains.

the class CheckerTestUtil method getDebugInfoDiagnostics.

@SuppressWarnings("TestOnlyProblems")
@NotNull
private static List<ActualDiagnostic> getDebugInfoDiagnostics(@NotNull PsiElement root, @NotNull BindingContext bindingContext, final boolean markDynamicCalls, @Nullable final List<DeclarationDescriptor> dynamicCallDescriptors, @Nullable final String platform) {
    final List<ActualDiagnostic> debugAnnotations = new ArrayList<ActualDiagnostic>();
    DebugInfoUtil.markDebugAnnotations(root, bindingContext, new DebugInfoUtil.DebugInfoReporter() {

        @Override
        public void reportElementWithErrorType(@NotNull KtReferenceExpression expression) {
            newDiagnostic(expression, DebugInfoDiagnosticFactory.ELEMENT_WITH_ERROR_TYPE);
        }

        @Override
        public void reportMissingUnresolved(@NotNull KtReferenceExpression expression) {
            newDiagnostic(expression, DebugInfoDiagnosticFactory.MISSING_UNRESOLVED);
        }

        @Override
        public void reportUnresolvedWithTarget(@NotNull KtReferenceExpression expression, @NotNull String target) {
            newDiagnostic(expression, DebugInfoDiagnosticFactory.UNRESOLVED_WITH_TARGET);
        }

        @Override
        public void reportDynamicCall(@NotNull KtElement element, DeclarationDescriptor declarationDescriptor) {
            if (dynamicCallDescriptors != null) {
                dynamicCallDescriptors.add(declarationDescriptor);
            }
            if (markDynamicCalls) {
                newDiagnostic(element, DebugInfoDiagnosticFactory.DYNAMIC);
            }
        }

        private void newDiagnostic(KtElement element, DebugInfoDiagnosticFactory factory) {
            debugAnnotations.add(new ActualDiagnostic(new DebugInfoDiagnostic(element, factory), platform));
        }
    });
    //noinspection unchecked
    for (Pair<? extends WritableSlice<? extends KtExpression, ?>, DebugInfoDiagnosticFactory> factory : Arrays.asList(TuplesKt.to(BindingContext.SMARTCAST, DebugInfoDiagnosticFactory.SMARTCAST), TuplesKt.to(BindingContext.IMPLICIT_RECEIVER_SMARTCAST, DebugInfoDiagnosticFactory.IMPLICIT_RECEIVER_SMARTCAST), TuplesKt.to(BindingContext.SMARTCAST_NULL, DebugInfoDiagnosticFactory.CONSTANT), TuplesKt.to(BindingContext.LEAKING_THIS, DebugInfoDiagnosticFactory.LEAKING_THIS), TuplesKt.to(BindingContext.IMPLICIT_EXHAUSTIVE_WHEN, DebugInfoDiagnosticFactory.IMPLICIT_EXHAUSTIVE))) {
        for (KtExpression expression : bindingContext.getSliceContents(factory.getFirst()).keySet()) {
            if (PsiTreeUtil.isAncestor(root, expression, false)) {
                debugAnnotations.add(new ActualDiagnostic(new DebugInfoDiagnostic(expression, factory.getSecond()), platform));
            }
        }
    }
    return debugAnnotations;
}

Example 3 with KtElement

use of org.jetbrains.kotlin.psi.KtElement in project kotlin by JetBrains.

the class AnalyzingUtils method formDebugNameForBindingTrace.

// --------------------------------------------------------------------------------------------------------------------------
public static String formDebugNameForBindingTrace(@NotNull String debugName, @Nullable Object resolutionSubjectForMessage) {
    if (WRITE_DEBUG_TRACE_NAMES) {
        StringBuilder debugInfo = new StringBuilder(debugName);
        if (resolutionSubjectForMessage instanceof KtElement) {
            KtElement element = (KtElement) resolutionSubjectForMessage;
            debugInfo.append(" ").append(DebugTextUtilKt.getDebugText(element));
            //debugInfo.append(" in ").append(element.getContainingFile().getName());
            debugInfo.append(" in ").append(element.getContainingKtFile().getName()).append(" ").append(element.getTextOffset());
        } else if (resolutionSubjectForMessage != null) {
            debugInfo.append(" ").append(resolutionSubjectForMessage);
        }
        return debugInfo.toString();
    }
    return "";
}

Example 4 with KtElement

use of org.jetbrains.kotlin.psi.KtElement in project kotlin by JetBrains.

the class CFGraphToDotFilePrinter method dumpDot.

public static void dumpDot(File file, Collection<Pseudocode> pseudocodes) throws FileNotFoundException {
    File target = KotlinTestUtils.replaceExtension(file, "dot");
    PrintStream out = new PrintStream(target);
    out.println("digraph " + FileUtil.getNameWithoutExtension(file) + " {");
    int[] count = new int[1];
    Map<Instruction, String> nodeToName = new HashMap<Instruction, String>();
    for (Pseudocode pseudocode : pseudocodes) {
        dumpNodes(pseudocode.getInstructionsIncludingDeadCode(), out, count, nodeToName, Sets.newHashSet(pseudocode.getInstructions()));
    }
    int i = 0;
    for (Pseudocode pseudocode : pseudocodes) {
        String label;
        KtElement correspondingElement = pseudocode.getCorrespondingElement();
        if (correspondingElement instanceof KtNamedDeclaration) {
            KtNamedDeclaration namedDeclaration = (KtNamedDeclaration) correspondingElement;
            label = namedDeclaration.getName();
        } else {
            label = "anonymous_" + i;
        }
        out.println("subgraph cluster_" + i + " {\n" + "label=\"" + label + "\";\n" + "color=blue;\n");
        dumpEdges(pseudocode.getInstructionsIncludingDeadCode(), out, count, nodeToName);
        out.println("}");
        i++;
    }
    out.println("}");
    out.close();
}

Example 5 with KtElement

use of org.jetbrains.kotlin.psi.KtElement in project kotlin by JetBrains.

the class KotlinDescriptorIconProvider method getIcon.

@NotNull
public static Icon getIcon(@NotNull DeclarationDescriptor descriptor, @Nullable PsiElement declaration, @Iconable.IconFlags int flags) {
    if (declaration != null && !(declaration instanceof KtElement)) {
        return declaration.getIcon(flags);
    }
    Icon result = getBaseIcon(descriptor);
    if ((flags & Iconable.ICON_FLAG_VISIBILITY) > 0) {
        RowIcon rowIcon = new RowIcon(2);
        rowIcon.setIcon(result, 0);
        rowIcon.setIcon(getVisibilityIcon(descriptor), 1);
        result = rowIcon;
    }
    return result;
}