Statistics
| Branch: | Revision:

root / toolbox / current.c @ db11713d

History | View | Annotate | Download (2.1 KB)

1
#include "current.h"
2
#include <avr/io.h>
3
#include <avr/interrupt.h>
4
#include <math.h>
5

    
6
#define CYCLES_PER_SECOND 60 // wall power
7
#define SAMPLES_PER_CYCLE 16
8
#define N_SAMPLES 48
9

    
10
unsigned int samples[N_SAMPLES];
11
int sample_idx;
12

    
13
unsigned int sum;
14
unsigned long sum_sq;
15

    
16
void current_init() {
17

    
18
  /* TODO reduce power consumption with DIDR* */
19

    
20
  /*
21
   * COM1A = COM1B = 0, disconnect pins
22
   * WGM1 = 4, clear timer on compare A
23
   * CS1 = 1, no prescaler
24
   */
25
  TCCR1B = _BV(WGM12) | _BV(CS10);
26

    
27
  /* Timer is cleared on A, ADC is triggered on B */
28
  OCR1A = F_CPU / SAMPLES_PER_CYCLE / CYCLES_PER_SECOND;
29
  OCR1B = OCR1A;
30

    
31
  /*
32
   * REFS = 0, Vcc reference (set to 2 for internal 1.1V reference)
33
   * MUX = 8, PB3(ADC8)
34
   */
35
  ADMUX = _BV(MUX3);
36
  DIDR1 |= _BV(ADC8D);
37

    
38
  /*
39
   * ADLAR = 0, right adjust result
40
   * ADTS = 5, start on timer 1 compare match B
41
   */
42
  ADCSRB = _BV(ADTS2) | _BV(ADTS0);
43

    
44
  /*
45
   * ADEN = 1, enable
46
   * ADSC = 0, don't start yet
47
   * ADATE = 1, auto trigger
48
   * ADIE = 1, enable interrupt
49
   * ADPS = 6, prescale clock by 64
50
   */
51
  ADCSRA = _BV(ADEN) | _BV(ADATE) | _BV(ADIE) | _BV(ADPS2) | _BV(ADPS1);
52
}
53

    
54
ISR(ADC_vect) {
55
  unsigned int old, new;
56

    
57
  /* clear the timer interrupt flag so that the ADC will be triggered again
58
   * next time the timer resets */
59
  TIFR |= _BV(OCF1B);
60

    
61
  new = ADC;
62

    
63
  /* put sample into ring buffer */
64
  old = samples[sample_idx];
65
  samples[sample_idx++] = new;
66
  if (sample_idx == N_SAMPLES)
67
    sample_idx = 0;
68

    
69
  /* keep a running total of samples and samples squared */
70
  sum += new;
71
  sum -= old;
72
  sum_sq += new*new;
73
  sum_sq -= old*old;
74
}
75

    
76
/*unsigned char isqrt(unsigned int x) {
77
  unsigned int sqrt, mulmask;
78
  sqrt = 0;
79
  mulmask = 0x80;
80
  if (x > 0) {
81
    while (mulmask) {
82
      sqrt |= mulmask;
83
      if (sqrt * sqrt > x)
84
        sqrt &= ~mulmask;
85
      mulmask >>= 1;
86
    }
87
  }
88
  return sqrt;
89
}*/
90

    
91
unsigned int current_read() {
92
  unsigned int _sum;
93
  unsigned long _sum_sq;
94

    
95
  cli();
96
  _sum = sum;
97
  _sum_sq = sum_sq;
98
  sei();
99

    
100
  /* calculate the variance using sum and sum_sq */
101
  return (N_SAMPLES*_sum_sq - (unsigned long)_sum*_sum) /
102
    (N_SAMPLES*(N_SAMPLES-1));
103
}