#include "stdio.h"
#include "simpletools.h"
#include <smartpins.h>
#include <stdint.h>
#include <stdlib.h>
#include <math.h>
#include <sys/time.h>

struct __using("jm_fullduplexserial.spin2") jm;

enum {
    _xtlfreq = 20_000_000,
    _clkfreq = 256_000_000,
    DOWNLOAD_BAUD = 230_400,
    DEBUG_BAUD = DOWNLOAD_BAUD,
    PINS = 20|7<<6,
    GROUP = 2,
    CLK_PIN = 6,
    TX_REGD = 1,
    CLK_REGD = 0,
    CPOL = 1,
    CPHA = 0,
    //CLK_DIV = 2,
    CLK_DIV = 8,
    M_NCO = 0x8000_0000UL / CLK_DIV + (0x8000_0000UL % CLK_DIV > 0UL ? 1UL : 0UL),    // round up
    //M_LEADIN = X_IMM_32X1_1DAC1 | 7 + CLK_DIV + CLK_REGD - TX_REGD - (1 ^ CPHA) * (CLK_DIV>>1),
    M_LEADIN = X_IMM_32X1_1DAC1 | 5 + CLK_DIV + CLK_REGD - TX_REGD - (1 ^ CPHA) * (CLK_DIV>>1),
};

#define buff_size 32768

static long stack1[32800];
static long stack2[32800];


char arr[buff_size] = {0};

int cogid1;
int cogid2;

int temp = 0;

unsigned int data_count = 0;
int chunk_cnt = 0;
int i = 0;

FILE * fp;

void print_data(){
	printf("Started new cog print_data()\n");
	mount("/sd", _vfs_open_sdcard());
	FILE *fp = fopen("/sd/data.txt", "a");
	FILE *fh ;
	if (fp == NULL) {
		printf("File failed to open in printdata..\n");
	} // checking file is present or not 
	else {
		printf("The file is now opened in printdata.\n");
	}
	
	
	while (true){
		while(_pollatn()==0);		
	    printf("Received attenuation from UART : Datasize = %d\n",sizeof(arr));
		int data_cnt = fwrite(arr, sizeof(char), sizeof(arr), fp);
		while(data_cnt<0)
		{
		   
		   data_cnt = fwrite(arr, sizeof(char), sizeof(arr), fh);
		}
		printf("DATA_CNT write= %d\n", data_cnt);
		
		printf("Chunk written\n");
		
		printf("Sending attenuation to PRINT second time: \n");
		_cogatn((1<<cogid1)); // Send ATN to cog1 for next chunk.
	}
	fclose(fp);
	printf("The file is now closed.\n");
}

void uart_func(){
	printf("Started new cog uart_func()\n");
	
	int Total_data = 0;
	jm.start(14, 15, 0, 2000000); //start(rxpin, txpin, mode, baud)
	
	while (true) {
		if (jm.rxcheck() == 255)
			break;
	}
	jm.tx(255);
	// Reciveing first chunk without getting attenuation from second cog.
	while (true){
		jm.tx(250);
		while (true){
			int rec = jm.rxcheck();
			if (rec == 254) {
				//One chunk received.
				printf("Received data = %d\n",i);
				chunk_cnt++;
				break;
			}
			if (rec != -1) {
				arr[i++] = rec;
			}
		}
		data_count += i;
		printf(" i : inside first while %d\n", i ); 
		i = 0;
		//_waitms(100);
		printf("Sending attenuation to PRINT: \n");
		_cogatn((1<<cogid2));
		break;
	}
	
	//_waitms(1000);
	// After one chunk recived, second cog will get attenuation. Then after: every time this cog receives attenuation, it start receiving data from R-pi.
	while(true){
	    printf("Wating for attenuation...\n");
		while(_pollatn()==0);
		memset(arr,0,sizeof(arr)); 		
		printf("Received attenuation from PRINT :\n");
		i = 0;
		jm.tx(250); //Receving next chunk
		
		while (true) {
			int rec = jm.rxcheck();
			if (rec == 254) {
			    chunk_cnt++;
           		printf("chunk_cnt: %d\n",chunk_cnt);
				break;
			}
			if (rec == 253 ){
				break; // Whole data received.
			}
			if (rec != -1) {
				arr[i++] = rec;
				data_count++;
			}
		}
		printf(" i : inside second while %d\n", i );
		printf("~~~~~~~~~~~~~~data_count = %d  ~~~~~~~~~~~~~~~~~\n\n\n",data_count);
				
		printf("Sending attenuation to PRINT second time : \n");
		//_waitms(100);
		_cogatn((1<<cogid2));
		_waitus(10);
	}
}
int main(){
	high(13);
	printf("Main start!\n");
	cogid1 = __builtin_cogstart(uart_func(), stack1);
	cogid2 = __builtin_cogstart(print_data(), stack2);
	while(true){
		_waitms(25);
	}
}