@@ -6,13 +6,14 @@ import org.utbot.framework.plugin.api.UtAssembleModel
6
6
import org.utbot.framework.plugin.api.UtClassRefModel
7
7
import org.utbot.framework.plugin.api.UtCompositeModel
8
8
import org.utbot.framework.plugin.api.UtCustomModel
9
+ import org.utbot.framework.plugin.api.UtDirectGetFieldModel
9
10
import org.utbot.framework.plugin.api.UtDirectSetFieldModel
10
11
import org.utbot.framework.plugin.api.UtEnumConstantModel
12
+ import org.utbot.framework.plugin.api.UtExecutableCallModel
11
13
import org.utbot.framework.plugin.api.UtLambdaModel
12
14
import org.utbot.framework.plugin.api.UtModel
13
15
import org.utbot.framework.plugin.api.UtNullModel
14
16
import org.utbot.framework.plugin.api.UtPrimitiveModel
15
- import org.utbot.framework.plugin.api.UtStatementCallModel
16
17
import org.utbot.framework.plugin.api.UtStatementModel
17
18
import org.utbot.framework.plugin.api.UtVoidModel
18
19
@@ -51,6 +52,9 @@ private fun UtModel.calculateSize(used: MutableSet<UtModel> = mutableSetOf()): I
51
52
52
53
private fun UtStatementModel.calculateSize (used : MutableSet <UtModel > = mutableSetOf()): Int =
53
54
when (this ) {
55
+ is UtExecutableCallModel -> 1 + params.sumOf { it.calculateSize(used) } + (instance?.calculateSize(used) ? : 0 )
54
56
is UtDirectSetFieldModel -> 1 + fieldModel.calculateSize(used) + instance.calculateSize(used)
55
- is UtStatementCallModel -> 1 + params.sumOf { it.calculateSize(used) } + (instance?.calculateSize(used) ? : 0 )
57
+
58
+ // -2 is added to encourage use of non-hardcoded values (including compensation for one extra `UtAssembleModel`)
59
+ is UtDirectGetFieldModel -> (- 2 + instance.calculateSize(used)).coerceAtLeast(0 )
56
60
}
0 commit comments