/* Compute and insert migration along directed edge (may fork child) */
static MIGRATION *
create_migration(RBFNODE *from_rbf, RBFNODE *to_rbf)
{
const double end_thresh = 5e-6;
PRICEMAT pmtx;
MIGRATION *newmig;
double *src_rem, *dst_rem;
double total_rem = 1., move_amt;
int i, j;
/* check if exists already */
for (newmig = from_rbf->ejl; newmig != NULL;
newmig = nextedge(from_rbf,newmig))
if (newmig->rbfv[1] == to_rbf)
return(NULL);
/* else allocate */
#ifdef DEBUG
fprintf(stderr, "Building path from (theta,phi) (%.0f,%.0f) ",
get_theta180(from_rbf->invec),
get_phi360(from_rbf->invec));
fprintf(stderr, "to (%.0f,%.0f) with %d x %d matrix\n",
get_theta180(to_rbf->invec),
get_phi360(to_rbf->invec),
from_rbf->nrbf, to_rbf->nrbf);
#endif
newmig = new_migration(from_rbf, to_rbf);
if (run_subprocess())
return(newmig); /* child continues */
price_routes(&pmtx, from_rbf, to_rbf);
src_rem = (double *)malloc(sizeof(double)*from_rbf->nrbf);
dst_rem = (double *)malloc(sizeof(double)*to_rbf->nrbf);
if ((src_rem == NULL) | (dst_rem == NULL)) {
fprintf(stderr, "%s: Out of memory in create_migration()\n",
progname);
exit(1);
}
/* starting quantities */
memset(newmig->mtx, 0, sizeof(float)*from_rbf->nrbf*to_rbf->nrbf);
for (i = from_rbf->nrbf; i--; )
src_rem[i] = rbf_volume(&from_rbf->rbfa[i]) / from_rbf->vtotal;
for (j = to_rbf->nrbf; j--; )
dst_rem[j] = rbf_volume(&to_rbf->rbfa[j]) / to_rbf->vtotal;
do { /* move a bit at a time */
move_amt = migration_step(newmig, src_rem, dst_rem, &pmtx);
total_rem -= move_amt;
} while ((total_rem > end_thresh) & (move_amt > 0));
for (i = from_rbf->nrbf; i--; ) { /* normalize final matrix */
double nf = rbf_volume(&from_rbf->rbfa[i]);
if (nf <= FTINY) continue;
nf = from_rbf->vtotal / nf;
for (j = to_rbf->nrbf; j--; )
mtx_coef(newmig,i,j) *= nf; /* row now sums to 1.0 */
}
end_subprocess(); /* exit here if subprocess */
free_routes(&pmtx); /* free working arrays */
free(src_rem);
free(dst_rem);
return(newmig);
}
/* Compute and insert migration along directed edge (may fork child) */
static MIGRATION *
create_migration(RBFNODE *from_rbf, RBFNODE *to_rbf)
{
MIGRATION *newmig;
int i, j;
/* check if exists already */
for (newmig = from_rbf->ejl; newmig != NULL;
newmig = nextedge(from_rbf,newmig))
if (newmig->rbfv[1] == to_rbf)
return(NULL);
/* else allocate */
#ifdef DEBUG
fprintf(stderr, "Building path from (theta,phi) (%.1f,%.1f) ",
get_theta180(from_rbf->invec),
get_phi360(from_rbf->invec));
fprintf(stderr, "to (%.1f,%.1f) with %d x %d matrix\n",
get_theta180(to_rbf->invec),
get_phi360(to_rbf->invec),
from_rbf->nrbf, to_rbf->nrbf);
#endif
newmig = new_migration(from_rbf, to_rbf);
if (run_subprocess())
return(newmig); /* child continues */
/* compute transport plan */
compute_nDSFs(from_rbf, to_rbf);
plan_transport(newmig);
for (i = from_rbf->nrbf; i--; ) { /* normalize final matrix */
double nf = rbf_volume(&from_rbf->rbfa[i]);
if (nf <= FTINY) continue;
nf = from_rbf->vtotal / nf;
for (j = to_rbf->nrbf; j--; )
mtx_coef(newmig,i,j) *= nf; /* row now sums to 1.0 */
}
end_subprocess(); /* exit here if subprocess */
return(newmig);
}
