rw wrapper

#include "randomwalker.h"

#ifdef USE_HYPRE
#include "randomwalker_hypre.h"
#else
#include "randomwalker_eigen.h"
#endif

#include <assert.h>

namespace NRandomWalker {

void MakeRandomWalk(std::size_t width, std::size_t height,
                    const std::vector<int>& vertices, const std::vector<int>& seedMask,
                    std::vector<int>& resultsMask)
{
    assert(vertices.size() > 0);
    assert(width * height == vertices.size());
    assert(vertices.size() == seedMask.size());

    RandomWalker rw;
    rw.initialize(width, height, 1, (int*)&vertices[0], (int*)&seedMask[0]);
    int* rwMask = rw.segment();
    rw.finalize();

    resultsMask.resize(vertices.size());
    for (size_t i = 0; i < resultsMask.size(); ++i) {
        resultsMask[i] = rwMask[i];
    }
    delete[] rwMask;
}

}