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 plot_xy_plot(const char* command, cairo_t* cairo,
plot_args_t* pargs, void* baton) {
plotxy_t* args = (plotxy_t*)baton;
// Plot it!
xylist_t* xyls;
starxy_t myxy;
starxy_t* xy = NULL;
starxy_t* freexy = NULL;
int Nxy;
int i;
#if 0
double t0;
#endif
plotstuff_builtin_apply(cairo, pargs);
if (args->fn && dl_size(args->xyvals)) {
ERROR("Can only plot one of xylist filename and xy_vals");
return -1;
}
if (!args->fn && !dl_size(args->xyvals)) {
ERROR("Neither xylist filename nor xy_vals given!");
return -1;
}
if (args->fn) {
#if 0
t0 = timenow();
#endif
// Open xylist.
xyls = xylist_open(args->fn);
if (!xyls) {
ERROR("Failed to open xylist from file \"%s\"", args->fn);
return -1;
}
// we don't care about FLUX and BACKGROUND columns.
xylist_set_include_flux(xyls, FALSE);
xylist_set_include_background(xyls, FALSE);
if (args->xcol)
xylist_set_xname(xyls, args->xcol);
if (args->ycol)
xylist_set_yname(xyls, args->ycol);
// Find number of entries in xylist.
xy = xylist_read_field_num(xyls, args->ext, NULL);
freexy = xy;
xylist_close(xyls);
if (!xy) {
ERROR("Failed to read FITS extension %i from file %s.\n", args->ext, args->fn);
return -1;
}
Nxy = starxy_n(xy);
// If N is specified, apply it as a max.
if (args->nobjs)
Nxy = MIN(Nxy, args->nobjs);
//logmsg("%g s to read xylist\n", timenow()-t0);
} else {
assert(dl_size(args->xyvals));
starxy_from_dl(&myxy, args->xyvals, FALSE, FALSE);
xy = &myxy;
Nxy = starxy_n(xy);
}
// Transform through WCSes.
if (args->wcs) {
double ra, dec, x, y;
assert(pargs->wcs);
/*
// check for any overlap.
double pralo,prahi,pdeclo,pdechi;
double ralo,rahi,declo,dechi;
anwcs_get_radec_bounds(pargs->wcs, 100, &pralo, &prahi, &pdeclo, &pdechi);
anwcs_get_radec_bounds(args->wcs, 100, &ralo, &rahi, &declo, &dechi);
if (
*/
for (i=0; i<Nxy; i++) {
anwcs_pixelxy2radec(args->wcs,
// I used to add 1 here
starxy_getx(xy, i), starxy_gety(xy, i),
&ra, &dec);
if (!plotstuff_radec2xy(pargs, ra, dec, &x, &y))
continue;
logverb(" xy (%g,%g) -> RA,Dec (%g,%g) -> plot xy (%g,%g)\n",
starxy_getx(xy,i), starxy_gety(xy,i), ra, dec, x, y);
// add shift and scale...
// FIXME -- not clear that we want to do this here...
/*
starxy_setx(xy, i, args->scale * (x - args->xoff));
starxy_sety(xy, i, args->scale * (y - args->yoff));
starxy_setx(xy, i, x-1);
starxy_sety(xy, i, y-1);
*/
// Output coords: FITS -> 0-indexed image
starxy_setx(xy, i, x-1);
starxy_sety(xy, i, y-1);
}
} else {
// Shift and scale xylist entries.
if (args->xoff != 0.0 || args->yoff != 0.0) {
for (i=0; i<Nxy; i++) {
starxy_setx(xy, i, starxy_getx(xy, i) - args->xoff);
starxy_sety(xy, i, starxy_gety(xy, i) - args->yoff);
}
}
if (args->scale != 1.0) {
for (i=0; i<Nxy; i++) {
starxy_setx(xy, i, args->scale * starxy_getx(xy, i));
starxy_sety(xy, i, args->scale * starxy_gety(xy, i));
}
}
}
// Plot markers.
#if 0
t0 = timenow();
#endif
for (i=args->firstobj; i<Nxy; i++) {
double x = starxy_getx(xy, i);
double y = starxy_gety(xy, i);
if (plotstuff_marker_in_bounds(pargs, x, y))
plotstuff_stack_marker(pargs, x, y);
}
plotstuff_plot_stack(pargs, cairo);
//logmsg("%g s to plot xylist\n", timenow()-t0);
starxy_free(freexy);
return 0;
}
