You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.3 KiB
79 lines
2.3 KiB
1 year ago
|
#include "min.h"
|
||
|
|
#include "for_each.h"
|
||
|
|
#include "find_zero.h"
|
||
|
|
|
||
|
|
template <typename AType, typename DerivedB, typename DerivedI>
|
||
|
|
IGL_INLINE void igl::min(
|
||
|
|
const Eigen::SparseMatrix<AType> & A,
|
||
|
|
const int dim,
|
||
|
|
Eigen::PlainObjectBase<DerivedB> & B,
|
||
|
|
Eigen::PlainObjectBase<DerivedI> & I)
|
||
|
|
{
|
||
|
|
const int n = A.cols();
|
||
|
|
const int m = A.rows();
|
||
|
|
B.resize(dim==1?n:m);
|
||
|
|
B.setConstant(std::numeric_limits<typename DerivedB::Scalar>::max());
|
||
|
|
I.resize(dim==1?n:m);
|
||
|
|
for_each(A,[&B,&I,&dim](int i, int j,const typename DerivedB::Scalar v)
|
||
|
|
{
|
||
|
|
if(dim == 2)
|
||
|
|
{
|
||
|
|
std::swap(i,j);
|
||
|
|
}
|
||
|
|
// Coded as if dim == 1, assuming swap for dim == 2
|
||
|
|
if(v < B(j))
|
||
|
|
{
|
||
|
|
B(j) = v;
|
||
|
|
I(j) = i;
|
||
|
|
}
|
||
|
|
});
|
||
|
|
Eigen::VectorXi Z;
|
||
|
|
find_zero(A,dim,Z);
|
||
|
|
for(int j = 0;j<I.size();j++)
|
||
|
|
{
|
||
|
|
if(Z(j) != (dim==1?m:n) && 0 < B(j))
|
||
|
|
{
|
||
|
|
B(j) = 0;
|
||
|
|
I(j) = Z(j);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
template <typename DerivedX, typename DerivedY, typename DerivedI>
|
||
|
|
IGL_INLINE void igl::min(
|
||
|
|
const Eigen::DenseBase<DerivedX> & X,
|
||
|
|
const int dim,
|
||
|
|
Eigen::PlainObjectBase<DerivedY> & Y,
|
||
|
|
Eigen::PlainObjectBase<DerivedI> & I)
|
||
|
|
{
|
||
|
|
assert(dim==1||dim==2);
|
||
|
|
|
||
|
|
// output size
|
||
|
|
int n = (dim==1?X.cols():X.rows());
|
||
|
|
// resize output
|
||
|
|
Y.resize(n,1);
|
||
|
|
I.resize(n,1);
|
||
|
|
|
||
|
|
// loop over dimension opposite of dim
|
||
|
|
for(int j = 0;j<n;j++)
|
||
|
|
{
|
||
|
|
typename DerivedX::Index PHONY,i;
|
||
|
|
typename DerivedX::Scalar m;
|
||
|
|
if(dim==1)
|
||
|
|
{
|
||
|
|
m = X.col(j).minCoeff(&i,&PHONY);
|
||
|
|
}else
|
||
|
|
{
|
||
|
|
m = X.row(j).minCoeff(&PHONY,&i);
|
||
|
|
}
|
||
|
|
Y(j) = m;
|
||
|
|
I(j) = i;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef IGL_STATIC_LIBRARY
|
||
|
|
// Explicit template instantiation
|
||
|
|
template void igl::min<Eigen::Array<bool, -1, 3, 0, -1, 3>, Eigen::Array<bool, -1, 1, 0, -1, 1>, Eigen::Matrix<int, -1, 1, 0, -1, 1> >(Eigen::DenseBase<Eigen::Array<bool, -1, 3, 0, -1, 3> > const&, int, Eigen::PlainObjectBase<Eigen::Array<bool, -1, 1, 0, -1, 1> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, 1, 0, -1, 1> >&);
|
||
|
|
template void igl::min<Eigen::Matrix<double, -1, -1, 0, -1, -1>, Eigen::Matrix<double, -1, 1, 0, -1, 1>, Eigen::Matrix<int, -1, 1, 0, -1, 1> >(Eigen::DenseBase<Eigen::Matrix<double, -1, -1, 0, -1, -1> > const&, int, Eigen::PlainObjectBase<Eigen::Matrix<double, -1, 1, 0, -1, 1> >&, Eigen::PlainObjectBase<Eigen::Matrix<int, -1, 1, 0, -1, 1> >&);
|
||
|
|
#endif
|