Partitioner.hpp

Go to the documentation of this file.

// Copyright (C) 2010-2013 von Karman Institute for Fluid Dynamics, Belgium
//
// This software is distributed under the terms of the
// GNU Lesser General Public License version 3 (LGPLv3).
// See doc/lgpl.txt and doc/gpl.txt for the license text.

#ifndef cf3_mesh_zoltan_Partitioner_hpp
#define cf3_mesh_zoltan_Partitioner_hpp


#include "mesh/MeshPartitioner.hpp"
#include "mesh/zoltan/LibZoltan.hpp"

namespace cf3 {
namespace common { template <typename KEY, typename DATA> class Map;}
namespace mesh {
namespace zoltan {


class mesh_zoltan_API Partitioner : public MeshPartitioner {

public: // typedefs




public: // functions

  Partitioner ( const std::string& name );

  virtual ~Partitioner();

  static std::string type_name () { return "Partitioner"; }


  virtual void build_graph() { /* Does nothing, as partition_graph() will use call-back functions to construct the graph */ }

  virtual void partition_graph();

private: // functions

  Zoltan& zoltan_handle();

  void set_partitioning_params();

  // query function
  static int  query_nb_of_objects(void *data, int *ierr);

  static void query_list_of_objects(void *data, int sizeGID, int sizeLID,
                                   ZOLTAN_ID_PTR globalID, ZOLTAN_ID_PTR localID,
                                   int wgt_dim, float *obj_wgts, int *ierr);

  static void query_nb_connected_objects(void *data, int sizeGID, int sizeLID, int num_obj,
                                         ZOLTAN_ID_PTR globalID, ZOLTAN_ID_PTR localID,
                                         int *numEdges, int *ierr);

  static void query_list_of_connected_objects(void *data, int sizeGID, int sizeLID, int num_obj,
                                              ZOLTAN_ID_PTR globalID, ZOLTAN_ID_PTR localID,
                                              int *num_edges,
                                              ZOLTAN_ID_PTR nborGID, int *nborProc,
                                              int wgt_dim, float *ewgts, int *ierr);




private: // data

  bool m_partitioned;
  int changes;
  int numGidEntries;
  int numLidEntries;
  int numImport;
  ZOLTAN_ID_PTR importGlobalIds;
  ZOLTAN_ID_PTR importLocalIds;
  int *importProcs;
  int *importToPart;
  int numExport;
  ZOLTAN_ID_PTR exportGlobalIds;
  ZOLTAN_ID_PTR exportLocalIds;
  int *exportProcs;
  int *exportToPart;

  boost::shared_ptr<Zoltan> m_zz;
  Real m_zoltan_version;
};


} // zoltan
} // mesh
} // cf3


#endif // cf3_mesh_zoltan_Partitioner_hpp