From 38c76e824d550bcadbd9ff1ab3feb04b28e50e23 Mon Sep 17 00:00:00 2001
From: Alexander Nozik <altavir@gmail.com>
Date: Wed, 14 Jul 2021 15:50:56 +0300
Subject: [PATCH] layer marking

---
 .../kotlin/space/kscience/visionforge/gdml/markLayers.kt   | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt b/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt
index a47374d0..dfee104e 100644
--- a/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt
+++ b/visionforge-gdml/src/commonMain/kotlin/space/kscience/visionforge/gdml/markLayers.kt
@@ -76,15 +76,14 @@ public fun SolidGroup.markLayers(thresholds: List<Int> = listOf(500, 1000, 20000
             if (layerIndex == 0) break
 
             node.vision.layer = layerIndex
-            val removedCount = node.selfCount * node.children.size
+            remaining -= node.selfCount * node.childrenCount
             logger?.apply {
                 if (node.selfCount > 1) {
-                    info { "Prototype with name ${node.name} moved to layer $layerIndex" }
+                    info { "Prototype with name ${node.name} moved to layer $layerIndex. $remaining nodes remains" }
                 } else {
-                    info { "Vision with name ${node.name} moved to layer $layerIndex" }
+                    info { "Vision with name ${node.name} moved to layer $layerIndex. $remaining nodes remains" }
                 }
             }
-            remaining -= removedCount
         }
     }
 }
\ No newline at end of file