EpetraExt Development
EpetraExt_Transpose_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_Transpose_CrsGraph.h>
00043 
00044 #include <Epetra_Export.h>
00045 #include <Epetra_CrsGraph.h>
00046 #include <Epetra_Map.h>
00047 
00048 #include <vector>
00049 
00050 namespace EpetraExt {
00051 
00052 CrsGraph_Transpose::
00053 ~CrsGraph_Transpose()
00054 {
00055   delete newObj_;
00056 }
00057 
00058 CrsGraph_Transpose::NewTypeRef
00059 CrsGraph_Transpose::
00060 operator()( OriginalTypeRef orig )
00061 {
00062   origObj_ = &orig;
00063 
00064   int nRows = orig.NumMyRows();
00065   int nCols = orig.NumMyCols();
00066 
00067   const Epetra_BlockMap & RowMap = orig.RowMap();
00068 
00069   int numIndices;
00070   int * Indices;
00071 
00072   Epetra_CrsGraph * TransposeGraph = 0;
00073 
00074   if( !ignoreNonLocalCols_ && orig.DistributedGlobal() )
00075   {
00076     std::vector<int> TransNumNZ( nCols, 0 );
00077     for( int i = 0; i < nRows; ++i )
00078     {
00079       orig.ExtractMyRowView( i, numIndices, Indices );
00080       for( int j = 0; j < numIndices; ++j ) ++TransNumNZ[ Indices[j] ];
00081     }
00082 
00083     std::vector< std::vector<int> > TransIndices( nCols );
00084     for( int i = 0; i < nCols; ++i )
00085       if( TransNumNZ[i] )
00086       {
00087         TransIndices[i].resize( TransNumNZ[i] );
00088         TransNumNZ[i] = 0;
00089       }
00090 
00091     for( int i = 0; i < nRows; ++i )
00092     {
00093       orig.ExtractMyRowView( i, numIndices, Indices );
00094       for( int j = 0; j < numIndices; ++j )
00095         TransIndices[ Indices[j] ][ TransNumNZ[ Indices[j] ]++ ] = i;
00096     }
00097 
00098     Epetra_CrsGraph SharedTransGraph( View, orig.ImportMap(), RowMap, &TransNumNZ[0] );
00099     for( int i = 0; i < nCols; ++i )
00100       if( TransNumNZ[i] ) SharedTransGraph.InsertMyIndices( i, TransNumNZ[i], &TransIndices[i][0] );
00101     SharedTransGraph.FillComplete();
00102 
00103     TransposeGraph = new Epetra_CrsGraph( Copy, RowMap, 0 );
00104     Epetra_Export Exporter( orig.ImportMap(), RowMap ); 
00105     TransposeGraph->Export( SharedTransGraph, Exporter, Add );
00106     TransposeGraph->FillComplete();
00107   }
00108   else
00109   {
00110     std::vector<int> TransNumNZ( nRows, 0 );
00111     for( int i = 0; i < nRows; ++i )
00112     {
00113       orig.ExtractMyRowView( i, numIndices, Indices );
00114       for( int j = 0; j < numIndices; ++j )
00115         if( Indices[j] < nRows ) ++TransNumNZ[ Indices[j] ];
00116     }
00117 
00118     std::vector< std::vector<int> > TransIndices( nRows );
00119     for( int i = 0; i < nRows; ++i )
00120       if( TransNumNZ[i] )
00121       {
00122         TransIndices[i].resize( TransNumNZ[i] );
00123         TransNumNZ[i] = 0;
00124       }
00125 
00126     for( int i = 0; i < nRows; ++i )
00127     {
00128       orig.ExtractMyRowView( i, numIndices, Indices );
00129       for( int j = 0; j < numIndices; ++j )
00130         if( Indices[j] < nRows ) TransIndices[ Indices[j] ][ TransNumNZ[ Indices[j] ]++ ] = i;
00131     }
00132 
00133     TransposeGraph = new Epetra_CrsGraph( Copy, RowMap, RowMap, &TransNumNZ[0] );
00134 
00135     for( int i = 0; i < nRows; ++i )
00136       if( TransNumNZ[i] ) TransposeGraph->InsertMyIndices( i, TransNumNZ[i], &TransIndices[i][0] );
00137 
00138     TransposeGraph->FillComplete();
00139   }
00140 
00141   newObj_ = TransposeGraph;
00142 
00143   return *TransposeGraph;
00144 }
00145 
00146 } // namespace EpetraExt
00147 
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines