Add PiktImageProcessingTest

Author: iamgio

core/src/test/kotlin/eu/iamgio/pikt/tests/PiktImageProcessingTest.kt

@@ -0,0 +1,53 @@
+package eu.iamgio.pikt.tests
+
+import eu.iamgio.pikt.command.commands.imageprocessing.MaskCommand
+import eu.iamgio.pikt.image.rgbToHex
+import org.junit.jupiter.api.Test
+import java.awt.image.BufferedImage
+import kotlin.test.assertTrue
+
+// These tests apply image processing filters to the same image (/imageprocessing-tests/source.png)
+// which generates the following pseudocode:
+// var str = "Pikt"
+// if true
+//     print str
+
+private const val EXPECTED_RESULT = "Pikt" // Expected result from all the programs.
+
+class PiktImageProcessingTest {
+
+    private val launcher = PiktImageProcessingTestLauncher()
+
+    // Launches a Pikt program.
+    private fun launch() = launcher.launch()
+
+    // Copies a resource from /imageprocessing-tests/ to the temp folder.
+    private fun copy(name: String) = launcher.copy(name)
+
+    // Gets the file path from file name within the temp folder.
+    private fun pathify(name: String) = launcher.pathify(name)
+
+    // Checks whether the transformed image outputs the correct result.
+    private fun checkCorrectOutput() = assertTrue { launch().firstOrNull() == EXPECTED_RESULT }
+
+    // Latest generated image
+    private val outImage: BufferedImage
+        get() = launcher.getOutputImage()
+
+    // Applies a mask image (-mask=<path>)
+    @Test
+    fun mask() {
+        copy("face-mask.png")
+        MaskCommand().fire(pathify("face-mask.png"))
+        checkCorrectOutput()
+
+        with(outImage) {
+            assertTrue {
+                getRGB(0, 0).rgbToHex() == "FFFFFF" // Top left corner, not in mask
+                        && getRGB(3, 4).rgbToHex() == "FF0000" // Left eye
+                        && getRGB(9, 4).rgbToHex() == "505050" // Right eye
+                        && getRGB(6, 7).rgbToHex() == "FF65FF" // Mouth
+            }
+        }
+    }
+}

core/src/test/kotlin/eu/iamgio/pikt/tests/PiktImageProcessingTestLauncher.kt

@@ -0,0 +1,55 @@
+package eu.iamgio.pikt.tests
+
+import eu.iamgio.pikt.command.Command
+import eu.iamgio.pikt.command.commands.imageprocessing.ImageOutputCommand
+import java.awt.image.BufferedImage
+import java.io.File
+import java.io.FileInputStream
+import java.io.InputStream
+import java.nio.file.Files
+import java.nio.file.StandardCopyOption
+import javax.imageio.ImageIO
+
+private const val TESTS_FOLDER = "/imageprocessing-tests"
+private const val SOURCE = "source.png"
+private const val OUT = "out.png"
+
+fun Command.fire(args: String? = null) = this.action(args)
+
+class PiktImageProcessingTestLauncher : PiktTestLauncher() {
+    private val folder: File = File(tempDirectory, "imageprocessing")
+    private val sourceFilePath: String
+        get() = folder.absolutePath + File.separator + OUT
+
+    init {
+        folder.mkdirs()
+        println("Storing to: $folder")
+
+        System.setProperty("source", sourceFilePath)
+        // Set output file
+        ImageOutputCommand().fire(sourceFilePath)
+        copy("source.png")
+    }
+
+    fun copy(name: String) {
+        Files.copy(
+                PiktImageProcessingTest::class.java.getResourceAsStream("$TESTS_FOLDER/$name")!!,
+                File(folder, name).toPath(),
+                StandardCopyOption.REPLACE_EXISTING
+        )
+    }
+
+    fun getOutputImage(): BufferedImage = ImageIO.read(getImage(""))
+
+    fun launch() = launch("")
+
+    fun pathify(name: String) = folder.absolutePath + File.separator + name
+
+    override fun getImage(name: String): InputStream {
+        return FileInputStream(sourceFilePath)
+    }
+
+    override fun getColorScheme(colorSchemeName: String): InputStream {
+        return PiktTest::class.java.getResourceAsStream("/schemes/$colorSchemeName.properties")!!
+    }
+}

