BadgeView.java

package com.xiaomi.shop2.widget;

import android.content.Context;
import android.graphics.Color;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.GradientDrawable;
import android.graphics.drawable.ShapeDrawable;
import android.graphics.drawable.shapes.OvalShape;
import android.util.AttributeSet;
import android.view.View;
import android.widget.TextView;

public class BadgeView extends TextView {

    private int mBadgeColor = Color.parseColor("#fe672c");
    private static final int DEFAULT_CORNER_RADIUS_DIP = 90;
    private float mCounterHeight = 17; // dp

    public BadgeView(Context context) {
        super(context);
    }

    public BadgeView(Context context, AttributeSet attrs) {
        super(context, attrs);
    }

    public BadgeView(Context context, AttributeSet attrs, int defStyle) {
        super(context, attrs, defStyle);
    }

    @Override
    public void setText(CharSequence text, BufferType type) {
        super.setText(text, type);
    }

    private Drawable getDefaultBackground() {
        GradientDrawable gd = new GradientDrawable();
        gd.setColor(mBadgeColor);
        gd.setCornerRadius(DEFAULT_CORNER_RADIUS_DIP);
        return gd;
    }

    private Drawable getOvalDrawable() {
        OvalShape oval = new OvalShape();
        oval.resize(DensityUtil.dip2px(mCounterHeight), DensityUtil.dip2px(mCounterHeight));
        ShapeDrawable drawable = new ShapeDrawable(oval);
        drawable.getPaint().setColor(mBadgeColor);
        drawable.setIntrinsicHeight(DensityUtil.dip2px(mCounterHeight));
        drawable.setIntrinsicWidth(DensityUtil.dip2px(mCounterHeight));
        return drawable;
    }

    public void setCount(int count) {
        if (count > 0) {
            if (count < 10) {
                setBackgroundDrawable(getOvalDrawable());
                setText(String.valueOf(count));
            } else if (count > 99) {
                setBackgroundDrawable(getDefaultBackground());
                setText("99+");
            } else {
                setBackgroundDrawable(getDefaultBackground());
                setText(String.valueOf(count));
            }
            this.setVisibility(View.VISIBLE);
        } else {
            this.setVisibility(View.GONE);
        }
    }
}