touchless_tracking.ino 2.4 KB
Newer Older
hackEns's avatar
hackEns committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/* Touchless tracking
 * ==================
 * This code takes as many measurements as it can at approximately
 * 10Hz = 60 Hz / (2 full cycles * 3 sensors).
 *
 * The code takes as many measurements as possible on the period of two
 * cycles of the main power frequency, in order to cancel out any potential
 * coupling.
 *
 * Original code was found on instructables :
 * http://www.instructables.com/id/DIY-3D-Controller/
 *
 * Code slightly modified and commented by Phyks for HackENS
 *
 * As I found the code on Instructables and I don't give a damn to the 
 * modifications I made, you may reuse it freely by writing where you 
 * found it.
 */

#define resolution 8
// Frequency of main power to avoid coupling
#define mains 50
// Time between measures
#define refresh 2 * 1000000 / mains
#define MAX_PINS 6

// Pins to use
// Must be between 8 and 13 to use this code without 
// modifications (PORTB is for pins 8 to 13)
//
// Note : If you change pin numbers, you *must* change the mask.
//        See arduino doc for more info.
int pins[MAX_PINS] = {8, 9, 10, 11, 12}

// Counter for the timer
extern volatile unsigned long timer0_overflow_count;

void startTimer() {
    timer0_overflow_count = 0;
    TCNT0 = 0; // Initialize counter value to 0
}

unsigned long checkTimer() {
    return ((timer0_overflow_count << 8) + TCNT0) << 2;
}

long time(int pin, byte mask) {
    unsigned long count = 0, total = 0;

    while(checkTimer() < refresh) {
        // Note : pinMode is about 6 times slower than assigning
        // DDRB directly, but that pause is important

        // Set pin as output and LOW, see arduino doc for info on PORTB
        pinMode(pin, OUTPUT);
        PORTB = 0;

        // Set it as INPUT to take measure
        pinMode(pin, INPUT);

        // While PINB is low, increment counter
        while((PINB & mask) == 0)
            count++;

        total++;
    }
    // Restart timer for next measurement
    startTimer();
    // Return the measurement result
    return (count << resolution) / total;
}

void setup() {
    // Initialize serial communication
    Serial.begin(115200);

    // INPUT on pins
    pinMode(PIN_R, INPUT);
    pinMode(PIN_G, INPUT);
    pinMode(PIN_B, INPUT);

    // Start timer
    startTimer();
}

void loop() {
    // Print output to serial in decimal
    for(int i = 0; i < MAX_PINS; i++) {
        Serial.print(time(PINS[i], i));
        Serial.print(" ");
    }
    Serial.println();
}