Add interface to set badge drawable

chenshixin · Sep 20, 2016 · 9aa70c2 · 9aa70c2
1 parent 95eb1d3
commit 9aa70c2
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/bottom-navigation/src/main/java/com/chenshixin/bottomnavigation/BottomNavigation.kt b/bottom-navigation/src/main/java/com/chenshixin/bottomnavigation/BottomNavigation.kt
@@ -1,8 +1,9 @@
 package com.chenshixin.bottomnavigation
 
 import android.content.Context
+import android.graphics.drawable.Drawable
 import android.support.design.widget.CoordinatorLayout
-import android.support.v4.app.FragmentStatePagerAdapter
+import android.support.v4.app.FragmentPagerAdapter
 import android.util.AttributeSet
 import android.view.LayoutInflater
 import android.view.ViewGroup
@@ -46,8 +47,10 @@ class BottomNavigation(context: Context?, attrs: AttributeSet?) : CoordinatorLay
         layoutParams = ViewGroup.LayoutParams(ViewGroup.LayoutParams.MATCH_PARENT, ViewGroup.LayoutParams.MATCH_PARENT)
     }
 
-    fun setFragmentPagerAdapter(fragmentPagerAdapter: FragmentStatePagerAdapter) {
+    fun setFragmentPagerAdapter(fragmentPagerAdapter: FragmentPagerAdapter) {
         bottom_navigation_view_pager.adapter = fragmentPagerAdapter
+        //Cache all pages
+        bottom_navigation_view_pager.offscreenPageLimit = fragmentPagerAdapter.count - 1
     }
 
     fun setTabItems(tabs: List<BottomNavigationItem>) {
@@ -86,6 +89,10 @@ class BottomNavigation(context: Context?, attrs: AttributeSet?) : CoordinatorLay
         bottom_navigation_bar.tabs[index].setBadgeNumber(number)
     }
 
+    fun setBadgeDrawable(drawable: Drawable) {
+        bottom_navigation_bar.setBadgeBackground(drawable)
+    }
+
     fun setItemTitle(index: Int, title: String) {
         bottom_navigation_bar.tabs[index].setTitle(title)
     }