jvmMain.plot.AwtLiveMapFactoryUtil.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lets-plot-common Show documentation

Lets-Plot JVM package without rendering part

There is a newer version: 4.5.3-alpha1

/*
 * Copyright (c) 2021. JetBrains s.r.o.
 * Use of this source code is governed by the MIT license that can be found in the LICENSE file.
 */

package jetbrains.datalore.plot

import jetbrains.datalore.base.geometry.DoubleVector
import jetbrains.datalore.plot.builder.PlotContainer
import jetbrains.datalore.plot.builder.assemble.PlotAssembler
import jetbrains.datalore.plot.config.LiveMapOptionsParser
import jetbrains.datalore.plot.livemap.CursorServiceConfig
import jetbrains.datalore.plot.livemap.LiveMapProvider
import jetbrains.datalore.vis.svg.SvgSvgElement
import javax.swing.JComponent

internal object AwtLiveMapFactoryUtil {
    fun buildLiveMapComponent(
        assembler: PlotAssembler,
        processedSpec: MutableMap,
        preferredSize: DoubleVector,
        svgComponentFactory: (svg: SvgSvgElement) -> JComponent,
        executor: (() -> Unit) -> Unit
    ): JComponent {
        val cursorServiceConfig = CursorServiceConfig()
        injectLiveMapProvider(assembler, processedSpec, cursorServiceConfig)

        val plot = assembler.createPlot()
        val plotContainer = PlotContainer(plot, preferredSize)
        val plotComponent = AwtPlotFactoryUtil.buildPlotComponent(
            plotContainer,
            svgComponentFactory,
            executor
        )

        @Suppress("UNCHECKED_CAST")
        return AwtLiveMapPanel(
            plotContainer,
            plotComponent,
            executor,
            cursorServiceConfig
        )
    }

    private fun injectLiveMapProvider(
        plotAssembler: PlotAssembler,
        processedPlotSpec: MutableMap,
        cursorServiceConfig: CursorServiceConfig
    ) {
        LiveMapOptionsParser.parseFromPlotSpec(processedPlotSpec)
            ?.let {
                LiveMapProvider.injectLiveMapProvider(
                    plotAssembler.coreLayersByTile,
                    it,
                    cursorServiceConfig
                )
            }
    }
}