/*
 * 
 *
 * Created: 21.1.2023 15:15:24
 * Author : Kryštof Keil
 * E-mail : keilkrys@fel.cvut.cz
 */

// fuses nastavit na
//D9
//EF
//07
#define F_CPU 8000000

#include <stdio.h>
#include <avr/io.h>
#include <util/delay.h>
#include <inttypes.h>
#include <stdlib.h>
#include <avr/interrupt.h>
#include <avr/sleep.h>
#include <avr/power.h>
#include "DHT.c"

#define myUBRR 16				// (F_CPU/(16*bps))-1	-> 57600bps
#define UBRR9600 104             // -> 9600bps
uint32_t maxWaitycles = 1000000;
void USART_init(int);
int LoRa_init(void);
int LoRa_send(void);
void LoRa_sleep(void);
void ReadPM25sensor(void);
char numberToCharLow(char);
char numberToCharHigh(char);
int readADCbattery(void);
int readPhotoResistor(void);

uint8_t resetprogramu, indexRecv, alreadyRegistered, firstRun;
uint16_t temperature = 22, humidity = 508;
uint32_t waitCycles, counter; 
char recvBytes[32];

void USART_init(int givenUBRR){
	UBRR0H = (unsigned char)(givenUBRR>>8);	//nastaveni BAUD
	UBRR0L = (unsigned char)givenUBRR;			//nastaveni BAUD
	UCSR0A = 0b00000010;					//Asynchronous Double Speed mode
	UCSR0B = 0b00011000;					//zapnut� vysilani a zapnuti prijmu
	UCSR0C = 0b00000110;					//asynchroni, bez parity, 1stop, 8bitu
}

