#include "SendCommand.h"
#include "stringex.h"

CSendCommand::CSendCommand(void)
{
}

CSendCommand::~CSendCommand(void)
{
}

bool CSendCommand::TestPacketCkSumOK(unsigned char Data[])
{
    long idxBuffer;
    long CS;
	long PktLen;
	unsigned char CHKSUM_MSB;
    //unsigned char CHKSUM_LSB;
    
	PktLen = (Data[4] * 256) + Data[5];
    CS = Data[0] + Data[1] + Data[2] + Data[3] + Data[4] + Data[5];
	if (PktLen > 0) {
		for (idxBuffer=0; idxBuffer<PktLen;idxBuffer++) {
            CS = CS + Data[idxBuffer + 6];
		}
    }
    CS = (CS ^ 0xFFFF) + 1;
    CHKSUM_MSB = (unsigned char)((CS & 0xFF00) / 256);	// calculated checksum
    //CHKSUM_LSB = (unsigned char)(CS & 0xFF);
	if ((Data[PktLen + 6] == CHKSUM_MSB) && (Data[PktLen + 7] == (CS & 0xFF))) {
		return true;
	} else {
		return false;
	}
}

//Extend Packet_Out to include entire message
//include send packet data finishing into dp5_cmd
//remove control and interface objects
//move send packet to separate function

