Skip to content

O
opencv
Commit c1a6cb62 authored by Maria Dimashova's avatar Maria Dimashova
Browse files
Options

added oni-files reading

parent 82a2a50f
Hide whitespace changes
Inline Side-by-side
Showing with 180 additions and 117 deletions
modules/highgui/src/cap_openni.cpp
View file @ c1a6cb62
...@@ -124,6 +124,8 @@ protected: ...@@ -124,6 +124,8 @@ protected:
xn::Context context; xn::Context context;
bool isContextOpened; bool isContextOpened;
xn::ProductionNode productionNode;
// Data generators with its metadata // Data generators with its metadata
xn::DepthGenerator depthGenerator; xn::DepthGenerator depthGenerator;
xn::DepthMetaData depthMetaData; xn::DepthMetaData depthMetaData;
...@@ -171,118 +173,158 @@ XnMapOutputMode defaultMapOutputMode() ...@@ -171,118 +173,158 @@ XnMapOutputMode defaultMapOutputMode()
CvCapture_OpenNI::CvCapture_OpenNI( int index ) CvCapture_OpenNI::CvCapture_OpenNI( int index )
{ {
XnStatus status;
isContextOpened = false; isContextOpened = false;
// Initialize and configure the context. // Initialize and configure the context.
if( context.Init() == XN_STATUS_OK ) status = context.Init();
if( status != XN_STATUS_OK )
{ {
// Find devices std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to initialize the context: "
xn::NodeInfoList devicesList; << std::string(xnGetStatusString(status)) << std::endl;
XnStatus status = context.EnumerateProductionTrees( XN_NODE_TYPE_DEVICE, NULL, devicesList, 0 ); return;
if( status != XN_STATUS_OK ) }
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to enumerate production trees: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
// Chose device according to index
xn::NodeInfoList::Iterator it = devicesList.Begin();
for( int i = 0; i < index; ++i ) it++;
xn::NodeInfo deviceNode = *it; // Find devices
status = context.CreateProductionTree( deviceNode ); xn::NodeInfoList devicesList;
if( status != XN_STATUS_OK ) status = context.EnumerateProductionTrees( XN_NODE_TYPE_DEVICE, NULL, devicesList, 0 );
{ if( status != XN_STATUS_OK )
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to create production tree: " {
<< std::string(xnGetStatusString(status)) << std::endl; std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to enumerate production trees: "
return; << std::string(xnGetStatusString(status)) << std::endl;
} return;
}
status = context.RunXmlScript( XMLConfig.c_str() ); // Chose device according to index
if( status != XN_STATUS_OK ) xn::NodeInfoList::Iterator it = devicesList.Begin();
{ for( int i = 0; i < index; ++i ) it++;
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to run xml script: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
// Associate generators with context. xn::NodeInfo deviceNode = *it;
// enumerate the nodes to find if depth generator is present status = context.CreateProductionTree( deviceNode, productionNode );
xn::NodeInfoList depthList; if( status != XN_STATUS_OK )
status = context.EnumerateExistingNodes( depthList, XN_NODE_TYPE_DEPTH ); {
if( status != XN_STATUS_OK ) std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to create production tree: "
{ << std::string(xnGetStatusString(status)) << std::endl;
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to enumerate depth generators: " return;
<< std::string(xnGetStatusString(status)) << std::endl; }
return;
}
if( depthList.IsEmpty() )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : The device doesn't have depth generator. Such devices aren't supported now." << std::endl;
return;
}
status = depthGenerator.Create( context );
if( status != XN_STATUS_OK )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to create depth generator: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
// enumerate the nodes to find if image generator is present status = context.RunXmlScript( XMLConfig.c_str() );
xn::NodeInfoList imageList; if( status != XN_STATUS_OK )
status = context.EnumerateExistingNodes( imageList, XN_NODE_TYPE_IMAGE ); {
if( status != XN_STATUS_OK ) std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to run xml script: "
{ << std::string(xnGetStatusString(status)) << std::endl;
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to enumerate image generators: " return;
<< std::string(xnGetStatusString(status)) << std::endl; }
return;
}
if( !imageList.IsEmpty() ) // Associate generators with context.
{ // enumerate the nodes to find if depth generator is present
status = imageGenerator.Create( context ); xn::NodeInfoList depthList;
if( status != XN_STATUS_OK ) status = context.EnumerateExistingNodes( depthList, XN_NODE_TYPE_DEPTH );
{ if( status != XN_STATUS_OK )
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to create image generator: " {
<< std::string(xnGetStatusString(status)) << std::endl; std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to enumerate depth generators: "
return; << std::string(xnGetStatusString(status)) << std::endl;
} return;
} }
if( depthList.IsEmpty() )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : The device doesn't have depth generator. Such devices aren't supported now." << std::endl;
return;
}
status = depthGenerator.Create( context );
if( status != XN_STATUS_OK )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to create depth generator: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
// Set map output mode. // enumerate the nodes to find if image generator is present
if( depthGenerator.IsValid() ) xn::NodeInfoList imageList;
CV_DbgAssert( depthGenerator.SetMapOutputMode(defaultMapOutputMode()) == XN_STATUS_OK ); // xn::DepthGenerator supports VGA only! (Jan 2011) status = context.EnumerateExistingNodes( imageList, XN_NODE_TYPE_IMAGE );
if( imageGenerator.IsValid() ) if( status != XN_STATUS_OK )
CV_DbgAssert( imageGenerator.SetMapOutputMode(defaultMapOutputMode()) == XN_STATUS_OK ); {
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to enumerate image generators: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
// Start generating data. if( !imageList.IsEmpty() )
status = context.StartGeneratingAll(); {
status = imageGenerator.Create( context );
if( status != XN_STATUS_OK ) if( status != XN_STATUS_OK )
{ {
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to start generating OpenNI data: " std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to create image generator: "
<< std::string(xnGetStatusString(status)) << std::endl; << std::string(xnGetStatusString(status)) << std::endl;
return; return;
} }
}
if( !readCamerasParams() ) // Set map output mode.
{ if( depthGenerator.IsValid() )
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Could not read cameras parameters" << std::endl; CV_DbgAssert( depthGenerator.SetMapOutputMode(defaultMapOutputMode()) == XN_STATUS_OK ); // xn::DepthGenerator supports VGA only! (Jan 2011)
return; if( imageGenerator.IsValid() )
} CV_DbgAssert( imageGenerator.SetMapOutputMode(defaultMapOutputMode()) == XN_STATUS_OK );
outputMaps.resize( outputMapsTypesCount ); // Start generating data.
status = context.StartGeneratingAll();
if( status != XN_STATUS_OK )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to start generating OpenNI data: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
isContextOpened = true; if( !readCamerasParams() )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Could not read cameras parameters" << std::endl;
return;
} }
outputMaps.resize( outputMapsTypesCount );
isContextOpened = true;
setProperty(CV_CAP_PROP_OPENNI_REGISTRATION, 1.0); setProperty(CV_CAP_PROP_OPENNI_REGISTRATION, 1.0);
} }
CvCapture_OpenNI::CvCapture_OpenNI(const char * filename) CvCapture_OpenNI::CvCapture_OpenNI(const char * filename)
{ {
CV_Assert(0); XnStatus status;
isContextOpened = false;
// Initialize and configure the context.
status = context.Init();
if( status != XN_STATUS_OK )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to initialize the context: "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
// Open file
status = context.OpenFileRecording( filename, productionNode );
if( status != XN_STATUS_OK )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Failed to open input file (" << filename << "): "
<< std::string(xnGetStatusString(status)) << std::endl;
return;
}
std::cout << context.FindExistingNode( XN_NODE_TYPE_DEPTH, depthGenerator ) << std::endl;
std::cout << context.FindExistingNode( XN_NODE_TYPE_IMAGE, imageGenerator ) << std::endl;
if( !readCamerasParams() )
{
std::cerr << "CvCapture_OpenNI::CvCapture_OpenNI : Could not read cameras parameters" << std::endl;
return;
}
outputMaps.resize( outputMapsTypesCount );
isContextOpened = true;
} }
CvCapture_OpenNI::~CvCapture_OpenNI() CvCapture_OpenNI::~CvCapture_OpenNI()
...@@ -424,6 +466,7 @@ double CvCapture_OpenNI::getDepthGeneratorProperty( int propIdx ) ...@@ -424,6 +466,7 @@ double CvCapture_OpenNI::getDepthGeneratorProperty( int propIdx )
break; break;
case CV_CAP_PROP_OPENNI_REGISTRATION : case CV_CAP_PROP_OPENNI_REGISTRATION :
propValue = depthGenerator.GetAlternativeViewPointCap().IsViewPointAs(imageGenerator) ? 1.0 : 0.0; propValue = depthGenerator.GetAlternativeViewPointCap().IsViewPointAs(imageGenerator) ? 1.0 : 0.0;
break;
default : default :
{ {
std::stringstream ss; std::stringstream ss;
...@@ -731,15 +774,14 @@ IplImage* CvCapture_OpenNI::retrieveValidDepthMask() ...@@ -731,15 +774,14 @@ IplImage* CvCapture_OpenNI::retrieveValidDepthMask()
inline void getBGRImageFromMetaData( const xn::ImageMetaData& imageMetaData, cv::Mat& bgrImage ) inline void getBGRImageFromMetaData( const xn::ImageMetaData& imageMetaData, cv::Mat& bgrImage )
{ {
int cols = imageMetaData.XRes(); if( imageMetaData.PixelFormat() != XN_PIXEL_FORMAT_RGB24 )
int rows = imageMetaData.YRes(); CV_Error( CV_StsUnsupportedFormat, "Unsupported format of grabbed image\n" );
cv::Mat rgbImage( rows, cols, CV_8UC3 );
cv::Mat rgbImage( imageMetaData.YRes(), imageMetaData.XRes(), CV_8UC3 );
const XnRGB24Pixel* pRgbImage = imageMetaData.RGB24Data(); const XnRGB24Pixel* pRgbImage = imageMetaData.RGB24Data();
// CV_Assert( 3*sizeof(uchar) == sizeof(XnRGB24Pixel) ); // CV_Assert( 3*sizeof(uchar) == sizeof(XnRGB24Pixel) );
memcpy( rgbImage.data, pRgbImage, cols*rows*sizeof(XnRGB24Pixel) ); memcpy( rgbImage.data, pRgbImage, rgbImage.total()*sizeof(XnRGB24Pixel) );
cv::cvtColor( rgbImage, bgrImage, CV_RGB2BGR ); cv::cvtColor( rgbImage, bgrImage, CV_RGB2BGR );
} }
......
samples/cpp/openni_capture.cpp
View file @ c1a6cb62
...@@ -93,9 +93,11 @@ void printCommandLineParams() ...@@ -93,9 +93,11 @@ void printCommandLineParams()
cout << "-m Mask to set which output images are need. It is a string of size 5. Each element of this is '0' or '1' and" << endl; cout << "-m Mask to set which output images are need. It is a string of size 5. Each element of this is '0' or '1' and" << endl;
cout << " determine: is depth map, disparity map, valid pixels mask, rgb image, gray image need or not (correspondently)?" << endl ; cout << " determine: is depth map, disparity map, valid pixels mask, rgb image, gray image need or not (correspondently)?" << endl ;
cout << " By default -m 01010 i.e. disparity map and rgb image will be shown." << endl ; cout << " By default -m 01010 i.e. disparity map and rgb image will be shown." << endl ;
cout << "-r Filename of .oni video file. The data will grabbed from it." << endl ;
} }
void parseCommandLine( int argc, char* argv[], bool& isColorizeDisp, bool& isFixedMaxDisp, int& imageMode, bool retrievedImageFlags[] ) void parseCommandLine( int argc, char* argv[], bool& isColorizeDisp, bool& isFixedMaxDisp, int& imageMode, bool retrievedImageFlags[],
string& filename, bool& isFileReading )
{ {
// set defaut values // set defaut values
isColorizeDisp = true; isColorizeDisp = true;
...@@ -108,6 +110,9 @@ void parseCommandLine( int argc, char* argv[], bool& isColorizeDisp, bool& isFix ...@@ -108,6 +110,9 @@ void parseCommandLine( int argc, char* argv[], bool& isColorizeDisp, bool& isFix
retrievedImageFlags[3] = true; retrievedImageFlags[3] = true;
retrievedImageFlags[4] = false; retrievedImageFlags[4] = false;
filename.clear();
isFileReading = false;
if( argc == 1 ) if( argc == 1 )
{ {
help(); help();
...@@ -154,6 +159,11 @@ void parseCommandLine( int argc, char* argv[], bool& isColorizeDisp, bool& isFix ...@@ -154,6 +159,11 @@ void parseCommandLine( int argc, char* argv[], bool& isColorizeDisp, bool& isFix
exit(0); exit(0);
} }
} }
else if( !strcmp( argv[i], "-r" ) )
{
filename = argv[++i];
isFileReading = true;
}
else else
{ {
cout << "Unsupported command line argument: " << argv[i] << "." << endl; cout << "Unsupported command line argument: " << argv[i] << "." << endl;
...@@ -172,10 +182,17 @@ int main( int argc, char* argv[] ) ...@@ -172,10 +182,17 @@ int main( int argc, char* argv[] )
bool isColorizeDisp, isFixedMaxDisp; bool isColorizeDisp, isFixedMaxDisp;
int imageMode; int imageMode;
bool retrievedImageFlags[5]; bool retrievedImageFlags[5];
parseCommandLine( argc, argv, isColorizeDisp, isFixedMaxDisp, imageMode, retrievedImageFlags ); string filename;
bool isVideoReading;
parseCommandLine( argc, argv, isColorizeDisp, isFixedMaxDisp, imageMode, retrievedImageFlags, filename, isVideoReading );
cout << "Device opening ..." << endl; cout << "Device opening ..." << endl;
VideoCapture capture( CV_CAP_OPENNI ); VideoCapture capture;
if( isVideoReading )
capture.open( filename );
else
capture.open( CV_CAP_OPENNI );
cout << "done." << endl; cout << "done." << endl;
if( !capture.isOpened() ) if( !capture.isOpened() )
...@@ -184,37 +201,41 @@ int main( int argc, char* argv[] ) ...@@ -184,37 +201,41 @@ int main( int argc, char* argv[] )
return -1; return -1;
} }
bool modeRes=false; if( !isVideoReading )
switch ( imageMode )
{ {
case 0: bool modeRes=false;
modeRes = capture.set( CV_CAP_OPENNI_IMAGE_GENERATOR_OUTPUT_MODE, CV_CAP_OPENNI_VGA_30HZ ); switch ( imageMode )
break; {
case 1: case 0:
modeRes = capture.set( CV_CAP_OPENNI_IMAGE_GENERATOR_OUTPUT_MODE, CV_CAP_OPENNI_SXGA_15HZ ); modeRes = capture.set( CV_CAP_OPENNI_IMAGE_GENERATOR_OUTPUT_MODE, CV_CAP_OPENNI_VGA_30HZ );
break; break;
case 2: case 1:
modeRes = capture.set( CV_CAP_OPENNI_IMAGE_GENERATOR_OUTPUT_MODE, CV_CAP_OPENNI_SXGA_30HZ ); modeRes = capture.set( CV_CAP_OPENNI_IMAGE_GENERATOR_OUTPUT_MODE, CV_CAP_OPENNI_SXGA_15HZ );
break; break;
default: case 2:
CV_Error( CV_StsBadArg, "Unsupported image mode property.\n"); modeRes = capture.set( CV_CAP_OPENNI_IMAGE_GENERATOR_OUTPUT_MODE, CV_CAP_OPENNI_SXGA_30HZ );
break;
default:
CV_Error( CV_StsBadArg, "Unsupported image mode property.\n");
}
if (!modeRes)
cout << "\nThis image mode is not supported by the device, the default value (CV_CAP_OPENNI_SXGA_15HZ) will be used.\n" << endl;
} }
if (!modeRes)
cout << "\nThis image mode is not supported by the device, the default value (CV_CAP_OPENNI_SXGA_15HZ) will be used.\n" << endl;
// Print some avalible device settings. // Print some avalible device settings.
cout << "\nDepth generator output mode:" << endl << cout << "\nDepth generator output mode:" << endl <<
"FRAME_WIDTH " << capture.get( CV_CAP_PROP_FRAME_WIDTH ) << endl << "FRAME_WIDTH " << capture.get( CV_CAP_PROP_FRAME_WIDTH ) << endl <<
"FRAME_HEIGHT " << capture.get( CV_CAP_PROP_FRAME_HEIGHT ) << endl << "FRAME_HEIGHT " << capture.get( CV_CAP_PROP_FRAME_HEIGHT ) << endl <<
"FRAME_MAX_DEPTH " << capture.get( CV_CAP_PROP_OPENNI_FRAME_MAX_DEPTH ) << " mm" << endl << "FRAME_MAX_DEPTH " << capture.get( CV_CAP_PROP_OPENNI_FRAME_MAX_DEPTH ) << " mm" << endl <<
"FPS " << capture.get( CV_CAP_PROP_FPS ) << endl; "FPS " << capture.get( CV_CAP_PROP_FPS ) << endl <<
"REGISTRATION " << capture.get( CV_CAP_PROP_OPENNI_REGISTRATION ) << endl;
if( capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR_PRESENT ) ) if( capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR_PRESENT ) )
{ {
cout << cout <<
"\nImage generator output mode:" << endl << "\nImage generator output mode:" << endl <<
"FRAME_WIDTH " << capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR+CV_CAP_PROP_FRAME_WIDTH ) << endl << "FRAME_WIDTH " << capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR+CV_CAP_PROP_FRAME_WIDTH ) << endl <<
"FRAME_HEIGHT " << capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR+CV_CAP_PROP_FRAME_HEIGHT ) << endl << "FRAME_HEIGHT " << capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR+CV_CAP_PROP_FRAME_HEIGHT ) << endl <<
"FPS " << capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR+CV_CAP_PROP_FPS ) << endl; "FPS " << capture.get( CV_CAP_OPENNI_IMAGE_GENERATOR+CV_CAP_PROP_FPS ) << endl;
} }
else else
{ {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment