Store calibration file and reinstate test for astrometry

pull/33/head
Cees Bassa 2019-08-02 22:15:24 +02:00
parent cbfa0c5ee3
commit 569b840dc0
2 changed files with 16 additions and 1 deletions

View File

@ -4,6 +4,7 @@ import glob
import numpy as np
from stvid.stio import fourframe
from stvid.stars import generate_star_catalog
from stvid.stars import store_calibration
from stvid.astrometry import calibrate_from_reference
from stvid.astrometry import is_calibrated
from stvid.astrometry import generate_reference_with_anet
@ -100,7 +101,7 @@ if __name__ == "__main__":
fnames = sorted(glob.glob("2*.fits"))
# Create reference calibration file
if os.path.exists("test.fits"):
if not os.path.exists("test.fits"):
solved = False
# Loop over files to find a suitable calibration file
for fname in fnames:
@ -125,6 +126,9 @@ if __name__ == "__main__":
# Calibrate from reference
calibrate_from_reference(fname, "test.fits", pix_catalog)
# Store calibration
store_calibration(pix_catalog, fname + ".cal")
# Generate satellite predictions
generate_satellite_predictions(fname)

View File

@ -47,3 +47,14 @@ def generate_star_catalog(fname):
os.rename("test.cat", fname+".cat")
return pixel_catalog(fname+".cat")
def store_calibration(pix_catalog, fname):
with open(fname, "w") as fp:
for i in range(pix_catalog.nstars):
if pix_catalog.flag[i]:
fp.write("%8.3f %8.3f %7.4f %10.6f %10.6f %6.3f\n" % (pix_catalog.x[i],
pix_catalog.y[i],
pix_catalog.mag[i],
pix_catalog.ra[i],
pix_catalog.dec[i],
pix_catalog.imag[i]))