hecmw_partitioner.c

Go to the documentation of this file.

/*****************************************************************************
 * Copyright (c) 2019 FrontISTR Commons
 * This software is released under the MIT License, see LICENSE.txt
 *****************************************************************************/
 
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <errno.h>
#include <limits.h>
 
#include "hecmw_util.h"
#include "hecmw_io.h"
#include "hecmw_init_for_partition.h"
#include "hecmw_partition.h"
 
int main(int argc, char **argv) {
  struct hecmwST_local_mesh *global_mesh = NULL;
  struct hecmwST_local_mesh *local_mesh  = NULL;
  int rtc;
 
  rtc = HECMW_init(&argc, &argv);
  if (rtc != 0) goto error;
 
  rtc = HECMW_init_for_partition(argc, argv);
  if (rtc != 0) goto error;
 
  HECMW_log(HECMW_LOG_INFO, "Reading mesh file...");
  global_mesh = HECMW_get_mesh("part_in");
  if (global_mesh == NULL) goto error;
 
  /* Debug output for large mesh analysis */
  HECMW_log(HECMW_LOG_INFO, "Mesh loaded: nodes=%d, elements=%d", 
            global_mesh->n_node, global_mesh->n_elem);
  
  if (global_mesh->elem_node_index && global_mesh->n_elem > 0) {
    size_t total_connectivity = global_mesh->elem_node_index[global_mesh->n_elem];
    if (total_connectivity > INT_MAX) {
      HECMW_log(HECMW_LOG_INFO, "Total element connectivity entries: %zu", total_connectivity);
      HECMW_log(HECMW_LOG_WARN, "WARNING: Element connectivity count exceeds 32-bit integer limit %d!", INT_MAX);
    }
  }
 
  local_mesh = HECMW_partition(global_mesh);
  if (local_mesh == NULL) goto error;
 
  HECMW_dist_free(global_mesh);
 
  HECMW_finalize();
 
  return 0;
 
error:
  HECMW_dist_free(global_mesh);
  HECMW_finalize();
  HECMW_abort(HECMW_comm_get_comm());
 
  return -1;
}