src/matrixmarket/mtxreader.cpp

Go to the documentation of this file.

/***************************************************************************
                          MtxReader.cpp  -  description
                             -------------------
    begin                : Tue Dec 2 2003
    copyright            : (C) 2003 by Roman Geus
    email                : roman.geus@psi.ch
***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/

#include <cassert>
#include "Epetra_Map.h"

#include "mtxdataiterator.h"
#include "mtxreader.h"

MtxReader::MtxReader(string filename, const Epetra_Map& row_map,
                     const Epetra_Map& domain_map, const Epetra_Map& range_map)
    : BaseMtxReader(filename, row_map.Comm()),
      _row_map(row_map),
      _domain_map(domain_map),
      _range_map(range_map) {}

MtxReader::MtxReader(istream& istr, const Epetra_Map& row_map,
                     const Epetra_Map& domain_map, const Epetra_Map& range_map)
    : BaseMtxReader(istr, row_map.Comm()),
      _row_map(row_map),
      _domain_map(domain_map),
      _range_map(range_map) {}

MtxReader::~MtxReader() {}

Epetra_CrsMatrix* MtxReader::read() {
    MtxDataIterator it(*_istr, true);

    assert(it.get_element_type() != MtxDataIterator::complex);
    assert(it.get_storage_scheme() == MtxDataIterator::general ||
           it.get_storage_scheme() == MtxDataIterator::symmetric);

    // Allocate matrix and read matrix data
    Epetra_CrsMatrix* mat = new Epetra_CrsMatrix (Copy, _row_map, 0);

    // read matrix entries
    while (!it.eof()) {
        MtxDataIterator::Triplet tr = *it;
        int row = tr.row;
        int col = tr.col;
        double val = tr.val;

        if (_row_map.MyGID(row))
            mat->InsertGlobalValues(row, 1, &val, &col);

        ++it;
    }

    // finalise Crs matrix
    mat->FillComplete(_domain_map, _range_map);

    return mat;
}

---