Eigen::MatrixMarketIterator< Scalar > Class Template Reference

Iterator to browse matrices from a specified folder. More...

Public Types
typedef SparseMatrix< Scalar, ColMajor >	MatrixType
typedef Matrix< Scalar, Dynamic, 1 >	VectorType

Public Member Functions
bool	hasrefX ()
bool	hasRhs ()
bool	isFolderValid ()
std::string &	matname ()
MatrixType &	matrix ()
	MatrixMarketIterator (const std::string &folder)
	operator bool () const
MatrixMarketIterator &	operator++ ()
VectorType &	refX ()
VectorType &	rhs ()
int	sym ()
	~MatrixMarketIterator ()

Protected Member Functions
bool	Fileexists (std::string file)
void	Getnextvalidmatrix ()

Protected Attributes
struct dirent *	m_curs_id
std::string	m_folder
DIR *	m_folder_id
bool	m_hasrefX
bool	m_hasRhs
bool	m_isvalid
MatrixType	m_mat
bool	m_matIsLoaded
std::string	m_matname
VectorType	m_refX
VectorType	m_rhs
int	m_sym

Private Types
typedef NumTraits< Scalar >::Real	RealScalar

Detailed Description

template<typename Scalar>
class Eigen::MatrixMarketIterator< Scalar >

Iterator to browse matrices from a specified folder.

This is used to load all the matrices from a folder. The matrices should be in Matrix Market format It is assumed that the matrices are named as matname.mtx and matname_SPD.mtx if the matrix is Symmetric and positive definite (or Hermitian) The right hand side vectors are loaded as well, if they exist. They should be named as matname_b.mtx. Note that the right hand side for a SPD matrix is named as matname_SPD_b.mtx

Sometimes a reference solution is available. In this case, it should be named as matname_x.mtx

Sample code

Template Parameters

Scalar

The scalar type

Definition at line 44 of file MatrixMarketIterator.h.

Member Typedef Documentation

MatrixType

template<typename Scalar >

typedef SparseMatrix<Scalar,ColMajor> Eigen::MatrixMarketIterator< Scalar >::MatrixType

Definition at line 49 of file MatrixMarketIterator.h.

RealScalar

template<typename Scalar >

typedef NumTraits<Scalar>::Real Eigen::MatrixMarketIterator< Scalar >::RealScalar

private

Definition at line 46 of file MatrixMarketIterator.h.

VectorType

template<typename Scalar >

typedef Matrix<Scalar,Dynamic,1> Eigen::MatrixMarketIterator< Scalar >::VectorType

Definition at line 48 of file MatrixMarketIterator.h.

Constructor & Destructor Documentation

MatrixMarketIterator()

template<typename Scalar >

Eigen::MatrixMarketIterator< Scalar >::MatrixMarketIterator ( const std::string &  folder )

inline

Definition at line 52 of file MatrixMarketIterator.h.

      : m_sym(0), m_isvalid(false), m_matIsLoaded(false), m_hasRhs(false), m_hasrefX(false), m_folder(folder)
    {
      m_folder_id = opendir(folder.c_str());
      if(m_folder_id)
        Getnextvalidmatrix();
    }

~MatrixMarketIterator()

template<typename Scalar >

Eigen::MatrixMarketIterator< Scalar >::~MatrixMarketIterator ( )

inline

Definition at line 60 of file MatrixMarketIterator.h.

    {
      if (m_folder_id) closedir(m_folder_id); 
    }

Member Function Documentation

Fileexists()

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::Fileexists ( std::string  file )

inlineprotected

Definition at line 175 of file MatrixMarketIterator.h.

    {
      std::ifstream file_id(file.c_str());
      if (!file_id.good() ) 
      {
        return false;
      }
      else 
      {
        file_id.close();
        return true;
      }
    }

Getnextvalidmatrix()

template<typename Scalar >

void Eigen::MatrixMarketIterator< Scalar >::Getnextvalidmatrix ( )

inlineprotected

