From 0cc68580b72ff568a0a73b3e04037fc24cf78d88 Mon Sep 17 00:00:00 2001
From: Thilina Rathnayake <thilinarmtb@users.noreply.github.com>
Date: Mon, 20 Jul 2020 05:38:53 -0500
Subject: [PATCH] Changes to parRCB interface (#21)

---
 src/parRCB.c | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/parRCB.c b/src/parRCB.c
index f6396648..5151c355 100644
--- a/src/parRCB.c
+++ b/src/parRCB.c
@@ -15,6 +15,7 @@ void fparRCB_partMesh(int *part,double *vtx,int *nel,int *nv,
   *err=parRCB_partMesh(part,vtx,*nel,*nv,options,c);
 }
 
+// vtx = [nel,nv,ndim]
 int parRCB_partMesh(int *part,double *vtx,int nel,int nv,
   int *options,MPI_Comm comm)
 {
@@ -33,12 +34,18 @@ int parRCB_partMesh(int *part,double *vtx,int nel,int nv,
 
   int ndim=(nv==8)?3:2;
 
-  int e, n;
+  int e,n,v;
+
   for(e=0;e<nel;++e){
     data[e].id=nelg_start+(e+1);
     data[e].orig=rank;
-    for(int n=0;n<ndim;n++)
-      data[e].coord[n]=vtx[e*ndim+n];
+    data[e].coord[0]=data[e].coord[1]=data[e].coord[2]=0.0;
+    for(v=0;v<nv;v++){
+      for(n=0;n<ndim;n++)
+        data[e].coord[n]+=vtx[e*ndim*nv+v*ndim+n];
+    }
+    for(n=0;n<ndim;n++)
+      data[e].coord[n]/=8;
   }
   a.n=nel;