/************************************
 * For msmr 
 * server.c
 * tesing the speed of bufferevent_write
 * 2015-02-03
 * author@tom
************************************/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <event2/event.h>
#include <event2/bufferevent.h>
#include <event2/buffer.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>

#include <unistd.h>

#include "ev_proto.h"
#include "face_daemon_proto.h"

#include <iostream>
#include <logger.h>
#include <MaterialBuffer.h>

Logger g_logger(std::cout);

/*
void make_msg(char* mesg, int& length)
{
	EVPHeader* evpHeader = new (mesg) EVPHeader;
	evpHeader->proto = EVPProto::EVPP_PROTOBUF;
	evpHeader->cmd = FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_PB;
	evpHeader->size = length;
	
	FILE* pFile = fopen("facelist-3.pb", "rb");
	size_t fsize = fread(mesg + sizeof(EVPHeader), 1, length - sizeof(EVPHeader), pFile);
	fclose(pFile);
}
*/

void make_msg(char* mesg, int& length)
{
	EVPHeader* evpHeader = new (mesg) EVPHeader;
	evpHeader->proto = EVPProto::EVPP_RAW_BIN; // 1
	evpHeader->cmd = FaceDaemonCommand::FDC_SENSETIMEFACEDETECT_COMPARE; // 134
	evpHeader->size = sizeof(EVPHeader);

	FDP_Image* fdpImage1 = nullptr;
	{
		fdpImage1 = new (mesg + evpHeader->size) FDP_Image;

		fdpImage1->db_id = 0; // -1
		fdpImage1->mb_type = MB_Frame::MBFT_JPEG; // 7
		fdpImage1->width = 291;
		fdpImage1->height = 194;
		
		FILE* pFile = fopen("compare1.jpg", "rb");
		fdpImage1->size = fread(fdpImage1->buff, 1, length, pFile);
		fclose(pFile);
		pFile = nullptr;

		evpHeader->size += sizeof(FDP_Image) + fdpImage1->size;
		fdpImage1->hton();
	}
	
	FDP_Image* fdpImage2 = nullptr;
	{
		fdpImage2 = new (mesg + evpHeader->size) FDP_Image;
    
		fdpImage2->db_id = 0; // -1
		fdpImage2->mb_type = MB_Frame::MBFT_JPEG; // 7
		fdpImage2->width = 198;
		fdpImage2->height = 154;
		
		FILE* pFile = fopen("compare2.jpg", "rb");
		fdpImage2->size = fread(fdpImage2->buff, 1, length, pFile);
		fclose(pFile);
		pFile = nullptr;
    
		evpHeader->size += sizeof(FDP_Image) + fdpImage2->size;
		fdpImage2->hton();
	}	
	
	//FDP_Image* fdpImage2 = nullptr;
	//{
	//	fdpImage2 = new (mesg + evpHeader->size) FDP_Image;
    //
	//	fdpImage2->db_id = 0; // -1
	//	fdpImage2->mb_type = MB_Frame::MBFT_RGB565; // 14
	//	fdpImage2->width = 52;
	//	fdpImage2->height = 52;
	//	
	//	FILE* pFile = fopen("face-13-w52-h52.rgb565", "rb");
	//	fdpImage2->size = fread(fdpImage2->buff, 1, length, pFile);
	//	fclose(pFile);
	//	pFile = nullptr;
    //
	//	evpHeader->size += sizeof(FDP_Image) + fdpImage2->size;
	//	fdpImage2->hton();
	//}
	
	length = evpHeader->size;
	evpHeader->hton();
}

int main()
{
    // build the message to be sent
    int length = 1024 * 1024; // the size of message
    char* mesg = (char*)malloc((length+1)*sizeof(char)); // Look out the end mark '/0' of a C string
    if (mesg == NULL)
        exit(1);
    int i;
    //for (i=0; i<length; i++) 
    //    strcat(mesg, "a");

	make_msg(mesg, length);

    printf("%s\n", mesg);
    printf("%d\n", length);

    // build socket
    int port = 15436;

    struct sockaddr_in server_addr;
    memset(&server_addr, 0, sizeof(server_addr));
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = inet_addr("127.0.0.1");
    server_addr.sin_port = htons(port);

    // build event base
    struct event_base* base = event_base_new();

    // set TCP_NODELAY to let data arrive at the server side quickly
    evutil_socket_t fd;
    fd = socket(AF_INET, SOCK_STREAM, 0);
    struct bufferevent* conn = bufferevent_socket_new(base,fd,BEV_OPT_CLOSE_ON_FREE);
    int enable = 1;
    if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (void*)&enable, sizeof(enable)) < 0)
       printf("ERROR: TCP_NODELAY SETTING ERROR!\n");
    //bufferevent_setcb(conn, NULL, NULL, NULL, NULL); // For client, we don't need callback function
    bufferevent_enable(conn, EV_WRITE);
    if(bufferevent_socket_connect(conn,(struct sockaddr*)&server_addr,sizeof(server_addr)) == 0)
            printf("connect success\n");

    // start to send data
    bufferevent_write(conn,mesg,length);
    // check the output evbuffer
    struct evbuffer* output = bufferevent_get_output(conn);

    event_base_dispatch(base);
 
	char readbuf[10];
	int readbufsize = read(fd, readbuf, sizeof(readbuf));
	while(readbufsize>0)
	{
		readbuf[readbufsize] = '\0';
		printf("%s", readbuf);
		readbufsize = read(fd, readbuf, sizeof(readbuf));
	}
	printf("\n");

    free(mesg);
    mesg = NULL;

    bufferevent_free(conn);
    event_base_free(base);

    printf("Client program is over\n");

    return 0;
}