#include "task_modbus.h"
#include "../msp/uart0.h"
#include "../tpc/modbus.h"
#include "../app/app_config.h"

U8 L3_pack_modbus_03(TS_PH4_modbus *pmodbus ,U8 slaverId, U16 reg, U16 num)
{
	U8 bufsize = 4;
	pmodbus->slaver = slaverId;
	pmodbus->oper = MODBUS_OPER_READ;
	pmodbus->buf[0] = reg >> 8 & 0xFF;
	pmodbus->buf[1] = reg >> 0 & 0xFF;
	pmodbus->buf[2] = num >> 8 * 0xFF;
	pmodbus->buf[3] = num >> 0 & 0xFF;
	crc16(pmodbus->crc,&pmodbus->slaver,2 + bufsize);
	pmodbus->buf[bufsize] = pmodbus->crc[0];
	pmodbus->buf[bufsize + 1] = pmodbus->crc[1];

	return bufsize + 2 + 2;
}

U8 L3_pack_modbus_06(TS_PH4_modbus *pmodbus ,U8 slaverId, U16 reg, U8 *buf)
{
	U8 bufsize = 4;
	pmodbus->slaver = slaverId;
	pmodbus->oper = MODBUS_OPER_WRITE;
	pmodbus->buf[0] = reg >> 8 & 0xFF;
	pmodbus->buf[1] = reg >> 0 & 0xFF;
	pmodbus->buf[2] = buf[0];
	pmodbus->buf[3] = buf[1];
	crc16(pmodbus->crc,&pmodbus->slaver,2 + bufsize);
	pmodbus->buf[bufsize] = pmodbus->crc[0];
	pmodbus->buf[bufsize + 1] = pmodbus->crc[1];

	return bufsize + 2 + 2;
}

U8 L3_pack_modbus_10(TS_PH4_modbus *pmodbus ,U8 slaverId, U16 reg, U16 num, U8 bytes, U8 *buf)
{
	U8 bufsize = 5,i = 0;
	pmodbus->slaver = slaverId;
	pmodbus->oper = MODBUS_OPER_WRITE_M;
	pmodbus->buf[0] = reg >> 8 & 0xFF;
	pmodbus->buf[1] = reg >> 0 & 0xFF;
	pmodbus->buf[2] = num >> 8 * 0xFF;
	pmodbus->buf[3] = num >> 0 & 0xFF;
	pmodbus->buf[4] = bytes;

	for(i=0;i<bytes;i++)
	{
		pmodbus->buf[5+i] = buf[i];
	}
	crc16(pmodbus->crc,&pmodbus->slaver,2 + bufsize + bytes);
	pmodbus->buf[bufsize + bytes] = pmodbus->crc[0];
	pmodbus->buf[bufsize + bytes + 1] = pmodbus->crc[1];

	return bufsize + bytes + 2 + 2;
}

U8 L3_pack_modbus(TS_PH4_modbus *pmodbus, MD_SLAVER_INFO *slaver_info)
{
	switch(slaver_info->oper)
	{
	case MODBUS_OPER_READ:
		return L3_pack_modbus_03(pmodbus,slaver_info->slaver,slaver_info->reg,slaver_info->regnum);
	case MODBUS_OPER_WRITE:
		return L3_pack_modbus_06(pmodbus,slaver_info->slaver,slaver_info->reg,slaver_info->buf);
	case MODBUS_OPER_WRITE_M:
		return L3_pack_modbus_10(pmodbus,slaver_info->slaver,slaver_info->reg,slaver_info->regnum,slaver_info->bytes,slaver_info->buf);
	default:
		return 0;
	}
}

U8 L3_pack_modbusack(TS_PH4_modbus *pmodbus ,U8 slaver, U8 oper, U8 bufsize)
{
	pmodbus->slaver = slaver;
	pmodbus->oper = oper;
	crc16(pmodbus->crc,&pmodbus->slaver,2 + bufsize);
	pmodbus->buf[bufsize] = pmodbus->crc[0];
	pmodbus->buf[bufsize + 1] = pmodbus->crc[1];

	return bufsize + 2 + 2;
}

