sky_plot.py

import numpy as np
import psrcat as pc
from bokeh.palettes import Viridis6 as palette
cat = pc.PSRCAT('data/psrcat.db')
raj, decj = zip(*[x['JCOORD_RAD'] for x in cat.blocks])
from bokeh.layouts import layout
from bokeh.models import (CustomJS, Slider, ColumnDataSource, WidgetBox, Toggle,
                          HoverTool, LogColorMapper)
from bokeh.plotting import figure, output_file, show
import sqlite3 as sql
import os

output_file('sky_plot.html')
tools = 'pan'

def load_psr_pos(db_file):
    cat = pc.PSRCAT(db_file)
    raj, decj = zip(*[x['JCOORD_RAD'] for x in cat.blocks])
    return raj, decj

def get_all_pointings(dbpath):
    '''
    access db at dbpath and read all pointings.
    columns required: datetime, ra, dec
    ra and dec are expected in degrees
    datetime format expected is similar to: 2017-Jul-20 00:08:17
    '''

    conn = sql.connect(dbpath)
    cur = conn.cursor()
    query = "SELECT datetime, ra, dec FROM pointings"
    cur.execute(query)
    results = cur.fetchall()
    return results

def get_observation_entry(summary_file):
    f = open(summary_file, 'r')
    entries = []
    info_list = ['date', 'time' ,'RAJ', 'DECJ', 'observer']
    for line in f.readlines():
        info = {}
        line = line.strip()
        l = line.split()
        num_ele = len(l)
        if num_ele < 5:
            raise RuntimeError("Observation summary file should have "
                    "'date', 'time' ,'RAJ', 'DECJ', 'observer' in the column.")
        for i in range(4):
            info[info_list[i]] = l[i]
        observers = ''
        for i in range(4, num_ele):
             observers += l[i] + ' '
        info[info_list[4]] = observers
        entries.append(info)
    return entries

def get_sky_cover_area(RAJ, DECJ, beam_size):
    half_beam = beam_size / 2.0
    RAJ = np.array(RAJ, ndmin=1, copy=False)
    DECJ = np.array(DECJ, ndmin=1, copy=False)
    if len(RAJ) != len(DECJ):
        raise ValueError("RAJ and DECJ have to be in the same dimension.")
    x_low = RAJ - half_beam
    x_up = RAJ + half_beam
    y_low = DECJ - half_beam
    y_up = DECJ + half_beam
    xs = np.zeros((len(RAJ), 4))
    ys = np.zeros((len(RAJ), 4))
    for ii in range(len(x_low)):
        xs[ii, :] = np.array([x_low[ii], x_up[ii], x_up[ii], x_low[ii]])
        ys[ii, :] = np.array ([y_low[ii], y_low[ii], y_up[ii], y_up[ii]])
    return xs, ys

def sky_plot(db_file, pointings):
    '''
    plot pointings in pulsar catalog pointed to by db_file.
    plot pointings in database retrieved in pointings parameter.

    THIS IS A DIRTY IMPLEMENTATION. WE SHOULD CLEAN THIS UP.
    '''
    raj, decj = load_psr_pos(db_file)

    source = ColumnDataSource(data=dict(x=raj, y=decj))

    # unpack pointings taken from database
    dates, ras, decs = zip(*pointings)


    cover_raj, cover_decj = get_sky_cover_area(ras,decs, 2.5/60.0)
    cover_raj *= np.pi/180.
    cover_decj *= np.pi/180.
    projects = ['P2030']*len(cover_decj)
    source2 = ColumnDataSource(data=dict(
                x=cover_raj.tolist(),
                y=cover_decj.tolist(),
                project=projects,
                date=dates,
                ))
    TOOLS = "pan,wheel_zoom,reset,hover,hover,save"
    plot = figure(
        x_range=(-1,7),
        y_range=(-2, 2), tools=TOOLS, toolbar_location="left",
        title="ANTF Catalogue",
        x_axis_label='RAJ (rad)',
        y_axis_label='DECJ (rad)')

    box_color = "navy"
    box_alpha = 0.4
    starplot = plot.asterisk('x', 'y', source=source, size=5)
    cover_box = plot.patches('x', 'y', source=source2, line_width=3,
                color=box_color,
                fill_alpha=box_alpha)

    pulsar_tooltips = [
        ("RA, DEC : ", "($x, $y)"),
        ]

    obs_tooltips = [
        ("Project", "@project"),
        ("Date", "@date"),
        ("Time", "@time"),
        ("Observer", "@name"),
        ("RA, DEC : ", "($x, $y)"),
        ]

    hover = plot.select(dict(type=HoverTool))
    hover[0].renderers = [starplot]
    hover[0].tooltips = pulsar_tooltips
    hover[1].renderers = [cover_box]
    hover[1].tooltips = obs_tooltips

    code2 = '''\
    object.visible = toggle.active
    '''

    callback_star = CustomJS.from_coffeescript(code=code2, args={})
    toggle1 = Toggle(label="Pulsars", button_type="success", callback=callback_star)
    callback_star.args = {'toggle': toggle1, 'object': starplot}

    callback_box = CustomJS.from_coffeescript(code=code2, args={})
    toggle2 = Toggle(label="Observation Scans", button_type="success", callback=callback_box)
    callback_box.args = {'toggle': toggle2, 'object': cover_box }
    return [plot, toggle1, toggle2]


obs_pointings = get_all_pointings(
    os.path.join(os.path.dirname(os.path.abspath(__file__)),
                 'db', 'p2030', 'pointings.db'))
print(np.shape(obs_pointings))
sp = sky_plot('data/psrcat.db', obs_pointings)
l = layout([
    sp[0],
    sp[1:3],
], sizing_mode='scale_width')

show(l)