Isorropia: Partitioning, Load Balancing and more
ispatest_epetra_utils.hpp
Go to the documentation of this file.
00001 //@HEADER
00002 //************************************************************************
00003 //
00004 //              Isorropia: Partitioning and Load Balancing Package
00005 //                Copyright (2006) Sandia Corporation
00006 //
00007 //Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
00008 //license for use of this work by or on behalf of the U.S. Government.
00009 //
00010 // Redistribution and use in source and binary forms, with or without
00011 // modification, are permitted provided that the following conditions are
00012 // met:
00013 //
00014 // 1. Redistributions of source code must retain the above copyright
00015 // notice, this list of conditions and the following disclaimer.
00016 //
00017 // 2. Redistributions in binary form must reproduce the above copyright
00018 // notice, this list of conditions and the following disclaimer in the
00019 // documentation and/or other materials provided with the distribution.
00020 //
00021 // 3. Neither the name of the Corporation nor the names of the
00022 // contributors may be used to endorse or promote products derived from
00023 // this software without specific prior written permission.
00024 //
00025 // THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
00026 // EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
00027 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
00028 // PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
00029 // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
00030 // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
00031 // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
00032 // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
00033 // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
00034 // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
00035 // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
00036 //
00037 //************************************************************************
00038 //@HEADER
00039 
00040 #ifndef _ispatest_epetra_test_utils_hpp_
00041 #define _ispatest_epetra_test_utils_hpp_
00042 
00043 #include <Isorropia_ConfigDefs.hpp>
00044 #include <string>
00045 #include <vector>
00046 #include <ostream>
00047 
00048 #ifdef HAVE_EPETRA
00049 
00050 class Epetra_CrsGraph;
00051 class Epetra_RowMatrix;
00052 class Epetra_LinearProblem;
00053 class Epetra_CrsMatrix;
00054 class Epetra_MultiVector;
00055 class Epetra_Map;
00056 class Epetra_BlockMap;
00057 class Epetra_Comm;
00058 
00059 namespace ispatest {
00060 
00067 int fill_matrix(Epetra_CrsMatrix& matrix,
00068                 int numNonzerosPerRow,
00069                 bool verbose);
00070 
00076 int fill_graph(Epetra_CrsGraph& graph,
00077                 int numNonzerosPerRow,
00078                 bool verbose);
00079 
00091 bool test_matrix_vector_multiply(Epetra_CrsMatrix &A);
00092 
00102 bool test_row_matrix_vector_multiply(Epetra_RowMatrix &R);
00103 
00115 bool test_matrix_vector_multiply(Epetra_CrsGraph &G);
00116 
00131 bool test_matrix_vector_multiply(Epetra_LinearProblem &LP);
00132 
00138 Epetra_Map *map_from_blockmap(const Epetra_BlockMap &b);
00139 
00152 int readCoordFile(const std::string &fname,
00153          std::vector<double> &x, std::vector<double> &y, std::vector<double> &z);
00154 
00162 Epetra_MultiVector *makeWeights(const Epetra_BlockMap &map, double (*wFunc)(const int, const int, const int, const int));
00163 
00166 double unitWeights(const int id, const int me, const int nids, const int nprocs);
00167 
00171 double veeWeights(const int id, const int me, const int nids, const int nprocs);
00172 
00175 double alternateWeights(const int id, const int me, const int nids, const int nprocs);
00176 
00180 Epetra_MultiVector *file2multivector(const Epetra_Comm &comm, const std::string &fname);
00181 
00185 int printMultiVector(const Epetra_MultiVector &mv, std::ostream &os, 
00186                      const char *s, int max=1000);
00187 
00193 int printRowMatrix(const Epetra_RowMatrix &m, std::ostream &os, 
00194                      const char *s, bool withGraphCuts, int max=1000);
00195 
00196 } //namespace ispatest
00197 
00198 #endif //HAVE_EPETRA
00199 
00200 #endif
00201