From 8e5f14d70e3f765552d6c932f42c7d0059c6be1d Mon Sep 17 00:00:00 2001
From: Ghost <>
Date: Fri, 20 Jun 2025 01:03:02 -0500
Subject: [PATCH] Added a sort by codec feature to animetorrentio extension
 (#961)

Checklist:

- [x] Updated `extVersionCode` value in `build.gradle` for individual extensions
- [ ] Updated `overrideVersionCode` or `baseVersionCode` as needed for all multisrc extensions
- [ ] Referenced all related issues in the PR body (e.g. "Closes #xyz")
- [ ] Added the `isNsfw = true` flag in `build.gradle` when appropriate
- [x] Have not changed source names
- [x] Have explicitly kept the `id` if a source's name or language were changed
- [x] Have tested the modifications by compiling and running the extension through Android Studio
- [ ] Have removed `web_hi_res_512.png` when adding a new extension
- [ ] Have made sure all the icons are in png format

Co-authored-by: worldInColors <almonzer1234567@gmail.com>
Reviewed-on: https://kohiden.xyz/Kohi-den/extensions-source/pulls/961
Co-authored-by: Ghost <>
Co-committed-by: Ghost <>
---
 src/all/torrentioanime/build.gradle           |  2 +-
 .../all/torrentioanime/Torrentio.kt           | 74 +++++++++++++++++--
 2 files changed, 68 insertions(+), 8 deletions(-)

diff --git a/src/all/torrentioanime/build.gradle b/src/all/torrentioanime/build.gradle
index 2245029d..d1e0fd47 100644
--- a/src/all/torrentioanime/build.gradle
+++ b/src/all/torrentioanime/build.gradle
@@ -1,7 +1,7 @@
 ext {
     extName = 'Torrentio Anime (Torrent / Debrid)'
     extClass = '.Torrentio'
-    extVersionCode = 16
+    extVersionCode = 17
     containsNsfw = false
 }
 
diff --git a/src/all/torrentioanime/src/eu/kanade/tachiyomi/animeextension/all/torrentioanime/Torrentio.kt b/src/all/torrentioanime/src/eu/kanade/tachiyomi/animeextension/all/torrentioanime/Torrentio.kt
index 2b30443a..1f10183a 100644
--- a/src/all/torrentioanime/src/eu/kanade/tachiyomi/animeextension/all/torrentioanime/Torrentio.kt
+++ b/src/all/torrentioanime/src/eu/kanade/tachiyomi/animeextension/all/torrentioanime/Torrentio.kt
@@ -456,18 +456,50 @@ class Torrentio : ConfigurableAnimeSource, AnimeHttpSource() {
         }.orEmpty()
     }
 
-    override fun List<Video>.sort(): List<Video> {
-        val isDub = preferences.getBoolean(IS_DUB_KEY, IS_DUB_DEFAULT)
-        val isEfficient = preferences.getBoolean(IS_EFFICIENT_KEY, IS_EFFICIENT_DEFAULT)
-
-        return sortedWith(
+ override fun List<Video>.sort(): List<Video> {
+    val isDub = preferences.getBoolean(IS_DUB_KEY, IS_DUB_DEFAULT)
+    val isEfficient = preferences.getBoolean(IS_EFFICIENT_KEY, IS_EFFICIENT_DEFAULT)
+    val codecPreferences = preferences.getStringSet(PREF_CODEC_KEY, PREF_CODEC_DEFAULT) ?: setOf()
+    
+    return if (codecPreferences.isNotEmpty()) {
+        // Filter to only show videos matching selected codecs
+        filter { video ->
+            codecPreferences.any { codec ->
+                when (codec) {
+                    "x264" -> video.quality.contains("264", true) || 
+                             (!video.quality.contains("265", true) && 
+                              !video.quality.contains("hevc", true) && 
+                              !video.quality.contains("av1", true) && 
+                              !video.quality.contains("vp9", true))
+                    "x265" -> video.quality.contains("265", true) || 
+                             video.quality.contains("hevc", true)
+                    "av1" -> video.quality.contains("av1", true)
+                    "vp9" -> video.quality.contains("vp9", true)
+                    "other" -> !video.quality.contains("264", true) && 
+                               !video.quality.contains("265", true) && 
+                               !video.quality.contains("hevc", true) && 
+                               !video.quality.contains("av1", true) && 
+                               !video.quality.contains("vp9", true)
+                    else -> false
+                }
+            }
+        }.sortedWith(
+            compareBy(
+                { Regex("\\[(.+?) download]").containsMatchIn(it.quality) },
+                { isDub && !it.quality.contains("dubbed", true) }
+            )
+        )
+    } else {
+        // If no codec preferences, use old sorting logic
+        sortedWith(
             compareBy(
                 { Regex("\\[(.+?) download]").containsMatchIn(it.quality) },
                 { isDub && !it.quality.contains("dubbed", true) },
                 { isEfficient && !arrayOf("hevc", "265", "av1").any { q -> it.quality.contains(q, true) } },
-            ),
+            )
         )
     }
+}
 
     private fun fetchTrackers(): String {
         val request = Request.Builder()
@@ -616,6 +648,18 @@ class Torrentio : ConfigurableAnimeSource, AnimeHttpSource() {
             }
             summary = "Codec: (HEVC / x265)  & AV1. High-quality video with less data usage."
         }.also(screen::addPreference)
+		MultiSelectListPreference(screen.context).apply {
+    key = PREF_CODEC_KEY
+    title = "Preferred Codecs"
+    entries = PREF_CODEC
+    entryValues = PREF_CODEC_VALUE
+    setDefaultValue(PREF_CODEC_DEFAULT)
+
+    setOnPreferenceChangeListener { _, newValue ->
+        @Suppress("UNCHECKED_CAST")
+        preferences.edit().putStringSet(key, newValue as Set<String>).commit()
+    }
+}.also(screen::addPreference)
     }
 
     companion object {
@@ -897,7 +941,22 @@ class Torrentio : ConfigurableAnimeSource, AnimeHttpSource() {
 
         private const val IS_EFFICIENT_KEY = "efficient"
         private const val IS_EFFICIENT_DEFAULT = false
-
+		private const val PREF_CODEC_KEY = "codec_selection"
+private val PREF_CODEC = arrayOf(
+    "x264",
+    "x265/HEVC",
+    "AV1",
+    "VP9",
+    "Other"
+)
+private val PREF_CODEC_VALUE = arrayOf(
+    "x264",
+    "x265",
+    "av1",
+    "vp9",
+    "other"
+)
+private val PREF_CODEC_DEFAULT = setOf<String>() // Empty by default to show al
         private val DATE_TIME_FORMATTER by lazy {
             SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss.SSS'Z'", Locale.ENGLISH)
         }
@@ -905,5 +964,6 @@ class Torrentio : ConfigurableAnimeSource, AnimeHttpSource() {
         private val DATE_FORMATTER by lazy {
             SimpleDateFormat("yyyy-MM-dd", Locale.ENGLISH)
         }
+		
     }
 }