mitkVnlVectorFromCvMat.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
#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