108 lines
3.5 KiB
C++
108 lines
3.5 KiB
C++
/*
|
|
* Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
/*
|
|
* vie_file_image.cc
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include "vie_file_image.h"
|
|
#include "jpeg.h"
|
|
#include "trace.h"
|
|
|
|
namespace webrtc {
|
|
|
|
int ViEFileImage::ConvertJPEGToVideoFrame(int engineId,
|
|
const char* fileNameUTF8,
|
|
VideoFrame& videoFrame)
|
|
{
|
|
// read jpeg file into temporary buffer
|
|
WebRtc_UWord8* imageBuffer;
|
|
WebRtc_UWord32 imageBufferSize;
|
|
FILE* imageFile = fopen(fileNameUTF8, "rb");
|
|
if (NULL == imageFile)
|
|
{
|
|
// error reading file
|
|
WEBRTC_TRACE(webrtc::kTraceError, webrtc::kTraceVideo, engineId,
|
|
"%s could not open file %s", __FUNCTION__, fileNameUTF8);
|
|
return -1;
|
|
}
|
|
fseek(imageFile, 0, SEEK_END);
|
|
imageBufferSize = ftell(imageFile);
|
|
fseek(imageFile, 0, SEEK_SET);
|
|
imageBuffer = new WebRtc_UWord8[imageBufferSize + 1];
|
|
if (imageBufferSize != fread(imageBuffer, sizeof(WebRtc_UWord8),
|
|
imageBufferSize, imageFile))
|
|
{
|
|
WEBRTC_TRACE(webrtc::kTraceError, webrtc::kTraceVideo, engineId,
|
|
"%s could not read file %s", __FUNCTION__, fileNameUTF8);
|
|
delete imageBuffer;
|
|
return -1;
|
|
}
|
|
fclose(imageFile);
|
|
|
|
// if this is a jpeg file, decode it
|
|
JpegDecoder decoder;
|
|
|
|
int ret = 0;
|
|
WebRtc_UWord8* imageDecodedBuffer = NULL;
|
|
WebRtc_UWord32 imageWidth, imageHeight = 0;
|
|
ret = decoder.Decode(imageBuffer, imageBufferSize, imageDecodedBuffer,
|
|
imageWidth, imageHeight);
|
|
|
|
// done with this.
|
|
delete imageBuffer;
|
|
|
|
if (-1 == ret)
|
|
{
|
|
// error decoding the file
|
|
WEBRTC_TRACE(webrtc::kTraceError, webrtc::kTraceVideo, engineId,
|
|
"%s could decode file %s from jpeg format", __FUNCTION__,
|
|
fileNameUTF8);
|
|
return -1;
|
|
} else if (-3 == ret)
|
|
{
|
|
WEBRTC_TRACE(webrtc::kTraceError, webrtc::kTraceVideo, engineId,
|
|
"%s could not convert jpeg's data to i420 format",
|
|
__FUNCTION__, fileNameUTF8);
|
|
}
|
|
|
|
WebRtc_UWord32 imageLength = (WebRtc_UWord32)(imageWidth * imageHeight
|
|
* 1.5);
|
|
if (-1 == videoFrame.Swap(imageDecodedBuffer, imageLength, imageLength))
|
|
{
|
|
WEBRTC_TRACE(
|
|
webrtc::kTraceDebug,
|
|
webrtc::kTraceVideo,
|
|
engineId,
|
|
"%s could not copy frame imageDecodedBuffer to frame videoFrame ",
|
|
__FUNCTION__, fileNameUTF8);
|
|
return -1;
|
|
}
|
|
videoFrame.SetWidth(imageWidth);
|
|
videoFrame.SetHeight(imageHeight);
|
|
return 0;
|
|
}
|
|
|
|
int ViEFileImage::ConvertPictureToVideoFrame(int engineId,
|
|
const ViEPicture& picture,
|
|
VideoFrame& videoFrame)
|
|
{
|
|
WebRtc_UWord32 pictureLength = (WebRtc_UWord32)(picture.width
|
|
* picture.height * 1.5);
|
|
videoFrame.CopyFrame(pictureLength, picture.data);
|
|
videoFrame.SetWidth(picture.width);
|
|
videoFrame.SetHeight(picture.height);
|
|
videoFrame.SetLength(pictureLength);
|
|
|
|
return 0;
|
|
}
|
|
} // namespace webrtc
|