fix LinePageIndicator #3

cpacm · Sep 22, 2016 · 104eb72 · 104eb72
1 parent d24b9e3
commit 104eb72
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/library/src/main/java/net/cpacm/library/indicator/ViewpagerIndicator/LinePageIndicator.java b/library/src/main/java/net/cpacm/library/indicator/ViewpagerIndicator/LinePageIndicator.java
@@ -368,7 +368,10 @@ private int measureWidth(int measureSpec) {
             result = specSize;
         } else {
             //Calculate the width according the views count
-            final int count = mViewPager.getAdapter().getCount();
+            int count = mViewPager.getAdapter().getCount();
+            if (mViewPager.getAdapter() instanceof InfinitePagerAdapter) {
+                count = ((InfinitePagerAdapter) mViewPager.getAdapter()).getRealCount();
+            }
             result = getPaddingLeft() + getPaddingRight() + (count * mLineWidth) + ((count - 1) * mGapWidth);
             //Respect AT_MOST value if that was what is called for by measureSpec
             if (specMode == MeasureSpec.AT_MOST) {