Definition at line 189 of file MatrixMarketIterator.h.

    {
      m_isvalid = false;
      // Here, we return with the next valid matrix in the folder
      while ( (m_curs_id = readdir(m_folder_id)) != NULL) {
        m_isvalid = false;
        std::string curfile;
        curfile = m_folder + "/" + m_curs_id->d_name;
        // Discard if it is a folder
        if (m_curs_id->d_type == DT_DIR) continue; //FIXME This may not be available on non BSD systems
//         struct stat st_buf; 
//         stat (curfile.c_str(), &st_buf);
//         if (S_ISDIR(st_buf.st_mode)) continue;
        
        // Determine from the header if it is a matrix or a right hand side 
        bool isvector,iscomplex=false;
        if(!getMarketHeader(curfile,m_sym,iscomplex,isvector)) continue;
        if(isvector) continue;
        if (!iscomplex)
        {
          if(internal::is_same<Scalar, std::complex<float> >::value || internal::is_same<Scalar, std::complex<double> >::value)
            continue; 
        }
        if (iscomplex)
        {
          if(internal::is_same<Scalar, float>::value || internal::is_same<Scalar, double>::value)
            continue; 
        }
        
        
        // Get the matrix name
        std::string filename = m_curs_id->d_name;
        m_matname = filename.substr(0, filename.length()-4); 
        
        // Find if the matrix is SPD 
        size_t found = m_matname.find("SPD");
        if( (found!=std::string::npos) && (m_sym != NonSymmetric) )
          m_sym = SPD;
       
        m_isvalid = true;
        break; 
      }
    }

hasrefX()

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::hasrefX ( )

inline

Definition at line 170 of file MatrixMarketIterator.h.

{return m_hasrefX; }

hasRhs()

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::hasRhs ( )

inline

Definition at line 169 of file MatrixMarketIterator.h.

{return m_hasRhs; }

isFolderValid()

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::isFolderValid ( )

inline

Definition at line 171 of file MatrixMarketIterator.h.

{ return bool(m_folder_id); }

matname()

template<typename Scalar >

std::string& Eigen::MatrixMarketIterator< Scalar >::matname ( )

inline

Definition at line 165 of file MatrixMarketIterator.h.

{ return m_matname; }

matrix()

template<typename Scalar >

MatrixType& Eigen::MatrixMarketIterator< Scalar >::matrix ( )

inline

Return the sparse matrix corresponding to the current file

Definition at line 76 of file MatrixMarketIterator.h.

    { 
      // Read the matrix
      if (m_matIsLoaded) return m_mat;
      
      std::string matrix_file = m_folder + "/" + m_matname + ".mtx";
      if ( !loadMarket(m_mat, matrix_file)) 
      {
        std::cerr << "Warning loadMarket failed when loading \"" << matrix_file << "\"" << std::endl;
        m_matIsLoaded = false;
        return m_mat;
      }
      m_matIsLoaded = true; 
 
      if (m_sym != NonSymmetric) 
      {
        // Check whether we need to restore a full matrix:
        RealScalar diag_norm  = m_mat.diagonal().norm();
        RealScalar lower_norm = m_mat.template triangularView<Lower>().norm();
        RealScalar upper_norm = m_mat.template triangularView<Upper>().norm();
        if(lower_norm>diag_norm && upper_norm==diag_norm)
        {
          // only the lower part is stored
          MatrixType tmp(m_mat);
          m_mat = tmp.template selfadjointView<Lower>();
        }
        else if(upper_norm>diag_norm && lower_norm==diag_norm)
        {
          // only the upper part is stored
          MatrixType tmp(m_mat);
          m_mat = tmp.template selfadjointView<Upper>();
        }
      }
      return m_mat; 
    }

operator bool()

template<typename Scalar >

Eigen::MatrixMarketIterator< Scalar >::operator bool ( ) const

inline

Definition at line 73 of file MatrixMarketIterator.h.

{ return m_isvalid;}

operator++()

template<typename Scalar >

MatrixMarketIterator& Eigen::MatrixMarketIterator< Scalar >::operator++ ( )

inline

Definition at line 65 of file MatrixMarketIterator.h.

    {
      m_matIsLoaded = false;
      m_hasrefX = false;
      m_hasRhs = false;
      Getnextvalidmatrix();
      return *this;
    }

refX()

template<typename Scalar >

VectorType& Eigen::MatrixMarketIterator< Scalar >::refX ( )

