File: RecorderNode.h

package info (click to toggle)
openni 1.5.4.0%2Bdfsg-8
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 45,580 kB
  • sloc: cpp: 116,706; ansic: 58,807; sh: 10,287; cs: 7,698; java: 7,402; python: 1,547; makefile: 492; xml: 167
file content (120 lines) | stat: -rw-r--r-- 5,248 bytes parent folder | download | duplicates (7)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
/****************************************************************************
*                                                                           *
*  OpenNI 1.x Alpha                                                         *
*  Copyright (C) 2011 PrimeSense Ltd.                                       *
*                                                                           *
*  This file is part of OpenNI.                                             *
*                                                                           *
*  OpenNI is free software: you can redistribute it and/or modify           *
*  it under the terms of the GNU Lesser General Public License as published *
*  by the Free Software Foundation, either version 3 of the License, or     *
*  (at your option) any later version.                                      *
*                                                                           *
*  OpenNI is distributed in the hope that it will be useful,                *
*  but WITHOUT ANY WARRANTY; without even the implied warranty of           *
*  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the             *
*  GNU Lesser General Public License for more details.                      *
*                                                                           *
*  You should have received a copy of the GNU Lesser General Public License *
*  along with OpenNI. If not, see <http://www.gnu.org/licenses/>.           *
*                                                                           *
****************************************************************************/
#ifndef __RECORDER_NODE_H__
#define __RECORDER_NODE_H__

#include <XnModuleCppInterface.h>
#include <XnStringsHashT.h>
#include <DataRecords.h>

class Record;

class RecorderNode : public xn::ModuleRecorder
{
public:
	RecorderNode(xn::Context &context);
	virtual ~RecorderNode();
	
	//public functions
	virtual XnStatus Init();
	virtual XnStatus Destroy();

	//xn::ModuleRecorder implementation
	virtual XnStatus SetOutputStream(void* pStreamToken, XnRecorderOutputStreamInterface* pStream);
	virtual XnStatus OnNodeAdded(const XnChar* strNodeName, XnProductionNodeType type, XnCodecID compression);
	virtual XnStatus OnNodeRemoved(const XnChar* strNodeName);
	virtual XnStatus OnNodeIntPropChanged(const XnChar* strNodeName, const XnChar* strPropName, XnUInt64 nValue);
	virtual XnStatus OnNodeRealPropChanged(const XnChar* strNodeName, const XnChar* strPropName, XnDouble dValue);
	virtual XnStatus OnNodeStringPropChanged(const XnChar* strNodeName, const XnChar* strPropName, const XnChar* strValue);
	virtual XnStatus OnNodeStateReady(const XnChar* strNodeName);
	virtual XnStatus OnNodeGeneralPropChanged(const XnChar* strNodeName, const XnChar* strPropName, XnUInt32 nBufferSize, const void* pBuffer);
	virtual XnStatus OnNodeNewData(const XnChar* strNodeName, XnUInt64 nTimeStamp, XnUInt32 nFrame, const void* pData, XnUInt32 nSize);

private:
	struct RecordedNodePropInfo
	{
		RecordedNodePropInfo() { Reset(); }
		void Reset()
		{
			nPos = 0;
		}

		XnUInt64 nPos;		//Position in stream of record that did this property change
	};

	typedef XnStringsHashT<RecordedNodePropInfo> RecordedNodePropInfoMap;
	typedef XnListT<DataIndexEntry> DataIndexEntryList;

	struct RecordedNodeInfo
	{
		RecordedNodeInfo();
		void Reset();

		XnUInt32 nNodeID;
		XnProductionNodeType type;
		XnUInt64 nNodeAddedPos;
		XnUInt32 nMaxFrameNum;
		XnUInt32 nCurFrameNum;
		XnUInt64 nMinTimeStamp;
		XnUInt64 nMaxTimeStamp;
		XnBool bGotData;
		XnCodecID compression;
		xn::Codec codec;
		RecordedNodePropInfoMap propInfoMap;
		DataIndexEntryList dataIndex;
	};

	typedef XnStringsHashT<RecordedNodeInfo> RecordedNodesInfo;

	XnStatus OpenStream();
	XnStatus WriteHeader(XnUInt64 nGlobalMaxTimeStamp, XnUInt32 nMaxNodeID);
	XnStatus WriteToStream(const XnChar* strNodeName, const void* pData, XnUInt32 nSize);
	XnStatus WriteRecordToStream(const XnChar* strNodeName, Record &record);
	XnStatus SeekStream(XnOSSeekType seekType, XnUInt64 nOffset);
	XnUInt64 TellStream();
	XnStatus FinalizeStream();
	XnStatus CloseStream();
	XnStatus WriteNodeDataBegin(const XnChar* strNodeName);
	XnStatus UpdateNodeSeekInfo(const XnChar* strNodeName, const RecordedNodeInfo& recordedNodeInfo);
	XnStatus RemoveNode(const XnChar* strNodeName);
	
	//UpdateNodePropInfo() returns, in nUndoPos, the position in the file you should read to undo the property update.
	XnStatus UpdateNodePropInfo(const XnChar* strNodeName, const XnChar* strPropName, RecordedNodeInfo*& pRecordedNodeInfo, XnUInt64& nUndoPos);
	RecordedNodeInfo* GetRecordedNodeInfo(const XnChar* strNodeName);

	static const XnUInt32 RECORD_MAX_SIZE;
	static const XnUInt32 PAYLOAD_DATA_SIZE;
	XnBool m_bOpen;
	XnUInt8* m_pRecordBuffer;
	XnUInt8* m_pPayloadData;
	void* m_pStreamCookie;
	XnRecorderOutputStreamInterface* m_pOutputStream;

	RecordedNodesInfo m_recordedNodesInfo;
	xn::Context m_context;
	XnUInt64 m_nGlobalStartTimeStamp;
	XnUInt64 m_nGlobalMaxTimeStamp;
	XnUInt32 m_nNumNodes;
	XnUInt32 m_nConfigurationID;
};

#endif //__RECORDER_NODE_H__