Matrix-Market: fix perf issue and infinite loop

This commit is contained in:
Gael Guennebaud 2012-06-15 09:07:13 +02:00
parent c858fb353f
commit 3fd2beebc8
2 changed files with 7 additions and 3 deletions

View File

@ -156,6 +156,9 @@ bool loadMarket(SparseMatrixType& mat, const std::string& filename)
char buffer[maxBuffersize];
bool readsizes = false;
typedef Triplet<Scalar,int> T;
std::vector<T> elements;
int M(-1), N(-1), NNZ(-1);
int count = 0;
@ -186,13 +189,13 @@ bool loadMarket(SparseMatrixType& mat, const std::string& filename)
if( internal::GetMarketLine(line, M, N, i, j, value) )
{
++ count;
mat.insert(i,j) = value;
elements.push_back(T(i,j,value));
}
else
std::cerr << "Invalid read: " << i << "," << j << "\n";
}
}
mat.makeCompressed();
mat.setFromTriplets(elements.begin(), elements.end());
if(count!=NNZ)
std::cerr << count << "!=" << NNZ << "\n";

View File

@ -85,7 +85,7 @@ class MatrixMarketIterator
Getnextvalidmatrix();
return *this;
}
inline operator bool() { return m_isvalid;}
inline operator bool() const { return m_isvalid;}
/** Return the sparse matrix corresponding to the current file */
inline MatrixType& matrix()
@ -186,6 +186,7 @@ class MatrixMarketIterator
void Getnextvalidmatrix( )
{
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;