コード例 #1
0
ファイル: Recorder.cpp プロジェクト: DZW314/NiWrapper.Net
	__declspec(dllexport) Status Recorder_create(Recorder*& rec, const char* fileName)
	{
		 Recorder* vsl = new Recorder();
		 Status status = vsl->create(fileName);
		 if (status == STATUS_OK)
			 rec = vsl;
		 else
			 Recorder_destroy(vsl);
		 return status;
	}
コード例 #2
0
ファイル: main.cpp プロジェクト: VIML/oniFixer
int main( int argc, char** argv )
{
	std::string	sSource, sTarget;
	#pragma region check input parameters
	if( argc == 3 )
	{
		sSource = argv[1];
		sTarget = argv[2];
	}
	else
	{
		cerr << "Usage: oniFixer source.oni target.oni\n" << endl;
	}
	#pragma endregion

	// Initial OpenNI 
	if( OpenNI::initialize() != STATUS_OK )
	{
		cerr << "OpenNI Initial Error: " << OpenNI::getExtendedError() << endl;
		return -1;
	}

	CNiTEProperty	mNiTEProp;

	#pragma region Open and check source file
	CNIDevice	mOniFile;
	if( !mOniFile.OpenDevice( sSource.c_str() ) )
	{
		cerr << "Can't open ONI file" << endl;
		return -1;
	}

	// create Playback controller
	PlaybackControl* pPlay = mOniFile.mDeivce.getPlaybackControl();
	int	iDepthFrameNum, iColorFrameNum;
	if( pPlay != nullptr )
	{
		pPlay->setRepeatEnabled( false );
		pPlay->setSpeed( 0.0f );

		iDepthFrameNum = pPlay->getNumberOfFrames( mOniFile.vsDepth );
		iColorFrameNum = pPlay->getNumberOfFrames( mOniFile.vsColor );
		cout << "This oni file have: " << iDepthFrameNum << " frames dpeth, " << iColorFrameNum << " frames color" << endl;
	}
	else
	{
		cerr << "Can't get playback controller, this may not a vaild oni file?" << endl;
		return -1;
	}

	// check NiTE required property
	if( mNiTEProp.LoadProperties( mOniFile.vsDepth ) )
	{
		cout << "All required properties works, don't need to fix" << endl;
		return 0;
	}
	#pragma endregion

	#pragma region Create a physical device to read property
	cout << "\nOpen a physical device to read property" << endl;
	CNIDevice mPhysical;
	if( !mPhysical.OpenDevice( ANY_DEVICE, true ) )
	{
		cerr << "Can't open physical device: " << OpenNI::getExtendedError() << endl;
		return -1;
	}

	// try to load property
	if( !mNiTEProp.LoadProperties( mPhysical.vsDepth ) )
	{
		cout << "Can't prepare all required properties" << endl;
		return -1;
	}
	mPhysical.Close();
	#pragma endregion
	
	#pragma region Create a virtual device to record
	// create recorder
	Recorder mRecorder;
	mRecorder.create( sTarget.c_str() );

	// create virtual device
	cout << "\nCreate a virtual device to record" << endl;
	CNIDevice mVirtual;
	if( !mVirtual.OpenDevice( "\\OpenNI2\\VirtualDevice\\PS1080", !(mOniFile.vsColor.isValid()) ) )
	{
		cerr << "Can't create virtual device" << endl;
		return -1;
	}

	// copy depth stream properties
	CopyGeneralProperties( mOniFile.vsDepth, mVirtual.vsDepth );
	mNiTEProp.WriteProperties( mVirtual.vsDepth );
	mOniFile.vsDepth.addNewFrameListener( new CFrameCopy( mVirtual.vsDepth ) );
	mRecorder.attach( mVirtual.vsDepth );

	// copy color stream properties
	if( mVirtual.vsColor.isValid() )
	{
		CopyGeneralProperties( mOniFile.vsColor, mVirtual.vsColor );
		mOniFile.vsColor.addNewFrameListener( new CFrameCopy( mVirtual.vsColor ) );
		mRecorder.attach( mVirtual.vsColor );
	}
	#pragma endregion

	// start
	cout << "Start to re-recorder\n";
	mRecorder.start();
	mVirtual.Start();
	mOniFile.Start();

	int iDepth = 0;
	while( true )
	{
		// just seek by depth videostream
		pPlay->seek( mOniFile.vsDepth, ++iDepth );

		cout << "." << flush;
		if( iDepth >= iDepthFrameNum )
			break;
	}
	cout << "Done" << endl;
	
	// stop
	mRecorder.stop();
	mOniFile.Stop();
	mVirtual.Stop();

	mOniFile.Close();
	mVirtual.Close();

	OpenNI::shutdown();
}