Set Badge count above notification icons

Layout.xml-   

<RelativeLayout

    android:layout_width="match_parent"

    android:layout_height="match_parent"

    android:layout_weight="1">



    <TextView

        android:id="@+id/tv_notification"

        android:layout_width="match_parent"

        android:layout_height="wrap_content"

        android:drawableTop="@drawable/notification_select"

        android:gravity="center"

        android:maxLines="1"

        android:text="Notification"

        android:textColor="@color/blue_color"

        android:textSize="10sp" />



    <com.nex3z.notificationbadge.NotificationBadge

        android:id="@+id/badge_count"

        android:layout_width="28dp"

        android:layout_height="28dp"

        android:layout_alignTop="@+id/tv_notification"

        android:layout_marginLeft="-30dp"

        android:layout_marginTop="-4dp"

        android:layout_toRightOf="@+id/tv_notification"

        app:badgeBackground="@drawable/badge_count"

        app:badgeTextSize="10dp"

        app:maxTextLength="2" />



</RelativeLayout>

Code- 

package com.nex3z.notificationbadge;



import android.content.Context;

import android.content.res.TypedArray;

import android.graphics.drawable.Drawable;

import android.util.AttributeSet;

import android.util.TypedValue;

import android.view.LayoutInflater;

import android.view.animation.Animation;

import android.view.animation.ScaleAnimation;

import android.widget.FrameLayout;

import android.widget.ImageView;

import android.widget.TextView;



public class NotificationBadge extends FrameLayout {

    private static final int DEFAULT_TEXT_COLOR = 0xFFFFFFFF;

    private static final int DEFAULT_TEXT_SIZE = 14;

    private static final boolean DEFAULT_ANIMATION_ENABLED = true;

    private static final int DEFAULT_ANIMATION_DURATION = 500;

    private static final int DEFAULT_MAX_TEXT_LENGTH = 2;

    private static final String DEFAULT_MAX_LENGTH_REACHED_TEXT = "...";



    private FrameLayout mContainer;

    private ImageView mIvBadgeBg;

    private TextView mTvBadgeText;

    private int mBadgeTextColor = DEFAULT_TEXT_COLOR;

    private float mBadgeTextSize = dpToPx(DEFAULT_TEXT_SIZE);

    private int mAnimationDuration = DEFAULT_ANIMATION_DURATION;

    private Animation mUpdate;

    private Animation mShow;

    private Animation mHide;

    private String mBadgeText;

    private boolean mIsBadgeShown;

    private boolean mAnimationEnabled = DEFAULT_ANIMATION_ENABLED;

    private int mMaxTextLength = DEFAULT_MAX_TEXT_LENGTH;

    private String mEllipsizeText = DEFAULT_MAX_LENGTH_REACHED_TEXT;



    public NotificationBadge(Context context, AttributeSet attrs) {

        super(context, attrs);



        LayoutInflater inflater = (LayoutInflater) context.getSystemService(

                Context.LAYOUT_INFLATER_SERVICE);

        inflater.inflate(R.layout.notification_badge, this, true);

        mContainer = (FrameLayout) findViewById(R.id.fl_container);

        mIvBadgeBg = (ImageView) findViewById(R.id.iv_badge_bg);

        mTvBadgeText = (TextView) findViewById(R.id.tv_badge_text);



        TypedArray a = context.getTheme().obtainStyledAttributes(

                attrs, R.styleable.NotificationBadge, 0, 0);

        try {

            mBadgeTextColor = a.getColor(R.styleable.NotificationBadge_badgeTextColor, DEFAULT_TEXT_COLOR);

            mTvBadgeText.setTextColor(mBadgeTextColor);



            mBadgeTextSize = a.getDimensionPixelSize(R.styleable.NotificationBadge_badgeTextSize, (int)dpToPx(DEFAULT_TEXT_SIZE));

            mTvBadgeText.setTextSize(TypedValue.COMPLEX_UNIT_PX, mBadgeTextSize);



            mAnimationEnabled = a.getBoolean(R.styleable.NotificationBadge_animationEnabled, DEFAULT_ANIMATION_ENABLED);



            mAnimationDuration = a.getInt(R.styleable.NotificationBadge_animationDuration, DEFAULT_ANIMATION_DURATION);

            Drawable badgeBackground = a.getDrawable(R.styleable.NotificationBadge_badgeBackground);

            if (badgeBackground != null) {

                mIvBadgeBg.setImageDrawable(badgeBackground);

            }



            mMaxTextLength = a.getInt(R.styleable.NotificationBadge_maxTextLength, DEFAULT_MAX_TEXT_LENGTH);

            mEllipsizeText = a.getString(R.styleable.NotificationBadge_ellipsizeText);

            if (mEllipsizeText == null) {

                mEllipsizeText = DEFAULT_MAX_LENGTH_REACHED_TEXT;

            }

        } finally {

            a.recycle();

        }



        if (mAnimationEnabled) {

            initUpdateAnimation();

        }

    }



