https://github.com/geodynamics/citcoms
Tip revision: db34189a4cc8afa725438397e42cb391338a2f06 authored by Leif Strand on 27 July 2005, 09:06:27 UTC
Merged changes fron trunk: "[...] uniprocessor examples work again [...]".
Merged changes fron trunk: "[...] uniprocessor examples work again [...]".
Tip revision: db34189
Drive_solvers.c
/*
*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*
*<LicenseText>
*=====================================================================
*
* CitcomS
* ---------------------------------
*
* Authors:
* Louis Moresi, Shijie Zhong, Lijie Han, Eh Tan,
* Clint Conrad, Michael Gurnis, and Eun-seo Choi
* (c) California Institute of Technology 1994-2005
*
* By downloading and/or installing this software you have
* agreed to the CitcomS.py-LICENSE bundled with this software.
* Free for non-commercial academic research ONLY.
* This program is distributed WITHOUT ANY WARRANTY whatsoever.
*
*=====================================================================
*
* Copyright June 2005, by the California Institute of Technology.
* ALL RIGHTS RESERVED. United States Government Sponsorship Acknowledged.
*
* Any commercial use must be negotiated with the Office of Technology
* Transfer at the California Institute of Technology. This software
* may be subject to U.S. export control laws and regulations. By
* accepting this software, the user agrees to comply with all
* applicable U.S. export laws and regulations, including the
* International Traffic and Arms Regulations, 22 C.F.R. 120-130 and
* the Export Administration Regulations, 15 C.F.R. 730-744. User has
* the responsibility to obtain export licenses, or other export
* authority as may be required before exporting such information to
* foreign countries or providing access to foreign nationals. In no
* event shall the California Institute of Technology be liable to any
* party for direct, indirect, special, incidental or consequential
* damages, including lost profits, arising out of the use of this
* software and its documentation, even if the California Institute of
* Technology has been advised of the possibility of such damage.
*
* The California Institute of Technology specifically disclaims any
* warranties, including the implied warranties or merchantability and
* fitness for a particular purpose. The software and documentation
* provided hereunder is on an "as is" basis, and the California
* Institute of Technology has no obligations to provide maintenance,
* support, updates, enhancements or modifications.
*
*=====================================================================
*</LicenseText>
*
*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
*/
#include <math.h>
#include <sys/types.h>
#include "element_definitions.h"
#include "global_defs.h"
#include "drive_solvers.h"
float global_fvdot();
float vnorm_nonnewt();
//***********************************************************
void general_stokes_solver_setup(struct All_variables *E)
{
int i, m;
if (E->control.NMULTIGRID || E->control.NASSEMBLE)
construct_node_maps(E);
else
for (i=E->mesh.gridmin;i<=E->mesh.gridmax;i++)
for (m=1;m<=E->sphere.caps_per_proc;m++)
E->elt_k[i][m]=(struct EK *)malloc((E->lmesh.NEL[i]+1)*sizeof(struct EK));
return;
}
void general_stokes_solver(struct All_variables *E)
{
void solve_constrained_flow_iterative();
void construct_stiffness_B_matrix();
void velocities_conform_bcs();
void assemble_forces();
void sphere_harmonics_layer();
double global_vdot(),kineticE_radial();
float global_fvdot();
float vnorm_nonnewt();
void get_system_viscosity();
float vmag;
double Udot_mag, dUdot_mag;
int m,count,i,j,k;
float *oldU[NCS], *delta_U[NCS];
const int nno = E->lmesh.nno;
const int nel = E->lmesh.nel;
const int nnov = E->lmesh.nnov;
const int neq = E->lmesh.neq;
const int vpts = vpoints[E->mesh.nsd];
const int dims = E->mesh.nsd;
const int addi_dof = additional_dof[dims];
E->monitor.elapsed_time_vsoln = E->monitor.elapsed_time;
velocities_conform_bcs(E,E->U);
assemble_forces(E,0);
if(E->monitor.solution_cycles==0 || E->viscosity.update_allowed) {
get_system_viscosity(E,1,E->EVI[E->mesh.levmax],E->VI[E->mesh.levmax]);
construct_stiffness_B_matrix(E);
}
solve_constrained_flow_iterative(E);
if (E->viscosity.SDEPV) {
for (m=1;m<=E->sphere.caps_per_proc;m++) {
delta_U[m] = (float *)malloc((neq+2)*sizeof(float));
oldU[m] = (float *)malloc((neq+2)*sizeof(float));
for(i=0;i<=neq;i++)
oldU[m][i]=0.0;
}
Udot_mag=dUdot_mag=0.0;
count=1;
while (1) {
for (m=1;m<=E->sphere.caps_per_proc;m++)
for (i=0;i<neq;i++) {
delta_U[m][i] = E->U[m][i] - oldU[m][i];
oldU[m][i] = E->U[m][i];
}
Udot_mag = sqrt(global_fvdot(E,oldU,oldU,E->mesh.levmax));
dUdot_mag = vnorm_nonnewt(E,delta_U,oldU,E->mesh.levmax);
if(E->parallel.me==0){
fprintf(stderr,"Stress dependent viscosity: DUdot = %.4e (%.4e) for iteration %d\n",dUdot_mag,Udot_mag,count);
fprintf(E->fp,"Stress dependent viscosity: DUdot = %.4e (%.4e) for iteration %d\n",dUdot_mag,Udot_mag,count);
fflush(E->fp);
}
if (count>50 || dUdot_mag>E->viscosity.sdepv_misfit)
break;
get_system_viscosity(E,1,E->EVI[E->mesh.levmax],E->VI[E->mesh.levmax]);
construct_stiffness_B_matrix(E);
solve_constrained_flow_iterative(E);
count++;
} //end while
for (m=1;m<=E->sphere.caps_per_proc;m++) {
free((void *) oldU[m]);
free((void *) delta_U[m]);
}
} //end if SDEPV
return;
}
void general_stokes_solver_pseudo_surf(struct All_variables *E)
{
void solve_constrained_flow_iterative_pseudo_surf();
void construct_stiffness_B_matrix();
void velocities_conform_bcs();
void assemble_forces_pseudo_surf();
float global_fvdot();
float vnorm_nonnewt();
void get_system_viscosity();
void std_timestep();
void get_STD_freesurf(struct All_variables *, float**);
float vmag;
double Udot_mag, dUdot_mag;
int m,count,i,j,k,topo_loop;
float *oldU[NCS], *delta_U[NCS];
const int nno = E->lmesh.nno;
const int nel = E->lmesh.nel;
const int nnov = E->lmesh.nnov;
const int neq = E->lmesh.neq;
const int vpts = vpoints[E->mesh.nsd];
const int dims = E->mesh.nsd;
const int addi_dof = additional_dof[dims];
E->monitor.elapsed_time_vsoln = E->monitor.elapsed_time;
velocities_conform_bcs(E,E->U);
E->monitor.stop_topo_loop = 0;
E->monitor.topo_loop = 0;
if(E->monitor.solution_cycles==0) std_timestep(E);
while(E->monitor.stop_topo_loop == 0) {
assemble_forces_pseudo_surf(E,0);
if(E->monitor.solution_cycles==0 || E->viscosity.update_allowed) {
get_system_viscosity(E,1,E->EVI[E->mesh.levmax],E->VI[E->mesh.levmax]);
construct_stiffness_B_matrix(E);
}
solve_constrained_flow_iterative_pseudo_surf(E);
if (E->viscosity.SDEPV) {
for (m=1;m<=E->sphere.caps_per_proc;m++) {
delta_U[m] = (float *)malloc((neq+2)*sizeof(float));
oldU[m] = (float *)malloc((neq+2)*sizeof(float));
for(i=0;i<=neq;i++)
oldU[m][i]=0.0;
}
Udot_mag=dUdot_mag=0.0;
count=1;
while (1) {
for (m=1;m<=E->sphere.caps_per_proc;m++)
for (i=0;i<neq;i++) {
delta_U[m][i] = E->U[m][i] - oldU[m][i];
oldU[m][i] = E->U[m][i];
}
Udot_mag = sqrt(global_fvdot(E,oldU,oldU,E->mesh.levmax));
dUdot_mag = vnorm_nonnewt(E,delta_U,oldU,E->mesh.levmax);
if(E->parallel.me==0){
fprintf(stderr,"Stress dependent viscosity: DUdot = %.4e (%.4e) for iteration %d\n",dUdot_mag,Udot_mag,count);
fprintf(E->fp,"Stress dependent viscosity: DUdot = %.4e (%.4e) for iteration %d\n",dUdot_mag,Udot_mag,count);
fflush(E->fp);
}
if (count>50 || dUdot_mag>E->viscosity.sdepv_misfit)
break;
get_system_viscosity(E,1,E->EVI[E->mesh.levmax],E->VI[E->mesh.levmax]);
construct_stiffness_B_matrix(E);
solve_constrained_flow_iterative_pseudo_surf(E);
count++;
} //end while
for (m=1;m<=E->sphere.caps_per_proc;m++) {
free((void *) oldU[m]);
free((void *) delta_U[m]);
}
} //end if SDEPV
E->monitor.topo_loop++;
}
get_STD_freesurf(E,E->slice.freesurf);
return;
}