int main(int argc, char** args) {
int c;
char* xylsfn = NULL;
char* rdlsfn = NULL;
char* corrfn = NULL;
char* outfn = NULL;
char* xcol = NULL;
char* ycol = NULL;
char* rcol = NULL;
char* dcol = NULL;
xylist_t* xyls = NULL;
rdlist_t* rdls = NULL;
rd_t rd;
starxy_t xy;
int fieldnum = 1;
int N;
double* fieldxy = NULL;
double* xyz = NULL;
sip_t wcs;
int rtn = -1;
int loglvl = LOG_MSG;
int siporder = 0;
int W=0, H=0;
anbool crpix_center = FALSE;
int i;
int doshift = 1;
fits_use_error_system();
while ((c = getopt(argc, args, OPTIONS)) != -1) {
switch (c) {
case 'h':
print_help(args[0]);
exit(0);
case 'W':
W = atoi(optarg);
break;
case 'H':
H = atoi(optarg);
break;
case 'C':
crpix_center = TRUE;
break;
case 's':
siporder = atoi(optarg);
break;
case 'c':
corrfn = optarg;
break;
case 'r':
rdlsfn = optarg;
break;
case 'R':
rcol = optarg;
break;
case 'D':
dcol = optarg;
break;
case 'x':
xylsfn = optarg;
break;
case 'X':
xcol = optarg;
break;
case 'Y':
ycol = optarg;
break;
case 'o':
outfn = optarg;
break;
case 'v':
loglvl++;
break;
}
}
if (optind != argc) {
print_help(args[0]);
exit(-1);
}
if (! ((xylsfn && rdlsfn) || corrfn) || !outfn) {
print_help(args[0]);
exit(-1);
}
log_init(loglvl);
if (corrfn) {
xylsfn = corrfn;
rdlsfn = corrfn;
if (!xcol)
xcol = "FIELD_X";
if (!ycol)
ycol = "FIELD_Y";
if (!rcol)
rcol = "INDEX_RA";
if (!dcol)
dcol = "INDEX_DEC";
}
// read XYLS.
xyls = xylist_open(xylsfn);
if (!xyls) {
ERROR("Failed to read an xylist from file %s", xylsfn);
goto bailout;
}
xylist_set_include_flux(xyls, FALSE);
xylist_set_include_background(xyls, FALSE);
if (xcol)
xylist_set_xname(xyls, xcol);
if (ycol)
xylist_set_yname(xyls, ycol);
// read RDLS.
rdls = rdlist_open(rdlsfn);
if (!rdls) {
ERROR("Failed to read an RA,Dec list from file %s", rdlsfn);
goto bailout;
}
if (rcol)
rdlist_set_raname(rdls, rcol);
if (dcol)
rdlist_set_decname(rdls, dcol);
if (!xylist_read_field_num(xyls, fieldnum, &xy)) {
ERROR("Failed to read xyls file %s, field %i", xylsfn, fieldnum);
goto bailout;
}
if (!rdlist_read_field_num(rdls, fieldnum, &rd)) {
ERROR("Failed to read rdls field %i", fieldnum);
goto bailout;
}
N = starxy_n(&xy);
if (rd_n(&rd) != N) {
ERROR("X,Y list and RA,Dec list must have the same number of entries, "
"but found %i vs %i", N, rd_n(&rd));
goto bailout;
}
logverb("Read %i points from %s and %s\n", N, rdlsfn, xylsfn);
xyz = (double*)malloc(sizeof(double) * 3 * N);
if (!xyz) {
ERROR("Failed to allocate %i xyz coords", N);
goto bailout;
}
radecdeg2xyzarrmany(rd.ra, rd.dec, xyz, N);
fieldxy = starxy_to_xy_array(&xy, NULL);
if (!fieldxy) {
ERROR("Failed to allocate %i xy coords", N);
goto bailout;
}
logverb("Fitting WCS\n");
if (siporder == 0) {
if (fit_tan_wcs(xyz, fieldxy, N, &(wcs.wcstan), NULL)) {
ERROR("Failed to fit for TAN WCS");
goto bailout;
}
} else {
if (W == 0) {
for (i=0; i<N; i++) {
W = MAX(W, (int)ceil(fieldxy[2*i + 0]));
}
}
if (H == 0) {
for (i=0; i<N; i++) {
H = MAX(H, (int)ceil(fieldxy[2*i + 1]));
}
}
logverb("Image size = %i x %i pix\n", W, H);
fit_sip_wcs_2(xyz, fieldxy, NULL, N,
siporder, siporder+1, W, H,
crpix_center, NULL, doshift, &wcs);
}
if (siporder <= 1) {
if (tan_write_to_file(&(wcs.wcstan), outfn)) {
ERROR("Failed to write TAN WCS header to file \"%s\"", outfn);
goto bailout;
}
} else {
if (sip_write_to_file(&wcs, outfn)) {
ERROR("Failed to write SIP WCS header to file \"%s\"", outfn);
goto bailout;
}
}
logverb("Wrote WCS to %s\n", outfn);
starxy_free_data(&xy);
rd_free_data(&rd);
rtn = 0;
bailout:
if (rdls)
rdlist_close(rdls);
if (xyls)
xylist_close(xyls);
if (fieldxy)
free(fieldxy);
if (xyz)
free(xyz);
return rtn;
}
startree_t* startree_build(fitstable_t* intable,
const char* racol, const char* deccol,
// keep RA,Dec in the tag-along table?
//anbool keep_radec,
// KDT_DATA_*, KDT_TREE_*
int datatype, int treetype,
// KD_BUILD_*
int buildopts,
int Nleaf,
char** args, int argc) {
double* ra = NULL;
double* dec = NULL;
double* xyz = NULL;
int N;
startree_t* starkd = NULL;
int tt;
int d;
double low[3];
double high[3];
qfits_header* hdr;
qfits_header* inhdr;
int i;
if (!racol)
racol = "RA";
if (!deccol)
deccol = "DEC";
if (!datatype)
datatype = KDT_DATA_U32;
if (!treetype)
treetype = KDT_TREE_U32;
if (!buildopts)
buildopts = KD_BUILD_SPLIT;
if (!Nleaf)
Nleaf = 25;
ra = fitstable_read_column(intable, racol, TFITS_BIN_TYPE_D);
if (!ra) {
ERROR("Failed to read RA from column %s", racol);
goto bailout;
}
dec = fitstable_read_column(intable, deccol, TFITS_BIN_TYPE_D);
if (!dec) {
ERROR("Failed to read RA from column %s", racol);
goto bailout;
}
N = fitstable_nrows(intable);
xyz = malloc(N * 3 * sizeof(double));
if (!xyz) {
SYSERROR("Failed to malloc xyz array to build startree");
goto bailout;
}
radecdeg2xyzarrmany(ra, dec, xyz, N);
free(ra);
ra = NULL;
free(dec);
dec = NULL;
starkd = startree_new();
if (!starkd) {
ERROR("Failed to allocate startree");
goto bailout;
}
tt = kdtree_kdtypes_to_treetype(KDT_EXT_DOUBLE, treetype, datatype);
starkd->tree = kdtree_new(N, 3, Nleaf);
for (d=0; d<3; d++) {
low[d] = -1.0;
high[d] = 1.0;
}
kdtree_set_limits(starkd->tree, low, high);
logverb("Building star kdtree...\n");
starkd->tree = kdtree_build(starkd->tree, xyz, N, 3, Nleaf, tt, buildopts);
if (!starkd->tree) {
ERROR("Failed to build star kdtree");
startree_close(starkd);
starkd = NULL;
goto bailout;
}
starkd->tree->name = strdup(STARTREE_NAME);
inhdr = fitstable_get_primary_header(intable);
hdr = startree_header(starkd);
an_fits_copy_header(inhdr, hdr, "HEALPIX");
an_fits_copy_header(inhdr, hdr, "HPNSIDE");
an_fits_copy_header(inhdr, hdr, "ALLSKY");
an_fits_copy_header(inhdr, hdr, "JITTER");
an_fits_copy_header(inhdr, hdr, "CUTNSIDE");
an_fits_copy_header(inhdr, hdr, "CUTMARG");
an_fits_copy_header(inhdr, hdr, "CUTDEDUP");
an_fits_copy_header(inhdr, hdr, "CUTNSWEP");
//fits_copy_header(inhdr, hdr, "CUTBAND");
//fits_copy_header(inhdr, hdr, "CUTMINMG");
//fits_copy_header(inhdr, hdr, "CUTMAXMG");
BOILERPLATE_ADD_FITS_HEADERS(hdr);
qfits_header_add(hdr, "HISTORY", "This file was created by the command-line:", NULL, NULL);
fits_add_args(hdr, args, argc);
qfits_header_add(hdr, "HISTORY", "(end of command line)", NULL, NULL);
qfits_header_add(hdr, "HISTORY", "** History entries copied from the input file:", NULL, NULL);
fits_copy_all_headers(inhdr, hdr, "HISTORY");
qfits_header_add(hdr, "HISTORY", "** End of history entries.", NULL, NULL);
for (i=1;; i++) {
char key[16];
int n;
sprintf(key, "SWEEP%i", i);
n = qfits_header_getint(inhdr, key, -1);
if (n == -1)
break;
an_fits_copy_header(inhdr, hdr, key);
}
bailout:
if (ra)
free(ra);
if (dec)
free(dec);
if (xyz)
free(xyz);
return starkd;
}