inline

Return a reference solution If it is not provided and if the right hand side is not available then refX is randomly generated such that A*refX = b where A and b are the matrix and the rhs. Note that when a rhs is provided, refX is not available

Definition at line 147 of file MatrixMarketIterator.h.

    { 
      // Check if a reference solution is provided
      if (m_hasrefX) return m_refX;
      
      std::string lhs_file;
      lhs_file = m_folder + "/" + m_matname + "_x.mtx"; 
      m_hasrefX = Fileexists(lhs_file);
      if (m_hasrefX)
      {
        m_refX.resize(m_mat.cols());
        m_hasrefX = loadMarketVector(m_refX, lhs_file);
      }
      else
        m_refX.resize(0);
      return m_refX; 
    }

rhs()

template<typename Scalar >

VectorType& Eigen::MatrixMarketIterator< Scalar >::rhs ( )

inline

Return the right hand side corresponding to the current matrix. If the rhs file is not provided, a random rhs is generated

Definition at line 115 of file MatrixMarketIterator.h.

    { 
       // Get the right hand side
      if (m_hasRhs) return m_rhs;
      
      std::string rhs_file;
      rhs_file = m_folder + "/" + m_matname + "_b.mtx"; // The pattern is matname_b.mtx
      m_hasRhs = Fileexists(rhs_file);
      if (m_hasRhs)
      {
        m_rhs.resize(m_mat.cols());
        m_hasRhs = loadMarketVector(m_rhs, rhs_file);
      }
      if (!m_hasRhs)
      {
        // Generate a random right hand side
        if (!m_matIsLoaded) this->matrix(); 
        m_refX.resize(m_mat.cols());
        m_refX.setRandom();
        m_rhs = m_mat * m_refX;
        m_hasrefX = true;
        m_hasRhs = true;
      }
      return m_rhs; 
    }

sym()

template<typename Scalar >

int Eigen::MatrixMarketIterator< Scalar >::sym ( )

inline

Definition at line 167 of file MatrixMarketIterator.h.

{ return m_sym; }

Member Data Documentation

m_curs_id

template<typename Scalar >

struct dirent* Eigen::MatrixMarketIterator< Scalar >::m_curs_id

protected

Definition at line 243 of file MatrixMarketIterator.h.

m_folder

template<typename Scalar >

std::string Eigen::MatrixMarketIterator< Scalar >::m_folder

protected

Definition at line 241 of file MatrixMarketIterator.h.

m_folder_id

template<typename Scalar >

DIR* Eigen::MatrixMarketIterator< Scalar >::m_folder_id

protected

Definition at line 242 of file MatrixMarketIterator.h.

m_hasrefX

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::m_hasrefX

protected

Definition at line 240 of file MatrixMarketIterator.h.

m_hasRhs

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::m_hasRhs

protected

Definition at line 239 of file MatrixMarketIterator.h.

m_isvalid

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::m_isvalid

protected

Definition at line 237 of file MatrixMarketIterator.h.

m_mat

template<typename Scalar >

MatrixType Eigen::MatrixMarketIterator< Scalar >::m_mat

protected

Definition at line 233 of file MatrixMarketIterator.h.

m_matIsLoaded

template<typename Scalar >

bool Eigen::MatrixMarketIterator< Scalar >::m_matIsLoaded

protected

Definition at line 238 of file MatrixMarketIterator.h.

m_matname

template<typename Scalar >

std::string Eigen::MatrixMarketIterator< Scalar >::m_matname

protected

Definition at line 236 of file MatrixMarketIterator.h.

m_refX

template<typename Scalar >

VectorType Eigen::MatrixMarketIterator< Scalar >::m_refX

protected

Definition at line 235 of file MatrixMarketIterator.h.

m_rhs

template<typename Scalar >

VectorType Eigen::MatrixMarketIterator< Scalar >::m_rhs

protected

Definition at line 234 of file MatrixMarketIterator.h.

m_sym

template<typename Scalar >

int Eigen::MatrixMarketIterator< Scalar >::m_sym

protected

Definition at line 232 of file MatrixMarketIterator.h.

---

The documentation for this class was generated from the following file:

• MatrixMarketIterator.h