python-notebooks/catalog_utils_v2.py at master · AngelGSoto/python-notebooks · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
'''
Taken from: https://github.com/will-henney/orion-bowshock-catalog/blob/master/luis-angel-ll.org
'''
region_hdr_lines = [
    '# Region file format: DS9 version 4.1',
    'global color=yellow dashlist=8 3 width=1 font=\"helvetica 20 normal roman\" '
    + 'select=1 highlite=1 dash=0 fixed=0 edit=1 move=1 delete=1 include=1 source=1',
    'fk5',
]

def region_box_to_string(ra='5:34:40.800', dec='-5:22:43.00', text='63864', text1='47481',
                         color='yellow', width=6):
                         pattern = 'circle({},{},{}\") # text={{{}-{}}} color={}'
                         return pattern.format(ra, dec, width, text, text1, color)


def region_lines_from_table(table, ra='RAJ2000', dec='DEJ2000', radius='kron', tile='Field', name='Name', color='yellow'):
    """Create DS9 region lines for all sources in an astropy `table`

    Coordinates come from the columns named `ra` and `dec`, while
    source names come the column named `name` and are formatted
    according to `name_fmt`"""

    region_lines = region_hdr_lines[:]
    for row in table:
        if row[radius] != 0.0:
            region_lines.append(
                                region_box_to_string(row[ra], row[dec], row[tile],
                                                  row[name], color, row[radius]))
        else:
            region_lines.append(
                                region_box_to_string(row[ra], row[dec], row[tile],
                                                  row[name], color))
    return region_lines