forked from avin2/SensorKinect
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathXnStreamDataSet.cpp
More file actions
156 lines (126 loc) · 5.39 KB
/
XnStreamDataSet.cpp
File metadata and controls
156 lines (126 loc) · 5.39 KB
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
/****************************************************************************
* *
* PrimeSense Sensor 5.x Alpha *
* Copyright (C) 2011 PrimeSense Ltd. *
* *
* This file is part of PrimeSense Sensor. *
* *
* PrimeSense Sensor 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. *
* *
* PrimeSense Sensor 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 PrimeSense Sensor. If not, see <http://www.gnu.org/licenses/>.*
* *
****************************************************************************/
//---------------------------------------------------------------------------
// Includes
//---------------------------------------------------------------------------
#include "XnStreamDataSetInternal.h"
#include <XnOS.h>
#include <XnDDKStatus.h>
//---------------------------------------------------------------------------
// Code
//---------------------------------------------------------------------------
XN_DDK_API XnStatus XnStreamDataSetCreate(XnStreamDataSet** ppStreamOutputSet)
{
XN_VALIDATE_OUTPUT_PTR(ppStreamOutputSet);
// allocate struct
XN_VALIDATE_CALLOC(*ppStreamOutputSet, XnStreamDataSet, 1);
XnStreamDataSet* pSet = (*ppStreamOutputSet);
// allocate hash table
pSet->pHash = XN_NEW(XnStreamDataHash);
if (pSet->pHash == NULL)
{
XnStreamDataSetDestroy(ppStreamOutputSet);
return XN_STATUS_ALLOC_FAILED;
}
return XN_STATUS_OK;
}
XN_DDK_API XnStatus XnStreamDataSetDestroy(XnStreamDataSet** ppStreamOutputSet)
{
XN_VALIDATE_INPUT_PTR(ppStreamOutputSet);
XnStreamDataSet* pSet = (*ppStreamOutputSet);
if (pSet != NULL)
{
for (XnStreamDataHash::Iterator it = pSet->pHash->Begin(); it != pSet->pHash->End(); ++it)
{
XnStreamData* pStreamData = it->Value();
XnStreamDataDestroy(&pStreamData);
}
// free hash table
XN_DELETE(pSet->pHash);
// free struct
XN_FREE_AND_NULL(*ppStreamOutputSet);
}
return XN_STATUS_OK;
}
XN_DDK_API XnStatus XnStreamDataSetAdd(XnStreamDataSet* pStreamOutputSet, XnStreamData* pStreamOutput)
{
XN_VALIDATE_INPUT_PTR(pStreamOutputSet);
XN_VALIDATE_INPUT_PTR(pStreamOutput);
// make sure another object of this stream is not already in set
XnStreamDataHash::Iterator it = pStreamOutputSet->pHash->End();
if (XN_STATUS_NO_MATCH != pStreamOutputSet->pHash->Find(pStreamOutput->StreamName, it))
return XN_STATUS_STREAM_OUTPUT_SET_ALREADY_IN_SET;
return pStreamOutputSet->pHash->Set(pStreamOutput->StreamName, pStreamOutput);
}
XN_DDK_API XnStatus XnStreamDataSetRemove(XnStreamDataSet* pStreamOutputSet, XnStreamData* pStreamOutput)
{
XN_VALIDATE_INPUT_PTR(pStreamOutputSet);
XN_VALIDATE_INPUT_PTR(pStreamOutput);
for (XnStreamDataHash::Iterator it = pStreamOutputSet->pHash->Begin(); it != pStreamOutputSet->pHash->End(); ++it)
{
if (pStreamOutput == it->Value())
{
pStreamOutputSet->pHash->Remove(it);
break;
}
}
return XN_STATUS_OK;
}
XN_DDK_API XnStatus XnStreamDataSetRemoveByName(XnStreamDataSet* pStreamOutputSet, const XnChar* StreamName)
{
XN_VALIDATE_INPUT_PTR(pStreamOutputSet);
XN_VALIDATE_INPUT_PTR(StreamName);
pStreamOutputSet->pHash->Remove(StreamName);
return XN_STATUS_OK;
}
XN_DDK_API XnStatus XnStreamDataSetGet(XnStreamDataSet* pStreamOutputSet, const XnChar* StreamName, XnStreamData** ppStreamOutput)
{
XN_VALIDATE_INPUT_PTR(pStreamOutputSet);
XN_VALIDATE_INPUT_PTR(StreamName);
XN_VALIDATE_OUTPUT_PTR(ppStreamOutput);
XnStreamData* pData;
XnStatus nRetVal = pStreamOutputSet->pHash->Get(StreamName, pData);
XN_IS_STATUS_OK(nRetVal);
*ppStreamOutput = pData;
return XN_STATUS_OK;
}
XN_DDK_API XnStatus XnStreamDataSetCopyToArray(const XnStreamDataSet* pStreamOutputSet, XnStreamData** apStreamOutputs, XnUInt32* pnCount)
{
XN_VALIDATE_INPUT_PTR(pStreamOutputSet);
XN_VALIDATE_INPUT_PTR(pnCount);
// first check if we have enough space
XnUInt32 nCount = pStreamOutputSet->pHash->Size();
XnUInt32 nArraySize = *pnCount;
*pnCount = nCount;
if (nArraySize < nCount)
{
return (XN_STATUS_OUTPUT_BUFFER_OVERFLOW);
}
// now copy
XnUInt32 nIndex = 0;
for (XnStreamDataHash::Iterator it = pStreamOutputSet->pHash->Begin(); it != pStreamOutputSet->pHash->End(); ++it)
{
apStreamOutputs[nIndex] = it->Value();
nIndex++;
}
return (XN_STATUS_OK);
}