Zoltan_LoadBalance.h

Go to the documentation of this file.

// @HEADER
// ************************************************************************
// 
//            Zoltan_CPP: An Object-Oriented Interface To Zoltan
//                    Copyright (2001) Sandia Corporation
// 
// Questions? Contact Robert J. Hoekstra (rjhoeks@sandia.gov)
// 
// ************************************************************************
// @HEADER

#ifndef ZOLTAN_LOADBALANCE_H_
#define ZOLTAN_LOADBALANCE_H_

#include <EpetraExt_ConfigDefs.h>

#include <zoltan.h>

#include <string>

namespace Zoltan {

class QueryObject;
class MigrationObject;


class LoadBalance
{

public:



 LoadBalance   ( int argc = 0,
                 char ** argv = 0,
                 float * ver = 0 );

 ~LoadBalance();


 
 int Create( MPI_Comm communicator );

 int Set_QueryObject( QueryObject * query_obj_ptr );

 int Set_MigrationObject( MigrationObject * migration_obj_ptr );

 int Set_Param( const std::string & param, const std::string & value );

 int Balance  ( int * changes,
                int * num_gid_entries,
                int * num_lid_entries,
    int * num_import,
    ZOLTAN_ID_PTR * import_global_ids,
    ZOLTAN_ID_PTR * import_local_ids,
    int ** import_procs,
    int * num_export,
    ZOLTAN_ID_PTR * export_global_ids,
    ZOLTAN_ID_PTR * export_local_ids,
    int ** export_procs );

#ifdef ZOLTAN_ORDER
 int Order    ( int * num_gid_entries,
                int * num_lid_entries,
                int num_objs,
    ZOLTAN_ID_PTR global_ids,
    ZOLTAN_ID_PTR local_ids,
                int * rank,
                int * iperm );
#endif

 void Evaluate( int print_stats,
    int * num_objects,
    float * object_weights,
        int * num_cut,
        float * cut_weights,
    int * num_boundary_objects,
    int * num_adj_procs );

 int Free_Data( ZOLTAN_ID_PTR * import_global_ids,
    ZOLTAN_ID_PTR * import_local_ids,
    int ** import_procs,
    ZOLTAN_ID_PTR * export_global_ids,
    ZOLTAN_ID_PTR * export_local_ids,
    int ** export_procs );



#ifdef ZOLTAN_OLD_CALLBACK

 int Set_CallBack_Fn  ( ZOLTAN_FN_TYPE fn_type,
      void * fn_ptr,
      void * data = 0 );

#else /* ZOLTAN_OLD_CALLBACK */


 int Set_Num_Edges_Fn        ( ZOLTAN_NUM_EDGES_FN * fn_ptr,
                              void * data = 0 );
 int Set_Edge_List_Fn        ( ZOLTAN_EDGE_LIST_FN * fn_ptr,
                              void * data = 0 );
 int Set_Num_Geom_Fn         ( ZOLTAN_NUM_GEOM_FN * fn_ptr,
                              void * data = 0 );
 int Set_Geom_Fn             ( ZOLTAN_GEOM_FN * fn_ptr,
                              void * data = 0 );
 int Set_Num_Obj_Fn          ( ZOLTAN_NUM_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Obj_List_Fn         ( ZOLTAN_OBJ_LIST_FN * fn_ptr,
                              void * data = 0 );
 int Set_First_Obj_Fn        ( ZOLTAN_FIRST_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Next_Obj_Fn         ( ZOLTAN_NEXT_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Num_Border_Obj_Fn   ( ZOLTAN_NUM_BORDER_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Border_Obj_List_Fn  ( ZOLTAN_BORDER_OBJ_LIST_FN * fn_ptr,
                              void * data = 0 );
 int Set_First_Border_Obj_Fn ( ZOLTAN_FIRST_BORDER_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Next_Border_Obj_Fn  ( ZOLTAN_NEXT_BORDER_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Num_Coarse_Obj_Fn   ( ZOLTAN_NUM_COARSE_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Coarse_Obj_List_Fn  ( ZOLTAN_COARSE_OBJ_LIST_FN * fn_ptr,
                              void * data = 0 );
 int Set_First_Coarse_Obj_Fn ( ZOLTAN_FIRST_COARSE_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Next_Coarse_Obj_Fn  ( ZOLTAN_NEXT_COARSE_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Num_Child_Fn        ( ZOLTAN_NUM_CHILD_FN * fn_ptr,
                              void * data = 0 );
 int Set_Child_List_Fn       ( ZOLTAN_CHILD_LIST_FN * fn_ptr,
                              void * data = 0 );
 int Set_Child_Weight_Fn     ( ZOLTAN_CHILD_WEIGHT_FN * fn_ptr,
                              void * data = 0 );
 int Set_Pre_Migrate_Fn      ( ZOLTAN_PRE_MIGRATE_FN * fn_ptr,
                              void * data = 0 );
 int Set_Mid_Migrate_Fn      ( ZOLTAN_MID_MIGRATE_FN * fn_ptr,
                              void * data = 0 );
 int Set_Post_Migrate_Fn     ( ZOLTAN_POST_MIGRATE_FN * fn_ptr,
                              void * data = 0 );
 int Set_Obj_Size_Fn         ( ZOLTAN_OBJ_SIZE_FN * fn_ptr,
                              void * data = 0 );
 int Set_Pack_Obj_Fn         ( ZOLTAN_PACK_OBJ_FN * fn_ptr,
                              void * data = 0 );
 int Set_Unpack_Obj_Fn       ( ZOLTAN_UNPACK_OBJ_FN * fn_ptr,
                              void * data = 0 );

#endif /* ZOLTAN_OLD_CALLBACK */



 int Point_Assign     ( double * coords,
      int * proc );

 int Box_Assign       ( double xmin,
      double ymin,
      double zmin,
      double xmax,
      double ymax,
      double zmax,
      int * procs,
      int * numprocs );

 

 int Compute_Destinations     ( int num_import,
        ZOLTAN_ID_PTR import_global_ids,
        ZOLTAN_ID_PTR import_local_ids,
        int * import_procs,
        int * num_export,
        ZOLTAN_ID_PTR * export_global_ids,
        ZOLTAN_ID_PTR * export_local_ids,
        int ** export_procs );

 int Help_Migrate     ( int num_import,
      ZOLTAN_ID_PTR import_global_ids,
      ZOLTAN_ID_PTR import_local_ids,
      int * import_procs,
      int num_export,
      ZOLTAN_ID_PTR export_global_ids,
      ZOLTAN_ID_PTR export_local_ids,
      int * export_procs );


 
 Zoltan_Struct * Return_Zoltan_Struct()
 { return LoadBalance_Ptr_; }


private:

 static int ObjectCount;

 int ObjectID;

 Zoltan_Struct * LoadBalance_Ptr_; 

 QueryObject * QueryObject_Ptr_;
 MigrationObject * MigrationObject_Ptr_;

};

} //namespace Zoltan

#endif