    public void clear() {

        if (mIsBadgeShown) {

            if (mAnimationEnabled) {

                mContainer.startAnimation(mHide);

            } else {

                mContainer.setVisibility(INVISIBLE);

            }

            mIsBadgeShown = false;

        }

    }



    public void show(String text) {

        mBadgeText = text;

        if (!mIsBadgeShown) {

            if (mAnimationEnabled) {

                mContainer.startAnimation(mShow);

            } else {

                mContainer.setVisibility(VISIBLE);

                mTvBadgeText.setText(text);

            }

            mIsBadgeShown = true;

        } else {

            mTvBadgeText.setText(text);

        }

    }



    public boolean isAnimationEnabled() {

        return mAnimationEnabled;

    }



    public void setAnimationEnabled(boolean animationEnabled) {

        mAnimationEnabled = animationEnabled;

        if (animationEnabled && (mUpdate == null || mShow == null || mHide == null)) {

            initUpdateAnimation();

        }

    }



    public void setMaxTextLength(int maxLength) {

        mMaxTextLength = maxLength;

    }



    public void setText(String text) {

        if (text != null && text.length() > mMaxTextLength) {

            mBadgeText = mEllipsizeText;

        } else {

            mBadgeText = text;

        }

        if (text == null || text.isEmpty()) {

            clear();

        } else {

            if (mIsBadgeShown) {

                if (mAnimationEnabled) {

                    mContainer.startAnimation(mUpdate);

                } else {

                    mTvBadgeText.setText(mBadgeText);

                }

            } else {

                show(text);

            }

        }

    }



    public void setNumber(int number) {

        if (number == 0) {

            clear();

        } else {

            setText(String.valueOf(number));

        }

    }



    public int getTextColor() {

        return mBadgeTextColor;

    }



    public void setTextColor(int textColor) {

        mBadgeTextColor = textColor;

        mTvBadgeText.setTextColor(textColor);

    }



    public void setTextSize(float textSize) {

        mBadgeTextSize = textSize;

        mTvBadgeText.setTextSize(TypedValue.COMPLEX_UNIT_PX, textSize);

    }



    public void setBadgeBackgroundDrawable(Drawable drawable) {

        mIvBadgeBg.setImageDrawable(drawable);

    }



    public void setBadgeBackgroundResource(int resId) {

        mIvBadgeBg.setImageResource(resId);

    }



    private void initUpdateAnimation() {

        mUpdate = new ScaleAnimation(1, 1.2f, 1, 1.2f,

                Animation.RELATIVE_TO_SELF, 0.5f, Animation.RELATIVE_TO_SELF, 0.5f);

        mUpdate.setDuration(mAnimationDuration / 2);

        mUpdate.setRepeatMode(Animation.REVERSE);

        mUpdate.setRepeatCount(1);

        mUpdate.setAnimationListener(new Animation.AnimationListener() {

            @Override

            public void onAnimationStart(Animation animation) {

                mTvBadgeText.setText(mBadgeText);

            }

            @Override

            public void onAnimationEnd(Animation animation) {}

            @Override

            public void onAnimationRepeat(Animation animation) {}

        });



        mShow = new ScaleAnimation(0, 1, 0, 1,

                Animation.RELATIVE_TO_SELF, 0.5f, Animation.RELATIVE_TO_SELF, 0.5f);

        mShow.setDuration(mAnimationDuration / 2);

        mShow.setAnimationListener(new Animation.AnimationListener() {

            @Override

            public void onAnimationStart(Animation animation) {

                mContainer.setVisibility(VISIBLE);

                mTvBadgeText.setText(mBadgeText);

            }

            @Override

            public void onAnimationEnd(Animation animation) {}

            @Override

            public void onAnimationRepeat(Animation animation) {}

        });



        mHide = new ScaleAnimation(1, 0, 1, 0,

                Animation.RELATIVE_TO_SELF, 0.5f, Animation.RELATIVE_TO_SELF, 0.5f);

        mHide.setDuration(mAnimationDuration / 2);

        mHide.setAnimationListener(new Animation.AnimationListener() {

            @Override

            public void onAnimationStart(Animation animation) {}

            @Override

            public void onAnimationEnd(Animation animation) {

                mContainer.setVisibility(INVISIBLE);

            }

            @Override

            public void onAnimationRepeat(Animation animation) {}

        });

    }



    private float dpToPx(float dp){

        return TypedValue.applyDimension(

                TypedValue.COMPLEX_UNIT_DIP, dp, getResources().getDisplayMetrics());

    }



}