static starxy_t* field1() {
starxy_t* starxy;
double field[14];
int i=0, N;
// star0 A: (0,0)
field[i++] = 0.0;
field[i++] = 0.0;
// star1 B: (2,2)
field[i++] = 2.0;
field[i++] = 2.0;
// star2
field[i++] = -1.0;
field[i++] = 3.0;
// star3
field[i++] = 0.5;
field[i++] = 1.5;
// star4
field[i++] = 1.0;
field[i++] = 1.0;
// star5
field[i++] = 1.5;
field[i++] = 0.5;
// star6
field[i++] = 3.0;
field[i++] = -1.0;
N = i/2;
starxy = starxy_new(N, FALSE, FALSE);
for (i=0; i<N; i++) {
starxy_setx(starxy, i, field[i*2+0]);
starxy_sety(starxy, i, field[i*2+1]);
}
return starxy;
}
void test_sorting(CuTest* tc) {
double flux[] = { 50, 100, 50, 100, 20, 20, 40, 40 };
double bg[] = { 0, 10, 10, 0, 10, 0, 5, 0 };
int trueorder[] = { 4, 5, 7, 6, 2, 0, 1, 3 };
int i, N;
starxy_t* s;
char* infn = "/tmp/test-resort-xylist";
char* outfn = "/tmp/test-resort-xylist-out";
xylist_t* xy;
xylist_t* xy2;
starxy_t* s2;
xy = xylist_open_for_writing(infn);
CuAssertPtrNotNull(tc, xy);
xylist_set_include_flux(xy, TRUE);
xylist_set_include_background(xy, TRUE);
if (xylist_write_primary_header(xy) ||
xylist_write_header(xy)) {
CuFail(tc, "write header");
}
N = sizeof(flux) / sizeof(double);
s = starxy_new(N, TRUE, TRUE);
for (i=0; i<N; i++) {
starxy_setx(s, i, random()%1000);
starxy_sety(s, i, random()%1000);
}
starxy_set_flux_array(s, flux);
starxy_set_bg_array(s, bg);
if (xylist_write_field(xy, s) ||
xylist_fix_header(xy) ||
xylist_fix_primary_header(xy) ||
xylist_close(xy)) {
CuFail(tc, "close xy");
}
CuAssertIntEquals(tc, 0,
resort_xylist(infn, outfn, NULL, NULL, TRUE));
xy2 = xylist_open(outfn);
s2 = xylist_read_field(xy2, NULL);
CuAssertPtrNotNull(tc, s2);
CuAssertPtrNotNull(tc, s2->x);
CuAssertPtrNotNull(tc, s2->y);
CuAssertPtrNotNull(tc, s2->flux);
CuAssertPtrNotNull(tc, s2->background);
for (i=0; i<N; i++) {
CuAssertDblEquals(tc, s->x[trueorder[i]], s2->x[i], 1e-6);
CuAssertDblEquals(tc, s->y[trueorder[i]], s2->y[i], 1e-6);
CuAssertDblEquals(tc, s->flux[trueorder[i]], s2->flux[i], 1e-6);
CuAssertDblEquals(tc, s->background[trueorder[i]], s2->background[i], 1e-6);
}
}
int main(int argc, char** args) {
int c;
dl* xys = dl_new(16);
dl* radecs = dl_new(16);
dl* otherradecs = dl_new(16);
double* xy;
double* xyz;
int i, N;
tan_t tan, tan2, tan3;
int W=0, H=0;
double crpix[] = { HUGE_VAL, HUGE_VAL };
int loglvl = LOG_MSG;
FILE* logstream = stderr;
int order = 1;
while ((c = getopt(argc, args, OPTIONS)) != -1) {
switch (c) {
case 'v':
loglvl++;
break;
case 'h':
exit(0);
case 'o':
order = atoi(optarg);
break;
case 'W':
W = atoi(optarg);
break;
case 'H':
H = atoi(optarg);
break;
case 'X':
crpix[0] = atof(optarg);
break;
case 'Y':
crpix[1] = atof(optarg);
break;
}
}
if (optind != argc) {
exit(-1);
}
log_init(loglvl);
log_to(logstream);
errors_log_to(logstream);
if (W == 0 || H == 0) {
logerr("Need -W, -H\n");
exit(-1);
}
if (crpix[0] == HUGE_VAL)
crpix[0] = W/2.0;
if (crpix[1] == HUGE_VAL)
crpix[1] = H/2.0;
while (1) {
double x,y,ra,dec;
if (fscanf(stdin, "%lf %lf %lf %lf\n", &x, &y, &ra, &dec) < 4)
break;
if (x == -1 && y == -1) {
dl_append(otherradecs, ra);
dl_append(otherradecs, dec);
} else {
dl_append(xys, x);
dl_append(xys, y);
dl_append(radecs, ra);
dl_append(radecs, dec);
}
}
logmsg("Read %i x,y,ra,dec tuples\n", dl_size(xys)/2);
N = dl_size(xys)/2;
xy = dl_to_array(xys);
xyz = malloc(3 * N * sizeof(double));
for (i=0; i<N; i++)
radecdeg2xyzarr(dl_get(radecs, 2*i), dl_get(radecs, 2*i+1), xyz + i*3);
dl_free(xys);
dl_free(radecs);
fit_tan_wcs(xyz, xy, N, &tan, NULL);
tan.imagew = W;
tan.imageh = H;
logmsg("Computed TAN WCS:\n");
tan_print_to(&tan, logstream);
sip_t* sip;
{
tweak_t* t = tweak_new();
starxy_t* sxy = starxy_new(N, FALSE, FALSE);
il* imginds = il_new(256);
il* refinds = il_new(256);
for (i=0; i<N; i++) {
starxy_set_x(sxy, i, xy[2*i+0]);
starxy_set_y(sxy, i, xy[2*i+1]);
}
tweak_init(t);
tweak_push_ref_xyz(t, xyz, N);
tweak_push_image_xy(t, sxy);
for (i=0; i<N; i++) {
il_append(imginds, i);
il_append(refinds, i);
}
// unweighted; no dist2s
tweak_push_correspondence_indices(t, imginds, refinds, NULL, NULL);
tweak_push_wcs_tan(t, &tan);
t->sip->a_order = t->sip->b_order = t->sip->ap_order = t->sip->bp_order = order;
for (i=0; i<10; i++) {
// go to TWEAK_HAS_LINEAR_CD -> do_sip_tweak
// t->image has the indices of corresponding image stars
// t->ref has the indices of corresponding catalog stars
tweak_go_to(t, TWEAK_HAS_LINEAR_CD);
logmsg("\n");
sip_print(t->sip);
t->state &= ~TWEAK_HAS_LINEAR_CD;
}
tan_write_to_file(&t->sip->wcstan, "kt1.wcs");
sip = t->sip;
}
for (i=0; i<dl_size(otherradecs)/2; i++) {
double ra, dec, x,y;
ra = dl_get(otherradecs, 2*i);
dec = dl_get(otherradecs, 2*i+1);
if (!sip_radec2pixelxy(sip, ra, dec, &x, &y)) {
logerr("Not in tangent plane: %g,%g\n", ra, dec);
exit(-1);
//continue;
}
printf("%g %g\n", x, y);
}
/*
blind_wcs_move_tangent_point(xyz, xy, N, crpix, &tan, &tan2);
blind_wcs_move_tangent_point(xyz, xy, N, crpix, &tan2, &tan3);
logmsg("Moved tangent point to (%g,%g):\n", crpix[0], crpix[1]);
tan_print_to(&tan3, logstream);
tan_write_to_file(&tan, "kt1.wcs");
tan_write_to_file(&tan3, "kt2.wcs");
*/
dl_free(otherradecs);
free(xy);
free(xyz);
return 0;
}
static sip_t* run_test(CuTest* tc, sip_t* sip, int N, double* xy, double* radec) {
int i;
starxy_t* sxy;
tweak_t* t;
sip_t* outsip;
il* imcorr;
il* refcorr;
dl* weights;
tan_t* tan = &(sip->wcstan);
printf("Input SIP:\n");
sip_print_to(sip, stdout);
fflush(NULL);
sxy = starxy_new(N, FALSE, FALSE);
starxy_set_xy_array(sxy, xy);
imcorr = il_new(256);
refcorr = il_new(256);
weights = dl_new(256);
for (i=0; i<N; i++) {
il_append(imcorr, i);
il_append(refcorr, i);
dl_append(weights, 1.0);
}
t = tweak_new();
tweak_push_wcs_tan(t, tan);
outsip = t->sip;
outsip->a_order = outsip->b_order = sip->a_order;
outsip->ap_order = outsip->bp_order = sip->ap_order;
t->weighted_fit = TRUE;
tweak_push_ref_ad_array(t, radec, N);
tweak_push_image_xy(t, sxy);
tweak_push_correspondence_indices(t, imcorr, refcorr, NULL, weights);
tweak_skip_shift(t);
// push correspondences
// push image xy
// push ref ra,dec
// push ref xy (tan)
// push tan
tweak_go_to(t, TWEAK_HAS_LINEAR_CD);
printf("Output SIP:\n");
sip_print_to(outsip, stdout);
CuAssertDblEquals(tc, tan->imagew, outsip->wcstan.imagew, 1e-10);
CuAssertDblEquals(tc, tan->imageh, outsip->wcstan.imageh, 1e-10);
// should be exactly equal.
CuAssertDblEquals(tc, tan->crpix[0], outsip->wcstan.crpix[0], 1e-10);
CuAssertDblEquals(tc, tan->crpix[1], outsip->wcstan.crpix[1], 1e-10);
t->sip = NULL;
tweak_free(t);
starxy_free(sxy);
return outsip;
}