#include <sys/mman.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/ioctl.h>

#include "hi_common.h"
#include "hi_comm_video.h"
#include "hi_comm_sys.h"
#include "mpi_sys.h"
#include "hi_comm_vb.h"
#include "mpi_vb.h"
#include "hi_comm_aio.h"
#include "mpi_ao.h"


#define VALUE_BETWEEN(x,min,max) (((x)>=(min)) && ((x) <= (max)))

HI_S32 main(int argc, char *argv[])
{
    AUDIO_DEV AoDevId;
    AO_CHN AoChn;
    HI_U32 u32Size = 1024;
    AUDIO_SAVE_FILE_INFO_S stSaveFileInfo;
    HI_S32 s32Ret;
    FILE *fp = NULL;
    HI_CHAR aCurPath[258]; /*256 + '\n'+'\0'*/  
    HI_S32 s32CurrentPath = -1;

    printf("Usage: ./ao_dump [dev] [chn] [name] [size] [path]\n");
    printf("dev: ao dev id.\n");
    printf("chn: ao chn id.\n");
	printf("name: file name (default).\n");
    printf("size: file size(KB)(defalutly 1M).\n");
    printf("path: path for saving(current path if NULL).\n\n\n");

    if ((argc < 3)||(argc > 6))
    {
    	printf("wrong arg!!!!\n\n");
    	return -1;
    }

    AoDevId = atoi(argv[1]);
    if (!VALUE_BETWEEN(AoDevId, 0, AO_DEV_MAX_NUM))
    {
    	printf("grp id must be [0,1]!!!!\n\n");
    	return -1;
    }

    AoChn = atoi(argv[2]);/* chn id*/
    if (!VALUE_BETWEEN(AoDevId, 0, AIO_MAX_CHN_NUM))
    {
    	printf("chn id must be [0,16]!!!!\n\n");
    	return -1;
    }

	if (argc>=4)
    {
        if (!VALUE_BETWEEN(strlen(argv[3]), 0, 256))
        {
        	printf("path name lenth must be [0,256]!!!!\n\n");
        	return -1;
        }
		
        memcpy(stSaveFileInfo.aFileName, argv[3], strlen(argv[3]));
        stSaveFileInfo.aFileName[strlen(argv[3])] = '\0';
    }
    else
    {
        memcpy(stSaveFileInfo.aFileName, "default", strlen("default")+1);
    }
	
    if (argc>=5)
    {
        if (!VALUE_BETWEEN(atoi(argv[4]), 1, 10*1024))
        {
        	printf("file size must be [1, 10*1024]!!!!\n\n");
        	return -1;
        }
        u32Size = atoi(argv[4]);/* frame count*/
    }
	
    if (argc>=6)
    {
        if (!VALUE_BETWEEN(strlen(argv[5]), 0, 256))
        {
        	printf("path lenth must be [0,256]!!!!\n\n");
        	return -1;
        }
        memcpy(stSaveFileInfo.aFilePath,argv[5],strlen(argv[5]));
        stSaveFileInfo.aFilePath[strlen(argv[5])] = '\0';

        if (stSaveFileInfo.aFilePath[0] == '.' && (stSaveFileInfo.aFilePath[1] == '\0' || stSaveFileInfo.aFilePath[1] == '/'))
        {
            s32CurrentPath = 0;
        }
    }
    else
    {
        s32CurrentPath = 1;
    }

    if (s32CurrentPath != -1)
    {
        fp = popen("pwd", "r");
        fgets(aCurPath, sizeof(aCurPath), fp);
        if (!VALUE_BETWEEN(strlen(aCurPath), 0, 256))
        {
            printf("path lenth must be [0,256]!!!!\n\n");
            return -1;
        }
        aCurPath[strlen(aCurPath) - 1] = '/'; /*replace '\n' with '/'*/

        if (s32CurrentPath == 0)
        {
            HI_U32 i = 2;
            HI_U32 u32Len = strlen(aCurPath);
            if (!VALUE_BETWEEN(strlen(aCurPath)+strlen(stSaveFileInfo.aFilePath), 0, 256))
            {
                printf("path lenth must be [0,256]!!!!\n\n");
                return -1;
            }

            while (strlen(stSaveFileInfo.aFilePath) > 1 && stSaveFileInfo.aFilePath[i] != '\0')
            {
                
                aCurPath[u32Len+i-2] = stSaveFileInfo.aFilePath[i];

                i ++;
            }

            aCurPath[u32Len+i-2] = '\0';
        }
        snprintf(stSaveFileInfo.aFilePath, 256, "%s", aCurPath);
        
        pclose(fp);
        //memcpy(stSaveFileInfo.aFilePath,"./",strlen("./")+1);
    }
	
    stSaveFileInfo.u32FileSize = u32Size;
    stSaveFileInfo.bCfg = HI_TRUE;
	
    printf("File path:%s, file name:%s, file size:%d*1024\n\n", stSaveFileInfo.aFilePath, stSaveFileInfo.aFileName, stSaveFileInfo.u32FileSize);
    s32Ret = HI_MPI_AO_SaveFile(AoDevId, AoChn, &stSaveFileInfo);
    if (s32Ret != HI_SUCCESS)
    {
    	printf("HI_MPI_AO_SaveFile() error,ret=%x!!!!\n\n",s32Ret);
    	return -1;
    }

    return HI_SUCCESS;
}