static void gfs_refine_surface_refine (GfsRefine * refine, GfsSimulation * sim)
{
RefineCut p;
p.refine = refine;
p.domain = GFS_DOMAIN (sim);
p.surface = GFS_REFINE_SURFACE (refine)->surface;
if (GFS_SURFACE (p.surface)->twod) {
if (GFS_SURFACE (p.surface)->s)
gfs_domain_traverse_cut_2D (GFS_DOMAIN (sim), GFS_REFINE_SURFACE (refine)->surface,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS,
(FttCellTraverseCutFunc) refine_cut_cell, &p);
else
g_assert_not_implemented ();
}
else {
if (GFS_SURFACE (p.surface)->s)
gfs_domain_traverse_cut (GFS_DOMAIN (sim), GFS_REFINE_SURFACE (refine)->surface,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS,
(FttCellTraverseCutFunc) refine_cut_cell, &p);
else
gfs_domain_cell_traverse (GFS_DOMAIN (sim),
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) refine_implicit_cell, &p);
}
}
static gboolean gfs_init_wave_event (GfsEvent * event, GfsSimulation * sim)
{
if ((* GFS_EVENT_CLASS (GTS_OBJECT_CLASS (gfs_init_wave_class ())->parent_class)->event)
(event, sim)) {
gfs_catch_floating_point_exceptions ();
gfs_domain_cell_traverse (GFS_DOMAIN (sim), FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) init_energy, event);
gfs_restore_fpe_for_function (GFS_INIT_WAVE (event)->d);
gfs_catch_floating_point_exceptions ();
gfs_domain_cell_traverse (GFS_DOMAIN (sim), FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) scale_energy, event);
gfs_restore_fpe_for_function (GFS_INIT_WAVE (event)->hs);
return TRUE;
}
return FALSE;
}
void gfs_source_darcy_implicit (GfsDomain * domain,
gdouble dt)
{
GfsSourceDarcy * s;
g_return_if_fail (domain != NULL);
if ((s = gfs_has_source_darcy (domain))) {
GfsSimulation * sim = GFS_SIMULATION (domain);
gdouble olddt = sim->advection_params.dt;
sim->advection_params.dt = dt;
gfs_catch_floating_point_exceptions ();
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) (FTT_DIMENSION==3?
implicit_darcy_3D : implicit_darcy_2D), s);
if (gfs_restore_floating_point_exceptions ()) {
gchar * c = g_strconcat ("\n", gfs_function_description (s->darcycoeff, FALSE), NULL);
if (s->forchhicoeff)
c = g_strconcat (c, "\n", gfs_function_description (s->forchhicoeff, FALSE), NULL);
/* fixme: memory leaks */
g_message ("floating-point exception in user-defined function(s):%s", c);
exit (1);
}
sim->advection_params.dt = olddt;
}
}
static void swap_face_fractions (GfsSimulation * sim)
{
GfsDomain * domain = GFS_DOMAIN (sim);
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_ALL, -1,
(FttCellTraverseFunc) swap_fractions,
GFS_SIMULATION_MOVING (sim)->old_solid);
gts_container_foreach (GTS_CONTAINER (domain), (GtsFunc) foreach_box, NULL);
}
static void gse_alleviation_diffusion (GfsDomain * domain, GfsVariable * F,
FttVector * cg, gdouble dt)
{
gfs_domain_timer_start (domain, "gse_alleviation");
gdouble ncg = sqrt (cg->x*cg->x + cg->y*cg->y);
gdouble dcg = (GFS_WAVE_GAMMA - 1./GFS_WAVE_GAMMA)*ncg/2.;
gdouble dtheta = 2.*M_PI/GFS_WAVE (domain)->ntheta;
#if 0
gdouble Ts = 4.*GFS_WAVE (domain)->alpha_s*GFS_WAVE (domain)->alpha_s*dt;
gdouble dtDss = dt*dcg*dcg*Ts/12.;
gdouble dtDnn = dt*(ncg*dtheta)*(ncg*dtheta)*Ts/12.;
#else
gdouble alpha = GFS_WAVE (domain)->alpha_s*dcg*dt;
gdouble beta = GFS_WAVE (domain)->alpha_s*ncg*dtheta*dt;
gdouble dtDss = alpha*alpha/3.;
gdouble dtDnn = beta*beta/3.;
#endif
GSEData p;
gdouble cost = cg->x/ncg, sint = cg->y/ncg;
p.D[0][0] = dtDss*cost*cost + dtDnn*sint*sint;
p.D[1][1] = dtDss*sint*sint + dtDnn*cost*cost;
p.D[0][1] = p.D[1][0] = (dtDss - dtDnn)*cost*sint;
p.F = F;
p.Fn = gfs_temporary_variable (domain);
p.dF = gfs_temporary_variable (domain);
gfs_domain_traverse_leaves (domain, (FttCellTraverseFunc) copy_F, &p);
gfs_domain_cell_traverse (domain, FTT_POST_ORDER, FTT_TRAVERSE_NON_LEAFS, -1,
(FttCellTraverseFunc) p.Fn->fine_coarse, p.Fn);
for (p.dF->component = 0; p.dF->component < 2; p.dF->component++) {
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_ALL, -1,
(FttCellTraverseFunc) compute_gradient, &p);
gfs_domain_bc (domain, FTT_TRAVERSE_ALL, -1, p.dF);
gfs_domain_traverse_leaves (domain, (FttCellTraverseFunc) diffusion, &p);
}
gts_object_destroy (GTS_OBJECT (p.Fn));
gts_object_destroy (GTS_OBJECT (p.dF));
gfs_domain_timer_stop (domain, "gse_alleviation");
}
void gfs_poisson_coefficients_por (GfsDomain * domain,
GfsPorous *por,
GfsFunction * alpha,
gboolean positive,
gboolean centered,
gboolean reset)
{
PoissonCoeff_por p;
FttComponent i;
g_return_if_fail (domain != NULL);
for (i = 0; i < FTT_DIMENSION; i++) {
gdouble lambda = (&domain->lambda.x)[i];
p.lambda2[i] = lambda*lambda;
}
p.alpha = alpha;
p.domain = domain;
p.positive = positive;
p.phi = por->porosity;
if (reset)
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_ALL, -1,
(FttCellTraverseFunc) reset_coeff_por, &p);
if (!centered)
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_ALL, -1,
(FttCellTraverseFunc) poisson_mixed_coeff_por, &p);
gfs_domain_face_traverse (domain, FTT_XYZ,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttFaceTraverseFunc) poisson_coeff_por, &p);
gfs_domain_cell_traverse (domain,
FTT_POST_ORDER, FTT_TRAVERSE_NON_LEAFS, -1,
(FttCellTraverseFunc) face_coeff_from_below_por, NULL);
}
void gfs_post_projection (GfsDomain *domain, GfsPorous *por, guint dimension)
{
GfsVariable **v;
FttComponent c;
gpointer data[3];
g_return_if_fail (domain != NULL);
data[0] = v = gfs_domain_velocity (domain);
data[1] = &dimension;
data[2] = por;
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1, (FttCellTraverseFunc) correct_por_undo, data);
for (c = 0; c < dimension; c++)
gfs_domain_bc (domain, FTT_TRAVERSE_LEAFS, -1, v[c]);
printf("\nvelocity correction undo !!!\n");
}
static gboolean gfs_init_stokes_wave_event (GfsEvent * event, GfsSimulation * sim)
{
if ((* GFS_EVENT_CLASS (GTS_OBJECT_CLASS (gfs_init_stokes_wave_class ())->parent_class)->event)
(event, sim)) {
GfsVariable ** velocity = gfs_domain_velocity (GFS_DOMAIN (sim));
GfsVariable * t = gfs_variable_from_name (GFS_DOMAIN (sim)->variables, "T");
g_assert (velocity);
g_assert (t);
gfs_domain_cell_traverse (GFS_DOMAIN (sim), FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) init_velocity, velocity);
GfsSurface * surface = GFS_SURFACE (gts_object_new (GTS_OBJECT_CLASS (gfs_surface_class ())));
surface->f = gfs_function_spatial_new (gfs_function_spatial_class (), stokes_height);
gfs_object_simulation_set (surface->f, sim);
gfs_domain_init_fraction (GFS_DOMAIN (sim), GFS_GENERIC_SURFACE (surface), t);
gts_object_destroy (GTS_OBJECT (surface));
return TRUE;
}
return FALSE;
}
static void gfs_refine_solid_refine (GfsRefine * refine, GfsSimulation * sim)
{
if (sim->solids) {
RefineCut p;
p.refine = refine;
p.domain = GFS_DOMAIN (sim);
GSList * i = sim->solids->items;
while (i) {
p.surface = GFS_SOLID (i->data)->s;
gfs_catch_floating_point_exceptions ();
if (GFS_SURFACE (p.surface)->s)
gfs_domain_traverse_cut (GFS_DOMAIN (sim), p.surface,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS,
(FttCellTraverseCutFunc) refine_cut_cell, &p);
else
gfs_domain_cell_traverse (GFS_DOMAIN (sim),
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) refine_implicit_cell, &p);
gfs_restore_fpe_for_function (refine->maxlevel);
i = i->next;
}
}
}
static void gfs_skew_symmetric_momentum (GfsSimulation * sim, FaceData * fd, GfsVariable **gmac)
{
GfsDomain * domain = GFS_DOMAIN (sim);
FttComponent c;
FttDirection d;
/* it is used for implementation of viscosity (improve?) */
GfsSourceDiffusion * dif = source_diffusion_viscosity (fd->u[0]);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) advance_face_values, fd);
/* boundary conditions */
for (d = 0; d < FTT_NEIGHBORS; d++)
gfs_domain_bc (domain, FTT_TRAVERSE_LEAFS, -1, fd->velfaces[d]);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) advection_term, fd);
if (dif) {
DataDif dd = { dif , sim->physical_params.alpha, fd };
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) diffusion_term, &dd);
}
/* regularize flux at faces */
for (c = 0; c < FTT_DIMENSION; c++)
gfs_domain_face_bc (domain, c, fd->u[c]);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) obtain_face_fluxes, NULL);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) update_vel, fd);
gfs_velocity_face_sources (domain, fd->u, (*fd->dt), sim->physical_params.alpha, gmac);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) correct_face_velocity, NULL);
}
static void swap_face_fractions_back (GfsSimulation * sim)
{
gfs_domain_cell_traverse (GFS_DOMAIN (sim), FTT_PRE_ORDER, FTT_TRAVERSE_ALL, -1,
(FttCellTraverseFunc) swap_fractions_back,
GFS_SIMULATION_MOVING (sim)->old_solid);
}
extern field_t* field_read_gfs(const char* file)
{
int argc = 0;
gfs_init(&argc,NULL);
FILE *st = fopen(file,"r");
if (!st)
{
fprintf(stderr,"failed to open %s\n",file);
return NULL;
}
GtsFile *fp = gts_file_new(st);
GfsSimulation *sim = gfs_simulation_read(fp);
if (!sim)
{
fprintf(stderr,
"file %s not a valid simulation file\n"
"line %d:%d: %s\n",
file,
fp->line,
fp->pos,
fp->error);
return NULL;
}
gts_file_destroy(fp);
fclose(st);
GfsDomain *gdom = GFS_DOMAIN(sim);
/* find the bounding box */
bbox_t *bb = NULL;
gfs_domain_cell_traverse(gdom,
FTT_PRE_ORDER,
FTT_TRAVERSE_NON_LEAFS,
0,
(FttCellTraverseFunc)ftt_bbox,
&bb);
if (!bb)
{
fprintf(stderr,"failed to determine bounding box\n");
return NULL;
}
#ifdef FRG_DEBUG
fprintf(stdout,
"%g %g %g %g\n",
bb->x.min,
bb->x.max,
bb->y.min,
bb->y.max);
#endif
/* tree depth and discretisation size */
int
depth = gfs_domain_depth(gdom),
nw = (int)(POW2(depth)*bbox_width(*bb)),
nh = (int)(POW2(depth)*bbox_height(*bb));
/* shave bbox for node-aligned rather than pixel */
double shave = bbox_width(*bb)/(2.0*nw);
bb->x.min += shave;
bb->x.max -= shave;
bb->y.min += shave;
bb->y.max -= shave;
/* create & intialise meshes */
bilinear_t* B[2];
int i;
for (i=0 ; i<2 ; i++)
{
if ((B[i] = bilinear_new()) == NULL) return NULL;
if (bilinear_dimension(nw,nh,*bb,B[i]) != ERROR_OK) return NULL;
}
ftts_t ftts;
ftts.B = B;
ftts.depth = depth;
if ((ftts.u = gfs_variable_from_name(gdom->variables,"U")) == NULL)
{
fprintf(stderr,"no variable U\n");
return NULL;
}
if ((ftts.v = gfs_variable_from_name(gdom->variables,"V")) == NULL)
{
fprintf(stderr,"no variable V\n");
return NULL;
}
gfs_domain_cell_traverse(gdom,
FTT_PRE_ORDER,
FTT_TRAVERSE_LEAFS,
-1,
(FttCellTraverseFunc)ftt_sample,
&ftts);
/* clean up */
free(bb);
gts_object_destroy(GTS_OBJECT(sim));
/* results */
field_t* F = malloc(sizeof(field_t));
if (!F) return NULL;
F->u = B[0];
F->v = B[1];
F->k = NULL;
return F;
}
static void wave_run (GfsSimulation * sim)
{
GfsDomain * domain = GFS_DOMAIN (sim);
GfsWave * wave = GFS_WAVE (sim);
SolidFluxParams par;
par.div = gfs_variable_from_name (domain->variables, "P");
g_assert (par.div);
par.p = &sim->advection_params;
par.fv = gfs_temporary_variable (domain);
gfs_simulation_refine (sim);
gfs_simulation_init (sim);
while (sim->time.t < sim->time.end &&
sim->time.i < sim->time.iend) {
gdouble tstart = gfs_clock_elapsed (domain->timer);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_do, sim);
/* get global timestep */
gfs_domain_face_traverse (domain, FTT_XYZ,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttFaceTraverseFunc) gfs_face_reset_normal_velocity, NULL);
gfs_simulation_set_timestep (sim);
gdouble dt = sim->advection_params.dt;
gdouble g = sim->physical_params.g/sim->physical_params.L;
gdouble tnext = sim->tnext;
/* spatial advection */
guint ik, ith;
for (ik = 0; ik < wave->nk; ik++) {
FttVector cg;
group_velocity (ik, 0, &cg, wave->ntheta, g);
gfs_domain_face_traverse (domain, FTT_XYZ,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttFaceTraverseFunc) set_group_velocity, &cg);
if (wave->alpha_s > 0.) {
/* stability criterion for GSE diffusion */
gdouble cfl = sim->advection_params.cfl;
sim->advection_params.cfl = MIN (cfl, 2./(4.*wave->alpha_s*M_PI/wave->ntheta));
/* fixme: this should be:
sim->advection_params.cfl = MIN (cfl, sqrt(3.)/(wave->alpha_s*2.*M_PI/wave->ntheta));
*/
gfs_simulation_set_timestep (sim);
sim->advection_params.cfl = cfl;
}
else
gfs_simulation_set_timestep (sim);
/* subcycling */
guint n = rint (dt/sim->advection_params.dt);
g_assert (fabs (sim->time.t + sim->advection_params.dt*n - tnext) < 1e-12);
while (n--) {
for (ith = 0; ith < wave->ntheta; ith++) {
FttVector cg;
group_velocity (ik, ith, &cg, wave->ntheta, g);
gfs_domain_face_traverse (domain, FTT_XYZ,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttFaceTraverseFunc) set_group_velocity, &cg);
GfsVariable * t = GFS_WAVE (sim)->F[ik][ith];
sim->advection_params.v = t;
gfs_domain_traverse_leaves (domain, (FttCellTraverseFunc) solid_flux, &par);
gfs_tracer_advection_diffusion (domain, &sim->advection_params, NULL);
sim->advection_params.fv = par.fv;
gfs_domain_traverse_merged (domain, (GfsMergedTraverseFunc) gfs_advection_update,
&sim->advection_params);
if (wave->alpha_s > 0.)
gse_alleviation_diffusion (domain, t, &cg, sim->advection_params.dt);
gfs_domain_bc (domain, FTT_TRAVERSE_LEAFS, -1, t);
gfs_domain_cell_traverse (domain,
FTT_POST_ORDER, FTT_TRAVERSE_NON_LEAFS, -1,
(FttCellTraverseFunc) t->fine_coarse, t);
}
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) redo_some_events, sim);
gfs_simulation_adapt (sim);
}
}
sim->advection_params.dt = dt;
/* source terms */
if (wave->source)
(* wave->source) (wave);
sim->time.t = sim->tnext = tnext;
sim->time.i++;
gts_range_add_value (&domain->timestep, gfs_clock_elapsed (domain->timer) - tstart);
gts_range_update (&domain->timestep);
gts_range_add_value (&domain->size, gfs_domain_size (domain, FTT_TRAVERSE_LEAFS, -1));
gts_range_update (&domain->size);
}
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_do, sim);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gts_object_destroy, NULL);
gts_object_destroy (GTS_OBJECT (par.fv));
}
static void gfs_skew_symmetric_run (GfsSimulation * sim)
{
GfsVariable * p, * res = NULL, * gmac[FTT_DIMENSION];
GfsDomain * domain;
GSList * i;
domain = GFS_DOMAIN (sim);
p = gfs_variable_from_name (domain->variables, "P");
g_assert (p);
FttComponent c;
for (c = 0; c < FTT_DIMENSION; c++)
gmac[c] = gfs_temporary_variable (domain);
gfs_variable_set_vector (gmac, FTT_DIMENSION);
gfs_simulation_refine (sim);
gfs_simulation_init (sim);
i = domain->variables;
while (i) {
if (GFS_IS_VARIABLE_RESIDUAL (i->data))
res = i->data;
i = i->next;
}
gfs_simulation_set_timestep (sim);
GfsVariable ** u = gfs_domain_velocity (domain);
GfsVariable ** velfaces = GFS_SKEW_SYMMETRIC(sim)->velfaces;
GfsVariable ** velold = GFS_SKEW_SYMMETRIC(sim)->velold;
FaceData fd = { velfaces, velold, u, p, &sim->advection_params.dt, GFS_SKEW_SYMMETRIC(sim)->beta};
if (sim->time.i == 0) {
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) reset_unold, &fd);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) get_face_values, &fd);
gfs_mac_projection (domain,
&sim->projection_params,
sim->advection_params.dt/2.,
p, sim->physical_params.alpha, gmac, NULL);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) get_velfaces, &fd);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) initialize_unold, &fd);
}
while (sim->time.t < sim->time.end && sim->time.i < sim->time.iend) {
gdouble tstart = gfs_clock_elapsed (domain->timer);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_do, sim);
gfs_skew_symmetric_momentum (sim, &fd, gmac);
gfs_mac_projection (domain,
&sim->projection_params,
sim->advection_params.dt/2.,
p, sim->physical_params.alpha, gmac, NULL);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) correct_face_velocity, NULL);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_half_do, sim);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) get_velfaces, &fd);
gfs_domain_cell_traverse (domain,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) get_cell_values, &fd);
gfs_domain_cell_traverse (domain,
FTT_POST_ORDER, FTT_TRAVERSE_NON_LEAFS, -1,
(FttCellTraverseFunc) gfs_cell_coarse_init, domain);
gfs_simulation_adapt (sim);
sim->time.t = sim->tnext;
sim->time.i++;
gfs_simulation_set_timestep (sim);
gfs_advance_tracers (sim, sim->advection_params.dt);
gts_range_add_value (&domain->timestep, gfs_clock_elapsed (domain->timer) - tstart);
gts_range_update (&domain->timestep);
gts_range_add_value (&domain->size, gfs_domain_size (domain, FTT_TRAVERSE_LEAFS, -1));
gts_range_update (&domain->size);
}
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_do, sim);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gts_object_destroy, NULL);
for (c = 0; c < FTT_DIMENSION; c++)
gts_object_destroy (GTS_OBJECT (gmac[c]));
}
static void gfs_porous_run (GfsSimulation * sim)
{
GfsVariable * p, * pmac, * res = NULL, * g[FTT_DIMENSION], * gmac[FTT_DIMENSION];
GfsVariable ** gc = sim->advection_params.gc ? g : NULL;
GfsDomain * domain;
GfsPorous *por;
GSList * i;
domain = GFS_DOMAIN (sim);
por = GFS_POROUS (sim);
p = gfs_variable_from_name (domain->variables, "P");
g_assert (p);
pmac = gfs_variable_from_name (domain->variables, "Pmac");
g_assert (pmac);
FttComponent c;
for (c = 0; c < FTT_DIMENSION; c++) {
gmac[c] = gfs_temporary_variable (domain);
if (sim->advection_params.gc)
g[c] = gfs_temporary_variable (domain);
else
g[c] = gmac[c];
}
gfs_variable_set_vector (gmac, FTT_DIMENSION);
gfs_variable_set_vector (g, FTT_DIMENSION);
gfs_simulation_refine (sim);
gfs_simulation_init (sim);
i = domain->variables;
while (i) {
if (GFS_IS_VARIABLE_RESIDUAL (i->data))
res = i->data;
i = i->next;
}
gfs_simulation_set_timestep (sim);
if (sim->time.i == 0) {
/*inserted changes inside this function*/
gfs_approximate_projection_por (domain, por,
&sim->approx_projection_params,
sim->advection_params.dt,
p, sim->physical_params.alpha, res, g, NULL);
gfs_simulation_set_timestep (sim);
gfs_advance_tracers (sim, sim->advection_params.dt/2.);
}
else if (sim->advection_params.gc)
gfs_update_gradients_por (domain, por, p, sim->physical_params.alpha, g);
while (sim->time.t < sim->time.end &&
sim->time.i < sim->time.iend) {
gdouble tstart = gfs_clock_elapsed (domain->timer);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_do, sim);
/*inserted changes */
gfs_pre_projection (domain, por, FTT_DIMENSION);
if (sim->advection_params.linear) {
/* linearised advection */
gfs_domain_face_traverse (domain, FTT_XYZ,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttFaceTraverseFunc) gfs_face_reset_normal_velocity, NULL);
gfs_domain_face_traverse (domain, FTT_XYZ,
FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttFaceTraverseFunc) gfs_face_interpolated_normal_velocity,
sim->u0);
}
else
gfs_predicted_face_velocities (domain, FTT_DIMENSION, &sim->advection_params);
gfs_variables_swap (p, pmac);
gfs_mac_projection_por (domain, por,
&sim->projection_params,
sim->advection_params.dt/2.,
p, sim->physical_params.alpha, gmac, NULL);
gfs_variables_swap (p, pmac);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_half_do, sim);
gfs_centered_velocity_advection_diffusion (domain,
FTT_DIMENSION,
&sim->advection_params,
gmac,
sim->time.i > 0 || !gc ? gc : gmac,
sim->physical_params.alpha);
if (gc) {
gfs_source_darcy_implicit (domain, sim->advection_params.dt);
gfs_correct_centered_velocities (domain, FTT_DIMENSION, sim->time.i > 0 ? gc : gmac,
-sim->advection_params.dt);
/*inserted changes*/
gfs_post_projection (domain, por, FTT_DIMENSION);
}
else if (gfs_has_source_coriolis (domain)) {
gfs_correct_centered_velocities (domain, FTT_DIMENSION, gmac, sim->advection_params.dt);
gfs_source_darcy_implicit (domain, sim->advection_params.dt);
gfs_correct_centered_velocities (domain, FTT_DIMENSION, gmac, -sim->advection_params.dt);
/*inserted changes*/
gfs_post_projection (domain, por, FTT_DIMENSION);
}
gfs_domain_cell_traverse (domain,
FTT_POST_ORDER, FTT_TRAVERSE_NON_LEAFS, -1,
(FttCellTraverseFunc) gfs_cell_coarse_init, domain);
gfs_simulation_adapt (sim);
/*inserted changes */
gfs_approximate_projection_por (domain, por,
&sim->approx_projection_params,
sim->advection_params.dt,
p, sim->physical_params.alpha, res, g, NULL);
/*inserted changes */
sim->time.t = sim->tnext;
sim->time.i++;
gfs_simulation_set_timestep (sim);
gfs_advance_tracers (sim, sim->advection_params.dt);
gts_range_add_value (&domain->timestep, gfs_clock_elapsed (domain->timer) - tstart);
gts_range_update (&domain->timestep);
gts_range_add_value (&domain->size, gfs_domain_size (domain, FTT_TRAVERSE_LEAFS, -1));
gts_range_update (&domain->size);
}
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gfs_event_do, sim);
gts_container_foreach (GTS_CONTAINER (sim->events), (GtsFunc) gts_object_destroy, NULL);
for (c = 0; c < FTT_DIMENSION; c++) {
gts_object_destroy (GTS_OBJECT (gmac[c]));
if (sim->advection_params.gc)
gts_object_destroy (GTS_OBJECT (g[c]));
}
}
static void mac_projection_por (GfsDomain * domain,
GfsPorous * por,
GfsMultilevelParams * par,
gdouble dt,
GfsVariable * p,
GfsFunction * alpha,
GfsVariable * res,
GfsVariable ** g,
void (* divergence_hook) (GfsDomain * domain,
gdouble dt,
GfsVariable * div)
)
{
/* Add face sources */
gfs_reset_gradients (domain, FTT_DIMENSION, g);
gfs_velocity_face_sources (domain, gfs_domain_velocity (domain), dt, alpha, g);
GfsVariable * dia = gfs_temporary_variable (domain);
GfsVariable * div = gfs_temporary_variable (domain);
GfsVariable * res1 = res ? res : gfs_temporary_variable (domain);
/* Initialize face coefficients */
gfs_poisson_coefficients_por (domain, por, alpha, TRUE, TRUE, TRUE);
/* hydrostatic pressure */
GSList * i = domain->variables;
while (i) {
if (GFS_IS_HYDROSTATIC_PRESSURE (i->data))
gfs_correct_normal_velocities (domain, FTT_DIMENSION, i->data, g, dt);
i = i->next;
}
/* Initialize diagonal coefficient */
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_ALL, -1,
(FttCellTraverseFunc) gfs_cell_reset, dia);
/* compute MAC divergence */
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) gfs_normal_divergence, div);
/* Divergence hook */
if (divergence_hook)
(* divergence_hook) (domain, dt, div);
/* add volume sources (if any) */
if (p->sources)
volume_sources (domain, p, div);
/* Scale divergence */
gpointer data[2];
data[0] = div;
data[1] = &dt;
gfs_domain_cell_traverse (domain, FTT_PRE_ORDER, FTT_TRAVERSE_LEAFS, -1,
(FttCellTraverseFunc) scale_divergence, data);
#if 0
{
FILE * fp = fopen ("/tmp/mac", "wt");
GfsNorm norm;
gfs_write_mac_velocity (domain, 0.9, FTT_TRAVERSE_LEAFS, -1, NULL, fp);
fclose (fp);
norm = gfs_domain_norm_variable (domain, div, FTT_TRAVERSE_LEAFS, -1);
fprintf (stderr, "mac div before: %g %g %g\n",
norm.first, norm.second, norm.infty);
}
#endif
par->poisson_solve (domain, par, p, div, res1, dia, dt);
gts_object_destroy (GTS_OBJECT (dia));
gts_object_destroy (GTS_OBJECT (div));
if (!res)
gts_object_destroy (GTS_OBJECT (res1));
gfs_correct_normal_velocities (domain, FTT_DIMENSION, p, g, dt);
gfs_scale_gradients (domain, FTT_DIMENSION, g);
}