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
157
158
159
|
/*****************************************************************************
* *
* OpenNI 2.x Alpha *
* Copyright (C) 2012 PrimeSense Ltd. *
* *
* This file is part of OpenNI. *
* *
* Licensed under the Apache License, Version 2.0 (the "License"); *
* you may not use this file except in compliance with the License. *
* You may obtain a copy of the License at *
* *
* http://www.apache.org/licenses/LICENSE-2.0 *
* *
* Unless required by applicable law or agreed to in writing, software *
* distributed under the License is distributed on an "AS IS" BASIS, *
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
* See the License for the specific language governing permissions and *
* limitations under the License. *
* *
*****************************************************************************/
/// @file
/// Contains the definition of Device class that implements a virtual OpenNI
/// device, capable of reading data from a *.ONI file.
#include "PlayerSource.h"
namespace oni_file {
/// Constructor.
PlayerSource::PlayerSource(const XnChar* strNodeName, OniSensorType sensorType) :
m_nodeName(strNodeName),
m_requiredFrameSize(0)
{
m_sourceInfo.sensorType = sensorType;
m_sourceInfo.numSupportedVideoModes = 0;
}
/// Destructor.
PlayerSource::~PlayerSource()
{
}
/// Return the source info associated with the source.
OniSensorInfo* PlayerSource::GetInfo()
{
return &m_sourceInfo;
}
const XnChar* PlayerSource::GetNodeName()
{
return m_nodeName.Data();
}
/// Get property.
OniStatus PlayerSource::GetProperty(int propertyId, void* data, int* pDataSize)
{
return m_properties.GetProperty(propertyId, data, pDataSize);
}
/// Set property.
OniStatus PlayerSource::SetProperty(int propertyId, const void* data, int dataSize)
{
if (propertyId == ONI_STREAM_PROPERTY_VIDEO_MODE)
{
if(m_sourceInfo.numSupportedVideoModes == 0)
{
// at first, we're gonna allocate and set.
// next time we'll just update the values.
m_sourceInfo.numSupportedVideoModes = 1;
m_sourceInfo.pSupportedVideoModes = XN_NEW(OniVideoMode);
}
memcpy(m_sourceInfo.pSupportedVideoModes, data, sizeof(OniVideoMode));
// Calculate bytes-per-pixel from format and number of maps per frame.
OniVideoMode* pVideoMode = (OniVideoMode*)data;
int stride;
int bytesPerPixel;
switch (pVideoMode->pixelFormat)
{
case ONI_PIXEL_FORMAT_GRAY8:
{
bytesPerPixel = 1;
stride = pVideoMode->resolutionX * bytesPerPixel;
break;
}
case ONI_PIXEL_FORMAT_DEPTH_1_MM:
case ONI_PIXEL_FORMAT_DEPTH_100_UM:
case ONI_PIXEL_FORMAT_SHIFT_9_2:
case ONI_PIXEL_FORMAT_SHIFT_9_3:
{
bytesPerPixel = 2;
stride = pVideoMode->resolutionX * bytesPerPixel;
break;
}
case ONI_PIXEL_FORMAT_GRAY16:
case ONI_PIXEL_FORMAT_YUV422:
case ONI_PIXEL_FORMAT_YUYV:
{
bytesPerPixel = 2;
stride = pVideoMode->resolutionX * bytesPerPixel;
break;
}
case ONI_PIXEL_FORMAT_RGB888:
{
bytesPerPixel = 3;
stride = pVideoMode->resolutionX * bytesPerPixel;
break;
}
case ONI_PIXEL_FORMAT_JPEG:
{
bytesPerPixel = 0;
stride = 0;
break;
}
default:
{
bytesPerPixel = 0;
stride = 0;
}
}
m_properties.SetProperty(ONI_STREAM_PROPERTY_BYTES_PER_PIXEL, &bytesPerPixel, sizeof(bytesPerPixel));
m_properties.SetProperty(ONI_STREAM_PROPERTY_STRIDE, &stride, sizeof(stride));
}
return m_properties.SetProperty(propertyId, data, dataSize);
}
// Process new data.
void PlayerSource::ProcessNewData(XnUInt64 nTimeStamp, XnUInt32 nFrameId, void* pData, XnUInt32 nSize)
{
// Raise the event to all registered callbacks.
NewDataEventArgs args;
args.nTimeStamp = nTimeStamp;
args.nFrameId = nFrameId;
args.pData = pData;
args.nSize = nSize;
m_newDataEvent.Raise(args);
}
// Register for new data event.
OniStatus PlayerSource::RegisterNewDataEvent(NewDataCallback callback, void* pCookie, OniCallbackHandle& handle)
{
XnStatus rc = m_newDataEvent.Register(callback, pCookie, (XnCallbackHandle&)handle);
if (rc != XN_STATUS_OK)
{
return ONI_STATUS_ERROR;
}
return ONI_STATUS_OK;
}
// Unregister from new data event.
void PlayerSource::UnregisterNewDataEvent(OniCallbackHandle handle)
{
m_newDataEvent.Unregister((XnCallbackHandle)handle);
}
} // namespace oni_files_player
|
