add pro support

digistump · Jun 23, 2015 · 68ad726 · 68ad726
1 parent 6ca6b11
commit 68ad726
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/digistump-avr/libraries/DigisparkRGB/DigisparkRGB.cpp b/digistump-avr/libraries/DigisparkRGB/DigisparkRGB.cpp
@@ -37,10 +37,19 @@ void DigisparkRGBBegin() {
   DigisparkPWMcompbuff[1] = 0x00;// set default PWM values
   DigisparkPWMcompbuff[2] = 0x00;// set default PWM values
 
-
+#ifdef TIFR
   TIFR = (1 << TOV0);           // clear interrupt flag
-  TIMSK = (1 << TOIE0);         // enable overflow interrupt
-  TCCR0B = (1 << CS00);         // start timer, no prescale
+#endif
+#ifdef TIFR0
+  TIFR0 = (1 << TOV0);           // clear interrupt flag
+#endif
+  #ifdef TIMSK
+  TIMSK = (1 << TOV0);           // enable overflow interrupt
+#endif
+#ifdef TIMSK0
+  TIMSK0 = (1 << TOV0);           // enable overflow interrupt
+#endif
+
 
   sei(); 
 }