Ifpack2 Templated Preconditioning Package Version 1.0
Ifpack2_LinearPartitioner_def.hpp
00001 /*@HEADER
00002 // ***********************************************************************
00003 //
00004 //       Ifpack2: Tempated Object-Oriented Algebraic Preconditioner Package
00005 //                 Copyright (2009) 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 // Questions? Contact Michael A. Heroux (maherou@sandia.gov)
00038 //
00039 // ***********************************************************************
00040 //@HEADER
00041 */
00042 
00043 #ifndef IFPACK2_LINEAR_PARTITIONER_DEF_HPP
00044 #define IFPACK2_LINEAR_PARTITIONER_DEF_HPP
00045 #include "Ifpack2_ConfigDefs.hpp"
00046 #include "Ifpack2_LinearPartitioner_decl.hpp"
00047 
00048 namespace Ifpack2 {
00049 
00050 //==============================================================================
00051 // Constructor
00052 template<class GraphType>
00053 LinearPartitioner<GraphType>::
00054 LinearPartitioner (const Teuchos::RCP<const row_graph_type>& graph) :
00055   OverlappingPartitioner<GraphType> (graph)
00056 {}
00057 
00058 
00059 template<class GraphType>
00060 LinearPartitioner<GraphType>::~LinearPartitioner() {}
00061 
00062 
00063 template<class GraphType>
00064 void
00065 LinearPartitioner<GraphType>::
00066 setPartitionParameters (Teuchos::ParameterList& List) {}
00067 
00068 
00069 template<class GraphType>
00070 void LinearPartitioner<GraphType>::LinearPartitioner::computePartitions()
00071 {
00072   using Teuchos::as;
00073   // Partition_ is an array of local_ordinal_type.  local_ordinal_type
00074   // may be signed or unsigned.  NumLocalParts_ is int, and needs to
00075   // be signed, since negative values are significant.  Comparisons
00076   // between signed and unsigned integers often result in compiler
00077   // warnings, which is why we use as() for explicit conversions
00078   // below.  We also use as() because in a debug build, it checks for
00079   // overflow.
00080   const int mod = as<int> (this->Graph_->getNodeNumRows () / 
00081          this->NumLocalParts_);
00082   for (size_t i = 0; i < this->Graph_->getNodeNumRows (); ++i) {
00083     this->Partition_[i] = as<local_ordinal_type> (i / mod);
00084     if (this->Partition_[i] >= as<local_ordinal_type> (this->NumLocalParts_)) {
00085       this->Partition_[i] = this->NumLocalParts_ - 1;
00086     }
00087   }
00088 }
00089 
00090 
00091 }// namespace Ifpack2
00092 
00093 #define IFPACK2_LINEARPARTITIONER_INSTANT(LO,GO,N) \
00094   template class Ifpack2::LinearPartitioner<Tpetra::CrsGraph< LO, GO, N > >; \
00095   template class Ifpack2::LinearPartitioner<Tpetra::RowGraph< LO, GO, N > >;
00096 
00097 #endif // IFPACK2_LINEARPARTITIONER_DEF_HPP
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends