Ejemplo n.º 1
0
def solve_field(
        image_files: Union[str, list],
        base_filename: str = "astrometry",
        overwrite: bool = True,
        tweak: bool = True,
        search_radius: units.Quantity = 1 * units.degree,
        centre: SkyCoord = None,
        guess_scale: bool = True,
        time_limit: units.Quantity = None,
        verify: bool = True,
        odds_to_tune_up: float = 1e6,
        odds_to_solve: float = 1e9,
        *flags,
        **params):
    """
    Returns True if successful (by checking whether the corrected file is generated); False if not.
    :param image_files:
    :param base_filename:
    :param overwrite:
    :param flags:
    :param params:
    :return:
    """

    params["o"] = base_filename
    params["odds-to-tune-up"] = odds_to_tune_up
    params["odds-to-solve"] = odds_to_solve
    if time_limit is not None:
        params["l"] = check_quantity(time_limit, units.second).value
    if search_radius is not None:
        params["radius"] = search_radius.to(units.deg).value
    if centre is not None:
        params["ra"] = centre.ra.to(units.deg).value
        params["dec"] = centre.dec.to(units.deg).value
    debug_print(1, "solve_field(): tweak ==", tweak)

    flags = list(flags)
    if overwrite:
        flags.append("O")
    if guess_scale:
        flags.append("g")
    if not tweak:
        flags.append("T")
    if not verify:
        flags.append("y")


    system_command("solve-field", image_files, False, True, False, *flags, **params)
    if isinstance(image_files, list):
        image_path = image_files[0]
    else:
        image_path = image_files
    check_dir = os.path.split(image_path)[0]
    check_path = os.path.join(check_dir, f"{base_filename}.new")
    print(f"Checking for result file at {check_path}...")
    return os.path.isfile(check_path)
Ejemplo n.º 2
0
def make_header(table_path: str, output_path: str):
    """
    Executes Montage task mMakeHdr <table_path> <output_path>
    :param table_path:
    :param output_path:
    :return:
    """
    u.sanitise_file_ext(output_path, ".hdr")
    return u.system_command("mMakeHdr", [table_path, output_path])
Ejemplo n.º 3
0
def build_astrometry_index(input_fits_catalog: str, unique_id: int, output_index: str = None,
                           scale_number: int = 0, sort_column: str = 'mag',
                           scan_through_catalog: bool = True, *flags, **params):
    print(input_fits_catalog)
    params["i"] = input_fits_catalog
    params["I"] = unique_id
    if output_index is not None:
        params["o"] = output_index
    if scale_number is not None:
        params["P"] = scale_number
    if sort_column is not None:
        params["s"] = sort_column

    flags = list(flags)
    if scan_through_catalog:
        flags.append("E")

    system_command("build-astrometry-index", None, False, True, *flags, **params)
Ejemplo n.º 4
0
def image_table(input_directory: str, output_path: str = "images.tbl"):
    """
    Executes the Montage task mImgtbl <input_directory> <output_path>
    :param input_directory:
    :param output_path:
    :return:
    """
    u.sanitise_file_ext(filename=output_path, ext=".tbl")
    return u.system_command("mImgtbl", [input_directory, output_path])
Ejemplo n.º 5
0
def psfex(catalog: str,
          output_name: str = None,
          output_dir: str = None,
          **kwargs):
    old_dir = os.getcwd()
    if output_dir is None:
        output_dir = os.getcwd()
    else:
        os.chdir(output_dir)
    if output_name is None:
        cat_name = os.path.split(catalog)[-1]
        output_name = cat_name.replace(".fits", ".psf")
    system_command(command="psfex",
                   arguments=[catalog],
                   force_single_dash=True,
                   **kwargs)
    os.chdir(old_dir)
    psfex_path = os.path.join(output_dir, output_name)
    return psfex_path
Ejemplo n.º 6
0
def overlaps(table_path: str, difference_table_path: str):
    """
    Executes mOverlaps <table_path> <difference_table_path>
    :param table_path:
    :param difference_table_path:
    :return:
    """
    table_path = u.sanitise_file_ext(filename=table_path, ext=".tbl")
    difference_table_path = u.sanitise_file_ext(filename=difference_table_path,
                                                ext=".tbl")
    return u.system_command(command="mOverlaps",
                            arguments=[table_path, difference_table_path])
