EpetraExt Development
EpetraExt_readEpetraLinearSystem.cpp
Go to the documentation of this file.
00001 //@HEADER
00002 // ***********************************************************************
00003 //
00004 //     EpetraExt: Epetra Extended - Linear Algebra Services Package
00005 //                 Copyright (2011) Sandia Corporation
00006 //
00007 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
00008 // the U.S. Government retains certain rights in this software.
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 // Questions? Contact Michael A. Heroux (maherou@sandia.gov)
00038 //
00039 // ***********************************************************************
00040 //@HEADER
00041 
00042 #include "EpetraExt_readEpetraLinearSystem.h"
00043 #include "Trilinos_Util.h"
00044 
00045 void EpetraExt::readEpetraLinearSystem(
00046   const std::string                               &fileName
00047   ,const Epetra_Comm                              &comm
00048   ,Teuchos::RefCountPtr<Epetra_CrsMatrix>         *A
00049   ,Teuchos::RefCountPtr<Epetra_Map>               *map
00050   ,Teuchos::RefCountPtr<Epetra_Vector>            *x
00051   ,Teuchos::RefCountPtr<Epetra_Vector>            *b
00052   ,Teuchos::RefCountPtr<Epetra_Vector>            *xExact
00053   )
00054 {
00055 
00056   Epetra_Map       *readMap;
00057   Epetra_CrsMatrix *readA; 
00058   Epetra_Vector    *readx; 
00059   Epetra_Vector    *readb;
00060   Epetra_Vector    *readxexact;
00061 
00062   const std::string::size_type ext_dot = fileName.rfind(".");
00063   TEUCHOS_TEST_FOR_EXCEPT( ext_dot == std::string::npos );
00064   std::string ext = fileName.substr(ext_dot+1);
00065   //std::cout << "\nfileName = " << fileName << "\next = " << ext << std::endl;
00066 
00067   char *hacked_file_str = const_cast<char*>(fileName.c_str());
00068   
00069   if ( ext == "triU" ) { 
00070     const bool NonContiguousMap = true; 
00071     TEUCHOS_TEST_FOR_EXCEPT(
00072       0!=Trilinos_Util_ReadTriples2Epetra(
00073         hacked_file_str, false, comm, readMap, readA, readx, 
00074         readb, readxexact, NonContiguousMap
00075         )
00076       );
00077   }
00078   else if ( ext == "triS" ) { 
00079     const bool NonContiguousMap = true; 
00080     TEUCHOS_TEST_FOR_EXCEPT(
00081       0!=Trilinos_Util_ReadTriples2Epetra(
00082         hacked_file_str, true, comm, readMap, readA, readx, 
00083         readb, readxexact, NonContiguousMap
00084         )
00085       );
00086   }
00087   else if( ext == "mtx" ) { 
00088     TEUCHOS_TEST_FOR_EXCEPT(
00089       0!=Trilinos_Util_ReadMatrixMarket2Epetra(
00090         hacked_file_str, comm, readMap, 
00091         readA, readx, readb, readxexact
00092         )
00093       );
00094   }
00095   else if ( ext == "hb" ) {
00096     Trilinos_Util_ReadHb2Epetra(
00097       hacked_file_str, comm, readMap, readA, readx, 
00098       readb, readxexact
00099       ); // No error return???
00100   }
00101   else {
00102     TEUCHOS_TEST_FOR_EXCEPTION(
00103       true, std::logic_error
00104       ,"Error, the file = \'"<<hacked_file_str<<"\' has the extension "
00105       "\'*."<<ext<<"\' is not \'*.triU\', \'*.triS\', \'*.mtx\', or \'*.hb\'!"
00106       );
00107   }
00108 
00109   Teuchos::RefCountPtr<Epetra_CrsMatrix>    loc_A         = Teuchos::rcp(readA);
00110   Teuchos::RefCountPtr<Epetra_Map>          loc_map       = Teuchos::rcp(readMap);
00111   Teuchos::RefCountPtr<Epetra_Vector>       loc_x         = Teuchos::rcp(readx);
00112   Teuchos::RefCountPtr<Epetra_Vector>       loc_b         = Teuchos::rcp(readb);
00113   Teuchos::RefCountPtr<Epetra_Vector>       loc_xExact    = Teuchos::rcp(readxexact);
00114 
00115   if(A)       *A       = loc_A;
00116   if(map)     *map     = loc_map;
00117   if(x)       *x       = loc_x;
00118   if(b)       *b       = loc_b;
00119   if(xExact)  *xExact  = loc_xExact;
00120   
00121 }
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines