commonMain.org.jetbrains.letsPlot.livemap.api.PieLayerBuilder.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of livemap Show documentation
A part of the Lets-Plot library.
The newest version!
/*
 * Copyright (c) 2019. JetBrains s.r.o.
 * Use of this source code is governed by the MIT license that can be found in the LICENSE file.
 */

package org.jetbrains.letsPlot.livemap.api

import org.jetbrains.letsPlot.commons.intern.spatial.LonLat
import org.jetbrains.letsPlot.commons.intern.typedGeometry.Vec
import org.jetbrains.letsPlot.commons.values.Color
import org.jetbrains.letsPlot.livemap.chart.ChartElementComponent
import org.jetbrains.letsPlot.livemap.chart.IndexComponent
import org.jetbrains.letsPlot.livemap.chart.LocatorComponent
import org.jetbrains.letsPlot.livemap.chart.PieSpecComponent
import org.jetbrains.letsPlot.livemap.chart.donut.Locator
import org.jetbrains.letsPlot.livemap.chart.donut.Renderer
import org.jetbrains.letsPlot.livemap.chart.donut.StrokeSide
import org.jetbrains.letsPlot.livemap.core.ecs.EcsEntity
import org.jetbrains.letsPlot.livemap.core.ecs.addComponents
import org.jetbrains.letsPlot.livemap.core.layers.LayerKind
import org.jetbrains.letsPlot.livemap.mapengine.LayerEntitiesComponent
import org.jetbrains.letsPlot.livemap.mapengine.MapProjection
import org.jetbrains.letsPlot.livemap.mapengine.RenderableComponent
import org.jetbrains.letsPlot.livemap.mapengine.placement.ScreenDimensionComponent
import org.jetbrains.letsPlot.livemap.mapengine.placement.WorldOriginComponent

@LiveMapDsl
class PieLayerBuilder(
    val factory: FeatureEntityFactory,
    val mapProjection: MapProjection
)

fun FeatureLayerBuilder.pies(block: PieLayerBuilder.() -> Unit) {
    val layerEntity = myComponentManager
        .createEntity("map_layer_pie")
        .addComponents {
            + layerManager.addLayer("geom_pie", LayerKind.FEATURES)
            + LayerEntitiesComponent()
        }

    PieLayerBuilder(
        FeatureEntityFactory(layerEntity, panningPointsMaxCount = 100),
        mapProjection
    ).apply(block)
}

fun PieLayerBuilder.pie(block: PieEntityBuilder.() -> Unit) {
    PieEntityBuilder(factory)
        .apply(block)
        .build()
}

@LiveMapDsl
class PieEntityBuilder(
    private val myFactory: FeatureEntityFactory
) {
    var sizeScalingRange: ClosedRange? = null
    var alphaScalingEnabled: Boolean = false

    var layerIndex: Int? = null
    var point: Vec? = null

    var indices: List = emptyList()
    var values: List = emptyList()
    var radius: Double = 0.0
    var holeSize: Double = 0.0
    var fillColors: List = emptyList()
    var strokeColors: List = emptyList()
    var strokeWidths: List = emptyList()
    var strokeSide: StrokeSide? = null
    var spacerColor: Color = Color.WHITE
    var spacerWidth: Double = 1.0
    var explodes: List? = null

    fun build(): EcsEntity {
        return when {
            point != null -> myFactory.createStaticFeatureWithLocation("map_ent_s_pie_sector", point!!)
            else -> error("Can't create pieSector entity. Coord is null.")
        }.run {
            myFactory.incrementLayerPointsTotalCount(1)
            setInitializer { worldPoint ->
                if (layerIndex != null) {
                    +IndexComponent(layerIndex!!, 0)
                }
                +LocatorComponent(Locator)
                +RenderableComponent().apply {
                    renderer = Renderer()
                }
                +ChartElementComponent().apply {
                    sizeScalingRange = [email protected]
                    alphaScalingEnabled = [email protected]
                }
                + PieSpecComponent().apply {
                    indices = [email protected]
                    sliceValues = [email protected]
                    radius = [email protected]
                    holeSize = [email protected]
                    fillColors = [email protected]
                    strokeColors = [email protected]
                    strokeWidths = [email protected]
                    strokeSide = [email protected]
                    spacerColor = [email protected]
                    spacerWidth = [email protected]
                    explodeValues = [email protected]
                }
                +WorldOriginComponent(worldPoint)
                +ScreenDimensionComponent()
            }
        }
    }
}