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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

package ml.adamsprogs.bimba.models

import java.util.*

data class Departure(val line: String, val mode: String, val time: String, val lowFloor: Boolean,
                     val modification: String?, val direction: String, val vm: Boolean = false,
                     var tomorrow: Boolean = false, val onStop: Boolean = false) {

    override fun toString(): String {
        return "$line|$mode|$time|$lowFloor|$modification|$direction|$vm|$tomorrow|$onStop"
    }

    fun copy(): Departure {
        return Departure.fromString(this.toString())
    }

    companion object {
        private fun filterDepartures(departures: List<Departure>?): ArrayList<Departure> {
            val filtered = ArrayList<Departure>()
            val lines = HashMap<String, Int>()
            val now = Calendar.getInstance()
            for (departure in departures!!) {
                val time = Calendar.getInstance()
                time.set(Calendar.HOUR_OF_DAY, Integer.parseInt(departure.time.split(":")[0]))
                time.set(Calendar.MINUTE, Integer.parseInt(departure.time.split(":")[1]))
                time.set(Calendar.SECOND, 0)
                time.set(Calendar.MILLISECOND, 0)
                if (departure.tomorrow)
                    time.add(Calendar.DAY_OF_MONTH, 1)
                var lineExistedTimes = lines[departure.line]
                if ((now.before(time) || now == time) && lineExistedTimes ?: 0 < 3) {
                    lineExistedTimes = (lineExistedTimes ?: 0) + 1
                    lines[departure.line] = lineExistedTimes
                    filtered.add(departure)
                }
            }
            return filtered
        }

        fun createDepartures(stopId: String): HashMap<String, ArrayList<Departure>> {
            val timetable = Timetable.getTimetable()
            val departures = timetable.getStopDepartures(stopId)
            val moreDepartures = timetable.getStopDepartures(stopId)
            val rolledDepartures = HashMap<String, ArrayList<Departure>>()

            for ((_, tomorrowDepartures) in moreDepartures) {
                tomorrowDepartures.forEach { it.tomorrow = true }
            }

            for ((mode, _) in departures) {
                rolledDepartures[mode] = (departures[mode] as ArrayList<Departure> +
                        moreDepartures[mode] as ArrayList<Departure>) as ArrayList<Departure>
                rolledDepartures[mode] = filterDepartures(rolledDepartures[mode])
            }

            return rolledDepartures
        }

        fun fromString(string: String): Departure {
            val array = string.split("|")
            return Departure(array[0], array[1], array[2], array[3] == "1", array[4], array[5],
                    array[6] == "true", array[7] == "true", array[8] == "true")
        }
    }

    fun timeTill(): Long {
        val time = Calendar.getInstance()
        time.set(Calendar.HOUR_OF_DAY, Integer.parseInt(this.time.split(":")[0]))
        time.set(Calendar.MINUTE, Integer.parseInt(this.time.split(":")[1]))
        time.set(Calendar.SECOND, 0)
        time.set(Calendar.MILLISECOND, 0)
        val now = Calendar.getInstance()
        if (this.tomorrow)
            time.add(Calendar.DAY_OF_MONTH, 1)
        return (time.timeInMillis - now.timeInMillis) / (1000 * 60)
    }
}