tri.covid19.coda.charts.DeathCaseChart.kt Maven / Gradle / Ivy

Go to download
/*-
 * #%L
 * coda-app
 * --
 * Copyright (C) 2020 - 2021 Elisha Peterson
 * --
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * #L%
 */
package tri.covid19.coda.charts

import javafx.event.EventTarget
import javafx.scene.chart.LineChart
import javafx.scene.chart.NumberAxis
import javafx.scene.layout.Priority
import tornadofx.attachTo
import tornadofx.gridpaneConstraints
import tri.covid19.coda.utils.chartContextMenu
import tri.covid19.coda.utils.dataSeries
import tri.covid19.coda.utils.series
import tri.timeseries.TimeSeries

/** Displays total on x-axis, doubling time on vertical axis. */
class DeathCaseChart : LineChart(NumberAxis(), NumberAxis()) {

    init {
        gridpaneConstraints { vhGrow = Priority.ALWAYS }
        animated = false
        createSymbols = false
        isLegendVisible = false
        axisSortingPolicy = SortingPolicy.NONE
        chartContextMenu()
    }

    /** Set mapping of deaths (first set of series) to cases (second set of series). */
    var series: Pair, List>
        get() = TODO()
        set(value) {
            val commonIndices = value.first.indices.intersect(value.second.indices)
            dataSeries = commonIndices.map { value.first[it] to value.second[it] }
                    .map { series(it.first.areaId, it.first, it.second) }
        }

}

/** Creates line chart. */
fun EventTarget.deathcasechart(title: String, xTitle: String, yTitle: String, op: DeathCaseChart.() -> Unit = {}) = DeathCaseChart()
        .attachTo(this, op) {
            it.title = title
            it.xAxis.label = xTitle
            it.yAxis.label = yTitle
        }