int main(int argc, char *argv[])
{
char *infile = "TPV7.fits";
char keyrec[81], header[288001], *disfn;
int dopoly, gotend, iblock, ikeyrec, inc, itest, j, k, n, naxis[2], naxis1,
naxis2, nClosure, nFail, nkeyrec, nsamp, nreject, nTest, nwcs, p1, p2,
status;
clock_t t0, tp2x, tx2p;
double absmax, dp1, dp2, *img, *img1, *img2, pix[8], pixblc[2], pixsamp[2],
pixtrc[2], px, *px0, *px1, pxi[8], rel, resid, relmax;
double *avgdis, *avgtot, *maxdis, *maxtot, *rmsdis, *rmstot, stats[9];
FILE *fptr;
struct linprm affine, *lin, *linpol, *lintpv;
struct wcsprm *wcs, wcspol;
wcserr_enable(1);
wcsprintf_set(stdout);
/* Set line buffering in case stdout is redirected to a file, otherwise
* stdout and stderr messages will be jumbled (stderr is unbuffered). */
setvbuf(stdout, NULL, _IOLBF, 0);
wcsprintf("Testing closure of WCSLIB distortion routines (tdis1.c)\n"
"-------------------------------------------------------\n");
/* List status return messages. */
wcsprintf("\nList of dis status return values:\n");
for (status = 1; status <= 5; status++) {
wcsprintf("%4d: %s.\n", status, dis_errmsg[status]);
}
wcsprintf("\n");
/* Optional file name specified? */
if (1 < argc) {
infile = argv[1];
}
/* Read in the FITS header, excluding COMMENT and HISTORY keyrecords. */
if ((fptr = fopen(infile, "r")) == 0) {
wcsprintf("ERROR opening %s\n", infile);
return 1;
}
memset(naxis, 0, 2*sizeof(int));
k = 0;
nkeyrec = 0;
gotend = 0;
for (iblock = 0; iblock < 100; iblock++) {
for (ikeyrec = 0; ikeyrec < 36; ikeyrec++) {
if (fgets(keyrec, 81, fptr) == 0) {
break;
}
if (strncmp(keyrec, " ", 8) == 0) continue;
if (strncmp(keyrec, "COMMENT ", 8) == 0) continue;
if (strncmp(keyrec, "HISTORY ", 8) == 0) continue;
if (strncmp(keyrec, "NAXIS", 5) == 0) {
if (keyrec[5] == ' ') {
sscanf(keyrec+10, "%d", &n);
if (n != 2) {
wcsprintf("ERROR, expecting a 2D image.\n");
return 1;
}
continue;
}
sscanf(keyrec+5, "%d = %d", &j, &n);
naxis[j-1] = n;
continue;
}
strncpy(header+k, keyrec, 80);
k += 80;
nkeyrec++;
if (strncmp(keyrec, "END ", 10) == 0) {
/* An END keyrecord was read, but read the rest of the block. */
gotend = 1;
}
}
if (gotend) break;
}
fclose(fptr);
/* Parse the header. */
if ((wcspih(header, nkeyrec, WCSHDR_none, 2, &nreject, &nwcs, &wcs))) {
wcsperr(wcs, 0x0);
return 1;
}
/* Is it TPV? */
dopoly = 0;
if (strcmp(wcs->ctype[0], "RA---TPV") == 0) {
/* Copy it and translate to Polynomial for later use. */
wcspol.flag = -1;
if (wcscopy(1, wcs, &wcspol)) {
wcsperr(wcs, 0x0);
return 1;
}
/* Translate TPV to Polynomial. */
tpv2poly(&wcspol);
wcspol.flag = -1;
if (wcsset(&wcspol)) {
wcsperr(&wcspol, 0x0);
return 1;
}
dopoly = 1;
}
/* wcsset() translates the TPV "projection" into a sequent distortion. */
if (wcsset(wcs)) {
wcsperr(wcs, 0x0);
return 1;
}
/* Henceforth, we will work with linprm. */
lin = &(wcs->lin);
/* Get statistics on the distortion in the inner quarter of the image. */
maxdis = stats;
maxtot = maxdis + 2;
avgdis = maxtot + 1;
avgtot = avgdis + 2;
rmsdis = avgtot + 1;
rmstot = rmsdis + 2;
pixblc[0] = 0.25 * naxis[0];
pixblc[1] = 0.25 * naxis[1];
pixtrc[0] = 0.75 * naxis[0];
pixtrc[1] = 0.75 * naxis[1];
pixsamp[0] = (pixtrc[0] - pixblc[0])/512.0;
pixsamp[1] = (pixtrc[1] - pixblc[1])/512.0;
if (pixsamp[0] < 1.0) pixsamp[0] = 1.0;
if (pixsamp[1] < 1.0) pixsamp[1] = 1.0;
if (linwarp(lin, pixblc, pixtrc, pixsamp, &nsamp,
maxdis, maxtot, avgdis, avgtot, rmsdis, rmstot)) {
linperr(lin, 0x0);
return 1;
}
for (k = 0; k < 9; k++) {
if (fabs(stats[k]) < 0.0005) stats[k] = 0.0;
}
wcsprintf("linwarp() statistics computed over %d sample points:\n"
" Max distortion, axis 1: %8.3f pixels\n"
" axis 2: %8.3f pixels\n"
" total: %8.3f pixels\n"
" Mean distortion, axis 1: %8.3f pixels\n"
" axis 2: %8.3f pixels\n"
" total: %8.3f pixels\n"
" RMS distortion, axis 1: %8.3f pixels\n"
" axis 2: %8.3f pixels\n"
" total: %8.3f pixels\n",
nsamp, maxdis[0], maxdis[1], *maxtot,
avgdis[0], avgdis[1], *avgtot,
rmsdis[0], rmsdis[1], *rmstot);
if (lin->disseq) {
/* Exercise diswarp() as well. */
wcsprintf("\n");
/* Define a rectangle in intermediate pixel coordinates that just */
/* encompasses the inner quarter of the image. For this we need */
/* to switch off CDELTia scaling and all distortions. */
affine.flag = -1;
if ((status = lincpy(1, lin, &affine))) {
linperr(lin, 0x0);
return 1;
}
affine.cdelt[0] = 1.0;
affine.cdelt[1] = 1.0;
if ((status = (lindis(1, &affine, 0x0) ||
lindis(2, &affine, 0x0) ||
linset(&affine)))) {
linperr(&affine, 0x0);
return 1;
}
pix[0] = pixblc[0];
pix[1] = pixblc[1];
pix[2] = pixtrc[0];
pix[3] = pixblc[1];
pix[4] = pixtrc[0];
pix[5] = pixtrc[1];
pix[6] = pixblc[0];
pix[7] = pixtrc[1];
if (linp2x(&affine, 4, 2, pix, pxi)) {
linperr(&affine, 0x0);
return 1;
}
linfree(&affine);
pixblc[0] = pxi[0];
pixblc[1] = pxi[1];
pixtrc[0] = pxi[0];
pixtrc[1] = pxi[1];
k = 2;
for (j = 1; j < 4; j++) {
if (pixblc[0] > pxi[k]) pixblc[0] = pxi[k];
if (pixtrc[0] < pxi[k]) pixtrc[0] = pxi[k];
k++;
if (pixblc[1] > pxi[k]) pixblc[1] = pxi[k];
if (pixtrc[1] < pxi[k]) pixtrc[1] = pxi[k];
k++;
}
pixsamp[0] = (pixtrc[0] - pixblc[0])/512.0;
pixsamp[1] = (pixtrc[1] - pixblc[1])/512.0;
if (diswarp(lin->disseq, pixblc, pixtrc, pixsamp, &nsamp,
maxdis, maxtot, avgdis, avgtot, rmsdis, rmstot)) {
wcserr_prt(lin->disseq->err, 0x0);
return 1;
}
for (k = 0; k < 9; k++) {
if (fabs(stats[k]) < 0.0005) stats[k] = 0.0;
}
wcsprintf("diswarp() statistics computed over %d sample points:\n"
" Max distortion, axis 1: %8.3f units\n"
" axis 2: %8.3f units\n"
" total: %8.3f units\n"
" Mean distortion, axis 1: %8.3f units\n"
" axis 2: %8.3f units\n"
" total: %8.3f units\n"
" RMS distortion, axis 1: %8.3f units\n"
" axis 2: %8.3f units\n"
" total: %8.3f units\n",
nsamp, maxdis[0], maxdis[1], *maxtot,
avgdis[0], avgdis[1], *avgtot,
rmsdis[0], rmsdis[1], *rmstot);
}
/* The image size determines the test domain. */
if ((naxis1 = naxis[0]) == 0) {
naxis1 = 2*wcs->crpix[0] + 1;
}
if ((naxis2 = naxis[1]) == 0) {
naxis2 = 2*wcs->crpix[1] + 1;
}
/* Limit the number of tests. */
inc = 1;
while ((naxis1/inc)*(naxis2/inc) > 800000) {
inc *= 2;
}
n = naxis1 / inc;
px0 = calloc(4*(2*n), sizeof(double));
px1 = px0 + 2*n ;
img = px1 + 2*n ;
img1 = img;
img2 = img + 2*n;
for (itest = 0; itest < 2; itest++) {
if (itest) {
if (!dopoly) break;
lin = &(wcspol.lin);
}
if (lin->dispre) {
disfn = lin->dispre->dtype[0];
} else if (lin->disseq) {
disfn = lin->disseq->dtype[0];
}
wcsprintf("\n");
/* Now the closure test. */
tp2x = 0;
tx2p = 0;
nTest = 0;
nFail = 0;
nClosure = 0;
absmax = 0.0;
relmax = 0.0;
for (p2 = 1; p2 <= naxis2; p2 += inc) {
k = 0;
for (p1 = 1; p1 <= naxis1; p1 += inc) {
px0[k++] = (double)p1;
px0[k++] = (double)p2;
}
t0 = clock();
if (linp2x(lin, n, 2, px0, img)) {
linperr(lin, 0x0);
nFail = 1;
break;
}
tp2x += clock() - t0;
t0 = clock();
if (linx2p(lin, n, 2, img, px1)) {
linperr(lin, 0x0);
nFail = 1;
break;
}
tx2p += clock() - t0;
/* Check closure. */
k = 0;
for (k = 0; k < 2*n ; k += 2) {
dp1 = fabs(px1[k] - px0[k]);
dp2 = fabs(px1[k+1] - px0[k+1]);
resid = (dp1 > dp2) ? dp1 : dp2;
if (resid > absmax) absmax = resid;
if (resid > ATOL) {
nClosure++;
wcsprintf("Absolute closure error:\n");
wcsprintf(" pix: %18.12f %18.12f\n", px0[k], px0[k+1]);
wcsprintf(" -> img: %18.12f %18.12f\n", img[k], img[k+1]);
wcsprintf(" -> pix: %18.12f %18.12f\n", px1[k], px1[k+1]);
wcsprintf("\n");
continue;
}
resid = 0.0;
if ((px = fabs(px0[k])) > 1.0) resid = dp1/px;
if ((px = fabs(px0[k+1])) > 1.0) {
if ((rel = dp2/px) > resid) resid = rel;
}
if (resid > relmax) relmax = resid;
if (resid > FTOL) {
nClosure++;
wcsprintf("Relative closure error:\n");
wcsprintf(" pix: %18.12f %18.12f\n", px0[k], px0[k+1]);
wcsprintf(" -> img: %18.12f %18.12f\n", img[k], img[k+1]);
wcsprintf(" -> pix: %18.12f %18.12f\n", px1[k], px1[k+1]);
wcsprintf("\n");
}
}
nTest += n;
}
if (nFail) {
wcsprintf("\nFAIL: The %s test failed to complete.\n", disfn);
} else {
wcsprintf("linp2x/linx2p with %s distortions:\n"
" Completed %d closure tests.\n"
" Maximum absolute closure residual = %.2e pixel.\n"
" Maximum relative closure residual = %.2e.\n", disfn,
nTest, absmax, relmax);
wcsprintf("\n");
wcsprintf(" linp2x time (ns): %6.0f\n linx2p time (ns): %6.0f\n\n",
1.0e9*((double)tp2x/CLOCKS_PER_SEC)/nTest,
1.0e9*((double)tx2p/CLOCKS_PER_SEC)/nTest);
if (nClosure) {
wcsprintf("FAIL: %d closure residuals exceed reporting tolerance.\n",
nClosure);
} else {
wcsprintf("PASS: All %s closure residuals are within reporting "
"tolerance.\n", disfn);
}
}
}
/* Compare TPV with Polynomial over the test domain. */
if (dopoly) {
wcsprintf("\n");
nTest = 0;
nFail = 0;
absmax = 0.0;
lintpv = &(wcs->lin);
linpol = &(wcspol.lin);
for (p2 = 1; p2 <= naxis2; p2 += inc) {
k = 0;
for (p1 = 1; p1 <= naxis1; p1 += inc) {
px0[k++] = (double)p1;
px0[k++] = (double)p2;
}
if (linp2x(lintpv, n, 2, px0, img1)) {
linperr(lintpv, 0x0);
break;
}
if (linp2x(linpol, n, 2, px0, img2)) {
linperr(linpol, 0x0);
break;
}
/* Check agreement. */
k = 0;
for (k = 0; k < 2*n ; k += 2) {
dp1 = fabs(img2[k] - img1[k]);
dp2 = fabs(img2[k+1] - img1[k+1]);
resid = (dp1 > dp2) ? dp1 : dp2;
if (resid > absmax) absmax = resid;
if (resid > ATOL) {
nFail++;
wcsprintf("TPV - Polynomial disagreement:\n");
wcsprintf(" pix: %18.12f %18.12f\n", px0[k], px0[k+1]);
wcsprintf(" -> TPV: %18.12f %18.12f\n", img1[k], img1[k+1]);
wcsprintf(" -> Pol: %18.12f %18.12f\n", img2[k], img2[k+1]);
wcsprintf("\n");
continue;
}
}
nTest += n;
}
wcsprintf("linp2x, TPV vs Polynomial distortions:\n"
" Completed %d comparisons.\n"
" Maximum absolute disagreement = %.2e units.\n", nTest, absmax);
wcsprintf("\n");
if (nFail) {
wcsprintf("FAIL: %d comparisons exceed reporting tolerance.\n", nFail);
} else {
wcsprintf("PASS: All TPV vs Polynomial comparisons are within "
"reporting tolerance.\n");
}
}
free(px0);
wcsvfree(&nwcs, &wcs);
wcsfree(&wcspol);
return nFail || nClosure;
}
int linwarp(
struct linprm *lin,
const double pixblc[],
const double pixtrc[],
const double pixsamp[],
int *nsamp,
double maxdis[],
double *maxtot,
double avgdis[],
double *avgtot,
double rmsdis[],
double *rmstot)
{
static const char *function = "linwarp";
int carry, i, j, naxis, ncoord, status = 0;
double dpix, dpx2, dssq, *img, *pix0, *pix0p, *pix1, *pix1p, *pixend,
*pixinc, pixspan, *ssqdis, ssqtot, *sumdis, sumtot, totdis;
struct linprm affine;
struct wcserr **err;
/* Initialize. */
if (lin == 0x0) return LINERR_NULL_POINTER;
err = &(lin->err);
naxis = lin->naxis;
if (nsamp) *nsamp = 0;
for (j = 0; j < naxis; j++) {
if (maxdis) maxdis[j] = 0.0;
if (avgdis) avgdis[j] = 0.0;
if (rmsdis) rmsdis[j] = 0.0;
}
if (maxtot) *maxtot = 0.0;
if (avgtot) *avgtot = 0.0;
if (rmstot) *rmstot = 0.0;
/* Quick return if no distortions. */
if (lin->affine) return 0;
/* It's easier if there are no sequent distortions! */
if (lin->disseq == 0x0) {
status = diswarp(lin->dispre, pixblc, pixtrc, pixsamp, nsamp,
maxdis, maxtot, avgdis, avgtot, rmsdis, rmstot);
return wcserr_set(LIN_ERRMSG(lin_diserr[status]));
}
/* Make a reference copy of lin without distortions. */
affine.flag = -1;
if ((status = (lincpy(1, lin, &affine) ||
lindis(1, &affine, 0x0) ||
lindis(2, &affine, 0x0) ||
linset(&affine)))) {
return wcserr_set(LIN_ERRMSG(status));
}
/* Work out increments on each axis. */
pixinc = lin->tmpcrd;
for (j = 0; j < naxis; j++) {
pixspan = pixtrc[j] - (pixblc ? pixblc[j] : 1.0);
if (pixsamp == 0x0) {
pixinc[j] = 1.0;
} else if (pixsamp[j] == 0.0) {
pixinc[j] = 1.0;
} else if (pixsamp[j] > 0.0) {
pixinc[j] = pixsamp[j];
} else if (pixsamp[j] > -1.5) {
pixinc[j] = 2.0*pixspan;
} else {
pixinc[j] = pixspan / ((int)(-pixsamp[j] - 0.5));
}
if (j == 0) {
/* Number of samples on axis 1. */
ncoord = 1 + (int)((pixspan/pixinc[0]) + 0.5);
}
}
/* Get memory for processing the image row by row. */
if ((pix0 = calloc((3*ncoord+4)*naxis, sizeof(double))) == 0x0) {
return wcserr_set(LIN_ERRMSG(LINERR_MEMORY));
}
img = pix0 + naxis*ncoord;
pix1 = img + naxis*ncoord;
pixinc = pix1 + naxis*ncoord;
pixend = pixinc + naxis;
sumdis = pixend + naxis;
ssqdis = sumdis + naxis;
/* Copy tmpcrd since linp2x() will overwrite it. */
memcpy(pixinc, lin->tmpcrd, naxis*sizeof(double));
/* Set up the array of pixel coordinates. */
for (j = 0; j < naxis; j++) {
pix0[j] = pixblc ? pixblc[j] : 1.0;
pixend[j] = pixtrc[j] + 0.5*pixinc[j];
}
pix0p = pix0 + naxis;
for (i = 1; i < ncoord; i++) {
*(pix0p++) = pix0[0] + i*pixinc[0];
for (j = 1; j < naxis; j++) {
*(pix0p++) = pix0[j];
}
}
/* Initialize accumulators. */
for (j = 0; j < naxis; j++) {
sumdis[j] = 0.0;
ssqdis[j] = 0.0;
}
sumtot = 0.0;
ssqtot = 0.0;
/* Loop over N dimensions. */
carry = 0;
while (carry == 0) {
if ((status = linp2x(lin, ncoord, naxis, pix0, img))) {
/* (Preserve the error message set by linp2x().) */
goto cleanup;
}
if ((status = linx2p(&affine, ncoord, naxis, img, pix1))) {
/* (Preserve the error message set by linx2p().) */
goto cleanup;
}
/* Accumulate statistics. */
pix0p = pix0;
pix1p = pix1;
for (i = 0; i < ncoord; i++) {
(*nsamp)++;
dssq = 0.0;
for (j = 0; j < naxis; j++) {
dpix = *(pix1p++) - *(pix0p++);
dpx2 = dpix*dpix;
sumdis[j] += dpix;
ssqdis[j] += dpx2;
if (maxdis && (dpix = fabs(dpix)) > maxdis[j]) maxdis[j] = dpix;
dssq += dpx2;
}
totdis = sqrt(dssq);
sumtot += totdis;
ssqtot += totdis*totdis;
if (maxtot && *maxtot < totdis) *maxtot = totdis;
}
/* Next array of pixel coordinates. */
for (j = 1; j < naxis; j++) {
pix0[j] += pixinc[j];
if ((carry = (pix0[j] > pixend[j]))) {
pix0[j] = pixblc ? pixblc[j] : 1.0;
}
pix0p = pix0 + naxis + j;
for (i = 1; i < ncoord; i++) {
*pix0p = pix0[j];
pix0p += naxis;
}
if (carry == 0) break;
}
}
/* Compute the means and RMSs. */
for (j = 0; j < naxis; j++) {
ssqdis[j] /= *nsamp;
sumdis[j] /= *nsamp;
if (avgdis) avgdis[j] = sumdis[j];
if (rmsdis) rmsdis[j] = sqrt(ssqdis[j] - sumdis[j]*sumdis[j]);
}
ssqtot /= *nsamp;
sumtot /= *nsamp;
if (avgtot) *avgtot = sumtot;
if (rmstot) *rmstot = sqrt(ssqtot - sumtot*sumtot);
cleanup:
free(pix0);
return status;
}
int tpv2poly(struct wcsprm *wcs)
{
const int tpvpow[40][3] = {
{0, 0, 0}, {1, 0, 0}, {0, 1, 0}, {0, 0, 1}, {2, 0, 0}, /* 0 - 4 */
{1, 1, 0}, {0, 2, 0}, {3, 0, 0}, {2, 1, 0}, {1, 2, 0}, /* 5 - 9 */
{0, 3, 0}, {0, 0, 3}, {4, 0, 0}, {3, 1, 0}, {2, 2, 0}, /* 10 - 14 */
{1, 3, 0}, {0, 4, 0}, {5, 0, 0}, {4, 1, 0}, {3, 2, 0}, /* 15 - 19 */
{2, 3, 0}, {1, 4, 0}, {0, 5, 0}, {0, 0, 5}, {6, 0, 0}, /* 20 - 24 */
{5, 1, 0}, {4, 2, 0}, {3, 3, 0}, {2, 4, 0}, {1, 5, 0}, /* 25 - 29 */
{0, 6, 0}, {7, 0, 0}, {6, 1, 0}, {5, 2, 0}, {4, 3, 0}, /* 30 - 34 */
{3, 4, 0}, {2, 5, 0}, {1, 6, 0}, {0, 7, 0}, {0, 0, 7}}; /* 35 - 39 */
char dqi[8], dqifield[32];
int i, k, m, naux, nterms;
struct disprm *dis;
struct dpkey *keyp;
/* Attach a disprm struct to linprm. Also inits it. */
if ((dis = calloc(1, sizeof(struct disprm))) == 0x0) {
return 1;
}
/* Set values in the disprm struct. */
dis->flag = -1;
disndp(22 + 3*wcs->npv);
lindis(2, &(wcs->lin), dis);
/* The asterisk after the name prevents translation to TPD. */
sprintf(dis->dtype[0], "Polynomial*");
sprintf(dis->dtype[1], "Polynomial*");
keyp = dis->dp;
for (i = 1; i <= 2; i++) {
/* Glean essential information from the PVi_ma. */
naux = 0;
nterms = 0;
for (k = 0; k < wcs->npv; k++) {
if (wcs->pv[k].i != i) continue;
m = wcs->pv[k].m;
if (tpvpow[m][2]) naux = 1;
nterms++;
}
/* Distortion parameters for axis i. */
sprintf(dqi, "DQ%d", i);
dpfill(keyp++, dqi, "NAXES", i, 0, 2, 0.0);
dpfill(keyp++, dqi, "AXIS.1", i, 0, (i==1)?1:2, 0.0);
dpfill(keyp++, dqi, "AXIS.2", i, 0, (i==1)?2:1, 0.0);
if (naux) {
dpfill(keyp++, dqi, "NAUX", i, 0, 1, 0.0);
dpfill(keyp++, dqi, "AUX.1.COEFF.0", i, 1, 0, 0.0);
dpfill(keyp++, dqi, "AUX.1.POWER.0", i, 1, 0, 0.5);
dpfill(keyp++, dqi, "AUX.1.COEFF.1", i, 1, 0, 1.0);
dpfill(keyp++, dqi, "AUX.1.POWER.1", i, 1, 0, 2.0);
dpfill(keyp++, dqi, "AUX.1.COEFF.2", i, 1, 0, 1.0);
dpfill(keyp++, dqi, "AUX.1.POWER.2", i, 1, 0, 2.0);
}
dpfill(keyp++, dqi, "NTERMS", i, 0, nterms, 0.0);
nterms = 0;
for (k = 0; k < wcs->npv; k++) {
if (wcs->pv[k].i != i) continue;
sprintf(dqifield, "%s.TERM.%d", dqi, ++nterms);
dpfill(keyp++, dqifield, "COEFF", i, 1, 0, wcs->pv[k].value);
m = wcs->pv[k].m;
if (tpvpow[m][0]) {
dpfill(keyp++, dqifield, "VAR.1", i, 1, 0, (double)tpvpow[m][0]);
}
if (tpvpow[m][1]) {
dpfill(keyp++, dqifield, "VAR.2", i, 1, 0, (double)tpvpow[m][1]);
}
if (tpvpow[m][2]) {
dpfill(keyp++, dqifield, "AUX.1", i, 1, 0, (double)tpvpow[m][2]);
}
}
}
dis->ndp = keyp - dis->dp;
/* TPV now becomes TAN. */
strcpy(wcs->ctype[0]+5, "TAN");
strcpy(wcs->ctype[1]+5, "TAN");
wcs->npv = 0;
return 0;
}
