TMSG44-CoolPi/command.c

#include "command.h"

#include <stdint.h>
#include <stdlib.h>
#include <inttypes.h>
#include <stdio.h>
#include <strings.h>
#include <string.h>
#include <sys/socket.h>
#include <unistd.h>
#include <time.h>

#include "Devices/lmx2594.h"
#include "Devices/dac8811.h"
#include "Devices/ad9912.h"

double f_pd = 200e6;

//Массив структур Command, который связывает строки команд с соответствующими функциями.
Command commands[] = {
	{"TMSG44:FREQ ", 	handleFreqCmd},
	{"TMSG44:LD?", 		handleLdCmd},
	{"TMSG44:POW ", 	handlePowCmd},
	{"TMSG44:ARM ", 	handleArmCmd},
	{"TMSG44:ATT ", 	handleAttCmd},
	{"*IDN?", 			handleIdnCmd},
	{NULL, NULL} // Завершающий элемент для обозначения конца массива
};

//handleXXXXCmd - обработчики команд
void handleFreqCmd(const char* recvBuff)
{
	printf("\nHandle command \"TMSG44:FREQ\"\n");
	double freq[1] = {0};

	splitLexeme(recvBuff, freq, sizeof(freq[0]), convertToDouble);

	printf("Frequency: %.2f Hz\n", freq[0]);
	f_pd = ad9912_set(bar1, freq[0], f_pd);
	printf("f_pd frequency is set to %.6f MHz\n", f_pd/1e6);
    lmx_freq_set(bar1, freq[0], f_pd);
    printf("The frequency is set to %.2f MHz\n", freq[0]/1e6);
}

void handleLdCmd(const char* recvBuff)
{
	int n = 0;
	char messageLd[] = "1\n";

	printf("\nHandle command \"TMSG44:LD?\"\n");

	uint32_t ld_status = lmx_ld_status(bar1);

	clock_t before = clock();
	clock_t difference;
	int difference_msec = 0;
	int trigger = 10000; //10ms

	while(!ld_status)
	{
		difference = clock() - before;
		difference_msec = difference * 1000 / CLOCKS_PER_SEC;

		if(difference_msec >= trigger)
		{
			strcpy(messageLd, "0\n");
			printf("LD timeout\n");
			break;
		}
		ld_status = lmx_ld_status(bar1);
		// printf("WHILE LD status: %d\n", ld_status);
	}

	send(conn_fd, messageLd, sizeof(messageLd), 0);
	
	printf("\nSend msg LD: %d!\n", ld_status);
}

void handlePowCmd(const char* recvBuff)
{
	printf("\nHandle command \"TMSG44:POW\"\n");
	double pow[1] = {0};

	splitLexeme(recvBuff, pow, sizeof(pow[0]), convertToDouble);
	printf("%f\n", pow[0]);
}

void handleArmCmd(const char* recvBuff)
{
	printf("\nHandle command \"TMSG44:ARM\"\n");
	uint16_t armCode[1] = {0};

	splitLexeme(recvBuff, armCode, sizeof(armCode[0]), convertToUInt16);
	printf("\n%u\n", armCode[0]);
	dac8811_set(bar1,armCode[0]);
}

void handleAttCmd(const char* recvBuff)
{
	printf("\nHandle command \"TMSG44:ATT\"\n");
	uint16_t attCode[1] = {0};

	splitLexeme(recvBuff, attCode, sizeof(attCode[0]), convertToUInt16);
	printf("\n%u\n", attCode[0]);
	dac8811_att_set(bar1,attCode[0]);
}

void handleIdnCmd(const char* recvBuff)
{
	printf("\nHandle command \"*IDN?\"\n");

	char messageIdn[] = "TMSG44_CoolPi\n";
	send(conn_fd, messageIdn, sizeof(messageIdn), 0);
}

//Проходим по массиву команд и ищем команду, которая совпадает с началом строки recvBuff. 
//Если команда найдена, вызывается соответствующая функция-обработчик
void processCommand(const char* recvBuff) 
{
	for (int i = 0; commands[i].command != NULL; i++) 
	{
		if (!strncasecmp(recvBuff, commands[i].command, strlen(commands[i].command))) 
		{
			commands[i].handler(recvBuff);
			return;
		}
	}
	printf("\nUnknown command: %s\n", recvBuff);
}

// Преобразование строки в uint16_t
void convertToUInt16(const char *str, void *output) 
{
	*(uint16_t *)output = (uint16_t)strtoul(str, NULL, 10);
}

// Преобразование строки в unsigned long long int
void convertToUint64(const char *str, void *output) 
{
	*(uint64_t *)output = (uint64_t)strtoull(str, NULL, 10);
}

// Преобразование строки в double
void convertToDouble(const char *str, void *output) 
{
	*(double *)output = strtod(str, NULL);
}

// Универсальная функция для разделения строки на лексемы
void splitLexeme(const char *ptrSCPI, void *numOutAndValue, size_t elementSize, ConvertFunc convertFunc) 
{
	uint8_t counter = 0;

	// Разделители лексем
	const char charSeparator[] = {" "};
	char *ptrLexeme = NULL;
	// Указатель для хранения контекста токенизации
	char *savePtr;

	// Инициализируем функцию
	ptrLexeme = strtok_r((char *)ptrSCPI, charSeparator, &savePtr);

	// Ищем лексемы разделенные разделителем
	ptrLexeme = strtok_r(NULL, charSeparator, &savePtr);

	// Ищем лексемы строки
	while (ptrLexeme) {
		// Проверяем, является ли первый символ лексемы числом
		if(('0' <= ptrLexeme[0]) && (ptrLexeme[0] <= '9')) {
			// Преобразуем строку с числом в число
			convertFunc(ptrLexeme, (uint8_t *)numOutAndValue + counter * elementSize);
			counter++;
		}
		// Ищем лексемы разделенные разделителем
		ptrLexeme = strtok_r(NULL, charSeparator, &savePtr);
	}
}