bool CSendCommand::DP5_CMD(unsigned char Buffer[], TRANSMIT_PACKET_TYPE XmtCmd)
{
    bool bCmdFound;
    string D;
    Packet_Out POUT; 
    
    bCmdFound = true;
    POUT.LEN = 0;
	string strCfg;
	long lLen;
	//long idxData;

	switch (XmtCmd) {
        ////REQUEST_PACKETS_TO_DP5
		case XMTPT_SEND_STATUS:
            POUT.PID1 = PID1_REQ_STATUS;
            POUT.PID2 = PID2_SEND_DP4_STYLE_STATUS;   // send status only
			break;
        //case XMTPT_SEND_SPECTRUM:
			//break;
        //case XMTPT_SEND_CLEAR_SPECTRUM:
			//break;
        case XMTPT_SEND_SPECTRUM_STATUS:
            POUT.PID1 = PID1_REQ_SPECTRUM;
            POUT.PID2 = PID2_SEND_SPECTRUM_STATUS;   // send spectrum & status
			break;
        case XMTPT_SEND_CLEAR_SPECTRUM_STATUS:
            POUT.PID1 = PID1_REQ_SPECTRUM;
            POUT.PID2 = PID2_SEND_CLEAR_SPECTRUM_STATUS;   // send & clear spectrum & status
			break;
        //case XMTPT_BUFFER_SPECTRUM:
			//break;
        //case XMTPT_BUFFER_CLEAR_SPECTRUM:
			//break;
        //case XMTPT_SEND_BUFFER:
			//break;
        //case XMTPT_SEND_DP4_STYLE_STATUS:
			//break;
        //case XMTPT_SEND_CONFIG:
			//break;
        case XMTPT_SEND_SCOPE_DATA:
            POUT.PID1 = PID1_REQ_SCOPE_MISC;
            POUT.PID2 = PID2_SEND_SCOPE_DATA;
			break;
        case XMTPT_SEND_512_BYTE_MISC_DATA:
            POUT.PID1 = PID1_REQ_SCOPE_MISC;
            POUT.PID2 = PID2_SEND_512_BYTE_MISC_DATA; // request misc data
			break;
        case XMTPT_SEND_SCOPE_DATA_REARM:
            POUT.PID1 = PID1_REQ_SCOPE_MISC;
            POUT.PID2 = PID2_SEND_SCOPE_DATA_REARM;
			break;
        //case XMTPT_SEND_ETHERNET_SETTINGS:
			//break;
        case XMTPT_SEND_DIAGNOSTIC_DATA:
            POUT.PID1 = PID1_REQ_SCOPE_MISC;
            POUT.PID2 = PID2_SEND_DIAGNOSTIC_DATA;   // Request Diagnostic Packet
            POUT.LEN = 0;
			break;
        case XMTPT_SEND_NETFINDER_PACKET:
            POUT.PID1 = PID1_REQ_SCOPE_MISC;
            POUT.PID2 = PID2_SEND_NETFINDER_READBACK;   // Request NetFinder Packet
            POUT.LEN = 0;
			break;
        //case XMTPT_SEND_HARDWARE_DESCRIPTION:
			//break;
        //case XMTPT_SEND_SCA:
			//break;
        //case XMTPT_LATCH_SEND_SCA:
			//break;
        //case XMTPT_LATCH_CLEAR_SEND_SCA:
			//break;
        //case XMTPT_SEND_ROI_OR_FIXED_BLOCK:
			//break;
        //case XMTPT_PX4_STYLE_CONFIG_PACKET:
			//break;
		case XMTPT_SCA_READ_CONFIG_PACKET:
			strCfg = "";
			strCfg = "SCAW=?;";
			strCfg +="SCAI=1;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=2;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=3;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=4;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=5;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=6;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=7;SCAL=?;SCAH=?;SCAO=?;";
			strCfg +="SCAI=8;SCAL=?;SCAH=?;SCAO=?;";
			lLen = (long)strCfg.length();
			if (lLen > 0) {
				strCfg = AsciiCmdUtil.MakeUpper(strCfg);
				AsciiCmdUtil.CopyAsciiData(POUT.DATA, strCfg, lLen);
			}
            POUT.PID1 = PID1_REQ_CONFIG;
            POUT.PID2 = PID2_CONFIG_READBACK_PACKET;   // read config packet
            POUT.LEN = (unsigned short)lLen;
			break;
		 case XMTPT_ERASE_FPGA_IMAGE:
            POUT.PID1 = PID1_REQ_FPGA_UC;
            POUT.PID2 = PID2_ERASE_FPGA_IMAGE;
            POUT.LEN = 2;
            POUT.DATA[0] = 0x12;
            POUT.DATA[1] = 0x34;
			break;
        //case XMTPT_UPLOAD_PACKET_FPGA:
			//break;
        //case XMTPT_REINITIALIZE_FPGA:
			//break;
        //case XMTPT_ERASE_UC_IMAGE_0:
			//break;
        case XMTPT_ERASE_UC_IMAGE_1:
            POUT.PID1 = PID1_REQ_FPGA_UC;
            POUT.PID2 = PID2_ERASE_UC_IMAGE_1;   // erase image #1 (sector 5)
            POUT.LEN = 2;
            POUT.DATA[0] = 0x12;
            POUT.DATA[1] = 0x34;
			break;
        //case XMTPT_ERASE_UC_IMAGE_2:
			//break;
        //case XMTPT_UPLOAD_PACKET_UC:
			//break;
        //case XMTPT_SWITCH_TO_UC_IMAGE_0:
			//break;
        case XMTPT_SWITCH_TO_UC_IMAGE_1:
            POUT.PID1 = PID1_REQ_FPGA_UC;
            POUT.PID2 = PID2_SWITCH_TO_UC_IMAGE_1;   // switch to uC image #1
            POUT.LEN = 2;
            POUT.DATA[0] = 0xA5; // uC FLASH unlock keys
            POUT.DATA[1] = 0xF1;
			break;
        //case XMTPT_SWITCH_TO_UC_IMAGE_2:
			//break;
        //case XMTPT_UC_FPGA_CHECKSUMS:
			//break;
        ////VENDOR_REQUESTS_TO_DP5
        //case XMTPT_CLEAR_SPECTRUM_BUFFER_A:
			//break;
        case XMTPT_ENABLE_MCA_MCS:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_ENABLE_MCA_MCS; 
            POUT.LEN = 0;
			break;
        case XMTPT_DISABLE_MCA_MCS:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_DISABLE_MCA_MCS; 
            POUT.LEN = 0;
			break;
        case XMTPT_ARM_DIGITAL_OSCILLOSCOPE:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_ARM_DIGITAL_OSCILLOSCOPE;   // arm trigger
			break;
        //case XMTPT_AUTOSET_INPUT_OFFSET:
			//break;
        case XMTPT_AUTOSET_FAST_THRESHOLD:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_AUTOSET_FAST_THRESHOLD; 
            POUT.LEN = 0;
			break;
        //case XMTPT_READ_IO3_0:
			//break;
        //case XMTPT_WRITE_IO3_0:
			//break;
        //case XMTPT_SET_DCAL:
			//break;
        //case XMTPT_SET_PZ_CORRECTION_UC_TEMP_CAL:
			//break;
        //case XMTPT_SET_PZ_CORRECTION_UC_TEMP_CAL:
			//break;
        //case XMTPT_SET_BOOT_FLAGS:
			//break;
        //case XMTPT_SET_HV_DP4_EMULATION:
			//break;
        //case XMTPT_SET_TEC_DP4_EMULATION:
			//break;
        //case XMTPT_SET_INPUT_OFFSET_DP4_EMULATION:
			//break;
        //case XMTPT_SET_ADC_CAL_GAIN_OFFSET:
			//break;
        //case XMTPT_SET_SPECTRUM_OFFSET:
			//break;
        //case XMTPT_REQ_SCOPE_DATA_MISC_DATA_SCA_PACKETS:
			//break;
        //case XMTPT_SET_SERIAL_NUMBER:
			//break;
        //case XMTPT_CLEAR_GP_COUNTER:
			//break;
        //case XMTPT_SWITCH_SUPPLIES:
			//break;
        //case XMTPT_SEND_TEST_PACKET:
			//break;
        case XMTPT_REQ_ACK_PACKET:
            POUT.PID1 = PID1_COMM_TEST;
            POUT.PID2 = PID2_ACK_OK; 
            POUT.LEN = 0;
			break;
		case XMTPT_FORCE_SCOPE_TRIGGER:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_GENERIC_FPGA_WRITE;	// generic FPGA write
            POUT.LEN = 2;
            POUT.DATA[0] = 119;						// force scope trigger
            POUT.DATA[1] = 0;						// data doesn't matter
			break;
		case XMTPT_AU34_2_RESTART:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_GENERIC_FPGA_WRITE;	// generic FPGA write
            POUT.LEN = 2;
            POUT.DATA[0] = 0x73;					// AU34_2_RESTART
            POUT.DATA[1] = 0;						// data doesn't matter
			break;
        case XMTPT_READ_MCA8000D_OPTION_PA_CAL:
            POUT.PID1 = PID1_REQ_SCOPE_MISC;
            POUT.PID2 = PID2_SEND_OPTION_PA_CALIBRATION; 
            POUT.LEN = 0;
			break;
		default:
            bCmdFound = false;
			break;
	}
	if (bCmdFound) {
        if (! POUT_Buffer(POUT, Buffer)) {
            bCmdFound = false;
        }
    }
	return bCmdFound;
}

