EpetraExt Development
EpetraExt_View_CrsGraph.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_View_CrsGraph.h>
00043 
00044 #include <Epetra_CrsGraph.h>
00045 #include <Epetra_BlockMap.h>
00046 
00047 #include <vector>
00048 
00049 namespace EpetraExt {
00050 
00051 CrsGraph_View::
00052 ~CrsGraph_View()
00053 {
00054   if( newObj_ ) delete newObj_;
00055 }
00056 
00057 CrsGraph_View::NewTypeRef
00058 CrsGraph_View::
00059 operator()( OriginalTypeRef orig )
00060 {
00061   origObj_ = &orig;
00062 
00063   //Error, must be local indices
00064   assert( !orig.IndicesAreGlobal() );
00065 
00066   //test maps, new std::map must be left subset of old
00067   const Epetra_BlockMap & oRowMap = orig.RowMap();
00068   const Epetra_BlockMap & oColMap = orig.ColMap();
00069 
00070   int nNumRows = NewRowMap_->NumMyElements();
00071   int nNumCols = 0;
00072   if( NewColMap_ ) nNumCols = NewColMap_->NumMyElements();
00073 
00074   bool matched = true;
00075   for( int i = 0; i < nNumRows; ++i )
00076     matched = matched && ( oRowMap.GID64(i) == NewRowMap_->GID64(i) );
00077   if( nNumCols )
00078     for( int i = 0; i < nNumCols; ++i )
00079       matched = matched && ( oColMap.GID64(i) == NewColMap_->GID64(i) );
00080 
00081   if( !matched ) std::cout << "EDT_CrsGraph_View: Bad Row or Col Mapping\n";
00082   assert( matched );
00083 
00084   //intial construction of graph
00085   std::vector<int> numIndices( nNumRows );
00086   std::vector<int*> indices( nNumRows );
00087   for( int i = 0; i < nNumRows; ++i )
00088   {
00089     orig.ExtractMyRowView( i, numIndices[i], indices[i] );
00090     int j = 0;
00091     if( nNumCols )
00092     {
00093       while( j < numIndices[i] && NewColMap_->GID64(indices[i][j]) != -1 ) ++j;
00094       numIndices[i] = j;
00095     }
00096   }
00097 
00098   Epetra_CrsGraph * newGraph( new Epetra_CrsGraph( View,
00099                                                    *NewRowMap_,
00100                                                    *NewColMap_,
00101                                                    &numIndices[0] ) );
00102 
00103   //insert views of row indices
00104   for( int i = 0; i < nNumRows; ++i )
00105     newGraph->InsertMyIndices( i, numIndices[i], indices[i] );
00106 
00107   newGraph->FillComplete();
00108 
00109   newObj_ = newGraph;
00110 
00111   return *newGraph;
00112 }
00113 
00114 } // namespace EpetraExt
00115 
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines