示例代码
基本原理
- 调用acldvppCreateChannel接口创建图片数据处理的通道、调用acldvppDestroyChannel接口销毁图片数据处理的通道。
- 调用acldvppJpegEncodeAsync异步接口,将YUV格式图片编码成.jpg图片。对于异步接口,还需调用aclrtSynchronizeStream接口阻塞程序运行,直到指定Stream中的所有任务都完成。
- 根据输入图片描述信息、图片编码配置数据,调用acldvppJpegPredictEncSize接口预估图片编码后所需的输出内存的大小。
实际输出内存大小可能与调用acldvppJpegPredictEncSize接口预估的内存大小存在差异,如果用户需要获取编码后的实际输出内存大小,可通过acldvppJpegEncodeAsync接口的出参size获取。
- 输入、输出相关的约束要求,请参见功能及约束说明。
示例代码
您可以从样例介绍中获取完整样例代码。
调用接口后,需增加异常处理的分支,并记录报错日志、提示日志,此处不一一列举。以下是关键步骤的代码示例,不可以直接拷贝编译运行,仅供参考。
// 1.AscendCL初始化
aclRet = aclInit(nullptr);
// 2.运行管理资源申请(依次申请Device、Context、Stream)
aclrtContext context_;
aclrtStream stream_;
aclrtSetDevice(0);
aclrtCreateContext(&context_, 0);
aclrtCreateStream(&stream_);
// 3.创建图片数据处理通道时的通道描述信息,dvppChannelDesc_是acldvppChannelDesc类型
dvppChannelDesc_ = acldvppCreateChannelDesc();
// 4.创建图片数据处理的通道
aclRet = acldvppCreateChannel(dvppChannelDesc_);
// 5.申请输入内存(区分运行状态)
// 调用aclrtGetRunMode接口获取软件栈的运行模式,如果调用aclrtGetRunMode接口获取软件栈的运行模式为ACL_HOST,则需要通过aclrtMemcpy接口将输入图片数据传输到Device,数据传输完成后,需及时释放内存;否则直接申请并使用Device的内存
aclrtRunMode runMode;
ret = aclrtGetRunMode(&runMode);
// inputPicWidth、inputPicHeight分别表示图片的对齐后宽、对齐后高,此处以YUV420SP格式的图片为例
uint32_t PicBufferSize = inputPicWidth * inputPicHeight * 3 / 2;
if(runMode == ACL_HOST){
// 申请Host内存vpcInHostBuffer
void* vpcInHostBuffer = nullptr;
vpcInHostBuffer = malloc(PicBufferSize);
// 将输入图片读入内存中,该自定义函数ReadPicFile由用户实现
ReadPicFile(picName, vpcInHostBuffer, PicBufferSize);
// 申请Device内存inDevBuffer_
aclRet = acldvppMalloc(&inDevBuffer_, PicBufferSize);
// 通过aclrtMemcpy接口将输入图片数据传输到Device
aclRet = aclrtMemcpy(inDevBuffer_, PicBufferSize, vpcInHostBuffer, PicBufferSize, ACL_MEMCPY_HOST_TO_DEVICE);
// 数据传输完成后,及时释放内存
free(vpcInHostBuffer);
} else {
// 申请Device输入内存inDevBuffer_
ret = acldvppMalloc(&inDevBuffer_, PicBufferSize);
// 将输入图片读入内存中,该自定义函数ReadPicFile由用户实现
ReadPicFile(picName, inDevBuffer_, PicBufferSize);
}
// 6. 创建编码输入图片的描述信息,并设置各属性值
// encodeInputDesc_是acldvppPicDesc类型
encodeInputDesc_ = acldvppCreatePicDesc();
acldvppSetPicDescData(encodeInputDesc_, reinterpret_cast<void *>(inDevBuffer_));
acldvppSetPicDescFormat(encodeInputDesc_, PIXEL_FORMAT_YUV_SEMIPLANAR_420);
acldvppSetPicDescWidth(encodeInputDesc_, inputWidth_);
acldvppSetPicDescHeight(encodeInputDesc_, inputHeight_);
acldvppSetPicDescWidthStride(encodeInputDesc_, encodeInWidthStride);
acldvppSetPicDescHeightStride(encodeInputDesc_, encodeInHeightStride);
acldvppSetPicDescSize(encodeInputDesc_, inDevBufferSizeE_);
// 7. 创建图片编码配置数据,设置编码质量
// 编码质量范围[0, 100],其中level 0编码质量与level 100差不多,而在[1, 100]内数值越小输出图片质量越差。
jpegeConfig_ = acldvppCreateJpegeConfig();
acldvppSetJpegeConfigLevel(jpegeConfig_, 100);
// 8. 申请输出内存,申请Device内存encodeOutBufferDev_,存放编码后的输出数据
uint32_t outBufferSize= 0;
ret = acldvppJpegPredictEncSize(encodeInputDesc_, jpegeConfig_, &outBufferSize);
ret = acldvppMalloc(&outBufferDev_, outBufferSize)
// 9. 执行异步编码,再调用aclrtSynchronizeStream接口阻塞程序运行,直到指定Stream中的所有任务都完成
aclRet = acldvppJpegEncodeAsync(dvppChannelDesc_, encodeInputDesc_, encodeOutBufferDev_,
&outBufferSize, jpegeConfig_, stream_);
aclRet = aclrtSynchronizeStream(stream_);
// 10. 编码结束后,释放资源,包括编码输入/输出图片的描述信息、编码输入/输出内存、通道描述信息、通道等
acldvppDestroyPicDesc(encodeInputDesc_);
if(runMode == ACL_HOST) {
// 该模式下,由于处理结果在Device侧,因此需要调用内存复制接口传输结果数据后,再释放Device侧内存
// 申请Host内存outputHostBuffer
void* outputHostBuffer = nullptr;
outputHostBuffer = malloc(outBufferSize_);
// 通过aclrtMemcpy接口将Device的处理结果数据传输到Host
aclRet = aclrtMemcpy(outputHostBuffer, outBufferSize_, encodeOutBufferDev_, outBufferSize_, ACL_MEMCPY_DEVICE_TO_HOST);
// 释放掉输入输出的device内存
(void)acldvppFree(inputDevBuff);
(void)acldvppFree(encodeOutBufferDev_);
// 数据使用完成后,释放内存
free(outputHostBuffer);
} else {
// 此时运行在device侧,处理结果也在Device侧,可以根据需要操作处理结果后,释放Device侧内存
(void)acldvppFree(inputDevBuff);
(void)acldvppFree(encodeOutBufferDev_);
}
acldvppDestroyChannel(dvppChannelDesc_);
(void)acldvppDestroyChannelDesc(dvppChannelDesc_);
dvppChannelDesc_ = nullptr;
// 11. 释放运行管理资源(依次释放Stream、Context、Device)
aclrtDestroyStream(stream_);
aclrtDestroyContext(context_);
aclrtResetDevice(0);
// 12.AscendCL去初始化
aclRet = aclFinalize();
// ....
父主题: JPEGE图片编码