Custom BottomNavigationView

PositionBottomNavigationView.java

import android.content.Context;
import android.support.annotation.NonNull;
import android.support.design.widget.BottomNavigationView;
import android.util.AttributeSet;
import android.util.SparseIntArray;
import android.view.Menu;
import android.view.MenuItem;

/**
 * Created by tysheng
 * Date: 2016/12/31 12:24.
 * Email: [email protected]
 */

public class PositionBottomNavigationView extends BottomNavigationView implements BottomNavigationView.OnNavigationItemSelectedListener {
    private SparseIntArray mMenuIds;
    private int prePosition = -1;
    private onPositionSelectedListener mListener;

    public PositionBottomNavigationView(Context context) {
        this(context, null, 0);
    }

    public PositionBottomNavigationView(Context context, AttributeSet attrs) {
        this(context, attrs, 0);
    }

    public PositionBottomNavigationView(Context context, AttributeSet attrs, int defStyleAttr) {
        super(context, attrs, defStyleAttr);
        mMenuIds = new SparseIntArray();
        setOnNavigationItemSelectedListener(this);
    }

    /**
     * ATTENTION: register the ids first.
     *
     * @param ids menu ids by order, like R.id.menu_0,R.id.menu_1...
     */
    public void registerIds(int... ids) {
        mMenuIds.clear();
        for (int i = 0; i < ids.length; i++) {
            mMenuIds.put(i, ids[i]);
        }
    }

    public void setOnPositionSelectedListener(onPositionSelectedListener listener) {
        mListener = listener;
    }

    public int getCurrentPosition() {
        return prePosition;
    }

    public void setCurrentPosition(int current) {
        if (prePosition == current) {
            if (mListener != null)
                mListener.onPositionReselected(current);
            return;
        }
        Menu menu = getMenu();
        for (int i = 0; i < mMenuIds.size(); i++) {
            if (prePosition >= 0)
                menu.getItem(prePosition).setChecked(false);
            menu.getItem(current).setChecked(true);
        }
        prePosition = current;
        if (mListener != null)
            mListener.onPositionSelected(current);
    }

    @Override
    public boolean onNavigationItemSelected(@NonNull MenuItem item) {
        int pos = mMenuIds.indexOfValue(item.getItemId());
        if (pos != -1) {
            setCurrentPosition(pos);
        }
        return true;
    }

    public interface onPositionSelectedListener {
        void onPositionSelected(int position);

        void onPositionReselected(int position);
    }

    public static class onSimplePositionSelectedListener implements onPositionSelectedListener {

        @Override
        public void onPositionSelected(int position) {

        }

        @Override
        public void onPositionReselected(int position) {

        }
    }
}