commonMain.jetbrains.datalore.vis.canvas.ScaledContext2d.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.vis.canvas

import jetbrains.datalore.base.geometry.DoubleRectangle
import jetbrains.datalore.base.values.Color
import jetbrains.datalore.vis.canvas.Canvas.Snapshot

internal class ScaledContext2d(
    private val ctx: Context2d,
    private val myScale: Double
) : Context2d {

    private fun scaled(value: Double) = myScale * value
    private fun descaled(value: Double) = value / myScale

    private fun scaled(values: DoubleArray): DoubleArray {
        if (myScale == 1.0) {
            return values
        }
        val res = DoubleArray(values.size)
        for (i in values.indices) {
            res[i] = scaled(values[i])
        }
        return res
    }

    private fun scaled(font: Context2d.Font): Context2d.Font = font.copy(fontSize = scaled(font.fontSize))
    override fun drawImage(snapshot: Snapshot, x: Double, y: Double) = ctx.drawImage(snapshot, scaled(x), scaled(y))

    override fun drawImage(snapshot: Snapshot, x: Double, y: Double, dw: Double, dh: Double) {
        ctx.drawImage(snapshot, scaled(x), scaled(y), scaled(dw), scaled(dh))
    }

    override fun drawImage(
        snapshot: Snapshot,
        sx: Double,
        sy: Double,
        sw: Double,
        sh: Double,
        dx: Double,
        dy: Double,
        dw: Double,
        dh: Double
    ) {
        ctx.drawImage(
            snapshot,
            scaled(sx),
            scaled(sy),
            scaled(sw),
            scaled(sh),
            scaled(dx),
            scaled(dy),
            scaled(dw),
            scaled(dh)
        )
    }

    override fun beginPath() = ctx.beginPath()
    override fun closePath() = ctx.closePath()
    override fun stroke() = ctx.stroke()
    override fun fill() = ctx.fill()

    override fun fillRect(x: Double, y: Double, w: Double, h: Double) {
        ctx.fillRect(scaled(x), scaled(y), scaled(w), scaled(h))
    }

    override fun moveTo(x: Double, y: Double) = ctx.moveTo(scaled(x), scaled(y))
    override fun lineTo(x: Double, y: Double) = ctx.lineTo(scaled(x), scaled(y))

    override fun arc(x: Double, y: Double, radius: Double, startAngle: Double, endAngle: Double, anticlockwise: Boolean) {
        ctx.arc(scaled(x), scaled(y), scaled(radius), startAngle, endAngle, anticlockwise)
    }

    override fun save() = ctx.save()
    override fun restore() = ctx.restore()
    override fun setFillStyle(color: Color?) = ctx.setFillStyle(color)
    override fun setStrokeStyle(color: Color?) = ctx.setStrokeStyle(color)
    override fun setGlobalAlpha(alpha: Double) = ctx.setGlobalAlpha(alpha)
    override fun setFont(f: Context2d.Font) = ctx.setFont(scaled(f))
    override fun setLineWidth(lineWidth: Double) = ctx.setLineWidth(scaled(lineWidth))

    override fun strokeRect(x: Double, y: Double, w: Double, h: Double) {
        ctx.strokeRect(scaled(x), scaled(y), scaled(w), scaled(h))
    }

    override fun strokeText(text: String, x: Double, y: Double) = ctx.strokeText(text, scaled(x), scaled(y))
    override fun fillText(text: String, x: Double, y: Double) = ctx.fillText(text, scaled(x), scaled(y))
    override fun scale(x: Double, y: Double) = ctx.scale(x, y)
    override fun scale(xy: Double) = scale(xy, xy)
    override fun rotate(angle: Double) = ctx.rotate(angle)
    override fun translate(x: Double, y: Double) = ctx.translate(scaled(x), scaled(y))

    override fun transform(m11: Double, m12: Double, m21: Double, m22: Double, dx: Double, dy: Double) {
        ctx.transform(m11, m12, m21, m22, scaled(dx), scaled(dy))
    }

    override fun bezierCurveTo(cp1x: Double, cp1y: Double, cp2x: Double, cp2y: Double, x: Double, y: Double) {
        ctx.bezierCurveTo(scaled(cp1x), scaled(cp1y), scaled(cp2x), scaled(cp2y), scaled(x), scaled(y))
    }

    override fun setLineJoin(lineJoin: Context2d.LineJoin) = ctx.setLineJoin(lineJoin)
    override fun setLineCap(lineCap: Context2d.LineCap) = ctx.setLineCap(lineCap)
    override fun setTextBaseline(baseline: Context2d.TextBaseline) = ctx.setTextBaseline(baseline)
    override fun setTextAlign(align: Context2d.TextAlign) = ctx.setTextAlign(align)

    override fun setTransform(m11: Double, m12: Double, m21: Double, m22: Double, dx: Double, dy: Double) {
        ctx.setTransform(m11, m12, m21, m22, scaled(dx), scaled(dy))
    }

    override fun fillEvenOdd() = ctx.fillEvenOdd()
    override fun setLineDash(lineDash: DoubleArray) = ctx.setLineDash(scaled(lineDash))
    override fun measureText(str: String): Double = descaled(ctx.measureText(str))

    override fun clearRect(rect: DoubleRectangle) {
        ctx.clearRect(
            DoubleRectangle(
                scaled(rect.left),
                scaled(rect.top),
                scaled(rect.width),
                scaled(rect.height)
            )
        )
    }
}