Main.cpp

#include "stdio.h"
#include "unistd.h"
#include "getopt.h"

#include "KEncoder.h"
#include "KDemuxer.h"
#include "InputParameter.h"
#include "KErrors.h"

const char *ENC = "enc";
const char *DMX = "dmx";

struct option longopts[] = 
{
    // 选择功能， enc表示编码
    {"func", required_argument, NULL, FUNCTION},
    // 比特率
    {"br", required_argument, NULL, BITRATE},
    // 帧率
    {"fr", required_argument, NULL, FRAMERATE},
    // 帧数
    {"fc", required_argument, NULL, FRAMECOUNT},
    // 分辨率
    {"vr", required_argument, NULL, VIDEO_RESOLUTION},
    // 输出视频es流
    {"vo", required_argument, NULL, VIDEO_OUTPUT_FILE},
    // 输出音频es流
    {"ao", required_argument, NULL, AUDIO_OUTPUT_FILE},
};

int main(int argc, char **argv)
{
    int oc;
    char ec;
    char *func = NULL;
    InputParameter parameter;
    while((oc = getopt_long(argc, argv, "i:o:", longopts, NULL)) != -1)
    {
        switch(oc)
        {
            case FUNCTION:
                func = optarg;
                break;

            case INPUT_FILE:
                parameter.mInputFileName = optarg;
                break;

            case OUTPUT_FILE:
                parameter.mOutputFileName = optarg;
                break;

            case BITRATE:
                parameter.mBitRate = atoi(optarg);
                break;

            case FRAMERATE:
                parameter.mFrameRate = atoi(optarg);
                break;

            case FRAMECOUNT:
                parameter.mFrameCount = atoi(optarg);
                break;

            case VIDEO_OUTPUT_FILE:
                parameter.mVideoOutputFileName = optarg;
                break;

            case AUDIO_OUTPUT_FILE:
                parameter.mAudioOutputFileName = optarg;
                break;

            case VIDEO_RESOLUTION:
                char *resolution = optarg;
                int length = strlen(resolution);
                int i;
                for (i = 0; i < length; i++)
                {
                    if (resolution[i] == 'x')
                    {
                        int widthLength = i;
                        char width[widthLength + 1];
                        char height[length - widthLength];
                        strncpy(width, resolution, widthLength);
                        width[widthLength] = '\0';
                        printf("w: %s\n", width);
                        strncpy(height, resolution + widthLength + 1, length - widthLength - 1);
                        height[length - widthLength - 1] = '\0';
                        printf("h: %s\n", height);
                        parameter.mWidth = atoi(width);
                        parameter.mHeight = atoi(height);
                    }
                }
                break;
        }
    }

    if (!strcmp(ENC, func))
    {
        printf("启动编码器\n");
        KErrors res = NO_ERROR;
        KEncoder *encoder = new KEncoder();
        res = encoder->configure(&parameter);
        if (res == NO_ERROR)
        {
            printf("配置编码器成功\n");
            res = encoder->start();
            if (res == NO_ERROR)
            {
                printf("开启编码器成功\n");
            }
            else
            {
                printf("开启编码器失败, error code: %d\n", res);
            }
        } 
        else 
        {
            printf("配置编码器失败, error code: %d\n", res);
        }

        if (encoder != NULL)
        {
            delete encoder;
            encoder = NULL;
        }
    }
    else if (!strcmp(DMX, func))
    {
        printf("启动解复用器\n");
        KErrors res = NO_ERROR;
        KDemuxer *demuxer = new KDemuxer();
        res = demuxer->configure(&parameter);
        if (res == NO_ERROR)
        {
            printf("配置解复用器成功\n");
            res = demuxer->start();
            if (res == NO_ERROR)
            {
                printf("开启解复用器成功\n");
            }
            else
            {
                printf("开启解复用器失败, error code: %d\n", res);
            }
        }
        else 
        {
            printf("配置解复用器失败, error code: %d\n", res);
        }

        if (demuxer != NULL)
        {
            delete demuxer;
            demuxer = NULL;
        }
    }
}