PPdata *pp_taxis_to_values(const PPtaxis *taxis, PPlist *heaplist)
{
PPdata *data;
PPlist *list;
PPtime *t;
int mean;
int i;
int n;
Freal val,val2;
PPlisthandle handle;
list=taxis->values;
n=pp_list_size(list);
CKP( data=pp_data_new(realtype,n,heaplist) );
mean = pp_is_time_mean(taxis->type);
pp_list_startwalk(list,&handle);
for (i=0; i < n; i++) {
t=pp_list_walk(&handle,0);
val=pp_time_diff(t->type,
&t->time1,
&taxis->time_orig);
if (mean) {
/* FIXME(?): with difference field (time 2 - time 1), maybe want to do likewise and
* use average time also?
*/
/* FIXME(?): in the event of a seasonal composite,
* (I think this is LBTIM==3 but doc not to hand at time of writing this),
* time1 is meaning start date and first year, time2 is meaning start date and end year;
* in event of even number of years, the date could end up 6 months out from the time
* of year (e.g. Jan mean over ten years 2000-2009 becomes average of 1 Jan 2000 and
* 1 Feb 2009, which is mid-July 2004 -- maybe more helpful to report e.g. mid-Jan 2004?)
*/
val2=pp_time_diff(t->type,
&t->time2,
&taxis->time_orig);
val = (val + val2) / 2;
}
((Freal*)(data->values))[i]=val;
}
return data;
ERRBLKP("pp_taxis_to_values");
}
PPdata *pp_regaxis_to_values(const PPregaxis *a, PPlist *heaplist)
{
PPdata *data;
int i;
CKP( data=pp_data_new(realtype,a->n,heaplist) );
for (i=0; i < a->n; i++) {
((Freal*)(data->values))[i]= a->start+i*a->interval;
}
return data;
ERRBLKP("pp_regaxis_to_values");
}
PPdata *pp_read_extradata(const PPrec *rec, const PPfile *ppfile, PPlist *heaplist, const PPextravec extra) {
const PPhdr *hdrp;
FILE *fh;
int pack, nread;
size_t pos, epos;
Fint ic, ia, ib;
PPdata *data;
CKP(ppfile);
CKP(rec);
CKP(heaplist);
data=NULL;
hdrp=&(rec->hdr);
fh = ppfile->fh;
pack = pp_get_var_packing(hdrp);
if (pack==0) {
pos=rec->datapos+(hdrp->LBROW*hdrp->LBNPT)*ppfile->wordsize;
epos=pos+hdrp->LBEXT*ppfile->wordsize;
fseek(fh, pos, SEEK_SET);
ia=1;
while (pos < epos && ia > 0) {
fseek(fh, pos, SEEK_SET);
nread=pp_read_words(&ic, 1, convert_int, ppfile);
ERRIF(nread != 1);
ia=ic/1000;
ib=ic-ia*1000;
if (ib == extra) {
CKP(data=pp_data_new(realtype,ia,heaplist));
if ((nread=pp_read_words(data->values,ia,convert_real, ppfile)) !=ia ) ERR;
break;
}
pos+=ia;
}
} else {
pp_error_mesg("pp_read_extra_dat","only do unpacked exta data");
data=NULL;
}
return data;
ERRBLKP("pp_read_extra_data");
}
int pp_add_rotmap_pole_vars(PPlist *rotmaps, CuVar *cuvars, int *nvarsp, PPlist *heaplist)
{
int rotmapid;
PPlisthandle handle;
CuVar *var;
PPvar *ppvar;
PProtmap *rotmap;
PPlist *catts;
/* add any rotated_pole variables */
rotmapid = 0;
pp_list_startwalk(rotmaps, &handle);
while ((rotmap = pp_list_walk(&handle, 0)) != NULL) {
var = &cuvars[*nvarsp];
ppvar = (PPvar*) var->internp;
sprintf(var->name, "rotated_pole%d", rotmapid);
strncpy(rotmap->name, var->name, CU_MAX_NAME);
rotmap->name[CU_MAX_NAME] = '\0';
/* single value of arbitrary type; set as integer = 0 */
var->datatype = inttype;
var->ndims = 0;
CKP( ppvar->data = pp_data_new(inttype, 1, heaplist) );
((Freal*)(ppvar->data->values))[0] = 0;
/* and add attributes */
catts = ppvar->atts;
CKI( pp_add_string_att(catts, "grid_mapping_name", "rotated_latitude_longitude", heaplist) );
CKI( pp_add_att(catts, "grid_north_pole_longitude", realtype, 1, &rotmap->pole_lon, heaplist) );
CKI( pp_add_att(catts, "grid_north_pole_latitude", realtype, 1, &rotmap->pole_lat, heaplist) );
CKI( pp_add_att(catts, "north_pole_grid_longitude", realtype, 1, &rotmap->truepole_gridlon, heaplist) );
rotmapid++;
(*nvarsp)++;
}
return 0;
ERRBLKI("pp_add_rotmap_pole_vars");
}
int pp_set_landmask(int nrec, PPrec **recs, PPfile *ppfile, PPlist *rotmaps, PPlist *heaplist)
{
/*
* find the land mask variable (if present) and if so, then store it in ppfile->landmask
* (also adding any associated rotated grid mappings to rotmaps)
*/
int rec;
PPrec *recp;
PPhdr *hdrp;
PPgenaxis *xaxis, *yaxis;
PPlandmask *landmask;
for (rec = 0; rec < nrec ; rec++) {
recp = recs[rec];
hdrp = &recp->hdr;
if (pp_var_is_land_mask(hdrp)) {
CKP( landmask = pp_malloc(sizeof(PPlandmask), heaplist) );
CKI( pp_set_horizontal_axes(recp, ppfile, &xaxis, &yaxis, rotmaps, heaplist) );
CKP( landmask->data = pp_data_new(inttype, pp_genaxis_len(xaxis) * pp_genaxis_len(yaxis), heaplist) ); /* JAK 2005-01-05 */
/* read in land mask data values */
landmask->xaxis = xaxis;
landmask->yaxis = yaxis;
CKP( landmask->data->values = pp_read_data_record(recp, ppfile, heaplist) );
ppfile->landmask = landmask;
}
}
return 0;
ERRBLKI("pp_set_landmask");
}
PPdata *pp_taxis_to_boundary_values(const PPtaxis *taxis, PPlist *heaplist)
{
PPdata *data;
PPlist *list;
PPtime *t;
int i;
int n;
Freal val,val2;
PPlisthandle handle;
/* only valid for time mean */
ERRIF(!pp_is_time_mean(taxis->type));
list=taxis->values;
n=pp_list_size(list);
CKP( data=pp_data_new(realtype,n*2,heaplist) );
pp_list_startwalk(list,&handle);
for (i=0; i < n; i++) {
t=pp_list_walk(&handle,0);
val=pp_time_diff(t->type,
&t->time1,
&taxis->time_orig);
val2=pp_time_diff(t->type,
&t->time2,
&taxis->time_orig);
((Freal*)(data->values))[ 2*i ]=val;
((Freal*)(data->values))[ 2*i+1 ]=val2;
}
return data;
ERRBLKP("pp_taxis_to_boundary_values");
}
int pp_add_p0_var(int have_hybrid_sigmap, CuVar *cuvars, int *nvarsp, PPlist *heaplist)
{
CuVar *var;
PPvar *ppvar;
/* add p0 variable if we had hybrid_sigmap coords */
if (have_hybrid_sigmap) {
var = &cuvars[*nvarsp];
ppvar = (PPvar*) var->internp;
sprintf(var->name, "p0");
var->ndims = 0;
CKI( pp_add_string_att(ppvar->atts, "long_name",
"reference pressure value for atmospheric hybrid sigma-pressure coordinates",
heaplist) );
/* single value consisting of p0 */
CKP( ppvar->data = pp_data_new(realtype, 1, heaplist) );
((Freal*)(ppvar->data->values))[0] = reference_pressure;
(*nvarsp)++;
}
return 0;
ERRBLKI("pp_add_p0_var");
}
int pp_calc_rot_grid(PProtgrid *rotgrid, PPdata **lons_return, PPdata **lats_return, PPlist *heaplist)
{
int nx, ny, i, j;
int offset, offset1;
PPgenaxis *xaxis;
PPgenaxis *yaxis;
PPdata *londata, *latdata, *rlondata, *rlatdata;
Freal *lons, *lats, *rlons, *rlats; /* "r" stands for rotated */
double latpole_rad, coslatpole, sinlatpole, cosrlat, sinrlat;
double *cosdrlon, *sindrlon;
double rlonN, lonpole, drlon_rad, dlon_rad, rlat_rad, lon;
double cycdx, sinlat;
const double dtor = M_PI / 180.;
CKP(rotgrid);
xaxis = rotgrid->xaxis;
yaxis = rotgrid->yaxis;
nx = pp_genaxis_len(xaxis);
ny = pp_genaxis_len(yaxis);
/* get input, output and workspace arrays */
CKP( rlondata = pp_genaxis_to_values(xaxis,0,heaplist) );
ERRIF(rlondata->type != realtype);
CKP( rlatdata = pp_genaxis_to_values(yaxis,0,heaplist) );
ERRIF(rlatdata->type != realtype);
CKP( londata = pp_data_new(realtype,nx*ny,heaplist) );
CKP( latdata = pp_data_new(realtype,nx*ny,heaplist) );
CKP( cosdrlon = pp_malloc(nx*sizeof(double),heaplist) );
CKP( sindrlon = pp_malloc(nx*sizeof(double),heaplist) );
/* some pointers for convenience (and speed?) */
rlons = (Freal*) rlondata->values;
rlats = (Freal*) rlatdata->values;
lons = londata->values;
lats = latdata->values;
latpole_rad = rotgrid->rotmap->pole_lat * dtor;
coslatpole = cos(latpole_rad);
sinlatpole = sin(latpole_rad);
rlonN = rotgrid->rotmap->truepole_gridlon;
lonpole = rotgrid->rotmap->pole_lon;
for (i=0; i<nx; i++) {
drlon_rad = (rlons[i] - rlonN) * dtor;
cosdrlon[i] = cos(drlon_rad);
sindrlon[i] = sin(drlon_rad);
}
for (j=0; j<ny; j++) {
rlat_rad = rlats[j] * dtor;
cosrlat = cos(rlat_rad);
sinrlat = sin(rlat_rad);
offset1 = j*nx;
for (i=0; i<nx; i++) {
offset = offset1 + i;
cycdx = cosrlat * cosdrlon[i];
dlon_rad = atan2( -cosrlat*sindrlon[i], sinrlat*coslatpole - cycdx*sinlatpole );
lon = (dlon_rad/dtor + lonpole);
/* put in range 0 <= lon < 360
* NOTE: This code previously put in range -180 to 180.
* The actual code was the following:
* lon -= lon_modulo * floor(lon / lon_modulo + 0.5);
* This was changed because the subsetting functions in CDAT
* didn't like the negative longitudes.
*/
lon -= lon_modulo * floor(lon / lon_modulo);
sinlat = cycdx * coslatpole + sinrlat * sinlatpole;
if (sinlat > 1.)
sinlat = 1.;
else if (sinlat < -1.)
sinlat = -1.;
lons[offset] = lon;
lats[offset] = asin(sinlat) / dtor;
}
}
/* free workspace arrays */
CKI( pp_free(rlondata, heaplist) );
CKI( pp_free(rlatdata, heaplist) );
CKI( pp_free(cosdrlon, heaplist) );
CKI( pp_free(sindrlon, heaplist) );
/* return pointers */
if (lons_return != NULL)
*lons_return = londata;
if (lats_return != NULL)
*lats_return = latdata;
return 0;
ERRBLKI("pp_calc_rot_grid");
}
PPdata *pp_zaxis_to_values(const PPzaxis *zaxis, PPlevvaltype vtype, PPlist *heaplist)
{
PPdata *data;
int i;
int n;
Freal val;
PPlevel *lev;
PPlist *list;
PPlisthandle handle;
list=zaxis->values;
n=pp_list_size(list);
CKP( data=pp_data_new(realtype,n,heaplist) );
pp_list_startwalk(list,&handle);
for (i=0; i < n; i++) {
lev=pp_list_walk(&handle,0);
switch (zaxis->lev_type) {
case hybrid_sigmap_lev_type:
switch (vtype) {
case lev_type:
val = lev->values.hybrid_sigmap.a / reference_pressure + lev->values.hybrid_sigmap.b;
break;
case hybrid_sigmap_a_type:
val = lev->values.hybrid_sigmap.a;
break;
case hybrid_sigmap_b_type:
val = lev->values.hybrid_sigmap.b;
break;
default:
ERR;
}
break;
case hybrid_height_lev_type:
switch (vtype) {
case lev_type:
val = lev->values.hybrid_height.a;
break;
case hybrid_height_a_type:
val = lev->values.hybrid_height.a;
break;
case hybrid_height_b_type:
val = lev->values.hybrid_height.b;
break;
default:
ERR;
}
break;
case pseudo_lev_type:
switch (vtype) {
case lev_type:
val = (Freal) lev->values.pseudo.index;
break;
default:
ERR;
}
break;
default:
switch (vtype) {
case lev_type:
val = lev->values.misc.level;
break;
default:
ERR;
}
break;
}
((Freal*)(data->values))[i]=val;
}
return data;
ERRBLKP("pp_zaxis_to_values");
}
int pp_process(CuFile *file)
{
int rec, nrec, at_start_rec, at_end_rec;
PPfile *ppfile;
PPfieldvar *fvar;
PPrec *recp;
PPhdr *hdrp;
PPgenaxis *xaxis, *yaxis, *zaxis, *taxis; /* JAK 2005-01-05 */
PPlist *heaplist;
PPlist *fieldvars;
PPlisthandle handle, thandle;
PPlist *gatts, *catts;
int ndims, dimid;
int idim; /* dim number of given type */
int have_time_mean, tmdimid; /* dimensions used for meaning (CF cell methods) */
CuDim *cudims,*dim;
PPdim *ppdim;
int nvars, varid, cvarid;
int ncvars; /* coord vars */
int nfvars; /* field vars */
int nvrec;
PPrec **recs, **vrecs;
CuVar *cuvars, *var;
PPvar *ppvar;
PPlist *atts;
PPlist *axislist;
PPlist *xaxes, *yaxes, *taxes, *zaxes; /* JAK 2005-01-05 */
PPgenaxis *axis; /*JAK 2005-01-10 */
int have_hybrid_sigmap;
PPaxistype axistype;
int rotmapid, rotgridid;
PPlist *rotgrids, *rotmaps;
PProtmap *rotmap;
PProtgrid *rotgrid;
CuVar *lonvar, *latvar;
PPvar *lonppvar, *latppvar;
PPlandmask *landmask;
char *varnamea, *varnameb;
char dimnamestem[CU_MAX_NAME], units[CU_MAX_NAME];
char formulaterms[MAX_ATT_LEN+1];
int dont_free_horiz_axes;
int added;
int zindex,tindex,svindex;
/* ------------------------------------------------------ */
/* initialisation constants which matter */
ncvars = 0;
have_hybrid_sigmap = 0;
have_time_mean = 0;
svindex = 1;
/* initialisation constants just to avoid compiler warnings
* (rather than get accustomed to ignoring warnings)
* but flow logic should mean that these vars do actually get
* initialised elsewhere before use
*/
at_end_rec=0;
xaxis=yaxis=NULL;
fvar=NULL;
zaxis=NULL;
taxis=NULL;
axislist=NULL;
tmdimid=-1;
dont_free_horiz_axes=0;
/* ------------------------------------------------------ */
ppfile = file->internp;
heaplist=ppfile->heaplist;
nrec = ppfile->nrec;
recs = ppfile->recs;
/* initialise elements in the records before sorting */
CKI( pp_initialise_records(recs, nrec, heaplist) );
/* sort the records */
qsort(recs, nrec, sizeof(PPrec*), pp_compare_records);
/* now sort out the list of variables and dimensions */
CKP( fieldvars=pp_list_new(heaplist) );
CKP( xaxes=pp_list_new(heaplist) );
CKP( yaxes=pp_list_new(heaplist) );
CKP( zaxes=pp_list_new(heaplist) );
CKP( taxes=pp_list_new(heaplist) );
CKP( rotmaps=pp_list_new(heaplist) );
CKP( rotgrids=pp_list_new(heaplist) );
/* before main loop over records, look for land mask */
for (rec=0; rec<nrec ; rec++) {
recp = recs[rec];
hdrp = &recp->hdr;
if (pp_var_is_land_mask(hdrp)) {
CKP( landmask = pp_malloc(sizeof(PPlandmask),heaplist) );
CKI( pp_set_horizontal_axes(recp,ppfile,&xaxis,&yaxis,rotmaps,heaplist) );
CKP( landmask->data = pp_data_new(inttype,pp_genaxis_len(xaxis) * pp_genaxis_len(yaxis),heaplist) ); /* JAK 2005-01-05 */
/* read in land mask data values */
landmask->xaxis = xaxis;
landmask->yaxis = yaxis;
CKP( landmask->data->values=pp_read_data_record(recp,ppfile,heaplist) );
ppfile->landmask = landmask;
}
}
/* ====== START LOOP OVER RECORDS ====== */
for (rec=0; rec<nrec ; rec++) {
recp = recs[rec];
hdrp = &recp->hdr;
/* we are at start record of a variable at the very start, or if at we were at the
* end record last time
*/
at_start_rec = ( rec == 0 || at_end_rec );
/* we are at end record of a variable at the very end, or if the header shows a
* difference from the next record which constitutes a different variable
*/
at_end_rec = ( rec == nrec-1 ||
pp_records_from_different_vars(recs[rec+1],recp));
/*------------------------------*/
/* allow for variables which are unsupported for some reason */
if (at_start_rec)
if (pp_test_skip_var(hdrp, ppfile->landmask))
continue;
/* -------
if (at_start_rec)
puts("++++++ START OF VARIABLE +++++++++");
printf("processing record %d / %d\n",rec,nrec);
pp_dump_header(hdrp);
------ */
if (at_start_rec) {
/* ====== THINGS DONE ONLY AT START RECORD OF EACH VARIABLE ====== */
/* get PPvar structure, and initialise certain structure members for tidiness */
CKP( fvar=pp_malloc(sizeof(PPfieldvar), heaplist) );
CKP( fvar->axes=pp_list_new(heaplist) ); /* JAK 2005-01-05 */
fvar->firstrecno = rec;
fvar->firstrec = recp;
if (pp_get_var_compression(hdrp) == 2) {
/* land/sea mask compression: for horiz axes use those of LSM */
xaxis = ppfile->landmask->xaxis;
yaxis = ppfile->landmask->yaxis;
dont_free_horiz_axes = 1;
} else {
CKI( pp_set_horizontal_axes(recp,ppfile,&xaxis,&yaxis,rotmaps,heaplist) );
dont_free_horiz_axes = 0;
}
CKP( zaxis=pp_genaxis_new(zaxis_type,zdir,heaplist) );
CKI( pp_zaxis_set(zaxis,hdrp) );
CKP( taxis=pp_genaxis_new(taxis_type,tdir,heaplist) );
CKI( pp_taxis_set(taxis,hdrp) );
}
/* construct pp_lev struct, and add it to the z axis if not already present
* (could already be present if field has multiple times on each level)
*/
/* ====== THINGS DONE FOR EVERY PP RECORD ====== */
CKI( pp_zaxis_add(zaxis, recp->lev, &zindex, heaplist) );
recp->zindex = zindex;
CKI( pp_taxis_add(taxis, recp->time, &tindex, heaplist) );
recp->tindex = tindex;
/* ===================================================== */
if (at_end_rec) {
/* ====== THINGS DONE ONLY AT END RECORD OF EACH VARIABLE ====== */
fvar->lastrecno = rec;
nvrec = fvar->lastrecno - fvar->firstrecno + 1;
vrecs = recs + fvar->firstrecno;
/* now if the axes are not regular, free the axes, split the variable into a number of variables and try again... */
if (pp_set_disambig_index(zaxis, taxis, vrecs, nvrec, svindex)) {
/* increment the supervar index, used later to show the connection between
* the separate variables into which this one will be split
*/
svindex++;
/* now re-sort this part of the record list, now that we have set the disambig index */
qsort(vrecs, nvrec, sizeof(PPrec*), pp_compare_records);
/* now go back to the start record of the variable; set to one less because it
* will get incremented in the "for" loop reinitialisation
*/
rec = fvar->firstrecno - 1;
/* and free the stuff assoc with the var we won't be using */
if (!dont_free_horiz_axes) {
CKI( pp_genaxis_free(xaxis,heaplist) );
CKI( pp_genaxis_free(yaxis,heaplist) );
}
CKI( pp_genaxis_free(zaxis,heaplist) );
CKI( pp_genaxis_free(taxis,heaplist) );
CKI( pp_free(fvar,heaplist) );
continue;
}
/*------------------------------------------------------------*/
/*
* For each axis, see if it matches an axis which already exists from a previous
* variable.
*
* If so, then free the structure and point to the existing occurrence instead.
*
* If not, then add to the list.
*/
/* x */
CKI( added = pp_list_add_or_find(xaxes, &xaxis, pp_genaxis_compare, 0,
(dont_free_horiz_axes ? NULL : (free_func) pp_genaxis_free),
NULL, heaplist) );
if (added)
ncvars++;
/* y */
CKI( added = pp_list_add_or_find(yaxes, &yaxis, pp_genaxis_compare, 0,
(dont_free_horiz_axes ? NULL : (free_func) pp_genaxis_free),
NULL, heaplist) );
if (added)
ncvars++;
/* z */
CKI( added = pp_list_add_or_find(zaxes, &zaxis, pp_genaxis_compare, 0,
(free_func) pp_genaxis_free,
NULL, heaplist) );
if (added) {
ncvars++;
if (pp_zaxis_lev_type(zaxis) == hybrid_sigmap_lev_type) {
/* two more coord vars (a and b coeffs) */
ncvars+=2;
have_hybrid_sigmap=1;
}
if (pp_zaxis_lev_type(zaxis) == hybrid_height_lev_type) {
/* two more coord vars (a and b coeffs) */
ncvars+=2;
}
}
/* t */
CKI( added = pp_list_add_or_find(taxes, &taxis, pp_genaxis_compare, 0,
(free_func) pp_genaxis_free,
NULL, heaplist) );
if (added) {
ncvars++;
if (pp_taxis_is_time_mean(taxis)) {
/* need to make sure we have the mean dim (size 2),
* also one more coordinate var
*/
have_time_mean=1;
ncvars++;
}
}
/* associate var with these axes */
CKI( pp_list_add(fvar->axes,xaxis,heaplist) );
CKI( pp_list_add(fvar->axes,yaxis,heaplist) );
CKI( pp_list_add(fvar->axes,zaxis,heaplist) );
CKI( pp_list_add(fvar->axes,taxis,heaplist) );
/* get the rotated grid, if any
* (NB this is done *after* the pp_list_add_or_find stuff above, because
* otherwise the axis pointers could get orphaned if the axes are found to
* be duplicates)
*/
CKP( fvar->rotgrid = pp_get_rotgrid(xaxis,yaxis,rotgrids,heaplist) );
/* add the variable */
CKI( pp_list_add(fieldvars, fvar, heaplist) );
/* ===================================================== */
}
}
/* ====================================================================
* Having completed the loop over records, we now know the number of
* dimensions and variables, so we can finally do the relevant calls
* to allocate these arrays and populate them usefully.
* ====================================================================
*/
/* FIRST ALLOCATE THE ARRAYS, and initialise some values */
nfvars = pp_list_size(fieldvars);
if (nfvars <= 0) {
CuError(CU_EOPEN,"No valid fields in file\n");
ERR; /* not the most elegant dealing with this error - ideally would free this file */
}
ndims = pp_list_size(xaxes) + pp_list_size(yaxes) + pp_list_size(zaxes) + pp_list_size(taxes);
if (have_time_mean){
tmdimid=ndims;
ndims++;
}
if (have_hybrid_sigmap) {
/* will need a scalar variable called "p0" */
ncvars++;
}
CKP( cudims = CuCreateDims(file,ndims) );
/* need a grid_mapping variable for every rotation mapping,
* and need lon and lat variables for every rotated grid
*/
ncvars += pp_list_size(rotmaps) + 2*pp_list_size(rotgrids);
nvars = ncvars + nfvars;
CKP( cuvars = CuCreateVars(file,nvars) );
for (dimid=0; dimid<ndims; dimid++) {
dim=&cudims[dimid];
dim->var = (CuVar*)0;
dim->coord = (CuVar*)0;
dim->datatype = realtype;
dim->dimtype = CuGlobalDim;
/* uncomment if internal structure is to be used
* CKP( dim->internp = pp_malloc(sizeof(PPdim), heaplist) );
* ppdim=(PPdim*)dim->internp;
*/
}
for (varid=0; varid<nvars; varid++) {
var=&cuvars[varid];
var->datatype = realtype;
CKP( var->internp = pp_malloc(sizeof(PPvar), heaplist) );
ppvar=(PPvar*)var->internp;
ppvar->firstrecno=-1;
ppvar->lastrecno=-1;
ppvar->data=NULL;
CKP( ppvar->atts = pp_list_new(heaplist) );
}
/*
* NOW POPULATE THE STRUCTURES
*
* The procedure will be to loop over all the axes, adding dimensions and
* variables associated with those axes.
*
* Having done that, any dimensions not associated with axes will be added,
* and then the field variables will be added.
*/
dimid=0;
varid=0;
for (axistype=0; axistype<num_axistype; axistype++) {
switch(axistype){
case xaxistype: axislist=xaxes; break;
case yaxistype: axislist=yaxes; break;
case zaxistype: axislist=zaxes; break;
case taxistype: axislist=taxes; break;
default: pp_switch_bug("cdunifpp_process");
}
pp_list_startwalk(axislist,&handle);
idim=0;
while ((axis=pp_list_walk(&handle,0))!=NULL) {
dim=&cudims[dimid];
var=&cuvars[varid];
ppdim=(PPdim*) dim->internp;
ppvar=(PPvar*) var->internp;
dim->coord = var;
axis->dimid=dimid;
dim->len=pp_genaxis_len(axis);
CKP( ppvar->data=pp_genaxis_getCF(axis,dimnamestem,units,ppvar->atts,heaplist) );
sprintf(dim->name,dimnamestem,idim);
if (units != NULL) {
strncpy(dim->units,units,CU_MAX_NAME);
dim->units[CU_MAX_NAME]='\0';
}
strncpy(var->name,dim->name,CU_MAX_NAME);
var->name[CU_MAX_NAME]='\0';
var->ndims=1;
var->dims[0] = dimid;
varid++;
/* now add certain variables for hybrid_sigmap z axis */
if (axistype == zaxistype && pp_zaxis_lev_type(axis) == hybrid_sigmap_lev_type) {
catts=ppvar->atts; /* attribute list for the main coord var */
/* Hybrid sigma-p A coefficient */
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"z%d_hybrid_sigmap_acoeff",idim);
varnamea=var->name;
CKP( ppvar->data = pp_genaxis_to_values(axis,hybrid_sigmap_a_type,heaplist) );
CKI( pp_add_string_att(ppvar->atts,"units","Pa",heaplist) );
CKI( pp_add_string_att(ppvar->atts,"long_name",
"atmospheric hybrid sigma-pressure 'A' coefficient",heaplist) );
var->ndims=1;
var->dims[0] = dimid;
varid++;
/* Hybrid sigma-p B coefficient */
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"z%d_hybrid_sigmap_bcoeff",idim);
varnameb=var->name;
CKP( ppvar->data = pp_genaxis_to_values(axis,hybrid_sigmap_b_type,heaplist) );
CKI( pp_add_string_att(ppvar->atts,"long_name",
"atmospheric hybrid sigma-pressure 'B' coefficient",heaplist) );
var->ndims=1;
var->dims[0] = dimid;
varid++;
snprintf(formulaterms,MAX_ATT_LEN,"ap: %s b: %s ps: ps p0: p0",varnamea,varnameb);
CKI( pp_add_string_att(catts,"formula_terms",formulaterms,heaplist) );
CKI( pp_add_string_att(catts,"standard_name","atmosphere_hybrid_sigma_pressure_coordinate",heaplist) );
CKI( pp_add_string_att(catts,"comments",
"The \"ps\" term in formula_terms is set to \"ps\" variable. "
"This variable may or may not be provided.",heaplist) );
}
/* now add certain variables for hybrid_height z axis */
if (axistype == zaxistype && pp_zaxis_lev_type(axis) == hybrid_height_lev_type) {
catts=ppvar->atts; /* attribute list for the main coord var */
/* Hybrid height A coefficient */
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"z%d_hybrid_height_acoeff",idim);
varnamea=var->name;
CKP( ppvar->data = pp_genaxis_to_values(axis,hybrid_height_a_type,heaplist) );
CKI( pp_add_string_att(ppvar->atts,"units","m",heaplist) );
var->ndims=1;
var->dims[0] = dimid;
varid++;
/* Hybrid height B coefficient */
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"z%d_hybrid_height_bcoeff",idim);
varnameb=var->name;
CKP( ppvar->data = pp_genaxis_to_values(axis,hybrid_height_b_type,heaplist) );
var->ndims=1;
var->dims[0] = dimid;
varid++;
snprintf(formulaterms,MAX_ATT_LEN,"a: %s b: %s orog: orography",varnamea,varnameb);
CKI( pp_add_string_att(catts,"formula_terms",formulaterms,heaplist) );
CKI( pp_add_string_att(catts,"standard_name","atmosphere_hybrid_sigma_pressure_coordinate",heaplist) );
CKI( pp_add_string_att(catts,"comments",
"The \"orog\" term in formula_terms is set to \"orography\" variable. "
"This variable may or may not be provided.",heaplist) );
}
/* add the boundary variable for time mean */
if (axistype == taxistype && pp_taxis_is_time_mean(axis)) {
catts=ppvar->atts; /* attribute list for the main coord var */
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"time_bnd%d",idim);
CKP( ppvar->data = pp_taxis_to_boundary_values(axis->axis,heaplist) );
var->ndims=2;
var->dims[0]=dimid;
var->dims[1]=tmdimid;
CKI( pp_add_string_att(catts,"bounds",var->name,heaplist) );
varid++;
}
dimid++;
idim++;
} /* end loop over axes of given */
} /* end loop over axis types */
/* add nv dimension if we had time mean */
if (have_time_mean) {
dim=&cudims[dimid];
strcpy(dim->name,"nv");
dim->len=2;
/* Should have tmdimid=dimid, but actually we already set it above (it evaluates to ndims)
* as we needed it before we got here. So just do a check here.
*/
if ( tmdimid != dimid ) {
pp_error_mesg("cdunifpp_process","ID wrong for 'nv' dimension?");
ERR;
}
dimid++;
}
/* add p0 variable if we had hybrid_sigmap coords */
if (have_hybrid_sigmap) {
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"p0");
var->ndims=0;
CKI( pp_add_string_att(ppvar->atts,"long_name",
"reference pressure value for atmospheric hybrid sigma-pressure coordinates",
heaplist) );
/* single value consisting of p0 */
CKP( ppvar->data = pp_data_new(realtype,1,heaplist) );
((Freal*)(ppvar->data->values))[0]=reference_pressure;
varid++;
}
/* add any rotated_pole variables */
rotmapid=0;
pp_list_startwalk(rotmaps,&handle);
while ((rotmap=pp_list_walk(&handle,0))!=NULL) {
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
sprintf(var->name,"rotated_pole%d",rotmapid);
strncpy(rotmap->name,var->name,CU_MAX_NAME);
rotmap->name[CU_MAX_NAME]='\0';
/* single value of arbitrary type; set as integer = 0 */
var->datatype=inttype;
var->ndims=0;
CKP( ppvar->data = pp_data_new(inttype,1,heaplist) );
((Freal*)(ppvar->data->values))[0]=0;
/* and add attributes */
catts=ppvar->atts;
CKI( pp_add_string_att(catts,"grid_mapping_name","rotated_latitude_longitude",heaplist) );
CKI( pp_add_att(catts,"grid_north_pole_longitude",realtype,1,&rotmap->pole_lon,heaplist) );
CKI( pp_add_att(catts,"grid_north_pole_latitude",realtype,1,&rotmap->pole_lat,heaplist) );
CKI( pp_add_att(catts,"north_pole_grid_longitude",realtype,1,&rotmap->truepole_gridlon,heaplist) );
rotmapid++;
varid++;
}
/* and add any lon, lat variables for rotated grids */
rotgridid=0;
pp_list_startwalk(rotgrids,&handle);
while ((rotgrid=pp_list_walk(&handle,0))!=NULL) {
lonvar=&cuvars[varid];
lonppvar=(PPvar*) lonvar->internp;
varid++;
latvar=&cuvars[varid];
latppvar=(PPvar*) latvar->internp;
varid++;
xaxis = rotgrid->xaxis;
yaxis = rotgrid->yaxis;
sprintf(lonvar->name,"true_lon%d",rotgridid);
sprintf(latvar->name,"true_lat%d",rotgridid);
sprintf(rotgrid->coords,"%s %s",lonvar->name,latvar->name);
lonvar->ndims=2;
latvar->ndims=2;
lonvar->dims[0]=yaxis->dimid;
latvar->dims[0]=yaxis->dimid;
lonvar->dims[1]=xaxis->dimid;
latvar->dims[1]=xaxis->dimid;
CKI( pp_calc_rot_grid(rotgrid,&lonppvar->data,&latppvar->data,heaplist) );
CKI( pp_add_string_att(lonppvar->atts,"long_name","longitude",heaplist) );
CKI( pp_add_string_att(latppvar->atts,"long_name","latitude",heaplist) );
CKI( pp_add_string_att(lonppvar->atts,"standard_name","longitude",heaplist) );
CKI( pp_add_string_att(latppvar->atts,"standard_name","latitude",heaplist) );
CKI( pp_add_string_att(lonppvar->atts,"units","degrees_east",heaplist) );
CKI( pp_add_string_att(latppvar->atts,"units","degrees_north",heaplist) );
CKI( pp_add_att(lonppvar->atts,"modulo",realtype,1,&lon_modulo,heaplist) );
rotgridid++;
}
/* sanity check - the variable ID for the next variable to be added should
* now match the number of coordinate variables
*/
if ( varid != ncvars ) {
pp_error_mesg("cdunifpp_process","wrong number of coord vars?");
ERR;
}
/* add all the attributes for coord variables
* (didn't do inside the loop because more complicated
* for hybrid z coords / t mean)
*/
for (cvarid=0; cvarid<ncvars; cvarid++) {
var=&cuvars[cvarid];
ppvar=(PPvar*) var->internp;
CKI( pp_copy_and_free_atts(file,var,ppvar->atts,heaplist) );
}
/*========================================================
* Okay we've done all the variables related to dimensions
* Add the field variables.
*========================================================
*/
pp_list_startwalk(fieldvars,&handle);
while ((fvar=pp_list_walk(&handle,0))!=NULL) {
var=&cuvars[varid];
ppvar=(PPvar*) var->internp;
atts = ppvar->atts;
hdrp = &fvar->firstrec->hdr;
CKI( pp_var_lookup(hdrp, &fvar->stashmeta) );
CKI( pp_get_var_name(varid, fvar->stashmeta.shortname, cuvars) );
var->ndims=4; /* rpw axeslist len */
/*
* Axes in fvar->axes list are fastest varying first (lon,lat,lev,time)
* But require them in netCDF-like order (time,lev,lat,lon), so
* reverse the order while copying into var->dims.
*/
idim=var->ndims;
pp_list_startwalk(fvar->axes,&thandle);
while ((axis=pp_list_walk(&thandle,0)) != NULL) {
var->dims[--idim] = axis->dimid;
}
var->datatype = pp_get_var_type(hdrp);
ppvar->firstrecno = fvar->firstrecno;
ppvar->lastrecno = fvar->lastrecno;
CKI( pp_var_get_extra_atts(var, fvar, cudims, atts, heaplist) );
CKI( pp_copy_and_free_atts(file, var, atts, heaplist) );
varid++;
}
/* sanity check - the variable ID for the next variable to be added (if there was one,
* which there isn't), should now match the total number of variables
*/
if ( varid != nvars ) {
pp_error_mesg("cdunifpp_process","wrong number of vars?");
ERR;
}
/* set numbers in file structure */
file->ndims=ndims;
file->nvars=nvars;
file->recdim=-1;
/* set global attributes */
CKP( gatts = pp_get_global_attributes(file->controlpath, ppfile, heaplist) );
CKI( pp_copy_and_free_atts(file,NULL,gatts,heaplist) );
/*========================================================
* All done and ready for dimget / varget.
*========================================================
*/
/* free what memory we can */
CKI( pp_free_tmp_vars(xaxes, yaxes, zaxes, taxes, fieldvars, heaplist) );
return 0;
ERRBLKI("pp_process");
}