works but color is wrong

This commit is contained in:
hugo LAMY
2025-03-10 15:16:39 +01:00
parent 048db02593
commit c3357fcfaa
6 changed files with 260 additions and 90 deletions

77
module03/ex03/colors.c Normal file
View File

@@ -0,0 +1,77 @@
#include "header.h"
// Table 14-9 : Port D Pins Alternate Functions
// OC2B (PD3) : Red
// OC0B (PD5) : Green
// OC0A (PD6) : Blue
void init_rgb() {
MODE_OUTPUT(RGB5_RED);
MODE_OUTPUT(RGB5_GEEN);
MODE_OUTPUT(RGB5_BLUE);
TCCR0A = 0; // 15.9.1 : Clear Timer0 control register A
TCCR0B = 0; // 15.9.2 : Clear Timer0 control register B
TCCR2A = 0; // Clear Timer2 control register A
TCCR0A |= (1 << WGM00) | (1 << WGM01); // Table 15-8 : Timer0 in Fast PWM mode
TCCR2A |= (1 << WGM20) | (1 << WGM21); // Table 18-8 : Timer2 in Fast PWM mode
TCCR0A |= (1 << COM0A1); // Table 15-3 : BLUE - Timer0 Non-inverting mode for Blue (OC0A -> PD6)
TCCR0A |= (1 << COM0B1); // Table 15-6 : GREEN - Timer0 Non-inverting mode for Green (OC0B -> PD5)
TCCR2A |= (1 << COM2B1); // Table 18-6 : RED - Timer2 Non-inverting mode for Red (OC2B -> PD3)
TCCR0B |= (1 << CS01); // Table 15-9 : Prescaler 8 → ~8kHz PWM frequency
TCCR2B |= (1 << CS21); // Table 18-9 : Prescaler 8 → ~8kHz PWM frequency
OCR0A = 0; // Blue OFF
OCR0B = 0; // Green OFF
OCR2B = 0; // Red OFF
}
void set_rgb(uint8_t r, uint8_t g, uint8_t b) {
OCR2B = r; // Red (PD3, Timer2 OCR2B)
OCR0B = g; // Green (PD5, Timer0 OCR0B)
OCR0A = b; // Blue (PD6, Timer0 OCR0A)
}
// convert hex to int : https://stackoverflow.com/a/39052987
uint8_t hex_to_int(char c) {
if (c >= '0' && c <= '9') {
return c - '0';
}
if (c >= 'A' && c <= 'F') {
return c - 'A' + 10;
}
if (c >= 'a' && c <= 'f') {
return c - 'a' + 10;
}
return 0;
}
// "<< 4" is equivalent to "* 16"
uint8_t get_red(char *input) {
if (input[0] != '#') {
return 0;
}
return (hex_to_int(input[1]) << 4) + hex_to_int(input[2]);
}
uint8_t get_green(char *input) {
if (input[0] != '#') {
return 0;
}
return (hex_to_int(input[3]) << 4) + hex_to_int(input[4]);
}
uint8_t get_blue(char *input) {
if (input[0] != '#') {
return 0;
}
return (hex_to_int(input[5]) << 4) + hex_to_int(input[6]);
}
void set_color(char *color_input) {
uint8_t r = get_red(color_input);
uint8_t g = get_green(color_input);
uint8_t b = get_blue(color_input);
set_rgb(r, g, b);
}

39
module03/ex03/header.h Normal file
View File

@@ -0,0 +1,39 @@
#ifndef HEADER_H
#define HEADER_H
#include <avr/io.h>
#include <util/delay.h>
#include <avr/interrupt.h>
#include "utils.h"
#include "bitmanip.h"
#include "timer.h"
#include "usart.h"
#include "interrupt.h"
//
// PROTOTYPES
//
// main.c
// colors.c
void init_rgb();
void set_rgb(uint8_t r, uint8_t g, uint8_t b);
uint8_t get_red(char *input);
uint8_t get_green(char *input);
uint8_t get_blue(char *input);
uint8_t hex_to_int(char c);
void uart_init();
void set_color(char *color_input);
// uart.c
void uart_init();
char uart_rx(void);
void uart_tx(char c);
// void uart_printstr(const char* str);
void remove_last_character();
void fill_str(char input);
int strlength(volatile char *str);
void reset_input();
int is_valid_color_input(char c);
#endif // HEADER_H

