Move processing parameters to config file (fixes #32)

pull/41/head
Cees Bassa 2019-09-21 14:26:50 +02:00
parent f2bda3ab25
commit 54b3baf681
2 changed files with 10 additions and 16 deletions

View File

@ -60,18 +60,12 @@ if __name__ == "__main__":
height=cfg.getfloat('Common', 'observer_el') * u.m)
# Extract settings
# Minimum predicted velocity (pixels/s)
drdtmin = 5.0
# Track selection region around prediction (pixels)
trkrmin = 10.0
# Track selection sigma
trksig = 5.0
# Minimum track points
ntrkmin = 10
drdtmin = cfg.getfloat('Processing', 'drdtmin')
trkrmin = cfg.getfloat('Processing', 'trkrmin')
trksig = cfg.getfloat('Processing', 'trksig')
ntrkmin = cfg.getint('Processing', 'ntrkmin')
nstarsmin = cfg.getint('Processing', 'nstarsmin')
# Move to processing directory
os.chdir(args.file_dir)
@ -110,7 +104,7 @@ if __name__ == "__main__":
pix_catalog = generate_star_catalog(fname)
# Solve
if pix_catalog.nstars > 10:
if pix_catalog.nstars > nstarsmin:
print(colored("Computing astrometric calibration for %s" % fname, "yellow"))
solved = generate_reference_with_anet(fname, "")
@ -136,7 +130,7 @@ if __name__ == "__main__":
generate_satellite_predictions(fname)
# Detect lines with 3D Hough transform
ids = find_hough3d_lines(fname)
ids = find_hough3d_lines(fname, ntrkmin, trkrmin)
# Get properties
ff = fourframe(fname)

View File

@ -17,7 +17,7 @@ def generate_satellite_predictions(fname):
return output
def find_hough3d_lines(fname, ntrkmin=20, dr=8):
def find_hough3d_lines(fname, ntrkmin, trkrmin):
# Read four frame
ff = fourframe(fname)
@ -38,7 +38,7 @@ def find_hough3d_lines(fname, ntrkmin=20, dr=8):
f.close()
# Run 3D Hough line-finding algorithm
command = "hough3dlines -dx %d -minvotes %d %s" % (dr, ntrkmin,
command = "hough3dlines -dx %d -minvotes %d %s" % (trkrmin, ntrkmin,
"/tmp/hough.dat")
try:
output = subprocess.check_output(command,