//MODBUS从设备数据处理函数
static Modbus03Ack *pModbus03Ack;
static Modbus06Ack *pModbus06Ack;
static Modbus10Ack *pModbus10Ack;
static U16 modbuslen = 0,reg,num,val,count;
U16 L3_modbus_slaver_ack(TS_PH4_modbus *pmodbus,TS_PH4_modbus *pModbusAck)
{
	switch(pmodbus->oper)
	{
	case MODBUS_OPER_READ:
	case MODBUS_OPER_READCONFIG:
	{
		U16 i = 0,j = 0,regval = 0;
		pModbus03Ack = (Modbus03Ack *)pModbusAck->buf;
		reg = ((U16)pmodbus->buf[0]) << 8 | ((U16)pmodbus->buf[1]);
		num = ((U16)pmodbus->buf[2]) << 8 | ((U16)pmodbus->buf[3]);
		if((num <= D_MODBUS_REG_MAX_NUM) && (reg <= sizeof(R)/2))	//每次最多读取D_MODBUS_REG_MAX_NUM个寄存器.防止越界
		{
			for(i=0; i<num; i++)
			{
				regval = *(REG_2_MEM(reg + i));
				pModbus03Ack->buf[j++] = regval >> 8 & 0xFF;
				pModbus03Ack->buf[j++] = regval & 0xFF;
			}
			pModbus03Ack->bytes = num * 2;
			modbuslen = L3_pack_modbusack(pModbusAck, pmodbus->slaver, pmodbus->oper, pModbus03Ack->bytes + 1);
		}
		break;
	}
#if 0
	case MODBUS_OPER_WRITE:
	{	
		pModbus06Ack = (Modbus06Ack *)pModbusAck->buf;
		reg = ((U16)pmodbus->buf[0]) << 8 | ((U16)pmodbus->buf[1]);
		val = ((U16)pmodbus->buf[2]) << 8 | ((U16)pmodbus->buf[3]);
		if(reg <= sizeof(R)/2)
		{
			*(REG_2_MEM(reg)) = val;
			pModbus06Ack->reg = reg;
			pModbus06Ack->val = *(REG_2_MEM(reg));
			modbuslen = L3_pack_modbusack(pModbusAck, pmodbus->slaver, pmodbus->oper, 4);
		}
		break;
	}
#endif
	case MODBUS_OPER_WRITE_M:
	{
		U16 i = 0;
		pModbus10Ack = (Modbus10Ack *)pModbusAck->buf;
		reg = ((U16)pmodbus->buf[0]) << 8 | ((U16)pmodbus->buf[1]);
		num = ((U16)pmodbus->buf[2]) << 8 | ((U16)pmodbus->buf[3]);
		count = pmodbus->buf[4];
		if((num <= D_MODBUS_REG_MAX_NUM) && (reg < sizeof(R)/2))	//每次最多写入20个寄存器 40个字节，以防止越界
		{
			for(i=0; i<num; i++)
			{
				*(REG_2_MEM(reg + i)) = ((U16)pmodbus->buf[5+i*2]) << 8 | ((U16)pmodbus->buf[5+i*2+1]);
			}
			pModbus10Ack->reg = reg;
			pModbus10Ack->num = num;
			modbuslen = L3_pack_modbusack(pModbusAck, pmodbus->slaver, pmodbus->oper, 4);
		}
		break;
	}
	default:break;
	}
	return modbuslen;
}

//MODBUS主设备轮询时，从设备返回数据处理函数
//MD_SLAVER_INFO需要被正确定义
void L3_modbus_master_handler(TS_PH4_modbus *pmodbus,MD_SLAVER_INFO *p_slaver_info)
{
	switch(pmodbus->oper)
	{
		case MODBUS_OPER_READ:
		case MODBUS_OPER_READCONFIG:
		{
			U8 i = 0;
			for(i=0;i<pmodbus->buf[0];i++)
			{
				//此处假设字节序一致
				p_slaver_info->buf[i] = pmodbus->buf[i + 1];
				//TODO 字节序不一致时的处理
			}
			break;
		}
		case MODBUS_OPER_WRITE:
		case MODBUS_OPER_WRITE_M:
		default: break;
	}
}


void L3_task_modbus_handler(TP_Handler_X *ph4)
{
	if(ph4->ok)
	{	
		TS_PH4_modbus *ts_modbus = (TS_PH4_modbus *)ph4->buf;
		//L0_uart0_sendArray(ph4->buf, 8);
		//L0_uart0_sendArray("abcdefgh", 8);
		ph4->ok = 0;
		if(ts_modbus->slaver == R.p.slaver_id || ts_modbus->slaver == D_UART0_485_SLAVER_ID_BROADCAST)	//从机
		{
			U16 acklen = L3_modbus_slaver_ack(ts_modbus,&s_uart0_ack);
			L0_uart0_sendArray((U8*)&s_uart0_ack, acklen);
		}
	}
}