Cloned SEACAS for EXODUS library with extra build files for internal package management.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

75 lines
2.3 KiB

2 years ago
/*
* Copyright(C) 1999-2020 National Technology & Engineering Solutions
* of Sandia, LLC (NTESS). Under the terms of Contract DE-NA0003525 with
* NTESS, the U.S. Government retains certain rights in this software.
*
* See packages/seacas/LICENSE for details
*/
#include "refine_map.h" // for refine_vdata
#include "structs.h" // for vtx_data
void compute_mesh_vdata(struct refine_vdata *vdata, /* preference data for a vertex */
struct vtx_data ** comm_graph, /* communication graph data structure */
int vtx, /* current vertex */
int * vtx2node, /* maps graph vtxs to mesh nodes */
int mesh_dims[3], /* size of mesh */
int dim /* dimension we are currently working in */
)
{
float above; /* my preference to move up in each dimension */
float below; /* my preference to move down in each dimension */
float same; /* my preference to stay where I am */
int my_loc; /* my location in mesh */
int neighb_loc; /* neighbor's location in mesh */
float ewgt; /* weight of an edge */
int node; /* set vertex is assigned to */
int neighbor; /* neighboring vtx in comm_graph */
int j; /* loop counter */
node = vtx2node[vtx];
neighb_loc = 0;
my_loc = 0;
if (dim == 0) {
my_loc = node % mesh_dims[0];
}
else if (dim == 1) {
my_loc = (node / mesh_dims[0]) % mesh_dims[1];
}
else if (dim == 2) {
my_loc = node / (mesh_dims[0] * mesh_dims[1]);
}
below = above = same = 0;
for (j = 1; j < comm_graph[vtx]->nedges; j++) {
neighbor = comm_graph[vtx]->edges[j];
ewgt = comm_graph[vtx]->ewgts[j];
node = vtx2node[neighbor];
if (dim == 0) {
neighb_loc = node % mesh_dims[0];
}
else if (dim == 1) {
neighb_loc = (node / mesh_dims[0]) % mesh_dims[1];
}
else if (dim == 2) {
neighb_loc = node / (mesh_dims[0] * mesh_dims[1]);
}
if (neighb_loc < my_loc) {
below += ewgt;
}
else if (neighb_loc > my_loc) {
above += ewgt;
}
else {
same += ewgt;
}
}
vdata->below = below;
vdata->above = above;
vdata->same = same;
}