Ejemplo n.º 7
0
def fit_execute(difference_table_path: str, fit_table_path: str,
                diff_dir: str):
    """
    Executes mFitExec <difference_table_path> <fit_table_path> <diff_dir>
    :param difference_table_path:
    :param fit_table_path:
    :param diff_dir:
    :return:
    """
    difference_table_path = u.sanitise_file_ext(filename=difference_table_path,
                                                ext=".tbl")
    fit_table_path = u.sanitise_file_ext(filename=fit_table_path, ext=".tbl")
    return u.system_command(
        command="mFitExec",
        arguments=[difference_table_path, fit_table_path, diff_dir])
Ejemplo n.º 8
0
def difference_execute(input_directory: str, difference_table_path: str,
                       header_path: str, diff_dir: str):
    """
    Executes mDiffExec <difference_table_path> <header_file> <diff_dir> -p <input_directory>
    :param input_directory:
    :return:
    """
    difference_table_path = u.sanitise_file_ext(filename=difference_table_path,
                                                ext=".tbl")
    header_path = u.sanitise_file_ext(filename=header_path, ext=".hdr")
    return u.system_command(
        command="mDiffExec",
        arguments=[difference_table_path, header_path, diff_dir],
        p=input_directory,
    )
Ejemplo n.º 9
0
def background_model(table_path: str, fit_table_path: str,
                     correction_table_path: str):
    """
    Executes mBGModel <table_path> <fit_table_path> <correction_table_path>
    :param table_path:
    :param fit_table_path:
    :param correction_table_path:
    :return:
    """
    table_path = u.sanitise_file_ext(filename=table_path, ext=".tbl")
    fit_table_path = u.sanitise_file_ext(filename=fit_table_path, ext=".tbl")
    correction_table_path = u.sanitise_file_ext(filename=correction_table_path,
                                                ext=".tbl")
    return u.system_command(
        command="mBgModel",
        arguments=[table_path, fit_table_path, correction_table_path])
Ejemplo n.º 10
0
def background_execute(input_directory: str, table_path: str,
                       correction_table_path: str, corr_dir: str):
    """
    Executes mBgExec <table_path> <corrections_table_path> <corr_dir> -p <input_directory>
    :param input_directory:
    :param table_path:
    :param correction_table_path:
    :param corr_dir:
    :return:
    """
    table_path = u.sanitise_file_ext(filename=table_path, ext=".tbl")
    correction_table_path = u.sanitise_file_ext(filename=correction_table_path,
                                                ext=".tbl")
    return u.system_command(
        command="mBgExec",
        arguments=[table_path, correction_table_path, corr_dir],
        p=input_directory)
Ejemplo n.º 11
0
def project_execute(input_directory: str, table_path: str, header_path: str,
                    proj_dir: str, stats_table_path: str):
    """
    Executes mProjExec <table_path> <header_path> <proj_dir> <stats_table_path> -p <input_directory>
    :param input_directory:
    :param table_path:
    :param header_path:
    :param proj_dir:
    :param stats_table_path:
    :return:
    """
    table_path = u.sanitise_file_ext(filename=table_path, ext=".tbl")
    header_path = u.sanitise_file_ext(filename=header_path, ext=".hdr")
    stats_table_path = u.sanitise_file_ext(filename=stats_table_path,
                                           ext=".tbl")
    return u.system_command(
        command="mProjExec",
        arguments=[table_path, header_path, proj_dir, stats_table_path],
        p=input_directory)
Ejemplo n.º 12
0
def add(input_directory: str,
        table_path: str,
        header_path: str,
        output_path: str = "coadded.fits",
        coadd_type: str = 'median'):
    """
    Executes mAdd  <table_path> <template_path> <output_path> -p <input_directory> -a <coadd_type>
    :param input_directory:
    :param table_path:
    :param header_path:
    :param output_path:
    :param coadd_type:
    :return:
    """
    table_path = u.sanitise_file_ext(filename=table_path, ext=".tbl")
    header_path = u.sanitise_file_ext(filename=header_path, ext=".hdr")
    output_path = u.sanitise_file_ext(filename=output_path, ext=".fits")
    return u.system_command(command="mAdd",
                            arguments=[table_path, header_path, output_path],
                            p=input_directory,
                            a=coadd_type)