mian.cpp

/**	@file       mian.cpp
 *	@note       Hikvision Digital Technology Co., Ltd. All Right Reserved.
 *	@brief
 *
 *	@author     lipengfei
 *	@date       2018/05/10
 *	@note       历史记录：
 *	@note       V1.0.0
 *	@warning
 */

#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <time.h>
#include "OSSocket.h"
#include "JsonParse.h"
#include "CmdParse.h"
#include "UAVAI.h"
#include "Point3.h"

const int MAX_SOCKET_BUFFER = (1024 * 1024 * 4);       /// 发送接受数据最大4M
using std::cout;
using std::endl;

/** @fn     int RecvJuderData(OS_SOCKET hSocket, char *pBuffer)
 *  @brief	接受数据
 *	@param  -I   - OS_SOCKET hSocket
 *	@param  -I   - char * pBuffer
 *	@return int
 */
int RecvJuderData(OS_SOCKET hSocket, char *pBuffer)
{
	int         nRecvLen = 0;
	int         nLen = 0;

	while (1)
	{
		// 接受头部长度
		nLen = OSRecv(hSocket, pBuffer + nRecvLen, MAX_SOCKET_BUFFER);
		if (nLen <= 0)
		{
			printf("recv error\n");
			return nLen;
		}

		nRecvLen += nLen;

		if (nRecvLen >= SOCKET_HEAD_LEN)
		{
			break;
		}
	}

	int         nJsonLen = 0;
	char        szLen[10] = { 0 };

	memcpy(szLen, pBuffer, SOCKET_HEAD_LEN);

	nJsonLen = atoi(szLen);

	while (nRecvLen < (SOCKET_HEAD_LEN + nJsonLen))
	{
		// 说明数据还没接受完
		nLen = OSRecv(hSocket, pBuffer + nRecvLen, MAX_SOCKET_BUFFER);
		if (nLen <= 0)
		{
			printf("recv error\n");
			return nLen;
		}

		nRecvLen += nLen;
	}

	return 0;
}

/** @fn     int SendJuderData(OS_SOCKET hSocket, char *pBuffer, int nLen)
 *  @brief	发送数据
 *	@param  -I   - OS_SOCKET hSocket
 *	@param  -I   - char * pBuffer
 *	@param  -I   - int nLen
 *	@return int
 */
int SendJuderData(OS_SOCKET hSocket, char *pBuffer, int nLen)
{
	int     nSendLen = 0;
	int     nLenTmp = 0;

	while (nSendLen < nLen)
	{
		nLenTmp = OSSend(hSocket, pBuffer + nSendLen, nLen - nSendLen);
		if (nLenTmp < 0)
		{
			return -1;
		}

		nSendLen += nLenTmp;
	}

	return 0;
}

/** @fn     void AlgorithmCalculationFun()
 *  @brief	学生的算法计算， 参数什么的都自己写，
 *	@return void
 */
void  AlgorithmCalculationFun(UAVAI *pstAI)
{
	pstAI->getNextAction();
}

int main(int argc, char *argv[])
{
	UAVAI *pstAI = new UAVAI;
#ifdef OS_WINDOWS
	// windows下，需要进行初始化操作
	WSADATA wsa;
	if (WSAStartup(MAKEWORD(2, 2), &wsa) != 0)
	{
		printf("WSAStartup failed\n");
		return false;
	}

#endif

	char        szIp[64] = { 0 };
	int         nPort = 0;
	char        szToken[128] = { 0 };
	int         nRet = 0;
	OS_SOCKET   hSocket;
	char        *pRecvBuffer = NULL;
	char        *pSendBuffer = NULL;
	int         nLen = 0;

	// 本地调试去掉这个
	if (argc != 4)
	{
		printf("error arg num\n");
		return -1;
	}

	// 解析参数
	strcpy(szIp, argv[1]);
	nPort = atoi(argv[2]);
	strcpy(szToken, argv[3]);

	//strcpy(szIp, "39.106.111.130");
	//nPort = 4010;
	//strcpy(szToken, "36d0a20b-7fab-4a93-b7e4-3247533b903a");

	printf("server ip %s, prot %d, token %s\n", szIp, nPort, szToken);

	// test

	// 开始连接服务器
	nRet = OSCreateSocket(szIp, (unsigned int)nPort, &hSocket);
	if (nRet != 0)
	{
		printf("connect server error\n");
		return nRet;
	}

	// 分配接受发送内存
	pRecvBuffer = (char*)malloc(MAX_SOCKET_BUFFER);
	if (pRecvBuffer == NULL)
	{
		return -1;
	}

	pSendBuffer = (char*)malloc(MAX_SOCKET_BUFFER);
	if (pSendBuffer == NULL)
	{
		free(pRecvBuffer);
		return -1;
	}

	memset(pRecvBuffer, 0, MAX_SOCKET_BUFFER);

	// 接受数据  连接成功后，Judger会返回一条消息：
	nRet = RecvJuderData(hSocket, pRecvBuffer);
	if (nRet != 0)
	{
		return nRet;
	}
	cout << pRecvBuffer << endl;
	// json 解析
	// 获取头部
	CONNECT_NOTICE  stNotice;

	nRet = ParserConnect(pRecvBuffer + SOCKET_HEAD_LEN, &stNotice);
	if (nRet != 0)
	{
		return nRet;
	}

	// 生成表明身份的json
	TOKEN_INFO  stToken;

	strcpy(stToken.szToken, szToken);  // 如果是调试阶段，请输入你调试的token，在我的对战中获取，
									   // 实际比赛，不要传入调试的，按照demo写的，有服务器调用传入。
	strcpy(stToken.szAction, "sendtoken");

	memset(pSendBuffer, 0, MAX_SOCKET_BUFFER);
	nRet = CreateTokenInfo(&stToken, pSendBuffer, &nLen);
	if (nRet != 0)
	{
		return nRet;
	}
	// 选手向裁判服务器表明身份(Player -> Judger)
	nRet = SendJuderData(hSocket, pSendBuffer, nLen);
	if (nRet != 0)
	{
		return nRet;
	}
	cout << pSendBuffer << endl;

	//身份验证结果(Judger -> Player)　
	memset(pRecvBuffer, 0, MAX_SOCKET_BUFFER);
	nRet = RecvJuderData(hSocket, pRecvBuffer);
	if (nRet != 0)
	{
		return nRet;
	}

	// 解析验证结果的json
	TOKEN_RESULT      stResult;
	nRet = ParserTokenResult(pRecvBuffer + SOCKET_HEAD_LEN, &stResult);
	if (nRet != 0)
	{
		return 0;
	}
	cout << pRecvBuffer << endl;

	// 是否验证成功
	if (stResult.nResult != 0)
	{
		printf("token check error\n");
		return -1;
	}
	// 选手向裁判服务器表明自己已准备就绪(Player -> Judger)
	READY_PARAM     stReady;

	strcpy(stReady.szToken, szToken);
	strcpy(stReady.szAction, "ready");

	memset(pSendBuffer, 0, MAX_SOCKET_BUFFER);
	nRet = CreateReadyParam(&stReady, pSendBuffer, &nLen);
	if (nRet != 0)
	{
		return nRet;
	}
	nRet = SendJuderData(hSocket, pSendBuffer, nLen);
	if (nRet != 0)
	{
		return nRet;
	}
	cout << pSendBuffer << endl;

	//对战开始通知(Judger -> Player)　
	memset(pRecvBuffer, 0, MAX_SOCKET_BUFFER);
	nRet = RecvJuderData(hSocket, pRecvBuffer);
	if (nRet != 0)
	{
		return nRet;
	}
	cout << pRecvBuffer << endl;

	// 解析数据
	//Mapinfo 结构体可能很大，不太适合放在栈中，可以定义为全局或者内存分配
	MAP_INFO            *pstMapInfo;
	MATCH_STATUS        *pstMatchStatus;
	FLY_PLANE          *pstFlyPlane;

	pstMapInfo = (MAP_INFO *)malloc(sizeof(MAP_INFO));
	if (pstMapInfo == NULL)
	{
		return -1;
	}

	pstMatchStatus = (MATCH_STATUS *)malloc(sizeof(MATCH_STATUS));
	if (pstMapInfo == NULL)
	{
		return -1;
	}

	pstFlyPlane = (FLY_PLANE *)malloc(sizeof(FLY_PLANE));
	if (pstFlyPlane == NULL)
	{
		return -1;
	}

	memset(pstMapInfo, 0, sizeof(MAP_INFO));
	memset(pstMatchStatus, 0, sizeof(MATCH_STATUS));
	memset(pstFlyPlane, 0, sizeof(FLY_PLANE));

	nRet = ParserMapInfo(pRecvBuffer + SOCKET_HEAD_LEN, pstMapInfo);
	if (nRet != 0)
	{
		return nRet;
	}

	// 第一次把无人机的初始赋值给flayplane
	pstFlyPlane->nPurchaseNum = 0;
	pstFlyPlane->nUavNum = pstMapInfo->nUavNum;
	pstMatchStatus->nUavWeNum = pstMapInfo->nUavNum;
	for (int i = 0; i < pstMapInfo->nUavNum; i++)
	{
		pstMatchStatus->astWeUav[i] = pstMapInfo->astUav[i];
		pstFlyPlane->astUav[i] = pstMapInfo->astUav[i];
	}

	pstAI->initPtr(pstMapInfo, pstMatchStatus, pstFlyPlane);
	pstAI->initMap();
	pstAI->setInitUavTarget();

	clock_t clockBegin, clockEnd;
	clock_t timeCount = 0;
	// 根据服务器指令，不停的接受发送数据
	while (1)
	{
		cout << "Current step: " << pstMatchStatus->nTime << endl;
		// 进行当前时刻的数据计算, 填充飞行计划结构体，注意：0时刻不能进行移动，即第一次进入该循环时
		if (pstMatchStatus->nTime != 0)
		{
			clockBegin = clock();
			AlgorithmCalculationFun(pstAI);
			clockEnd = clock();
			cout << "Cost Time: " << clockEnd - clockBegin << endl;
			timeCount += clockEnd - clockBegin;
		}

		//发送飞行计划结构体
		memset(pSendBuffer, 0, MAX_SOCKET_BUFFER);
		nRet = CreateFlayPlane(pstFlyPlane, szToken, pSendBuffer, &nLen);
		if (nRet != 0)
		{
			return nRet;
		}
		nRet = SendJuderData(hSocket, pSendBuffer, nLen);
		if (nRet != 0)
		{
			return nRet;
		}

		//printf("%s\n", pSendBuffer);

		// 接受当前比赛状态
		memset(pRecvBuffer, 0, MAX_SOCKET_BUFFER);
		nRet = RecvJuderData(hSocket, pRecvBuffer);
		if (nRet != 0)
		{
			return nRet;
		}

		// 解析
		nRet = ParserMatchStatus(pRecvBuffer + SOCKET_HEAD_LEN, pstMatchStatus);
		if (nRet != 0)
		{
			return nRet;
		}

		if (pstMatchStatus->nMacthStatus == 1)
		{
			// 比赛结束
			printf("game over, we value %d, enemy value %d\n", pstMatchStatus->nWeValue, pstMatchStatus->nEnemyValue);
			cout << "Total time: " << timeCount << endl;
			return 0;
		}
	}

	delete pstAI;
	// 关闭socket
	OSCloseSocket(hSocket);
	// 资源回收
	free(pRecvBuffer);
	free(pSendBuffer);
	free(pstMapInfo);
	free(pstMatchStatus);
	free(pstFlyPlane);

	return 0;
}