X-Git-Url: https://www.fi.muni.cz/~kas/git//home/kas/public_html/git/?p=bike-lights.git;a=blobdiff_plain;f=firmware%2Fpwm.c;h=201770f8152f440fe7ef21a219c336fb9f382b5f;hp=2df2e6bb8b9abe762c1bb6208a13e6a4dde4cd7c;hb=320b394fa14a973bcb68fcbd2af65a79462e30bc;hpb=ddc8f4be4d90e7e709832b5135b2e7e4631b1554

diff --git a/firmware/pwm.c b/firmware/pwm.c
index 2df2e6b..201770f 100644
--- a/firmware/pwm.c
+++ b/firmware/pwm.c
@@ -4,8 +4,6 @@
 
 #include "lights.h"
 
-#define PWM_MAX 0x1FF
-
 void init_pwm()
 {
 	/* Async clock */
@@ -23,10 +21,11 @@ void init_pwm()
 	TCCR1A = _BV(COM1A1) | _BV(COM1B1) | _BV(PWM1A) | _BV(PWM1B);
 	TCCR1D = 0;
 	TCCR1B = _BV(CS10); 			// no clock prescaling
+
 	TC1H = PWM_MAX >> 8;
 	OCR1C = PWM_MAX & 0xFF;				// TOP value
 
-	TC1H = PWM_MAX >> 8;
+	TC1H = PWM_MAX >> 8;		// PWM3 is inverted
 	OCR1D = PWM_MAX & 0xFF;
 
 	TC1H = 0x00;
@@ -36,13 +35,12 @@ void init_pwm()
 	PORTB &= ~(_BV( PB1 ) | _BV( PB3 ) | _BV( PB5 )); // set to zero
 }
 
-void pwm_on(unsigned char n)
+void susp_pwm()
 {
-	switch (n) {
-	case 0: DDRB |= _BV(PB1); break;
-	case 1: DDRB |= _BV(PB3); break;
-	case 2: DDRB |= _BV(PB5); break;
-	}
+	DDRB &= ~(_BV( PB1 ) | _BV( PB3 ) | _BV( PB5 ));
+	TCCR1D = TCCR1C = TCCR1B = TCCR1A = 0;
+	TIMSK = 0;
+	TIFR = 0;
 }
 
 void pwm_off(unsigned char n)
@@ -54,20 +52,34 @@ void pwm_off(unsigned char n)
 	}
 }
 
-void pwm_set(unsigned char n, unsigned char stride)
+void pwm_set(unsigned char n, uint16_t stride)
 {
-	TC1H = 0x00;
+	unsigned char hi, lo;
+
+	if (stride > PWM_MAX)
+		stride = PWM_MAX;
+
+	if (n == 2)
+		stride = PWM_MAX - stride;
+
+	hi = stride >> 8;
+	lo = stride & 0xFF;
+
 	switch (n) {
-	case 0: OCR1A = stride; break;
-	case 1: OCR1B = stride; break;
-	case 2: {
-			uint16_t s16 = PWM_MAX - (uint16_t)stride;
-			volatile unsigned char hi, lo;
-			hi = s16 >> 8;
-			lo = s16 & 0xFF;
-			TC1H = hi;
-			OCR1D = lo;
-		}
+	case 0:
+		TC1H = hi;
+		OCR1A = lo;
+		DDRB |= _BV(PB1);
+		break;
+	case 1:
+		TC1H = hi;
+		OCR1B = lo;
+		DDRB |= _BV(PB3);
+		break;
+	case 2:
+		TC1H = hi;
+		OCR1D = lo;
+		DDRB |= _BV(PB5);
 		break;
 	}
 }