diff --git a/sample/src/main/AndroidManifest.xml b/sample/src/main/AndroidManifest.xml
index d65cb6c..7fff270 100644
--- a/sample/src/main/AndroidManifest.xml
+++ b/sample/src/main/AndroidManifest.xml
@@ -17,6 +17,9 @@
+
+
diff --git a/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AdvancedDatasetActivity.java b/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AdvancedDatasetActivity.java
new file mode 100644
index 0000000..d8daa41
--- /dev/null
+++ b/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AdvancedDatasetActivity.java
@@ -0,0 +1,155 @@
+package com.timehop.stickyheadersrecyclerview.sample;
+
+import android.app.Activity;
+import android.graphics.Color;
+import android.os.Bundle;
+import android.support.v7.widget.LinearLayoutManager;
+import android.support.v7.widget.RecyclerView;
+import android.view.LayoutInflater;
+import android.view.View;
+import android.view.ViewGroup;
+import android.widget.TextView;
+
+import com.timehop.stickyheadersrecyclerview.StickyRecyclerHeadersAdapter;
+import com.timehop.stickyheadersrecyclerview.StickyRecyclerHeadersDecoration;
+import com.timehop.stickyheadersrecyclerview.sample.AnimalsAdapter;
+import com.timehop.stickyheadersrecyclerview.sample.DividerDecoration;
+import com.timehop.stickyheadersrecyclerview.sample.R;
+
+import java.security.SecureRandom;
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.HashMap;
+import java.util.Map;
+
+public class AdvancedDatasetActivity extends Activity {
+ final Map mAnimalColors = new HashMap<>();
+ final AnimalsColorHeadersAdapter mAdapter = new AnimalsColorHeadersAdapter();
+
+ final String COLORNAME_BLACK = "Black";
+ final String COLORNAME_BLUE = "Blue";
+ final String COLORNAME_BROWN = "Brown";
+ final String COLORNAME_OTHER = "Other";
+
+ Comparator mColorComparator = new Comparator() {
+ @Override
+ public int compare(String lhsName, String rhsName) {
+ // order by (color, name)
+ String lhsColor = getColorForAnimalName(lhsName);
+ String rhsColor = getColorForAnimalName(rhsName);
+
+ if (lhsColor.compareTo(rhsColor) < 0) {
+ // lhsColor < rhsColor
+ return -1;
+ } else if (lhsColor.compareTo(rhsColor) > 0) {
+ // lhsColor > rhsColor
+ return 1;
+ } else {
+ // color is equal
+ return lhsName.compareTo(rhsName);
+ }
+ }
+ };
+
+ @Override
+ protected void onCreate(Bundle savedInstanceState) {
+ super.onCreate(savedInstanceState);
+ setContentView(R.layout.activity_advanced);
+
+ final RecyclerView recyclerView = (RecyclerView) findViewById(R.id.recyclerview);
+
+ // Add initial data to color map
+ mAnimalColors.put("Aye Aye", COLORNAME_BROWN);
+ mAnimalColors.put("Black Bear", COLORNAME_BLACK);
+ mAnimalColors.put("Black Rhinoceros", COLORNAME_BLACK);
+ mAnimalColors.put("Black Russian Terrier", COLORNAME_BLACK);
+ mAnimalColors.put("Black Widow Spider", COLORNAME_BLACK);
+ mAnimalColors.put("Blue Lacy Dog", COLORNAME_BLUE);
+ mAnimalColors.put("Blue Whale", COLORNAME_BLUE);
+ mAnimalColors.put("Highland Cattle", COLORNAME_BROWN);
+
+ // Populate adapter with example dummy data
+ mAdapter.addAll(getDummyDataSet());
+ mAdapter.sort(mColorComparator);
+ recyclerView.setAdapter(mAdapter);
+
+ // Set layout manager
+ final LinearLayoutManager layoutManager = new LinearLayoutManager(this);
+ recyclerView.setLayoutManager(layoutManager);
+
+ // Add the sticky headers decoration
+ final StickyRecyclerHeadersDecoration headersDecor = new StickyRecyclerHeadersDecoration(mAdapter);
+ recyclerView.addItemDecoration(headersDecor);
+
+ // Add decoration for dividers between list items
+ recyclerView.addItemDecoration(new DividerDecoration(this));
+
+ /*
+ mAdapter.registerAdapterDataObserver(new RecyclerView.AdapterDataObserver() {
+ @Override
+ public void onChanged() {
+ //headersDecor.invalidateHeaders();
+ }
+ });
+ */
+ }
+
+ private String[] getDummyDataSet() {
+ return getResources().getStringArray(R.array.animals);
+ }
+
+ private String getColorForAnimalName(String animalName) {
+ if (mAnimalColors.containsKey(animalName)) {
+ return mAnimalColors.get(animalName);
+ } else {
+ return COLORNAME_OTHER;
+ }
+ }
+
+ public void toggleAyeAyeColorClicked(View v) {
+ if (mAnimalColors.get("Aye Aye").equals(COLORNAME_BROWN)) {
+ mAnimalColors.put("Aye Aye", COLORNAME_BLACK);
+ } else {
+ mAnimalColors.put("Aye Aye", COLORNAME_BROWN);
+ }
+ mAdapter.sort(mColorComparator);
+ }
+
+ private class AnimalsColorHeadersAdapter extends AnimalsAdapter
+ implements StickyRecyclerHeadersAdapter {
+ @Override
+ public RecyclerView.ViewHolder onCreateViewHolder(ViewGroup parent, int viewType) {
+ View view = LayoutInflater.from(parent.getContext())
+ .inflate(R.layout.view_item, parent, false);
+ return new RecyclerView.ViewHolder(view) {
+ };
+ }
+
+ @Override
+ public void onBindViewHolder(RecyclerView.ViewHolder holder, int position) {
+ TextView textView = (TextView) holder.itemView;
+ textView.setText(getItem(position));
+ }
+
+ @Override
+ public long getHeaderId(int position) {
+ String colorName = getColorForAnimalName(getItem(position));
+ return colorName.hashCode();
+ }
+
+ @Override
+ public RecyclerView.ViewHolder onCreateHeaderViewHolder(ViewGroup parent) {
+ View view = LayoutInflater.from(parent.getContext())
+ .inflate(R.layout.view_header, parent, false);
+ return new RecyclerView.ViewHolder(view) {
+ };
+ }
+
+ @Override
+ public void onBindHeaderViewHolder(RecyclerView.ViewHolder holder, int position) {
+ String colorName = getColorForAnimalName(getItem(position));
+ TextView textView = (TextView) holder.itemView;
+ textView.setText(colorName);
+ }
+ }
+}
diff --git a/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AnimalsAdapter.java b/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AnimalsAdapter.java
index 1872b4c..ff7de5a 100644
--- a/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AnimalsAdapter.java
+++ b/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/AnimalsAdapter.java
@@ -5,6 +5,8 @@
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
+import java.util.Collections;
+import java.util.Comparator;
/**
@@ -49,6 +51,11 @@ public void remove(String object) {
notifyDataSetChanged();
}
+ public void sort(Comparator comparator) {
+ Collections.sort(items, comparator);
+ notifyDataSetChanged();
+ }
+
public String getItem(int position) {
return items.get(position);
}
diff --git a/sample/src/main/res/layout/activity_advanced.xml b/sample/src/main/res/layout/activity_advanced.xml
new file mode 100644
index 0000000..cd21eef
--- /dev/null
+++ b/sample/src/main/res/layout/activity_advanced.xml
@@ -0,0 +1,36 @@
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/sample/src/main/res/values/strings.xml b/sample/src/main/res/values/strings.xml
index 0c90349..d832c30 100644
--- a/sample/src/main/res/values/strings.xml
+++ b/sample/src/main/res/values/strings.xml
@@ -4,5 +4,6 @@
StickyHeadersRecyclerView
Start Update
Is reverse
+ Make Aye Aye brown/black