void startree_search_for(const startree_t* s, const double* xyzcenter, double radius2,
double** xyzresults, double** radecresults,
int** starinds, int* nresults) {
kdtree_qres_t* res = NULL;
int opts;
double* xyz;
int i, N;
opts = KD_OPTIONS_SMALL_RADIUS;
if (xyzresults || radecresults)
opts |= KD_OPTIONS_RETURN_POINTS;
res = kdtree_rangesearch_options(s->tree, xyzcenter, radius2, opts);
if (!res || !res->nres) {
if (xyzresults)
*xyzresults = NULL;
if (radecresults)
*radecresults = NULL;
if (starinds)
*starinds = NULL;
*nresults = 0;
if (res)
kdtree_free_query(res);
return;
}
xyz = res->results.d;
N = res->nres;
*nresults = N;
if (radecresults) {
*radecresults = malloc(N * 2 * sizeof(double));
for (i=0; i<N; i++)
xyzarr2radecdegarr(xyz + i*3, (*radecresults) + i*2);
}
if (xyzresults) {
// Steal the results array.
*xyzresults = xyz;
res->results.d = NULL;
}
if (starinds) {
*starinds = malloc(res->nres * sizeof(int));
for (i=0; i<N; i++)
(*starinds)[i] = res->inds[i];
}
kdtree_free_query(res);
}
static
int fit_tan_wcs_solve(const double* starxyz,
const double* fieldxy,
const double* weights,
int N,
const double* crpix,
const tan_t* tanin,
tan_t* tanout,
double* p_scale) {
int i, j, k;
double field_cm[2] = {0, 0};
double cov[4] = {0, 0, 0, 0};
double R[4] = {0, 0, 0, 0};
double scale;
// projected star coordinates
double* p;
// relative field coordinates
double* f;
double pcm[2] = {0, 0};
double w = 0;
double totalw;
gsl_matrix* A;
gsl_matrix* U;
gsl_matrix* V;
gsl_vector* S;
gsl_vector* work;
gsl_matrix_view vcov;
gsl_matrix_view vR;
double crxyz[3];
double star_cm[3] = {0, 0, 0};
assert(((tanin != NULL) && (crpix != NULL)) || ((tanin == NULL) && (crpix == NULL)));
if (tanin) {
// default vals...
memcpy(tanout, tanin, sizeof(tan_t));
} else {
memset(tanout, 0, sizeof(tan_t));
}
// -allocate and fill "p" and "f" arrays. ("projected" and "field")
p = malloc(N * 2 * sizeof(double));
f = malloc(N * 2 * sizeof(double));
// -get field center-of-mass
totalw = 0.0;
for (i=0; i<N; i++) {
w = (weights ? weights[i] : 1.0);
field_cm[0] += w * fieldxy[i*2 + 0];
field_cm[1] += w * fieldxy[i*2 + 1];
totalw += w;
}
field_cm[0] /= totalw;
field_cm[1] /= totalw;
// Subtract it out.
for (i=0; i<N; i++) {
f[2*i+0] = fieldxy[2*i+0] - field_cm[0];
f[2*i+1] = fieldxy[2*i+1] - field_cm[1];
}
if (tanin) {
// Use original WCS to set the center of projection to the new crpix.
tan_pixelxy2xyzarr(tanin, crpix[0], crpix[1], crxyz);
for (i=0; i<N; i++) {
Unused anbool ok;
// -project the stars around crval
ok = star_coords(starxyz + i*3, crxyz, TRUE, p + 2*i, p + 2*i + 1);
assert(ok);
}
} else {
// -get the star center-of-mass (this will become the tangent point CRVAL)
for (i=0; i<N; i++) {
w = (weights ? weights[i] : 1.0);
star_cm[0] += w * starxyz[i*3 + 0];
star_cm[1] += w * starxyz[i*3 + 1];
star_cm[2] += w * starxyz[i*3 + 2];
}
normalize_3(star_cm);
// -project the stars around their center of mass
for (i=0; i<N; i++) {
Unused anbool ok;
ok = star_coords(starxyz + i*3, star_cm, TRUE, p + 2*i, p + 2*i + 1);
assert(ok);
}
}
// -compute the center of mass of the projected stars and subtract it out.
for (i=0; i<N; i++) {
w = (weights ? weights[i] : 1.0);
pcm[0] += w * p[2*i + 0];
pcm[1] += w * p[2*i + 1];
}
pcm[0] /= totalw;
pcm[1] /= totalw;
for (i=0; i<N; i++) {
p[2*i + 0] -= pcm[0];
p[2*i + 1] -= pcm[1];
}
// -compute the covariance between field positions and projected
// positions of the corresponding stars.
for (i=0; i<N; i++)
for (j=0; j<2; j++)
for (k=0; k<2; k++)
cov[j*2 + k] += p[i*2 + k] * f[i*2 + j];
for (i=0; i<4; i++)
assert(isfinite(cov[i]));
// -run SVD
V = gsl_matrix_alloc(2, 2);
S = gsl_vector_alloc(2);
work = gsl_vector_alloc(2);
vcov = gsl_matrix_view_array(cov, 2, 2);
vR = gsl_matrix_view_array(R, 2, 2);
A = &(vcov.matrix);
// The Jacobi version doesn't always compute an orthonormal U if S has zeros.
//gsl_linalg_SV_decomp_jacobi(A, V, S);
gsl_linalg_SV_decomp(A, V, S, work);
// the U result is written to A.
U = A;
gsl_vector_free(S);
gsl_vector_free(work);
// R = V U'
gsl_blas_dgemm(CblasNoTrans, CblasTrans, 1.0, V, U, 0.0, &(vR.matrix));
gsl_matrix_free(V);
for (i=0; i<4; i++)
assert(isfinite(R[i]));
// -compute scale: make the variances equal.
{
double pvar, fvar;
pvar = fvar = 0.0;
for (i=0; i<N; i++) {
w = (weights ? weights[i] : 1.0);
for (j=0; j<2; j++) {
pvar += w * square(p[i*2 + j]);
fvar += w * square(f[i*2 + j]);
}
}
scale = sqrt(pvar / fvar);
}
// -compute WCS parameters.
scale = rad2deg(scale);
tanout->cd[0][0] = R[0] * scale; // CD1_1
tanout->cd[0][1] = R[1] * scale; // CD1_2
tanout->cd[1][0] = R[2] * scale; // CD2_1
tanout->cd[1][1] = R[3] * scale; // CD2_2
assert(isfinite(tanout->cd[0][0]));
assert(isfinite(tanout->cd[0][1]));
assert(isfinite(tanout->cd[1][0]));
assert(isfinite(tanout->cd[1][1]));
if (tanin) {
// CRPIX is fixed.
tanout->crpix[0] = crpix[0];
tanout->crpix[1] = crpix[1];
// Set CRVAL temporarily...
tan_pixelxy2radec(tanin, crpix[0], crpix[1],
tanout->crval+0, tanout->crval+1);
// Shift CRVAL so that the center of the quad is in the right place.
{
double ix,iy;
double dx,dy;
double dxyz[3];
tan_pixelxy2iwc(tanout, field_cm[0], field_cm[1], &ix, &iy);
dx = rad2deg(pcm[0]) - ix;
dy = rad2deg(pcm[1]) - iy;
tan_iwc2xyzarr(tanout, dx, dy, dxyz);
xyzarr2radecdeg(dxyz, tanout->crval + 0, tanout->crval + 1);
}
} else {
tanout->crpix[0] = field_cm[0];
tanout->crpix[1] = field_cm[1];
xyzarr2radecdegarr(star_cm, tanout->crval);
// FIXME -- we ignore pcm. It should get added back in (after
// multiplication by CD in the appropriate units) to either crval or
// crpix. It's a very small correction probably of the same size
// as the other approximations we're making.
}
if (p_scale) *p_scale = scale;
free(p);
free(f);
return 0;
}
int main(int argc, char** args) {
int argchar;
char* outfn = NULL;
char* fn;
rdlist_t* rdls;
startree_t* skdt = NULL;
int i;
int loglvl = LOG_MSG;
while ((argchar = getopt (argc, args, OPTIONS)) != -1)
switch (argchar) {
case 'v':
loglvl++;
break;
case 'r':
outfn = optarg;
break;
case 'h':
print_help(args[0]);
exit(0);
}
log_init(loglvl);
if (!outfn || (optind == argc)) {
print_help(args[0]);
exit(-1);
}
rdls = rdlist_open_for_writing(outfn);
if (!rdls) {
ERROR("Failed to open RDLS file %s for output", outfn);
exit(-1);
}
if (rdlist_write_primary_header(rdls)) {
ERROR("Failed to write RDLS header");
exit(-1);
}
for (; optind<argc; optind++) {
int Nstars;
fn = args[optind];
logmsg("Opening star kdtree %s...\n", fn);
skdt = startree_open(fn);
if (!skdt) {
ERROR("Failed to read star kdtree %s", fn);
exit(-1);
}
Nstars = startree_N(skdt);
if (rdlist_write_header(rdls)) {
ERROR("Failed to write new RDLS field header");
exit(-1);
}
logmsg("Reading stars...\n");
for (i=0; i<Nstars; i++) {
double xyz[3];
double radec[2];
if (!(i % 200000)) {
printf(".");
fflush(stdout);
}
startree_get(skdt, i, xyz);
xyzarr2radecdegarr(xyz, radec);
if (rdlist_write_one_radec(rdls, radec[0], radec[1])) {
ERROR("Failed to write a RA,Dec entry");
exit(-1);
}
}
printf("\n");
startree_close(skdt);
if (rdlist_fix_header(rdls)) {
ERROR("Failed to fix RDLS field header");
exit(-1);
}
}
if (rdlist_fix_primary_header(rdls) ||
rdlist_close(rdls)) {
ERROR("Failed to close RDLS file");
exit(-1);
}
return 0;
}
int main(int argc, char** args) {
int argchar;
char* basename;
char* outfn = NULL;
index_t* index;
quadfile* qf;
rdlist_t* rdls;
startree_t* skdt = NULL;
anbool addradius = FALSE;
int i;
int radcolumn = -1;
while ((argchar = getopt (argc, args, OPTIONS)) != -1)
switch (argchar) {
case 'R':
addradius = TRUE;
break;
case 'r':
outfn = optarg;
break;
case 'h':
print_help(args[0]);
exit(0);
}
if (!outfn || (optind == argc)) {
print_help(args[0]);
exit(-1);
}
rdls = rdlist_open_for_writing(outfn);
if (!rdls) {
fprintf(stderr, "Failed to open RDLS file %s for output.\n", outfn);
exit(-1);
}
if (rdlist_write_primary_header(rdls)) {
fprintf(stderr, "Failed to write RDLS header.\n");
exit(-1);
}
if (addradius) {
radcolumn = rdlist_add_tagalong_column(rdls, fitscolumn_double_type(),
1, fitscolumn_double_type(),
"QUADRADIUS", "deg");
}
for (; optind<argc; optind++) {
//int Nstars;
int dimquads;
basename = args[optind];
printf("Reading files with basename %s\n", basename);
index = index_load(basename, 0, NULL);
if (!index) {
fprintf(stderr, "Failed to read index with base name \"%s\"\n", basename);
exit(-1);
}
qf = index->quads;
skdt = index->starkd;
//Nstars = startree_N(skdt);
if (rdlist_write_header(rdls)) {
fprintf(stderr, "Failed to write new RDLS field header.\n");
exit(-1);
}
dimquads = quadfile_dimquads(qf);
printf("Reading quads...\n");
for (i=0; i<qf->numquads; i++) {
unsigned int stars[dimquads];
double axyz[3], bxyz[3];
double midab[3];
double radec[2];
if (!(i % 200000)) {
printf(".");
fflush(stdout);
}
quadfile_get_stars(qf, i, stars);
startree_get(skdt, stars[0], axyz);
startree_get(skdt, stars[1], bxyz);
star_midpoint(midab, axyz, bxyz);
xyzarr2radecdegarr(midab, radec);
if (rdlist_write_one_radec(rdls, radec[0], radec[1])) {
fprintf(stderr, "Failed to write a RA,Dec entry.\n");
exit(-1);
}
if (addradius) {
double rad = arcsec2deg(distsq2arcsec(distsq(midab, axyz, 3)));
if (rdlist_write_tagalong_column(rdls, radcolumn, i, 1, &rad, 0)) {
fprintf(stderr, "Failed to write quad radius.\n");
exit(-1);
}
}
}
printf("\n");
index_close(index);
if (rdlist_fix_header(rdls)) {
fprintf(stderr, "Failed to fix RDLS field header.\n");
exit(-1);
}
rdlist_next_field(rdls);
}
if (rdlist_fix_primary_header(rdls) ||
rdlist_close(rdls)) {
fprintf(stderr, "Failed to close RDLS file.\n");
exit(-1);
}
return 0;
}