core/src/test/kotlin/eu/iamgio/pikt/tests/PiktTest.kt

@@ -1,12 +1,22 @@
 package eu.iamgio.pikt.tests
 
 import org.junit.jupiter.api.Test
+import java.io.InputStream
 import kotlin.test.assertTrue
 
+private class InternalTestLauncher : PiktTestLauncher() {
+    override fun getImage(name: String): InputStream {
+        return PiktTest::class.java.getResourceAsStream("/$name.png")!!
+    }
+
+    override fun getColorScheme(colorSchemeName: String): InputStream {
+        return PiktTest::class.java.getResourceAsStream("/schemes/$colorSchemeName.properties")!!
+    }
+}
 
 class PiktTest {
 
-    private val launcher = PiktTestLauncher()
+    private val launcher = InternalTestLauncher()
     private fun launch(name: String, scheme: String? = null) = launcher.launch(name, scheme)
 
     @Test

core/src/test/kotlin/eu/iamgio/pikt/tests/PiktTestLauncher.kt

@@ -9,6 +9,7 @@ import eu.iamgio.pikt.properties.PiktProperties
 import eu.iamgio.pikt.properties.PiktPropertiesRetriever
 import eu.iamgio.pikt.registerStatements
 import java.io.File
+import java.io.InputStream
 import javax.imageio.ImageIO
 
 
@@ -17,15 +18,27 @@ import javax.imageio.ImageIO
  *
  * @author Giorgio Garofalo
  */
-class PiktTestLauncher {
+abstract class PiktTestLauncher {
 
-    private val tempDirectory = File(System.getProperty("java.io.tmpdir") + File.separator + "pikt-test")
+    protected val tempDirectory = File(System.getProperty("java.io.tmpdir") + File.separator + "pikt-test")
 
     init {
         tempDirectory.mkdir()
         registerStatements()
     }
 
+    /**
+     * @param name name of the source image
+     * @return source image as [InputStream]
+     */
+    abstract fun getImage(name: String): InputStream
+
+    /**
+     * @param colorSchemeName name of the color scheme
+     * @return color scheme as [InputStream]
+     */
+    abstract fun getColorScheme(colorSchemeName: String): InputStream
+
     /**
      * Launches the interpreter and returns all the non-error messages.
      * @param name PNG image name without extension
@@ -53,14 +66,14 @@ class PiktTestLauncher {
                 nativeCompilerPath = null,
                 colors = ColorsPropertiesRetriever(libraries).also {
                     if(colorSchemeName != null) {
-                        it.loadProperties(PiktTest::class.java.getResourceAsStream("/schemes/$colorSchemeName.properties")!!)
+                        it.loadProperties(getColorScheme(colorSchemeName))
                     }
                 }.retrieve()
         )
 
         val lines = mutableListOf<String>()
 
-        val image = PiktImage(ImageIO.read(PiktTest::class.java.getResourceAsStream("/$name.png")), properties.colors)
+        val image = PiktImage(ImageIO.read(getImage(name)), properties.colors)
 
         val evaluator = Evaluator()
         evaluator.evaluate(image, properties.libraries)

core/src/test/resources/imageprocessing-tests/face-mask.png

@@ -0,0 +1,16 @@
+<component name="ProjectRunConfigurationManager">
+  <configuration default="false" name="PiktImageProcessingTest" type="JUnit" factoryName="JUnit" nameIsGenerated="true">
+    <module name="core" />
+    <extension name="coverage">
+      <pattern>
+        <option name="PATTERN" value="eu.iamgio.pikt.tests.*" />
+        <option name="ENABLED" value="true" />
+      </pattern>
+    </extension>
+    <option name="PACKAGE_NAME" value="eu.iamgio.pikt.tests" />
+    <option name="MAIN_CLASS_NAME" value="eu.iamgio.pikt.tests.PiktImageProcessingTest" />
+    <option name="METHOD_NAME" value="" />
+    <option name="TEST_OBJECT" value="class" />
+    <option name="VM_PARAMETERS" value="-Djvmcompiler=SET_HERE -Dlib=../stdlib/target/stdlib.jar" />
+  </configuration>
+</component>