58 lines
2.1 KiB
C++
58 lines
2.1 KiB
C++
|
// This file is part of libigl, a simple c++ geometry processing library.
|
||
|
|
//
|
||
|
|
// Copyright (C) 2017 Daniele Panozzo <daniele.panozzo@gmail.com>
|
||
|
|
//
|
||
|
|
// This Source Code Form is subject to the terms of the Mozilla Public License
|
||
|
|
// v. 2.0. If a copy of the MPL was not distributed with this file, You can
|
||
|
|
// obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
#include "slice_cached.h"
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
#include <vector>
|
||
|
|
#include <utility>
|
||
|
|
#include "slice.h"
|
||
|
|
|
||
|
|
template <typename TX, typename TY, typename DerivedI>
|
||
|
|
IGL_INLINE void igl::slice_cached_precompute(
|
||
|
|
const Eigen::SparseMatrix<TX>& X,
|
||
|
|
const Eigen::Matrix<int,Eigen::Dynamic,1> & R,
|
||
|
|
const Eigen::Matrix<int,Eigen::Dynamic,1> & C,
|
||
|
|
Eigen::MatrixBase<DerivedI>& data,
|
||
|
|
Eigen::SparseMatrix<TY>& Y
|
||
|
|
)
|
||
|
|
{
|
||
|
|
// Create a sparse matrix whose entries are the ids
|
||
|
|
Eigen::SparseMatrix<unsigned> TS = X.template cast<unsigned>();
|
||
|
|
|
||
|
|
TS.makeCompressed();
|
||
|
|
for (unsigned i=0;i<TS.nonZeros();++i)
|
||
|
|
*(TS.valuePtr() + i) = i;
|
||
|
|
|
||
|
|
Eigen::SparseMatrix<unsigned> TS_sliced;
|
||
|
|
igl::slice(TS,R,C,TS_sliced);
|
||
|
|
Y = TS_sliced.cast<TY>();
|
||
|
|
|
||
|
|
data.resize(TS_sliced.nonZeros());
|
||
|
|
for (unsigned i=0;i<data.size();++i)
|
||
|
|
{
|
||
|
|
data[i] = *(TS_sliced.valuePtr() + i);
|
||
|
|
*(Y.valuePtr() + i) = *(X.valuePtr() + data[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename TX, typename TY, typename DerivedI>
|
||
|
|
IGL_INLINE void igl::slice_cached(
|
||
|
|
const Eigen::SparseMatrix<TX>& X,
|
||
|
|
const Eigen::MatrixBase<DerivedI>& data,
|
||
|
|
Eigen::SparseMatrix<TY>& Y
|
||
|
|
)
|
||
|
|
{
|
||
|
|
for (unsigned i=0; i<data.size(); ++i)
|
||
|
|
*(Y.valuePtr() + i) = *(X.valuePtr() + data[i]);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef IGL_STATIC_LIBRARY
|
||
|
|
template void igl::slice_cached<double, double, Eigen::Matrix<int, -1, 1, 0, -1, 1> >(Eigen::SparseMatrix<double, 0, int> const&, Eigen::MatrixBase<Eigen::Matrix<int, -1, 1, 0, -1, 1> > const&, Eigen::SparseMatrix<double, 0, int>&);
|
||
|
|
template void igl::slice_cached_precompute<double, double, Eigen::Matrix<int, -1, 1, 0, -1, 1> >(Eigen::SparseMatrix<double, 0, int> const&, Eigen::Matrix<int, -1, 1, 0, -1, 1> const&, Eigen::Matrix<int, -1, 1, 0, -1, 1> const&, Eigen::MatrixBase<Eigen::Matrix<int, -1, 1, 0, -1, 1> >&, Eigen::SparseMatrix<double, 0, int>&);
|
||
|
|
#endif
|