[AssistantTool]_6_添加生成YUV文件的功能

Posted 2021-09-08 暮色1994

[AssistantTool]_6_添加生成YUV文件的功能

创建时间：20200615 22:14:00

功能

• 生成yuv数据，用来编码使用。

主要代码

• 生成数据，写入文件中

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;
}