Welcome aboard

Author: Iain Connor

library/src/main/java/com/tippingcanoe/divergence/SectionAdapter.java

@@ -96,20 +96,48 @@ public void onBindViewHolder(RecyclerView.ViewHolder holder, int position) {
     @Override
     public RecyclerView.ViewHolder onCreateViewHolder(ViewGroup parent, int viewType) {
         if (getLoadingResultsLayout() != null && viewType == getLoadingResultsLayout()) {
-            return new ViewHolder(parent, getLoadingResultsLayout());
+            return onCreateLoadingResultsViewHolder(parent, viewType);
         }
 
         if (getErrorResultsLayout() != null && viewType == getErrorResultsLayout()) {
-            return new ViewHolder(parent, getErrorResultsLayout());
+            return onCreateErrorResultsViewHolder(parent, viewType);
         }
 
         if (getNoResultsLayout() != null && viewType == getNoResultsLayout()) {
-            return new ViewHolder(parent, getNoResultsLayout());
+            return onCreateNoResultsViewHolder(parent, viewType);
         }
 
         return onCreateDataViewHolder(parent, viewType);
     }
 
+    protected RecyclerView.ViewHolder onCreateLoadingResultsViewHolder(ViewGroup parent, int viewType) {
+        if (getLoadingResultsLayout() != null) {
+            return new ViewHolder(parent, getLoadingResultsLayout());
+        }
+
+        return null;
+    }
+
+    protected RecyclerView.ViewHolder onCreateErrorResultsViewHolder(ViewGroup parent, int viewType) {
+        if (getErrorResultsLayout() != null) {
+            return new ViewHolder(parent, getErrorResultsLayout());
+        }
+
+        return null;
+    }
+
+    protected RecyclerView.ViewHolder onCreateNoResultsViewHolder(ViewGroup parent, int viewType) {
+        if (getNoResultsLayout() != null) {
+            return new ViewHolder(parent, getNoResultsLayout());
+        }
+
+        return null;
+    }
+
+    public boolean isLoading() {
+        return isLoading;
+    }
+
     public void setLoading(boolean loading) {
         if (getLoadingResultsLayout() == null) {
             throw new InvalidParameterException("You did not provide a layout for the loading cell.");
@@ -130,6 +158,10 @@ public void setLoading(boolean loading) {
         }
     }
 
+    public boolean isError() {
+        return isError;
+    }
+
     public void setError(boolean error) {
         if (getErrorResultsLayout() == null) {
             throw new InvalidParameterException("You did not provide a layout for the error cell.");