source-engine/video/quicktime_recorder.cpp

2147 lines
63 KiB
C++
Raw Normal View History

2020-04-22 16:56:21 +00:00
//========= Copyright Valve Corporation, All rights reserved. ============//
//
//----------------------------------------------------------------------------------------
#define WIN32_LEAN_AND_MEAN
#include "quicktime_recorder.h"
#include "filesystem.h"
#ifdef _WIN32
#include "windows.h"
#endif
//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
#define SAFE_DISPOSE_HANDLE( _handle ) if ( _handle != nullptr ) { DisposeHandle( (Handle) _handle ); _handle = nullptr; }
#define SAFE_DISPOSE_GWORLD( _gworld ) if ( _gworld != nullptr ) { DisposeGWorld( _gworld ); _gworld = nullptr; }
#define SAFE_DISPOSE_MOVIE( _movie ) if ( _movie != nullptr ) { DisposeMovie( _movie ); ThreadSleep(10); Assert( GetMoviesError() == noErr ); _movie = nullptr; }
// Platform check
#if defined ( OSX ) || defined ( WIN32 )
// platform is supported
#else
#error "Unsupported Platform for QuickTime"
#endif
//-----------------------------------------------------------------------------
// Helper functions for copying and converting bitmaps
//-----------------------------------------------------------------------------
enum PixelComponent_t
{
RED = 0,
GREEN,
BLUE,
ALPHA
};
int GetBytesPerPixel( OSType pixelFormat )
{
int bpp = ( pixelFormat == k24BGRPixelFormat || pixelFormat == k24RGBPixelFormat ) ? 3 :
( pixelFormat == k32BGRAPixelFormat || pixelFormat == k32RGBAPixelFormat ) ? 4 : 0;
Assert( bpp > 0 );
return bpp;
}
int GetPixelCompnentByteOffset( OSType format, PixelComponent_t component )
{
if ( component == RED )
{
return ( format == k24RGBPixelFormat || format == k32RGBAPixelFormat ) ? 0 :
( format == k24BGRPixelFormat || format == k32BGRAPixelFormat ) ? 2 : -1;
}
if ( component == GREEN )
{
return ( format == k24RGBPixelFormat || format == k32RGBAPixelFormat ) ? 1 :
( format == k24BGRPixelFormat || format == k32BGRAPixelFormat ) ? 1 : -1;
}
if ( component == BLUE )
{
return ( format == k24RGBPixelFormat || format == k32RGBAPixelFormat ) ? 2 :
( format == k24BGRPixelFormat || format == k32BGRAPixelFormat ) ? 0 : -1;
}
if ( component == ALPHA )
{
return ( format == k32BGRAPixelFormat || format == k32RGBAPixelFormat ) ? 3 : -1;
}
Assert( false );
return -1;
}
bool CopyBitMapPixels( int width, int height, OSType srcFmt, byte *srcBase, int srcStride, OSType dstFmt, byte *dstBase, int dstStride )
{
AssertExitF( width > 0 && height > 0 && srcBase != nullptr && srcStride > 0 && dstBase != nullptr && dstStride > 0 );
// copy the bitmap pixels into our GWorld
if ( srcFmt == dstFmt ) // identical formats, memcopy each line
{
int srcLineSize = width * GetBytesPerPixel( srcFmt );
AssertExitF( srcLineSize <= dstStride && srcLineSize <= srcStride );
for ( int y = 0; y < height; y++ )
{
byte *src = srcBase + srcStride * y;
byte *dst = dstBase + dstStride * y;
memcpy( dst, src, srcLineSize );
}
return true;
}
// ok, we got some byte swizzling to do.. get the info we need
int srcBPP = GetBytesPerPixel( srcFmt );
int dstBPP = GetBytesPerPixel( dstFmt );
int rSrcIndex = GetPixelCompnentByteOffset( srcFmt, RED );
int gSrcIndex = GetPixelCompnentByteOffset( srcFmt, GREEN );
int bSrcIndex = GetPixelCompnentByteOffset( srcFmt, BLUE );
int aSrcIndex = GetPixelCompnentByteOffset( srcFmt, ALPHA );
int rDstIndex = GetPixelCompnentByteOffset( dstFmt, RED );
int gDstIndex = GetPixelCompnentByteOffset( dstFmt, GREEN );
int bDstIndex = GetPixelCompnentByteOffset( dstFmt, BLUE );
int aDstIndex = GetPixelCompnentByteOffset( dstFmt, ALPHA );
Assert( rSrcIndex >= 0 && gSrcIndex >= 0 && bSrcIndex >= 0 );
// 3 byte format to 3 byte format or a 4 byte format to a 3 byte format?
if ( dstBPP == 3 )
{
for ( int y = 0; y < height; y++ )
{
byte *src = srcBase + srcStride * y;
byte *dst = dstBase + dstStride * y;
for ( int x = 0; x < width; x++, dst+=dstBPP, src+=srcBPP )
{
dst[rDstIndex] = src[rSrcIndex];
dst[gDstIndex] = src[gSrcIndex];
dst[bDstIndex] = src[bSrcIndex];
}
}
return true;
}
AssertExitF( aDstIndex >= 0 );
// 3 byte format to 4 byte format?
if ( srcBPP == 3 && dstBPP == 4 )
{
for ( int y = 0; y < height; y++ )
{
byte *src = srcBase + srcStride * y;
byte *dst = dstBase + dstStride * y;
for ( int x = 0; x < width; x++, dst+=dstBPP, src+=srcBPP )
{
dst[rDstIndex] = src[rSrcIndex];
dst[gDstIndex] = src[gSrcIndex];
dst[bDstIndex] = src[bSrcIndex];
dst[aDstIndex] = 0xFF;
}
}
return true;
}
// 4 byte format to 4 byte format?
if ( srcBPP == 4 && dstBPP == 4 )
{
for ( int y = 0; y < height; y++ )
{
byte *src = srcBase + srcStride * y;
byte *dst = dstBase + dstStride * y;
for ( int x = 0; x < width; x++, dst+=dstBPP, src+=srcBPP )
{
dst[rDstIndex] = src[rSrcIndex];
dst[gDstIndex] = src[gSrcIndex];
dst[bDstIndex] = src[bSrcIndex];
dst[aDstIndex] = src[aSrcIndex];
}
}
return true;
}
// didn't find the format?
Assert( false );
return false;
}
//-----------------------------------------------------------------------------
// Utility functions to save targa images
//-----------------------------------------------------------------------------
#pragma pack( push, 1 )
struct TGA_Header
{
public:
byte identsize; // size of ID field that follows 18 byte header (0 usually)
byte colourmaptype; // type of colour map 0=none, 1=has palette
byte imagetype; // type of image 0=none,1=indexed,2=rgb,3=grey,+8=rle packed
short colourmapstart; // first colour map entry in palette
short colourmaplength; // number of colours in palette
byte colourmapbits; // number of bits per palette entry 15,16,24,32
short xstart; // image x origin
short ystart; // image y origin
short width; // image width in pixels
short height; // image height in pixels
byte bits; // image bits per pixel 8,16,24,32
byte descriptor; // image descriptor bits (vh flip bits)
// pixel data follows header
};
#pragma pack(pop)
void SaveToTargaFile( int frameNum, const char* pBaseFileName, int width, int height, void *pPixels, OSType PixelFormat, int strideAdjust )
{
if ( pBaseFileName == nullptr || pPixels== nullptr ) return;
Assert( sizeof( TGA_Header ) == 18 );
TGA_Header theHeader;
ZeroVar( theHeader );
int BytesPerPixel = GetBytesPerPixel( PixelFormat );
Assert( BytesPerPixel > 0 );
theHeader.imagetype = 2;
theHeader.width = (short) width;
theHeader.height = (short) height;
theHeader.colourmapbits = BytesPerPixel * 8;
theHeader.bits = BytesPerPixel * 8;
theHeader.descriptor = ( BytesPerPixel == 4) ? ( 8 | 32 ) : 32; // Targa32, Upper Left Origin, attribute (alpha) bits in bits 0-3
char TGAFileName[MAX_PATH];
V_snprintf( TGAFileName, MAX_PATH, "%s%.4d.tga", pBaseFileName, frameNum );
FileHandle_t TGAFile = g_pFullFileSystem->Open( TGAFileName, "wb" );
g_pFullFileSystem->Write( &theHeader, sizeof( theHeader ), TGAFile );
// is the buffer in BGR format?
if ( PixelFormat == k24BGRPixelFormat || PixelFormat == k32BGRAPixelFormat )
{
if ( strideAdjust == 0 )
{
g_pFullFileSystem->Write( pPixels, width * height * BytesPerPixel, TGAFile );
}
else
{
int lineWidth = width * BytesPerPixel;
int lineOffset = lineWidth + strideAdjust;
for ( int y = 0; y < height; y++ )
{
byte *pData = (byte*) pPixels + ( y * lineOffset );
g_pFullFileSystem->Write( pData, lineWidth, TGAFile );
}
}
}
else // we need to convert the bits from RGB to BGR
{
byte *pData = new byte[width * height * BytesPerPixel];
OSType tgaFormat = ( PixelFormat == k24RGBPixelFormat ) ? k24BGRPixelFormat :
( PixelFormat == k32RGBAPixelFormat ) ? k32BGRAPixelFormat : 0;
CopyBitMapPixels( width, height, PixelFormat, (byte*) pPixels, width * BytesPerPixel + strideAdjust, tgaFormat, pData, width * BytesPerPixel );
g_pFullFileSystem->Write( pData, width * height * BytesPerPixel, TGAFile );
delete [] pData;
}
g_pFullFileSystem->Close( TGAFile );
}
// ===========================================================================
// Data tables used to estimate file size
// ===========================================================================
enum EstVideoEncodeQuality_t
{
cVEQuality_Min = 0,
cVEQuality_Low = 25,
cVEQuality_Normal = 50,
cVEQuality_High = 75,
cVEQuality_Max = 100
};
struct EncodingDataRateInfo_t
{
EstVideoEncodeQuality_t m_QualitySetting;
int m_XResolution;
int m_YResolution;
float m_DataRate; // in MBits / second
};
struct VideoRes_t
{
int X, Y;
};
static EstVideoEncodeQuality_t s_QualityPresets[] =
{
cVEQuality_Min,
cVEQuality_Low,
cVEQuality_Normal,
cVEQuality_High,
cVEQuality_Max
};
static VideoRes_t s_ResolutionPresets[] =
{
{ 16, 16 },
{ 720, 480 },
{ 640, 960 },
{ 960, 640 },
{ 1280, 720 },
{ 1920, 1080 },
{ 2048, 2048 },
};
static EncodingDataRateInfo_t s_H264EncodeRates[] =
{
{ cVEQuality_Min, 16, 160, 2.00f },
{ cVEQuality_Min, 720, 480, 2.26f },
{ cVEQuality_Min, 640, 960, 2.73f },
{ cVEQuality_Min, 960, 640, 2.91f },
{ cVEQuality_Min, 1280, 720, 3.56f },
{ cVEQuality_Min, 1920, 1080, 5.6f },
{ cVEQuality_Min, 2048, 2048, 6.6f },
{ cVEQuality_Low, 16, 160, 3.00f },
{ cVEQuality_Low, 720, 480, 3.65f },
{ cVEQuality_Low, 640, 960, 4.57f },
{ cVEQuality_Low, 960, 640, 5.03f },
{ cVEQuality_Low, 1280, 720, 6.41f },
{ cVEQuality_Low, 1920, 1080, 10.57f },
{ cVEQuality_Low, 2048, 2048, 13.0f },
{ cVEQuality_Normal, 16, 160, 5.00f },
{ cVEQuality_Normal, 720, 480, 6.4f },
{ cVEQuality_Normal, 640, 960, 8.25f },
{ cVEQuality_Normal, 960, 640, 9.24f },
{ cVEQuality_Normal, 1280, 720, 12.1f },
{ cVEQuality_Normal, 1920, 1080, 20.64f },
{ cVEQuality_Normal, 2048, 2048, 25.0f },
{ cVEQuality_High, 16, 160, 9.50f },
{ cVEQuality_High, 720, 480, 11.3f },
{ cVEQuality_High, 640, 960, 15.06f },
{ cVEQuality_High, 960, 640, 16.9f },
{ cVEQuality_High, 1280, 720, 22.72 },
{ cVEQuality_High, 1920, 1080, 40.06f },
{ cVEQuality_High, 2048, 2048, 52.5f },
{ cVEQuality_Max, 16, 160, 15.50f },
{ cVEQuality_Max, 720, 480, 19.33f },
{ cVEQuality_Max, 640, 960, 29.89f },
{ cVEQuality_Max, 960, 640, 26.82f },
{ cVEQuality_Max, 1280, 720, 41.08f },
{ cVEQuality_Max, 1920, 1080, 75.14f },
{ cVEQuality_Max, 2048, 2048, 90.0f },
};
// ===========================================================================
// CQuickTimeVideoRecorder class - implements IVideoRecorder interface for
// QuickTime, and buffers commands to the actual encoder object
// ===========================================================================
CQuickTimeVideoRecorder::CQuickTimeVideoRecorder() :
m_pEncoder( nullptr ),
m_LastResult( VideoResult::SUCCESS ),
m_bHasAudio( false ),
m_bMovieFinished( false )
{
}
CQuickTimeVideoRecorder::~CQuickTimeVideoRecorder()
{
if ( m_pEncoder != nullptr )
{
// Abort any encoding in progress
if ( !m_bMovieFinished )
{
AbortMovie();
}
SAFE_DELETE( m_pEncoder );
}
}
bool CQuickTimeVideoRecorder::CreateNewMovieFile( const char *pFilename, bool hasAudio )
{
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_NOT_EMPTY( pFilename ) );
SetResult( VideoResult::OPERATION_ALREADY_PERFORMED );
AssertExitF( m_pEncoder == nullptr && !m_bMovieFinished );
// Create new video recorder
m_pEncoder = new CQTVideoFileComposer();
if ( !m_pEncoder->CreateNewMovie( pFilename, hasAudio ) )
{
SetResult( m_pEncoder->GetResult() ); // save the error result for after the encoder goes poof
SAFE_DELETE( m_pEncoder );
return false;
}
m_bHasAudio = hasAudio;
SetResult( VideoResult::SUCCESS );
return true;
}
bool CQuickTimeVideoRecorder::SetMovieVideoParameters( VideoEncodeCodec_t theCodec, int videoQuality, int movieFrameWidth, int movieFrameHeight, VideoFrameRate_t movieFPS, VideoEncodeGamma_t gamma )
{
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_IN_RANGECOUNT( theCodec, VideoEncodeCodec::DEFAULT_CODEC, VideoEncodeCodec::CODEC_COUNT ) );
AssertExitF( IS_IN_RANGE( videoQuality, VideoEncodeQuality::MIN_QUALITY, VideoEncodeQuality::MAX_QUALITY ) );
AssertExitF( IS_IN_RANGE( movieFrameWidth, cMinVideoFrameWidth, cMaxVideoFrameWidth ) && IS_IN_RANGE( movieFrameHeight, cMinVideoFrameHeight, cMaxVideoFrameHeight ) );
AssertExitF( IS_IN_RANGE( movieFPS.GetFPS(), cMinFPS, cMaxFPS ) );
AssertExitF( IS_IN_RANGECOUNT( gamma, VideoEncodeGamma::NO_GAMMA_ADJUST, VideoEncodeGamma::GAMMA_COUNT ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_pEncoder != nullptr && !m_bMovieFinished );
return m_pEncoder->SetMovieVideoParameters( movieFrameWidth, movieFrameHeight, movieFPS, theCodec, videoQuality, gamma );
}
bool CQuickTimeVideoRecorder::SetMovieSourceImageParameters( VideoEncodeSourceFormat_t srcImageFormat, int imgWidth, int imgHeight )
{
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_IN_RANGECOUNT( srcImageFormat, VideoEncodeSourceFormat::VIDEO_FORMAT_FIRST, VideoEncodeSourceFormat::VIDEO_FORMAT_COUNT ) );
AssertExitF( IS_IN_RANGE( imgWidth, cMinVideoFrameWidth, cMaxVideoFrameWidth ) && IS_IN_RANGE( imgHeight, cMinVideoFrameHeight, cMaxVideoFrameHeight ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_pEncoder != nullptr && !m_bMovieFinished );
return m_pEncoder->SetMovieSourceImageParameters( imgWidth, imgHeight, srcImageFormat );
}
bool CQuickTimeVideoRecorder::SetMovieSourceAudioParameters( AudioEncodeSourceFormat_t srcAudioFormat, int audioSampleRate, AudioEncodeOptions_t audioOptions, int audioSampleGroupSize )
{
SetResult( VideoResult::ILLEGAL_OPERATION );
AssertExitF( m_bHasAudio );
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_IN_RANGECOUNT( srcAudioFormat, AudioEncodeSourceFormat::AUDIO_NONE, AudioEncodeSourceFormat::AUDIO_FORMAT_COUNT ) );
AssertExitF( audioSampleRate == 0 || IS_IN_RANGE( audioSampleRate, cMinSampleRate, cMaxSampleRate ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_pEncoder != nullptr && !m_bMovieFinished );
bool result = m_pEncoder->SetMovieSourceAudioParameters( srcAudioFormat, audioSampleRate, audioOptions, audioSampleGroupSize );
m_bHasAudio = m_pEncoder->HasAudio(); // Audio can be turned off after specifying, so reload status
return result;
}
bool CQuickTimeVideoRecorder::IsReadyToRecord()
{
return ( m_pEncoder == nullptr || m_bMovieFinished ) ? false : m_pEncoder->IsReadyToRecord();
}
VideoResult_t CQuickTimeVideoRecorder::GetLastResult()
{
return ( m_pEncoder == nullptr ) ? m_LastResult : m_pEncoder->GetResult();
}
void CQuickTimeVideoRecorder::SetResult( VideoResult_t resultCode )
{
m_LastResult = resultCode;
if ( m_pEncoder != nullptr )
{
m_pEncoder->SetResult( resultCode );
}
}
bool CQuickTimeVideoRecorder::AppendVideoFrame( void *pFrameBuffer, int nStrideAdjustBytes )
{
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( pFrameBuffer != nullptr );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( IsReadyToRecord() );
return m_pEncoder->AppendVideoFrameToMedia( pFrameBuffer, nStrideAdjustBytes );
}
bool CQuickTimeVideoRecorder::AppendAudioSamples( void *pSampleBuffer, size_t sampleSize )
{
SetResult( VideoResult::ILLEGAL_OPERATION );
AssertExitF( m_bHasAudio );
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( pSampleBuffer != nullptr );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( IsReadyToRecord() );
return m_pEncoder->AppendAudioSamplesToMedia( pSampleBuffer, sampleSize );
}
int CQuickTimeVideoRecorder::GetFrameCount()
{
return ( m_pEncoder == nullptr ) ? 0 : m_pEncoder->GetFrameCount();
}
int CQuickTimeVideoRecorder::GetSampleCount()
{
return ( m_pEncoder == nullptr ) ? 0 : m_pEncoder->GetSampleCount();
}
VideoFrameRate_t CQuickTimeVideoRecorder::GetFPS()
{
return ( m_pEncoder == nullptr ) ? VideoFrameRate_t( 0 ) : m_pEncoder->GetFPS();
}
int CQuickTimeVideoRecorder::GetSampleRate()
{
return ( m_pEncoder == nullptr ) ? 0 : m_pEncoder->GetSampleRate();
}
bool CQuickTimeVideoRecorder::AbortMovie()
{
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_pEncoder != nullptr && !m_bMovieFinished );
m_bMovieFinished = true;
return m_pEncoder->AbortMovie();
}
bool CQuickTimeVideoRecorder::FinishMovie( bool SaveMovieToDisk )
{
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_pEncoder != nullptr && !m_bMovieFinished );
m_bMovieFinished = true;
return m_pEncoder->FinishMovie( SaveMovieToDisk );
}
#ifdef ENABLE_EXTERNAL_ENCODER_LOGGING
bool CQuickTimeVideoRecorder::LogMessage( const char *pMsg )
{
if ( m_pEncoder != nullptr )
{
m_pEncoder->LogMessage( pMsg );
}
return true;
}
#endif
bool CQuickTimeVideoRecorder::EstimateMovieFileSize( size_t *pEstSize, int movieWidth, int movieHeight, VideoFrameRate_t movieFps, float movieDuration, VideoEncodeCodec_t theCodec, int videoQuality, AudioEncodeSourceFormat_t srcAudioFormat, int audioSampleRate )
{
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertPtrExitF( pEstSize );
*pEstSize = 0;
AssertExitF( IS_IN_RANGE( movieWidth, cMinVideoFrameWidth, cMaxVideoFrameWidth ) && IS_IN_RANGE( movieHeight, cMinVideoFrameHeight, cMaxVideoFrameHeight ) );
AssertExitF( IS_IN_RANGE( movieFps.GetFPS(), cMinFPS, cMaxFPS ) && movieDuration > 0.0f );
AssertExitF( IS_IN_RANGECOUNT( theCodec, VideoEncodeCodec::DEFAULT_CODEC, VideoEncodeCodec::CODEC_COUNT ) );
AssertExitF( IS_IN_RANGE( videoQuality, VideoEncodeQuality::MIN_QUALITY, VideoEncodeQuality::MAX_QUALITY ) );
AssertExitF( IS_IN_RANGECOUNT( srcAudioFormat, AudioEncodeSourceFormat::AUDIO_NONE, AudioEncodeSourceFormat::AUDIO_FORMAT_COUNT ) );
AssertExitF( audioSampleRate == 0 || IS_IN_RANGE( audioSampleRate, cMinSampleRate, cMaxSampleRate ) );
// Determine the Quality LERP
int Q1 = VideoEncodeQuality::MIN_QUALITY, Q2 = VideoEncodeQuality::MAX_QUALITY;
float Qlerp = 0.0f;
bool bQLerp = true;
for ( int i = 0; i < ARRAYSIZE( s_QualityPresets ); i++ )
{
if ( s_QualityPresets[i] == videoQuality )
{
Q1 = videoQuality;
Q2 = videoQuality;
Qlerp = 0.0f;
bQLerp = false;
break;
}
else if ( s_QualityPresets[i] < videoQuality && s_QualityPresets[i] > Q1 )
{
Q1 = s_QualityPresets[i];
}
else if ( s_QualityPresets[i] > videoQuality && s_QualityPresets[i] < Q2 )
{
Q2 = s_QualityPresets[i];
}
}
if ( bQLerp )
{
Qlerp = ( (float) videoQuality - (float) Q1 ) / ( (float) Q2 - (float) Q1 ) ;
}
// determine the resolution lerp
VideoRes_t RES1 = { cMinVideoFrameWidth, cMinVideoFrameHeight }, RES2 = { cMaxVideoFrameWidth, cMaxVideoFrameHeight };
float RLerp = 0.0f;
bool bRLerp = true;
int nPixels = movieHeight * movieWidth;
int R1pixels = RES1.X * RES1.Y;
int R2pixels = RES2.X * RES2.Y;
for ( int i = 0; i < ARRAYSIZE( s_ResolutionPresets ); i++ )
{
if ( s_ResolutionPresets[i].X == movieWidth && s_ResolutionPresets[i].Y == movieHeight )
{
RES1 = s_ResolutionPresets[i];
RES2 = s_ResolutionPresets[i];
RLerp = 0.0f;
bRLerp = false;
break;
}
int rPixels = s_ResolutionPresets[i].X * s_ResolutionPresets[i].Y;
if ( rPixels <= nPixels && rPixels > R1pixels )
{
RES1 = s_ResolutionPresets[i];
R1pixels = rPixels;
}
else if ( rPixels > nPixels && rPixels < R2pixels )
{
RES2 = s_ResolutionPresets[i];
R2pixels = rPixels;
}
}
if ( bRLerp )
{
RLerp = (float) (nPixels - R1pixels) / (float) ( R2pixels - R1pixels );
}
// Now we see what we need to do
// We determine the estimated Data Rate
float DR = 0.0f;
if ( bQLerp == false && bRLerp == false )
{
DR = GetDataRate( videoQuality, movieWidth, movieHeight );
}
else if ( bQLerp == true && bRLerp == false )
{
float D1 = GetDataRate( Q1, movieWidth, movieHeight );
float D2 = GetDataRate( Q2, movieWidth, movieHeight );
DR = D1 + Qlerp * ( D2 - D1 );
}
else if ( bQLerp == false && bRLerp == true )
{
float D1 = GetDataRate( videoQuality, RES1.X, RES1.Y );
float D2 = GetDataRate( videoQuality, RES2.X, RES2.Y );
DR = D1 + RLerp * ( D2 - D1 );
}
else // need the full filter
{
float D1 = GetDataRate( Q1, RES1.X, RES1.Y );
float D2 = GetDataRate( Q1, RES2.X, RES2.Y );
float D3 = GetDataRate( Q2, RES1.X, RES1.Y );
float D4 = GetDataRate( Q2, RES2.X, RES2.Y );
float I1 = D1 + Qlerp * ( D3 - D1 );
float I2 = D2 + Qlerp * ( D4 - D2 );
DR = I1 + RLerp * ( I2 - I1 );
}
// Now do the big computation
// should this be 1024 * 1024?
double VideoData = DR * 1000000 / 8 * movieDuration ;
// Quick hack to guess at audio data size
double audioData = 0;
if ( srcAudioFormat == AudioEncodeSourceFormat::AUDIO_16BIT_PCMStereo )
{
audioData = ( audioSampleRate * 2 ) * ( 0.05 * DR );
}
*pEstSize = (size_t) VideoData + (size_t) audioData;
SetResult( VideoResult::SUCCESS );
return true;
}
float CQuickTimeVideoRecorder::GetDataRate( int quality, int width, int height )
{
for (int i = 0; i < ARRAYSIZE( s_H264EncodeRates ); i++ )
{
if ( s_H264EncodeRates[i].m_QualitySetting == quality && s_H264EncodeRates[i].m_XResolution == width && s_H264EncodeRates[i].m_YResolution == height )
{
return s_H264EncodeRates[i].m_DataRate;
}
}
Assert( false );
return 0.0f;
}
// ------------------------------------------------------------------------
// CQTVideoFileComposer - Class to encapsulate the creation of a QuickTime
// Movie from a sequence of uncompressed images and (future) audio samples
// ------------------------------------------------------------------------
CQTVideoFileComposer::CQTVideoFileComposer() :
m_LastResult( VideoResult::SUCCESS ),
m_bMovieCreated( false ),
m_bHasAudioTrack( false ),
m_bMovieConfigured( false ),
m_bSourceImagesConfigured( false ),
m_bSourceAudioConfigured( false ),
m_bComposingMovie( false ),
m_bMovieCompleted( false ),
m_nFramesAdded( 0 ),
m_nAudioFramesAdded( 0 ),
m_nSamplesAdded( 0 ),
m_nSamplesAddedToMedia( 0 ),
m_MovieFrameWidth( 0 ),
m_MovieFrameHeight( 0 ),
m_MovieTimeScale( 0 ),
m_DurationPerFrame( 0 ),
m_AudioOptions( AudioEncodeOptions::NO_AUDIO_OPTIONS ),
m_SampleGrouping( AG_NONE ),
m_nAudioSampleGroupSize( 0 ),
m_AudioSourceFrequency( 0 ),
m_AudioBytesPerSample( 0 ),
m_bBufferSourceAudio( false ),
m_bLimitAudioDurationToVideo( false ),
m_srcAudioBuffer( nullptr ),
m_srcAudioBufferSize( 0 ),
m_srcAudioBufferCurrentSize( 0 ),
m_AudioSampleFrameCounter( 0 ),
m_FileName( nullptr ),
m_SrcImageWidth( 0 ),
m_SrcImageHeight( 0 ),
m_ScrImageMaxCompressedSize( 0 ),
m_SrcImageBuffer( nullptr ),
m_SrcImageCompressedBuffer( nullptr ),
m_SrcPixelFormat( 0 ),
m_SrcBytesPerPixel( 0 ),
m_GWorldPixelFormat( 0 ),
m_GWorldBytesPerPixel( 0 ),
m_GWorldImageWidth( 0 ),
m_GWorldImageHeight( 0 ),
m_srcSoundDescription( nullptr ),
m_EncodeQuality( CQTVideoFileComposer::DEFAULT_ENCODE_QUALITY ),
m_VideoCodecToUse( CQTVideoFileComposer::DEFAULT_CODEC ),
m_EncodeGamma( CQTVideoFileComposer::DEFAULT_GAMMA ),
m_theSrcGWorld( nullptr ),
m_MovieFileDataRef( nullptr ),
m_MovieFileDataRefType( 0 ),
m_MovieFileDataHandler( nullptr ),
m_theMovie( nullptr ),
m_theVideoTrack( nullptr),
m_theAudioTrack( nullptr ),
m_theVideoMedia( nullptr ),
m_theAudioMedia( nullptr )
#ifdef LOG_ENCODER_OPERATIONS
,m_LogFile( FILESYSTEM_INVALID_HANDLE )
#endif
{
m_MovieRecordFPS.SetFPS( 0, false );
m_GWorldRect.top = m_GWorldRect.left = m_GWorldRect.bottom = m_GWorldRect.right = 0;
#ifdef LOG_FRAMES_TO_TGA
ZeroVar( m_TGAFileBase );
#endif
}
CQTVideoFileComposer::~CQTVideoFileComposer()
{
if ( m_bComposingMovie )
{
AbortMovie();
}
#ifdef LOG_ENCODER_OPERATIONS
if ( m_LogFile != FILESYSTEM_INVALID_HANDLE )
{
g_pFullFileSystem->Close( m_LogFile );
m_LogFile = FILESYSTEM_INVALID_HANDLE;
}
#endif
SAFE_DELETE_ARRAY( m_FileName );
SAFE_DELETE_ARRAY( m_SrcImageBuffer );
SAFE_DELETE_ARRAY( m_srcAudioBuffer );
SAFE_DISPOSE_HANDLE( m_MovieFileDataRef );
SAFE_DISPOSE_HANDLE( m_srcSoundDescription );
SAFE_DISPOSE_HANDLE( m_SrcImageCompressedBuffer );
SAFE_DISPOSE_GWORLD( m_theSrcGWorld );
}
#ifdef LOG_ENCODER_OPERATIONS
void CQTVideoFileComposer::LogMsg( const char* pMsg, ... )
{
const int MAX_TEXT = 8192;
static char messageBuf[MAX_TEXT];
if ( m_LogFile == FILESYSTEM_INVALID_HANDLE || pMsg == nullptr )
{
return;
}
va_list marker;
va_start( marker, pMsg );
#ifdef _WIN32
int len = _vsnprintf( messageBuf, MAX_TEXT, pMsg, marker );
#elif POSIX
int len = vsnprintf( messageBuf, MAX_TEXT, pMsg, marker );
#else
#error "define vsnprintf type."
#endif
// Len < 0 represents an overflow
if( len < 0 )
{
((char*) pMsg)[MAX_TEXT-1] = nullchar;
}
va_end( marker );
g_pFullFileSystem->Write( messageBuf, V_strlen( messageBuf ), m_LogFile );
}
#endif
bool CQTVideoFileComposer::CreateNewMovie( const char *fileName, bool hasAudio )
{
// Validate input and state
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_NOT_EMPTY( fileName ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( !m_bMovieCreated && !m_bMovieCompleted );
#ifdef LOG_ENCODER_OPERATIONS
char logFileName[MAX_PATH];
V_strncpy( logFileName, fileName, MAX_PATH );
V_SetExtension( logFileName, ".log", MAX_PATH );
m_LogFile = g_pFullFileSystem->Open( logFileName, "wb" );
const char* aMsg = (hasAudio) ? "HAS" : "DOES NOT HAVE";
LogMsg( "Creating Video File: '%s' - %s AUDIO TRACK\n", fileName, aMsg );
#endif
// now create the movie file
OSErr status = noErr;
OSType movieType = FOUR_CHAR_CODE('TVOD'); // todo - change movie type??
m_MovieFileDataRef = nullptr;
m_MovieFileDataRefType = 0;
m_MovieFileDataHandler = nullptr;
CFStringRef imageStrRef = CFStringCreateWithCString ( NULL, fileName, 0 );
status = QTNewDataReferenceFromFullPathCFString( imageStrRef, (QTPathStyle) kQTNativeDefaultPathStyle, 0, &m_MovieFileDataRef, &m_MovieFileDataRefType );
AssertExitF( status == noErr );
status = CreateMovieStorage( m_MovieFileDataRef, m_MovieFileDataRefType, movieType, smCurrentScript, createMovieFileDeleteCurFile | createMovieFileDontCreateResFile, &m_MovieFileDataHandler, &m_theMovie );
AssertExitF( status == noErr );
CFRelease( imageStrRef );
m_FileName = COPY_STRING( fileName );
#ifdef LOG_FRAMES_TO_TGA
V_strncpy( m_TGAFileBase, m_FileName, sizeof( m_TGAFileBase ) );
V_StripExtension( m_TGAFileBase, m_TGAFileBase, sizeof( m_TGAFileBase ) );
#endif
// we did it! party on...
SetResult( VideoResult::SUCCESS );
m_bMovieCreated = true;
m_bHasAudioTrack = hasAudio;
return m_bMovieCreated;
}
bool CQTVideoFileComposer::SetMovieVideoParameters( int width, int height, VideoFrameRate_t movieFPS, VideoEncodeCodec_t desiredCodec, int encodeQuality, VideoEncodeGamma_t gamma )
{
// Validate input and state
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_IN_RANGE( width, cMinVideoFrameWidth, cMaxVideoFrameWidth ) && IS_IN_RANGE( height, cMinVideoFrameHeight, cMaxVideoFrameHeight ) );
AssertExitF( IS_IN_RANGE( movieFPS.GetFPS(), cMinFPS, cMaxFPS ) );
AssertExitF( IS_IN_RANGECOUNT( desiredCodec, VideoEncodeCodec::DEFAULT_CODEC, VideoEncodeCodec::CODEC_COUNT ) );
AssertExitF( IS_IN_RANGE( encodeQuality, VideoEncodeQuality::MIN_QUALITY, VideoEncodeQuality::MAX_QUALITY ) );
AssertExitF( IS_IN_RANGECOUNT( gamma, VideoEncodeGamma::NO_GAMMA_ADJUST, VideoEncodeGamma::GAMMA_COUNT ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bMovieCreated && !m_bMovieConfigured );
// Configure video parameters
m_MovieFrameWidth = width;
m_MovieFrameHeight = height;
// map the requested codec in
switch( desiredCodec )
{
case VideoEncodeCodec::MPEG2_CODEC:
{
m_VideoCodecToUse = kMpegYUV420CodecType;
break;
}
case VideoEncodeCodec::MPEG4_CODEC:
{
m_VideoCodecToUse = kMPEG4VisualCodecType;
break;
}
case VideoEncodeCodec::H261_CODEC:
{
m_VideoCodecToUse = kH261CodecType;
break;
}
case VideoEncodeCodec::H263_CODEC:
{
m_VideoCodecToUse = kH263CodecType;
break;
}
case VideoEncodeCodec::H264_CODEC:
{
m_VideoCodecToUse = kH264CodecType;
break;
}
case VideoEncodeCodec::MJPEG_A_CODEC:
{
m_VideoCodecToUse = kMotionJPEGACodecType;
break;
}
case VideoEncodeCodec::MJPEG_B_CODEC:
{
m_VideoCodecToUse = kMotionJPEGBCodecType;
break;
}
case VideoEncodeCodec::SORENSON3_CODEC:
{
m_VideoCodecToUse = kSorenson3CodecType;
break;
}
case VideoEncodeCodec::CINEPACK_CODEC:
{
m_VideoCodecToUse = kCinepakCodecType;
break;
}
default: // should never hit this because we are already range checked
{
m_VideoCodecToUse = CQTVideoFileComposer::DEFAULT_CODEC;
break;
}
}
// Determine if codec is available...
CodecInfo theInfo;
OSErr status = GetCodecInfo( &theInfo, m_VideoCodecToUse, 0 );
if ( status == noCodecErr )
{
SetResult( VideoResult::CODEC_NOT_AVAILABLE );
return false;
}
AssertExitF( status == noErr );
#ifdef LOG_ENCODER_OPERATIONS
char codecName[64];
ZeroVar( codecName );
V_memcpy( codecName, &theInfo.typeName[1], (int) theInfo.typeName[0] );
LogMsg( "Video Image Size is (%d x %d)\n", m_MovieFrameWidth, m_MovieFrameHeight );
LogMsg( "Codec selected is %s\n", codecName );
LogMsg( "Encoding Quality = %d\n", (int) encodeQuality );
LogMsg( "Encode Gamma = %d\n", (int) gamma );
#endif
// convert encoding quality into quicktime specific value
int Q = (int) encodeQuality; - (int) VideoEncodeQuality::MIN_QUALITY;
int MaxQ = (int) VideoEncodeQuality::MAX_QUALITY - (int) VideoEncodeQuality::MIN_QUALITY;
m_EncodeQuality = codecLosslessQuality * ( (float) Q / (float) MaxQ ) ;
clamp( m_EncodeQuality, codecMinQuality, codecMaxQuality );
// convert the gamma correction value into quicktime specific values
switch( gamma )
{
case VideoEncodeGamma::NO_GAMMA_ADJUST:
{
m_EncodeGamma = kQTUseSourceGammaLevel;
break;
}
case VideoEncodeGamma::PLATFORM_STANDARD_GAMMA:
{
m_EncodeGamma = kQTUsePlatformDefaultGammaLevel;
break;
}
case VideoEncodeGamma::GAMMA_1_8:
{
m_EncodeGamma = 0x0001CCCC; // (Fixed) Gamma 1.8
break;
}
case VideoEncodeGamma::GAMMA_2_2:
{
m_EncodeGamma = kQTCCIR601VideoGammaLevel;
break;
}
case VideoEncodeGamma::GAMMA_2_5:
{
m_EncodeGamma = 0x00028000; // (Fixed) Gamma 2.5
break;
}
default:
{
m_EncodeGamma = CQTVideoFileComposer::DEFAULT_GAMMA;
break;
}
}
// Process the framerate into usable values
m_MovieRecordFPS = movieFPS;
m_DurationPerFrame = m_MovieRecordFPS.GetUnitsPerFrame();
m_MovieTimeScale = m_MovieRecordFPS.GetUnitsPerSecond();
AssertExitF( m_DurationPerFrame > 0 && m_MovieTimeScale > 0 );
/* if ( movieFPS.IsNTSCRate() )
{
m_MovieTimeScale = movieFPS.GetIntFPS() * 1000;
m_DurationPerFrame = 1001;
}
else if ( movieFPS.GetUnitsPerSecond() % movieFPS.GetUnitsPerFrame() == 0 ) // integer frame rate?
{
m_MovieTimeScale = movieFPS.GetIntFPS() * 1000;
m_DurationPerFrame = 1000;
}
else // round to nearest .001 second
{
m_MovieTimeScale = (int) ( movieFPS.GetFPS() * 1000 );
m_DurationPerFrame = 1000;
}
*/
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Video Frame Rate = %f FPS\n %d time units per second\n %d time units per frame\n", m_MovieRecordFPS.GetFPS(), m_MovieRecordFPS.GetUnitsPerSecond(), m_MovieRecordFPS.GetUnitsPerFrame() );
if ( m_MovieRecordFPS.IsNTSCRate() )
LogMsg( " IS CONSIDERED NTSC RATE\n");
LogMsg( "MovieTimeScale is being set to %d\nDuration Per Frame is %d\n\n", m_MovieTimeScale, m_DurationPerFrame );
#endif
// Create the video track and media
SetResult( VideoResult::VIDEO_ERROR_OCCURED );
m_theVideoTrack = NewMovieTrack( m_theMovie, FixRatio( width, 1 ), FixRatio( height, 1 ), kNoVolume );
AssertExitF( GetMoviesError() == noErr );
m_theVideoMedia = NewTrackMedia( m_theVideoTrack, VideoMediaType, m_MovieTimeScale, NULL, 0 );
AssertExitF( GetMoviesError() == noErr );
// we have successfully configured the output movie
SetResult( VideoResult::SUCCESS );
m_bMovieConfigured = true;
return true;
}
bool CQTVideoFileComposer::SetMovieSourceImageParameters( int srcWidth, int srcHeight, VideoEncodeSourceFormat_t srcImageFormat )
{
// Validate input and state
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_IN_RANGE( srcWidth, cMinVideoFrameWidth, cMaxVideoFrameWidth ) && IS_IN_RANGE( srcHeight, cMinVideoFrameHeight, cMaxVideoFrameHeight ) );
AssertExitF( IS_IN_RANGECOUNT( srcImageFormat, VideoEncodeSourceFormat::VIDEO_FORMAT_FIRST, VideoEncodeSourceFormat::VIDEO_FORMAT_COUNT ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bMovieCreated && !m_bMovieCompleted && m_bMovieConfigured && !m_bSourceImagesConfigured );
// Setup source image format related stuff
m_SrcPixelFormat = ( srcImageFormat == VideoEncodeSourceFormat::BGRA_32BIT ) ? k32BGRAPixelFormat :
( srcImageFormat == VideoEncodeSourceFormat::BGR_24BIT ) ? k24BGRPixelFormat :
( srcImageFormat == VideoEncodeSourceFormat::RGB_24BIT ) ? k24RGBPixelFormat :
( srcImageFormat == VideoEncodeSourceFormat::RGBA_32BIT ) ? k32RGBAPixelFormat : 0;
m_SrcBytesPerPixel = GetBytesPerPixel( m_SrcPixelFormat );
// Setup source image size related stuff
m_SrcImageWidth = srcWidth;
m_SrcImageHeight = srcHeight;
m_SrcImageSize = srcWidth * srcHeight * m_SrcBytesPerPixel;
// Setup the GWorld to hold the frame to video compress
m_GWorldPixelFormat = k32BGRAPixelFormat; // can use k24BGRPixelFormat on Win32.. but it compresses wrong on OSX?
m_GWorldBytesPerPixel = 4;
m_GWorldImageWidth = CONTAINING_MULTIPLE_OF( srcWidth, 4 ); // make sure the encoded surface is a multiple of 4 in each dimensions
m_GWorldImageHeight = CONTAINING_MULTIPLE_OF( srcHeight, 4 );
m_GWorldRect.top = m_GWorldRect.left = 0;
m_GWorldRect.bottom = m_GWorldImageHeight;
m_GWorldRect.right = m_GWorldImageWidth;
// Setup the QuiuckTime Graphics World for incoming frames of video
// Always use a 32-bit GWORD to avoid encoding bugs
SetResult( VideoResult::VIDEO_ERROR_OCCURED );
OSErr status = QTNewGWorld( &m_theSrcGWorld, m_GWorldPixelFormat, &m_GWorldRect, nil, nil, 0 );
AssertExitF( status == noErr );
PixMapHandle thePixMap = GetGWorldPixMap( m_theSrcGWorld );
AssertPtrExitF( thePixMap );
status = QTSetPixMapHandleRequestedGammaLevel( thePixMap, m_EncodeGamma );
AssertExitF( status == noErr );
// Set encoding buffer to max size at max quality
// Should we try it with the actual quality setting?
status = GetMaxCompressionSize( thePixMap, &m_GWorldRect, 0, m_EncodeQuality, m_VideoCodecToUse,
(CompressorComponent)anyCodec, (long*) &m_ScrImageMaxCompressedSize );
AssertExitF( status == noErr && m_ScrImageMaxCompressedSize > 0 );
// allocated buffers for the uncompressed and compressed images
m_SrcImageBuffer = new byte[ m_SrcImageSize ];
m_SrcImageCompressedBuffer = NewHandle( m_ScrImageMaxCompressedSize );
// we have successfully configured the video input images
SetResult( VideoResult::SUCCESS );
m_bSourceImagesConfigured = true;
return CheckForReadyness(); // We are ready to go if audio is...
}
bool CQTVideoFileComposer::SetMovieSourceAudioParameters( AudioEncodeSourceFormat_t srcAudioFormat, int audioSampleRate, AudioEncodeOptions_t audioOptions, int audioSampleGroupSize )
{
SetResult( VideoResult::ILLEGAL_OPERATION );
AssertExitF( m_bHasAudioTrack );
// Validate input and state
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( IS_IN_RANGECOUNT( srcAudioFormat, AudioEncodeSourceFormat::AUDIO_NONE, AudioEncodeSourceFormat::AUDIO_FORMAT_COUNT ) );
AssertExitF( audioSampleRate == 0 || IS_IN_RANGE( audioSampleRate, cMinSampleRate, cMaxSampleRate ) );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bMovieCreated && !m_bMovieCompleted && m_bMovieConfigured && !m_bSourceAudioConfigured );
// it is possible to disable audio here by passing in AudioEncodeSourceFormat::AUDIO_NONE even
// if the movie was created with the hasHadio flag set to true, or by setting the sample rate to 0
if ( srcAudioFormat == AudioEncodeSourceFormat::AUDIO_NONE || audioSampleRate == 0 )
{
m_bHasAudioTrack = false;
}
else
{
m_AudioOptions = audioOptions;
// Setup the audio frequency
m_AudioSourceFrequency = audioSampleRate;
// Create the audio track and media
SetResult( VideoResult::AUDIO_ERROR_OCCURED );
m_theAudioTrack = NewMovieTrack( m_theMovie, 0, 0, kFullVolume );
AssertExitF( GetMoviesError() == noErr );
m_theAudioMedia = NewTrackMedia( m_theAudioTrack, SoundMediaType, (TimeScale) audioSampleRate, NULL, 0 );
AssertExitF( GetMoviesError() == noErr );
// Setup the Audio Sound description
AudioStreamBasicDescription inASBD;
switch( srcAudioFormat )
{
case AudioEncodeSourceFormat::AUDIO_16BIT_PCMStereo:
{
inASBD.mSampleRate = Float64( audioSampleRate );
inASBD.mFormatID = kAudioFormatLinearPCM;
inASBD.mFormatFlags = kAudioFormatFlagIsSignedInteger | kAudioFormatFlagIsPacked;
inASBD.mBytesPerPacket = 4;
inASBD.mFramesPerPacket = 1;
inASBD.mBytesPerPacket = 4;
inASBD.mChannelsPerFrame = 2;
inASBD.mBitsPerChannel = 16;
inASBD.mReserved = 0;
break;
}
default:
{
Assert( false ); // Impossible.. we hope
return false;
}
}
m_AudioBytesPerSample = inASBD.mBytesPerPacket;
OSStatus result = QTSoundDescriptionCreate( &inASBD, NULL, 0, NULL, 0, kQTSoundDescriptionKind_Movie_LowestPossibleVersion, (SoundDescriptionHandle*) &m_srcSoundDescription );
AssertExitF( result == noErr );
// Setup audio sample buffering if needed
m_bLimitAudioDurationToVideo = BITFLAGS_SET( audioOptions, AudioEncodeOptions::LIMIT_AUDIO_TRACK_TO_VIDEO_DURATION );
m_SampleGrouping = ( BITFLAGS_SET( audioOptions, AudioEncodeOptions::GROUP_SIZE_IS_VIDEO_FRAME ) ) ? CQTVideoFileComposer::AG_PER_FRAME :
( BITFLAGS_SET( audioOptions, AudioEncodeOptions::USE_AUDIO_ENCODE_GROUP_SIZE ) ) ? CQTVideoFileComposer::AG_FIXED_SIZE : AG_NONE;
// check for invalid sample grouping duration
if ( m_SampleGrouping == AG_FIXED_SIZE && ( audioSampleGroupSize < MIN_AUDIO_SAMPLE_GROUP_SIZE || audioSampleGroupSize > MAX_AUDIO_GROUP_SIZE_IN_SEC * m_AudioSourceFrequency ) )
{
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
Assert( false );
return false;
}
m_bBufferSourceAudio = ( m_SampleGrouping != AG_NONE ) || m_bLimitAudioDurationToVideo;
// Set up an audio buffer than can hold the maxium specified duration
if ( m_bBufferSourceAudio )
{
m_srcAudioBufferSize = m_AudioSourceFrequency * m_AudioBytesPerSample * MAX_AUDIO_GROUP_SIZE_IN_SEC;
m_srcAudioBuffer = new byte[m_srcAudioBufferSize];
m_srcAudioBufferCurrentSize = 0;
}
if ( m_SampleGrouping == AG_FIXED_SIZE )
{
// Set up to emit audio after fixed number of samples
m_nAudioSampleGroupSize = audioSampleGroupSize;
}
if ( m_SampleGrouping == AG_PER_FRAME )
{
m_AudioSampleFrameCounter = 0;
}
m_bSourceAudioConfigured = true;
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Audio Sample Grouping Mode = %d\nSample Group Size = %d \n", (int) m_SampleGrouping, m_nAudioSampleGroupSize );
LogMsg( "Audio Track Sample Rate is %d samples per second\n", m_AudioSourceFrequency );
LogMsg( "Estimated Samples per frame = %d\n\n", (int) ( m_AudioSourceFrequency / m_MovieRecordFPS.GetFPS() ) );
#endif
// finish up
SetResult( VideoResult::SUCCESS );
return CheckForReadyness(); // We are ready to go if video is...
}
// Returns true if we are not ready, or if we began movie creation successfully
// This ONLY returns false if it tried to begin the movie creation process and failed
bool CQTVideoFileComposer::CheckForReadyness()
{
return ( m_bMovieCreated && !m_bMovieCompleted && !m_bComposingMovie && m_bMovieConfigured && m_bSourceImagesConfigured &&
m_bSourceAudioConfigured == m_bHasAudioTrack ) ? BeginMovieCreation() : true;
}
bool CQTVideoFileComposer::BeginMovieCreation()
{
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bMovieCreated && !m_bMovieCompleted && !m_bComposingMovie &&
m_bMovieConfigured && m_bSourceImagesConfigured && m_bSourceAudioConfigured == m_bHasAudioTrack );
// Open the tracks up for editing
SetResult( VideoResult::VIDEO_ERROR_OCCURED );
OSErr status = BeginMediaEdits( m_theVideoMedia );
AssertExitF( status == noErr );
if ( m_bHasAudioTrack )
{
OSErr status = BeginMediaEdits( m_theAudioMedia );
AssertExitF( status == noErr );
}
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Media Tracks opened for editing\n\n" );
#endif
// We are now ready to take in data to make a movie with
SetResult( VideoResult::SUCCESS );
m_bComposingMovie = true;
return true;
}
bool CQTVideoFileComposer::AppendVideoFrameToMedia( void *ImageBuffer, int strideAdjustBytes )
{
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "AppendVideoFrameToMedia( %8.8x ) called for %d --- ", ImageBuffer, m_nFramesAdded+1 );
#endif
// Validate input and state
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( ImageBuffer != nullptr );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bComposingMovie && !m_bMovieCompleted );
SetResult( VideoResult::VIDEO_ERROR_OCCURED );
// Get the pixmap
PixMapHandle thePixMap = GetGWorldPixMap( m_theSrcGWorld );
AssertPtrExitF( thePixMap );
// copy the raw image into our bitmap
AssertExitF( LockPixels( thePixMap ) );
byte *srcBase = (byte*) ImageBuffer;
int srcStride = m_SrcImageWidth * m_SrcBytesPerPixel + strideAdjustBytes;
byte *dstBase = nullptr;
int dstStride = 0;
#if defined ( WIN32 )
// Get the HBITMAP of our GWorld
HBITMAP theHBITMAP = (HBITMAP) GetPortHBITMAP( (GrafPtr) m_theSrcGWorld );
// retrieve the bitmap info header information
BITMAP bmp;
AssertExitF( GetObject( theHBITMAP, sizeof(BITMAP), (LPSTR) &bmp) );
// validate the BMP we just got
AssertExitF( bmp.bmWidth == m_SrcImageWidth && bmp.bmHeight == m_SrcImageHeight && bmp.bmBitsPixel == 8 * m_GWorldBytesPerPixel );
// setup the pixel copy info
dstBase = (byte*)bmp.bmBits;
dstStride = bmp.bmWidthBytes;
#elif defined ( OSX )
// setup the pixel copy info
dstBase = (byte*) GetPixBaseAddr( thePixMap );
dstStride = GetPixRowBytes( thePixMap );
#endif
AssertExitF( dstBase != nullptr && dstStride > 0 );
// save a TGA if we are running diagnostics
#if defined ( LOG_FRAMES_TO_TGA )
if ( ( m_nFramesAdded % LOG_FRAMES_TO_TGA_INTERVAL ) == 0 )
{
SaveToTargaFile( m_nFramesAdded, m_TGAFileBase, m_SrcImageWidth, m_SrcImageHeight, srcBase, m_SrcPixelFormat, strideAdjustBytes );
}
#endif
// copy the supplied pixel buffer into our GWORLD data
if ( !CopyBitMapPixels( m_SrcImageWidth, m_SrcImageHeight,
m_SrcPixelFormat, srcBase, srcStride,
m_GWorldPixelFormat, dstBase, dstStride ) )
{
Assert( false );
return false;
}
// You are now free to move about the cabin...
UnlockPixels( thePixMap );
// allocate a handle which CompressImage will resize...
ImageDescriptionHandle theImageDescHandle = (ImageDescriptionHandle) NewHandle( sizeof(ImageDescriptionHandle) );
AssertExitF( theImageDescHandle != nullptr );
// compress the single image
OSErr status = CompressImage( thePixMap, &m_GWorldRect, m_EncodeQuality,
m_VideoCodecToUse, theImageDescHandle, *m_SrcImageCompressedBuffer );
if ( status != noErr )
{
Assert( false ); // tell the user
SAFE_DISPOSE_HANDLE( theImageDescHandle );
return false;
}
TimeValue addedTime = 0;
// Lets add gamma info the image description
if ( m_EncodeGamma != kQTUseSourceGammaLevel )
{
Fixed newGamma = m_EncodeGamma;
status = ICMImageDescriptionSetProperty( theImageDescHandle, kQTPropertyClass_ImageDescription, kICMImageDescriptionPropertyID_GammaLevel, sizeof( newGamma ), &newGamma );
AssertExitF( status == noErr );
}
// add the compressed image to the movie stream
status = AddMediaSample( m_theVideoMedia, m_SrcImageCompressedBuffer, 0, (**theImageDescHandle).dataSize, m_DurationPerFrame,
(SampleDescriptionHandle) theImageDescHandle, 1, 0, &addedTime );
if ( status != noErr )
{
Assert( false ); // tell the user
SAFE_DISPOSE_HANDLE( theImageDescHandle );
return false;
}
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Video Frame %d added to Video Media: Duration = %d, Inserted at Time %d\n", m_nFramesAdded+1, m_DurationPerFrame, addedTime );
#endif
// free up dynamic resources
SAFE_DISPOSE_HANDLE( theImageDescHandle );
// Report success
SetResult( VideoResult::SUCCESS );
m_nFramesAdded++;
return true;
}
int CQTVideoFileComposer::GetAudioSampleCountThruFrame( int frameNo )
{
if ( frameNo < 1 )
{
return 0;
}
double secondsSoFar = (double) ( frameNo * m_DurationPerFrame ) / (double) m_MovieTimeScale;
int nAudioSamples = (int) floor( secondsSoFar * (double) m_AudioSourceFrequency );
return nAudioSamples;
}
bool CQTVideoFileComposer::AppendAudioSamplesToMedia( void *soundBuffer, size_t bufferSize )
{
SetResult( VideoResult::ILLEGAL_OPERATION );
AssertExitF( m_bHasAudioTrack );
// Validate input and state
SetResult( VideoResult::BAD_INPUT_PARAMETERS );
AssertExitF( soundBuffer != nullptr && bufferSize % m_AudioBytesPerSample == 0 );
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bComposingMovie && !m_bMovieCompleted );
int nSamples = bufferSize / m_AudioBytesPerSample;
Assert( bufferSize % m_AudioBytesPerSample == 0 );
TimeValue64 insertTime64 = 0;
OSErr status = noErr;
bool retest;
int samplesToEmit, MaxCanAdd, nSamplesAvailable;
m_nSamplesAdded+= nSamples; // track samples given to encoder
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "%d Audio Samples Submitted (%d total) -- ", nSamples, m_nSamplesAdded );
#endif
// We can pass in 0 bytes to trigger a flush...
if ( nSamples == 0 )
{
if ( !m_bBufferSourceAudio || m_srcAudioBufferCurrentSize == 0 )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "NO SAMPLES TO PROCESS. EXIT\n" );
#endif
goto finish_up;
}
}
// Are we not buffering audio?
if ( !m_bBufferSourceAudio && nSamples > 0 )
{
SetResult( VideoResult::AUDIO_ERROR_OCCURED );
status = AddMediaSample2( m_theAudioMedia, (const UInt8 *) soundBuffer, (ByteCount) bufferSize,
(TimeValue64) 1, (TimeValue64) 0, (SampleDescriptionHandle) m_srcSoundDescription,
(ItemCount) nSamples, 0, &insertTime64 );
AssertExitF( status == noErr );
m_nSamplesAddedToMedia+= nSamples;
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "%d samples (%d total) inserted into Video at time %ld\n", nSamples, m_nSamplesAddedToMedia, insertTime64 );
#endif
goto finish_up;
}
// Buffering audio ....
if ( nSamples > 0 )
{
memaddr_t pSrc = (memaddr_t) soundBuffer;
size_t bytesToCopy = nSamples * m_AudioBytesPerSample;
// Is buffer big enough to hold it all?
if ( m_srcAudioBufferCurrentSize + bytesToCopy > m_srcAudioBufferSize )
{
// get a bigger buffer
size_t newBufferSize = m_srcAudioBufferSize * 2 + bytesToCopy;
byte *newBuffer = new byte[newBufferSize];
// copy buffered sound and swap out buffers
V_memcpy( newBuffer, m_srcAudioBuffer, m_srcAudioBufferCurrentSize );
delete[] m_srcAudioBuffer;
m_srcAudioBuffer = newBuffer;
m_srcAudioBufferSize = newBufferSize;
}
// Append samples to buffer
V_memcpy( m_srcAudioBuffer + m_srcAudioBufferCurrentSize, pSrc, bytesToCopy );
m_srcAudioBufferCurrentSize += bytesToCopy;
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "%d Samples buffered. Buffer=%d Samples -- ", nSamples, ( m_srcAudioBufferCurrentSize / m_AudioBytesPerSample ) );
#endif
retest_here:
nSamplesAvailable = m_srcAudioBufferCurrentSize / m_AudioBytesPerSample;
samplesToEmit = 0;
retest = false;
MaxCanAdd = ( m_bLimitAudioDurationToVideo ) ? ( GetAudioSampleCountThruFrame( m_nFramesAdded ) - m_nSamplesAddedToMedia ) : INT32_MAX;
if ( MaxCanAdd <= 0 )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Can't Add Audio Now.\n" );
#endif
goto finish_up;
}
// Now.. we determine if we are ready to insert the audio samples into the media..and if so, how much...
if ( m_SampleGrouping == AG_NONE )
{
// are we keeping audio from getting ahead of video?
Assert( m_bLimitAudioDurationToVideo );
samplesToEmit = MIN( MaxCanAdd, nSamplesAvailable );
}
else if ( m_SampleGrouping == AG_FIXED_SIZE )
{
// do we have enough to emit a sample?
if ( ( nSamplesAvailable < m_nAudioSampleGroupSize ) || ( m_bLimitAudioDurationToVideo && ( MaxCanAdd < m_nAudioSampleGroupSize ) ) )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
if ( nSamplesAvailable < m_nAudioSampleGroupSize )
LogMsg( "Need %d Samples to emit sample group\n", m_nAudioSampleGroupSize );
else
LogMsg( "Audio is caught up to Video (can add %d) \n", MaxCanAdd );
#endif
goto finish_up;
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "emitting 1 group of audio (%d samples)\n", m_nAudioSampleGroupSize );
#endif
samplesToEmit = m_nAudioSampleGroupSize;
retest = true;
}
else if ( m_SampleGrouping == AG_PER_FRAME )
{
// is the audio already caught up with the current video frame?
if ( m_bLimitAudioDurationToVideo && m_AudioSampleFrameCounter >= m_nFramesAdded )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Audio is caught up to Video\n" );
#endif
goto finish_up;
}
int curSampleCount = GetAudioSampleCountThruFrame( m_AudioSampleFrameCounter );
int nextSampleCount = GetAudioSampleCountThruFrame( m_AudioSampleFrameCounter+1 );
int thisGroupSize = nextSampleCount - curSampleCount;
Assert( m_nSamplesAddedToMedia == curSampleCount );
if ( nSamplesAvailable < thisGroupSize )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Not enough samples to fill video frame (need %d)\n", thisGroupSize );
#endif
goto finish_up;
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "emitting 1 video frame of audio (%d samples)\n", thisGroupSize );
#endif
samplesToEmit = thisGroupSize;
m_AudioSampleFrameCounter++;
retest = true;
}
else
{
Assert( false );
}
if ( samplesToEmit > 0 )
{
SetResult( VideoResult::AUDIO_ERROR_OCCURED );
status = AddMediaSample2( m_theAudioMedia, (const UInt8 *) m_srcAudioBuffer, (ByteCount) samplesToEmit * m_AudioBytesPerSample,
(TimeValue64) 1, (TimeValue64) 0, (SampleDescriptionHandle) m_srcSoundDescription,
(ItemCount) samplesToEmit, 0, &insertTime64 );
AssertExitF( status == noErr );
m_nSamplesAddedToMedia+= samplesToEmit;
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "%d samples inserted into Video (%d total) at time %ld -- ", samplesToEmit, m_nSamplesAddedToMedia, insertTime64 );
#endif
// remove added samples from sound buffer
// (this really should be a circular buffer.. but that has its own problems)
m_srcAudioBufferCurrentSize -= samplesToEmit * m_AudioBytesPerSample;
nSamplesAvailable -= samplesToEmit;
if ( nSamplesAvailable > 0 )
{
V_memcpy( m_srcAudioBuffer, m_srcAudioBuffer + (samplesToEmit * m_AudioBytesPerSample), nSamplesAvailable * m_AudioBytesPerSample );
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Buffer now =%d samples", nSamplesAvailable );
#endif
if ( retest )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( " -- rechecking -- " );
#endif
goto retest_here;
}
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "\n" );
#endif
finish_up:
// Report success
SetResult( VideoResult::SUCCESS );
return true;
}
bool CQTVideoFileComposer::SyncAndFlushAudio()
{
if ( !m_bHasAudioTrack )
{
return false;
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Resolving Audio Track...\n" );
#endif
restart_sync:
bool bPadWithSilence = BITFLAGS_SET( m_AudioOptions, AudioEncodeOptions::PAD_AUDIO_WITH_SILENCE );
int VideoDurationInSamples = GetAudioSampleCountThruFrame( m_nFramesAdded );
int CurShortfall = VideoDurationInSamples - m_nSamplesAddedToMedia;
int SilenceToEmit = 0;
bool forceFlush = false;
bool forcePartialGroupFlush = false;
int nSamplesInBuffer = ( m_bBufferSourceAudio ) ? m_srcAudioBufferCurrentSize / m_AudioBytesPerSample : 0;
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Video duration is %d frames, which is %d Audio Samples\n", m_nFramesAdded, VideoDurationInSamples );
LogMsg( "%d Samples emitted to Audio track so far. %d samples remain in audio buffer\n", m_nSamplesAddedToMedia, nSamplesInBuffer );
LogMsg( "Delta to sync end of audio to end of video is %d Samples\n", CurShortfall );
LogMsg( "Pad With Silence Mode = %d, Align End of Audio With Video Mode = %d\n", (int) bPadWithSilence, (int) m_bLimitAudioDurationToVideo );
#endif
// not grouping samples mode
if ( m_SampleGrouping == AG_NONE )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "No sample grouping Mode\n" );
#endif
Assert( m_bLimitAudioDurationToVideo == m_bBufferSourceAudio ); // if we're not limiting, we're not buffering
if ( m_bLimitAudioDurationToVideo && CurShortfall > 0 && bPadWithSilence )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Padding with %d samples to match video duration", CurShortfall );
#endif
SilenceToEmit = CurShortfall; // pad with silence
}
if ( nSamplesInBuffer > 0 || SilenceToEmit > 0 ) // force if we have something to add
{
forceFlush = true;
}
}
// Fixed sized grouping (and buffering)
if ( m_SampleGrouping == AG_FIXED_SIZE )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Fixed sample grouping mode. Group size of %d\n", m_nAudioSampleGroupSize );
#endif
// No matter what, if we have a partially filled buffer, we add silence to it to make a complete group
// do we have a partially full buffer? if so pad with silence to make a full group
if ( nSamplesInBuffer > 0 && nSamplesInBuffer % m_nAudioSampleGroupSize != 0 )
{
SilenceToEmit = m_nAudioSampleGroupSize - ( nSamplesInBuffer % m_nAudioSampleGroupSize );
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Adding %d silence samples to complete group\n", SilenceToEmit );
#endif
}
int bufferedSamples = nSamplesInBuffer + SilenceToEmit;
int newShortFall = VideoDurationInSamples - m_nSamplesAddedToMedia - bufferedSamples;
if ( bPadWithSilence && newShortFall > 0 )
{
SilenceToEmit += newShortFall; // pad with silence until audio matches video duration
forceFlush = true;
forcePartialGroupFlush = true;
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Adding %d silence samples to pad to match audio to video duration\n", newShortFall );
#endif
}
}
if ( m_SampleGrouping == AG_PER_FRAME )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Video Frame duraiton Audio grouping mode\n" );
#endif
// Have we already enough audio to match the video
if ( m_bLimitAudioDurationToVideo && m_AudioSampleFrameCounter >= m_nFramesAdded )
{
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Audio is caught up to Video\n" );
#endif
goto audio_complete;
}
// if we have anything in the buffer... pad it out with zeros
if ( nSamplesInBuffer > 0 )
{
// get the group size for the video frame the audio is currently on
int thisGroupSize = GetAudioSampleCountThruFrame( m_AudioSampleFrameCounter+1 ) - GetAudioSampleCountThruFrame( m_AudioSampleFrameCounter );
Assert( m_nSamplesAddedToMedia == GetAudioSampleCountThruFrame( m_AudioSampleFrameCounter ) );
// if we already have 1 (or more) groups in the buffer.. emit them, and restart
if ( nSamplesInBuffer >= thisGroupSize )
{
char n = nullchar;
AppendAudioSamplesToMedia( &n, 0 );
goto restart_sync;
}
SilenceToEmit = thisGroupSize - nSamplesInBuffer;
forceFlush = true;
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Adding %d silence samples to pad current group to match video frame duration\n", SilenceToEmit );
#endif
}
// with the output being aligned to a video frame, do we need to add more to pad to end of the video
int bufferedSamples = nSamplesInBuffer + SilenceToEmit;
int newShortFall = VideoDurationInSamples - m_nSamplesAddedToMedia - bufferedSamples;
if ( bPadWithSilence && newShortFall > 0 )
{
SilenceToEmit += newShortFall; // pad with silence until audio matches video duration
forceFlush = true;
forcePartialGroupFlush = true;
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Adding %d silence samples to pad audio to match video duration", newShortFall );
#endif
}
}
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "\n" );
#endif
// now we append any needed silence to the audio stream...
if ( SilenceToEmit > 0 )
{
int bufferSize = SilenceToEmit * m_AudioBytesPerSample;
byte *pSilenceBuf = new byte[ bufferSize ];
V_memset( pSilenceBuf, nullchar, bufferSize );
#ifdef LOG_ENCODER_AUDIO_OPERATIONS
LogMsg( "Appending %d Silence samples\n", SilenceToEmit );
#endif
AppendAudioSamplesToMedia( pSilenceBuf, bufferSize );
}
else
{
if ( forceFlush )
{
char n = nullchar;
AppendAudioSamplesToMedia( &n, 0 );
}
}
if ( forcePartialGroupFlush && m_srcAudioBufferCurrentSize >0 )
{
int nSamplesThisAdd = m_srcAudioBufferCurrentSize / m_AudioBytesPerSample;
TimeValue64 insertTime64 = 0;
OSErr status = AddMediaSample2( m_theAudioMedia, (const UInt8 *) m_srcAudioBuffer, (ByteCount) m_srcAudioBufferCurrentSize,
(TimeValue64) 1, (TimeValue64) 0, (SampleDescriptionHandle) m_srcSoundDescription,
(ItemCount) nSamplesThisAdd, 0, &insertTime64 );
AssertExitF( status == noErr );
m_srcAudioBufferCurrentSize = 0;
m_nSamplesAddedToMedia+= nSamplesThisAdd;
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "FORCED FLUSH - Audio Samples added to media. %d added, %d total samples, inserted at time %ld\n", nSamplesThisAdd, m_nSamplesAddedToMedia, insertTime64 );
#endif
}
audio_complete:
return true;
}
bool CQTVideoFileComposer::EndMovieCreation( bool saveMovieData )
{
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nEndMovieCreation Called Composing=%d Completed=%d\n\n", (int) m_bComposingMovie, (int) m_bMovieCompleted );
#endif
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bComposingMovie && !m_bMovieCompleted );
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nEndMovieCreation Called\n\n" );
#endif
// Stop adding to and (optionally) save the video media into the file
SetResult( VideoResult::VIDEO_ERROR_OCCURED );
if ( m_nFramesAdded > 0 )
{
TimeValue VideoDuration = GetMediaDuration( m_theVideoMedia );
AssertExitF( VideoDuration == m_nFramesAdded * m_DurationPerFrame );
OSErr status = EndMediaEdits( m_theVideoMedia );
AssertExitF( status == noErr );
if ( saveMovieData )
{
status = InsertMediaIntoTrack( m_theVideoTrack, 0, 0, VideoDuration, fixed1 );
Assert( status == noErr );
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nVideo Media inserted into Track\n" );
#endif
}
}
// Stop adding to and (optionally) save the audio media into the file
SetResult( VideoResult::AUDIO_ERROR_OCCURED );
if ( m_bHasAudioTrack && m_nSamplesAdded > 0 )
{
// flush any remaining samples in the buffer to the media
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Calling SyncAndFlushAudio()\n" );
#endif
SyncAndFlushAudio();
TimeValue AudioDuration = GetMediaDuration( m_theAudioMedia );
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Audio Duration = %d nSamples Added = %d\n", AudioDuration, m_nSamplesAdded );
#endif
// AssertExitF( AudioDuration == m_nSamplesAdded );
OSErr status = EndMediaEdits( m_theAudioMedia );
AssertExitF( status == noErr );
if ( saveMovieData )
{
status = InsertMediaIntoTrack( m_theAudioTrack, 0, 0, AudioDuration, fixed1 );
AssertExitF( status == noErr );
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nAudio Media inserted into Track\n" );
#endif
}
}
if ( saveMovieData )
{
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Saving Movie Data...\n" );
#endif
SetResult( VideoResult::FILE_ERROR_OCCURED );
OSErr status = AddMovieToStorage( m_theMovie, m_MovieFileDataHandler );
AssertExitF( status == noErr );
if ( status != noErr )
{
DataHDeleteFile( m_MovieFileDataHandler );
}
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nMovie Resource added to file. Returned Status = %d\n", (int) status );
#endif
}
// free our resources
if ( m_MovieFileDataHandler != nullptr )
{
OSErr status = CloseMovieStorage( m_MovieFileDataHandler );
m_MovieFileDataHandler = nullptr;
AssertExitF( status == noErr );
}
SAFE_DISPOSE_HANDLE( m_MovieFileDataRef );
SAFE_DISPOSE_HANDLE( m_SrcImageCompressedBuffer );
SAFE_DISPOSE_GWORLD( m_theSrcGWorld );
SAFE_DISPOSE_HANDLE( m_srcSoundDescription );
SetResult( VideoResult::SUCCESS );
m_bComposingMovie = false;
return true;
}
// The movie can be aborted at any time before completion
bool CQTVideoFileComposer::AbortMovie()
{
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( !m_bMovieCompleted );
// Shut down the movie if we are recording
if ( m_bComposingMovie )
{
if ( !EndMovieCreation( false ) )
{
return false;
}
}
if ( m_bMovieCreated )
{
if ( m_MovieFileDataHandler != nullptr )
{
SetResult( VideoResult::FILE_ERROR_OCCURED );
OSErr status = CloseMovieStorage( m_MovieFileDataHandler );
AssertExitF( status == noErr );
m_MovieFileDataHandler = nullptr;
}
SAFE_DISPOSE_HANDLE( m_MovieFileDataRef );
SAFE_DISPOSE_MOVIE( m_theMovie );
}
if ( m_FileName )
{
g_pFullFileSystem->RemoveFile( m_FileName );
}
SetResult( VideoResult::SUCCESS );
m_bMovieCompleted = true;
return true;
}
bool CQTVideoFileComposer::FinishMovie( bool SaveMovieToDisk )
{
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nFinish Movie Called\n" );
#endif
SetResult( VideoResult::OPERATION_OUT_OF_SEQUENCE );
AssertExitF( m_bComposingMovie && !m_bMovieCompleted );
// Shutdown movie creation
if ( !EndMovieCreation( SaveMovieToDisk ) )
{
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "\nEndMovieCreation Aborted\n" );
#endif
return false;
}
// todo: check on Disposing of theMovie and theMedia
if ( m_MovieFileDataHandler != nullptr )
{
SetResult( VideoResult::FILE_ERROR_OCCURED );
OSErr status = CloseMovieStorage( m_MovieFileDataHandler );
AssertExitF( status == noErr );
m_MovieFileDataHandler = nullptr;
#ifdef LOG_ENCODER_OPERATIONS
LogMsg( "Movie File Closed\n" );
#endif
}
SAFE_DISPOSE_HANDLE( m_MovieFileDataRef );
SAFE_DISPOSE_MOVIE( m_theMovie );
// if no frames have been added.. delete files
if ( SaveMovieToDisk == false || ( m_nFramesAdded <= 0 && m_nSamplesAdded <= 0) )
{
g_pFullFileSystem->RemoveFile( m_FileName );
}
SetResult( VideoResult::SUCCESS );
m_bMovieCompleted = true;
#ifdef LOG_ENCODER_OPERATIONS
g_pFullFileSystem->Close( m_LogFile );
m_LogFile = FILESYSTEM_INVALID_HANDLE;
#endif
return true;
}
void CQTVideoFileComposer::SetResult( VideoResult_t status )
{
m_LastResult = status;
}
VideoResult_t CQTVideoFileComposer::GetResult()
{
return m_LastResult;
}
bool CQTVideoFileComposer::IsReadyToRecord()
{
return ( m_bComposingMovie && !m_bMovieCompleted );
}
#ifdef ENABLE_EXTERNAL_ENCODER_LOGGING
bool CQTVideoFileComposer::LogMessage( const char *msg )
{
#ifdef LOG_ENCODER_OPERATIONS
if ( IS_NOT_EMPTY(msg) && m_LogFile != FILESYSTEM_INVALID_HANDLE )
{
g_pFullFileSystem->Write( msg, V_strlen( msg ), m_LogFile );
}
#endif
return true;
}
#endif