自定義計時器UIButton

1. 創建PWTimerButton 繼承與UIButton

2. 添加成員方法和屬性

3. 頭文件如下：

@interface PWTimerButton : UIButton

{

    NSInteger _timerInterval;

    NSTimer *_timer;

    NSString *_formatString;

}

- (void)setFormatString:(NSString *)formatString;

- (void)startTimer:(NSInteger)interval;

- (void)stopTimer;

4.源文件如下：

- (void)startTimer:(NSInteger)interval

{

    if (_timer) {

        [_timer invalidate];

        _timer = nil;

    }

    if (interval > 0) {

        _timerInterval = interval;

        self.enabled = NO;

        [self updateButtonState];

        _timer = [NSTimer scheduledTimerWithTimeInterval:1

                                                  target:self

                                                selector:@selector(timeOut:)

                                                userInfo:nil

                                                 repeats:YES];

    }

}

- (void)stopTimer

{

    if (_timer) {

        [_timer invalidate];

        _timer = nil;

        _timerInterval = 0;

    }

    [self updateButtonState];

}

- (void)timeOut:(NSTimer *)timer

{

    _timerInterval--;

    if (_timerInterval > 0) {

    }

    else {

        [self stopTimer];

    }

    [self updateButtonState];

}

- (void)setFormatString:(NSString *)formatString

{

    _formatString = formatString;

}

- (void)updateButtonState

{

    if (_timerInterval > 0) {

        [self setTitle:[NSString stringWithFormat:@"%d", _timerInterval] forState:UIControlStateDisabled];

    }

    else {

        [self setTitle:@"驗證" forState:UIControlStateNormal];

        self.enabled = YES;

    }

}

標記***的為重點知識，設置按鈕的什么狀態就要再什么狀態下更改按鈕的文字，顏色或者背景圖片等。

調了n長時間終于調試出來了，哎，自己粗心大意了。。。