From 2a433a0cbe30da9ec2d502fb0c77bebb9cfce853 Mon Sep 17 00:00:00 2001
From: minjaesong <alswo9628@gmail.com>
Date: Tue, 21 Dec 2021 00:03:13 +0900
Subject: [PATCH] get mugshot for assemblying sprites

---
 .../spriteassembler/AssembleSheetPixmap.kt    | 61 +++++++++++++++----
 .../modulebasegame/gameactors/IngamePlayer.kt |  6 +-
 2 files changed, 53 insertions(+), 14 deletions(-)

diff --git a/src/net/torvald/spriteassembler/AssembleSheetPixmap.kt b/src/net/torvald/spriteassembler/AssembleSheetPixmap.kt
index 5d1a43cf6..6f17dbcd8 100644
--- a/src/net/torvald/spriteassembler/AssembleSheetPixmap.kt
+++ b/src/net/torvald/spriteassembler/AssembleSheetPixmap.kt
@@ -23,6 +23,9 @@ import java.util.*
  */
 object AssembleSheetPixmap {
 
+    const val MUGSHOT_PIXMAP_W = 48
+    const val MUGSHOT_PIXMAP_H = 48
+
     /**
      * The name of the Bodypart here may or may not be case-sensitive (depends on your actual filesystem -- NTFS, APFS, Ext4, ...)
      */
@@ -65,28 +68,64 @@ object AssembleSheetPixmap {
         return drawAndGetCanvas(properties, getVirtualDiskFileGetter(bodypartMapping, disk))
     }
 
-    fun getPartTex(getFile: (String) -> InputStream?, partName: String): TextureRegion? {
-        (getFile(partName) ?: throw FileNotFoundException("file for '$partName' is not found")).let {
+    fun getPartPixmap(getFile: (String) -> InputStream?, partName: String): Pixmap? {
+        getFile(partName)?.let {
             val bytes = it.readAllBytes()
-            val pixmap = Pixmap(bytes, 0, bytes.size)
-            val tr = TextureRegion(Texture(pixmap))
-            pixmap.dispose()
-            return tr
-
+            return Pixmap(bytes, 0, bytes.size)
         }
         return null
     }
 
-    fun getHeadFromAssetsDir(properties: ADProperties): TextureRegion? {
+    fun getMugshotFromAssetsDir(properties: ADProperties): TextureRegion? {
         // TODO assemble from HAIR_FORE (optional), HAIR (optional) then HEAD (mandatory)
-        return getPartTex(getAssetsDirFileGetter(properties), "HEAD")
+        val getter = getAssetsDirFileGetter(properties)
+        val headPixmap = getPartPixmap(getter, "HEAD")
+        val hairPixmap = getPartPixmap(getter, "HAIR")
+        val hair2Pixmap = getPartPixmap(getter, "HAIR_FORE")
+
+        if (headPixmap == null) throw FileNotFoundException("Bodyparts file of HEAD is not found!")
+        return composeMugshot(properties, headPixmap, hairPixmap, hair2Pixmap)
     }
 
-    fun getHeadFromVirtualDisk(disk: SimpleFileSystem, entrynum: Long, properties: ADProperties): TextureRegion? {
+    fun getMugshotFromVirtualDisk(disk: SimpleFileSystem, entrynum: Long, properties: ADProperties): TextureRegion? {
         // TODO assemble from HAIR_FORE (optional), HAIR (optional) then HEAD (mandatory)
         val bodypartMapping = Properties()
         bodypartMapping.load(ByteArray64Reader(disk.getFile(entrynum)!!.bytes, Common.CHARSET))
-        return getPartTex(getVirtualDiskFileGetter(bodypartMapping, disk), "HEAD")
+        val getter = getVirtualDiskFileGetter(bodypartMapping, disk)
+        val headPixmap = getPartPixmap(getter, "HEAD")
+        val hairPixmap = getPartPixmap(getter, "HAIR")
+        val hair2Pixmap = getPartPixmap(getter, "HAIR_FORE")
+
+        if (headPixmap == null) throw FileNotFoundException("Bodyparts file of HEAD is not found!")
+        return composeMugshot(properties, headPixmap, hairPixmap, hair2Pixmap)
+    }
+
+    private fun composeMugshot(properties: ADProperties, head: Pixmap, hair: Pixmap?, hair2: Pixmap?): TextureRegion {
+        val canvas = Pixmap(MUGSHOT_PIXMAP_W, MUGSHOT_PIXMAP_H, Pixmap.Format.RGBA8888)
+        val drawX = (canvas.width - head.width) / 2
+        val drawY = (canvas.height - head.height) / 2
+        val headOffset = properties.bodypartJoints["HEAD"]!!
+
+        // TODO shift drawing pos using the properties BODYPARTS
+
+        canvas.drawPixmap(head, drawX, drawY)
+        hair?.let {
+            val offset = properties.bodypartJoints["HAIR"]!! - headOffset
+            canvas.drawPixmap(it, drawX - offset.x, drawY - offset.y)
+        }
+        hair2?.let {
+            val offset = properties.bodypartJoints["HAIR_FORE"]!! - headOffset
+            canvas.drawPixmap(it, drawX - offset.x, drawY - offset.y)
+        }
+
+        val tr = TextureRegion(Texture(canvas))
+
+        canvas.dispose()
+        head.dispose()
+        hair?.dispose()
+        hair2?.dispose()
+
+        return tr
     }
 
     private fun drawThisFrame(frameName: String,
diff --git a/src/net/torvald/terrarum/modulebasegame/gameactors/IngamePlayer.kt b/src/net/torvald/terrarum/modulebasegame/gameactors/IngamePlayer.kt
index d7a680e7b..9991ad841 100644
--- a/src/net/torvald/terrarum/modulebasegame/gameactors/IngamePlayer.kt
+++ b/src/net/torvald/terrarum/modulebasegame/gameactors/IngamePlayer.kt
@@ -85,7 +85,7 @@ class IngamePlayer : ActorHumanoid {
     fun reassembleSprite(sprite: SpriteAnimation?, spriteGlow: SpriteAnimation? = null) {
         if (animDesc != null && sprite != null) {
             _rebuild(animDesc!!, sprite)
-            spriteHeadTexture = AssembleSheetPixmap.getHeadFromAssetsDir(animDesc!!)
+            spriteHeadTexture = AssembleSheetPixmap.getMugshotFromAssetsDir(animDesc!!)
         }
         if (animDescGlow != null && spriteGlow != null)
             _rebuild(animDescGlow!!, spriteGlow)
@@ -97,9 +97,9 @@ class IngamePlayer : ActorHumanoid {
             _rebuild(disk, -1025L, animDesc!!, sprite)
 
             if (disk.getEntry(-1025L) != null)
-                spriteHeadTexture = AssembleSheetPixmap.getHeadFromVirtualDisk(disk, -1025L, animDesc!!)
+                spriteHeadTexture = AssembleSheetPixmap.getMugshotFromVirtualDisk(disk, -1025L, animDesc!!)
             else
-                spriteHeadTexture = AssembleSheetPixmap.getHeadFromAssetsDir(animDesc!!)
+                spriteHeadTexture = AssembleSheetPixmap.getMugshotFromAssetsDir(animDesc!!)
         }
         if (animDescGlow != null && spriteGlow != null)
             _rebuild(disk, -1026L, animDescGlow!!, spriteGlow)