From b46da2f58a57acb2e6cf566b71d1ce98bed7c9f8 Mon Sep 17 00:00:00 2001
From: tibbi <tibor@kaputa.sk>
Date: Mon, 20 Feb 2017 21:59:55 +0100
Subject: [PATCH] try speeding up the process of obtaining all images and
 videos

---
 .../gallery/asynctasks/GetMediaAsynctask.kt      | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/app/src/main/kotlin/com/simplemobiletools/gallery/asynctasks/GetMediaAsynctask.kt b/app/src/main/kotlin/com/simplemobiletools/gallery/asynctasks/GetMediaAsynctask.kt
index e51909a40..5bfdf5075 100644
--- a/app/src/main/kotlin/com/simplemobiletools/gallery/asynctasks/GetMediaAsynctask.kt
+++ b/app/src/main/kotlin/com/simplemobiletools/gallery/asynctasks/GetMediaAsynctask.kt
@@ -2,9 +2,6 @@ package com.simplemobiletools.gallery.asynctasks
 
 import android.content.Context
 import android.os.AsyncTask
-import com.simplemobiletools.commons.extensions.isGif
-import com.simplemobiletools.commons.extensions.isImageFast
-import com.simplemobiletools.commons.extensions.isVideoFast
 import com.simplemobiletools.gallery.extensions.config
 import com.simplemobiletools.gallery.extensions.getParents
 import com.simplemobiletools.gallery.helpers.IMAGES
@@ -20,6 +17,8 @@ class GetMediaAsynctask(val context: Context, val mPath: String, val isPickVideo
     var config = context.config
     var showMedia = IMAGES_AND_VIDEOS
     var fileSorting = 0
+    val photoExtensions = arrayOf("jpg", "png", "jpeg", "bmp", "webp", "tiff")
+    val videoExtensions = arrayOf("webm", "mkv", "flv", "vob", "avi", "wmv", "mp4", "ogv", "qt", "m4p", "mpg", "m4v", "mp2", "mpeg", "3gp")
 
     override fun onPreExecute() {
         super.onPreExecute()
@@ -49,8 +48,9 @@ class GetMediaAsynctask(val context: Context, val mPath: String, val isPickVideo
         val dir = File(path)
         val files = dir.listFiles() ?: return media
         for (file in files) {
-            val isImage = file.isImageFast() || file.isGif()
-            val isVideo = if (isImage) false else file.isVideoFast()
+            val filePath = file.absolutePath
+            val isImage = localIsImage(filePath) || localIsGif(filePath)
+            val isVideo = if (isImage) false else localIsVideo(filePath)
 
             if (!isImage && !isVideo)
                 continue
@@ -68,11 +68,15 @@ class GetMediaAsynctask(val context: Context, val mPath: String, val isPickVideo
             val name = file.name
             val absolutePath = file.absolutePath
             val dateModified = file.lastModified()
-            media.add(Medium(name, absolutePath, isVideo, dateModified, dateModified, size))
+            media.add(Medium(name, absolutePath, isVideo, dateModified, dateModified, 0))
         }
         return media
     }
 
+    private fun localIsImage(path: String) = photoExtensions.any { path.endsWith(".$it", true) }
+    private fun localIsGif(path: String) = path.endsWith(".gif", true)
+    private fun localIsVideo(path: String) = videoExtensions.any { path.endsWith(".$it", true) }
+
     override fun onPostExecute(media: ArrayList<Medium>) {
         super.onPostExecute(media)
         callback.invoke(media)