cmiss_parmetis.f90

Go to the documentation of this file.

MODULE cmiss_parmetis
  
  USE base_routines
  USE kinds
  USE iso_varying_string
  
#include "macros.h"

  IMPLICIT NONE
 
  PRIVATE

  !Module parameters

  !Module types

  !Module variables

  !Interfaces
  INTERFACE

    SUBROUTINE parmetis_v3_partkway(vtxdist, xadj, adjncy, vwgt, adjwgt, wgtflag, numflag, ncon, nparts, tpwgts, ubvec, &
      & options, edgecut, part, comm)
#ifdef WIN32
      !DEC$ ATTRIBUTES C, reference, alias:'_ParMETIS_V3_PartKway' :: ParMETIS_V3_PartKway
#endif      
      USE kinds
      INTEGER(INTG) :: vtxdist(*)
      INTEGER(INTG) :: xadj(*)
      INTEGER(INTG) :: adjncy(*)
      INTEGER(INTG) :: vwgt(*)
      INTEGER(INTG) :: adjwgt(*)
      INTEGER(INTG) :: wgtflag
      INTEGER(INTG) :: numflag
      INTEGER(INTG) :: ncon
      INTEGER(INTG) :: nparts
      !REAL(SP) :: tpwgts(*)
      !REAL(SP) :: ubvec(*)
      REAL(DP) :: tpwgts(*)
      REAL(DP) :: ubvec(*)
      INTEGER(INTG) :: options(*)
      INTEGER(INTG) :: edgecut
      INTEGER(INTG) :: part(*)
      INTEGER(INTG) :: comm
    END SUBROUTINE parmetis_v3_partkway

    SUBROUTINE parmetis_v3_partmeshkway(elmdist, eptr, eind, elmwgt, wgtflag, numflag, ncon, ncommonnodes, nparts, tpwgts, &
      & ubvec, options, edgecut, part, comm)
#ifdef WIN32
      !DEC$ ATTRIBUTES C, reference, alias:'_ParMETIS_V3_PartMeshKway' :: ParMETIS_V3_PartMeshKway
#endif      
      USE kinds
      INTEGER(INTG) :: elmdist(*)
      INTEGER(INTG) :: eptr(*)
      INTEGER(INTG) :: eind(*)
      INTEGER(INTG) :: elmwgt(*)
      INTEGER(INTG) :: wgtflag
      INTEGER(INTG) :: numflag
      INTEGER(INTG) :: ncon
      INTEGER(INTG) :: ncommonnodes
      INTEGER(INTG) :: nparts
      !REAL(SP) :: tpwgts(*)
      !REAL(SP) :: ubvec(*)
      REAL(DP) :: tpwgts(*)
      REAL(DP) :: ubvec(*)
      INTEGER(INTG) :: options(*)
      INTEGER(INTG) :: edgecut
      INTEGER(INTG) :: part(*)
      INTEGER(INTG) :: comm      
    END SUBROUTINE parmetis_v3_partmeshkway
    
  END INTERFACE

  PUBLIC parmetis_partkway,parmetis_partmeshkway
  
