Merge pull request #141 from Omega-R/bugfix/140-databinding-crash

fixed crash when using data binding

Author: anton-knyazev

omegarecyclerviewlibs/src/main/java/com/omega_r/libs/omegarecyclerview/OmegaRecyclerView.java

@@ -65,6 +65,7 @@ public class OmegaRecyclerView extends ExpandedRecyclerView implements SwipeMenu
     private DividerItemDecoration mDividerItemDecoration;
     private int mItemSpace;
     private int mDividerSize;
+    private boolean mIsAdapterConnected;
 
     public OmegaRecyclerView(Context context) {
         super(context);
@@ -175,6 +176,7 @@ private void initStickyMode(TypedArray a) {
     @Override
     @SuppressWarnings("unchecked")
     public void setAdapter(RecyclerView.Adapter adapter) {
+        mIsAdapterConnected = true;
         unregisterObservers();
 
         if (adapter == null) {
@@ -303,6 +305,38 @@ public void addView(View view, int index, ViewGroup.LayoutParams params) {
         }
     }
 
+    @Override
+    public int getChildCount() {
+        if (!mIsAdapterConnected && areSectionsInitialized())  {
+            return super.getChildCount() + mHeadersList.size() + mFooterList.size();
+        }
+
+        return super.getChildCount();
+    }
+
+    @Override
+    public View getChildAt(int index) {
+        if (!mIsAdapterConnected && areSectionsInitialized()) {
+            int realChildrenCount = super.getChildCount();
+            int headersCount = mHeadersList.size();
+            int footersCount = mFooterList.size();
+
+            if (index < realChildrenCount) {
+                return super.getChildAt(index);
+            } else if (index < realChildrenCount + headersCount) {
+                return mHeadersList.get(index - realChildrenCount);
+            } else if (index < realChildrenCount + headersCount + footersCount) {
+                return mFooterList.get(index - realChildrenCount - headersCount);
+            }
+
+        }
+        return super.getChildAt(index);
+    }
+
+    private boolean areSectionsInitialized() {
+        return mHeadersList != null && mFooterList != null;
+    }
+
     @Override
     protected void onFinishInflate() {
         super.onFinishInflate();