/* theta, phi version */
void thetaphipv2cv(HC_PREC theta, float phi,
HC_PREC *polar_vec,HC_PREC *cart_vec)
{
HC_HIGH_PREC polar_base[9];
calc_polar_base_at_theta_phi(theta,phi,polar_base);
lonlatpv2cv_with_base(polar_vec,polar_base,cart_vec);
}
int main(int argc, char **argv)
{
int ilayer,nvsol,ndsol=0,mode,shps,loop,i1,i2,nlat,nlon,
ivec,lc,ndata,ndata_all,ndata_d,npoints,i,j,
poff,shps_read=0,shps_read_d=0;
FILE *in;
struct sh_lms *vsol=NULL,*dsol=NULL;
struct hcs *model;
HC_PREC zlabel;
hc_boolean binary_in = TRUE, verbose = FALSE,read_dsol=FALSE;
HC_PREC *data,*plm=NULL,*xpos,*xvec,lon,lat,theta,phi,xtmp[3],pvec[3],
*xscalar;
HC_PREC polar_base[9];
hc_struc_init(&model);
/*
deal with parameters
*/
ilayer = 0;
mode = 1;
switch(argc){
case 3:
sscanf(argv[2],"%i",&ilayer);
break;
case 4:
sscanf(argv[2],"%i",&ilayer);
sscanf(argv[3],"%i",&mode);
break;
case 5:
sscanf(argv[2],"%i",&ilayer);
sscanf(argv[3],"%i",&mode);
read_dsol = TRUE;
break;
default:
fprintf(stderr,"%s: usage\n%s sol.file layer [mode,%i] [scalar.sol]\n\n",
argv[0],argv[0],mode);
fprintf(stderr,"extracts spatial solution (velocity or stress, v) from output file sol.file\n");
fprintf(stderr," if scalar.sol argument is given, will also read in a scalar for VTK output\n");
fprintf(stderr,"layer: 1...nset\n");
fprintf(stderr,"\tif ilayer= 1..nset, will print one layer\n");
fprintf(stderr,"\t -1, will select nset (the top layer)\n");
fprintf(stderr,"\t -2, will print all layers\n");
fprintf(stderr,"mode: 1...4\n");
fprintf(stderr,"\tif mode = 1, will print lon lat z v_r \n");
fprintf(stderr,"\t 2, will print lon lat z v_theta v_phi \n");
fprintf(stderr,"\t 3, will print lon lat z v_r v_theta v_phi\n");
fprintf(stderr,"\t 4, will print the depth levels of all layers\n");
fprintf(stderr,"\t 5, compute all depth levels (set ilayer=-2) and write VTK file, ASCII\n");
fprintf(stderr,"\t 6, compute all depth levels (set ilayer=-2) and write VTK file, BINARY\n");
exit(-1);
break;
}
if((mode == 4)||(mode==5)||(mode==6))
ilayer = -2;
/*
read in velocity/traction solution
*/
in = hc_fopen(argv[1],"r","hc_extract_spatial");
shps_read = hc_read_sh_solution(model,&vsol,in,binary_in,verbose);
fclose(in);
nvsol = model->nradp2 * shps_read;
/*
deal with selection
*/
loop = 0;
if(ilayer == -1)
ilayer = model->nradp2;
else if(ilayer == -2){
ilayer = model->nradp2;
loop =1;
}
if((ilayer < 1)||(ilayer > model->nradp2)){
fprintf(stderr,"%s: ilayer (%i) out of range, use 1 ... %i\n",
argv[0],ilayer,model->nradp2);
exit(-1);
}
/* set up layer bounds */
if(loop){
i1=0;i2=model->nradp2-1;
}else{
i1=ilayer-1;i2 = i1;
}
/* detect number of expansions */
if(mode == 1){
shps = 1; /* r */
}else if(mode == 2){
shps = 2; /* theta,phi */
}else if((mode == 3)||(mode == 5)||(mode==6)){
shps = 3; /* r,theta,phi */
}else{
shps = 1;
}
if(shps > shps_read){
fprintf(stderr,"%s: solution file only had %i expansions, mode %i requests %i\n",
argv[0],shps_read,mode,shps);
exit(-1);
}
/*
density solution or other scalar
*/
if(read_dsol){
if((mode != 5)&&(mode != 6))
HC_ERROR("hc_extract_spatial","error, only mode 5 and can handle scalar input");
in = hc_fopen(argv[4],"r","hc_extract_spatial");
shps_read_d = hc_read_sh_solution(model,&dsol,in,binary_in,
verbose);
fclose(in);
ndsol = model->nradp2 * shps_read_d;
}
/*
room for spatial expansion
*/
npoints = (vsol+i1*shps_read)->npoints;
if((vsol+i1*shps_read)->type != SH_RICK)
HC_ERROR("sh_extract_spatial","SH_RICK type required");
/* geographic set up */
nlat = (vsol+i1*shps_read)->rick.nlat;
nlon = (vsol+i1*shps_read)->rick.nlon;
ndata = npoints * shps ;
ndata_d = npoints * shps_read_d;
ndata_all = npoints * (shps + shps_read_d);
if((mode == 5)||(mode==6)){ /* save all layers */
hc_vecalloc(&data,model->nradp2 * ndata_all,"hc_extract_spatial");
}else
hc_vecalloc(&data, ndata_all,"hc_extract_spatial");
for(lc=0,ilayer=i1;ilayer <= i2;ilayer++,lc++){
/*
output
*/
zlabel = HC_Z_DEPTH(model->r[ilayer]);
switch(mode){
case 1:
/* */
if(verbose)
fprintf(stderr,"%s: printing v_r at layer %i (depth: %g)\n",argv[0],ilayer,
(double)zlabel);
ivec=FALSE;sh_compute_spatial((vsol+ilayer*shps_read),ivec,TRUE,&plm,data,verbose);
sh_print_spatial_data_to_stream((vsol+ilayer*shps_read),shps,data,TRUE,zlabel,stdout);
break;
case 2:
/* */
if(verbose)
fprintf(stderr,"%s: printing v_theta v_phi SHE at layer %i (depth: %g)\n",argv[0],ilayer,(double)zlabel);
ivec=TRUE;sh_compute_spatial((vsol+ilayer*shps_read+1),ivec,TRUE,&plm,data,verbose);
sh_print_spatial_data_to_stream((vsol+ilayer*shps_read+1),shps,data,TRUE,zlabel,stdout);
break;
case 3:
if(verbose)
fprintf(stderr,"%s: printing v_r v_theta v_phi SHE at layer %i (depth: %g)\n",argv[0],ilayer,(double)zlabel);
ivec=FALSE;sh_compute_spatial((vsol+ilayer*shps_read), ivec,TRUE,&plm,data,verbose); /* radial */
ivec=TRUE; sh_compute_spatial((vsol+ilayer*shps_read+1),ivec,TRUE,&plm,(data+npoints),verbose); /* theta,phi */
sh_print_spatial_data_to_stream((vsol+ilayer*shps_read),shps,data,TRUE,zlabel,stdout);
break;
case 4:
fprintf(stdout,"%5i %11g\n",ilayer,(double)HC_Z_DEPTH(model->r[ilayer]));
break;
case 5: /* compute all and store */
case 6:
ivec=FALSE;sh_compute_spatial((vsol+ilayer*shps_read), ivec,TRUE,&plm,(data+lc*ndata_all),verbose); /* radial */
ivec=TRUE; sh_compute_spatial((vsol+ilayer*shps_read+1),ivec,TRUE,&plm,(data+lc*ndata_all+npoints),verbose); /* theta,phi */
if(read_dsol){
if(!shps_read_d)
HC_ERROR("sh_extract_spatial","logic error");
ivec=FALSE;sh_compute_spatial((dsol+ilayer*shps_read_d),ivec,TRUE,&plm,(data+lc*ndata_all+npoints*shps),verbose); /* radial */
}
break;
default:
fprintf(stderr,"%s: error, mode %i undefined\n",argv[0],mode);
exit(-1);
break;
}
}
/* clear and exit */
sh_free_expansion(vsol,nvsol);
if(read_dsol)
sh_free_expansion(dsol,ndsol);
free(plm);
/* */
if((mode == 5)||(mode==6)){
/*
print the already stored properties
*/
if(shps != 3)HC_ERROR("hc_extract_spatial","shps has to be 3 for mode 5 and 6");
/* convert */
hc_vecalloc(&xpos,model->nradp2 * ndata,"hc_extract_spatial");
hc_vecalloc(&xvec,model->nradp2 * ndata,"hc_extract_spatial");
if(read_dsol)
hc_vecalloc(&xscalar,model->nradp2 * ndata_d,"hc_extract_spatial");
for(i=0;i < npoints;i++){ /* loop through all points */
/* lon lat coordinates */
sh_get_coordinates((vsol+i1*3),i,&lon,&lat);
theta = LAT2THETA(lat);phi = LON2PHI(lon);
xtmp[0] = xtmp[1] = sin(theta);
xtmp[0] *= cos(phi); /* x */
xtmp[1] *= sin(phi); /* y */
xtmp[2] = cos(theta); /* z */
/* for conversion */
calc_polar_base_at_theta_phi(theta,phi,polar_base);
for(ilayer=0;ilayer < model->nradp2;ilayer++){
/* this is the slow data storage loop but it avoids
recomputing the polar basis vector */
poff = ilayer * ndata + i*shps; /* point offset */
for(j=0;j < 3;j++){
xpos[poff+j] = xtmp[j] * model->r[ilayer]; /* cartesian coordinates */
}
/* data are stored a bit weirdly, this makes for lots of
jumping around in memory ... */
pvec[0] = data[ilayer*ndata_all + i];
pvec[1] = data[ilayer*ndata_all + npoints +i];
pvec[2] = data[ilayer*ndata_all + npoints*2+i];
lonlatpv2cv_with_base(pvec,polar_base,(xvec+poff));
/* assign scalar fata if any */
for(j=0;j < shps_read_d;j++)
xscalar[j * model->nradp2 * ndata_d + ilayer * npoints + i] =
data[ilayer * ndata_all + npoints*(shps+j) + i];
}
}
free(data);
/* print in VTK format */
hc_print_vtk(stdout,xpos,xvec,npoints,model->nradp2,(mode==6),
shps_read_d,xscalar,nlon,nlat);
free(xvec);free(xpos);
if(shps_read_d)
free(xscalar);
}else{
free(data);
}
return 0;
}
