[FIR] Add default visitors and transformers with useful part of visited hierarchy

2019-09-17 14:58:53 +03:00
parent e70c1d6959
commit 61ae98df40
15 changed files with 283 additions and 267 deletions
@@ -0,0 +1,89 @@
+/*
+ * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.fir.visitors
+
+import org.jetbrains.kotlin.fir.declarations.FirDeclaration
+import org.jetbrains.kotlin.fir.declarations.FirEnumEntry
+import org.jetbrains.kotlin.fir.expressions.*
+import org.jetbrains.kotlin.fir.types.*
+
+abstract class FirDefaultTransformer<D> : FirTransformer<D>() {
+    override fun transformDelegatedTypeRef(delegatedTypeRef: FirDelegatedTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRef(delegatedTypeRef, data)
+    }
+
+    override fun transformImplicitTypeRef(implicitTypeRef: FirImplicitTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRef(implicitTypeRef, data)
+    }
+
+    override fun transformResolvedTypeRef(resolvedTypeRef: FirResolvedTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRef(resolvedTypeRef, data)
+    }
+
+    override fun transformErrorTypeRef(errorTypeRef: FirErrorTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformResolvedTypeRef(errorTypeRef, data)
+    }
+
+    override fun transformResolvedFunctionTypeRef(resolvedFunctionTypeRef: FirResolvedFunctionTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformResolvedTypeRef(resolvedFunctionTypeRef, data)
+    }
+
+    override fun transformTypeRefWithNullability(typeRefWithNullability: FirTypeRefWithNullability, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRef(typeRefWithNullability, data)
+    }
+
+    override fun transformDynamicTypeRef(dynamicTypeRef: FirDynamicTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRefWithNullability(dynamicTypeRef, data)
+    }
+
+    override fun transformFunctionTypeRef(functionTypeRef: FirFunctionTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRefWithNullability(functionTypeRef, data)
+    }
+
+    override fun transformUserTypeRef(userTypeRef: FirUserTypeRef, data: D): CompositeTransformResult<FirTypeRef> {
+        return transformTypeRefWithNullability(userTypeRef, data)
+    }
+
+    override fun transformCallableReferenceAccess(
+        callableReferenceAccess: FirCallableReferenceAccess,
+        data: D
+    ): CompositeTransformResult<FirStatement> {
+        return transformQualifiedAccessExpression(callableReferenceAccess, data)
+    }
+
+    override fun transformComponentCall(componentCall: FirComponentCall, data: D): CompositeTransformResult<FirStatement> {
+        return transformFunctionCall(componentCall, data)
+    }
+
+    override fun transformReturnExpression(returnExpression: FirReturnExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformJump(returnExpression, data)
+    }
+
+    override fun transformContinueExpression(continueExpression: FirContinueExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformJump(continueExpression, data)
+    }
+
+    override fun transformBreakExpression(breakExpression: FirBreakExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformJump(breakExpression, data)
+    }
+
+    override fun transformEnumEntry(enumEntry: FirEnumEntry, data: D): CompositeTransformResult<FirDeclaration> {
+        return transformRegularClass(enumEntry, data)
+    }
+
+    override fun transformLambdaArgumentExpression(lambdaArgumentExpression: FirLambdaArgumentExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformWrappedArgumentExpression(lambdaArgumentExpression, data)
+    }
+
+    override fun transformSpreadArgumentExpression(spreadArgumentExpression: FirSpreadArgumentExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformWrappedArgumentExpression(spreadArgumentExpression, data)
+    }
+
+    override fun transformNamedArgumentExpression(namedArgumentExpression: FirNamedArgumentExpression, data: D): CompositeTransformResult<FirStatement> {
+        return transformWrappedArgumentExpression(namedArgumentExpression, data)
+    }
+}
+
@@ -0,0 +1,85 @@
+/*
+ * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.fir.visitors
+
+import org.jetbrains.kotlin.fir.declarations.FirEnumEntry
+import org.jetbrains.kotlin.fir.expressions.*
+import org.jetbrains.kotlin.fir.types.*
+
+abstract class FirDefaultVisitor<R, D> : FirVisitor<R, D>() {
+    override fun visitDelegatedTypeRef(delegatedTypeRef: FirDelegatedTypeRef, data: D): R {
+        return visitTypeRef(delegatedTypeRef, data)
+    }
+
+    override fun visitImplicitTypeRef(implicitTypeRef: FirImplicitTypeRef, data: D): R {
+        return visitTypeRef(implicitTypeRef, data)
+    }
+
+    override fun visitResolvedTypeRef(resolvedTypeRef: FirResolvedTypeRef, data: D): R {
+        return visitTypeRef(resolvedTypeRef, data)
+    }
+
+    override fun visitErrorTypeRef(errorTypeRef: FirErrorTypeRef, data: D): R {
+        return visitResolvedTypeRef(errorTypeRef, data)
+    }
+
+    override fun visitResolvedFunctionTypeRef(resolvedFunctionTypeRef: FirResolvedFunctionTypeRef, data: D): R {
+        return visitResolvedTypeRef(resolvedFunctionTypeRef, data)
+    }
+
+    override fun visitTypeRefWithNullability(typeRefWithNullability: FirTypeRefWithNullability, data: D): R {
+        return visitTypeRef(typeRefWithNullability, data)
+    }
+
+    override fun visitDynamicTypeRef(dynamicTypeRef: FirDynamicTypeRef, data: D): R {
+        return visitTypeRefWithNullability(dynamicTypeRef, data)
+    }
+
+    override fun visitFunctionTypeRef(functionTypeRef: FirFunctionTypeRef, data: D): R {
+        return visitTypeRefWithNullability(functionTypeRef, data)
+    }
+
+    override fun visitUserTypeRef(userTypeRef: FirUserTypeRef, data: D): R {
+        return visitTypeRefWithNullability(userTypeRef, data)
+    }
+
+    override fun visitCallableReferenceAccess(callableReferenceAccess: FirCallableReferenceAccess, data: D): R {
+        return visitQualifiedAccessExpression(callableReferenceAccess, data)
+    }
+
+    override fun visitComponentCall(componentCall: FirComponentCall, data: D): R {
+        return visitFunctionCall(componentCall, data)
+    }
+
+    override fun visitReturnExpression(returnExpression: FirReturnExpression, data: D): R {
+        return visitJump(returnExpression, data)
+    }
+
+    override fun visitContinueExpression(continueExpression: FirContinueExpression, data: D): R {
+        return visitJump(continueExpression, data)
+    }
+
+    override fun visitBreakExpression(breakExpression: FirBreakExpression, data: D): R {
+        return visitJump(breakExpression, data)
+    }
+
+    override fun visitEnumEntry(enumEntry: FirEnumEntry, data: D): R {
+        return visitRegularClass(enumEntry, data)
+    }
+
+    override fun visitLambdaArgumentExpression(lambdaArgumentExpression: FirLambdaArgumentExpression, data: D): R {
+        return visitWrappedArgumentExpression(lambdaArgumentExpression, data)
+    }
+
+    override fun visitSpreadArgumentExpression(spreadArgumentExpression: FirSpreadArgumentExpression, data: D): R {
+        return visitWrappedArgumentExpression(spreadArgumentExpression, data)
+    }
+
+    override fun visitNamedArgumentExpression(namedArgumentExpression: FirNamedArgumentExpression, data: D): R {
+        return visitWrappedArgumentExpression(namedArgumentExpression, data)
+    }
+}
+
@@ -0,0 +1,84 @@
+/*
+ * Copyright 2010-2019 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.fir.visitors
+
+import org.jetbrains.kotlin.fir.declarations.FirEnumEntry
+import org.jetbrains.kotlin.fir.expressions.*
+import org.jetbrains.kotlin.fir.types.*
+
+abstract class FirDefaultVisitorVoid : FirVisitorVoid() {
+    override fun visitDelegatedTypeRef(delegatedTypeRef: FirDelegatedTypeRef) {
+        return visitTypeRef(delegatedTypeRef)
+    }
+
+    override fun visitImplicitTypeRef(implicitTypeRef: FirImplicitTypeRef) {
+        return visitTypeRef(implicitTypeRef)
+    }
+
+    override fun visitResolvedTypeRef(resolvedTypeRef: FirResolvedTypeRef) {
+        return visitTypeRef(resolvedTypeRef)
+    }
+
+    override fun visitErrorTypeRef(errorTypeRef: FirErrorTypeRef) {
+        return visitResolvedTypeRef(errorTypeRef)
+    }
+
+    override fun visitResolvedFunctionTypeRef(resolvedFunctionTypeRef: FirResolvedFunctionTypeRef) {
+        return visitResolvedTypeRef(resolvedFunctionTypeRef)
+    }
+
+    override fun visitTypeRefWithNullability(typeRefWithNullability: FirTypeRefWithNullability) {
+        return visitTypeRef(typeRefWithNullability)
+    }
+
+    override fun visitDynamicTypeRef(dynamicTypeRef: FirDynamicTypeRef) {
+        return visitTypeRefWithNullability(dynamicTypeRef)
+    }
+
+    override fun visitFunctionTypeRef(functionTypeRef: FirFunctionTypeRef) {
+        return visitTypeRefWithNullability(functionTypeRef)
+    }
+
+    override fun visitUserTypeRef(userTypeRef: FirUserTypeRef) {
+        return visitTypeRefWithNullability(userTypeRef)
+    }
+
+    override fun visitCallableReferenceAccess(callableReferenceAccess: FirCallableReferenceAccess) {
+        return visitQualifiedAccessExpression(callableReferenceAccess)
+    }
+
+    override fun visitComponentCall(componentCall: FirComponentCall) {
+        return visitFunctionCall(componentCall)
+    }
+
+    override fun visitReturnExpression(returnExpression: FirReturnExpression) {
+        return visitJump(returnExpression)
+    }
+
+    override fun visitContinueExpression(continueExpression: FirContinueExpression) {
+        return visitJump(continueExpression)
+    }
+
+    override fun visitBreakExpression(breakExpression: FirBreakExpression) {
+        return visitJump(breakExpression)
+    }
+
+    override fun visitEnumEntry(enumEntry: FirEnumEntry) {
+        return visitRegularClass(enumEntry)
+    }
+
+    override fun visitLambdaArgumentExpression(lambdaArgumentExpression: FirLambdaArgumentExpression) {
+        return visitWrappedArgumentExpression(lambdaArgumentExpression)
+    }
+
+    override fun visitSpreadArgumentExpression(spreadArgumentExpression: FirSpreadArgumentExpression) {
+        return visitWrappedArgumentExpression(spreadArgumentExpression)
+    }
+
+    override fun visitNamedArgumentExpression(namedArgumentExpression: FirNamedArgumentExpression) {
+        return visitWrappedArgumentExpression(namedArgumentExpression)
+    }
+}