astrophotography/pipeline/astrometry.py
laurence 3c4ae5c77e Solve by pointing where the header allows it, and refuse doubtful solutions
Three fixes and one refusal, all found by working on the wide-field
narrowband session that would not solve.

The detection floor was a fixed 12 pixels, which quietly assumed a well
sampled star. At 0.53 arcsec/px with 5.9 px seeing that is right; at 3.5
arcsec/px the stars are undersampled at 1.75 px FWHM and cover a handful
of pixels each, so the floor discarded nearly every real star and kept
blends and galaxies instead. It now scales with the measured seeing, and
that session went from 305 usable detections to 600.

Matching now happens over the frame's inscribed circle rather than a cone
reaching its corners. On a 4 degree field the old cone covered 33 square
degrees of sky against 16 of image, so half the catalogue was not in the
picture at all.

Where the header records a roll angle - and iTelescope's does - the
orientation no longer has to be recovered from scratch. Rotation, scale
and parity are applied directly and only the residual pointing error is
searched, by histogramming every detection-to-catalogue offset and taking
the peak. Asterism matching remains as the fallback for headers that say
nothing about orientation.

The refusal matters most. With catalogue depth slices added, the wide
field produced a "solution" of 25 stars at 2.50 px residual, claiming a
centre half a degree from the pointing. A genuine solve on that field
matches hundreds of stars and refines to well under a pixel. Accepting it
would have silently corrupted every position in the science catalogue, so
a solution must now reach 40 stars AND 1.5 px or it is discarded and the
session is reported as unsolved. A wrong WCS is worse than no WCS.

NGC 2030 improves from 0.31 to 0.12 arcsec on 123 stars under the
pointing-based match. NGC 2070 is honestly unsolved.
2026-07-21 22:41:36 +01:00

422 lines
18 KiB
Python

