[AssistantTool]_4_生成原始Color数据文件

最新推荐文章于 2021-08-23 12:47:48 发布

小砖头1994

最新推荐文章于 2021-08-23 12:47:48 发布

阅读量135

点赞数

分类专栏：工具编写_AssistantTool

本文链接：https://blog.csdn.net/u013209302/article/details/119858818

版权

工具编写_AssistantTool 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

该代码段展示了如何创建并渲染不同格式的颜色数据文件，包括FMT_NV12, FMT_YUYV和FMT_UYVY。通过循环遍历每一帧，根据指定的格式填充颜色缓冲区，并将内容写入文件。主要函数`frame_render`根据格式进行颜色填充，`createColor`负责分配内存，打开文件，调用`frame_render`并关闭文件。

摘要由CSDN通过智能技术生成

[AssistantTool]_4_生成原始Color数据文件

创建时间：20200525 23:35:58

界面介绍

变量

    enum{
        FMT_NV12,
        FMT_UYVY,
        FMT_YUYV,
    };

    int frameSize;
    int width;
    int height;
    QString fileName;
    int frameCount;
    int fmt;
    char *frameBuff;
    int fb;

主要函数

void CreateColorForm::frame_render(int frameCount)
{
    int colorIndex = 0;
    int frame = 0;
    int ret;
    char *disY = NULL;
    char *disUV = NULL;
    int tw = 0;
    while(frame < frameCount)
    {
        qDebug() << "frame_render frame:" << frame;
        switch( fmt )
        {
            case FMT_NV12:
                disY = frameBuff;
                disUV = frameBuff+width*height;
                for(int i = 0; i < height; i++)
                {
                    for(int j = 0; j < width; j++)
                    {
                        disY[i*width+j] = (j+colorIndex)%255;
                    }
                }

                for(int i = 0; i < height; i++)
                {
                    for(int j = 0; j < width/4; j++)
                    {
                        disUV[(i*width/4+j)*2+0] = 90;
                        disUV[(i*width/4+j)*2+1] = 90;
                    }
                }
                colorIndex++;
            break;
            case FMT_YUYV:
                tw = width/2;
                for(int i = 0; i < height; i++)
                {
                    for(int j = 0; j < width; j++)
                    {
                        frameBuff[(i*width+j)*2] = (j+colorIndex)%255;
                    }
                }

                for(int i = 0; i < height; i++)
                {
                    for(int j = 0; j < tw; j++)
                    {
                        frameBuff[(i*tw+j)*4+1] = 90;
                        frameBuff[(i*tw+j)*4+3] = 90;
                    }
                }
                colorIndex++;
            break;
            case FMT_UYVY:
                tw = width/2;
                for(int i = 0; i < height; i++)
                {
                    for(int j = 0; j < width; j++)
                    {
                        frameBuff[(i*width+j)*2+1] = (j+colorIndex)%255;
                    }
                }
                for(int i = 0; i < height; i++)
                {
                    for(int j = 0; j < tw; j++)
                    {
                        frameBuff[(i*tw+j)*4+0] = 90;
                        frameBuff[(i*tw+j)*4+2] = 90;
                    }
                }
                colorIndex++;
            break;
        }
        //frameBuff
        ret = write(fb, frameBuff, frameSize);
        qDebug() << "frame_render write ret = " << ret;
        frame++;
    }
}


/**
 * @brief CreateColorForm::createColor
 * 创建颜色文件， 分配内存，并调用渲染数据
 * @return 0:Ok -1:Err
 */
int CreateColorForm::createColor()
{
    frameBuff = (char*)malloc(frameSize);
    if( frameBuff == NULL )
    {
        printf("frameBuff == NULL , malloc Err\n");
    }

    qDebug() << "fileName:" << fileName;
    fb = ::open(fileName.toStdString().c_str(), O_CREAT|O_RDWR, 0777);
    if( fb < 0 )
    {
        qDebug() << "fileName:" << fileName << "open err";
    }
    else
    {
        qDebug() << "fileName:" << fileName << "open ok";
    }

    frame_render(frameCount);

    ::close(fb);
    qDebug() << "fileName:" << fileName << "close ok";
    free(frameBuff);
    qDebug() << "createColor run end ok\n";
    return 0;
}