Isorropia_EpetraPartitioner2D.hpp

Go to the documentation of this file.
00001 //@HEADER
00002 /*
00003 ************************************************************************
00004 
00005               Isorropia: Partitioning and Load Balancing Package
00006                 Copyright (2006) Sandia Corporation
00007 
00008 Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
00009 license for use of this work by or on behalf of the U.S. Government.
00010 
00011 This library is free software; you can redistribute it and/or modify
00012 it under the terms of the GNU Lesser General Public License as
00013 published by the Free Software Foundation; either version 2.1 of the
00014 License, or (at your option) any later version.
00015 
00016 This library is distributed in the hope that it will be useful, but
00017 WITHOUT ANY WARRANTY; without even the implied warranty of
00018 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
00019 Lesser General Public License for more details.
00020 
00021 You should have received a copy of the GNU Lesser General Public
00022 License along with this library; if not, write to the Free Software
00023 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
00024 USA
00025 
00026 ************************************************************************
00027 */
00028 //@HEADER
00029 
00030 #ifndef _Isorropia_EpetraPartitioner2D_hpp_
00031 #define _Isorropia_EpetraPartitioner2D_hpp_
00032 
00033 #include <Isorropia_ConfigDefs.hpp>
00034 #include <Teuchos_RCP.hpp>
00035 #include <Teuchos_ParameterList.hpp>
00036 
00037 #include <Isorropia_EpetraCostDescriber.hpp>
00038 #include <Isorropia_EpetraOperator.hpp>
00039 #include <Isorropia_Partitioner2D.hpp>
00040 
00041 #ifdef HAVE_EPETRA
00042 class Epetra_Map;
00043 class Epetra_BlockMap;
00044 class Epetra_Import;
00045 class Epetra_Vector;
00046 class Epetra_MultiVector;
00047 class Epetra_CrsGraph;
00048 class Epetra_CrsMatrix;
00049 class Epetra_RowMatrix;
00050 class Epetra_LinearProblem;
00051 
00052 namespace Isorropia {
00053 
00054 namespace Epetra {
00055   class CostDescriber;
00056 
00062 class Partitioner2D : virtual public Isorropia::Partitioner2D, virtual public Isorropia::Epetra::Operator  {
00063 public:
00087   Partitioner2D(Teuchos::RCP<const Epetra_CrsGraph> input_graph,
00088               const Teuchos::ParameterList& paramlist,
00089               bool compute_partitioning_now=true);
00090 
00117   Partitioner2D(Teuchos::RCP<const Epetra_CrsGraph> input_graph,
00118               Teuchos::RCP<CostDescriber> costs,
00119               const Teuchos::ParameterList& paramlist,
00120               bool compute_partitioning_now=true);
00121 
00146   Partitioner2D(Teuchos::RCP<const Epetra_RowMatrix> input_matrix,
00147               const Teuchos::ParameterList& paramlist,
00148               bool compute_partitioning_now=true);
00149 
00177   Partitioner2D(Teuchos::RCP<const Epetra_RowMatrix> input_matrix,
00178               Teuchos::RCP<CostDescriber> costs,
00179               const Teuchos::ParameterList& paramlist,
00180               bool compute_partitioning_now=true);
00181 
00182 
00183 
00184   //  MMW: Missing the following constructors that are in Partitioner
00185   //
00186   //  Partitioner(Teuchos::RCP<const Epetra_CrsGraph> input_graph,
00187   //            bool compute_partitioning_now=true);
00188   //
00189   //  Partitioner(Teuchos::RCP<const Epetra_RowMatrix> input_matrix,
00190   //            bool compute_partitioning_now=true);
00191   //
00192   //  Partitioner(Teuchos::RCP<const Epetra_MultiVector> coords,
00193   //            bool compute_partitioning_now=true);
00194   //
00195   //  Partitioner(Teuchos::RCP<const Epetra_MultiVector> coords,
00196   //            const Teuchos::ParameterList& paramlist,
00197   //            bool compute_partitioning_now=true);
00198   //
00199   //  Partitioner(Teuchos::RCP<const Epetra_MultiVector> coords,
00200   //            Teuchos::RCP<const Epetra_MultiVector> weights,
00201   //            const Teuchos::ParameterList& paramlist,
00202   //            bool compute_partitioning_now=true);
00203   
00204 
00205 
00206 
00208   virtual ~Partitioner2D();
00209 
00210 
00211   // MMW: Missing functions that are in EpetraPartioner, might need to implement
00212   //void setPartSizes(int len, int *global_part_id, float *part_size);
00213   //void clearPartSizes();
00214 
00226   void partition(bool force_repartitioning=false);
00227 
00228   virtual void compute(bool forceRecomputing=false);
00229 
00230 
00235   int numElemsInPart(int part) const;
00236 
00237 
00243   void elemsInPart(int part, int* elementList, int len) const;
00244 
00245 
00256   Teuchos::RCP<Epetra_Map> createNewMap();
00257 
00258   int createNewMaps(Teuchos::RCP<Epetra_Map> domainMap, 
00259         Teuchos::RCP<Epetra_Map> rangeMap);
00260 
00261   int partitionVectors();
00262 
00263 
00264 
00265 };//class Partitioner2D
00266 
00267 }//namespace Epetra
00268 }//namespace Isorropia
00269 
00270 #endif //HAVE_EPETRA
00271 
00272 #endif
00273 
00274