Convert Entity/Expr/Stmt/Cfg to Proto

Author: Lipen

jacodb-ets/src/main/kotlin/org/jacodb/ets/proto/ConvertToProto.kt

@@ -31,6 +31,7 @@ import org.jacodb.ets.model.EtsModifiers
 import org.jacodb.ets.model.EtsNamespace
 import org.jacodb.ets.model.EtsNamespaceSignature
 import org.jacodb.ets.model.EtsScene
+import model.Block as ProtoBlock
 import model.BlockCfg as ProtoBlockCfg
 import model.Class as ProtoClass
 import model.ClassSignature as ProtoClassSignature
@@ -148,4 +149,15 @@ fun EtsModifiers.toProto(): Int {
     return mask
 }
 
-fun EtsBlockCfg.toProto(): ProtoBlockCfg = TODO()
+fun EtsBlockCfg.toProto(): ProtoBlockCfg {
+    return ProtoBlockCfg(
+        blocks = blocks.map { block ->
+            ProtoBlock(
+                id = block.id,
+                statements = block.statements.map { stmt -> stmt.toProto() },
+                successors = successors.getValue(block.id),
+                // Note: predecessors are omitted
+            )
+        },
+    )
+}

jacodb-ets/src/main/kotlin/org/jacodb/ets/proto/EtsEntityToProto.kt

@@ -0,0 +1,43 @@
+/*
+ *  Copyright 2022 UnitTestBot contributors (utbot.org)
+ * <p>
+ *  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
+ * <p>
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ *  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 org.jacodb.ets.proto
+
+import org.jacodb.ets.model.EtsEntity
+import org.jacodb.ets.model.EtsExpr
+import org.jacodb.ets.model.EtsRawEntity
+import org.jacodb.ets.model.EtsValue
+import model.RawValue as ProtoRawValue
+import model.Value as ProtoValue
+
+fun EtsEntity.toProto(): ProtoValue = accept(EtsEntityToProto)
+
+internal object EtsEntityToProto :
+    EtsEntity.Visitor<ProtoValue>,
+    EtsValue.Visitor.Default<ProtoValue>,
+    EtsExpr.Visitor.Default<ProtoValue> {
+
+    override fun visit(value: EtsRawEntity): ProtoValue {
+        val rawValue = ProtoRawValue(
+            kind = value.kind,
+            type = value.type.toProto(),
+        )
+        return ProtoValue(raw_value = rawValue)
+    }
+
+    override fun defaultVisit(value: EtsValue): ProtoValue = value.toProto()
+    override fun defaultVisit(expr: EtsExpr): ProtoValue = expr.toProto()
+}