dima-avdeev-jb
3 years ago
committed by
GitHub
7 changed files with 381 additions and 5 deletions
@ -0,0 +1,120 @@
|
||||
/* |
||||
* Copyright 2020-2022 JetBrains s.r.o. and respective authors and developers. |
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE.txt file. |
||||
*/ |
||||
|
||||
package com.jetbrains.compose.color |
||||
|
||||
import androidx.compose.foundation.layout.fillMaxSize |
||||
import androidx.compose.material.Surface |
||||
import androidx.compose.runtime.mutableStateOf |
||||
import androidx.compose.ui.Modifier |
||||
import androidx.compose.ui.awt.ComposePanel |
||||
import com.intellij.codeInsight.daemon.LineMarkerInfo |
||||
import com.intellij.codeInsight.daemon.LineMarkerProvider |
||||
import com.intellij.icons.AllIcons |
||||
import com.intellij.openapi.editor.markup.GutterIconRenderer |
||||
import com.intellij.openapi.ui.DialogWrapper |
||||
import com.intellij.psi.PsiElement |
||||
import org.jetbrains.kotlin.psi.KtPsiFactory |
||||
import org.jetbrains.uast.* |
||||
import javax.swing.JComponent |
||||
import androidx.compose.ui.graphics.Color |
||||
import androidx.compose.ui.graphics.ImageBitmap |
||||
import androidx.compose.ui.graphics.painter.ColorPainter |
||||
import androidx.compose.ui.graphics.toArgb |
||||
import com.intellij.openapi.application.ApplicationManager |
||||
import com.jetbrains.compose.theme.WidgetTheme |
||||
import org.intellij.datavis.r.inlays.components.GraphicsManager |
||||
import java.awt.Component |
||||
import java.awt.Graphics |
||||
import javax.swing.Icon |
||||
|
||||
class ColorLineMarkerProvider : LineMarkerProvider { |
||||
|
||||
override fun getLineMarkerInfo(element: PsiElement): LineMarkerInfo<*>? { |
||||
val project = element.project |
||||
val ktPsiFactory = KtPsiFactory(project) |
||||
val uElement: UElement = element.toUElement() ?: return null |
||||
if (uElement is UCallExpression) { |
||||
if (uElement.kind == UastCallKind.METHOD_CALL && uElement.methodIdentifier?.name == "Color") { |
||||
val colorLongValue = (uElement.valueArguments.firstOrNull() as? ULiteralExpression)?.getLongValue() |
||||
val previousColor = try { |
||||
Color(colorLongValue!!) |
||||
} catch (t: Throwable) { |
||||
Color(0xffffffff) |
||||
} |
||||
|
||||
val iconSize = 20 |
||||
return LineMarkerInfo( |
||||
element, |
||||
element.textRange, |
||||
object : Icon { |
||||
override fun paintIcon(c: Component?, g: Graphics?, x: Int, y: Int) { |
||||
g?.color = java.awt.Color( |
||||
previousColor.red, |
||||
previousColor.green, |
||||
previousColor.blue, |
||||
previousColor.alpha |
||||
) |
||||
g?.fillRect(0, 0, iconSize, iconSize) |
||||
} |
||||
override fun getIconWidth(): Int = iconSize |
||||
override fun getIconHeight(): Int = iconSize |
||||
}, |
||||
null, |
||||
{ _, psiElement: PsiElement -> |
||||
val isDarkMode = try { |
||||
GraphicsManager.getInstance(project)?.isDarkModeEnabled ?: false |
||||
} catch (t: Throwable) { |
||||
false |
||||
} |
||||
class ChooseColorDialog() : DialogWrapper(project) { |
||||
val colorState = mutableStateOf(previousColor) |
||||
|
||||
init { |
||||
title = "Choose color" |
||||
init() |
||||
} |
||||
|
||||
override fun createCenterPanel(): JComponent = |
||||
ComposePanel().apply { |
||||
setBounds(0, 0, 400, 400) |
||||
setContent { |
||||
WidgetTheme(darkTheme = isDarkMode) { |
||||
Surface(modifier = Modifier.fillMaxSize()) { |
||||
ColorPicker(colorState) |
||||
} |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
val chooseColorDialog = ChooseColorDialog() |
||||
val result = chooseColorDialog.showAndGet() |
||||
if (result) { |
||||
val color = chooseColorDialog.colorState.value |
||||
ApplicationManager.getApplication().runWriteAction { |
||||
psiElement.replace( |
||||
ktPsiFactory.createExpression( |
||||
"Color(${color.toHexString()})" |
||||
) |
||||
) |
||||
} |
||||
} |
||||
}, |
||||
GutterIconRenderer.Alignment.RIGHT, |
||||
{ "change color literal" } |
||||
) |
||||
} |
||||
} |
||||
return null |
||||
} |
||||
|
||||
override fun collectSlowLineMarkers( |
||||
elements: MutableList<out PsiElement>, |
||||
result: MutableCollection<in LineMarkerInfo<*>> |
||||
) { |
||||
|
||||
} |
||||
} |
@ -0,0 +1,128 @@
|
||||
/* |
||||
* Copyright 2020-2022 JetBrains s.r.o. and respective authors and developers. |
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE.txt file. |
||||
*/ |
||||
|
||||
package com.jetbrains.compose.color |
||||
|
||||
import androidx.compose.foundation.Canvas |
||||
import androidx.compose.foundation.background |
||||
import androidx.compose.foundation.clickable |
||||
import androidx.compose.foundation.layout.* |
||||
import androidx.compose.material.Divider |
||||
import androidx.compose.material.Text |
||||
import androidx.compose.material.TextField |
||||
import androidx.compose.runtime.* |
||||
import androidx.compose.ui.Alignment |
||||
import androidx.compose.ui.Modifier |
||||
import androidx.compose.ui.geometry.Offset |
||||
import androidx.compose.ui.geometry.Size |
||||
import androidx.compose.ui.graphics.Color |
||||
import androidx.compose.ui.graphics.drawscope.Stroke |
||||
import androidx.compose.ui.graphics.toArgb |
||||
import androidx.compose.ui.input.pointer.isPrimaryPressed |
||||
import androidx.compose.ui.input.pointer.pointerInput |
||||
import androidx.compose.ui.unit.dp |
||||
|
||||
private const val VALUE_BAND_RATIO = 0.07f |
||||
private val DEFAULT_COLORS = |
||||
listOf(Color.Red, Color.Green, Color.Blue, Color.Black, Color.Gray, Color.Yellow, Color.Cyan, Color.Magenta) |
||||
|
||||
@Composable |
||||
fun ColorPicker(colorState: MutableState<Color>) { |
||||
var currentColor: Color by remember { colorState } |
||||
Column { |
||||
Row { |
||||
DEFAULT_COLORS.forEach { |
||||
Box(Modifier.size(30.dp).background(color = it).clickable { |
||||
currentColor = it |
||||
}) |
||||
} |
||||
} |
||||
Divider(Modifier.size(2.dp)) |
||||
Row(Modifier.fillMaxWidth(), verticalAlignment = Alignment.CenterVertically) { |
||||
Text("Result color:") |
||||
Divider(Modifier.size(2.dp)) |
||||
TextField(modifier = Modifier.width(120f.dp), value = currentColor.toHexString(), onValueChange = {}) |
||||
Divider(Modifier.size(2.dp)) |
||||
val size = 60f |
||||
Box(Modifier.size(size.dp).background(color = currentColor)) |
||||
} |
||||
Divider(Modifier.size(2.dp)) |
||||
var width by remember { mutableStateOf(300) } |
||||
var height by remember { mutableStateOf(256) } |
||||
val rainbowWidth by derivedStateOf { (width * (1 - VALUE_BAND_RATIO)).toInt() } |
||||
val bandWidth by derivedStateOf { width * VALUE_BAND_RATIO } |
||||
fun calcHue(x: Float) = limit0to1(x / rainbowWidth) * HSV.HUE_MAX_VALUE |
||||
fun calcSaturation(y: Float) = 1 - limit0to1(y / height) |
||||
fun calcValue(y: Float) = 1 - limit0to1(y / height) |
||||
Row(Modifier.fillMaxSize()) { |
||||
Canvas(Modifier.fillMaxSize().pointerInput(Unit) { |
||||
width = size.width |
||||
height = size.height |
||||
awaitPointerEventScope { |
||||
while (true) { |
||||
val event = awaitPointerEvent() |
||||
if (event.buttons.isPrimaryPressed) { |
||||
val position = event.changes.first().position |
||||
if (position.x < rainbowWidth) { |
||||
currentColor = try { |
||||
currentColor.toHsv().copy( |
||||
hue = calcHue(position.x), |
||||
saturation = calcSaturation(position.y) |
||||
).toRgb() |
||||
} catch (t: Throwable) { |
||||
t.printStackTrace() |
||||
println("exception $t") |
||||
currentColor |
||||
} |
||||
} else { |
||||
currentColor = |
||||
currentColor.toHsv().copy( |
||||
value = calcValue(position.y) |
||||
).toRgb() |
||||
} |
||||
} |
||||
} |
||||
} |
||||
}) { |
||||
for (x in 0..rainbowWidth) { |
||||
for (y in 0..height) { |
||||
drawRect( |
||||
color = currentColor.toHsv().copy( |
||||
hue = calcHue(x.toFloat()), |
||||
saturation = calcSaturation(y.toFloat()) |
||||
).toRgb(), |
||||
topLeft = Offset(x.toFloat(), y.toFloat()), |
||||
size = Size(1f, 1f) |
||||
) |
||||
} |
||||
} |
||||
val valueBandX = rainbowWidth + 1 |
||||
for (y in 0..height) { |
||||
drawRect( |
||||
color = currentColor.toHsv().copy(value = calcValue(y.toFloat())).toRgb(), |
||||
topLeft = Offset(valueBandX.toFloat(), y.toFloat()), |
||||
size = Size(bandWidth, 1f) |
||||
) |
||||
} |
||||
val circleX = (currentColor.toHsv().hue / 360) * rainbowWidth |
||||
val circleY = (1 - currentColor.toHsv().saturation) * height |
||||
drawCircle( |
||||
center = Offset(circleX, circleY), |
||||
color = Color.Black, |
||||
radius = 5f, |
||||
style = Stroke(width = 3f) |
||||
) |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
fun Color.toHexString() = "0x" + toArgb().toUInt().toString(16) |
||||
fun limit(value: Float, min: Float, max: Float) = minOf( |
||||
maxOf(value, min), |
||||
max |
||||
) |
||||
|
||||
fun limit0to1(value: Float) = limit(value = value, 0f, 1f) |
@ -0,0 +1,75 @@
|
||||
/* |
||||
* Copyright 2020-2022 JetBrains s.r.o. and respective authors and developers. |
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE.txt file. |
||||
*/ |
||||
|
||||
package com.jetbrains.compose.color |
||||
|
||||
import androidx.compose.ui.graphics.Color |
||||
import kotlin.math.abs |
||||
|
||||
data class HSV( |
||||
/** |
||||
* 0.0 .. 360.0 |
||||
*/ |
||||
val hue: Float, |
||||
/** |
||||
* 0.0 .. 1.0 |
||||
*/ |
||||
val saturation: Float, |
||||
/** |
||||
* 0.0 . 1.0¬ |
||||
*/ |
||||
val value: Float |
||||
) { |
||||
companion object { |
||||
const val HUE_MAX_VALUE = 360f |
||||
} |
||||
} |
||||
|
||||
/** |
||||
* Convert to HSV color space |
||||
* https://www.rapidtables.com/convert/color/rgb-to-hsv.html |
||||
*/ |
||||
fun Color.toHsv(): HSV { |
||||
val max = maxOf(red, green, blue) |
||||
val min = minOf(red, green, blue) |
||||
val delta = max - min |
||||
val h = when { |
||||
delta == 0f -> 0f |
||||
max == red -> 60 * ((green - blue) / delta).mod(6f) |
||||
max == green -> 60 * ((blue - red) / delta + 2) |
||||
max == blue -> 60 * ((red - green) / delta + 4) |
||||
else -> 0f |
||||
} |
||||
val s = when { |
||||
max == 0f -> 0f |
||||
else -> delta / max |
||||
} |
||||
val v = max |
||||
return HSV( |
||||
hue = h, |
||||
saturation = s, |
||||
value = v |
||||
) |
||||
} |
||||
|
||||
/** |
||||
* Convert to RGB color space |
||||
* https://www.rapidtables.com/convert/color/hsv-to-rgb.html |
||||
*/ |
||||
fun HSV.toRgb(): Color { |
||||
val c = value * saturation |
||||
val x = minOf(c * (1 - abs((hue / 60).mod(2f) - 1)), 1f) |
||||
val m = value - c |
||||
val tempColor = when { |
||||
hue >= 0 && hue < 60 -> Color(c, x, 0f) |
||||
hue >= 60 && hue < 120 -> Color(x, c, 0f) |
||||
hue >= 120 && hue < 180 -> Color(0f, c, x) |
||||
hue >= 180 && hue < 240 -> Color(0f, x, c) |
||||
hue >= 240 && hue < 300 -> Color(x, 0f, c) |
||||
else -> Color(c, 0f, x) |
||||
} |
||||
return Color(minOf(m + tempColor.red, 1f), minOf(m + tempColor.green, 1f), minOf(m + tempColor.blue, 1f)) |
||||
} |
||||
|
@ -0,0 +1,25 @@
|
||||
/* |
||||
* Copyright 2020-2022 JetBrains s.r.o. and respective authors and developers. |
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE.txt file. |
||||
*/ |
||||
|
||||
package com.jetbrains.compose.color |
||||
|
||||
import androidx.compose.runtime.mutableStateOf |
||||
import androidx.compose.runtime.remember |
||||
import androidx.compose.ui.graphics.Color |
||||
import androidx.compose.ui.unit.dp |
||||
import androidx.compose.ui.window.Window |
||||
import androidx.compose.ui.window.WindowState |
||||
import androidx.compose.ui.window.application |
||||
|
||||
fun main() = application { |
||||
val windowState = remember { WindowState(width = 400.dp, height = 400.dp) } |
||||
Window( |
||||
onCloseRequest = ::exitApplication, |
||||
title = "ColorPicker", |
||||
state = windowState |
||||
) { |
||||
ColorPicker(mutableStateOf(Color(0xffaabbcc))) |
||||
} |
||||
} |
@ -0,0 +1,22 @@
|
||||
/* |
||||
* Copyright 2020-2022 JetBrains s.r.o. and respective authors and developers. |
||||
* Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE.txt file. |
||||
*/ |
||||
|
||||
package com.jetbrains.compose.color |
||||
|
||||
import androidx.compose.ui.graphics.Color |
||||
import org.junit.Test |
||||
import kotlin.test.assertEquals |
||||
|
||||
class HSVTest { |
||||
|
||||
@Test |
||||
fun testGreenToHsv() { |
||||
val greenRgb = Color(0xff00ff00) |
||||
val result = greenRgb.toHsv() |
||||
assertEquals(HSV(120f, 1f, 1f), result) |
||||
assertEquals(greenRgb, result.toRgb()) |
||||
} |
||||
|
||||
} |
Loading…
Reference in new issue