From a0d085a0d0872abd53a523b26d2f246e5e548a7f Mon Sep 17 00:00:00 2001
From: zloihach <72695485+zloihach@users.noreply.github.com>
Date: Thu, 22 Jun 2023 15:41:24 +0300
Subject: [PATCH] extended pwm library

---
 main.cpp | 103 +++++++++++++++++++++++--------------------------------
 1 file changed, 42 insertions(+), 61 deletions(-)

diff --git a/main.cpp b/main.cpp
index 867e600..5a31bfc 100644
--- a/main.cpp
+++ b/main.cpp
@@ -1,37 +1,45 @@
-//VERSION #2
 #include <avr/io.h>
 
-
 void pwm_init() {
-    TCCR1A |= (1 << COM1A1) | (1 << WGM11); // non-inverting mode, Fast PWM (mode 14)
-    TCCR1B |= (1 << WGM12) | (1 << WGM13) | (1 << CS10); // Fast PWM (mode 14), prescaler = 1
-    DDRB |= (1 << PB1); // set pin PB1 as output
+    // Configure pin PB1 as output
+    DDRB |= (1 << PB1);
+    // Configure Timer1 in Fast PWM (mode 14) with non-inverting mode
+    TCCR1A |= (1 << COM1A1) | (1 << WGM11);
+    TCCR1B |= (1 << WGM12) | (1 << WGM13) | (1 << CS10);
 }
 
 void pwm_set_frequency(uint16_t frequency) {
+    // Calculate the top value for Timer1
     uint16_t prescaler = 1;
     uint32_t top = F_CPU / (prescaler * frequency) - 1;
     ICR1 = top;
 }
 
-void pwm_set_duty_cycle(uint8_t dutyCycle) {
-    uint16_t value = ICR1 * dutyCycle / 100.0;
+void pwm_set_duty_cycle(uint8_t duty_cycle) {
+    // Calculate the OCR1A value corresponding to the duty cycle
+    uint16_t value = ICR1 * duty_cycle / 100;
     OCR1A = value;
 }
 
 void pwm_enable() {
-    pwm_set_duty_cycle(50); // Начальная скважность
+    pwm_set_duty_cycle(50); // Initial duty cycle
 }
 
 void pwm_disable() {
-    pwm_set_duty_cycle(0); // Выключение ШИМ
+    pwm_set_duty_cycle(0); // Turn off PWM
+}
+
+uint16_t pwm_get_frequency() {
+    uint16_t prescaler = 1;
+    uint32_t frequency = F_CPU / (prescaler * (ICR1 + 1));
+    return frequency;
+}
+
+uint8_t pwm_get_duty_cycle() {
+    uint8_t duty_cycle = OCR1A * 100 / ICR1;
+    return duty_cycle;
 }
 
-void pwm_init();
-void pwm_set_frequency(uint16_t frequency);
-void pwm_set_duty_cycle(uint8_t dutyCycle);
-void pwm_enable();
-void pwm_disable();
 
 #define F_CPU 16000000UL
 #define I2C_FREQ 100000UL
@@ -91,12 +99,8 @@ uint8_t i2c_read_nack() {
     return TWDR;
 }
 
-void i2c_init();
-void i2c_start();
-void i2c_stop();
-void i2c_write(uint8_t data);
-uint8_t i2c_read_ack();
-uint8_t i2c_read_nack();
+
+
 
 const uint8_t BUTTON_PIN[] = {0, 1, 2, 3, 4, 5};
 const uint8_t PWM_SLAVE_ADDR = 9;
@@ -116,6 +120,8 @@ void setup() {
 
     Serial.begin(9600); // Инициализируем Serial монитор
     Serial.println("PWM Controller started!");
+    Serial.println("PWM on master device initial successful!");
+
     sendCommand(0x01, 0.0); // Включить ШИМ при запуске
 }
 
@@ -155,47 +161,26 @@ void checkButton(uint8_t pin, const char* message, uint16_t cmd, float value) {
     }
 }
 
+
 void sendCommand(uint16_t cmd, float value) {
-    if (cmd != 0x01 && cmd != 0x02) { // если команда изменения параметров
-        switch (cmd) {
-            case 0x03: // Увеличить частоту на 25% от текущего.
-                value = frequency;
-                value *= 0.25;
-                break;
-            case 0x04: // Уменьшить частоту на 20% от текущего.
-                value = frequency;
-                value *= 0.2;
-                break;
-            case 0x05: // Увеличить скважность на 10% от текущего.
-                value = dutyCycle;
-                value *= 0.1;
-                break;
-            case 0x06: // Уменьшить скважность на 10% от текущего.
-                value = dutyCycle;
-                value *= 0.1;
-                break;
-        }
+    // Проверяем, является ли команда изменением частоты или коэффициента заполнения
+    if (cmd == 0x03 || cmd == 0x04) {
+        value = pwm_get_frequency(); // Если да, получаем текущее значение частоты
+    } else if (cmd == 0x05 || cmd == 0x06) {
+        value = pwm_get_duty_cycle(); // Если да, получаем текущее значение коэффициента заполнения
     }
-    // uint16_t data = (uint16_t)(value * 16.0);
-    // data |= cmd << 4;
-    // i2c_start();
-    // i2c_write(PWM_SLAVE_ADDR << 1);
-    // i2c_write(data >> 8);
-    // i2c_write(data & 0xFF);
-    // i2c_stop();
-    // Serial.print("Sent command: ");
-    // Serial.print(cmd, HEX);
-    // Serial.print(", value: ");
-    // Serial.println(value);
-
 
+    // Кодируем команду и значение в 16-битное число
     uint16_t cmd_value = cmd << 12 | (uint16_t)(value * 16.0f);
-    cmd_value |= cmd;
-    i2c_start();
-    i2c_write(PWM_SLAVE_ADDR << 1);
-    i2c_write(cmd_value >> 8);    // старший байт команды
-    i2c_write(cmd_value & 0xFF);  // младший байт команды
-    i2c_stop();
+
+    // Отправляем команду через шину I2C
+    i2c_start(); // Начинаем передачу
+    i2c_write(PWM_SLAVE_ADDR << 1); // Отправляем адрес устройства
+    i2c_write(cmd_value >> 8);    // Передаем старший байт команды
+    i2c_write(cmd_value & 0xFF);  // Передаем младший байт команды
+    i2c_stop(); // Завершаем передачу
+
+    // Выводим информацию о команде в монитор последовательного порта
     Serial.print("Sent command: 0x");
     Serial.print(cmd, HEX);
     Serial.print(", value: ");
@@ -204,8 +189,4 @@ void sendCommand(uint16_t cmd, float value) {
     Serial.print(cmd_value >> 8, HEX);
     Serial.print(" ");
     Serial.println(cmd_value & 0xFF, HEX);
-
 }
-
-
-