1 files changed, 10 insertions, 10 deletions
diff --git a/androidApp/src/main/java/mx/trackermap/TrackerMap/android/map/UnitMapFragment.kt b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/map/UnitMapFragment.kt
index b6396c8..b67bca8 100644
--- a/androidApp/src/main/java/mx/trackermap/TrackerMap/android/map/UnitMapFragment.kt
+++ b/androidApp/src/main/java/mx/trackermap/TrackerMap/android/map/UnitMapFragment.kt
@@ -14,7 +14,6 @@ import kotlinx.coroutines.DelicateCoroutinesApi
 import mx.trackermap.TrackerMap.android.R
 import mx.trackermap.TrackerMap.android.databinding.UnitMapFragmentBinding
 import mx.trackermap.TrackerMap.android.details.DetailsActivity
-import mx.trackermap.TrackerMap.android.shared.MarkerTransformations
 import mx.trackermap.TrackerMap.android.shared.UnitRenderData
 import mx.trackermap.TrackerMap.android.units.UnitsViewModel
 import mx.trackermap.TrackerMap.client.models.MapLayer
@@ -30,7 +29,7 @@ class UnitMapFragment : Fragment() {
 
     private var _binding: UnitMapFragmentBinding? = null
     private val binding get() = _binding!!
-    private lateinit var unitsMapFragment: MapFragment
+    private lateinit var mapFragment: MapWrapperFragment
 
     private var shouldCenter = true
 
@@ -63,8 +62,9 @@ class UnitMapFragment : Fragment() {
 
     private fun initializeMap() {
         shouldCenter = true
-        unitsMapFragment = childFragmentManager.findFragmentById(R.id.unitsMap) as MapFragment
-        unitsMapFragment.markerCallback = unitsViewModel::selectUnitWith
+        mapFragment = childFragmentManager.findFragmentById(R.id.unitsMap) as MapWrapperFragment
+        mapFragment.setMarkerCallback(unitsViewModel::selectUnitWith)
+        mapFragment.setOverlayPaddingTop(resources.getDimensionPixelSize(R.dimen.nav_height))
 
         val layer = MapLayer.defaultLayer
         binding.attributionText.text = HtmlCompat.fromHtml(layer.attribution, 0)
@@ -75,8 +75,8 @@ class UnitMapFragment : Fragment() {
 
         unitsViewModel.units.observe(viewLifecycleOwner) { units ->
             Log.d("UnitMapFragment", "Available units: $units")
-            unitsMapFragment.display(
-                units.mapNotNull(MarkerTransformations::unitToMarker).toTypedArray(),
+            mapFragment.display(
+                units.mapNotNull { it.position }.toTypedArray(),
                 isReport = false,
                 center = shouldCenter
             )
@@ -107,22 +107,22 @@ class UnitMapFragment : Fragment() {
                     ), binding.mapUnitCard.context, unit, this::itemAction
                 )
 
-                unitsMapFragment.focusOn(unit.position!!.latitude!!, unit.position!!.longitude!!)
+                mapFragment.focusOn(unit.position!!.latitude!!, unit.position!!.longitude!!)
             }
         }
 
         unitsViewModel.mapLayerType.observe(viewLifecycleOwner) { type ->
             Log.d("UnitMapFragment", "Loading layer!")
-            unitsMapFragment.updateLayer(type)
+            mapFragment.updateLayer(type)
             unitsViewModel.selectedUnit.value?.let {
-                unitsMapFragment.focusOn(it.position!!.latitude!!, it.position!!.longitude!!)
+                mapFragment.focusOn(it.position!!.latitude!!, it.position!!.longitude!!)
             }
             val layer = MapLayer.layers[type]!!
             binding.attributionText.text = HtmlCompat.fromHtml(layer.attribution, 0)
         }
 
         unitsViewModel.geofences.observe(viewLifecycleOwner) { geofences ->
-            unitsMapFragment.displayGeofences(geofences.values.toTypedArray())
+            mapFragment.display(geofences.values.toTypedArray())
         }
     }