CODEWITHSUNDEEP
c++sortingeigen
Jendas
Jendas

Reputation: 3579

Eigen - Sort matrix diagonal

What I have is diagonal matrix of type Eigen::MatrixXi. I need elements on the diagonal to be sorted in ascending order. For example like this:

2 0 0      1 0 0
0 7 0  >>> 0 2 0
0 0 1      0 0 7

I thought I would simply do:

std::sort(matrix.diagonal().begin(), matrix.diagonal().end());

But apparently Eigen::Diagonal does not have begin nor end function. So the question is, is there any way of sorting elements on diagonal using internal std::sort or anything similarly elegant?

I went through official documentation but did not find anything useful.

Upvotes: 7

Views: 1638

Answers (2)

chtz
chtz

Reputation: 18807

There is no native support for sorting matrices as of today. There are two long-pending feature requests relevant to this functionality:

As suggested by @NicolasM in the comments, currently, the most elegant solution is to provide custom iterators yourself, e.g.:

namespace Eigen {
  template<typename Scalar>
  class iterator {
    Scalar* ptr;
    int stride;
  public: 
    iterator(Scalar* ptr_, int stride_) : ptr(ptr_), stride(stride_) {}
    Scalar& operator*() const { return *ptr;}
    iterator& operator++() { ptr += stride; return *this;}
    bool operator<(const iterator& other) { return ptr < other.ptr; }
    // implement further operations, required for random access iterators ...
  }

  template<class Derived>
  iterator begin(MatrixBase<Derived>&& mat)
  { return iterator(mat.data(), mat.innerStride()); }
  template<class Derivde>
  iterator end(MatrixBase<Derived>&& mat)
  { return iterator(mat.data() + mat.size()*mat.innerStride(), mat.innerStride()); }

} // namespace Eigen

Upvotes: 1

user1935024
user1935024

Reputation:

What I tried to say is something like this. 

void insertion_sort(int arr[3][3], int length) {

int i, j, tmp;
for (i = 1; i < length; i++) {
    j = i;
    while (j > 0 && arr[j - 1][j- 1] > arr[j][j]) {
        tmp = arr[j][j];
        arr[j][j] = arr[j - 1][j - 1];
        arr[j - 1][j - 1] = tmp;
        j--;
    }       
}
}

int main() {    

int a[3][3] = { { 2, 0, 0 }, { 0, 7, 0 }, { 0, 0, 1 } };
insertion_sort(a, 3);

for (int i = 0; i < 3; ++i)
{
    for (int j = 0; j < 3; ++j)
        cout << a[i][j] << " ";
    cout << endl;
}

I used insertion sort because it's easy to implement.

Upvotes: 0

Related Questions