Bimba.git

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

package xyz.apiote.bimba.czwek.search.ui

import androidx.lifecycle.LiveData
import androidx.lifecycle.MutableLiveData
import androidx.lifecycle.ViewModel
import dev.bandb.graphview.graph.Graph
import dev.bandb.graphview.graph.Node
import xyz.apiote.bimba.czwek.repo.LineGraph

class PageViewModel : ViewModel() {

	private val _data = MutableLiveData<Graph>()
	val data: LiveData<Graph> = _data

	fun setupGraphView(lineGraph: LineGraph) {
		val graph = Graph()
		val nodes = lineGraph.stops.map { Node(it) }
		lineGraph.nextNodes.filter { it.key != -1L }.forEach { (from, tos) ->
			tos.filter { it != -1L }.forEach { to ->
				graph.addEdge(nodes[from.toInt()], nodes[to.toInt()])
			}
		}
		_data.value = graph
	}
}