//DP5_CMD_Config is for: (requires pc5 and device info)
//		XMTPT_SEND_CONFIG_PACKET_TO_HW
//		XMTPT_SEND_CONFIG_PACKET_EX
//		XMTPT_FULL_READ_CONFIG_PACKET
//		XMTPT_READ_CONFIG_PACKET
//		XMTPT_READ_CONFIG_PACKET_EX
bool CSendCommand::DP5_CMD_Config(unsigned char Buffer[], TRANSMIT_PACKET_TYPE XmtCmd, CONFIG_OPTIONS CfgOptions)
{
    bool bCmdFound;
    string D;
    Packet_Out POUT; 
    bCmdFound = true;
    POUT.LEN = 0;
	string strCfg;
	long lLen;

	switch (XmtCmd) {
        case XMTPT_SEND_CONFIG_PACKET_TO_HW:
			// CONFIG_OPTIONS Needed:
			//		CfgOptions.HwCfgDP5Out
			//		CfgOptions.SendCoarseFineGain
			//		CfgOptions.PC5_PRESENT
			//		CfgOptions.DppType
			strCfg = "";
			strCfg = CfgOptions.HwCfgDP5Out;

			if (CfgOptions.SendCoarseFineGain) {
				strCfg = AsciiCmdUtil.RemoveCmd("GAIN",strCfg);
			} else {
				strCfg = AsciiCmdUtil.RemoveCmd("GAIA",strCfg);
				strCfg = AsciiCmdUtil.RemoveCmd("GAIF",strCfg);
			}
			strCfg = AsciiCmdUtil.RemoveCmdByDeviceType(strCfg, CfgOptions.PC5_PRESENT, CfgOptions.DppType, CfgOptions.isDP5_RevDxGains, CfgOptions.DPP_ECO);
			lLen = (long)strCfg.length();
			if (lLen > 0) {
				strCfg = AsciiCmdUtil.MakeUpper(strCfg);
				AsciiCmdUtil.CopyAsciiData(POUT.DATA, strCfg, lLen);
			}
            POUT.PID1 = PID1_REQ_CONFIG;
            POUT.PID2 = PID2_TEXT_CONFIG_PACKET;   // text config packet
            POUT.LEN = (unsigned short)lLen;
			break;
        case XMTPT_SEND_CONFIG_PACKET_EX:			// bypass any filters
			// CONFIG_OPTIONS Needed:
			//		CfgOptions.HwCfgDP5Out
			strCfg = "";
			strCfg = CfgOptions.HwCfgDP5Out;
			lLen = (long)strCfg.length();
			if (lLen > 0) {
				strCfg = AsciiCmdUtil.MakeUpper(strCfg);
				AsciiCmdUtil.CopyAsciiData(POUT.DATA, strCfg, lLen);
			}
            POUT.PID1 = PID1_REQ_CONFIG;
            POUT.PID2 = PID2_TEXT_CONFIG_PACKET;   // text config packet
            POUT.LEN = (unsigned short)lLen;
 			break;
         case XMTPT_FULL_READ_CONFIG_PACKET:
			// CONFIG_OPTIONS Needed:
			//		CfgOptions.PC5_PRESENT
			//		CfgOptions.DppType
			strCfg = "";
			strCfg = AsciiCmdUtil.CreateFullReadBackCmd(CfgOptions.PC5_PRESENT, CfgOptions.DppType, CfgOptions.isDP5_RevDxGains, CfgOptions.DPP_ECO);
			lLen = (long)strCfg.length();
			if (lLen > 0) {
				strCfg = AsciiCmdUtil.MakeUpper(strCfg);
				AsciiCmdUtil.CopyAsciiData(POUT.DATA, strCfg, lLen);
			}
            POUT.PID1 = PID1_REQ_CONFIG;
            POUT.PID2 = PID2_CONFIG_READBACK_PACKET;   // read config packet
            POUT.LEN = (unsigned short)lLen;
			break;
        case XMTPT_READ_CONFIG_PACKET_EX:					// bypass any filters
			strCfg = "";
			strCfg = CfgOptions.HwCfgDP5Out;
			lLen = (long)strCfg.length();
			if (lLen > 0) {
				strCfg = AsciiCmdUtil.MakeUpper(strCfg);
				AsciiCmdUtil.CopyAsciiData(POUT.DATA, strCfg, lLen);	//ForceASCII_20110606
			}
            POUT.PID1 = PID1_REQ_CONFIG;
            POUT.PID2 = PID2_CONFIG_READBACK_PACKET;   // text config packet
            POUT.LEN = (unsigned short)lLen;
			break;
		case XMTPT_READ_CONFIG_PACKET:
			strCfg = "";
			strCfg += "CLCK=?;"; // FPGA clock
			strCfg += "TPEA=?;"; // peak time
			strCfg += "GAIN=?;"; // gain
			if (CfgOptions.DppType == dppMCA8000D) {
				strCfg += "GAIA=?;"; // coarse gain
			}
			strCfg += "MCAS=?;"; // mca mode
			strCfg += "MCAC=?;"; // channels
			strCfg += "INOF=?;"; // osc. Input offset
			strCfg += "THSL=?;"; // LLD thresh
			strCfg += "TFLA=?;"; // flat top width
			strCfg += "THFA=?;"; // fast thresh
			strCfg += "DACO=?;"; // osc. DAC output
			strCfg += "DACF=?;"; // osc. DAC offset
			strCfg += "AUO1=?;"; // osc. AUX_OUT1
			strCfg += "PRET=?;"; // preset actual time
			strCfg += "PRER=?;"; // preset real time
			strCfg += "PREC=?;"; // preset count
			if (CfgOptions.DppType == dppMCA8000D) {
				strCfg += "PREL=?;"; // preset real time
			}
			strCfg += "HVSE=?;"; // high voltage setting for manufacuting test
			strCfg += "SCOE=?;"; // osc. Scope trigger edge
			strCfg += "SCOT=?;"; // osc. Scope trigger position
			strCfg += "SCOG=?;"; // osc. Scope gain

			lLen = (long)strCfg.length();
			if (lLen > 0) {
				strCfg = AsciiCmdUtil.MakeUpper(strCfg);
				AsciiCmdUtil.CopyAsciiData(POUT.DATA, strCfg, lLen);
			}
            POUT.PID1 = PID1_REQ_CONFIG;
            POUT.PID2 = PID2_CONFIG_READBACK_PACKET;   // read config packet
            POUT.LEN = (unsigned short)lLen;
			break;
		default:
            bCmdFound = false;
			break;
	}
	if (bCmdFound) {
        if (! POUT_Buffer(POUT, Buffer)) {
            bCmdFound = false;
        }
    }
	return bCmdFound;
}