int LoRa_init(void){ //inicializace modulu					
    PORTD &= ~(1<<PD2); //reset modulu
    _delay_ms(100);
    PORTD |= (1<<PD2); //resetPIN modulu HIGH aby byl aktivni
    _delay_ms(2000);
    volatile unsigned int buff;

	char mac[] = "mac join otaa";				//prikaz pro pripojeni se k lorawan siti
	for (int i = 0;i<sizeof(mac);i++){
		while (!( UCSR0A & (1<<UDRE0))) {}
		UDR0 = mac[i];
	}
		while (!( UCSR0A & (1<<UDRE0))) {}
		UDR0 = 0x0D;
		while (!( UCSR0A & (1<<UDRE0))) {}
		UDR0 = 0x0A;

	while((buff != 'o') && (waitCycles < maxWaitycles)) {		//zprava o prijeti pozadavku na registraci v siti
        while (!(UCSR0A & (1<<RXC0)) && (waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
                if(buff == 'd' || buff == 'n' || buff == 'i' || buff == 'k' || buff == 's' || buff == 'b' || buff == 'm'){
                    return 0;
                }
	}
	waitCycles = 0;
    while((buff != 'k')&&(waitCycles < maxWaitycles)) {		//zprava o prijeti pozadavku na registraci v siti
        while (!(UCSR0A & (1<<RXC0))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
	}
	waitCycles = 0;
    while((buff != 0x0D) && (waitCycles < maxWaitycles)){		//zprava o prijeti pozadavku na registraci v siti
        while (!(UCSR0A & (1<<RXC0))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
	}
	waitCycles = 0;
    while((buff != 0x0A) && (waitCycles < maxWaitycles)) {		//zprava o prijeti pozadavku na registraci v siti
        while (!(UCSR0A & (1<<RXC0))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
	}
	waitCycles = 0;
    char accepted[] = "accepted";
    uint8_t pocitadloAccepted = 0;
	buff = 0;
    while(pocitadloAccepted < sizeof(accepted)){
        while((buff != accepted[pocitadloAccepted]) && (waitCycles < maxWaitycles)) {		//zprava o prijeti pozadavku na registraci v siti
            while ((!(UCSR0A & (1<<RXC0)))) {waitCycles++;}
            buff = UDR0;
			if(buff == 'd' && pocitadloAccepted < 1){
				PORTC &= ~(1<<PORTC2);	//zhasnout debug LEDku
				_delay_ms(200);
				PORTC |= (1<<PORTC2);	//rozsvitit debug LEDku
				_delay_ms(200);
				PORTC &= ~(1<<PORTC2);	//zhasnout debug LEDku
				_delay_ms(200);
				PORTC |= (1<<PORTC2);	//rozsvitit debug LEDku
                return 0;
            }
        }
		waitCycles = 0;
        buff = 0;
        pocitadloAccepted++;
    }
    while((buff != 0x0D) && (waitCycles < maxWaitycles)){		//zprava o prijeti pozadavku na registraci v siti
        while (!(UCSR0A & (1<<RXC0))) {waitCycles++;}
				buff = UDR0;
    }
	waitCycles = 0;
    while((buff != 0x0A) && (waitCycles < maxWaitycles)){		//zprava o prijeti pozadavku na registraci v siti
        while (!(UCSR0A & (1<<RXC0))) {waitCycles++;}
				buff = UDR0;
    }
	if(waitCycles == maxWaitycles) {
		waitCycles = 0;
		return 0;
	}
	waitCycles = 0;
    return 1;
}

int LoRa_send(void){
	volatile unsigned int buff;
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0D;
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0A; // posleme jeden dummy prikaz abysme meli cisty canvas pro dalsi prikaz
	buff = 0;
	int battVoltage = readADCbattery(); 
	int solarIntensity = readPhotoResistor();
	for (int i = 0; i < 2; i++){
		while(dht_GetTempUtil(&temperature, &humidity) == -1) {}
	}
	char rx[] = "mac tx uncnf 50 aaabacadaeafbabbbcbdbebfcacb";		//vysilat zpravu pres LoRaWAN sit
	rx[16] = numberToCharHigh(recvBytes[6]); //jeden bajt informace se zakoduje do 2 charů prikazu  PPPPP   MM   MM   222        55555555
	rx[17] = numberToCharLow(recvBytes[6]);  //PMS5003 zakoduje kazdou velicinu do 2 bajtu          P   PP  M M M M  2   2       5
	rx[18] = numberToCharHigh(recvBytes[7]);  //													PPPPP   M  M  M     2        55555
	rx[19] = numberToCharLow(recvBytes[7]);   //                                                    P       M     M   2       ..       55
	                                          //													P       M     M  222222   .. 555555
	rx[20] = numberToCharHigh(recvBytes[4]);  //        PPPPP   MM   MM       1        00000           
	rx[21] = numberToCharLow(recvBytes[4]);   //        P   PP  M M M M     1 1       0     0
	rx[22] = numberToCharHigh(recvBytes[5]);  //		PPPPP   M  M  M    1  1       0     0
	rx[23] = numberToCharLow(recvBytes[5]);   //		P       M     M       1   ..  0     0
											  //        P       M     M       1   ..   00000
	rx[24] = numberToCharHigh(recvBytes[8]);  //        PPPPP   MM   MM       1    00000
	rx[25] = numberToCharLow(recvBytes[8]);   //        P   PP  M M M M     1 1   0     0
	rx[26] = numberToCharHigh(recvBytes[9]);  //		PPPPP   M  M  M    1  1   0     0
	rx[27] = numberToCharLow(recvBytes[9]);   //		P       M     M       1   0     0
											  //        P       M     M       1    00000			
	rx[28] = numberToCharHigh(battVoltage >> 8);
	rx[29] = numberToCharLow(battVoltage >> 8);
	rx[30] = numberToCharHigh(battVoltage & 0xff);	
	rx[31] = numberToCharLow(battVoltage & 0xff);
	rx[32] = numberToCharHigh(solarIntensity >> 8);
	rx[33] = numberToCharLow(solarIntensity >> 8);
	rx[34] = numberToCharHigh(solarIntensity & 0xff);
	rx[35] = numberToCharLow(solarIntensity & 0xff);
	rx[36] = numberToCharHigh(temperature >> 8);
	rx[37] = numberToCharLow(temperature >> 8);
	rx[38] = numberToCharHigh(temperature & 0xff);
	rx[39] = numberToCharLow(temperature & 0xff);
	rx[40] = numberToCharHigh(humidity >> 8);
	rx[41] = numberToCharLow(humidity >> 8);
	rx[42] = numberToCharHigh(humidity & 0xff);
	rx[43] = numberToCharLow(humidity & 0xff);				  
	for (int i = 0;i<sizeof(rx);i++)
	{
		while (!( UCSR0A & (1<<UDRE0))) {}
		UDR0 = rx[i];
	}
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0D;
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0A;
    while((buff != 'o')&&(waitCycles < maxWaitycles)) {		//zprava o uspesnem odvysilani
        while ((!(UCSR0A & (1<<RXC0)))&&(waitCycles < maxWaitycles)) {waitCycles++;}
            buff = UDR0;
           // if(buff == 'n' || buff == 'i' || buff == 'b' || buff == 's'|| buff == 'f'|| buff == 'm'){ //detekovano n coz znaci no_free_channel
			  /* if (buff == 0x6e){
                while(buff != 0x0A) {		//cekame na posledni odvysilany znak CR/LF
                while (!(UCSR0A & (1<<RXC0))) {}
                buff = UDR0;
            }
            return 1; //vrátíme návratovou hodnotu 1 a za daný čas zkusím odvysílat znova, ten LORa modul nemá rád když to zkoušíme furt dokola
        }*/
    }
	waitCycles = 0;
    while((buff != 'k')&&(waitCycles < maxWaitycles)) {		//zprava o uspesnem odvysilani
        while ((!(UCSR0A & (1<<RXC0)))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
    }
	waitCycles = 0;
    while((buff != 0x0D)&&(waitCycles < maxWaitycles)){		//zprava o uspesnem odvysilani
        while ((!(UCSR0A & (1<<RXC0)))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
    }
	waitCycles = 0;
    while((buff != 0x0A)&&(waitCycles < maxWaitycles)) {		//zprava o uspesnem odvysilani
        while ((!(UCSR0A & (1<<RXC0)))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
    }
	waitCycles = 0;
	buff = 0;
	_delay_ms(1000);
    char mac[] = "mac_"; // TODO (pripadne) ulozeni RX prichozich dat
    uint8_t pocitadloMac = 0;
    while(pocitadloMac < sizeof(mac)){
	    while(buff != mac[pocitadloMac] && (waitCycles < maxWaitycles)) {
		    while ((!(UCSR0A & (1<<RXC0)))&&(waitCycles < maxWaitycles)) {waitCycles++;}
		    buff = UDR0;
	    }
		waitCycles = 0;
	    buff = 0;
	    pocitadloMac++;
    }
    while(buff != 0x0A &&(waitCycles < maxWaitycles)) {		//zprava o uspesnem odvysilani cekani na posledni znak
        while ((!(UCSR0A & (1<<RXC0)))&&(waitCycles < maxWaitycles)) {waitCycles++;}
				buff = UDR0;
    }
	waitCycles = 0;
    return 1;
}
void LoRa_sleep(void){
    char sleep[] = "sys sleep 1800000";		//uvest modul do spanku na dany pocet ms
	for (int i = 0;i<sizeof(sleep);i++)
	{
		while (!( UCSR0A & (1<<UDRE0))) {}
		UDR0 = sleep[i];
	}
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0D;
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0A;
    _delay_ms(10);
}
void LoRa_sleep_short(void){
	char sleep[] = "sys sleep 11000";		//uvest modul do spanku na dany pocet ms
	for (int i = 0;i<sizeof(sleep);i++)
	{
		while (!( UCSR0A & (1<<UDRE0))) {}
		UDR0 = sleep[i];
	}
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0D;
	while (!( UCSR0A & (1<<UDRE0))) {}
	UDR0 = 0x0A;
	_delay_ms(10);
}
void ReadPM25sensor(void){
	PORTC |= (1<<PORTC4); //enable PM2.5 sensor by puling reset pin high
	for(int i = 0; i < 10; i++){
		indexRecv = 0;
		recvBytes[0] = 0;
		recvBytes[1] = 0;
		while (recvBytes[indexRecv] != 0x42){ // detekujeme prvni znak sekvecnce co každou sekundu vysila senzor
			while (! (UCSR0A & (1<<RXC0))) {} // z teto smycky se vyskoci kdyz jsou k dispozici v bufferu nova dosud neprectena data ekvivalent k Serial.available() v Arduinu
			recvBytes[indexRecv] = UDR0;
		}
		indexRecv = 1; 
		while (recvBytes[indexRecv] != 0x4d){ // detekujeme druhy znak sekvecnce co každou sekundu vysila senzor
			while (!(UCSR0A & (1<<RXC0))) {} // z teto smycky se vyskoci kdyz jsou k dispozici v bufferu nova dosud neprectena data ekvivalent k Serial.available() v Arduinu
			recvBytes[indexRecv] = UDR0;
		}
		indexRecv = 2;
		while(indexRecv < 32) {
			while (!(UCSR0A & (1<<RXC0))) {}
			recvBytes[indexRecv] = UDR0;
			indexRecv++;
		}
		indexRecv = 0;
	}
	PORTC &= ~(1<<PORTC4);	// disable PM2.5 sensor by pulling reset pin low
}
char numberToCharHigh(char inChar){  //funkce prevede prvni 4 bity hodnoty vyjadrenou v 8-bitovem datovem formatu do tisknutelne podoby
	if ((inChar >> 4) <= 9){
		return (inChar >> 4) + 0x30; // 0x30 je ASCII kod znaku '0' nula
	}
	else if ((inChar >> 4) >= 10){
		return (inChar >> 4) + 55; //65 dec je znak A a 0xa je 10 dec, tedy 55 + 10 je znak 'A'
	}
	return '0';
}
char numberToCharLow(char inChar){  //funkce prevede prvni 4 bity hodnoty vyjadrenou v 8-bitovem datovem formatu do tisknutelne podoby
	if ((inChar & 0x0f) <= 9){
		return (inChar & 0x0f) + 0x30; // 0x30 je ASCII kod znaku '0' nula
	}
	else if ((inChar & 0x0f) >= 10){
		return (inChar & 0x0f) + 55; //65 dec je znak A a 0xa je 10 dec, tedy 55 + 10 je znak 'A'
	}
	return '0';
}
int readADCbattery(){
	uint16_t ADCvysledek;
	ADMUX = 0x46;
	ADCSRA = 0xC7;
	ADCSRB = 0x00;
	while((ADCSRA & (1 << ADIF)) == 0){}
	ADCvysledek = (int) ADCH << 8;
	ADCvysledek |= (int) ADCL;
	ADCSRA = 0x00;
	return ADCvysledek;
}
int readPhotoResistor(){
	uint16_t ADCvysledek;
	ADCSRA = (1 << ADEN)|(0x07);
	ADMUX = 0x45;
	ADCSRB = 0x00;
	ADCSRA |= (1 << ADSC);
	while(ADCSRA & (1 << ADSC)){}
	ADCvysledek = (int) ADCH << 8;
	ADCvysledek |= (int) ADCL;
	ADCSRA = 0x00;
	return ADCvysledek;
}

int main(void)
{
	if (~resetprogramu)
	{

	}
	DDRB = 0b11111111;
	DDRC = 0b11011111;
	DDRD = 0b11111110;
	DDRE = 0b11111011; 
	PORTC |= (1<<PORTC2);	//rozsvitit debug LEDku
	PORTC &= ~(1<<PORTC4);	// disable PM2.5 sensor
	volatile unsigned int buff;
	
    
    TCCR2A = 0x00; // Wave Form Generation Mode 0: Normal Mode, OC2A disconnected
    TCCR2B = 0;
    TCCR2B = (1<<CS22)|(1<<CS21)|(1<<CS20); // prescaler = 1024
    TIMSK2 = (1<<TOIE2);
	if(alreadyRegistered != 1){
		USART_init(myUBRR);			//inicializace USART modulu
		while(0 == LoRa_init()){}		//inicializace LoRa modulu
	}
    

    while (1)
    {
		if(counter==51450 || firstRun == 0) { //550 preteceni TIMER2 je neco pod 20 sekund pro Fosc = 8MHz; 28,95 overflow/sec
			PORTC |= (1 << 2);
			PORTC &= ~(1<<PORTC4);
			cli();
			waitCycles = 0;
			if(firstRun >= 1){
				while(buff != 0x0A && (waitCycles < 20*maxWaitycles)) {		//vyprazdnime si buffer kde ocekavame zpravu o probuzeni LoRa modulu
					while (!(UCSR0A & (1<<RXC0))) {}
						buff = UDR0;
						waitCycles++;
				}
			}
			if(waitCycles >= 11*maxWaitycles) {  //LoRaWAN modul se z nejakeho duvodu neprobouzi, provedeme reset
				while(0 == LoRa_init()){}		//inicializace LoRa modulu
			}
			waitCycles = 0;
			TIMSK2 &= ~(1<<TOIE2);		
			USART_init(UBRR9600);
			_delay_ms(10);
			ReadPM25sensor();
			_delay_ms(10);
			USART_init(myUBRR); //nastavit UART rozhrani na 57600 bps pro komunikaci s LoRaWAN transceiverem
			while(UCSR0A &(1 << RXC0)){ //dump RX buffer
				buff = UDR0;
			}
			while(0 == LoRa_send()) {}
			LoRa_sleep(); //send command to LoRa module to sleep
			PORTC &= ~(1<<PORTC2);	//zhasnout debug LEDku
			firstRun = 1;
			counter = 0;
			UCSR0D |= (1 << SFDE);
			UCSR0D |= (1 << RXCIE0);
			sei();
			
		}
		TIMSK2 = (1<<TOIE2); // interrupt when TCNT2 is overflowed
		TCNT2 = 0;
		set_sleep_mode(SLEEP_MODE_PWR_SAVE); // choose power down mode
		//power_timer2_disable(); // if you uncomment the MCU will sleep forever
		sleep_mode(); // sleep now!
    }
}
// TIMER2 interrupt service routine
ISR(TIMER2_OVF_vect){
    counter++;
}
// USART0 frame start interrupt service routine
ISR(USART0_START_vect){
  alreadyRegistered = 1;
}