tmk_core/common/avr/timer.c

   1 /*
   2 Copyright 2011 Jun Wako <wakojun@gmail.com>
   3
   4 This program is free software: you can redistribute it and/or modify
   5 it under the terms of the GNU General Public License as published by
   6 the Free Software Foundation, either version 2 of the License, or
   7 (at your option) any later version.
   8
   9 This program is distributed in the hope that it will be useful,
  10 but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 GNU General Public License for more details.
  13
  14 You should have received a copy of the GNU General Public License
  15 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 */
  17
  18 #include <avr/io.h>
  19 #include <avr/interrupt.h>
  20 #include <util/atomic.h>
  21 #include <stdint.h>
  22 #include "timer_avr.h"
  23 #include "timer.h"
  24
  25
  26 // counter resolution 1ms
  27 // NOTE: union { uint32_t timer32; struct { uint16_t dummy; uint16_t timer16; }}
  28 volatile uint32_t timer_count;
  29
  30 void timer_init(void)
  31 {
  32 #if TIMER_PRESCALER == 1
  33     uint8_t prescaler = 0x01;
  34 #elif TIMER_PRESCALER == 8
  35     uint8_t prescaler = 0x02;
  36 #elif TIMER_PRESCALER == 64
  37     uint8_t prescaler = 0x03;
  38 #elif TIMER_PRESCALER == 256
  39     uint8_t prescaler = 0x04;
  40 #elif TIMER_PRESCALER == 1024
  41     uint8_t prescaler = 0x05;
  42 #else
  43 #   error "Timer prescaler value is NOT vaild."
  44 #endif
  45
  46 #ifndef __AVR_ATmega32A__
  47     // Timer0 CTC mode
  48     TCCR0A = 0x02;
  49
  50     TCCR0B = prescaler;
  51
  52     OCR0A = TIMER_RAW_TOP;
  53     TIMSK0 = (1<<OCIE0A);
  54 #else
  55     // Timer0 CTC mode
  56     TCCR0 = (1 << WGM01) | prescaler;
  57
  58     OCR0 = TIMER_RAW_TOP;
  59     TIMSK = (1 << OCIE0);
  60 #endif
  61 }
  62
  63 inline
  64 void timer_clear(void)
  65 {
  66   ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
  67     timer_count = 0;
  68   }
  69 }
  70
  71 inline
  72 uint16_t timer_read(void)
  73 {
  74     uint32_t t;
  75
  76     ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
  77       t = timer_count;
  78     }
  79
  80     return (t & 0xFFFF);
  81 }
  82
  83 inline
  84 uint32_t timer_read32(void)
  85 {
  86     uint32_t t;
  87
  88     ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
  89       t = timer_count;
  90     }
  91
  92     return t;
  93 }
  94
  95 inline
  96 uint16_t timer_elapsed(uint16_t last)
  97 {
  98     uint32_t t;
  99
 100     ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
 101       t = timer_count;
 102     }
 103
 104     return TIMER_DIFF_16((t & 0xFFFF), last);
 105 }
 106
 107 inline
 108 uint32_t timer_elapsed32(uint32_t last)
 109 {
 110     uint32_t t;
 111
 112     ATOMIC_BLOCK(ATOMIC_RESTORESTATE) {
 113       t = timer_count;
 114     }
 115
 116     return TIMER_DIFF_32(t, last);
 117 }
 118
 119 // excecuted once per 1ms.(excess for just timer count?)
 120 #ifndef __AVR_ATmega32A__
 121 #define TIMER_INTERRUPT_VECTOR TIMER0_COMPA_vect
 122 #else
 123 #define TIMER_INTERRUPT_VECTOR TIMER0_COMP_vect
 124 #endif
 125 ISR(TIMER_INTERRUPT_VECTOR, ISR_NOBLOCK)
 126 {
 127     timer_count++;
 128 }