core/vnl/io/vnl_io_matrix.txx

Go to the documentation of this file.

// This is core/vnl/io/vnl_io_matrix.txx
#ifndef vnl_io_matrix_txx_
#define vnl_io_matrix_txx_
//:
// \file

#include "vnl_io_matrix.h"
#include <vnl/vnl_matrix.h>
#include <vsl/vsl_b_read_block_old.h>
#include <vsl/vsl_block_binary.h>
#include <vsl/vsl_indent.h>

//=================================================================================
//: Binary save self to stream.
template<class T>
void vsl_b_write(vsl_b_ostream & os, const vnl_matrix<T> & p)
{
  const short version_no = 2;
  vsl_b_write(os, version_no);
  vsl_b_write(os, p.rows());
  vsl_b_write(os, p.cols());

  // Calling p.begin() on empty matrix causes segfault
  if (p.size()>0)
    vsl_block_binary_write(os, p.begin(), p.size());
}

//=================================================================================
//: Binary load self from stream.
template<class T>
void vsl_b_read(vsl_b_istream &is, vnl_matrix<T> & p)
{
  if (!is) return;

  short v;
  unsigned m, n;
  vsl_b_read(is, v);
  switch (v)
  {
   case 1:
    vsl_b_read(is, m);
    vsl_b_read(is, n);
    p.set_size(m, n);
    // Calling begin() on empty matrix causes segfault
    if (m*n>0)
      vsl_b_read_block_old(is, p.begin(), p.size());
    break;

   case 2:
    vsl_b_read(is, m);
    vsl_b_read(is, n);
    p.set_size(m, n);
    // Calling begin() on empty matrix causes segfault
    if (m*n>0)
      vsl_block_binary_read(is, p.data_block(), p.size());
    break;

   default:
    vcl_cerr << "I/O ERROR: vsl_b_read(vsl_b_istream&, vnl_matrix<T>&)\n"
             << "           Unknown version number "<< v << '\n';
    is.is().clear(vcl_ios::badbit); // Set an unrecoverable IO error on stream
    return;
  }
}

//====================================================================================
//: Output a human readable summary to the stream
template<class T>
void vsl_print_summary(vcl_ostream & os,const vnl_matrix<T> & p)
{
  os<<"Size: "<<p.rows()<<" x "<<p.cols()<<vcl_endl;

  unsigned int m = 5; unsigned int n = 5;


  if (m>p.rows()) m=p.rows();
  if (n>p.cols()) n=p.cols();

  vsl_indent_inc(os);
  for (unsigned int i=0;i<m;i++)
  {
     os<<vsl_indent()<<" (";

     for ( unsigned int j=0; j<n; j++)
        os<<p(i,j)<<' ';
      if (p.cols()>n) os<<"...";
        os<<")\n";
  }
  if (p.rows()>m) os <<vsl_indent()<<" (...\n";
  vsl_indent_dec(os);
}


#define VNL_IO_MATRIX_INSTANTIATE(T) \
template void vsl_print_summary(vcl_ostream &, const vnl_matrix<T > &); \
template void vsl_b_read(vsl_b_istream &, vnl_matrix<T > &); \
template void vsl_b_write(vsl_b_ostream &, const vnl_matrix<T > &)

#endif // vnl_io_matrix_txx_