EpetraExt Package Browser (Single Doxygen Collection) Development
EpetraExt_Overlap_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_Overlap_CrsGraph.h>
00043 
00044 #include <Epetra_Import.h>
00045 #include <Epetra_CrsGraph.h>
00046 #include <Epetra_Map.h>
00047 
00048 namespace EpetraExt {
00049 
00050 CrsGraph_Overlap::
00051 ~CrsGraph_Overlap()
00052 {
00053   if( newObj_ ) delete newObj_;
00054 
00055   if( OverlapMap_ ) delete OverlapMap_;
00056 }
00057 
00058 CrsGraph_Overlap::NewTypeRef
00059 CrsGraph_Overlap::
00060 operator()( OriginalTypeRef orig )
00061 {
00062   origObj_ = &orig;
00063 
00064   //check that this is a distributed graph and overlap level is not zero
00065   if( orig.DomainMap().DistributedGlobal() && levelOverlap_ )
00066   {
00067     Epetra_CrsGraph * OverlapGraph = new Epetra_CrsGraph( orig );
00068     OverlapMap_ = new Epetra_BlockMap( orig.RowMap() );
00069 
00070     Epetra_BlockMap * DomainMap = &(const_cast<Epetra_BlockMap&>(orig.DomainMap()));
00071     Epetra_BlockMap * RangeMap = &(const_cast<Epetra_BlockMap&>(orig.RangeMap()));
00072 
00073     for( int level = 0; level < levelOverlap_; ++level )
00074     {
00075       Epetra_BlockMap * OldRowMap = OverlapMap_;
00076       Epetra_CrsGraph * OldGraph = OverlapGraph;
00077 
00078       Epetra_Import & OverlapImporter = *(const_cast<Epetra_Import *>( OldGraph->Importer() ));
00079       OverlapMap_ = new Epetra_BlockMap( OverlapImporter.TargetMap() );
00080 
00081       //filter to local square block on last level if required
00082       if( squareLocalBlock_ && level==(levelOverlap_-1) )
00083         OverlapGraph = new Epetra_CrsGraph( Copy, *OverlapMap_, *OverlapMap_, 0 );
00084       else
00085         OverlapGraph = new Epetra_CrsGraph( Copy, *OverlapMap_, 0 );
00086 
00087       OverlapGraph->Import( *OldGraph, OverlapImporter, Insert );
00088       OverlapGraph->FillComplete( *DomainMap, *RangeMap );
00089 
00090       delete OldGraph;
00091       delete OldRowMap;
00092     }
00093 
00094     newObj_ = OverlapGraph;
00095   }
00096   else //just create a copy since this is not a InPlaceTransform
00097     newObj_ = new Epetra_CrsGraph( orig );
00098 
00099   return *newObj_;
00100 }
00101 
00102 } // namespace EpetraExt
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines