1 files changed, 10 insertions, 0 deletions
diff --git a/androidApp/src/main/java/mx/trackermap/TrackerMap/android/devices/UnitRenderData.kt b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/devices/UnitRenderData.kt
index 39ef472..beeba6f 100644
--- a/androidApp/src/main/java/mx/trackermap/TrackerMap/android/devices/UnitRenderData.kt
+++ b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/devices/UnitRenderData.kt
@@ -13,6 +13,7 @@ import androidx.cardview.widget.CardView
 import androidx.core.content.ContextCompat
 import com.zerobranch.layout.SwipeLayout
 import mx.trackermap.TrackerMap.android.R
+import mx.trackermap.TrackerMap.android.map.MarkerTransformations
 import mx.trackermap.TrackerMap.client.models.UnitInformation
 import mx.trackermap.TrackerMap.utils.Formatter
 import mx.trackermap.TrackerMap.utils.SpeedUnit
@@ -22,6 +23,7 @@ typealias ActionCallback = (unit: UnitInformation, action: Action) -> Unit
 class UnitRenderData {
     data class UnitRenderViewHolder(
         val unitName: TextView,
+        val unitIcon: ImageView? = null,
         val statusIcon: ImageView,
         val engineStopIcon: ImageView,
         val gridLayout: GridLayout,
@@ -45,6 +47,14 @@ class UnitRenderData {
                 /* Device name */
                 unitName.text = unit.device.name
 
+                /* Device icon */
+                unitIcon?.setImageResource(
+                    MarkerTransformations.categoryToResourceId(unit.device.category)
+                )
+                unitIcon?.contentDescription = context.getString(
+                        MarkerTransformations.categoryToStringId(unit.device.category)
+                )
+
                 /* Contact */
                 unit.device.contact?.let { contact ->
                     if (contact.isNotEmpty()) {