"""Stage 4: plate solve the deepest master and copy the WCS to the others.
iTelescope's frames arrive with a PinPoint HISTORY line but no WCS keywords at
all, so the astrometry has to be redone locally. A blind solve is unnecessary:
the header gives the pointing to arcminutes and usually the plate scale to four
figures, so a catalogue can be fetched for the right patch of sky and matched
directly.
The match is asterism-based, which is invariant to rotation and scale - so the
camera angle never has to be guessed, and a session at an unknown orientation
solves as easily as one at a known angle. It is NOT invariant to a mirror flip,
because a similarity transform cannot include reflection, so both parities are
tried and the one that matches wins.
The seed match returns only a handful of stars. That is enough to establish an
approximate solution but too few to trust, so the solution is then used to pair
every catalogue source with its nearest detection and refit, twice, with a
shrinking tolerance. The residual that comes out of that is the honest measure
of whether the solve is real.
Failure here is not fatal to the pipeline: an unsolved session still produces
images, it just cannot produce positions.
"""
import os
import astroalign as aa
import numpy as np
from astropy import units as u
from astropy.coordinates import SkyCoord
from astropy.io import fits
from astropy.wcs import WCS
from astropy.wcs.utils import fit_wcs_from_points
import layout
import measure as measure_mod
SEED_STARS = 600
CATALOGUE_FETCH = 6000 # rich fields need far more than the match uses
def pointing(header):
"""Approximate field centre from whatever the header offers."""
for ra_key, dec_key in (("OBJCTRA", "OBJCTDEC"), ("RA", "DEC"),
("CRVAL1", "CRVAL2")):
ra, dec = header.get(ra_key), header.get(dec_key)
if ra is None or dec is None:
continue
try:
if isinstance(ra, str) and (":" in ra or " " in ra):
return SkyCoord(ra, dec, unit=(u.hourangle, u.deg))
return SkyCoord(float(ra) * u.deg, float(dec) * u.deg)
except Exception: # noqa: BLE001
continue
return None
def plate_scale(header, session=None):
"""Arcsec per pixel: from the header if present, else from the optics."""
val = header.get("HIERARCH iTelescopePlateScaleH")
if val:
try:
return float(val)
except (TypeError, ValueError):
pass
focal = header.get("FOCALLEN")
pixel = header.get("XPIXSZ")
if focal and pixel:
# 206265 arcsec per radian; XPIXSZ is already binned microns.
return 206.265 * float(pixel) / float(focal)
return session.scale if session else None
def fetch_catalogue(centre, radius_deg, cache, limit=SEED_STARS, deep=False):
"""Gaia sources around the pointing, cached so a re-run costs nothing."""
if os.path.exists(cache):
z = np.load(cache)
return z["ra"], z["dec"], z["g"]
ra = dec = g = None
mag_cut = 20.5 if deep else 18.0
# VizieR first, deliberately. It serves the identical DR3 catalogue over a
# plain HTTP query and answers in under a second. Gaia's own archive is
# excellent but launch_job_async submits to a job QUEUE and polls for the
# result, which for a few hundred rows means waiting minutes for work that
# takes milliseconds - and it hung for ten minutes on a query of 600 stars
# while the network was demonstrably healthy.
try:
from astroquery.vizier import Vizier
# Ask for far more rows than are needed and pick the brightest
# locally. VizieR's row_limit truncates the result BEFORE any sort, so
# asking for 600 rows in a crowded field returns 600 arbitrary stars
# rather than the 600 brightest - and asterism matching only works if
# both lists contain the same bright stars. In the LMC field of
# NGC 2030 that silently produced no match at all.
v = Vizier(columns=["RA_ICRS", "DE_ICRS", "Gmag"],
column_filters={"Gmag": f"<{mag_cut}"}, row_limit=50000)
v.TIMEOUT = 60
res = v.query_region(centre, radius=radius_deg * u.deg,
catalog="I/355/gaiadr3")
if res:
t = res[0]
ra = np.asarray(t["RA_ICRS"], float)
dec = np.asarray(t["DE_ICRS"], float)
g = np.asarray(t["Gmag"], float)
except Exception as exc: # noqa: BLE001
print(f" VizieR unavailable ({type(exc).__name__})")
if ra is None or len(ra) == 0:
try:
from astroquery.gaia import Gaia
Gaia.ROW_LIMIT = limit
# Synchronous, not async: no job queue for a query this small.
job = Gaia.launch_job(f"""
SELECT TOP {limit} ra, dec, phot_g_mean_mag
FROM gaiadr3.gaia_source
WHERE 1 = CONTAINS(POINT('ICRS', ra, dec),
CIRCLE('ICRS', {centre.ra.deg}, {centre.dec.deg},
{radius_deg}))
AND phot_g_mean_mag IS NOT NULL
AND phot_g_mean_mag < {mag_cut}
ORDER BY phot_g_mean_mag ASC""")
t = job.get_results()
ra = np.asarray(t["ra"], float)
dec = np.asarray(t["dec"], float)
g = np.asarray(t["phot_g_mean_mag"], float)
except Exception as exc: # noqa: BLE001
print(f" Gaia archive unavailable ({type(exc).__name__})")
return None, None, None
order = np.argsort(g)[:limit]
ra, dec, g = ra[order], dec[order], g[order]
np.savez_compressed(cache, ra=ra, dec=dec, g=g)
return ra, dec, g
def project(ra, dec, centre, scale, parity):
"""Gnomonic projection to pixel-like coordinates for asterism matching."""
c = SkyCoord(ra * u.deg, dec * u.deg)
dx, dy = centre.spherical_offsets_to(c)
return np.column_stack([dx.to_value(u.arcsec) / scale * parity,
dy.to_value(u.arcsec) / scale])
def _roll_angle(header):
"""Camera angle from the header, if the telescope recorded it."""
for key in ("HIERARCH RollAngle", "RollAngle", "POSANG", "CROTA2"):
val = header.get(key)
if val is None:
continue
try:
return float(val)
except (TypeError, ValueError):
continue
return None
def match_by_pointing(src, ra, dec, centre, scale, roll, shape, verbose=True):
"""Pair detections with catalogue stars using a known camera angle.
Asterism matching exists to recover an UNKNOWN orientation. When the header
records the roll angle - and iTelescope's does - the whole geometry is
already known bar a small pointing error, and a direct search is both more
robust and far faster. Rotation, scale and parity are applied, then the
residual translation is found by histogramming every detection-to-catalogue
offset and taking the peak: a real solution puts thousands of pairs in one
bin, and noise spreads flat.
"""
ny, nx = shape
# Slide down the catalogue's magnitude ranking as well as trying the four
# geometric conventions. In a field as rich as the LMC the brightest few
# hundred catalogue stars are far brighter than anything a 60 second
# narrowband frame records, so the two "brightest N" lists describe
# different populations and overlap barely at all.
best = None
slices = [(0, 400), (200, 800), (600, 1400), (1200, 2400), (2000, 4000)]
for lo, hi in slices:
if lo >= len(ra):
break
ra_s, dec_s = ra[lo:hi], dec[lo:hi]
for parity in (1.0, -1.0):
for sign in (1.0, -1.0):
th = np.radians(sign * roll)
cat = project(ra_s, dec_s, centre, scale, parity)
rot = np.column_stack([
cat[:, 0] * np.cos(th) - cat[:, 1] * np.sin(th),
cat[:, 0] * np.sin(th) + cat[:, 1] * np.cos(th)])
cat_px = rot + np.array([nx / 2.0, ny / 2.0])
# Every pairwise offset, histogrammed. Limited to the brightest of
# each list to keep this to a few million comparisons.
a = src[:400]
b = cat_px[:400]
dx = (a[:, None, 0] - b[None, :, 0]).ravel()
dy = (a[:, None, 1] - b[None, :, 1]).ravel()
lim = 0.25 * max(nx, ny)
keep = (np.abs(dx) < lim) & (np.abs(dy) < lim)
if keep.sum() < 50:
continue
bins = np.arange(-lim, lim + 12, 12)
H, xe, ye = np.histogram2d(dx[keep], dy[keep], bins=[bins, bins])
iy, ix = np.unravel_index(np.argmax(H), H.shape)
peak = H[iy, ix]
if best is None or peak > best[0]:
off = (0.5 * (xe[iy] + xe[iy + 1]),
0.5 * (ye[ix] + ye[ix + 1]))
best = (peak, parity, sign, off, cat_px, lo, hi)
if best is None:
return None
peak, parity, sign, off, cat_px, lo, hi = best
if verbose:
print(f" pointing match: catalogue[{lo}:{hi}], parity "
f"{parity:+.0f}, roll {sign:+.0f}, offset "
f"({off[0]:+.0f}, {off[1]:+.0f}) px, peak {int(peak)} pairs")
if peak < 12:
return None
shifted = cat_px + np.array(off)
idx_base = lo
pairs = []
for i, (cx, cy) in enumerate(shifted):
if not (0 <= cx < nx and 0 <= cy < ny):
continue
d = np.hypot(src[:, 0] - cx, src[:, 1] - cy)
j = int(np.argmin(d))
if d[j] < 18:
pairs.append((i + idx_base, j))
if len(pairs) < 10:
return None
return np.array([p[0] for p in pairs]), np.array([p[1] for p in pairs])
def run(session, verbose=True):
"""Solve the deepest master; write the WCS into every master."""
masters_dir = os.path.join(session.root, layout.MASTERS)
deepest = session.filters[0]
path = os.path.join(masters_dir, f"master-{deepest}.fit")
if not os.path.exists(path):
raise RuntimeError("no masters found - run the register stage first")
with fits.open(path) as hd:
image = hd[0].data.astype(np.float32)
header = hd[0].header
ny, nx = image.shape
centre = pointing(header)
scale = plate_scale(header, session)
if centre is None or not scale:
print(" no pointing or plate scale in the header - "
"cannot solve without a blind solver")
return None
# Match over the frame's INSCRIBED circle, not its circumscribed one. A
# cone big enough to reach the corners also reaches well outside the
# frame - on a 4 degree field that is 33 square degrees of catalogue
# against 16 of image, so half the catalogue's brightest stars are not in
# the picture at all and the two "brightest N" lists barely overlap.
# The inscribed circle is inside the frame whatever the camera angle, so
# both lists then describe the same patch of sky.
radius = 0.95 * 0.5 * min(nx, ny) * scale / 3600.0
if verbose:
print(f" pointing {centre.to_string('hmsdms')}, "
f"{scale:.4f} arcsec/px, search {radius:.3f} deg")
xy, _ = measure_mod.load(session)
# Detect on the master itself: it is deeper than any single frame.
from measure import measure_frame
det = measure_frame(path, max_stars=SEED_STARS * 3)
src = det["xy"]
# Same restriction on the detections, so the two lists cover one region.
r_pix = 0.95 * 0.5 * min(nx, ny)
inside = np.hypot(src[:, 0] - nx / 2.0, src[:, 1] - ny / 2.0) < r_pix
src = src[inside][:SEED_STARS]
if len(src) < 12:
print(f" only {len(src)} stars in the master - too few")
return None
cache = os.path.join(session.root, layout.INTERMEDIATES, "_gaia.npz")
os.makedirs(os.path.dirname(cache), exist_ok=True)
ra, dec, gmag = fetch_catalogue(centre, radius, cache,
limit=CATALOGUE_FETCH)
if ra is None:
print(" no catalogue available - skipping the solve")
return None
if verbose:
print(f" {len(ra)} catalogue stars, {len(src)} detected")
# If the header recorded the camera angle, use it: a direct geometric
# match is far more reliable than recovering the orientation from scratch.
roll = _roll_angle(header)
seed = None
if roll is not None:
if verbose:
print(f" header roll angle {roll:.2f} deg")
seed = match_by_pointing(src, ra, dec, centre, scale, roll,
(ny, nx), verbose=verbose)
if seed is not None:
ci, ii = seed
world0 = SkyCoord(ra[ci] * u.deg, dec[ci] * u.deg)
wcs = fit_wcs_from_points((src[ii, 0], src[ii, 1]), world0,
proj_point="center", projection="TAN")
best = "pointing"
else:
best = None
# Matching needs the two lists to contain the SAME stars, and "brightest"
# does not guarantee that. The detector deliberately rejects saturated
# cores, so on a well exposed frame the brightest detections are not the
# brightest stars in the sky - while the catalogue's brightest are. The
# two top-N lists can therefore barely overlap, which is exactly what
# happened on the crowded LMC field of NGC 2030.
#
# So slide a window down the catalogue's magnitude ranking as well as
# varying the sample size, and take the best match found.
if best == "pointing":
pass
else:
best = None
for offset in (0, 15, 40, 80):
for n in (120, 200, 60):
hi = min(offset + n, len(ra))
if hi - offset < 25 or len(src) < 25:
continue
cat_slice = slice(offset, hi)
for parity in (1.0, -1.0):
cat_xy = project(ra[cat_slice], dec[cat_slice], centre, scale,
parity)
try:
tform, (cat_m, img_m) = aa.find_transform(
cat_xy, src[:max(n, 120)], max_control_points=60)
except Exception: # noqa: BLE001
continue
if best is None or len(cat_m) > len(best[0]):
# Record which catalogue rows matched, in FULL-list terms.
best = (cat_m, img_m, parity, offset)
if verbose:
print(f" catalogue[{offset}:{hi}] parity "
f"{parity:+.0f}: matched {len(cat_m)} stars")
if best is not None and len(best[0]) >= 8:
break
if best is not None and len(best[0]) >= 8:
break
if best is None:
print(" no match by pointing or asterism - solve failed")
return None
if best != "pointing":
cat_m, img_m, parity, _offset = best
cat_xy = project(ra, dec, centre, scale, parity)
idx = [int(np.argmin(np.hypot(cat_xy[:, 0] - px, cat_xy[:, 1] - py)))
for px, py in cat_m]
world = SkyCoord(ra[idx] * u.deg, dec[idx] * u.deg)
wcs = fit_wcs_from_points((img_m[:, 0], img_m[:, 1]), world,
proj_point="center", projection="TAN")
# Refine: the seed match is a handful of stars, so use the approximate
# solution to pair every catalogue source with its nearest detection.
all_world = SkyCoord(ra * u.deg, dec * u.deg)
nmatch, resid = (len(seed[0]) if seed is not None else len(cat_m)), None
for tol in (4.0, 2.0):
px, py = wcs.world_to_pixel(all_world)
pairs = []
for i, (cx, cy) in enumerate(zip(px, py)):
if not (0 <= cx < nx and 0 <= cy < ny):
continue
d = np.hypot(src[:, 0] - cx, src[:, 1] - cy)
j = int(np.argmin(d))
if d[j] <= tol:
pairs.append((i, j))
if len(pairs) < 20:
break
ci = np.array([p[0] for p in pairs])
ii = np.array([p[1] for p in pairs])
wcs = fit_wcs_from_points((src[ii, 0], src[ii, 1]), all_world[ci],
proj_point="center", projection="TAN")
qx, qy = wcs.world_to_pixel(all_world[ci])
resid = np.hypot(qx - src[ii, 0], qy - src[ii, 1])
nmatch = len(pairs)
# A quality gate, because a WRONG solution is worse than none: every
# position in the science catalogue would inherit the error, silently. A
# genuine solve on a field like this matches hundreds of stars and refines
# to well under a pixel; a spurious alignment matches a couple of dozen and
# stalls at several. Both conditions must hold.
MIN_STARS, MAX_RESID_PX = 40, 1.5
if resid is None or nmatch < MIN_STARS or np.median(resid) > MAX_RESID_PX:
got = "n/a" if resid is None else f"{np.median(resid):.2f} px"
print(f" REJECTED: {nmatch} stars at {got} residual "
f"(need >= {MIN_STARS} stars and <= {MAX_RESID_PX} px). "
f"Treating as unsolved rather than trusting a doubtful WCS.")
return None
cd = wcs.pixel_scale_matrix * 3600.0
solved_scale = float(np.sqrt(abs(np.linalg.det(cd))))
rot = float(np.degrees(np.arctan2(cd[0, 1], cd[1, 1])))
med_resid = float(np.median(resid)) if resid is not None else float("nan")
field = wcs.pixel_to_world(nx / 2.0, ny / 2.0)
if verbose:
print(f" solved on {nmatch} stars, residual "
f"{med_resid:.2f} px ({med_resid * solved_scale:.2f} arcsec)")
print(f" centre {field.to_string('hmsdms')}, "
f"{solved_scale:.4f} arcsec/px, PA {rot:.2f} deg, "
f"{nx * solved_scale / 60:.1f}' x {ny * solved_scale / 60:.1f}'")
whdr = wcs.to_header()
for filt in session.filters:
p = os.path.join(masters_dir, f"master-{filt}.fit")
if not os.path.exists(p):
continue
with fits.open(p, mode="update") as hd:
for card in whdr.cards:
hd[0].header[card.keyword] = (card.value, card.comment)
hd[0].header["ASTRSOLV"] = (
f"Gaia DR3 / {nmatch} stars / "
f"{med_resid * solved_scale:.2f} arcsec", "local plate solve")
hd.flush()
return dict(nstars=nmatch, residual_px=med_resid,
residual_arcsec=med_resid * solved_scale,
scale=solved_scale, position_angle=rot,
centre=field.to_string("hmsdms"),
fov_arcmin=(nx * solved_scale / 60, ny * solved_scale / 60))