#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "JZsdkLib.h"
#include "version_choose.h"

#if APP_VERSION == APP_PSDK


#include "dji_typedef.h"
#include "dji_payload_camera.h"
#include "dji_logger.h"
#include "dji_error.h"
#include "utils/util_misc.h"

#include "MediaProc/MediaProc.h"
#include "camera_emu/test_payload_cam_emu_base.h"

#define VIDEO_FRAME_MAX_COUNT 18000                   //视频最大帧数18000帧
#define DATA_SEND_FROM_VIDEO_STREAM_MAX_LEN  60000    //大疆一次最多推6w长度的数据
#define COMMAND_STOP    0                             //停止指令
#define COMMAND_PAUSE   1                             //暂停指令
#define COMMAND_START   2                             //开始指令
#define VIDEO_FRAME_AUD_LEN                  6
static const uint8_t s_frameAudInfo[VIDEO_FRAME_AUD_LEN] = {0x00, 0x00, 0x00, 0x01, 0x09, 0x10};
#define Temp_H264 "/root/sdcard/DCIM/Videos/out.h264"   //临时h264的存放地址

struct FrameInfo
{
    float DurationTime ;
    int   size;
    int   pos;
}FrameInfo;

struct FrameInfo VideoFrameInfo[VIDEO_FRAME_MAX_COUNT];
int FrameNumber = 0;        //视频总帧数
int VideoFrameOneSecond = 0;   //视频一秒的帧数
int PlayFrameNumber = 0;    //当前播放的帧数

int transcode_flag = JZ_FLAGCODE_OFF;
unsigned int old_command_timeMS = 0;

//大疆推流函数，将输入的视频帧，输入到DjiPayloadCamera_SendVideoStream
int DJI_VideoDeal_Push_DJIH264Frame(unsigned char* h264_data, uint32_t data_len)
{
    //USER_LOG_INFO("推流%d",data_len);

    static T_DjiReturnCode returnCode;
    static T_DjiDataChannelState videoStreamState = {0};

    int lengthOfDataHaveBeenSent = 0;
    uint16_t lengthOfDataToBeSent = 0;

    unsigned int FrameBufSize = 0;
    FrameBufSize = data_len;

    lengthOfDataHaveBeenSent = 0;

    //高速传输通道 每秒传输 1048576字节，分0.8给视频流  /25帧数 33554.4

    //JZSDK_LOG_INFO("推送一帧");

    //视频发送函数
    while (FrameBufSize - lengthOfDataHaveBeenSent)
    {
        lengthOfDataToBeSent = USER_UTIL_MIN(DATA_SEND_FROM_VIDEO_STREAM_MAX_LEN,
                                            FrameBufSize - lengthOfDataHaveBeenSent);
        returnCode = DjiPayloadCamera_SendVideoStream(h264_data + lengthOfDataHaveBeenSent,
                                                    lengthOfDataToBeSent);
        if (returnCode != DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS){
            USER_LOG_ERROR("send video stream error: 0x%08llX.", returnCode);
            JZSDK_LOG_INFO("发送视频失败：0x%08llX.", returnCode);
        }
        lengthOfDataHaveBeenSent += lengthOfDataToBeSent;
    }


    // 查看视频流通道是否回复
    returnCode = DjiPayloadCamera_GetVideoStreamState(&videoStreamState);
    if (returnCode == DJI_ERROR_SYSTEM_MODULE_CODE_SUCCESS) {

        //realtime band width limit 实时带宽限制
        //realtimeBandwidthBeforeFlowController 流量控制器前的实时带宽
        //busyState 
        USER_LOG_DEBUG(
            "video stream state: realtimeBandwidthLimit: %d, realtimeBandwidthBeforeFlowController: %d, realtimeBandwidthAfterFlowController:%d busyState: %d.",
            videoStreamState.realtimeBandwidthLimit, videoStreamState.realtimeBandwidthBeforeFlowController,
            videoStreamState.realtimeBandwidthAfterFlowController,
            videoStreamState.busyState);
    } else {
        USER_LOG_ERROR("get video stream state error.");
    }

}