CONTAINS

  !
  !================================================================================================================================
  !

  SUBROUTINE parmetis_partkway(VERTEX_DISTANCE,XADJ,ADJNCY,VERTEX_WEIGHT,ADJ_WEIGHT,WEIGHT_FLAG,NUM_FLAG,NCON, &
    & number_parts,tp_weights,ub_vec,options,number_edges_cut,partition,communicator,err,error,*)

    !Argument Variables
    INTEGER(INTG), INTENT(IN) :: VERTEX_DISTANCE(:)
    INTEGER(INTG), INTENT(IN) :: XADJ(:)
    INTEGER(INTG), INTENT(IN) :: ADJNCY(:)
    INTEGER(INTG), INTENT(IN) :: VERTEX_WEIGHT(:)
    INTEGER(INTG), INTENT(IN) :: ADJ_WEIGHT(:)
    INTEGER(INTG), INTENT(IN) :: WEIGHT_FLAG
    INTEGER(INTG), INTENT(IN) :: NUM_FLAG
    INTEGER(INTG), INTENT(IN) :: NCON
    INTEGER(INTG), INTENT(IN) :: NUMBER_PARTS
    !REAL(SP), INTENT(IN) :: TP_WEIGHTS(:)
    !REAL(SP), INTENT(IN) :: UB_VEC(:)
    REAL(DP), INTENT(IN) :: TP_WEIGHTS(:)
    REAL(DP), INTENT(IN) :: UB_VEC(:)
    INTEGER(INTG), INTENT(IN) :: OPTIONS(:)
    INTEGER(INTG), INTENT(OUT) :: NUMBER_EDGES_CUT
    INTEGER(INTG), INTENT(OUT) :: PARTITION(:)
    INTEGER(INTG), INTENT(IN) :: COMMUNICATOR
    INTEGER(INTG), INTENT(OUT) :: ERR
    TYPE(varying_string), INTENT(OUT) :: ERROR
    !Local Variables

    enters("PARMETIS_PARTKWAY",err,error,*999)

    CALL parmetis_v3_partkway(vertex_distance,xadj,adjncy,vertex_weight,adj_weight,weight_flag,num_flag,ncon, &
      & number_parts,tp_weights,ub_vec,options,number_edges_cut,partition,communicator)
    IF(err/=0) THEN
      CALL flagerror("ParMetis error in ParMETIS_V3_PartKway",err,error,*999)
    ENDIF
    
    exits("PARMETIS_PARTKWAY")
    RETURN
999 errorsexits("PARMETIS_PARTKWAY",err,error)
    RETURN 1
  END SUBROUTINE parmetis_partkway

  !
  !================================================================================================================================
  !

  SUBROUTINE parmetis_partmeshkway(ELEMENT_DISTANCE,ELEMENT_PTR,ELEMENT_INDEX,ELEMENT_WEIGHT,WEIGHT_FLAG,NUM_FLAG,NCON, &
    & number_common_nodes,number_parts,tp_weights,ub_vec,options,number_edges_cut,partition,communicator,err,error,*)

    !Argument Variables
    INTEGER(INTG), INTENT(IN) :: ELEMENT_DISTANCE(:)
    INTEGER(INTG), INTENT(IN) :: ELEMENT_PTR(:)
    INTEGER(INTG), INTENT(IN) :: ELEMENT_INDEX(:)
    INTEGER(INTG), INTENT(IN) :: ELEMENT_WEIGHT(:)
    INTEGER(INTG), INTENT(IN) :: WEIGHT_FLAG
    INTEGER(INTG), INTENT(IN) :: NUM_FLAG
    INTEGER(INTG), INTENT(IN) :: NCON
    INTEGER(INTG), INTENT(IN) :: NUMBER_COMMON_NODES
    INTEGER(INTG), INTENT(IN) :: NUMBER_PARTS
    !REAL(SP), INTENT(IN) :: TP_WEIGHTS(:)
    !REAL(SP), INTENT(IN) :: UB_VEC(:)
    REAL(DP), INTENT(IN) :: TP_WEIGHTS(:)
    REAL(DP), INTENT(IN) :: UB_VEC(:)
    INTEGER(INTG), INTENT(IN) :: OPTIONS(:)
    INTEGER(INTG), INTENT(OUT) :: NUMBER_EDGES_CUT
    INTEGER(INTG), INTENT(OUT) :: PARTITION(:)
    INTEGER(INTG), INTENT(IN) :: COMMUNICATOR
    INTEGER(INTG), INTENT(OUT) :: ERR
    TYPE(varying_string), INTENT(OUT) :: ERROR
    !Local Variables

    enters("PARMETIS_PARTMESHKWAY",err,error,*999)
    
    CALL parmetis_v3_partmeshkway(element_distance,element_ptr,element_index,element_weight,weight_flag,num_flag,ncon, &
      & number_common_nodes,number_parts,tp_weights,ub_vec,options,number_edges_cut,partition,communicator)
    
    IF(err/=0) THEN
      CALL flagerror("ParMetis error in ParMETIS_V3_PartMeshKway",err,error,*999)
    ENDIF
    
    exits("PARMETIS_PARTMESHKWAY")
    RETURN
999 errorsexits("PARMETIS_PARTMESHKWAY",err,error)
    RETURN 1
  END SUBROUTINE parmetis_partmeshkway

  !
  !================================================================================================================================
  !
    
END MODULE cmiss_parmetis