MueLu_ZoltanInterface_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//        MueLu: A package for multigrid based preconditioning
//
// Copyright 2012 NTESS and the MueLu contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef MUELU_ZOLTANINTERFACE_DECL_HPP
#define MUELU_ZOLTANINTERFACE_DECL_HPP
 
#include "MueLu_ConfigDefs.hpp"
 
#if defined(HAVE_MUELU_ZOLTAN) && defined(HAVE_MPI)
 
#include <zoltan_cpp.h>
 
#include <Xpetra_Matrix_fwd.hpp>
#include <Xpetra_VectorFactory_fwd.hpp>
 
#include "MueLu_SingleLevelFactoryBase.hpp"
#include "MueLu_ZoltanInterface_fwd.hpp"
 
#include "MueLu_Level_fwd.hpp"
#include "MueLu_FactoryBase_fwd.hpp"
 
namespace MueLu {
 
// FIXME: this class should not be templated
template <class Scalar,
          class LocalOrdinal  = DefaultLocalOrdinal,
          class GlobalOrdinal = DefaultGlobalOrdinal,
          class Node          = DefaultNode>
class ZoltanInterface : public SingleLevelFactoryBase {
#undef MUELU_ZOLTANINTERFACE_SHORT
#include "MueLu_UseShortNames.hpp"
 
 public:
 
 
  ZoltanInterface() {}
 
  virtual ~ZoltanInterface() {}
 
  RCP<const ParameterList> GetValidParameterList() const;
 
 
  void DeclareInput(Level &level) const;
 
 
  void Build(Level &level) const;
 
 
 
 
  static int GetLocalNumberOfRows(void *data, int *ierr);
 
  static void GetLocalNumberOfNonzeros(void *data, int NumGidEntries, int NumLidEntries, ZOLTAN_ID_PTR gids,
                                       ZOLTAN_ID_PTR lids, int wgtDim, float *weights, int *ierr);
 
  static int GetProblemDimension(void *data, int *ierr);
 
  static void GetProblemGeometry(void *data, int numGIDEntries, int numLIDEntries, int numObjectIDs,
                                 ZOLTAN_ID_PTR gids, ZOLTAN_ID_PTR lids, int dim, double *coordinates, int *ierr);
 
 
 private:
  static ArrayRCP<typename Teuchos::ScalarTraits<Scalar>::magnitudeType> coalesceCoordinates(ArrayRCP<typename Teuchos::ScalarTraits<Scalar>::magnitudeType> coord, LocalOrdinal blksize);
 
};  // class ZoltanInterface
 
}  // namespace MueLu
 
#define MUELU_ZOLTANINTERFACE_SHORT
#endif  // if defined(HAVE_MUELU_ZOLTAN) && defined(HAVE_MPI)
 
#endif  // MUELU_ZOLTANINTERFACE_DECL_HPP