安卓TextView圆形背景、圆角矩形背景根据文字长度自适应

设计师要求TextView高度固定，当文字长度为1时，显示圆形背景；当文字长度大于1时显示圆角矩形背景。如图所示。

困难在于，设定宽度wrap_content，当文字长度为1时，不一定能保证宽高比正好为1:1

比较容易想到的方法是，弄两个TextView，圆形背景的tv永远保持1:1大小不变，圆角矩形背景的tv的宽度设为wrap_content，使用时根据文字长度分别设置两者的Visibility即可。但这样做过于麻烦，其实一个View就可以解决。

public class MinWidthCircleTextView extends FrameLayout{

    private TextView tv;

    public MinWidthCircleTextView(Context context) {
        super(context);
        init();
    }

    public MinWidthCircleTextView(Context context, @Nullable AttributeSet attrs) {
        super(context, attrs);
        init();
    }

    public MinWidthCircleTextView(Context context, @Nullable AttributeSet attrs, int defStyleAttr) {
        super(context, attrs, defStyleAttr);
        init();
    }

    private void init() {
        tv = new TextView(getContext());
        FrameLayout.LayoutParams layoutParams = new FrameLayout.LayoutParams(WRAP_CONTENT, MATCH_PARENT);
        layoutParams.gravity = Gravity.CENTER;
        addView(tv, layoutParams);
        tv.setText("00");
        tv.setGravity(Gravity.CENTER);
        tv.setTextSize(TypedValue.COMPLEX_UNIT_DIP, 10);
        tv.setTypeface(ResourcesCompat.getFont(getContext(), R.font.my_font));
        tv.setTextColor(Color.parseColor("#FFFFFF"));
    }

    public void setText(String text) {
        tv.setText(text);
    }

    @Override
    protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {
        super.onMeasure(widthMeasureSpec, heightMeasureSpec);
        if (tv.getText().length() <= 1) {
            setMeasuredDimension(getMeasuredHeight(), getMeasuredHeight());
            setBackgroundResource(R.drawable.shape_oval);
        } else {
            setBackgroundResource(R.drawable.shape_rect);
        }
    }
}