/*---------------------------------------------------------------
CLSLAM_RBPF_2DLASER
Implements a 2D local SLAM method based on a RBPF
over an occupancy grid map. A part of HMT-SLAM.
\param LMH The local metric hypothesis which must be updated by this SLAM
algorithm.
\param act The action to process (or nullptr).
\param sf The observations to process (or nullptr).
WE ALREADY HAVE CONTROL OVER THE CRITICAL SECTION OF THE LMHs!
--------------------------------------------------------------- */
void CLSLAM_RBPF_2DLASER::processOneLMH(
CLocalMetricHypothesis* LMH, const CActionCollection::Ptr& actions,
const CSensoryFrame::Ptr& sf)
{
MRPT_START
// Get the current robot pose estimation:
TPoseID currentPoseID = LMH->m_currentRobotPose;
// If this is the first iteration, just create a new robot pose at the
// origin:
if (currentPoseID == POSEID_INVALID)
{
currentPoseID = CHMTSLAM::generatePoseID();
LMH->m_currentRobotPose = currentPoseID; // Change it in the LMH
// Create a new robot pose:
CPose3D initPose(0, 0, 0);
ASSERT_(LMH->m_particles.size() > 0);
for (auto& m_particle : LMH->m_particles)
m_particle.d->robotPoses[currentPoseID] = initPose;
ASSERT_(m_parent->m_map.nodeCount() == 1);
m_parent->m_map_cs.lock();
CHMHMapNode::Ptr firstArea = m_parent->m_map.getFirstNode();
ASSERT_(firstArea);
LMH->m_nodeIDmemberships[currentPoseID] = firstArea->getID();
// Set annotation for the reference pose:
firstArea->m_annotations.setElemental(
NODE_ANNOTATION_REF_POSEID, currentPoseID, LMH->m_ID);
m_parent->m_map_cs.unlock();
}
bool insertNewRobotPose = false;
if (sf)
{
if (LMH->m_nodeIDmemberships.size() < 2) // If there is just 1 node
// (the current robot pose),
// then there is no
// observation in the map yet!
{ // Update map if this is the first observation!
insertNewRobotPose = true;
}
else
{
// Check minimum distance from current robot pose to those in the
// neighborhood:
TMapPoseID2Pose3D lstRobotPoses;
LMH->getMeans(lstRobotPoses);
CPose3D* currentRobotPose = &lstRobotPoses[currentPoseID];
float minDistLin = 1e6f;
float minDistAng = 1e6f;
// printf("Poses in graph:\n");
for (auto& lstRobotPose : lstRobotPoses)
{
if (lstRobotPose.first != currentPoseID)
{
float linDist =
lstRobotPose.second.distanceTo(*currentRobotPose);
float angDist = fabs(math::wrapToPi(
lstRobotPose.second.yaw() - currentRobotPose->yaw()));
minDistLin = min(minDistLin, linDist);
if (linDist < m_parent->m_options.SLAM_MIN_DIST_BETWEEN_OBS)
minDistAng = min(minDistAng, angDist);
}
}
// time to insert a new node??
insertNewRobotPose =
(minDistLin > m_parent->m_options.SLAM_MIN_DIST_BETWEEN_OBS) ||
(minDistAng > m_parent->m_options.SLAM_MIN_HEADING_BETWEEN_OBS);
}
} // end if there are SF
// Save data in members so PF callback "prediction_and_update_pfXXXX" have
// access to them:
m_insertNewRobotPose = insertNewRobotPose;
// ------------------------------------------------
// Execute RBPF method:
// 1) PROCESS ACTION
// 2) PROCESS OBSERVATIONS
// ------------------------------------------------
CParticleFilter pf;
pf.m_options = m_parent->m_options.pf_options;
pf.executeOn(*LMH, actions.get(), sf.get());
// 3) The appearance observation: update the log likelihood
// ...
// -----------------------------------------------------------
// 4) UPDATE THE MAP
// -----------------------------------------------------------
if (insertNewRobotPose)
{
m_parent->logStr(
mrpt::system::LVL_INFO,
"[CLSLAM_RBPF_2DLASER] Adding new pose...\n");
// Leave the up-to-now "current pose" in the map, insert the SF in it,
// and...
// ----------------------------------------------------------------------------
TPoseID newCurrentPoseID = CHMTSLAM::generatePoseID();
// ...and create a new "current pose" making a copy of the previous
// one:
// and insert the observations into the metric maps:
// ----------------------------------------------------------------------------
for (auto& m_particle : LMH->m_particles)
{
const CPose3D* curRobotPose =
&m_particle.d->robotPoses[currentPoseID];
m_particle.d->robotPoses[newCurrentPoseID] = *curRobotPose;
sf->insertObservationsInto(&m_particle.d->metricMaps, curRobotPose);
}
// Add node membership: for now, a copy of the current pose:
LMH->m_nodeIDmemberships[newCurrentPoseID] =
LMH->m_nodeIDmemberships[currentPoseID];
// Now, insert the SF in the just added robot pose:
// -----------------------------------------------------
LMH->m_SFs[currentPoseID] = *sf;
// Sets the new poseID as current robot pose:
// ----------------------------------------------------
TPoseID newlyAddedPose = currentPoseID;
currentPoseID = LMH->m_currentRobotPose = newCurrentPoseID;
// Mark the "newlyAddedPose" as pending to reconsider in the graph-cut
// method
// (Done in the main loop @ LSLAM thread)
// --------------------------------------------------------------------------------
LMH->m_posesPendingAddPartitioner.push_back(newlyAddedPose);
m_parent->logFmt(
mrpt::system::LVL_INFO, "[CLSLAM_RBPF_2DLASER] Added pose %i.\n",
(int)newlyAddedPose);
// Notice LC detectors:
// ------------------------------
{
std::lock_guard<std::mutex> lock(m_parent->m_topLCdets_cs);
for (auto& m_topLCdet : m_parent->m_topLCdets)
m_topLCdet->OnNewPose(newlyAddedPose, sf.get());
}
} // end of insertNewRobotPose
MRPT_END
}
/*---------------------------------------------------------------
initializeEmptyMap
---------------------------------------------------------------*/
void CHMTSLAM::initializeEmptyMap()
{
THypothesisIDSet LMH_hyps;
THypothesisID newHypothID = generateHypothesisID();
LMH_hyps.insert(COMMON_TOPOLOG_HYP);
LMH_hyps.insert(newHypothID);
// ------------------------------------
// CLEAR HIERARCHICAL MAP
// ------------------------------------
CHMHMapNode::TNodeID firstAreaID;
{
std::lock_guard<std::mutex> lock(m_map_cs);
// Initialize hierarchical structures:
// -----------------------------------------------------
m_map.clear();
// Create a single node for the starting area:
CHMHMapNode::Ptr firstArea =
mrpt::make_aligned_shared<CHMHMapNode>(&m_map);
firstAreaID = firstArea->getID();
firstArea->m_hypotheses = LMH_hyps;
CMultiMetricMap::Ptr emptyMap = CMultiMetricMap::Ptr(
new CMultiMetricMap(&m_options.defaultMapsInitializers));
firstArea->m_nodeType = "Area";
firstArea->m_label = generateUniqueAreaLabel();
firstArea->m_annotations.set(
NODE_ANNOTATION_METRIC_MAPS, emptyMap, newHypothID);
firstArea->m_annotations.setElemental(
NODE_ANNOTATION_REF_POSEID, POSEID_INVALID, newHypothID);
} // end of lock m_map_cs
// ------------------------------------
// CLEAR LIST OF HYPOTHESES
// ------------------------------------
{
std::lock_guard<std::mutex> lock(m_LMHs_cs);
// Add to the list:
m_LMHs.clear();
CLocalMetricHypothesis& newLMH = m_LMHs[newHypothID];
newLMH.m_parent = this;
newLMH.m_currentRobotPose =
POSEID_INVALID; // Special case: map is empty
newLMH.m_log_w = 0;
newLMH.m_ID = newHypothID;
newLMH.m_neighbors.clear();
newLMH.m_neighbors.insert(firstAreaID);
newLMH.clearRobotPoses();
} // end of cs
// ------------------------------------------
// Create the local SLAM algorithm object
// -----------------------------------------
switch (m_options.SLAM_METHOD)
{
case lsmRBPF_2DLASER:
{
if (m_LSLAM_method) delete m_LSLAM_method;
m_LSLAM_method = new CLSLAM_RBPF_2DLASER(this);
}
break;
default:
THROW_EXCEPTION_FMT(
"Invalid selection for LSLAM method: %i",
(int)m_options.SLAM_METHOD);
};
// ------------------------------------
// Topological LC detectors:
// ------------------------------------
{
std::lock_guard<std::mutex> lock(m_topLCdets_cs);
// Clear old list:
for (std::deque<CTopLCDetectorBase*>::iterator it = m_topLCdets.begin();
it != m_topLCdets.end(); ++it)
delete *it;
m_topLCdets.clear();
// Create new list:
// 1: Occupancy Grid matching.
// 2: Cummins' image matching.
for (std::vector<std::string>::const_iterator d =
m_options.TLC_detectors.begin();
d != m_options.TLC_detectors.end(); ++d)
m_topLCdets.push_back(loopClosureDetector_factory(*d));
}
// ------------------------------------
// Other variables:
// ------------------------------------
m_LSLAM_queue.clear();
// ------------------------------------
// Delete log files:
// ------------------------------------
if (!m_options.LOG_OUTPUT_DIR.empty())
{
mrpt::system::deleteFilesInDirectory(m_options.LOG_OUTPUT_DIR);
mrpt::system::createDirectory(m_options.LOG_OUTPUT_DIR);
mrpt::system::createDirectory(m_options.LOG_OUTPUT_DIR + "/HMAP_txt");
mrpt::system::createDirectory(m_options.LOG_OUTPUT_DIR + "/HMAP_3D");
mrpt::system::createDirectory(m_options.LOG_OUTPUT_DIR + "/LSLAM_3D");
mrpt::system::createDirectory(m_options.LOG_OUTPUT_DIR + "/ASSO");
mrpt::system::createDirectory(
m_options.LOG_OUTPUT_DIR + "/HMTSLAM_state");
}
}