//Extend Packet_Out to include entire message
//include send packet data finishing into dp5_cmd
//remove control and interface objects
//move send packet to separate function
bool CSendCommand::DP5_CMD_Data(unsigned char Buffer[], TRANSMIT_PACKET_TYPE XmtCmd, unsigned char DataOut[])
{
    bool bCmdFound;
    short idxMiscData;
    Packet_Out POUT; 
    long PktLen;
    
    bCmdFound = false;
    POUT.LEN = 0;
	string strCfg;
	long idxData;
	switch (XmtCmd) {	//REQUEST_PACKETS_TO_DP5
        case XMTPT_WRITE_512_BYTE_MISC_DATA:
            POUT.PID1 = PID1_VENDOR_REQ;
            POUT.PID2 = PID2_WRITE_512_BYTE_MISC_DATA;				// write misc data
            POUT.LEN = 512;
			for (idxMiscData=0;idxMiscData<=511;idxMiscData++) {	// byte array padded w/NULLs
                POUT.DATA[idxMiscData] = DataOut[idxMiscData];
			}
			bCmdFound = true;
			if (! POUT_Buffer(POUT, Buffer)) {
				bCmdFound = false;
			}
			break;
         case XMTPT_SEND_TEST_PACKET:
			PktLen = (DataOut[4] * 256) + DataOut[5] + 8;		// get entire packet size
			if ((PktLen >= 8) && (PktLen <= 12)) {				// test data len 0-4 bytes
				if (TestPacketCkSumOK(DataOut)) {				// check the message for correct check sum
					for(idxData=0;idxData<PktLen;idxData++) {	// load the data into the command buffer
						Buffer[idxData] = DataOut[idxData];
					}
					bCmdFound = true;
				}
			}
			break;
		default:
            bCmdFound = false;
			break;
	}
  	return bCmdFound;
}

bool CSendCommand::POUT_Buffer(Packet_Out POUT, unsigned char Buffer[])
{
    long idxBuffer;
    long CS;
    
    Buffer[0] = SYNC1_;
    Buffer[1] = SYNC2_;
    Buffer[2] = POUT.PID1;
    Buffer[3] = POUT.PID2;
    Buffer[4] = (POUT.LEN & 0xFF00) / 256;
    Buffer[5] = POUT.LEN & 0xFF;

    CS = SYNC1_ + SYNC2_ + POUT.PID1 + POUT.PID2 + ((POUT.LEN & 0xFF00) / 256) + (POUT.LEN & 0xFF);

	if (POUT.LEN > 0) {
		for (idxBuffer=0; idxBuffer<POUT.LEN;idxBuffer++) {
            Buffer[idxBuffer + 6] = POUT.DATA[idxBuffer];
            CS = CS + POUT.DATA[idxBuffer];
		}
    }
    CS = (CS ^ 0xFFFF) + 1;
    Buffer[POUT.LEN + 6] = (unsigned char)((CS & 0xFF00) / 256);
    Buffer[POUT.LEN + 7] = (unsigned char)(CS & 0xFF);
    return true;
}