Add image and byte utilities

This commit introduces toBase64() and encodeGzip() methods for ByteArray and image resizing methods for BufferedImage.

Author: ermolenkodev

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/BytesUtils.kt

@@ -0,0 +1,5 @@
+package org.jetbrains.kotlinx.dataframe.impl.io
+
+import java.util.Base64
+
+internal fun ByteArray.toBase64(): String = Base64.getEncoder().encodeToString(this)

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/compression.kt

@@ -0,0 +1,11 @@
+package org.jetbrains.kotlinx.dataframe.impl.io
+
+import java.io.ByteArrayOutputStream
+import java.util.zip.GZIPOutputStream
+
+internal fun ByteArray.encodeGzip(): ByteArray {
+    val bos = ByteArrayOutputStream()
+    GZIPOutputStream(bos).use { it.write(this) }
+
+    return bos.toByteArray()
+}

core/generated-sources/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/image.kt

@@ -0,0 +1,60 @@
+package org.jetbrains.kotlinx.dataframe.impl.io
+
+import java.awt.Graphics2D
+import java.awt.RenderingHints
+import java.awt.image.BufferedImage
+import java.awt.image.ImageObserver
+import java.io.ByteArrayOutputStream
+import javax.imageio.ImageIO
+import kotlin.math.max
+import kotlin.math.min
+
+internal fun BufferedImage.resizeKeepingAspectRatio(
+    maxSize: Int,
+    resultImageType: Int = BufferedImage.TYPE_INT_ARGB,
+    interpolation: Any = RenderingHints.VALUE_INTERPOLATION_NEAREST_NEIGHBOR,
+    renderingQuality: Any = RenderingHints.VALUE_RENDER_QUALITY,
+    antialiasing: Any = RenderingHints.VALUE_ANTIALIAS_ON,
+    observer: ImageObserver? = null
+): BufferedImage {
+    val aspectRatio = width.toDouble() / height.toDouble()
+    val size = min(maxSize, max(width, height))
+
+    val (nWidth, nHeight) = if (width > height) {
+        Pair(size, (size / aspectRatio).toInt())
+    } else {
+        Pair((size * aspectRatio).toInt(), size)
+    }
+
+    return resize(nWidth, nHeight, resultImageType, interpolation, renderingQuality, antialiasing, observer)
+}
+
+internal fun BufferedImage.resize(
+    width: Int,
+    height: Int,
+    resultImageType: Int = BufferedImage.TYPE_INT_ARGB,
+    interpolation: Any = RenderingHints.VALUE_INTERPOLATION_NEAREST_NEIGHBOR,
+    renderingQuality: Any = RenderingHints.VALUE_RENDER_QUALITY,
+    antialiasing: Any = RenderingHints.VALUE_ANTIALIAS_ON,
+    observer: ImageObserver? = null
+): BufferedImage {
+    val resized = BufferedImage(width, height, resultImageType)
+    val g: Graphics2D = resized.createGraphics()
+
+    g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, interpolation)
+    g.setRenderingHint(RenderingHints.KEY_RENDERING, renderingQuality)
+    g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, antialiasing)
+
+    g.drawImage(this, 0, 0, width, height, observer)
+    g.dispose()
+
+    return resized
+}
+
+internal const val DEFAULT_IMG_FORMAT = "png"
+
+internal fun BufferedImage.toByteArray(format: String = DEFAULT_IMG_FORMAT): ByteArray =
+    ByteArrayOutputStream().use { bos ->
+        ImageIO.write(this, format, bos)
+        bos.toByteArray()
+    }

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/BytesUtils.kt

@@ -0,0 +1,5 @@
+package org.jetbrains.kotlinx.dataframe.impl.io
+
+import java.util.Base64
+
+internal fun ByteArray.toBase64(): String = Base64.getEncoder().encodeToString(this)

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/compression.kt

@@ -0,0 +1,11 @@
+package org.jetbrains.kotlinx.dataframe.impl.io
+
+import java.io.ByteArrayOutputStream
+import java.util.zip.GZIPOutputStream
+
+internal fun ByteArray.encodeGzip(): ByteArray {
+    val bos = ByteArrayOutputStream()
+    GZIPOutputStream(bos).use { it.write(this) }
+
+    return bos.toByteArray()
+}

core/src/main/kotlin/org/jetbrains/kotlinx/dataframe/impl/io/image.kt

@@ -0,0 +1,60 @@
+package org.jetbrains.kotlinx.dataframe.impl.io
+
+import java.awt.Graphics2D
+import java.awt.RenderingHints
+import java.awt.image.BufferedImage
+import java.awt.image.ImageObserver
+import java.io.ByteArrayOutputStream
+import javax.imageio.ImageIO
+import kotlin.math.max
+import kotlin.math.min
+
+internal fun BufferedImage.resizeKeepingAspectRatio(
+    maxSize: Int,
+    resultImageType: Int = BufferedImage.TYPE_INT_ARGB,
+    interpolation: Any = RenderingHints.VALUE_INTERPOLATION_NEAREST_NEIGHBOR,
+    renderingQuality: Any = RenderingHints.VALUE_RENDER_QUALITY,
+    antialiasing: Any = RenderingHints.VALUE_ANTIALIAS_ON,
+    observer: ImageObserver? = null
+): BufferedImage {
+    val aspectRatio = width.toDouble() / height.toDouble()
+    val size = min(maxSize, max(width, height))
+
+    val (nWidth, nHeight) = if (width > height) {
+        Pair(size, (size / aspectRatio).toInt())
+    } else {
+        Pair((size * aspectRatio).toInt(), size)
+    }
+
+    return resize(nWidth, nHeight, resultImageType, interpolation, renderingQuality, antialiasing, observer)
+}
+
+internal fun BufferedImage.resize(
+    width: Int,
+    height: Int,
+    resultImageType: Int = BufferedImage.TYPE_INT_ARGB,
+    interpolation: Any = RenderingHints.VALUE_INTERPOLATION_NEAREST_NEIGHBOR,
+    renderingQuality: Any = RenderingHints.VALUE_RENDER_QUALITY,
+    antialiasing: Any = RenderingHints.VALUE_ANTIALIAS_ON,
+    observer: ImageObserver? = null
+): BufferedImage {
+    val resized = BufferedImage(width, height, resultImageType)
+    val g: Graphics2D = resized.createGraphics()
+
+    g.setRenderingHint(RenderingHints.KEY_INTERPOLATION, interpolation)
+    g.setRenderingHint(RenderingHints.KEY_RENDERING, renderingQuality)
+    g.setRenderingHint(RenderingHints.KEY_ANTIALIASING, antialiasing)
+
+    g.drawImage(this, 0, 0, width, height, observer)
+    g.dispose()
+
+    return resized
+}
+
+internal const val DEFAULT_IMG_FORMAT = "png"
+
+internal fun BufferedImage.toByteArray(format: String = DEFAULT_IMG_FORMAT): ByteArray =
+    ByteArrayOutputStream().use { bos ->
+        ImageIO.write(this, format, bos)
+        bos.toByteArray()
+    }