model_search.cpp

Go to the documentation of this file.

/* +------------------------------------------------------------------------+
   |                     Mobile Robot Programming Toolkit (MRPT)            |
   |                          https://www.mrpt.org/                         |
   |                                                                        |
   | Copyright (c) 2005-2020, Individual contributors, see AUTHORS file     |
   | See: https://www.mrpt.org/Authors - All rights reserved.               |
   | Released under BSD License. See: https://www.mrpt.org/License          |
   +------------------------------------------------------------------------+ */

#include "math-precomp.h"  // Precompiled headers

#include <mrpt/math/model_search.h>
#include <mrpt/random/random_shuffle.h>

using namespace mrpt;
using namespace mrpt::math;

//----------------------------------------------------------------------
//! Select random (unique) indices from the 0..p_size sequence
void ModelSearch::pickRandomIndex(
    size_t p_size, size_t p_pick, std::vector<size_t>& p_ind)
{
    ASSERT_(p_size >= p_pick);

    std::vector<size_t> a(p_size);
    for (size_t i = 0; i < p_size; i++) a[i] = i;

    mrpt::random::shuffle(a.begin(), a.end());
    p_ind.resize(p_pick);
    for (size_t i = 0; i < p_pick; i++) p_ind[i] = a[i];
}

//----------------------------------------------------------------------
//! Select random (unique) indices from the set.
//! The set is destroyed during pick
void ModelSearch::pickRandomIndex(
    std::set<size_t> p_set, size_t p_pick, std::vector<size_t>& p_ind)
{
    p_ind.resize(p_pick);
    std::vector<size_t> inds(p_set.begin(), p_set.end());

    mrpt::random::shuffle(inds.begin(), inds.end());
    p_ind.resize(p_pick);
    for (size_t i = 0; i < p_pick; i++) p_ind[i] = inds[i];
}