static HRESULT CreateAAFFile(
aafWChar * pFileName,
aafUID_constref fileKind,
testRawStorageType_t rawStorageType,
aafProductIdentification_constref productID)
{
// IAAFSession *pSession = NULL;
IAAFFile *pFile = NULL;
IAAFHeader *pHeader = NULL;
IAAFDictionary *pDictionary = NULL;
IAAFLocator *pLocator = NULL;
IAAFLocator *pLocator2 = NULL;
IAAFSourceMob *pSourceMob = NULL;
IAAFMob *pMob = NULL;
IAAFEssenceDescriptor *edesc = NULL;
IEnumAAFLocators *pEnumLocators = NULL;
aafUInt32 numLocators, numLocators2;
aafUInt32 i;
HRESULT hr = AAFRESULT_SUCCESS,
localhr = AAFRESULT_SUCCESS;
bool bFileOpen = false;
// aafUID_t ddef = kAAFDataDef_Sound;
try
{
// Remove the previous test file if any.
RemoveTestFile(pFileName);
// Create the file.
checkResult(CreateTestFile( pFileName, fileKind, rawStorageType, productID, &pFile ));
bFileOpen = true;
// We can't really do anthing in AAF without the header.
checkResult(pFile->GetHeader(&pHeader));
// Get the AAF Dictionary so that we can create valid AAF objects.
checkResult(pHeader->GetDictionary(&pDictionary));
CAAFBuiltinDefs defs (pDictionary);
//Make the first mob
// Create a Mob
checkResult(defs.cdSourceMob()->
CreateInstance(IID_IAAFSourceMob,
(IUnknown **)&pSourceMob));
// Initialize mob properties:
checkResult(pSourceMob->QueryInterface (IID_IAAFMob, (void **)&pMob));
checkResult(pMob->SetMobID(TEST_MobID));
checkResult(pMob->SetName(L"EssenceDescriptorTest"));
// Create the descriptor:
// instantiate a concrete subclass of EssenceDescriptor
checkResult(defs.cdAIFCDescriptor()->
CreateInstance(IID_IAAFEssenceDescriptor,
(IUnknown **)&edesc));
IAAFAIFCDescriptor* pAIFCDesc = NULL;
checkResult(edesc->QueryInterface (IID_IAAFAIFCDescriptor, (void **)&pAIFCDesc));
checkResult(pAIFCDesc->SetSummary (5, (unsigned char*)"TEST"));
pAIFCDesc->Release();
pAIFCDesc = NULL;
checkResult(pSourceMob->SetEssenceDescriptor (edesc));
/* CountLocators() ******************************************/
localhr = AAFRESULT_SUCCESS;
// Verify AAFRESULT_NULL_PARAM is returned
if (edesc->CountLocators(NULL) != AAFRESULT_NULL_PARAM)
localhr = AAFRESULT_TEST_FAILED;
// Verify that there are no locators
if (edesc->CountLocators(&numLocators) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
if (0 != numLocators)
localhr = AAFRESULT_TEST_FAILED;
if (localhr == AAFRESULT_SUCCESS)
cout<< " CountLocators() ... Passed"<< endl;
else
{
cout<< " CountLocators() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/* AppendLocator() ******************************************/
localhr = AAFRESULT_SUCCESS;
// Verify AAFRESULT_NULL_PARAM is returned
if (edesc->AppendLocator(NULL) != AAFRESULT_NULL_PARAM)
localhr = AAFRESULT_TEST_FAILED;
// Append and Count a bunch of Locators
for (i=1; i<=10; i++)
{
// Make a concrete subclass of locator, and attach it to
// the EssenceDescriptor
checkResult(defs.cdNetworkLocator()->
CreateInstance(IID_IAAFLocator,
(IUnknown **)&pLocator));
if (edesc->AppendLocator(pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
// Verify the number of locators
numLocators = 0;
edesc->CountLocators(&numLocators);
if (i != numLocators)
localhr = AAFRESULT_TEST_FAILED;
// Verify that locator was appended
edesc->GetLocatorAt(i-1, &pLocator2);
if (pLocator2 != pLocator)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
}
// Make sure we can't add it again
// if (edesc->AppendLocator(pLocator) != AAFRESULT_OBJECT_ALREADY_ATTACHED)
// localhr = AAFRESULT_TEST_FAILED;
if (localhr == AAFRESULT_SUCCESS)
cout<< " AppendLocator() ... Passed"<< endl;
else
{
cout<< " AppendLocator() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/* PrependLocator() ******************************************/
localhr = AAFRESULT_SUCCESS;
// Verify AAFRESULT_NULL_PARAM is returned
if (edesc->PrependLocator(NULL) != AAFRESULT_NULL_PARAM)
localhr = AAFRESULT_TEST_FAILED;
for (; i<=20; i++)
{
// Make a concrete subclass of locator, and attach it to
// the EssenceDescriptor
checkResult(defs.cdNetworkLocator()->
CreateInstance(IID_IAAFLocator,
(IUnknown **)&pLocator));
if (edesc->PrependLocator(pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
// Verify the number of locators
numLocators = 0;
edesc->CountLocators(&numLocators);
if (i != numLocators)
localhr = AAFRESULT_TEST_FAILED;
// Verify that locator was prepended
edesc->GetLocatorAt(0, &pLocator2);
if (pLocator2 != pLocator)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
}
if (localhr == AAFRESULT_SUCCESS)
cout<< " PrependLocator() ... Passed"<< endl;
else
{
cout<< " PrependLocator() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/* InsertLocatorAt() **************************************/
localhr = AAFRESULT_SUCCESS;
// Make a concrete subclass of locator to attach
checkResult(defs.cdNetworkLocator()->
CreateInstance(IID_IAAFLocator,
(IUnknown **)&pLocator));
// Verify that we can't remove an index value that is out of range
if (edesc->InsertLocatorAt(numLocators+1, pLocator) != AAFRESULT_BADINDEX)
localhr = AAFRESULT_TEST_FAILED;
// Verify behavior when NULL is passed in
if (edesc->InsertLocatorAt(1, NULL) != AAFRESULT_NULL_PARAM)
localhr = AAFRESULT_TEST_FAILED;
edesc->CountLocators(&numLocators);
// Insert it
if (edesc->InsertLocatorAt(0, pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
// Check it
edesc->GetLocatorAt(0, &pLocator2);
if (pLocator2 != pLocator)
localhr = AAFRESULT_TEST_FAILED;
// Count it
edesc->CountLocators(&numLocators2);
if (numLocators2 != numLocators+1)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
edesc->CountLocators(&numLocators);
// Make a concrete subclass of locator to attach in the middle
checkResult(defs.cdNetworkLocator()->
CreateInstance(IID_IAAFLocator,
(IUnknown **)&pLocator));
// Insert it
if (edesc->InsertLocatorAt(numLocators/2, pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
// Check it
edesc->GetLocatorAt(numLocators/2, &pLocator2);
if (pLocator2 != pLocator)
localhr = AAFRESULT_TEST_FAILED;
// Count it
edesc->CountLocators(&numLocators2);
if (numLocators2 != numLocators+1)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
edesc->CountLocators(&numLocators);
// Make a concrete subclass of locator to attach to the end
checkResult(defs.cdNetworkLocator()->
CreateInstance(IID_IAAFLocator,
(IUnknown **)&pLocator));
// Insert it. note: its 0 based so the end is numLocators - 1
if (edesc->InsertLocatorAt(numLocators-1, pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
// Check it
edesc->GetLocatorAt(numLocators-1, &pLocator2);
if (pLocator2 != pLocator)
localhr = AAFRESULT_TEST_FAILED;
// Count it
edesc->CountLocators(&numLocators2);
if (numLocators2 != numLocators+1)
localhr = AAFRESULT_TEST_FAILED;
// Make sure we can't add it again
if (edesc->InsertLocatorAt(numLocators+1, pLocator) != AAFRESULT_OBJECT_ALREADY_ATTACHED)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
if (localhr == AAFRESULT_SUCCESS)
cout<< " InsertLocatorAt() ... Passed"<< endl;
else
{
cout<< " InsertLocatorAt() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/* GetLocatorAt() ******************************************/
localhr = AAFRESULT_SUCCESS;
edesc->CountLocators(&numLocators);
// Verify that we can't remove an index value that is out of range
// note: Locators index is 0 based so the index numLocators is out of range
if (edesc->GetLocatorAt(numLocators, &pLocator) != AAFRESULT_BADINDEX)
localhr = AAFRESULT_TEST_FAILED;
// Verify behavior when NULL is passed in
if (edesc->GetLocatorAt(1, NULL) != AAFRESULT_NULL_PARAM)
localhr = AAFRESULT_TEST_FAILED;
for (i=0; i<numLocators; i++)
{
pLocator = NULL;
if (edesc->GetLocatorAt(i, &pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
if (pLocator != NULL)
pLocator->Release();
}
if (localhr == AAFRESULT_SUCCESS)
cout<< " GetLocatorAt() ... Passed"<< endl;
else
{
cout<< " GetLocatorAt() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/* GetLocators() ******************************************/
if (edesc->GetLocators(NULL) != AAFRESULT_NULL_PARAM)
localhr = AAFRESULT_TEST_FAILED;
pEnumLocators = NULL;
if (edesc->GetLocators(&pEnumLocators) == AAFRESULT_SUCCESS)
{
if (pEnumLocators != NULL)
{
// Try a simple test to confirm
if (pEnumLocators->NextOne(&pLocator) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
edesc->GetLocatorAt(0, &pLocator2);
if (pLocator != pLocator2)
localhr = AAFRESULT_TEST_FAILED;
else
pLocator2->Release();
pLocator->Release();
pLocator = 0;
pEnumLocators->Release();
}
else
localhr = AAFRESULT_TEST_FAILED;
}
else
localhr = AAFRESULT_TEST_FAILED;
if (localhr == AAFRESULT_SUCCESS)
cout<< " GetLocators() ... Passed"<< endl;
else
{
cout<< " GetLocators() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/* RemoveLocatorAt() ******************************************/
localhr = AAFRESULT_SUCCESS;
// Verify that we can't remove an index value that is out of range
// note: Locators index is 0 based so the index numLocators is out of range
if (edesc->RemoveLocatorAt (numLocators) != AAFRESULT_BADINDEX)
localhr = AAFRESULT_TEST_FAILED;
// Remove locator at beginning, but Release it first
edesc->CountLocators(&numLocators);
edesc->GetLocatorAt(0, &pLocator);
pLocator->Release();
edesc->GetLocatorAt(1, &pLocator);
if (edesc->RemoveLocatorAt (0) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
// Verify the count
edesc->CountLocators(&numLocators2);
if (numLocators2 != (numLocators - 1))
localhr = AAFRESULT_TEST_FAILED;
edesc->GetLocatorAt(0, &pLocator2);
// Verify that the locators shifted properly
if (pLocator != pLocator2)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
// Remove locator in middle, but Release it first
edesc->CountLocators(&numLocators);
edesc->GetLocatorAt((numLocators/2), &pLocator);
pLocator->Release();
edesc->GetLocatorAt((numLocators/2 +1), &pLocator);
if (edesc->RemoveLocatorAt (numLocators/2) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
edesc->CountLocators(&numLocators2);
if (numLocators2 != (numLocators - 1))
localhr = AAFRESULT_TEST_FAILED;
edesc->GetLocatorAt(numLocators/2, &pLocator2);
// Verify that the locators shifted properly
if (pLocator != pLocator2)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
// Remove locator at end, but Release it first
edesc->CountLocators(&numLocators);
edesc->GetLocatorAt(numLocators-1, &pLocator);
pLocator->Release();
edesc->GetLocatorAt(numLocators-2, &pLocator);
if (edesc->RemoveLocatorAt (numLocators-1) != AAFRESULT_SUCCESS)
localhr = AAFRESULT_TEST_FAILED;
edesc->CountLocators(&numLocators2);
if (numLocators2 != (numLocators - 1))
localhr = AAFRESULT_TEST_FAILED;
edesc->GetLocatorAt(numLocators2-1, &pLocator2);
// Verify that the locators shifted properly
if (pLocator != pLocator2)
localhr = AAFRESULT_TEST_FAILED;
pLocator->Release();
pLocator = 0;
pLocator2->Release();
pLocator2 = 0;
if (localhr == AAFRESULT_SUCCESS)
cout<< " RemoveLocatorAt() ... Passed"<< endl;
else
{
cout<< " RemoveLocatorAt() ... FAILED"<< endl;
hr = AAFRESULT_TEST_FAILED;
}
/*************************************************************/
// Add the source mob into the tree
checkResult(pHeader->AddMob(pMob));
}
catch (HRESULT& rResult)
{
hr = rResult;
}
// Cleanup object references
if (edesc)
edesc->Release();
if (pMob)
pMob->Release();
if (pSourceMob)
pSourceMob->Release();
if (pDictionary)
pDictionary->Release();
if (pHeader)
pHeader->Release();
if (pFile)
{ // Close file, clean-up and return
if (bFileOpen)
{
pFile->Save();
pFile->Close();
}
pFile->Release();
}
return hr;
}
static HRESULT CreateAAFFile(
aafWChar * pFileName,
aafUID_constref fileKind,
testRawStorageType_t rawStorageType,
aafProductIdentification_constref productID)
{
IAAFFile * pFile = NULL;
bool bFileOpen = false;
IAAFHeader * pHeader = NULL;
IAAFDictionary* pDictionary = NULL;
IAAFLocator * pLocator = NULL;
IAAFNetworkLocator * pNetLocator = NULL;
IAAFSourceMob *pSourceMob = NULL;
IAAFMob *pMob = NULL;
IAAFEssenceDescriptor *edesc = NULL;
aafUInt32 numLocators;
HRESULT hr = AAFRESULT_SUCCESS;
try
{
// Remove the previous test file if any.
RemoveTestFile(pFileName);
// Create the file.
checkResult(CreateTestFile( pFileName, fileKind, rawStorageType, productID, &pFile ));
bFileOpen = true;
// We can't really do anthing in AAF without the header.
checkResult(pFile->GetHeader(&pHeader));
// Get the AAF Dictionary so that we can create valid AAF objects.
checkResult(pHeader->GetDictionary(&pDictionary));
CAAFBuiltinDefs defs (pDictionary);
//Make the first mob
// Create a Mob
checkResult(defs.cdSourceMob()->
CreateInstance(IID_IAAFSourceMob,
(IUnknown **)&pSourceMob));
checkResult(pSourceMob->QueryInterface (IID_IAAFMob, (void **)&pMob));
checkResult(pMob->SetMobID(TEST_MobID));
checkResult(pMob->SetName(L"SourceMOBTest"));
// Create a concrete subclass of EssenceDescriptor
checkResult(defs.cdAIFCDescriptor()->
CreateInstance(IID_IAAFEssenceDescriptor,
(IUnknown **)&edesc));
IAAFAIFCDescriptor* pAIFCDesc = NULL;
checkResult(edesc->QueryInterface (IID_IAAFAIFCDescriptor, (void **)&pAIFCDesc));
checkResult(pAIFCDesc->SetSummary (5, (unsigned char*)"TEST"));
pAIFCDesc->Release();
pAIFCDesc = NULL;
// Verify that there are no locators
checkResult(edesc->CountLocators(&numLocators));
checkExpression(0 == numLocators, AAFRESULT_TEST_FAILED);
// Make a locator, and attach it to the EssenceDescriptor
checkResult(defs.cdNetworkLocator()->
CreateInstance(IID_IAAFNetworkLocator,
(IUnknown **)&pNetLocator));
checkResult(pNetLocator->QueryInterface (IID_IAAFLocator, (void **)&pLocator));
checkResult(pLocator->SetPath (TEST_PATH));
checkResult(edesc->AppendLocator(pLocator));
checkResult(pSourceMob->SetEssenceDescriptor (edesc));
// Verify that there is now one locator
checkResult(edesc->CountLocators(&numLocators));
checkExpression(1 == numLocators, AAFRESULT_TEST_FAILED);
// Add the source mob into the tree
checkResult(pHeader->AddMob(pMob));
}
catch (HRESULT& rResult)
{
hr = rResult;
}
// cleanup
if (pLocator)
pLocator->Release();
if (pNetLocator)
pNetLocator->Release();
if (edesc)
edesc->Release();
if (pMob)
pMob->Release();
if (pSourceMob)
pSourceMob->Release();
if (pDictionary)
pDictionary->Release();
if (pHeader)
pHeader->Release();
if (pFile)
{
if (bFileOpen)
{
pFile->Save();
pFile->Close();
}
pFile->Release();
}
// hr = pSession->EndSession();
// if (AAFRESULT_SUCCESS != hr)
// return hr;
// if (pSession) pSession->Release();
return hr;
}