//视频添加帧头，并推送给飞机
static int DJI_VideoDeal_H264_TO_PUSH_DJIH264Frame(char* h264_data, uint32_t data_len)
{
    memcpy(h264_data + data_len, s_frameAudInfo, VIDEO_FRAME_AUD_LEN);
    data_len += VIDEO_FRAME_AUD_LEN;
    DJI_VideoDeal_Push_DJIH264Frame(h264_data, data_len);
}

//继续播放录像
static int VideoReplay_ContinuePlay()
{
    FILE *video_fp;
    char *dataBuffer = NULL; //视频队列
    int  dataLength = 0;     //队列长度

    //1、打开视频文件

    video_fp = fopen(Temp_H264, "r");
    if (video_fp == NULL)
    {
        printf("没有读到转码后的h264");
    }

    //2、从视频中读取此帧的数据
    //注册内存
    dataBuffer = calloc(VideoFrameInfo[PlayFrameNumber].size + 6, 1);
    if (dataBuffer == NULL)
    {
        printf("回放视频队列注册失败\n");
    }
    
    //移动fp的文件光标
    fseek(video_fp, VideoFrameInfo[PlayFrameNumber].pos, SEEK_SET);
    
    //读取视频
    dataLength = fread(dataBuffer, 1, VideoFrameInfo[PlayFrameNumber].size, video_fp);
    if (dataLength != VideoFrameInfo[PlayFrameNumber].size) 
    {
        printf("回放视频队列长度获取失败\n");
    } else {
        printf("回放视频队列长度获取成功:%d\n", dataLength);
    }

    //3.1关闭文件
    fclose(video_fp);

    //4、将视频推送给飞机
    DJI_VideoDeal_H264_TO_PUSH_DJIH264Frame(dataBuffer, dataLength);

    //5、释放视频队列
    free(dataBuffer);

    //6、帧数加一
    PlayFrameNumber++;
}

//重新播放录像
static int VideoReplay_AgainPlay(unsigned int timeMs)
{
    FILE *video_fp;
    char *dataBuffer = NULL; //视频队列
    int  dataLength = 0;     //队列长度
    int proportion; //每帧占比

    //1、打开视频文件

    video_fp = fopen(Temp_H264, "r");
    if (video_fp == NULL)
    {
        printf("没有读到转码后的h264");
    }
    
    //2、将时间转换为帧数 timeMS 一秒数值为1000
    proportion = 1000/VideoFrameOneSecond;     
    PlayFrameNumber = timeMs/proportion;           //当前播放到第几帧

    //3、从视频中读取此帧的数据
    //注册内存
    dataBuffer = calloc(VideoFrameInfo[PlayFrameNumber].size + 6, 1);
    if (dataBuffer == NULL)
    {
        printf("回放视频队列注册失败\n");
    }
    
    //移动fp的文件光标
    fseek(video_fp, VideoFrameInfo[PlayFrameNumber].pos, SEEK_SET);
    
    //读取视频
    dataLength = fread(dataBuffer, 1, VideoFrameInfo[PlayFrameNumber].size, video_fp);
    if (dataLength != VideoFrameInfo[PlayFrameNumber].size) 
    {
        printf("回放视频队列长度获取失败\n");
    } else {
        printf("回放视频队列长度获取成功:%d\n", dataLength);
    }

    //3.1关闭文件
    fclose(video_fp);

    //4、将视频推送给飞机
    DJI_VideoDeal_H264_TO_PUSH_DJIH264Frame(dataBuffer, dataLength);

    //5、释放视频队列
    free(dataBuffer);

    //6、帧数加一
    PlayFrameNumber++;
}

