Joren/pokesearch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add bin search

Browse Source
This commit is contained in:
Joren
2024-04-30 02:02:39 +02:00
parent fd122ec376
commit 84e801ed01
1 changed files with 18 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
app/src/main/java/com/ti/mobpo/ui/viewmodels/PokeSearchViewModel.kt
View File

@@ -39,10 +39,11 @@ class PokeSearchViewModel(private val favouritesRepository: FavouritesRepository
fun search(query: String) {
viewModelScope.launch {
try {
val filteredList = _initialPokemonList.value?.filter { it.name.startsWith(query, ignoreCase = true) }
val filteredList = _initialPokemonList.value?.binarySearch(query)
val detailsList = mutableListOf<PokemonDetails>()
if (filteredList != null) {
for (pokemonSpecies in filteredList.take(SHOW_LIMIT)) {
for (index in filteredList until minOf(_initialPokemonList.value?.size ?: 0, filteredList + SHOW_LIMIT)) {
val pokemonSpecies = _initialPokemonList.value!![index]
val details = service.getPokemonDetails(extractPokemonId(pokemonSpecies.url))
val isFavorite = favouritesRepository.isFavourite(details.id)
detailsList.add(details.copy(isFavorite = isFavorite))
@@ -55,6 +56,21 @@ class PokeSearchViewModel(private val favouritesRepository: FavouritesRepository
}
}
private fun List<PokemonSpecies>.binarySearch(query: String): Int {
var low = 0
var high = this.size - 1
while (low <= high) {
val mid = (low + high) ushr 1
val comparison = this[mid].name.compareTo(query, ignoreCase = true)
when {
comparison == 0 -> return mid
comparison < 0 -> low = mid + 1
else -> high = mid - 1
}
}
return low
}
private fun extractPokemonId(url: String): Int {
val parts = url.split("/")
return parts[parts.size - 2].toInt()