ProgressButton的实现

一向对UI不是太感兴趣，不过项目中遇到了还是得去做，做这个ProgressButton还是费了不少功夫，不过效果还可以，这里先记下，以后可能还会用到。只贴代码，原理就不多说了。

public class ProgressButton extends Button {

    public static final int MODE_BUTTON = 0;

    public static final int MODE_PROGRESS = 1;

    private Paint mBackgroundPaint;

    private Paint mTextPaint;

    private int mProgress = -1;

    /**
     * 进度条填充色
     */
    private int mPbColor = Color.GREEN;

    /**
     * 进度条字体颜色
     */
    private int mPbTextColor = Color.WHITE;

    /**
     * 原始按钮字体颜色
     */
    private int mTextColor;

    private int mWidth, mHeight;

    private Drawable mBackgroundDrawable;

    private Bitmap mBackgroundBitmap;

    private RectF mRectSrc;

    private int mMode;

    private PorterDuffXfermode mXfermode;

    public ProgressButton(Context context) {
        this(context, null);
    }

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

    public ProgressButton(Context context, AttributeSet attrs, int defStyleAttr) {
        super(context, attrs, defStyleAttr);
        init(context, attrs);
    }

    private void initAttrs(Context context, AttributeSet attrs) {
        TypedArray a = context.obtainStyledAttributes(attrs, R.styleable.ProgressButton);
        mPbColor = a.getColor(R.styleable.ProgressButton_ProgressButton_pbColor, Color.GREEN);
        mPbTextColor = a.getColor(R.styleable.ProgressButton_ProgressButton_pbTextColor, Color.WHITE);
        a.recycle();
    }

    private void init(Context context, AttributeSet attrs) {
        initAttrs(context, attrs);

        mTextColor = getCurrentTextColor();

        mBackgroundPaint = new Paint();
        mBackgroundPaint.setAntiAlias(true);
        mBackgroundPaint.setStyle(Paint.Style.FILL);

        mTextPaint = new Paint();
        mTextPaint.setAntiAlias(true);
        mTextPaint.setTextSize(getTextSize());
        mTextPaint.setColor(mTextColor);

        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.HONEYCOMB) {
            //解决文字有时候画不出问题
            setLayerType(LAYER_TYPE_SOFTWARE, mTextPaint);
        }

        mBackgroundDrawable = getBackground();
        mXfermode = new PorterDuffXfermode(PorterDuff.Mode.SRC_ATOP);
    }

    public void setProgress(int progress) {
        if (progress > mProgress) {
            mProgress = progress;
            invalidate();
        }
    }

    public void setButtonMode(int mode) {
        mMode = mode;

        if (mode == MODE_BUTTON) {
            setBackgroundDrawable(mBackgroundDrawable);
            mProgress = -1;
        } else if (mode == MODE_PROGRESS) {
            setBackgroundResource(0);
        } else {
            throw new UnsupportedOperationException("Unknown mode " + mode);
        }
    }

    private void releaseBgBitmap() {
        if (mBackgroundBitmap != null) {
            mBackgroundBitmap.recycle();
            mBackgroundBitmap = null;
        }
    }

    public boolean isProgressMode() {
        return mMode == MODE_PROGRESS;
    }

    @Override
    protected void onSizeChanged(int w, int h, int oldw, int oldh) {
        super.onSizeChanged(w, h, oldw, oldh);

        if (w > 0 && h > 0 && w != oldw && h != oldh) {
            mWidth = w;
            mHeight = h;
            mRectSrc = new RectF(0, 0, mWidth, mHeight);

            if (mBackgroundBitmap != null) {
                releaseBgBitmap();
            }

            mBackgroundBitmap = Bitmap.createBitmap(mWidth, mHeight, Bitmap.Config.ARGB_8888);
            Canvas canvas = new Canvas(mBackgroundBitmap);

            /**
             * 这个不能掉
             */
            mBackgroundDrawable.setBounds(0, 0, mWidth, mHeight);

            mBackgroundDrawable.draw(canvas);
        }
    }

    @Override
    protected void onDraw(Canvas canvas) {
        super.onDraw(canvas);

        if (mWidth > 0 && mHeight > 0 && isProgressMode()) {
            int saveCount = canvas.saveLayer(mRectSrc, mBackgroundPaint, Canvas.ALL_SAVE_FLAG);

            canvas.drawBitmap(mBackgroundBitmap, null, mRectSrc, mBackgroundPaint);

            mBackgroundPaint.setXfermode(mXfermode);

            mBackgroundPaint.setColor(mPbColor);
            mBackgroundPaint.setStyle(Paint.Style.FILL);
            canvas.drawRect(0, 0, mWidth * mProgress / 100, mHeight, mBackgroundPaint);

            mBackgroundPaint.setXfermode(null);

            drawText(canvas);

            canvas.restoreToCount(saveCount);
        }
    }

    private void drawText(Canvas canvas) {
        float progress = mProgress * 0.01f;
        LinearGradient shader = new LinearGradient(0, 0, mWidth, 0, new int[]{
                mPbTextColor, mTextColor
        }, new float[]{
                progress, progress + 0.001f
        }, Shader.TileMode.CLAMP);

        mTextPaint.setShader(shader);

        String text = getText().toString();
        float textWidth = mTextPaint.measureText(text);
        float baseX = (mWidth - textWidth) / 2;
        float baseY = mHeight / 2 - (mTextPaint.descent() + mTextPaint.ascent()) / 2;
        canvas.drawText(text, baseX, baseY, mTextPaint);
    }

    public void destroy() {
        releaseBgBitmap();
    }
}

值得一提的是如何将Nine Patch Drawable转化为Bitmap，如下：

mBackgroundBitmap = Bitmap.createBitmap(mWidth, mHeight, Bitmap.Config.ARGB_8888);
Canvas canvas = new Canvas(mBackgroundBitmap);
mBackgroundDrawable.setBounds(0, 0, mWidth, mHeight);
mBackgroundDrawable.draw(canvas);