mitkCvMatFromVnlMatrix.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 mitkCvMatFromVnlMatrix_h
#define mitkCvMatFromVnlMatrix_h

#include <mitkEndoMacros.h>
#include <mitkEndoDebug.h>
#include <vnl/vnl_matrix.h>
#include <mitkAlgorithm.h>

#include "opencv2/core.hpp"

namespace mitk
{
  template <class T>
  class CvMatFromVnlMatrix:
    virtual public Algorithm
  {
  public:
    CvMatFromVnlMatrix(
      const vnl_matrix<T>* _VnlMatrix,
      cv::Mat* _CvMat):
        m_VnlMatrix(_VnlMatrix),
        m_CvMat(_CvMat)
    {
    }

    template <class T_CvType, class T_VnlType>
    static void Cast( const vnl_matrix<T_VnlType>& vnlMat, cv::Mat& cvMat )
    {
      endodebugvar( vnlMat )
      cvMat = cv::Mat(vnlMat.rows(), vnlMat.cols(), cv::DataType<T_CvType>::type);
      for(unsigned int i=0; i<vnlMat.rows(); ++i)
      {
        for(unsigned int j=0; j<vnlMat.cols(); ++j)
        {
          cvMat.at<T_CvType>(i,j) = static_cast<T_CvType>( vnlMat(i,j) );
//          endodebugvar( vnlMat(i,j) )
//          endodebugvar( cvMat.at<T_CvType>(i,j) )
        }
      }
    }

    void Update() override
    {
      Cast<T, T>( *m_VnlMatrix, *m_CvMat );
    }
  private:
    const vnl_matrix<T>* m_VnlMatrix;
    cv::Mat* m_CvMat;
  };
} // namespace mitk

#endif // mitkCvMatFromVnlMatrix_h