Встановіть діапазон для елементів у GridLayoutManager за допомогою SpanSizeLookup

Я хочу реалізувати сітчастий макет із заголовками розділів. Подумайте про https://github.com/TonicArtos/StickyGridHeaders

Що я зараз роблю:

mRecyclerView = (RecyclerView) view.findViewById(R.id.grid);
mLayoutManager = new GridLayoutManager(getActivity(), 2);
mLayoutManager.setSpanSizeLookup(new GridLayoutManager.SpanSizeLookup() {
            @Override
            public int getSpanSize(int position) {
                switch(mAdapter.getItemViewType(position)){
                    case MyAdapter.TYPE_HEADER:
                        return 1;
                    case MyAdapter.TYPE_ITEM:
                        return 2;
                    default:
                        return -1;
                }
            }
        });

mRecyclerView.setLayoutManager(mLayoutManager);

Тепер і звичайні елементи, і заголовки мають розмір 1. Як це вирішити?

Проблема полягала в тому, що заголовок повинен мати розмір 2, а звичайний елемент - 1. Тож правильні реалізації:

mLayoutManager.setSpanSizeLookup(new GridLayoutManager.SpanSizeLookup() {
            @Override
            public int getSpanSize(int position) {
                switch(mAdapter.getItemViewType(position)){
                    case MyAdapter.TYPE_HEADER:
                        return 2;
                    case MyAdapter.TYPE_ITEM:
                        return 1;
                    default:
                        return -1;
                }
            }
        });

Заголовок повинен мати інтервал, рівний кількості діапазонів усього списку.

mLayoutManager.setSpanSizeLookup(new GridLayoutManager.SpanSizeLookup() {
    @Override
    public int getSpanSize(int position) {
           switch(mAdapter.getItemViewType(position)){
                    case MyAdapter.TYPE_HEADER:
                        return mLayoutManager.getSpanCount();
                    case MyAdapter.TYPE_ITEM:
                        return 1;
                    default:
                        return -1;
                }
    }
});

Відповідь на моє власне запитання: Замінити getSpanSizeLookup () з Activity після налаштування адаптера.