Isorropia: Partitioning, Load Balancing and more Version 3.0
Isorropia_EpetraZoltanLib.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_EpetraZoltanLib_hpp_
00031 #define _Isorropia_EpetraZoltanLib_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_EpetraLibrary.hpp>
00039 
00040 #include <QueryObject.hpp>
00041 #include <zoltan_cpp.h>
00042 
00043 #ifdef HAVE_EPETRA
00044 class Epetra_Map;
00045 class Epetra_BlockMap;
00046 class Epetra_Import;
00047 class Epetra_Vector;
00048 class Epetra_MultiVector;
00049 class Epetra_CrsGraph;
00050 class Epetra_CrsMatrix;
00051 class Epetra_RowMatrix;
00052 class Epetra_LinearProblem;
00053 
00054 namespace Isorropia {
00055 
00056 namespace Epetra {
00057   class CostDescriber;
00058 
00059 
00060 class ZoltanLibClass : public Library {
00061 public:
00062 
00063   ZoltanLibClass(Teuchos::RCP<const Epetra_CrsGraph> input_graph, int inputType=unspecified_input_);
00064   ZoltanLibClass(Teuchos::RCP<const Epetra_CrsGraph> input_graph, 
00065      Teuchos::RCP<const Epetra_MultiVector> input_coords, int inputType=unspecified_input_);
00066   ZoltanLibClass(Teuchos::RCP<const Epetra_CrsGraph> input_graph,
00067            Teuchos::RCP<CostDescriber> costs, int inputType=unspecified_input_);
00068   ZoltanLibClass(Teuchos::RCP<const Epetra_CrsGraph> input_graph, Teuchos::RCP<CostDescriber> costs, 
00069                  Teuchos::RCP<const Epetra_MultiVector> input_coords, Teuchos::RCP<const Epetra_MultiVector> weights, 
00070                  int inputType=unspecified_input_);
00071   ZoltanLibClass(Teuchos::RCP<const Epetra_RowMatrix> input_matrix, int inputType=unspecified_input_);
00072   ZoltanLibClass(Teuchos::RCP<const Epetra_RowMatrix> input_matrix, 
00073                  Teuchos::RCP<const Epetra_MultiVector> input_coords, int inputType=unspecified_input_);
00074   ZoltanLibClass(Teuchos::RCP<const Epetra_RowMatrix> input_matrix,
00075            Teuchos::RCP<CostDescriber> costs, int inputType=unspecified_input_);
00076   ZoltanLibClass(Teuchos::RCP<const Epetra_RowMatrix> input_matrix, Teuchos::RCP<CostDescriber> costs, 
00077      Teuchos::RCP<const Epetra_MultiVector> input_coords, Teuchos::RCP<const Epetra_MultiVector> weights,
00078                  int inputType=unspecified_input_);
00079   ZoltanLibClass(Teuchos::RCP<const Epetra_MultiVector> input_coords, int inputType=unspecified_input_);
00080   ZoltanLibClass(Teuchos::RCP<const Epetra_MultiVector> input_coords,
00081             Teuchos::RCP<const Epetra_MultiVector> weights, int inputType=unspecified_input_);
00082   ZoltanLibClass(Teuchos::RCP<const Epetra_BlockMap> input_map, int inputType=unspecified_input_);
00083 
00084 
00085 
00086 
00105   virtual int
00106   repartition(Teuchos::ParameterList& paramlist,
00107         std::vector<int>& newPartitions,
00108         int& exportsSize,
00109         std::vector<int>& imports);
00110 
00118   virtual int
00119   color(Teuchos::ParameterList& paramlist,
00120   std::vector<int>& colorAssignment);
00121 
00129   virtual int
00130   order(Teuchos::ParameterList& paramlist,
00131   std::vector<int>& orderAssignment);
00132 
00133 protected:
00134   virtual int precompute();
00135   virtual int postcompute();
00136   void computeCost();
00137   void preCheckPartition();
00138 
00139   void setParameterList(Teuchos::ParameterList& zoltanParamList);
00140 
00141 private:
00142   Teuchos::ParameterList zoltanParamList_;
00143   std::string partMethod_; // stores partitioning method used, perhaps should be in EpetraLibrary?
00144   Zoltan *zz_;
00145   Teuchos::RCP<ZoltanLib::QueryObject> queryObject_;
00146   int num_obj_;
00147 
00148 };//class ZoltanLibClass
00149 
00150 }//namespace Epetra
00151 }//namespace Isorropia
00152 
00153 #endif //HAVE_EPETRA
00154 
00155 #endif
00156