View File

@@ -1,94 +1,11 @@
#include <avr/io.h>
#include <util/delay.h>
#include <avr/interrupt.h>
#include "header.h"
#include "utils.h"
#include "bitmanip.h"
// prototypes
void init_rgb();
void set_rgb(uint8_t r, uint8_t g, uint8_t b);
uint8_t get_red(char *input);
uint8_t get_green(char *input);
uint8_t get_blue(char *input);
uint8_t hex_to_int(char c);
// Table 14-9 : Port D Pins Alternate Functions
// OC2B (PD3) : Red
// OC0B (PD5) : Green
// OC0A (PD6) : Blue
void init_rgb() {
MODE_OUTPUT(RGB5_RED);
MODE_OUTPUT(RGB5_GEEN);
MODE_OUTPUT(RGB5_BLUE);
TCCR0A = 0; // 15.9.1 : Clear Timer0 control register A
TCCR0B = 0; // 15.9.2 : Clear Timer0 control register B
TCCR2A = 0; // Clear Timer2 control register A
TCCR0A |= (1 << WGM00) | (1 << WGM01); // Table 15-8 : Timer0 in Fast PWM mode
TCCR2A |= (1 << WGM20) | (1 << WGM21); // Table 18-8 : Timer2 in Fast PWM mode
TCCR0A |= (1 << COM0A1); // Table 15-3 : BLUE - Timer0 Non-inverting mode for Blue (OC0A -> PD6)
TCCR0A |= (1 << COM0B1); // Table 15-6 : GREEN - Timer0 Non-inverting mode for Green (OC0B -> PD5)
TCCR2A |= (1 << COM2B1); // Table 18-6 : RED - Timer2 Non-inverting mode for Red (OC2B -> PD3)
TCCR0B |= (1 << CS01); // Table 15-9 : Prescaler 8 → ~8kHz PWM frequency
TCCR2B |= (1 << CS21); // Table 18-9 : Prescaler 8 → ~8kHz PWM frequency
OCR0A = 0; // Blue OFF
OCR0B = 0; // Green OFF
OCR2B = 0; // Red OFF
}
void set_rgb(uint8_t r, uint8_t g, uint8_t b) {
OCR2B = r; // Red (PD3, Timer2 OCR2B)
OCR0B = g; // Green (PD5, Timer0 OCR0B)
OCR0A = b; // Blue (PD6, Timer0 OCR0A)
}
// convert hex to int : https://stackoverflow.com/a/39052987
uint8_t hex_to_int(char c) {
if (c >= '0' && c <= '9') {
return c - '0';
}
if (c >= 'A' && c <= 'F') {
return c - 'A' + 10;
}
if (c >= 'a' && c <= 'f') {
return c - 'a' + 10;
}
return 0;
}
// "<< 4" is equivalent to "* 16"
uint8_t get_red(char *input) {
if (input[0] == '#') {
return (hex_to_int(input[1]) << 4) + hex_to_int(input[2]);
}
return 0;
}
uint8_t get_green(char *input) {
if (input[0] == '#') {
return (hex_to_int(input[3]) << 4) + hex_to_int(input[4]);
}
return 0;
}
uint8_t get_blue(char *input) {
if (input[0] == '#') {
return (hex_to_int(input[5]) << 4) + hex_to_int(input[6]);
}
return 0;
}
// led RGB D5 must turns on in a loop of colors using PWM
int main() {
init_rgb();
uart_init();
char *input = "#FFFF00";
uint8_t r = get_red(input);
uint8_t g = get_green(input);
uint8_t b = get_blue(input);
set_rgb(r, g, b);
while (1);
}

121
module03/ex03/uart.c Normal file
View File

@@ -0,0 +1,121 @@
#include "header.h"
// MACROS
// USART
#define USART_BAUDRATE 115200
#define INPUT_SIZE 7
// GLOBAL VARIABLES
volatile char color_input[INPUT_SIZE + 1] = {0};
volatile int input_index = 0;
//
// FUNCTIONS
//
void uart_init() {
UBRR0H = (unsigned char) (BAUD_PRESCALER(USART_BAUDRATE) >> 8); // 20.11.5 : UBRRnL and UBRRnH USART Baud Rate Registers
UBRR0L = (unsigned char) BAUD_PRESCALER(USART_BAUDRATE);
UCSR0C |= ASYNCHRONOUS | PARITY_DISABLED | STOP_ONE_BIT | DATA_EIGHT_BIT; // 20.11.4 : set Frame Format
UCSR0B |= RECEIVER_ENABLED | TRANSMITTER_ENABLED | INTERRUPT_RECEIVER_ENABLED; // 20.11.3 : enable Receiver and Transmitter, and interrupt on receiver
SREG |= ENABLE_GLOBAL_INTERRUPT; // 7.3.1 : Status Register, bit 7 : I Global Interrupt Enable
}
char uart_rx(void) {
while (TEST(UCSR0A, RXC0) == 0); // 20.11.2 : do nothing until there are unread data in the receive buffer (UDRn), (RXCn flag in UCSRnA register set to 1 when buffer has data)
return UDR0; // 20.11.1 : get data in buffer, UDRn USART I/O Data Register (read and write)
}
void uart_tx(char c) {
while (TEST(UCSR0A, UDRE0) == 0); // 20.11.2 : do nothing until UDRn buffer is empty, (UDREn flag in UCSRnA register set to 1 when buffer empty)
UDR0 = (unsigned char) c; // 20.11.1 : Put data into buffer, UDRn USART I/O Data Register (read and write)
}
// void uart_printstr(const char* str) {
// while (*str) {
// uart_tx(*str);
// str++;
// }
// }
ISR(USART_RX_vect) { // Table 12-7 : we select the code for USART Receive
char received_char = UDR0; // read received character
if (received_char == '\b' || received_char == 127) { // if backspace is received
if (input_index <= 0) {
return;
}
if (input_index >= INPUT_SIZE) {
return;
}
remove_last_character();
input_index--;
uart_tx('\b'); // Move cursor back
uart_tx(' '); // Erase the character on screen
uart_tx('\b'); // Move cursor back again
} else if (received_char == '\n' || received_char == '\r') { // if enter is received
if (input_index != INPUT_SIZE) {
return;
}
set_color((char *)color_input);
reset_input();
} else { // any other character
if (input_index >= INPUT_SIZE) {
return;
}
if (!is_valid_color_input(received_char)) {
return;
}
fill_str(received_char);
uart_tx(received_char);
input_index++;
}
}
int is_valid_color_input(char c) {
if (input_index == 0) {
if (c == '#') {
return TRUE;
} else {
return FALSE;
}
}
if (input_index >= INPUT_SIZE) {
return FALSE;
}
if (c >= '0' && c <= '9') {
return TRUE;
} else if (c >= 'A' && c <= 'F') {
return TRUE;
} else if (c >= 'a' && c <= 'f') {
return TRUE;
}
return FALSE;
}
void fill_str(char input) {
if (input_index >= INPUT_SIZE) {
return;
}
color_input[input_index] = input;
}
void remove_last_character() {
color_input[input_index] = 0;
}
int strlength(volatile char *str) {
int length = 0;
while (*str) {
length++;
str++;
}
return length;
}
void reset_input() {
for(int i = 0; i < INPUT_SIZE; i++) {
color_input[i] = 0;
}
uart_tx('\r');
uart_tx('\n');
input_index = 0;
}