int ESMC_VMGet(ESMC_VM vm, int *localPet, int *petCount, int *peCount,
MPI_Comm *mpiCommunicator, int *pthreadsEnabledFlag, int *openMPEnabledFlag){
#undef ESMC_METHOD
#define ESMC_METHOD "ESMC_VMGet()"
// initialize return code; assume routine not implemented
int rc = ESMC_RC_NOT_IMPL; // final return code
ESMCI::VM *vmp = (ESMCI::VM*)(vm.ptr);
*localPet = vmp->getLocalPet();
*petCount = vmp->getPetCount();
//compute peCount
int npets = vmp->getNpets();
*peCount = 0; // reset
for (int i=0; i<npets; i++)
*peCount += vmp->getNcpet(i);
*mpiCommunicator = vmp->getMpi_c();
*pthreadsEnabledFlag = vmp->isPthreadsEnabled();
*openMPEnabledFlag = vmp->isOpenMPEnabled();
// return successfully
rc = ESMF_SUCCESS;
return rc;
}
extern "C" void FTN(c_esmc_regrid_getiwts)(ESMCI::VM **vmpp, Grid **gridpp,
Mesh **meshpp, ESMCI::Array **arraypp, int *staggerLoc,
int *regridScheme, int*rc) {
#undef ESMC_METHOD
#define ESMC_METHOD "c_esmc_regrid_getiwts()"
Trace __trace(" FTN(regrid_getiwts)()");
ESMCI::VM *vm = *vmpp;
ESMCI::Array &array = **arraypp;
int localPet = vm->getLocalPet();
int petCount = vm->getPetCount();
Mesh &mesh = **meshpp;
Grid &grid = **gridpp;
try {
// Get the integration weights
MEField<> *iwts = mesh.GetField("iwts");
if (!iwts) Throw() << "Could not find integration weights field on this mesh"
<<std::endl;
if(!get_iwts(mesh, iwts, regridScheme))
Throw() << "Online regridding error" << std::endl;
CpMeshDataToArray(grid, *staggerLoc, mesh, array, iwts);
} catch(std::exception &x) {
// catch Mesh exception return code
if (x.what()) {
ESMC_LogDefault.ESMC_LogMsgFoundError(ESMC_RC_INTNRL_BAD,
x.what(), rc);
} else {
ESMC_LogDefault.ESMC_LogMsgFoundError(ESMC_RC_INTNRL_BAD,
"UNKNOWN", rc);
}
return;
} catch(int localrc){
// catch standard ESMF return code
ESMC_LogDefault.ESMC_LogMsgFoundError(localrc, ESMF_ERR_PASSTHRU, rc);
return;
} catch(...){
ESMC_LogDefault.ESMC_LogMsgFoundError(ESMC_RC_INTNRL_BAD,
"- Caught unknown exception", rc);
return;
}
// Set return code
if (rc!=NULL) *rc = ESMF_SUCCESS;
}
extern "C" void FTN(c_esmc_regrid_create)(ESMCI::VM **vmpp,
Mesh **meshsrcpp, ESMCI::Array **arraysrcpp, int *srcstaggerLoc,
Mesh **meshdstpp, ESMCI::Array **arraydstpp, int *dststaggerLoc,
int *regridMethod, int *regridConserve,
int *regridScheme, int *unmappedaction,
ESMCI::RouteHandle **rh, int *has_rh, int *has_iw,
int *nentries, ESMCI::TempWeights **tweights,
int*rc) {
#undef ESMC_METHOD
#define ESMC_METHOD "c_esmc_regrid_create()"
Trace __trace(" FTN(regrid_test)(ESMCI::VM **vmpp, ESMCI::Grid **gridsrcpp, int *srcstaggerLoc, ESMCI::Grid **griddstcpp, int *dststaggerLoc, int*rc");
ESMCI::VM *vm = *vmpp;
ESMCI::Array &srcarray = **arraysrcpp;
ESMCI::Array &dstarray = **arraydstpp;
int localPet = vm->getLocalPet();
int petCount = vm->getPetCount();
Mesh &srcmesh = **meshsrcpp;
Mesh &dstmesh = **meshdstpp;
try {
// Weights matrix
IWeights wts;
if(!online_regrid(srcmesh, dstmesh, wts, regridConserve, regridMethod,
regridScheme, unmappedaction))
Throw() << "Online regridding error" << std::endl;
//wts.Print(Par::Out());
// We have the weights, now set up the sparsemm object
// Firstly, the index list
std::pair<UInt,UInt> iisize = wts.count_matrix_entries();
int num_entries = iisize.first;
int *iientries = new int[2*iisize.first];
int larg[2] = {2, iisize.first};
// Gather the list
ESMCI::InterfaceInt ii(iientries, 2, larg);
ESMCI::InterfaceInt *iiptr = ⅈ
double *factors = new double[iisize.first];
// Gather the data
WMat::WeightMap::iterator wi = wts.begin_row(), we = wts.end_row();
UInt i = 0;
for (; wi != we; ++wi) {
const WMat::Entry &w = wi->first;
std::vector<WMat::Entry> &wcol = wi->second;
for (UInt j = 0; j < wcol.size(); ++j) {
UInt twoi = 2*i;
const WMat::Entry &wc = wcol[j];
// I expected things to be (dst,src), but they are (src_id,dst_id)
//iientries[twoi] = w.id; iientries[twoi+1] = wc.id;
iientries[twoi+1] = w.id; iientries[twoi] = wc.id;
factors[i] = wc.value;
i++;
} // for j
} // for wi
/*
Par::Out() << "Matrix entries" << std::endl;
for (UInt n = 0; n < num_entries; ++n) {
Par::Out() << std::setw(5) << iientries[2*n] << std::setw(5) << iientries[2*n+1] << std::setw(15) << factors[n] << std::endl;
}
wts.Print(Par::Out());
*/
// Build the ArraySMM
if (*has_rh != 0) {
int localrc;
enum ESMC_TypeKind tk = ESMC_TYPEKIND_R8;
FTN(c_esmc_arraysmmstore)(arraysrcpp, arraydstpp, rh, &tk, factors,
&num_entries, &iiptr, &localrc);
if (ESMC_LogDefault.ESMC_LogMsgFoundError(localrc,ESMF_ERR_PASSTHRU,NULL))
throw localrc; // bail out with exception
}
*nentries = num_entries;
// Clean up. If has_iw, then we will use the arrays to
// fill out the users pointers. These will be deleted following a copy.
if (*has_iw == 0) {
delete [] factors;
delete [] iientries;
*nentries = 0;
} else {
// Save off the weights so the F90 caller can allocate arrays and
// copy the values.
*tweights = new ESMCI::TempWeights;
(*tweights)->nentries = num_entries;
(*tweights)->factors = factors;
(*tweights)->iientries = iientries;
}
} catch(std::exception &x) {
// catch Mesh exception return code
if (x.what()) {
ESMC_LogDefault.ESMC_LogMsgFoundError(ESMC_RC_INTNRL_BAD,
x.what(), rc);
} else {
ESMC_LogDefault.ESMC_LogMsgFoundError(ESMC_RC_INTNRL_BAD,
"UNKNOWN", rc);
}
return;
} catch(int localrc){
// catch standard ESMF return code
ESMC_LogDefault.ESMC_LogMsgFoundError(localrc, ESMF_ERR_PASSTHRU, rc);
return;
} catch(...){
ESMC_LogDefault.ESMC_LogMsgFoundError(ESMC_RC_INTNRL_BAD,
"- Caught unknown exception", rc);
return;
}
// Set return code
if (rc!=NULL) *rc = ESMF_SUCCESS;
}