From c5a9dc7288982d8711cf7d363ad641ce2deea6ed Mon Sep 17 00:00:00 2001
From: dece <shgck@pistache.land>
Date: Tue, 4 Apr 2023 18:58:12 +0200
Subject: [PATCH] MainActivity: move to FFmpegKit

---
 .../harvestdawn/MainActivity.kt               | 54 +++++++++++--------
 1 file changed, 33 insertions(+), 21 deletions(-)

diff --git a/app/src/main/java/io/lowrespalmtree/harvestdawn/MainActivity.kt b/app/src/main/java/io/lowrespalmtree/harvestdawn/MainActivity.kt
index 25660a4..b98d789 100644
--- a/app/src/main/java/io/lowrespalmtree/harvestdawn/MainActivity.kt
+++ b/app/src/main/java/io/lowrespalmtree/harvestdawn/MainActivity.kt
@@ -21,15 +21,15 @@ import android.widget.VideoView
 import androidx.appcompat.app.AppCompatActivity
 import androidx.core.content.FileProvider
 import androidx.core.net.toUri
-import com.arthenica.mobileffmpeg.Config.RETURN_CODE_CANCEL
-import com.arthenica.mobileffmpeg.Config.RETURN_CODE_SUCCESS
-import com.arthenica.mobileffmpeg.FFmpeg
-import com.arthenica.mobileffmpeg.FFprobe
+import com.arthenica.ffmpegkit.FFmpegKit
+import com.arthenica.ffmpegkit.FFprobeKit
+import com.arthenica.ffmpegkit.ReturnCode
 import com.google.android.material.floatingactionbutton.FloatingActionButton
 import java.io.File
 import java.io.FileInputStream
 import java.io.FileOutputStream
 import java.lang.ref.WeakReference
+import kotlin.math.roundToLong
 
 class MainActivity : AppCompatActivity() {
     private val spinner: ProgressBar? get() = findViewById(R.id.progressBar)
@@ -171,25 +171,26 @@ class MainActivity : AppCompatActivity() {
         currentVideoPath = null
         currentVideoUri = null
 
-        val mediaInfo = FFprobe.getMediaInformation(inputVideoPath)
         // Detect vertical/horizontal resolution, because at the moment we downscale everything to
         // fit Telegram's media preview requirements. If FFprobe fails to returns data, we consider
         // by default the video to be vertical.
-        val isVertical = mediaInfo?.let {
-            it.streams.getOrNull(0)?.let { streamInformation ->
+        val session = FFprobeKit.getMediaInformation(inputVideoPath)
+        var isVertical = true
+        var duration: Double = 0.0
+        if (ReturnCode.isSuccess(session.returnCode)) {
+            session.mediaInformation.streams.getOrNull(0)?.let { streamInformation ->
                 if (streamInformation.height != null && streamInformation.width != null)
-                    streamInformation.height > streamInformation.width
-                else
-                    true
+                    isVertical = streamInformation.height > streamInformation.width
             }
-        } ?: true
-        val duration = mediaInfo?.duration?.toLong() ?: 0L
+            duration = session.mediaInformation.duration.toDouble()
+        }
 
         // Save the soundtrack as well cause we can't pass a resource to FFMPEG.
         // Should be done only once, then it's kept in app internal storage.
         val themeId = getThemeId()
         val soundFile = File(filesDir, "$themeId.m4a")
         if (!soundFile.exists()) {
+            Log.i(TAG, "Prepare sound file with a filesystem path.")
             resources.openRawResource(getThemeRawResource(themeId)).use { resource ->
                 FileOutputStream(soundFile).use { fos ->
                     fos.buffered().write(resource.buffered().readBytes())
@@ -224,7 +225,7 @@ class MainActivity : AppCompatActivity() {
 
     private class MixTask(
         private val activity: WeakReference<MainActivity>,
-        private val duration: Long,
+        private val duration: Double,
         private val videoIsVertical: Boolean
     ): AsyncTask<String, Void, Boolean>() {
         private lateinit var videoPath: String
@@ -238,8 +239,12 @@ class MainActivity : AppCompatActivity() {
             videoPath = params[0]!!
             val audioPath = params[1]!!
             outputPath = params[2]!!
-            val width = if (videoIsVertical) 480 else 640
-            val durationOpt = if (duration > 0) "-t ${duration}ms" else ""
+            //val width = if (videoIsVertical) 480 else 640
+            val durationOpt = if (duration > 0) {
+                "-t ${(duration * 1000).roundToLong()}ms"
+            } else {
+                ""
+            }
             val command = (
                 "-i $videoPath -i $audioPath"
                 + " -filter_complex amix=duration=longest"
@@ -248,20 +253,27 @@ class MainActivity : AppCompatActivity() {
                 + " $durationOpt -y $outputPath"
             )
             Log.d(TAG, "Calling FFmpeg with command: $command")
-            return when (val rc = FFmpeg.execute(command)) {
-                RETURN_CODE_SUCCESS -> true.also { Log.i(TAG, "Mix succeeded.") }
-                RETURN_CODE_CANCEL -> false.also { Log.i(TAG, "Mix cancelled.") }
-                else -> false .also { Log.e(TAG, "Mix failed! rc = $rc") }
+            val rc = FFmpegKit.execute(command).returnCode
+            if (ReturnCode.isSuccess(rc)) {
+                Log.i(TAG, "Mix succeeded.")
+                return true
             }
+            if (ReturnCode.isCancel(rc))
+                Log.i(TAG, "Mix cancelled.")
+            else
+                Log.e(TAG, "Mix failed! rc = $rc")
+            return false
         }
 
         override fun onPostExecute(result: Boolean?) {
             File(videoPath).delete()
             activity.get()?.let {
-                if (result == true)
+                if (result == true) {
                     it.handleFfmpegOutput(outputPath)
-                else
+                } else {
+                    it.toast("Crasse de brin !!!")
                     it.setLoading(false)
+                }
             }
         }
     }