//播放录像
static int VideoReplay_PlayRecord(unsigned int timeMS)
{
    //1、处理时间
    //如果新传入的指令时间与旧的不同，  说明是重新播放视频，或者拉动的视频进度条
    if ( old_command_timeMS != timeMS)
    {
        //1、调整指令时间
        old_command_timeMS = timeMS;

        VideoReplay_AgainPlay(timeMS);
    }
    
    //否则为继续播放视频
    else
    {
        VideoReplay_ContinuePlay();
    }
    
}

static int VideoReplay_GetFrameInfo()
{
    char *FrameInfoString = NULL; //帧信息
    int ret;

    FrameInfoString = (char *)malloc(VIDEO_FRAME_MAX_COUNT * 1024);
    if (FrameInfoString == NULL) {
        printf("帧信息字符串内存注册失败\n");
        return -1;
    }
    memset(FrameInfoString, 0, VIDEO_FRAME_MAX_COUNT * 1024);

    char ffmpegCmdStr[256];
    char *str1 = Temp_H264;
    snprintf(ffmpegCmdStr, 256, "ffprobe -show_packets %s 2>/dev/null", str1);
    //snprintf(ffmpegCmdStr, 256, "ffprobe -show_packets ../test.h264 2>/dev/null");

    FILE *fp;
    //fp = JZsdk_RunSystemCmd(ffmpegCmdStr);

    fp = popen(ffmpegCmdStr, "r");
    if (fp == NULL) {
            printf("空\n");
    }

    ret = (long) fread(FrameInfoString, 1, VIDEO_FRAME_MAX_COUNT * 1024, fp);

    FrameInfoString[ret] = '\0';
    printf("ret:%d\n",ret);

    

    //show_packets读取的信息
    // [PACKET]
    // codec_type=video
    // stream_index=0
    // pts=N/A
    // pts_time=N/A
    // dts=N/A
    // dts_time=N/A
    // duration=48000
    // duration_time=0.040000
    // size=52632
    // pos=7902242
    // flags=K_
    // [/PACKET]
    const char *FrameKeyHead =    "[PACKET]";         //帧头  
    char *FrameKeyDurationTime =   "duration_time";                //当前帧的时间长度
    char *FrameKeyPos          =    "pos"     ;                        //当前帧的开头的视频长度 从0开始
    char *FrameKeySize            =    "size"  ;                        //当前帧的长度

    char *FrameLocation = NULL;
    char *FrameLocalDurationTime    = NULL; 
    char *FrameLocalPos =   NULL;
    char *FrameLocalSize =  NULL;
    char frameParameterFormat[50] = {0};
    int frameDurationTimeS = 0;
    //解析FrameInfoString
    FrameLocation = FrameInfoString;
    while (1)
    {
        //1、寻找帧头
        FrameLocation = strstr(FrameLocation, FrameKeyHead);    //返回第一次出现FrameKeyChar的地址
        if (FrameLocation == NULL) {
            printf("已到达视频尾\n");
            break;
        }

        //2、寻找此帧的一帧时间长度 的地址
        FrameLocalDurationTime = strstr(FrameLocation, FrameKeyDurationTime);    //返回第一次出现FrameKeyDurationTime
        if (FrameLocalDurationTime == NULL) {
            printf("无法获取当前帧的时间长度\n");
        }
        //3、寻找当前帧的位置 的地址
        FrameLocalPos = strstr(FrameLocation, FrameKeyPos);    //返回第一次出现FrameKeyPos的地址
        if (FrameLocalPos == NULL) {
            printf("无法获取当前帧的位置\n");
        }
    
        //4、寻找当前帧的长度   的地址
        FrameLocalSize = strstr(FrameLocation, FrameKeySize);    //返回第一次出现FrameKeySize的地址
        if (FrameLocalPos == NULL) {
            printf("无法获取当前帧的长度\n");
        }

        //解析参数
        //帧时间长度
        snprintf(frameParameterFormat, sizeof(frameParameterFormat), "%s=%%f", FrameKeyDurationTime);
        sscanf(FrameLocalDurationTime, frameParameterFormat, &VideoFrameInfo[FrameNumber].DurationTime);
        //printf("帧时间长度:%f\n",VideoFrameInfo[FrameNumber].DurationTime);
        //当前帧的字节位
        snprintf(frameParameterFormat, sizeof(frameParameterFormat), "%s=%%d", FrameKeyPos);
        sscanf(FrameLocalPos, frameParameterFormat, &VideoFrameInfo[FrameNumber].pos);
        //printf("帧字节位:%d\n",VideoFrameInfo[FrameNumber].pos);
        //当前帧长度
        snprintf(frameParameterFormat, sizeof(frameParameterFormat), "%s=%%d", FrameKeySize);
        sscanf(FrameLocalSize, frameParameterFormat, &VideoFrameInfo[FrameNumber].size);
        //printf("帧长度:%d\n",VideoFrameInfo[FrameNumber].size);

        FrameLocation += strlen(FrameKeyHead);
        FrameNumber++;
    }

    //计算视频一秒多少帧数
    VideoFrameOneSecond = 1/VideoFrameInfo[0].DurationTime;
    free(FrameInfoString);
}

