commonMain.jetbrains.datalore.plot.config.LiveMapOptionsParser.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) 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 jetbrains.datalore.plot.config

import jetbrains.datalore.plot.base.livemap.LiveMapOptions
import jetbrains.datalore.plot.base.livemap.LivemapConstants.DisplayMode
import jetbrains.datalore.plot.base.livemap.LivemapConstants.Projection
import jetbrains.datalore.plot.config.Option.Geom.LiveMap
import jetbrains.datalore.plot.config.Option.Plot.LAYERS

class LiveMapOptionsParser {
    companion object {

        fun parseFromPlotSpec(plotSpec: Map): LiveMapOptions? {
            fun Map<*, *>.isLiveMap(): Boolean = this[Option.Layer.GEOM] == Option.GeomName.LIVE_MAP

            val layers = plotSpec.getMaps(LAYERS)!!
            if (layers.any { it.isLiveMap() }) {
                require(layers.count { it.isLiveMap() } == 1) { "Only one geom_livemap is allowed per plot" }
                require(layers.first().isLiveMap()) { "geom_livemap should be a first geom" }
                @Suppress("UNCHECKED_CAST")
                return parseFromLayerOptions(OptionsAccessor(layers.first() as Map))
            }

            return null
        }

        fun parseFromLayerOptions(opts: OptionsAccessor): LiveMapOptions {

            return LiveMapOptions(
                zoom = opts.getInteger(LiveMap.ZOOM),
                location = opts.get(LiveMap.LOCATION),
                stroke = opts.getDouble(LiveMap.STROKE),
                interactive = opts.getBoolean(LiveMap.INTERACTIVE, true),
                displayMode = opts.getString(LiveMap.DISPLAY_MODE)?.let(::parseDisplayMode) ?: DisplayMode.POINT,
                scaled = opts.getBoolean(LiveMap.SCALED, false),
                clustering = opts.getBoolean(LiveMap.CLUSTERING, false),
                labels = opts.getBoolean(LiveMap.LABELS, true),
                projection = opts.getString(LiveMap.PROJECTION)?.let(::parseProjection) ?: Projection.EPSG3857,
                geodesic = opts.getBoolean(LiveMap.GEODESIC, true),
                geocodingService = opts.getMap(LiveMap.GEOCODING),
                tileProvider = opts.getMap(LiveMap.TILES),
                devParams = opts.getMap(LiveMap.DEV_PARAMS)
            )
        }

        private fun > formatValues(values: Array): String {
            return values.joinToString(prefix = "=[", separator = "|", postfix = "]") { "'${it.name.lowercase()}'" }
        }

        private fun parseDisplayMode(displayMode: String): DisplayMode {
            try {
                return DisplayMode.valueOf(displayMode.uppercase())
            } catch (ignored: Exception) {
                throw IllegalArgumentException("geom" + formatValues(DisplayMode.values()))
            }
        }

        private fun parseProjection(projection: String): Projection {
            try {
                return Projection.valueOf(projection.uppercase())
            } catch (ignored: Exception) {
                throw IllegalArgumentException(LiveMap.PROJECTION + formatValues(Projection.values()))
            }
        }
    }
}