XnStatus PlayerNode::HandleDataIndexRecord(DataIndexRecordHeader record, XnBool bReadPayload)
{
XnStatus nRetVal = XN_STATUS_OK;
XN_VALIDATE_INPUT_PTR(m_pNodeNotifications);
nRetVal = record.Decode();
XN_IS_STATUS_OK(nRetVal);
DEBUG_LOG_RECORD(record, "DataIndex");
XN_ASSERT(record.GetNodeID() != INVALID_NODE_ID);
PlayerNodeInfo* pPlayerNodeInfo = GetPlayerNodeInfo(record.GetNodeID());
XN_VALIDATE_PTR(pPlayerNodeInfo, XN_STATUS_CORRUPT_FILE);
XnUInt32 nRecordTotalSize = record.GetSize() + record.GetPayloadSize();
if (nRecordTotalSize > RECORD_MAX_SIZE)
{
XN_ASSERT(FALSE);
XN_LOG_ERROR_RETURN(XN_STATUS_INTERNAL_BUFFER_TOO_SMALL, XN_MASK_OPEN_NI, "Record size %u is larger than player internal buffer", nRecordTotalSize);
}
if (bReadPayload)
{
// make sure node exists
if (!pPlayerNodeInfo->bValid)
{
XN_ASSERT(FALSE);
return XN_STATUS_CORRUPT_FILE;
}
if (record.GetPayloadSize() != (pPlayerNodeInfo->nFrames+1) * sizeof(DataIndexEntry))
{
XN_ASSERT(FALSE);
XN_LOG_WARNING_RETURN(XN_STATUS_CORRUPT_FILE, XN_MASK_OPEN_NI, "Seek table has %u entries, but node has %u frames!", record.GetPayloadSize() / sizeof(DataIndexEntry), pPlayerNodeInfo->nFrames);
}
// allocate our data index
pPlayerNodeInfo->pDataIndex = (DataIndexEntry*)xnOSCalloc(pPlayerNodeInfo->nFrames+1, sizeof(DataIndexEntry));
XN_VALIDATE_ALLOC_PTR(pPlayerNodeInfo->pDataIndex);
//Now read the actual data
XnUInt32 nBytesRead = 0;
nRetVal = Read(pPlayerNodeInfo->pDataIndex, record.GetPayloadSize(), nBytesRead);
XN_IS_STATUS_OK(nRetVal);
if (nBytesRead < record.GetPayloadSize())
{
XN_LOG_ERROR_RETURN(XN_STATUS_CORRUPT_FILE, XN_MASK_OPEN_NI, "Not enough bytes read");
}
}
else
{
//Just skip the data
nRetVal = SkipRecordPayload(record);
XN_IS_STATUS_OK(nRetVal);
}
return XN_STATUS_OK;
}
XnStatus PlayerNode::SeekToRecordByType(XnUInt32 nNodeID, RecordType type)
{
XnStatus nRetVal = XN_STATUS_OK;
Record record(m_pRecordBuffer, RECORD_MAX_SIZE);
XnUInt32 nStartPos = TellStream();
XnBool bFound = FALSE;
XnUInt32 nPosBeforeRecord = 0;
while (!bFound && nRetVal == XN_STATUS_OK)
{
nPosBeforeRecord = TellStream();
nRetVal = ReadRecord(record);
XN_IS_STATUS_OK(nRetVal);
if ((record.GetType() == type) && (record.GetNodeID() == nNodeID))
{
bFound = TRUE;
}
else if (record.GetType() == RECORD_END)
{
nRetVal = XN_STATUS_NO_MATCH;
}
else
{
// if record has payload, skip it
nRetVal = SkipRecordPayload(record);
}
}
if (bFound)
{
// seek to before requested record
nRetVal = SeekStream(XN_OS_SEEK_SET, nPosBeforeRecord);
XN_IS_STATUS_OK(nRetVal);
}
else
{
// seek back to starting position
SeekStream(XN_OS_SEEK_SET, nStartPos);
return (nRetVal);
}
return (XN_STATUS_OK);
}
XnStatus PlayerNode::HandleNewDataRecord(NewDataRecordHeader record, XnBool bReadPayload)
{
XN_VALIDATE_INPUT_PTR(m_pNodeNotifications);
XnStatus nRetVal = record.Decode();
XN_IS_STATUS_OK(nRetVal);
DEBUG_LOG_RECORD(record, "NewData");
XN_ASSERT(record.GetNodeID() != INVALID_NODE_ID);
PlayerNodeInfo* pPlayerNodeInfo = GetPlayerNodeInfo(record.GetNodeID());
XN_VALIDATE_PTR(pPlayerNodeInfo, XN_STATUS_CORRUPT_FILE);
if (!pPlayerNodeInfo->bValid)
{
XN_ASSERT(FALSE);
return XN_STATUS_CORRUPT_FILE;
}
XnUInt32 nRecordTotalSize = record.GetSize() + record.GetPayloadSize();
if (nRecordTotalSize > RECORD_MAX_SIZE)
{
XN_ASSERT(FALSE);
XN_LOG_ERROR_RETURN(XN_STATUS_INTERNAL_BUFFER_TOO_SMALL, XN_MASK_OPEN_NI, "Record size %u is larger than player internal buffer", nRecordTotalSize);
}
pPlayerNodeInfo->nLastDataPos = TellStream() - record.GetSize();
pPlayerNodeInfo->newDataUndoInfo.nRecordPos = pPlayerNodeInfo->nLastDataPos;
pPlayerNodeInfo->newDataUndoInfo.nUndoRecordPos = record.GetUndoRecordPos();
if (record.GetFrameNumber() > pPlayerNodeInfo->nFrames)
{
XN_ASSERT(FALSE);
return XN_STATUS_CORRUPT_FILE;
}
pPlayerNodeInfo->nCurFrame = record.GetFrameNumber();
if (record.GetTimeStamp() > m_nGlobalMaxTimeStamp)
{
XN_ASSERT(FALSE);
XN_LOG_ERROR_RETURN(XN_STATUS_CORRUPT_FILE, XN_MASK_OPEN_NI, "Record timestamp for record in position %u is larger than reported max timestamp", pPlayerNodeInfo->nLastDataPos);
}
m_nTimeStamp = record.GetTimeStamp();
if (bReadPayload)
{
//Now read the actual data
XnUInt32 nBytesRead = 0;
nRetVal = Read(record.GetPayload(), record.GetPayloadSize(), nBytesRead);
XN_IS_STATUS_OK(nRetVal);
if (nBytesRead < record.GetPayloadSize())
{
XN_LOG_ERROR_RETURN(XN_STATUS_CORRUPT_FILE, XN_MASK_OPEN_NI, "Not enough bytes read");
}
const XnUInt8* pCompressedData = record.GetPayload(); //The new (compressed) data is right at the end of the header
XnUInt32 nCompressedDataSize = record.GetPayloadSize();
const XnUInt8* pUncompressedData = NULL;
XnUInt32 nUncompressedDataSize = 0;
XnCodecID compression = pPlayerNodeInfo->codec.GetCodecID();
if (compression == XN_CODEC_UNCOMPRESSED)
{
pUncompressedData = pCompressedData;
nUncompressedDataSize = nCompressedDataSize;
}
else
{
//Decode data with codec
nRetVal = pPlayerNodeInfo->codec.DecodeData(pCompressedData, nCompressedDataSize,
m_pUncompressedData, DATA_MAX_SIZE, &nUncompressedDataSize);
XN_IS_STATUS_OK(nRetVal);
pUncompressedData = m_pUncompressedData;
}
nRetVal = m_pNodeNotifications->OnNodeNewData(m_pNotificationsCookie, pPlayerNodeInfo->strName,
record.GetTimeStamp(), record.GetFrameNumber(), pUncompressedData, nUncompressedDataSize);
XN_IS_STATUS_OK(nRetVal);
}
else
{
//Just skip the data
nRetVal = SkipRecordPayload(record);
XN_IS_STATUS_OK(nRetVal);
}
return XN_STATUS_OK;
}