static int VideoReplay_MP4_to_H264(char *MP4_path, char *H264_path)
{
    char ffmpegCmdStr[256];
    FILE *fpCommand;
    snprintf(ffmpegCmdStr, 256, "echo \"y\" | ffmpeg -i \"%s\" -codec copy -f \"%s\" \"%s\" 1>/dev/null 2>&1", MP4_path, "h264", H264_path);
    printf("%s\n",ffmpegCmdStr);
    fpCommand = popen(ffmpegCmdStr, "r");
    if (fpCommand == NULL) {
        printf("录像转码失败");
    }
    
    pclose(fpCommand);
}

//大疆录像回放相关操作的入口
int DJI_VideoReplay_input(int command, unsigned int Command_timeMS, char* path)
{
    switch (command)
    {
        //退出播放，将参数归0
        case COMMAND_STOP:
            //printf("COMMAND_STOP\n");
            FrameNumber = JZ_FLAGCODE_OFF;        //视频总帧数
            VideoFrameOneSecond = JZ_FLAGCODE_OFF;   //视频一秒的帧数
            PlayFrameNumber = JZ_FLAGCODE_OFF;    //当前播放的帧数
            transcode_flag = JZ_FLAGCODE_OFF;      //转码标志位
#ifdef MEDIA_PROC_CONFIG_STATUS_ON
            VideoMgmt_VideoStreamFlow_RecordPlay(JZ_FLAGCODE_OFF);
#endif
        break;

        case COMMAND_PAUSE:
            //JZSDK_LOG_DEBUG("dji录像推流暂停\n");
#ifdef MEDIA_PROC_CONFIG_STATUS_ON
            VideoMgmt_VideoStreamFlow_RecordPlay(JZ_FLAGCODE_OFF);
#endif
        break;

        case COMMAND_START:
            /* 第一次进入时要进行视频转码*/
            if (transcode_flag == JZ_FLAGCODE_OFF)
            {
                //视频转码
                VideoReplay_MP4_to_H264(path,  Temp_H264);

                //转码后要获取帧信息
                VideoReplay_GetFrameInfo();

                transcode_flag = JZ_FLAGCODE_ON;
            }

#ifdef MEDIA_PROC_CONFIG_STATUS_ON
            VideoMgmt_VideoStreamFlow_RecordPlay(JZ_FLAGCODE_ON);
#endif
            
            //播放录像
            VideoReplay_PlayRecord(Command_timeMS);

            //JZSDK_LOG_DEBUG("dji录像推流开始\n");
        break;
    
        default:
        break;
    }
}

//大疆psdk下，设置转码状态
int DJI_Set_CameraTransStatus(int status)
{
    DJI_CamEmu_base_CameraTransLock(status);
}


#endif