VTK/Examples/Cxx/Math/HomogeneousLeastSquares

From KitwarePublic

Jump to: navigation, search

HomogeneousLeastSquares.cxx

#include <vtkSmartPointer.h>
#include <vtkMath.h>
 
  /* allocate memory for an nrow x ncol matrix */
  template<class TReal>
      TReal **create_matrix ( long nrow, long ncol )
{
  typedef TReal* TRealPointer;
  TReal **m = new TRealPointer[nrow];
 
  TReal* block = ( TReal* ) calloc ( nrow*ncol, sizeof ( TReal ) );
  m[0] = block;
  for ( int row = 1; row < nrow; ++row )
  {
    m[ row ] = &block[ row * ncol ];
  }
  return m;
}
 
  /* free a TReal matrix allocated with matrix() */
  template<class TReal>
      void free_matrix ( TReal **m )
{
  free ( m[0] );
  delete[] m;
}
 
int main (int, char *[])
{
 
  //Solve XM = Y;
 
  int numberOfSamples = 3;
  int numberOfVariables = 2;
  double **x = create_matrix<double> (numberOfSamples, numberOfVariables);
  x[0][0] = 1; x[0][1] = 2;
  x[1][0] = 2; x[1][1] = 4;
  x[2][0] = 3; x[2][1] = 6;
 
  double **m = create_matrix<double> ( numberOfVariables, 1 );
 
  vtkMath::SolveHomogeneousLeastSquares(numberOfSamples, x, numberOfVariables, m);
 
  std::cout << "Solution is: " << m[0][0] << " " << m[1][0] << std::endl;
 
  free_matrix(x);
  free_matrix(m);
 
  return EXIT_SUCCESS;
}

CMakeLists.txt

cmake_minimum_required(VERSION 2.6)
PROJECT(HomogeneousLeastSquares)
 
FIND_PACKAGE(VTK REQUIRED)
INCLUDE(${VTK_USE_FILE})
 
ADD_EXECUTABLE(HomogeneousLeastSquares HomogeneousLeastSquares.cxx)
TARGET_LINK_LIBRARIES(HomogeneousLeastSquares vtkHybrid)
Personal tools