kscience/visionforge
8
1
Fork 1
Code Issues Pull Requests Packages Projects Releases 3 Wiki Activity

WIP Root to Solid conversion

Browse Source
This commit is contained in:
Alexander Nozik 2021-08-26 09:52:35 +03:00
parent 193665b99a
commit 68704086e9
4 changed files with 30 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cern-root-loader/src/commonMain/kotlin/ru/mipt/npm/root/TGeoManager.kt
View File

@ -9,14 +9,14 @@ import kotlinx.serialization.Serializable
public class TGeoManager : TNamed() {
@Contextual
public val fMatrices: TObjArray<TGeoMatrix> = TObjArray.getEmpty()
public val fMatrices: TObjArray<@Contextual TGeoMatrix> = TObjArray.getEmpty()
@Contextual
public val fShapes: TObjArray<TGeoShape> = TObjArray.getEmpty()
public val fShapes: TObjArray<@Contextual TGeoShape> = TObjArray.getEmpty()
@Contextual
public val fVolumes: TObjArray<TGeoVolume> = TObjArray.getEmpty()
public val fVolumes: TObjArray<@Contextual TGeoVolume> = TObjArray.getEmpty()
@Contextual
public val fNodes: TObjArray<TGeoNode> = TObjArray.getEmpty()
public val fNodes: TObjArray<@Contextual TGeoNode> = TObjArray.getEmpty()
}

23
cern-root-loader/src/commonMain/kotlin/ru/mipt/npm/root/jsonToRoot.kt
View File

@ -81,19 +81,19 @@ private object RootDecoder {
): SerializersModule = SerializersModule {
include(serializersModule)
contextual(TGeoManager.serializer().unref(refCache))
//contextual(TGeoManager.serializer().unref(refCache))
contextual(TObjArray::class) { TObjArray.serializer(it[0]).unref(refCache) }
contextual(TGeoVolumeAssembly.serializer().unref(refCache))
contextual(TGeoShapeAssembly.serializer().unref(refCache))
//contextual(TGeoVolumeAssembly.serializer().unref(refCache))
//contextual(TGeoShapeAssembly.serializer().unref(refCache))
contextual(TGeoRotation.serializer().unref(refCache))
contextual(TGeoMedium.serializer().unref(refCache))
contextual(TGeoVolume.serializer().unref(refCache))
contextual(TGeoMatrix.serializer().unref(refCache))
contextual(TGeoNode.serializer().unref(refCache))
contextual(TGeoNodeOffset.serializer().unref(refCache))
contextual(TGeoNodeMatrix.serializer().unref(refCache))
contextual(TGeoShape.serializer().unref(refCache))
contextual(TObject.serializer().unref(refCache))
//contextual(TGeoVolume.serializer().unref(refCache))
//contextual(TGeoMatrix.serializer().unref(refCache))
//contextual(TGeoNode.serializer().unref(refCache))
//contextual(TGeoNodeOffset.serializer().unref(refCache))
//contextual(TGeoNodeMatrix.serializer().unref(refCache))
//contextual(TGeoShape.serializer().unref(refCache))
//contextual(TObject.serializer().unref(refCache))
polymorphicDefault(TGeoShape::class) {
@ -245,7 +245,8 @@ private object RootDecoder {
subclass(TGeoNodeOffset.serializer())
}
polymorphic(TGeoVolume::class, TGeoVolume.serializer()) {
polymorphic(TGeoVolume::class) {
subclass(TGeoVolume.serializer())
subclass(TGeoVolumeAssembly.serializer())
}
}

15
cern-root-loader/src/commonMain/kotlin/ru/mipt/npm/root/rootToSolid.kt
View File

@ -115,7 +115,7 @@ private fun SolidGroup.addShape(shape: TGeoShape) {
private fun SolidGroup.node(obj: TGeoNode) {
if (obj.fVolume != null) {
volume(obj.fVolume).apply {
volume(obj.fVolume, obj.fName).apply {
when (obj) {
is TGeoNodeMatrix -> {
useMatrix(obj.fMatrix)
@ -145,15 +145,22 @@ private fun buildGroup(volume: TGeoVolume): SolidGroup {
private val SolidGroup.rootPrototypes: SolidGroup get() = (parent as? SolidGroup)?.rootPrototypes ?: this
private fun SolidGroup.volume(volume: TGeoVolume, cache: Boolean = true): Solid {
private fun SolidGroup.volume(volume: TGeoVolume, name: String? = null, cache: Boolean = true): Solid {
val group = buildGroup(volume)
val combinedName = if (volume.fName.isEmpty()) {
name
} else if (name == null) {
volume.fName
} else {
"${name}_${volume.fName}"
}
return if (!cache) {
group
} else newRef(
name = volume.fName.ifEmpty { null },
name = combinedName,
obj = group,
prototypeHolder = rootPrototypes,
templateName = volumesName + Name.parse(volume.fName.ifEmpty { group.toString() })
templateName = volumesName + Name.parse(combinedName ?: "volume[${group.hashCode()}]")
)
}

6
cern-root-loader/src/jvmTest/kotlin/ru/mipt/npm/root/loadBMN.kt
View File

@ -1,12 +1,12 @@
package ru.mipt.npm.root
import kotlinx.serialization.json.*
import space.kscience.visionforge.solid.Solids
import java.time.Duration
import kotlin.system.measureTimeMillis
private fun JsonElement.countTypes(): Sequence<String> = sequence {
val json = this@countTypes
when (json){
when (val json = this@countTypes){
is JsonObject -> {
json["_typename"]?.let { yield(it.jsonPrimitive.content) }
json.values.forEach { yieldAll(it.countTypes()) }
@ -33,7 +33,7 @@ fun main() {
val geo = TObject.decodeFromString(TGeoManager.serializer(), string)
val solid = geo.toSolid()
println(solid)
println(Solids.encodeToString(solid))
}
println(Duration.ofMillis(time))