2016-02-15 19:35:24 -05:00
/*
Copyright ( c ) 2010 - 2016 , Mathieu Labbe - IntRoLab - Universite de Sherbrooke
All rights reserved .
Redistribution and use in source and binary forms , with or without
modification , are permitted provided that the following conditions are met :
* Redistributions of source code must retain the above copyright
notice , this list of conditions and the following disclaimer .
* Redistributions in binary form must reproduce the above copyright
notice , this list of conditions and the following disclaimer in the
documentation and / or other materials provided with the distribution .
* Neither the name of the Universite de Sherbrooke nor the
names of its contributors may be used to endorse or promote products
derived from this software without specific prior written permission .
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS " AS IS " AND
ANY EXPRESS OR IMPLIED WARRANTIES , INCLUDING , BUT NOT LIMITED TO , THE IMPLIED
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED . IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY
DIRECT , INDIRECT , INCIDENTAL , SPECIAL , EXEMPLARY , OR CONSEQUENTIAL DAMAGES
( INCLUDING , BUT NOT LIMITED TO , PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES ;
LOSS OF USE , DATA , OR PROFITS ; OR BUSINESS INTERRUPTION ) HOWEVER CAUSED AND
ON ANY THEORY OF LIABILITY , WHETHER IN CONTRACT , STRICT LIABILITY , OR TORT
( INCLUDING NEGLIGENCE OR OTHERWISE ) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE , EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE .
*/
# include <tango-gl/conversions.h>
# include "RTABMapApp.h"
# include <rtabmap/core/Rtabmap.h>
2016-11-04 16:39:52 -04:00
# include <rtabmap/core/util2d.h>
2016-02-15 19:35:24 -05:00
# include <rtabmap/core/util3d.h>
# include <rtabmap/core/util3d_transforms.h>
# include <rtabmap/core/util3d_filtering.h>
# include <rtabmap/core/Graph.h>
# include <rtabmap/utilite/UEventsManager.h>
# include <rtabmap/utilite/UStl.h>
2016-03-21 20:03:03 -04:00
# include <rtabmap/utilite/UDirectory.h>
# include <rtabmap/utilite/UFile.h>
2016-02-15 19:35:24 -05:00
# include <opencv2/opencv_modules.hpp>
# include <rtabmap/core/util3d_surface.h>
# include <rtabmap/utilite/UConversion.h>
# include <rtabmap/utilite/UTimer.h>
# include <rtabmap/core/ParamEvent.h>
2016-03-21 20:03:03 -04:00
# include <rtabmap/core/Compression.h>
2016-04-03 21:35:30 -04:00
# include <rtabmap/core/Optimizer.h>
2016-04-09 18:57:49 -04:00
# include <rtabmap/core/VWDictionary.h>
# include <rtabmap/core/Memory.h>
2016-09-07 17:19:21 -04:00
# include <rtabmap/core/GainCompensator.h>
2016-02-15 19:35:24 -05:00
# include <pcl/filters/extract_indices.h>
# include <pcl/io/ply_io.h>
2016-03-21 20:03:03 -04:00
# include <pcl/io/obj_io.h>
2016-02-15 19:35:24 -05:00
const int kVersionStringLength = 128 ;
2016-09-10 18:45:46 -04:00
const int minPolygonClusterSize = 200 ;
2016-02-15 19:35:24 -05:00
static JavaVM * jvm ;
static jobject RTABMapActivity = 0 ;
2016-05-12 14:34:51 -04:00
namespace {
constexpr int kTangoCoreMinimumVersion = 9377 ;
} // anonymous namespace.
2016-02-15 19:35:24 -05:00
rtabmap : : ParametersMap RTABMapApp : : getRtabmapParameters ( )
{
rtabmap : : ParametersMap parameters ;
2016-04-02 15:17:40 -04:00
parameters . insert ( mappingParameters_ . begin ( ) , mappingParameters_ . end ( ) ) ;
2016-02-15 19:35:24 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kKpDetectorStrategy ( ) , std : : string ( " 6 " ) ) ) ; // GFTT/BRIEF
2016-09-10 18:45:46 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kKpMaxFeatures ( ) , std : : string ( " 200 " ) ) ) ;
2016-02-15 19:35:24 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kGFTTQualityLevel ( ) , std : : string ( " 0.0001 " ) ) ) ;
2016-04-08 18:58:18 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemImagePreDecimation ( ) , std : : string ( fullResolution_ ? " 2 " : " 1 " ) ) ) ;
2016-02-15 19:35:24 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kFASTThreshold ( ) , std : : string ( " 1 " ) ) ) ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kBRIEFBytes ( ) , std : : string ( " 64 " ) ) ) ;
2016-04-08 18:58:18 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRtabmapTimeThr ( ) , std : : string ( " 800 " ) ) ) ;
2016-09-10 18:45:46 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRtabmapPublishLikelihood ( ) , std : : string ( " false " ) ) ) ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRtabmapPublishPdf ( ) , std : : string ( " false " ) ) ) ;
2016-11-21 17:09:04 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRtabmapStartNewMapOnLoopClosure ( ) , uBool2Str ( appendMode_ ) ) ) ;
2016-02-15 19:35:24 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemBinDataKept ( ) , uBool2Str ( ! trajectoryMode_ ) ) ) ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemNotLinkedNodesKept ( ) , std : : string ( " false " ) ) ) ;
2016-04-03 21:35:30 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kOptimizerIterations ( ) , graphOptimization_ ? " 10 " : " 0 " ) ) ;
2016-02-15 19:35:24 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemIncrementalMemory ( ) , uBool2Str ( ! localizationMode_ ) ) ) ;
2016-09-02 22:37:12 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRtabmapMaxRetrieved ( ) , " 1 " ) ) ;
2016-09-03 20:38:27 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDMaxLocalRetrieved ( ) , " 0 " ) ) ;
2016-02-15 19:35:24 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kKpMaxDepth ( ) , std : : string ( " 10 " ) ) ) ; // to avoid extracting features in invalid depth (as we compute transformation directly from the words)
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDOptimizeFromGraphEnd ( ) , std : : string ( " true " ) ) ) ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kDbSqlite3InMemory ( ) , std : : string ( " true " ) ) ) ;
2016-11-21 17:09:04 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kVisMinInliers ( ) , std : : string ( " 25 " ) ) ) ;
2016-04-11 10:43:28 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kVisEstimationType ( ) , std : : string ( " 0 " ) ) ) ; // 0=3D-3D 1=PnP
2016-11-16 10:13:16 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDOptimizeMaxError ( ) , std : : string ( " 0.1 " ) ) ) ;
2016-09-02 22:37:12 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDProximityPathMaxNeighbors ( ) , std : : string ( " 0 " ) ) ) ; // disable scan matching to merged nodes
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDProximityBySpace ( ) , std : : string ( " false " ) ) ) ; // just keep loop closure detection
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDNeighborLinkRefining ( ) , uBool2Str ( driftCorrection_ ) ) ) ;
2016-11-21 17:09:04 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRegStrategy ( ) , std : : string ( driftCorrection_ ? " 2 " : " 0 " ) ) ) ;
2016-09-07 17:19:21 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpPointToPlane ( ) , std : : string ( " true " ) ) ) ;
2016-11-21 17:09:04 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemLaserScanNormalK ( ) , std : : string ( " 10 " ) ) ) ;
2016-09-02 22:37:12 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpIterations ( ) , std : : string ( " 10 " ) ) ) ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpEpsilon ( ) , std : : string ( " 0.001 " ) ) ) ;
2016-09-07 17:19:21 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpMaxRotation ( ) , std : : string ( " 0.17 " ) ) ) ; // 10 degrees
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpMaxTranslation ( ) , std : : string ( " 0.05 " ) ) ) ;
2016-11-21 17:09:04 -05:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpCorrespondenceRatio ( ) , std : : string ( " 0.5 " ) ) ) ;
2016-09-02 22:37:12 -04:00
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kIcpMaxCorrespondenceDistance ( ) , std : : string ( " 0.05 " ) ) ) ;
2016-02-15 19:35:24 -05:00
2016-10-29 18:14:25 -04:00
parameters . insert ( * rtabmap : : Parameters : : getDefaultParameters ( ) . find ( rtabmap : : Parameters : : kKpMaxFeatures ( ) ) ) ;
parameters . insert ( * rtabmap : : Parameters : : getDefaultParameters ( ) . find ( rtabmap : : Parameters : : kMemRehearsalSimilarity ( ) ) ) ;
parameters . insert ( * rtabmap : : Parameters : : getDefaultParameters ( ) . find ( rtabmap : : Parameters : : kMemMapLabelsAdded ( ) ) ) ;
if ( dataRecorderMode_ )
{
uInsert ( parameters , rtabmap : : ParametersPair ( rtabmap : : Parameters : : kKpMaxFeatures ( ) , std : : string ( " -1 " ) ) ) ;
uInsert ( parameters , rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemRehearsalSimilarity ( ) , std : : string ( " 1.0 " ) ) ) ; // deactivate rehearsal
uInsert ( parameters , rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemMapLabelsAdded ( ) , " false " ) ) ; // don't create map labels
}
2016-02-15 19:35:24 -05:00
return parameters ;
}
RTABMapApp : : RTABMapApp ( ) :
camera_ ( 0 ) ,
rtabmapThread_ ( 0 ) ,
2016-04-03 21:35:30 -04:00
rtabmap_ ( 0 ) ,
2016-02-15 19:35:24 -05:00
logHandler_ ( 0 ) ,
odomCloudShown_ ( true ) ,
graphOptimization_ ( true ) ,
2016-04-09 18:57:49 -04:00
nodesFiltering_ ( false ) ,
2016-09-02 22:37:12 -04:00
driftCorrection_ ( false ) ,
2016-02-15 19:35:24 -05:00
localizationMode_ ( false ) ,
trajectoryMode_ ( false ) ,
2016-09-08 11:55:44 -04:00
autoExposure_ ( true ) ,
2016-04-02 15:17:40 -04:00
fullResolution_ ( false ) ,
2016-11-21 17:09:04 -05:00
appendMode_ ( true ) ,
2016-04-02 15:17:40 -04:00
maxCloudDepth_ ( 0.0 ) ,
2016-04-04 13:03:43 -04:00
meshTrianglePix_ ( 1 ) ,
2016-09-02 22:37:12 -04:00
meshAngleToleranceDeg_ ( 15.0 ) ,
2016-09-08 11:55:44 -04:00
paused_ ( false ) ,
2016-10-29 18:14:25 -04:00
dataRecorderMode_ ( false ) ,
2016-02-15 19:35:24 -05:00
clearSceneOnNextRender_ ( false ) ,
2016-09-03 20:38:27 -04:00
filterPolygonsOnNextRender_ ( false ) ,
2016-10-27 15:35:19 -04:00
gainCompensationOnNextRender_ ( 0 ) ,
2016-11-04 16:39:52 -04:00
bilateralFilteringOnNextRender_ ( false ) ,
2016-09-08 11:55:44 -04:00
cameraJustInitialized_ ( false ) ,
2016-02-15 19:35:24 -05:00
totalPoints_ ( 0 ) ,
totalPolygons_ ( 0 ) ,
2016-05-12 14:34:51 -04:00
lastDrawnCloudsCount_ ( 0 ) ,
2016-09-08 11:55:44 -04:00
renderingTime_ ( 0.0f )
2016-05-12 14:34:51 -04:00
2016-02-15 19:35:24 -05:00
{
}
RTABMapApp : : ~ RTABMapApp ( ) {
if ( camera_ )
{
delete camera_ ;
}
if ( rtabmapThread_ )
{
rtabmapThread_ - > close ( false ) ;
delete rtabmapThread_ ;
}
if ( logHandler_ )
{
delete logHandler_ ;
}
}
2016-05-12 14:34:51 -04:00
void RTABMapApp : : onCreate ( JNIEnv * env , jobject caller_activity )
2016-02-15 19:35:24 -05:00
{
env - > GetJavaVM ( & jvm ) ;
RTABMapActivity = env - > NewGlobalRef ( caller_activity ) ;
2016-05-12 14:34:51 -04:00
LOGI ( " RTABMapApp::onCreate() " ) ;
2016-02-15 19:35:24 -05:00
createdMeshes_ . clear ( ) ;
rawPoses_ . clear ( ) ;
clearSceneOnNextRender_ = true ;
totalPoints_ = 0 ;
totalPolygons_ = 0 ;
lastDrawnCloudsCount_ = 0 ;
2016-09-08 11:55:44 -04:00
renderingTime_ = 0.0f ;
2016-02-15 19:35:24 -05:00
if ( camera_ )
{
delete camera_ ;
}
if ( rtabmapThread_ )
{
rtabmapThread_ - > close ( false ) ;
delete rtabmapThread_ ;
rtabmapThread_ = 0 ;
2016-04-03 21:35:30 -04:00
rtabmap_ = 0 ;
2016-02-15 19:35:24 -05:00
}
if ( logHandler_ = = 0 )
{
logHandler_ = new LogHandler ( ) ;
}
this - > registerToEventsManager ( ) ;
2016-04-02 15:17:40 -04:00
camera_ = new rtabmap : : CameraTango ( fullResolution_ ? 1 : 2 , autoExposure_ ) ;
2016-02-15 19:35:24 -05:00
}
void RTABMapApp : : openDatabase ( const std : : string & databasePath )
{
this - > unregisterFromEventsManager ( ) ; // to ignore published init events when closing rtabmap
status_ . first = rtabmap : : RtabmapEventInit : : kInitializing ;
rtabmapMutex_ . lock ( ) ;
if ( rtabmapThread_ )
{
rtabmapThread_ - > close ( false ) ;
delete rtabmapThread_ ;
rtabmapThread_ = 0 ;
2016-04-03 21:35:30 -04:00
rtabmap_ = 0 ;
2016-02-15 19:35:24 -05:00
}
//Rtabmap
2016-04-03 21:35:30 -04:00
rtabmap_ = new rtabmap : : Rtabmap ( ) ;
2016-02-15 19:35:24 -05:00
rtabmap : : ParametersMap parameters = getRtabmapParameters ( ) ;
2016-04-03 21:35:30 -04:00
rtabmap_ - > init ( parameters , databasePath ) ;
rtabmapThread_ = new rtabmap : : RtabmapThread ( rtabmap_ ) ;
2016-10-29 18:14:25 -04:00
if ( parameters . find ( rtabmap : : Parameters : : kRtabmapDetectionRate ( ) ) ! = parameters . end ( ) )
{
rtabmapThread_ - > setDetectorRate ( uStr2Float ( parameters . at ( rtabmap : : Parameters : : kRtabmapDetectionRate ( ) ) ) ) ;
}
2016-02-15 19:35:24 -05:00
// Generate all meshes
std : : map < int , rtabmap : : Signature > signatures ;
std : : map < int , rtabmap : : Transform > poses ;
std : : multimap < int , rtabmap : : Link > links ;
2016-04-03 21:35:30 -04:00
rtabmap_ - > get3DMap (
2016-02-15 19:35:24 -05:00
signatures ,
poses ,
links ,
true ,
true ) ;
clearSceneOnNextRender_ = true ;
rtabmap : : Statistics stats ;
stats . setSignatures ( signatures ) ;
2016-04-09 18:57:49 -04:00
stats . addStatistic ( rtabmap : : Statistics : : kMemoryWorking_memory_size ( ) , ( float ) rtabmap_ - > getWMSize ( ) ) ;
stats . addStatistic ( rtabmap : : Statistics : : kKeypointDictionary_size ( ) , ( float ) rtabmap_ - > getMemory ( ) - > getVWDictionary ( ) - > getVisualWords ( ) . size ( ) ) ;
stats . addStatistic ( rtabmap : : Statistics : : kMemoryDatabase_memory_used ( ) , ( float ) rtabmap_ - > getMemory ( ) - > getDatabaseMemoryUsed ( ) ) ;
2016-02-15 19:35:24 -05:00
stats . setPoses ( poses ) ;
stats . setConstraints ( links ) ;
rtabmapEvents_ . push_back ( stats ) ;
// Start threads
LOGI ( " Start rtabmap thread " ) ;
this - > registerToEventsManager ( ) ;
rtabmapThread_ - > registerToEventsManager ( ) ;
rtabmapThread_ - > start ( ) ;
status_ . first = rtabmap : : RtabmapEventInit : : kInitialized ;
status_ . second = " " ;
rtabmapMutex_ . unlock ( ) ;
}
2016-05-12 14:34:51 -04:00
bool RTABMapApp : : onTangoServiceConnected ( JNIEnv * env , jobject iBinder )
2016-02-15 19:35:24 -05:00
{
2016-05-12 14:34:51 -04:00
LOGW ( " onTangoServiceConnected() " ) ;
2016-02-15 19:35:24 -05:00
if ( camera_ )
{
camera_ - > join ( true ) ;
2016-05-12 14:34:51 -04:00
if ( TangoService_setBinder ( env , iBinder ) ! = TANGO_SUCCESS ) {
LOGE ( " TangoHandler::ConnectTango, TangoService_setBinder error " ) ;
return false ;
}
2016-02-15 19:35:24 -05:00
if ( camera_ - > init ( ) )
{
LOGI ( " Start camera thread " ) ;
2016-09-08 11:55:44 -04:00
if ( ! paused_ )
{
camera_ - > start ( ) ;
}
cameraJustInitialized_ = true ;
2016-05-12 14:34:51 -04:00
return true ;
2016-02-15 19:35:24 -05:00
}
LOGE ( " Failed camera initialization! " ) ;
}
2016-05-12 14:34:51 -04:00
return false ;
2016-02-15 19:35:24 -05:00
}
void RTABMapApp : : onPause ( )
{
LOGW ( " onPause() " ) ;
if ( camera_ )
{
camera_ - > join ( true ) ;
camera_ - > close ( ) ;
}
}
void RTABMapApp : : TangoResetMotionTracking ( ) {
TangoService_resetMotionTracking ( ) ;
}
// OpenGL thread
void RTABMapApp : : InitializeGLContent ( )
{
UINFO ( " " ) ;
main_scene_ . InitGLContent ( ) ;
}
// OpenGL thread
void RTABMapApp : : SetViewPort ( int width , int height )
{
UINFO ( " " ) ;
main_scene_ . SetupViewPort ( width , height ) ;
}
2016-04-09 18:57:49 -04:00
class PostRenderEvent : public UEvent
{
public :
PostRenderEvent ( const rtabmap : : Statistics & stats ) :
stats_ ( stats )
{
}
virtual std : : string getClassName ( ) const { return " PostRenderEvent " ; }
const rtabmap : : Statistics & getStats ( ) const { return stats_ ; }
private :
rtabmap : : Statistics stats_ ;
} ;
2016-11-04 16:39:52 -04:00
// OpenGL thread
bool RTABMapApp : : smoothMesh ( int id , Mesh & mesh )
{
UTimer t ;
// reconstruct depth image
cv : : Mat depth = cv : : Mat : : zeros ( mesh . height , mesh . width , CV_32FC1 ) ;
rtabmap : : Transform localTransformInv = mesh . cameraModel . localTransform ( ) . inverse ( ) ;
for ( unsigned int i = 0 ; i < mesh . denseToOrganizedIndices . size ( ) ; + + i )
{
// FastBilateralFilter works in camera frame
pcl : : PointXYZRGB pt = rtabmap : : util3d : : transformPoint ( mesh . cloud - > at ( i ) , localTransformInv ) ;
depth . at < float > ( mesh . denseToOrganizedIndices [ i ] ) = pt . z ;
}
depth = rtabmap : : util2d : : fastBilateralFiltering ( depth , 2.0f , 0.075f ) ;
LOGI ( " smoothMesh() Bilateral filtering of %d, time=%fs " , id , t . ticks ( ) ) ;
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr cloudOrganized = rtabmap : : util3d : : cloudFromDepthRGB ( mesh . texture . rows > 1 ? mesh . texture : rtabmap : : uncompressImage ( mesh . texture ) , depth , mesh . cameraModel , 1 , maxCloudDepth_ ) ;
cloudOrganized = rtabmap : : util3d : : transformPointCloud ( cloudOrganized , mesh . cameraModel . localTransform ( ) ) ;
//reconstruct the mesh with smoothed surfaces
std : : vector < pcl : : Vertices > polygons = rtabmap : : util3d : : organizedFastMesh ( cloudOrganized , meshAngleToleranceDeg_ * M_PI / 180.0 , false , meshTrianglePix_ ) ;
// filter NaN points
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr outputCloud ( new pcl : : PointCloud < pcl : : PointXYZRGB > ) ;
std : : vector < pcl : : Vertices > outputPolygons ;
std : : vector < int > denseToOrganizedIndices = rtabmap : : util3d : : filterNaNPointsFromMesh ( * cloudOrganized , polygons , * outputCloud , outputPolygons ) ;
LOGI ( " smoothMesh() Reconstructing the mesh of %d, time=%fs " , id , t . ticks ( ) ) ;
if ( outputPolygons . size ( ) )
{
mesh . cloud = outputCloud ;
mesh . polygons = outputPolygons ;
mesh . denseToOrganizedIndices = denseToOrganizedIndices ;
}
else
{
LOGE ( " smoothMesh() Failed to smooth surface %d " , id ) ;
return false ;
}
return true ;
}
2016-02-15 19:35:24 -05:00
// OpenGL thread
int RTABMapApp : : Render ( )
{
2016-09-08 11:55:44 -04:00
UTimer fpsTime ;
2016-11-18 14:18:04 -05:00
bool notifyCameraStarted = false ;
2016-09-08 11:55:44 -04:00
bool notifyDataLoaded = false ;
2016-09-07 17:19:21 -04:00
boost : : mutex : : scoped_lock lock ( renderingMutex_ ) ;
2016-02-15 19:35:24 -05:00
// should be before clearSceneOnNextRender_ in case openDatabase is called
std : : list < rtabmap : : Statistics > rtabmapEvents ;
{
boost : : mutex : : scoped_lock lock ( rtabmapMutex_ ) ;
rtabmapEvents = rtabmapEvents_ ;
rtabmapEvents_ . clear ( ) ;
}
if ( clearSceneOnNextRender_ )
{
odomMutex_ . lock ( ) ;
odomEvents_ . clear ( ) ;
odomMutex_ . unlock ( ) ;
poseMutex_ . lock ( ) ;
poseEvents_ . clear ( ) ;
poseMutex_ . unlock ( ) ;
main_scene_ . clear ( ) ;
clearSceneOnNextRender_ = false ;
createdMeshes_ . clear ( ) ;
rawPoses_ . clear ( ) ;
totalPoints_ = 0 ;
totalPolygons_ = 0 ;
lastDrawnCloudsCount_ = 0 ;
2016-09-08 11:55:44 -04:00
renderingTime_ = 0.0f ;
}
// Did we lose OpenGL context? If so, recreate the context;
if ( main_scene_ . getAddedClouds ( ) . size ( ) ! = createdMeshes_ . size ( ) )
{
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ; iter ! = createdMeshes_ . end ( ) ; + + iter )
{
if ( ! main_scene_ . hasCloud ( iter - > first ) )
{
cv : : Mat compressed = iter - > second . texture ;
iter - > second . texture = rtabmap : : uncompressImage ( iter - > second . texture ) ;
2016-11-18 14:18:04 -05:00
main_scene_ . addMesh ( iter - > first , iter - > second , opengl_world_T_rtabmap_world * iter - > second . pose ) ;
2016-09-08 11:55:44 -04:00
main_scene_ . setCloudVisible ( iter - > first , iter - > second . visible ) ;
iter - > second . texture = compressed ;
}
}
2016-02-15 19:35:24 -05:00
}
// Process events
rtabmap : : Transform pose ;
{
boost : : mutex : : scoped_lock lock ( poseMutex_ ) ;
if ( poseEvents_ . size ( ) )
{
pose = poseEvents_ . back ( ) ;
poseEvents_ . clear ( ) ;
}
}
if ( ! pose . isNull ( ) )
{
// update camera pose?
2016-11-16 10:13:16 -05:00
main_scene_ . SetCameraPose ( opengl_world_T_tango_world * pose ) ;
2016-09-08 11:55:44 -04:00
if ( ! camera_ - > isRunning ( ) & & cameraJustInitialized_ )
{
2016-11-18 14:18:04 -05:00
notifyCameraStarted = true ;
2016-09-08 11:55:44 -04:00
cameraJustInitialized_ = false ;
}
}
rtabmap : : OdometryEvent odomEvent ;
{
boost : : mutex : : scoped_lock lock ( odomMutex_ ) ;
if ( odomEvents_ . size ( ) )
{
LOGI ( " Process odom events " ) ;
odomEvent = odomEvents_ . back ( ) ;
odomEvents_ . clear ( ) ;
if ( cameraJustInitialized_ )
{
2016-11-18 14:18:04 -05:00
notifyCameraStarted = true ;
2016-09-08 11:55:44 -04:00
cameraJustInitialized_ = false ;
}
}
2016-02-15 19:35:24 -05:00
}
if ( rtabmapEvents . size ( ) )
{
LOGI ( " Process rtabmap events " ) ;
// update buffered signatures
std : : map < int , rtabmap : : SensorData > bufferedSensorData ;
2016-10-29 18:14:25 -04:00
if ( ! trajectoryMode_ & & ! dataRecorderMode_ )
2016-02-15 19:35:24 -05:00
{
for ( std : : list < rtabmap : : Statistics > : : iterator iter = rtabmapEvents . begin ( ) ; iter ! = rtabmapEvents . end ( ) ; + + iter )
{
for ( std : : map < int , rtabmap : : Signature > : : const_iterator jter = iter - > getSignatures ( ) . begin ( ) ; jter ! = iter - > getSignatures ( ) . end ( ) ; + + jter )
{
if ( ! jter - > second . sensorData ( ) . imageRaw ( ) . empty ( ) & &
! jter - > second . sensorData ( ) . depthRaw ( ) . empty ( ) )
{
uInsert ( bufferedSensorData , std : : make_pair ( jter - > first , jter - > second . sensorData ( ) ) ) ;
uInsert ( rawPoses_ , std : : make_pair ( jter - > first , jter - > second . getPose ( ) ) ) ;
}
2016-11-18 13:07:53 -05:00
else if ( totalPoints_ = = 0 & &
! jter - > second . sensorData ( ) . imageCompressed ( ) . empty ( ) & &
2016-02-15 19:35:24 -05:00
! jter - > second . sensorData ( ) . depthOrRightCompressed ( ) . empty ( ) )
{
// uncompress
rtabmap : : SensorData data = jter - > second . sensorData ( ) ;
2016-11-21 17:09:04 -05:00
cv : : Mat tmpA , depth ;
data . uncompressData ( & tmpA , & depth ) ;
// do post-processing bilateral filtering now
UTimer t ;
depth = rtabmap : : util2d : : fastBilateralFiltering ( depth , 2.0f , 0.075f ) ;
data . setDepthOrRightRaw ( depth ) ;
LOGI ( " Bilateral filtering of %d, time=%fs " , jter - > first , t . ticks ( ) ) ;
2016-02-15 19:35:24 -05:00
uInsert ( bufferedSensorData , std : : make_pair ( jter - > first , data ) ) ;
uInsert ( rawPoses_ , std : : make_pair ( jter - > first , jter - > second . getPose ( ) ) ) ;
2016-11-18 13:07:53 -05:00
LOGI ( " Detecting that we are loading a database, so do some post-processing... " ) ;
2016-02-15 19:35:24 -05:00
notifyDataLoaded = true ;
}
}
}
}
std : : map < int , rtabmap : : Transform > poses = rtabmapEvents . back ( ) . poses ( ) ;
// Transform pose in OpenGL world
for ( std : : map < int , rtabmap : : Transform > : : iterator iter = poses . begin ( ) ; iter ! = poses . end ( ) ; + + iter )
{
if ( ! graphOptimization_ )
{
std : : map < int , rtabmap : : Transform > : : iterator jter = rawPoses_ . find ( iter - > first ) ;
if ( jter ! = rawPoses_ . end ( ) )
{
iter - > second = opengl_world_T_rtabmap_world * jter - > second ;
}
}
else
{
iter - > second = opengl_world_T_rtabmap_world * iter - > second ;
}
}
2016-04-09 18:57:49 -04:00
const std : : multimap < int , rtabmap : : Link > & links = rtabmapEvents . back ( ) . constraints ( ) ;
2016-02-15 19:35:24 -05:00
if ( poses . size ( ) )
{
//update graph
main_scene_ . updateGraph ( poses , links ) ;
// update clouds
2016-04-09 18:57:49 -04:00
boost : : mutex : : scoped_lock lock ( meshesMutex_ ) ;
2016-02-15 19:35:24 -05:00
std : : set < std : : string > strIds ;
for ( std : : map < int , rtabmap : : Transform > : : iterator iter = poses . begin ( ) ; iter ! = poses . end ( ) ; + + iter )
{
int id = iter - > first ;
if ( ! iter - > second . isNull ( ) )
{
if ( main_scene_ . hasCloud ( id ) )
{
//just update pose
main_scene_ . setCloudPose ( id , iter - > second ) ;
main_scene_ . setCloudVisible ( id , true ) ;
2016-04-09 18:57:49 -04:00
std : : map < int , Mesh > : : iterator meshIter = createdMeshes_ . find ( id ) ;
2016-09-08 11:55:44 -04:00
UASSERT ( meshIter ! = createdMeshes_ . end ( ) ) ;
2016-11-04 16:39:52 -04:00
meshIter - > second . pose = opengl_world_T_rtabmap_world . inverse ( ) * iter - > second ;
2016-09-08 11:55:44 -04:00
meshIter - > second . visible = true ;
2016-02-15 19:35:24 -05:00
}
else if ( uContains ( bufferedSensorData , id ) )
{
rtabmap : : SensorData & data = bufferedSensorData . at ( id ) ;
if ( ! data . imageRaw ( ) . empty ( ) & & ! data . depthRaw ( ) . empty ( ) )
{
// Voxelize and filter depending on the previous cloud?
2016-04-02 15:17:40 -04:00
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr cloud ;
2016-02-15 19:35:24 -05:00
pcl : : IndicesPtr indices ( new std : : vector < int > ) ;
2016-09-07 19:59:25 -04:00
LOGI ( " Creating node cloud %d (depth=%dx%d rgb=%dx%d) " , id , data . depthRaw ( ) . cols , data . depthRaw ( ) . rows , data . imageRaw ( ) . cols , data . imageRaw ( ) . rows ) ;
2016-09-02 22:37:12 -04:00
cloud = rtabmap : : util3d : : cloudRGBFromSensorData ( data , 1 , maxCloudDepth_ , 0 , indices . get ( ) ) ;
2016-02-15 19:35:24 -05:00
if ( cloud - > size ( ) & & indices - > size ( ) )
{
UTimer time ;
// pcl::organizedFastMesh doesn't take indices, so set to NaN points we don't need to mesh
2016-04-02 15:17:40 -04:00
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr output ( new pcl : : PointCloud < pcl : : PointXYZRGB > ) ;
pcl : : ExtractIndices < pcl : : PointXYZRGB > filter ;
filter . setIndices ( indices ) ;
2016-02-15 19:35:24 -05:00
filter . setKeepOrganized ( true ) ;
filter . setInputCloud ( cloud ) ;
filter . filter ( * output ) ;
2016-04-04 13:03:43 -04:00
std : : vector < pcl : : Vertices > polygons = rtabmap : : util3d : : organizedFastMesh ( output , meshAngleToleranceDeg_ * M_PI / 180.0 , false , meshTrianglePix_ ) ;
2016-04-02 15:17:40 -04:00
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr outputCloud ( new pcl : : PointCloud < pcl : : PointXYZRGB > ) ;
2016-02-15 19:35:24 -05:00
std : : vector < pcl : : Vertices > outputPolygons ;
2016-03-21 15:45:26 -04:00
2016-11-04 16:39:52 -04:00
std : : vector < int > denseToOrganizedIndices = rtabmap : : util3d : : filterNaNPointsFromMesh ( * output , polygons , * outputCloud , outputPolygons ) ;
2016-04-02 15:17:40 -04:00
2016-02-15 19:35:24 -05:00
LOGI ( " Creating mesh, %d polygons (%fs) " , ( int ) outputPolygons . size ( ) , time . ticks ( ) ) ;
2016-04-02 15:17:40 -04:00
if ( outputCloud - > size ( ) & & outputPolygons . size ( ) )
2016-02-15 19:35:24 -05:00
{
totalPolygons_ + = outputPolygons . size ( ) ;
2016-04-02 15:17:40 -04:00
2016-03-21 20:03:03 -04:00
std : : pair < std : : map < int , Mesh > : : iterator , bool > inserted = createdMeshes_ . insert ( std : : make_pair ( id , Mesh ( ) ) ) ;
UASSERT ( inserted . second ) ;
inserted . first - > second . cloud = outputCloud ;
2016-09-07 19:59:25 -04:00
inserted . first - > second . denseToOrganizedIndices = denseToOrganizedIndices ;
inserted . first - > second . width = cloud - > width ;
inserted . first - > second . height = cloud - > height ;
2016-03-21 20:03:03 -04:00
inserted . first - > second . polygons = outputPolygons ;
2016-11-04 16:39:52 -04:00
inserted . first - > second . pose = opengl_world_T_rtabmap_world . inverse ( ) * iter - > second ;
2016-09-08 11:55:44 -04:00
inserted . first - > second . visible = true ;
2016-09-07 19:59:25 -04:00
inserted . first - > second . texture = data . imageRaw ( ) ;
2016-11-04 16:39:52 -04:00
inserted . first - > second . cameraModel = data . cameraModels ( ) [ 0 ] ;
2016-09-07 19:59:25 -04:00
2016-11-21 17:09:04 -05:00
if ( notifyDataLoaded )
{
// gain compensation is done, so don't compress the texture yet
inserted . first - > second . texture = data . imageRaw ( ) ; // keep raw
// mesh will be added in gain compensation below
}
else
{
main_scene_ . addMesh ( id , inserted . first - > second , iter - > second ) ;
inserted . first - > second . texture = data . imageCompressed ( ) ; // keep compressed
}
2016-09-07 19:59:25 -04:00
2016-02-15 19:35:24 -05:00
}
else
{
LOGE ( " Not mesh could be created for node %d " , id ) ;
}
}
totalPoints_ + = indices - > size ( ) ;
}
}
}
}
}
2016-04-09 18:57:49 -04:00
//filter poses?
if ( poses . size ( ) > 2 )
{
if ( nodesFiltering_ )
{
for ( std : : multimap < int , rtabmap : : Link > : : const_iterator iter = links . begin ( ) ; iter ! = links . end ( ) ; + + iter )
{
if ( iter - > second . type ( ) ! = rtabmap : : Link : : kNeighbor )
{
int oldId = iter - > second . to ( ) > iter - > second . from ( ) ? iter - > second . from ( ) : iter - > second . to ( ) ;
poses . erase ( oldId ) ;
}
}
}
}
2016-11-21 17:09:04 -05:00
if ( poses . size ( ) )
2016-02-15 19:35:24 -05:00
{
2016-11-21 17:09:04 -05:00
//update cloud visibility
std : : set < int > addedClouds = main_scene_ . getAddedClouds ( ) ;
for ( std : : set < int > : : const_iterator iter = addedClouds . begin ( ) ;
iter ! = addedClouds . end ( ) ;
+ + iter )
2016-02-15 19:35:24 -05:00
{
2016-11-21 17:09:04 -05:00
if ( * iter > 0 & & poses . find ( * iter ) = = poses . end ( ) )
{
main_scene_ . setCloudVisible ( * iter , false ) ;
std : : map < int , Mesh > : : iterator meshIter = createdMeshes_ . find ( * iter ) ;
UASSERT ( meshIter ! = createdMeshes_ . end ( ) ) ;
meshIter - > second . visible = true ;
}
2016-02-15 19:35:24 -05:00
}
}
}
else
{
2016-11-04 16:39:52 -04:00
main_scene_ . setCloudVisible ( - 1 , odomCloudShown_ & & ! trajectoryMode_ & & ! paused_ ) ;
2016-04-04 13:03:43 -04:00
2016-02-15 19:35:24 -05:00
//just process the last one
2016-09-08 11:55:44 -04:00
if ( ! odomEvent . pose ( ) . isNull ( ) )
2016-02-15 19:35:24 -05:00
{
if ( odomCloudShown_ & & ! trajectoryMode_ )
{
2016-09-08 11:55:44 -04:00
if ( ! odomEvent . data ( ) . imageRaw ( ) . empty ( ) & & ! odomEvent . data ( ) . depthRaw ( ) . empty ( ) )
2016-02-15 19:35:24 -05:00
{
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr cloud ;
2016-09-08 11:55:44 -04:00
cloud = rtabmap : : util3d : : cloudRGBFromSensorData ( odomEvent . data ( ) , 1 , maxCloudDepth_ ) ;
2016-02-15 19:35:24 -05:00
if ( cloud - > size ( ) )
{
2016-04-04 13:03:43 -04:00
LOGI ( " Created odom cloud (rgb=%dx%d depth=%dx%d cloud=%dx%d) " ,
2016-09-08 11:55:44 -04:00
odomEvent . data ( ) . imageRaw ( ) . cols , odomEvent . data ( ) . imageRaw ( ) . rows ,
odomEvent . data ( ) . depthRaw ( ) . cols , odomEvent . data ( ) . depthRaw ( ) . rows ,
( int ) cloud - > width , ( int ) cloud - > height ) ;
main_scene_ . addCloud ( - 1 , cloud , opengl_world_T_rtabmap_world * odomEvent . pose ( ) ) ;
2016-02-15 19:35:24 -05:00
main_scene_ . setCloudVisible ( - 1 , true ) ;
}
else
{
LOGE ( " Generated cloud is empty! " ) ;
}
}
else
{
LOGE ( " Odom data images are empty! " ) ;
}
}
}
}
2016-10-27 15:35:19 -04:00
if ( notifyDataLoaded | | gainCompensationOnNextRender_ > 0 )
2016-09-07 17:19:21 -04:00
{
2016-11-21 17:09:04 -05:00
UTimer tGainCompensation ;
2016-09-07 19:59:25 -04:00
LOGI ( " Gain compensation... " ) ;
boost : : mutex : : scoped_lock lock ( meshesMutex_ ) ;
2016-11-21 17:09:04 -05:00
rtabmap : : GainCompensator compensator ;
std : : map < int , pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr > clouds ;
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ; iter ! = createdMeshes_ . end ( ) ; + + iter )
2016-09-07 17:19:21 -04:00
{
2016-11-21 17:09:04 -05:00
clouds . insert ( std : : make_pair ( iter - > first , iter - > second . cloud ) ) ;
}
std : : map < int , rtabmap : : Transform > poses ;
std : : multimap < int , rtabmap : : Link > links ;
rtabmap_ - > getGraph ( poses , links , false , true ) ;
if ( gainCompensationOnNextRender_ = = 2 )
{
// full compensation
links . clear ( ) ;
for ( std : : map < int , pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr > : : const_iterator iter = clouds . begin ( ) ; iter ! = clouds . end ( ) ; + + iter )
2016-10-27 15:35:19 -04:00
{
2016-11-21 17:09:04 -05:00
int from = iter - > first ;
std : : map < int , pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr > : : const_iterator jter = iter ;
+ + jter ;
for ( ; jter ! = clouds . end ( ) ; + + jter )
2016-10-27 15:35:19 -04:00
{
2016-11-21 17:09:04 -05:00
int to = jter - > first ;
links . insert ( std : : make_pair ( from , rtabmap : : Link ( from , to , rtabmap : : Link : : kUserClosure , poses . at ( from ) . inverse ( ) * poses . at ( to ) ) ) ) ;
2016-10-27 15:35:19 -04:00
}
}
2016-11-21 17:09:04 -05:00
}
2016-10-27 15:35:19 -04:00
2016-11-21 17:09:04 -05:00
if ( clouds . size ( ) > 1 & & links . size ( ) )
{
2016-09-07 19:59:25 -04:00
compensator . feed ( clouds , links ) ;
2016-11-21 17:09:04 -05:00
LOGI ( " Gain compensation... compute gain: links=%d, time=%fs " , ( int ) links . size ( ) , tGainCompensation . ticks ( ) ) ;
}
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ; iter ! = createdMeshes_ . end ( ) ; + + iter )
{
cv : : Mat compressedImage = iter - > second . texture ;
iter - > second . texture = compressedImage . rows = = 1 ? rtabmap : : uncompressImage ( compressedImage ) : compressedImage ;
if ( ! iter - > second . cloud - > empty ( ) )
2016-09-07 19:59:25 -04:00
{
2016-11-21 17:09:04 -05:00
if ( clouds . size ( ) > 1 & & links . size ( ) )
2016-09-07 17:19:21 -04:00
{
2016-09-07 19:59:25 -04:00
compensator . apply ( iter - > first , iter - > second . cloud ) ;
if ( ! iter - > second . texture . empty ( ) )
{
compensator . apply ( iter - > first , iter - > second . texture ) ;
}
2016-09-07 17:19:21 -04:00
}
2016-11-21 17:09:04 -05:00
}
2016-11-04 16:39:52 -04:00
2016-11-21 17:09:04 -05:00
if ( notifyDataLoaded )
{
main_scene_ . addMesh ( iter - > first , iter - > second , opengl_world_T_rtabmap_world * poses . at ( iter - > first ) ) ;
}
else
{
main_scene_ . updateMesh ( iter - > first , iter - > second ) ;
2016-09-07 17:19:21 -04:00
}
2016-11-21 17:09:04 -05:00
iter - > second . texture = rtabmap : : compressImage2 ( iter - > second . texture , " .jpg " ) ;
2016-09-07 17:19:21 -04:00
}
2016-11-21 17:09:04 -05:00
LOGI ( " Gain compensation... applying gain: meshes=%d, time=%fs " , ( int ) createdMeshes_ . size ( ) , tGainCompensation . ticks ( ) ) ;
2016-10-27 15:35:19 -04:00
gainCompensationOnNextRender_ = 0 ;
2016-09-07 17:19:21 -04:00
notifyDataLoaded = true ;
}
2016-11-04 16:39:52 -04:00
if ( bilateralFilteringOnNextRender_ )
{
LOGI ( " Bilateral filtering... " ) ;
bilateralFilteringOnNextRender_ = false ;
boost : : mutex : : scoped_lock lock ( meshesMutex_ ) ;
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ; iter ! = createdMeshes_ . end ( ) ; + + iter )
{
if ( iter - > second . cloud - > size ( ) )
{
if ( smoothMesh ( iter - > first , iter - > second ) )
{
main_scene_ . addMesh ( iter - > first , iter - > second , opengl_world_T_rtabmap_world * iter - > second . pose ) ;
}
}
}
notifyDataLoaded = true ;
}
2016-09-03 20:38:27 -04:00
if ( filterPolygonsOnNextRender_ )
{
2016-09-07 19:59:25 -04:00
LOGI ( " Polygon filtering... " ) ;
2016-09-03 20:38:27 -04:00
filterPolygonsOnNextRender_ = false ;
boost : : mutex : : scoped_lock lock ( meshesMutex_ ) ;
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ; iter ! = createdMeshes_ . end ( ) ; + + iter )
{
if ( iter - > second . polygons . size ( ) )
{
// filter polygons
std : : vector < std : : set < int > > neighbors ;
std : : vector < std : : set < int > > vertexToPolygons ;
rtabmap : : util3d : : createPolygonIndexes (
iter - > second . polygons ,
iter - > second . cloud - > size ( ) ,
neighbors ,
vertexToPolygons ) ;
std : : list < std : : list < int > > clusters = rtabmap : : util3d : : clusterPolygons (
neighbors ,
minPolygonClusterSize ) ;
std : : vector < pcl : : Vertices > filteredPolygons ( iter - > second . polygons . size ( ) ) ;
int oi = 0 ;
for ( std : : list < std : : list < int > > : : iterator jter = clusters . begin ( ) ; jter ! = clusters . end ( ) ; + + jter )
{
for ( std : : list < int > : : iterator kter = jter - > begin ( ) ; kter ! = jter - > end ( ) ; + + kter )
{
filteredPolygons [ oi + + ] = iter - > second . polygons . at ( * kter ) ;
}
}
filteredPolygons . resize ( oi ) ;
iter - > second . polygons = filteredPolygons ;
main_scene_ . updateCloudPolygons ( iter - > first , iter - > second . polygons ) ;
}
}
notifyDataLoaded = true ;
}
2016-02-15 19:35:24 -05:00
lastDrawnCloudsCount_ = main_scene_ . Render ( ) ;
2016-09-08 11:55:44 -04:00
if ( renderingTime_ < fpsTime . elapsed ( ) )
{
renderingTime_ = fpsTime . elapsed ( ) ;
}
2016-02-15 19:35:24 -05:00
2016-04-09 18:57:49 -04:00
if ( rtabmapEvents . size ( ) )
{
// send statistics to GUI
LOGI ( " Posting PostRenderEvent! " ) ;
UEventsManager : : post ( new PostRenderEvent ( rtabmapEvents . back ( ) ) ) ;
}
2016-11-18 14:18:04 -05:00
return notifyDataLoaded | | notifyCameraStarted ? 1 : 0 ;
2016-02-15 19:35:24 -05:00
}
void RTABMapApp : : SetCameraType (
tango_gl : : GestureCamera : : CameraType camera_type ) {
main_scene_ . SetCameraType ( camera_type ) ;
}
void RTABMapApp : : OnTouchEvent ( int touch_count ,
tango_gl : : GestureCamera : : TouchEvent event ,
float x0 , float y0 , float x1 , float y1 ) {
main_scene_ . OnTouchEvent ( touch_count , event , x0 , y0 , x1 , y1 ) ;
}
void RTABMapApp : : setPausedMapping ( bool paused )
{
2016-09-08 11:55:44 -04:00
paused_ = paused ;
2016-02-15 19:35:24 -05:00
if ( camera_ )
{
2016-09-08 11:55:44 -04:00
if ( paused_ )
2016-02-15 19:35:24 -05:00
{
LOGW ( " Pause! " ) ;
camera_ - > kill ( ) ;
2016-09-08 11:55:44 -04:00
2016-02-15 19:35:24 -05:00
}
else
{
LOGW ( " Resume! " ) ;
2016-09-08 11:55:44 -04:00
UEventsManager : : post ( new rtabmap : : RtabmapEventCmd ( rtabmap : : RtabmapEventCmd : : kCmdTriggerNewMap ) ) ;
2016-02-15 19:35:24 -05:00
camera_ - > start ( ) ;
}
}
}
void RTABMapApp : : setMapCloudShown ( bool shown )
{
2016-04-04 13:03:43 -04:00
main_scene_ . setMapRendering ( shown ) ;
2016-02-15 19:35:24 -05:00
}
void RTABMapApp : : setOdomCloudShown ( bool shown )
{
odomCloudShown_ = shown ;
main_scene_ . setTraceVisible ( shown ) ;
}
2016-09-02 22:37:12 -04:00
void RTABMapApp : : setMeshRendering ( bool enabled , bool withTexture )
2016-02-15 19:35:24 -05:00
{
2016-09-02 22:37:12 -04:00
main_scene_ . setMeshRendering ( enabled , withTexture ) ;
2016-02-15 19:35:24 -05:00
}
void RTABMapApp : : setLocalizationMode ( bool enabled )
{
localizationMode_ = enabled ;
this - > post ( new rtabmap : : ParamEvent ( rtabmap : : Parameters : : kMemIncrementalMemory ( ) , uBool2Str ( ! localizationMode_ ) ) ) ;
}
void RTABMapApp : : setTrajectoryMode ( bool enabled )
{
if ( trajectoryMode_ ! = enabled )
{
main_scene_ . SetCameraType ( enabled ? tango_gl : : GestureCamera : : kTopDown : tango_gl : : GestureCamera : : kThirdPersonFollow ) ;
}
trajectoryMode_ = enabled ;
this - > post ( new rtabmap : : ParamEvent ( rtabmap : : Parameters : : kMemBinDataKept ( ) , uBool2Str ( ! trajectoryMode_ ) ) ) ;
}
void RTABMapApp : : setGraphOptimization ( bool enabled )
{
graphOptimization_ = enabled ;
2016-04-09 18:57:49 -04:00
if ( ! camera_ - > isRunning ( ) )
{
std : : map < int , rtabmap : : Transform > poses ;
std : : multimap < int , rtabmap : : Link > links ;
rtabmap_ - > getGraph ( poses , links , true , true ) ;
if ( poses . size ( ) )
{
boost : : mutex : : scoped_lock lock ( rtabmapMutex_ ) ;
rtabmap : : Statistics stats = rtabmap_ - > getStatistics ( ) ;
stats . setPoses ( poses ) ;
stats . setConstraints ( links ) ;
rtabmapEvents_ . push_back ( stats ) ;
rtabmap_ - > setOptimizedPoses ( poses ) ;
}
}
}
void RTABMapApp : : setNodesFiltering ( bool enabled )
{
nodesFiltering_ = enabled ;
setGraphOptimization ( graphOptimization_ ) ; // this will resend the graph if paused
2016-02-15 19:35:24 -05:00
}
2016-09-02 22:37:12 -04:00
void RTABMapApp : : setDriftCorrection ( bool enabled )
{
driftCorrection_ = enabled ;
rtabmap : : ParametersMap parameters ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRGBDNeighborLinkRefining ( ) , uBool2Str ( driftCorrection_ ) ) ) ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRegStrategy ( ) , std : : string ( driftCorrection_ ? " 1 " : " 0 " ) ) ) ;
this - > post ( new rtabmap : : ParamEvent ( parameters ) ) ;
}
2016-02-15 19:35:24 -05:00
void RTABMapApp : : setGraphVisible ( bool visible )
{
main_scene_ . setGraphVisible ( visible ) ;
2016-09-02 22:37:12 -04:00
main_scene_ . setTraceVisible ( visible ) ;
}
void RTABMapApp : : setGridVisible ( bool visible )
{
main_scene_ . setGridVisible ( visible ) ;
2016-02-15 19:35:24 -05:00
}
void RTABMapApp : : setAutoExposure ( bool enabled )
{
if ( autoExposure_ ! = enabled )
{
autoExposure_ = enabled ;
if ( camera_ )
{
camera_ - > setAutoExposure ( autoExposure_ ) ;
}
2016-04-02 15:17:40 -04:00
}
}
void RTABMapApp : : setFullResolution ( bool enabled )
{
if ( fullResolution_ ! = enabled )
{
fullResolution_ = enabled ;
if ( camera_ )
{
camera_ - > setDecimation ( fullResolution_ ? 1 : 2 ) ;
}
2016-04-09 12:08:30 -04:00
rtabmap : : ParametersMap parameters ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kMemImagePreDecimation ( ) , std : : string ( fullResolution_ ? " 2 " : " 1 " ) ) ) ;
this - > post ( new rtabmap : : ParamEvent ( parameters ) ) ;
2016-04-02 15:17:40 -04:00
}
}
2016-11-21 17:09:04 -05:00
void RTABMapApp : : setAppendMode ( bool enabled )
{
if ( appendMode_ ! = enabled )
{
appendMode_ = enabled ;
rtabmap : : ParametersMap parameters ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRtabmapStartNewMapOnLoopClosure ( ) , uBool2Str ( appendMode_ ) ) ) ;
this - > post ( new rtabmap : : ParamEvent ( parameters ) ) ;
}
}
2016-10-29 18:14:25 -04:00
void RTABMapApp : : setDataRecorderMode ( bool enabled )
{
if ( dataRecorderMode_ ! = enabled )
{
dataRecorderMode_ = enabled ; // parameters will be set when resuming (we assume we are paused)
}
}
2016-04-02 15:17:40 -04:00
void RTABMapApp : : setMaxCloudDepth ( float value )
{
maxCloudDepth_ = value ;
}
2016-04-04 13:03:43 -04:00
void RTABMapApp : : setMeshAngleTolerance ( float value )
{
meshAngleToleranceDeg_ = value ;
}
void RTABMapApp : : setMeshTriangleSize ( int value )
{
meshTrianglePix_ = value ;
}
2016-04-02 15:17:40 -04:00
int RTABMapApp : : setMappingParameter ( const std : : string & key , const std : : string & value )
{
2016-11-24 15:09:16 -05:00
std : : string compatibleKey = key ;
// Backward compatibility
std : : map < std : : string , std : : pair < bool , std : : string > > : : const_iterator iter = rtabmap : : Parameters : : getRemovedParameters ( ) . find ( key ) ;
if ( iter ! = rtabmap : : Parameters : : getRemovedParameters ( ) . end ( ) )
{
if ( iter - > second . first )
{
// can be migrated
compatibleKey = iter - > second . second ;
LOGW ( " Parameter name changed: \" %s \" -> \" %s \" . Please update the code accordingly. Value \" %s \" is still set to the new parameter name. " ,
iter - > first . c_str ( ) , iter - > second . second . c_str ( ) , value . c_str ( ) ) ;
}
else
{
if ( iter - > second . second . empty ( ) )
{
LOGE ( " Parameter \" %s \" doesn't exist anymore! " ,
iter - > first . c_str ( ) ) ;
}
else
{
LOGE ( " Parameter \" %s \" doesn't exist anymore! You may look at this similar parameter: \" %s \" " ,
iter - > first . c_str ( ) , iter - > second . second . c_str ( ) ) ;
}
}
}
if ( rtabmap : : Parameters : : getDefaultParameters ( ) . find ( compatibleKey ) ! = rtabmap : : Parameters : : getDefaultParameters ( ) . end ( ) )
2016-04-02 15:17:40 -04:00
{
2016-11-24 15:09:16 -05:00
LOGI ( uFormat ( " Setting param \" %s \" to \" %s \" " , compatibleKey . c_str ( ) , value . c_str ( ) ) . c_str ( ) ) ;
2016-04-02 15:17:40 -04:00
uInsert ( mappingParameters_ , rtabmap : : ParametersPair ( key , value ) ) ;
UEventsManager : : post ( new rtabmap : : ParamEvent ( mappingParameters_ ) ) ;
return 0 ;
}
else
{
2016-11-24 15:09:16 -05:00
LOGE ( uFormat ( " Key \" %s \" doesn't exist! " , compatibleKey . c_str ( ) ) . c_str ( ) ) ;
2016-04-02 15:17:40 -04:00
return - 1 ;
2016-02-15 19:35:24 -05:00
}
}
void RTABMapApp : : resetMapping ( )
{
LOGW ( " Reset! " ) ;
status_ . first = rtabmap : : RtabmapEventInit : : kInitializing ;
status_ . second = " " ;
clearSceneOnNextRender_ = true ;
UEventsManager : : post ( new rtabmap : : RtabmapEventCmd ( rtabmap : : RtabmapEventCmd : : kCmdResetMemory ) ) ;
}
2016-11-21 17:09:04 -05:00
void RTABMapApp : : save ( const std : : string & databasePath )
2016-02-15 19:35:24 -05:00
{
2016-11-21 17:09:04 -05:00
rtabmapThread_ - > join ( true ) ;
2016-11-23 15:13:09 -05:00
if ( dataRecorderMode_ )
{
// to save mapping parameters in the database
dataRecorderMode_ = false ;
rtabmap : : ParametersMap parameters = getRtabmapParameters ( ) ;
rtabmap_ - > parseParameters ( parameters ) ;
dataRecorderMode_ = true ;
}
2016-11-21 17:09:04 -05:00
rtabmap_ - > close ( true , databasePath ) ;
2016-11-23 15:13:09 -05:00
rtabmap_ - > init ( getRtabmapParameters ( ) , dataRecorderMode_ ? " " : databasePath ) ;
if ( dataRecorderMode_ )
{
clearSceneOnNextRender_ = true ;
}
2016-11-21 17:09:04 -05:00
rtabmapThread_ - > start ( ) ;
2016-11-23 15:13:09 -05:00
2016-02-15 19:35:24 -05:00
}
bool RTABMapApp : : exportMesh ( const std : : string & filePath )
{
bool success = false ;
//Assemble the meshes
2016-04-02 15:17:40 -04:00
if ( UFile : : getExtension ( filePath ) . compare ( " obj " ) = = 0 )
2016-02-15 19:35:24 -05:00
{
2016-03-21 20:03:03 -04:00
pcl : : TextureMesh textureMesh ;
std : : vector < cv : : Mat > textures ;
2016-04-09 18:57:49 -04:00
int totalPolygons = 0 ;
2016-03-21 20:03:03 -04:00
pcl : : PointCloud < pcl : : PointXYZRGBNormal > : : Ptr mergedClouds ( new pcl : : PointCloud < pcl : : PointXYZRGBNormal > ) ;
2016-02-15 19:35:24 -05:00
{
2016-03-21 20:03:03 -04:00
boost : : mutex : : scoped_lock lock ( meshesMutex_ ) ;
textureMesh . tex_materials . resize ( createdMeshes_ . size ( ) ) ;
textureMesh . tex_polygons . resize ( createdMeshes_ . size ( ) ) ;
textureMesh . tex_coordinates . resize ( createdMeshes_ . size ( ) ) ;
textures . resize ( createdMeshes_ . size ( ) ) ;
int polygonsStep = 0 ;
int oi = 0 ;
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ;
iter ! = createdMeshes_ . end ( ) ;
+ + iter )
2016-02-15 19:35:24 -05:00
{
2016-03-21 20:03:03 -04:00
if ( ! iter - > second . texture . empty ( ) & &
iter - > second . cloud - > size ( ) & &
2016-09-07 19:59:25 -04:00
iter - > second . polygons . size ( ) & &
( ! iter - > second . cloud - > is_dense | | ( iter - > second . cloud - > is_dense & & iter - > second . denseToOrganizedIndices . size ( ) = = iter - > second . cloud - > size ( ) ) ) )
2016-03-21 20:03:03 -04:00
{
2016-04-02 15:17:40 -04:00
// OBJ format requires normals
2016-09-07 19:59:25 -04:00
pcl : : PointCloud < pcl : : Normal > : : Ptr normals = rtabmap : : util3d : : computeNormals ( iter - > second . cloud , 6 ) ;
2016-08-01 16:00:58 -04:00
2016-09-07 19:59:25 -04:00
pcl : : PointCloud < pcl : : PointXYZRGBNormal > : : Ptr cloudWithNormals ( new pcl : : PointCloud < pcl : : PointXYZRGBNormal > ) ;
2016-08-01 16:00:58 -04:00
pcl : : concatenateFields ( * iter - > second . cloud , * normals , * cloudWithNormals ) ;
2016-04-02 15:17:40 -04:00
2016-03-21 20:03:03 -04:00
// polygons
2016-09-07 19:59:25 -04:00
UASSERT ( iter - > second . polygons . size ( ) ) ;
unsigned int polygonSize = iter - > second . polygons . front ( ) . vertices . size ( ) ;
textureMesh . tex_polygons [ oi ] . resize ( iter - > second . polygons . size ( ) ) ;
textureMesh . tex_coordinates [ oi ] . resize ( iter - > second . polygons . size ( ) * polygonSize ) ;
for ( unsigned int j = 0 ; j < iter - > second . polygons . size ( ) ; + + j )
2016-03-21 20:03:03 -04:00
{
2016-09-07 19:59:25 -04:00
pcl : : Vertices vertices = iter - > second . polygons [ j ] ;
2016-03-21 20:03:03 -04:00
UASSERT ( polygonSize = = vertices . vertices . size ( ) ) ;
for ( unsigned int k = 0 ; k < vertices . vertices . size ( ) ; + + k )
{
//uv
2016-09-07 19:59:25 -04:00
UASSERT ( vertices . vertices [ k ] < iter - > second . denseToOrganizedIndices . size ( ) ) ;
int originalVertex = iter - > second . denseToOrganizedIndices [ vertices . vertices [ k ] ] ;
2016-03-21 20:03:03 -04:00
textureMesh . tex_coordinates [ oi ] [ j * vertices . vertices . size ( ) + k ] = Eigen : : Vector2f (
2016-09-07 19:59:25 -04:00
float ( originalVertex % iter - > second . width ) / float ( iter - > second . width ) , // u
float ( iter - > second . height - originalVertex / iter - > second . width ) / float ( iter - > second . height ) ) ; // v
2016-03-21 20:03:03 -04:00
vertices . vertices [ k ] + = polygonsStep ;
}
textureMesh . tex_polygons [ oi ] [ j ] = vertices ;
}
2016-09-07 19:59:25 -04:00
totalPolygons + = iter - > second . polygons . size ( ) ;
polygonsStep + = iter - > second . cloud - > size ( ) ;
2016-03-21 20:03:03 -04:00
2016-09-07 19:59:25 -04:00
pcl : : PointCloud < pcl : : PointXYZRGBNormal > : : Ptr transformedCloud = rtabmap : : util3d : : transformPointCloud ( cloudWithNormals , iter - > second . pose ) ;
2016-03-21 20:03:03 -04:00
if ( mergedClouds - > size ( ) = = 0 )
{
* mergedClouds = * transformedCloud ;
}
else
{
* mergedClouds + = * transformedCloud ;
}
textures [ oi ] = iter - > second . texture ;
textureMesh . tex_materials [ oi ] . tex_illum = 1 ;
textureMesh . tex_materials [ oi ] . tex_name = uFormat ( " material_%d " , iter - > first ) ;
+ + oi ;
}
else
{
UERROR ( " Texture not set for mesh %d " , iter - > first ) ;
}
2016-02-15 19:35:24 -05:00
}
2016-03-21 20:03:03 -04:00
textureMesh . tex_materials . resize ( oi ) ;
textureMesh . tex_polygons . resize ( oi ) ;
textures . resize ( oi ) ;
if ( textures . size ( ) )
2016-02-15 19:35:24 -05:00
{
2016-03-21 20:03:03 -04:00
pcl : : toPCLPointCloud2 ( * mergedClouds , textureMesh . cloud ) ;
std : : string textureDirectory = uSplit ( filePath , ' . ' ) . front ( ) ;
UINFO ( " Saving %d textures to %s. " , textures . size ( ) , textureDirectory . c_str ( ) ) ;
UDirectory : : makeDir ( textureDirectory ) ;
for ( unsigned int i = 0 ; i < textures . size ( ) ; + + i )
{
2016-11-21 17:09:04 -05:00
cv : : Mat rawImage = textures [ i ] . rows > 1 ? textures [ i ] : rtabmap : : uncompressImage ( textures [ i ] ) ;
2016-03-21 20:03:03 -04:00
std : : string texFile = textureDirectory + " / " + textureMesh . tex_materials [ i ] . tex_name + " .png " ;
cv : : imwrite ( texFile , rawImage ) ;
UINFO ( " Saved %s (%d bytes). " , texFile . c_str ( ) , rawImage . total ( ) * rawImage . channels ( ) ) ;
// relative path
textureMesh . tex_materials [ i ] . tex_file = uSplit ( UFile : : getName ( filePath ) , ' . ' ) . front ( ) + " / " + textureMesh . tex_materials [ i ] . tex_name + " .png " ;
}
2016-04-09 18:57:49 -04:00
UINFO ( " Saving obj (%d vertices, %d polygons) to %s. " , ( int ) mergedClouds - > size ( ) , totalPolygons , filePath . c_str ( ) ) ;
2016-03-21 20:03:03 -04:00
success = pcl : : io : : saveOBJFile ( filePath , textureMesh ) = = 0 ;
2016-04-04 13:03:43 -04:00
if ( success )
{
UINFO ( " Saved obj to %s! " , filePath . c_str ( ) ) ;
}
else
{
UERROR ( " Failed saving obj to %s! " , filePath . c_str ( ) ) ;
}
2016-02-15 19:35:24 -05:00
}
}
}
2016-03-21 20:03:03 -04:00
else
2016-02-15 19:35:24 -05:00
{
2016-04-02 15:17:40 -04:00
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr mergedClouds ( new pcl : : PointCloud < pcl : : PointXYZRGB > ) ;
2016-03-21 20:03:03 -04:00
std : : vector < pcl : : Vertices > mergedPolygons ;
2016-02-15 19:35:24 -05:00
2016-03-21 20:03:03 -04:00
{
boost : : mutex : : scoped_lock lock ( meshesMutex_ ) ;
2016-02-15 19:35:24 -05:00
2016-03-21 20:03:03 -04:00
for ( std : : map < int , Mesh > : : iterator iter = createdMeshes_ . begin ( ) ;
iter ! = createdMeshes_ . end ( ) ;
+ + iter )
{
2016-09-07 19:59:25 -04:00
pcl : : PointCloud < pcl : : PointXYZRGB > : : Ptr transformedCloud = rtabmap : : util3d : : transformPointCloud ( iter - > second . cloud , iter - > second . pose ) ;
2016-03-21 20:03:03 -04:00
if ( mergedClouds - > size ( ) = = 0 )
{
* mergedClouds = * transformedCloud ;
2016-09-07 19:59:25 -04:00
mergedPolygons = iter - > second . polygons ;
2016-03-21 20:03:03 -04:00
}
else
{
2016-09-07 19:59:25 -04:00
rtabmap : : util3d : : appendMesh ( * mergedClouds , mergedPolygons , * transformedCloud , iter - > second . polygons ) ;
2016-03-21 20:03:03 -04:00
}
}
}
if ( mergedClouds - > size ( ) & & mergedPolygons . size ( ) )
{
pcl : : PolygonMesh mesh ;
pcl : : toPCLPointCloud2 ( * mergedClouds , mesh . cloud ) ;
mesh . polygons = mergedPolygons ;
2016-04-09 18:57:49 -04:00
UINFO ( " Saving ply (%d vertices, %d polygons) to %s. " , ( int ) mergedClouds - > size ( ) , ( int ) mergedPolygons . size ( ) , filePath . c_str ( ) ) ;
2016-03-21 20:03:03 -04:00
success = pcl : : io : : savePLYFileBinary ( filePath , mesh ) = = 0 ;
2016-04-04 13:03:43 -04:00
if ( success )
{
UINFO ( " Saved ply to %s! " , filePath . c_str ( ) ) ;
}
else
{
UERROR ( " Failed saving ply to %s! " , filePath . c_str ( ) ) ;
}
2016-03-21 20:03:03 -04:00
}
2016-02-15 19:35:24 -05:00
}
return success ;
}
2016-04-04 13:03:43 -04:00
int RTABMapApp : : postProcessing ( int approach )
2016-04-03 21:35:30 -04:00
{
2016-09-02 22:37:12 -04:00
LOGI ( " postProcessing(%d) " , approach ) ;
2016-04-04 13:03:43 -04:00
int returnedValue = 0 ;
2016-04-03 21:35:30 -04:00
if ( rtabmap_ )
{
std : : map < int , rtabmap : : Transform > poses ;
std : : multimap < int , rtabmap : : Link > links ;
2016-09-02 22:37:12 -04:00
// detect more loop closures
if ( approach = = - 1 | | approach = = 2 )
2016-04-03 21:35:30 -04:00
{
2016-09-02 22:37:12 -04:00
// detect more loop closures
2016-11-21 17:09:04 -05:00
returnedValue = rtabmap_ - > detectMoreLoopClosures ( 1.0f , M_PI / 6.0f , approach = = - 1 ? 5 : 1 ) ;
2016-09-02 22:37:12 -04:00
}
2016-04-04 13:03:43 -04:00
2016-09-02 22:37:12 -04:00
// ICP refining
2016-11-21 17:09:04 -05:00
if ( returnedValue > = 0 & & approach = = 3 )
2016-09-02 22:37:12 -04:00
{
rtabmap : : ParametersMap parameters ;
parameters . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRegStrategy ( ) , std : : string ( " 1 " ) ) ) ; // ICP
rtabmap_ - > parseParameters ( parameters ) ;
int r = rtabmap_ - > refineLinks ( ) ;
if ( approach = = 3 )
2016-04-04 13:03:43 -04:00
{
2016-09-02 22:37:12 -04:00
returnedValue = r ;
2016-04-04 13:03:43 -04:00
}
2016-09-02 22:37:12 -04:00
// reset back default registration (visual)
uInsert ( parameters , rtabmap : : ParametersPair ( rtabmap : : Parameters : : kRegStrategy ( ) , std : : string ( driftCorrection_ ? " 1 " : " 0 " ) ) ) ; // Visual
rtabmap_ - > parseParameters ( parameters ) ;
2016-04-03 21:35:30 -04:00
}
2016-09-02 22:37:12 -04:00
// graph optimization
if ( returnedValue > = 0 )
2016-04-03 21:35:30 -04:00
{
2016-09-02 22:37:12 -04:00
if ( approach = = 1 )
2016-04-03 21:35:30 -04:00
{
2016-09-02 22:37:12 -04:00
if ( rtabmap : : Optimizer : : isAvailable ( rtabmap : : Optimizer : : kTypeG2O ) )
{
std : : map < int , rtabmap : : Signature > signatures ;
rtabmap_ - > getGraph ( poses , links , false , true , & signatures ) ;
rtabmap : : ParametersMap param ;
param . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kOptimizerIterations ( ) , " 30 " ) ) ;
param . insert ( rtabmap : : ParametersPair ( rtabmap : : Parameters : : kOptimizerEpsilon ( ) , " 0 " ) ) ;
rtabmap : : Optimizer * sba = rtabmap : : Optimizer : : create ( rtabmap : : Optimizer : : kTypeG2O , param ) ;
poses = sba - > optimizeBA ( poses . rbegin ( ) - > first , poses , links , signatures ) ;
delete sba ;
}
else
{
LOGE ( " g2o not available! " ) ;
}
2016-04-03 21:35:30 -04:00
}
2016-11-04 16:39:52 -04:00
else if ( approach ! = 4 & & approach ! = 5 & & approach ! = 7 )
2016-04-04 13:03:43 -04:00
{
2016-09-02 22:37:12 -04:00
// simple graph optmimization
rtabmap_ - > getGraph ( poses , links , true , true ) ;
2016-04-04 13:03:43 -04:00
}
}
2016-04-03 21:35:30 -04:00
if ( poses . size ( ) )
{
boost : : mutex : : scoped_lock lock ( rtabmapMutex_ ) ;
2016-04-09 18:57:49 -04:00
rtabmap : : Statistics stats = rtabmap_ - > getStatistics ( ) ;
2016-04-03 21:35:30 -04:00
stats . setPoses ( poses ) ;
stats . setConstraints ( links ) ;
rtabmapEvents_ . push_back ( stats ) ;
rtabmap_ - > setOptimizedPoses ( poses ) ;
}
2016-11-04 16:39:52 -04:00
else if ( approach ! = 4 & & approach ! = 5 & & approach ! = 7 )
2016-04-04 13:03:43 -04:00
{
returnedValue = - 1 ;
}
2016-09-03 20:38:27 -04:00
2016-09-07 17:19:21 -04:00
if ( returnedValue > = 0 )
{
2016-11-04 16:39:52 -04:00
boost : : mutex : : scoped_lock lock ( renderingMutex_ ) ;
2016-09-07 17:19:21 -04:00
// filter polygons
2016-11-21 17:09:04 -05:00
if ( approach = = 4 )
2016-09-07 17:19:21 -04:00
{
filterPolygonsOnNextRender_ = true ;
}
// gain compensation
2016-10-27 15:35:19 -04:00
if ( approach = = - 1 | | approach = = 5 | | approach = = 6 )
2016-09-07 17:19:21 -04:00
{
2016-10-27 15:35:19 -04:00
gainCompensationOnNextRender_ = approach = = 6 ? 2 : 1 ; // 2 = full, 1 = fast
2016-09-07 17:19:21 -04:00
}
2016-11-04 16:39:52 -04:00
// bilateral filtering
2016-11-04 16:46:30 -04:00
if ( approach = = - 1 | | approach = = 7 )
2016-11-04 16:39:52 -04:00
{
bilateralFilteringOnNextRender_ = true ;
}
2016-09-07 17:19:21 -04:00
}
2016-09-03 20:38:27 -04:00
}
2016-04-04 13:03:43 -04:00
return returnedValue ;
2016-04-03 21:35:30 -04:00
}
2016-02-15 19:35:24 -05:00
void RTABMapApp : : handleEvent ( UEvent * event )
{
if ( camera_ & & camera_ - > isRunning ( ) )
{
// called from events manager thread, so protect the data
if ( event - > getClassName ( ) . compare ( " OdometryEvent " ) = = 0 )
{
2016-04-09 18:57:49 -04:00
LOGI ( " Received OdometryEvent! " ) ;
2016-02-15 19:35:24 -05:00
if ( odomMutex_ . try_lock ( ) )
{
odomEvents_ . clear ( ) ;
if ( camera_ - > isRunning ( ) )
{
odomEvents_ . push_back ( * ( ( rtabmap : : OdometryEvent * ) ( event ) ) ) ;
}
odomMutex_ . unlock ( ) ;
}
}
if ( status_ . first = = rtabmap : : RtabmapEventInit : : kInitialized & &
event - > getClassName ( ) . compare ( " RtabmapEvent " ) = = 0 )
{
2016-04-09 18:57:49 -04:00
LOGI ( " Received RtabmapEvent! " ) ;
if ( camera_ - > isRunning ( ) )
2016-02-15 19:35:24 -05:00
{
2016-04-09 18:57:49 -04:00
boost : : mutex : : scoped_lock lock ( rtabmapMutex_ ) ;
rtabmapEvents_ . push_back ( ( ( rtabmap : : RtabmapEvent * ) event ) - > getStats ( ) ) ;
2016-02-15 19:35:24 -05:00
}
}
}
if ( event - > getClassName ( ) . compare ( " PoseEvent " ) = = 0 )
{
if ( poseMutex_ . try_lock ( ) )
{
poseEvents_ . clear ( ) ;
poseEvents_ . push_back ( ( ( rtabmap : : PoseEvent * ) event ) - > pose ( ) ) ;
poseMutex_ . unlock ( ) ;
}
}
if ( event - > getClassName ( ) . compare ( " CameraTangoEvent " ) = = 0 )
{
rtabmap : : CameraTangoEvent * tangoEvent = ( rtabmap : : CameraTangoEvent * ) event ;
// Call JAVA callback with tango event msg
bool success = false ;
if ( jvm & & RTABMapActivity )
{
JNIEnv * env = 0 ;
jint rs = jvm - > AttachCurrentThread ( & env , NULL ) ;
if ( rs = = JNI_OK & & env )
{
jclass clazz = env - > GetObjectClass ( RTABMapActivity ) ;
if ( clazz )
{
jmethodID methodID = env - > GetMethodID ( clazz , " tangoEventCallback " , " (ILjava/lang/String;Ljava/lang/String;)V " ) ;
if ( methodID )
{
env - > CallVoidMethod ( RTABMapActivity , methodID ,
tangoEvent - > type ( ) ,
env - > NewStringUTF ( tangoEvent - > key ( ) . c_str ( ) ) ,
env - > NewStringUTF ( tangoEvent - > value ( ) . c_str ( ) ) ) ;
success = true ;
}
}
}
jvm - > DetachCurrentThread ( ) ;
}
if ( ! success )
{
UERROR ( " Failed to call RTABMapActivity::tangoEventCallback " ) ;
}
}
if ( event - > getClassName ( ) . compare ( " RtabmapEventInit " ) = = 0 )
{
2016-04-09 18:57:49 -04:00
LOGI ( " Received RtabmapEventInit! " ) ;
2016-02-15 19:35:24 -05:00
status_ . first = ( ( rtabmap : : RtabmapEventInit * ) event ) - > getStatus ( ) ;
status_ . second = ( ( rtabmap : : RtabmapEventInit * ) event ) - > getInfo ( ) ;
// Call JAVA callback with init msg
bool success = false ;
if ( jvm & & RTABMapActivity )
{
JNIEnv * env = 0 ;
jint rs = jvm - > AttachCurrentThread ( & env , NULL ) ;
if ( rs = = JNI_OK & & env )
{
jclass clazz = env - > GetObjectClass ( RTABMapActivity ) ;
if ( clazz )
{
jmethodID methodID = env - > GetMethodID ( clazz , " rtabmapInitEventCallback " , " (ILjava/lang/String;)V " ) ;
if ( methodID )
{
env - > CallVoidMethod ( RTABMapActivity , methodID ,
status_ . first ,
env - > NewStringUTF ( status_ . second . c_str ( ) ) ) ;
success = true ;
}
}
}
jvm - > DetachCurrentThread ( ) ;
}
if ( ! success )
{
UERROR ( " Failed to call RTABMapActivity::rtabmapInitEventsCallback " ) ;
}
}
2016-04-09 18:57:49 -04:00
if ( event - > getClassName ( ) . compare ( " PostRenderEvent " ) = = 0 )
{
LOGI ( " Received PostRenderEvent! " ) ;
const rtabmap : : Statistics & stats = ( ( PostRenderEvent * ) event ) - > getStats ( ) ;
int nodes = ( int ) uValue ( stats . data ( ) , rtabmap : : Statistics : : kMemoryWorking_memory_size ( ) , 0.0f ) +
uValue ( stats . data ( ) , rtabmap : : Statistics : : kMemoryShort_time_memory_size ( ) , 0.0f ) ;
int words = ( int ) uValue ( stats . data ( ) , rtabmap : : Statistics : : kKeypointDictionary_size ( ) , 0.0f ) ;
float updateTime = uValue ( stats . data ( ) , rtabmap : : Statistics : : kTimingTotal ( ) , 0.0f ) ;
int loopClosureId = stats . loopClosureId ( ) > 0 ? stats . loopClosureId ( ) : stats . proximityDetectionId ( ) > 0 ? stats . proximityDetectionId ( ) : 0 ;
2016-04-11 10:43:28 -04:00
int highestHypId = ( int ) uValue ( stats . data ( ) , rtabmap : : Statistics : : kLoopHighest_hypothesis_id ( ) , 0.0f ) ;
2016-04-09 18:57:49 -04:00
int databaseMemoryUsed = ( int ) uValue ( stats . data ( ) , rtabmap : : Statistics : : kMemoryDatabase_memory_used ( ) , 0.0f ) ;
int inliers = ( int ) uValue ( stats . data ( ) , rtabmap : : Statistics : : kLoopVisual_inliers ( ) , 0.0f ) ;
2016-04-11 10:43:28 -04:00
int rejected = ( int ) uValue ( stats . data ( ) , rtabmap : : Statistics : : kLoopRejectedHypothesis ( ) , 0.0f ) ;
2016-04-09 18:57:49 -04:00
int featuresExtracted = stats . getSignatures ( ) . size ( ) ? stats . getSignatures ( ) . rbegin ( ) - > second . getWords ( ) . size ( ) : 0 ;
float hypothesis = uValue ( stats . data ( ) , rtabmap : : Statistics : : kLoopHighest_hypothesis_value ( ) , 0.0f ) ;
// Call JAVA callback with some stats
UINFO ( " Send statistics to GUI " ) ;
bool success = false ;
if ( jvm & & RTABMapActivity )
{
JNIEnv * env = 0 ;
jint rs = jvm - > AttachCurrentThread ( & env , NULL ) ;
if ( rs = = JNI_OK & & env )
{
jclass clazz = env - > GetObjectClass ( RTABMapActivity ) ;
if ( clazz )
{
2016-05-12 14:34:51 -04:00
jmethodID methodID = env - > GetMethodID ( clazz , " updateStatsCallback " , " (IIIIFIIIIIFIFI)V " ) ;
2016-04-09 18:57:49 -04:00
if ( methodID )
{
env - > CallVoidMethod ( RTABMapActivity , methodID ,
nodes ,
words ,
totalPoints_ ,
totalPolygons_ ,
updateTime ,
loopClosureId ,
2016-04-11 10:43:28 -04:00
highestHypId ,
2016-04-09 18:57:49 -04:00
databaseMemoryUsed ,
inliers ,
featuresExtracted ,
hypothesis ,
2016-04-11 10:43:28 -04:00
lastDrawnCloudsCount_ ,
2016-09-08 11:55:44 -04:00
renderingTime_ > 0.0f ? 1.0f / renderingTime_ : 0.0f ,
2016-04-11 10:43:28 -04:00
rejected ) ;
2016-04-09 18:57:49 -04:00
success = true ;
}
}
}
jvm - > DetachCurrentThread ( ) ;
}
if ( ! success )
{
UERROR ( " Failed to call RTABMapActivity::updateStatsCallback " ) ;
}
2016-09-08 11:55:44 -04:00
renderingTime_ = 0.0f ;
2016-04-09 18:57:49 -04:00
}
2016-02-15 19:35:24 -05:00
}