mitkVnlVectorFromCvMat.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/
#ifndef mitkVnlVectorFromCvMat_h
#define mitkVnlVectorFromCvMat_h

#include <mitkVnlMatrixFromCvMat.h>
#include <vnl/vnl_vector.h>

namespace mitk
{
  template <class T>
  class VnlVectorFromCvMat:
    virtual public Algorithm
  {
  public:
    VnlVectorFromCvMat(
      const cv::Mat* _CvMat,
      vnl_vector<T>* _VnlVector):
        m_VnlMatrixFromCvMat( _CvMat, &m_VnlMatrix ),
        m_VnlVector(_VnlVector)
    {
    }

    void Update() override
    {
      m_VnlMatrixFromCvMat.Update();

      if( m_VnlMatrix.rows() == 1 )
        *m_VnlVector = m_VnlMatrix.get_row(0);
      else if( m_VnlMatrix.cols() == 1 )
        *m_VnlVector = m_VnlMatrix.get_column(0);

    }
  private:
    vnl_matrix<T> m_VnlMatrix;
    VnlMatrixFromCvMat<T> m_VnlMatrixFromCvMat;
    vnl_vector<T>* m_VnlVector;
  };
} // namespace mitk

#endif // mitkVnlVectorFromCvMat_h