plot_demo.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "0dc262d8-c338-4b9d-8619-0736cfa4e0c0",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "\n",
    "from astropy.io import fits\n",
    "from astropy.table import Table\n",
    "from astropy.coordinates import Angle, SkyCoord\n",
    "import astropy.units as u\n",
    "import geoviews\n",
    "import pandas as pd"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6b6337c1-4293-4ab0-bf96-05d191f6a194",
   "metadata": {},
   "outputs": [],
   "source": [
    "# tbl = fits.open(\"/users/elowe/repos/antenna-plots/stack_table.fits\")\n",
    "# antenna_data=Table(tbl[1].data).to_pandas()\n",
    "# tbl.close()\n",
    "import pandas as pd\n",
    "filename = \"/home/scratch/elowe/antenna_data/ant_pos_with_az_al_small.parquet\"\n",
    "antenna_data = pd.read_parquet(filename)\n",
    "antenna_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "79907aeb-f190-4223-803e-9d4fe57f7e8d",
   "metadata": {},
   "outputs": [],
   "source": [
    "points = geoviews.Points(antenna_data, [\"RAJ2000\", \"DECJ2000\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "83eef9c8-6f1b-4153-aca9-1cf3075da0f3",
   "metadata": {},
   "outputs": [],
   "source": [
    "import datashader\n",
    "import colorcet as cc\n",
    "\n",
    "canvas = datashader.Canvas(\n",
    "    plot_width=800,\n",
    "    plot_height=400\n",
    ")\n",
    "canvas_points = canvas.points(points.data, \"RAJ2000\", \"DECJ2000\")\n",
    "shaded_points = datashader.transfer_functions.shade(canvas_points, cmap=cc.fire)\n",
    "shaded_points"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "7f3910cd-c7e1-4f66-b6ba-d2ebb58131a2",
   "metadata": {},
   "outputs": [],
   "source": [
    "from cartopy import crs\n",
    "projected = geoviews.operation.project_points(points, projection=crs.Mollweide())\n",
    "canvas = datashader.Canvas(\n",
    "    plot_width=800,\n",
    "    plot_height=400\n",
    ")\n",
    "canvas_points = canvas.points(projected.data, \"RAJ2000\", \"DECJ2000\")\n",
    "shaded_points = datashader.transfer_functions.shade(canvas_points, cmap=cc.fire)\n",
    "shaded_points"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a06bfc4e-09cb-4f78-bf2c-ba9c394fa5ef",
   "metadata": {},
   "outputs": [],
   "source": [
    "geoviews.extension('bokeh')\n",
    "\n",
    "points"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a27722a6-586d-4800-8ab1-0622752f6b14",
   "metadata": {},
   "outputs": [],
   "source": [
    "projected.opts(tools=['hover'],projection=crs.Mollweide(), width=800, height=400)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "950558a1-b82c-403b-a9c7-094df32bc5fd",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Frame for RA and DEC\n",
    "sc = SkyCoord(antenna_data['RAJ2000'], antenna_data['DECJ2000'], unit=u.deg, frame=\"icrs\", equinox=\"J2000\")\n",
    "sc"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8a158676-e04c-4f72-b16d-419618d33d41",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Get skycoord into a dataframe\n",
    "galactic_df = sc.galactic.to_table().to_pandas()\n",
    "galactic_df.tail()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "327a6618-9a9a-4ec8-9959-2d71a31212df",
   "metadata": {},
   "outputs": [],
   "source": [
    "antenna_data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "f73d560f-4501-42eb-a606-7aa4acbec0e0",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Creating one file with all the columns we need\n",
    "\n",
    "gal_ra = galactic_df['l'].values.tolist()\n",
    "gal_dec = galactic_df['b'].values.tolist()\n",
    "\n",
    "antenna_data[\"gal_ra\"] = gal_ra\n",
    "antenna_data[\"gal_dec\"] = gal_dec\n",
    "antenna_data2 = antenna_data\n",
    "\n",
    "antenna_data2.to_parquet('modified_ant_pos_with_az_al_small.parquet', engine='fastparquet')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a73b4899-3a63-4d79-8e3d-848635aab596",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Double checking that the file works\n",
    "\n",
    "filename1 = \"/users/elowe/repos/antenna-plots/modified_ant_pos_with_az_al_small.parquet\"\n",
    "antenna_data1 = pd.read_parquet(filename1)\n",
    "antenna_data2.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "53a355d0-dfa5-4e2a-af94-50eb0ea5f51a",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Get points for geoview\n",
    "sc_points = geoviews.Points(galactic_df, ['l', 'b'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6e40c058-8fac-4f1d-a9c0-a29f772957f8",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Shade the points on canvas\n",
    "sc_canvas = datashader.Canvas(\n",
    "    plot_width=800,\n",
    "    plot_height=400\n",
    ")\n",
    "sc_canvas_points = sc_canvas.points(sc_points.data, \"l\", \"b\")\n",
    "sc_shaded_points = datashader.transfer_functions.shade(sc_canvas_points, cmap=cc.fire)\n",
    "sc_shaded_points"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2b62975e-4afe-4496-99c7-b332eeff61a0",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Project the points on Mollweide\n",
    "sc_projected = geoviews.operation.project_points(sc_points, projection=crs.Mollweide())\n",
    "sc_canvas = datashader.Canvas(\n",
    "    plot_width=800,\n",
    "    plot_height=400\n",
    ")\n",
    "sc_canvas_points = sc_canvas.points(sc_projected.data, \"l\", \"b\")\n",
    "sc_shaded_points = datashader.transfer_functions.shade(sc_canvas_points, cmap=cc.fire)\n",
    "sc_shaded_points"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5ab28c02-3c07-45e0-91a3-1086d2d0d9ba",
   "metadata": {},
   "outputs": [],
   "source": [
    "# interactive view\n",
    "sc_projected.opts(projection=crs.Mollweide(), width=800, height=400)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d33822a6-bfba-4c93-a5e5-5d4f658c22d1",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Frame for azi and alt\n",
    "sc_azi_alt = SkyCoord(antenna_data['azimuth'], antenna_data['altitude'], unit=u.deg)\n",
    "sc_azi_alt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d4e76de9-0e26-4f47-bf80-05c4a2cf531f",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Getting values from skycoord to dataframe\n",
    "azi_df = sc_azi_alt.to_table().to_pandas()\n",
    "azi_df"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "1caf8f5f-ae74-405c-9b17-c96806ab727c",
   "metadata": {},
   "outputs": [],
   "source": [
    "az_el_points = geoviews.Points(azi_df, ['ra', 'dec'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d841fb66-a8df-4bf9-aa59-4b24229d7c8c",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Shade the points\n",
    "az_el_canvas = datashader.Canvas(\n",
    "    plot_width=800,\n",
    "    plot_height=400\n",
    ")\n",
    "az_el_canvas_points = sc_canvas.points(az_el_points.data, \"ra\", \"dec\")\n",
    "az_el_shaded = datashader.transfer_functions.shade(az_el_canvas_points, cmap=cc.rainbow)\n",
    "az_el_shaded"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "d99f560b-02fd-4d2f-945f-8a124e27f43e",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Project the points on AzimuthalEquidistant\n",
    "az_el_points = geoviews.Points(azi_df, ['ra', 'dec'])\n",
    "\n",
    "az_el_projected = geoviews.operation.project_points(az_el_points, projection=crs.AzimuthalEquidistant())\n",
    "az_el_canvas = datashader.Canvas(\n",
    "    # Since it is a circle, we want a sqaure canvas\n",
    "    plot_width=800,\n",
    "    plot_height=800\n",
    ")\n",
    "az_el_canvas_points = az_el_canvas.points(az_el_projected.data, \"ra\", \"dec\")\n",
    "az_el_points = datashader.transfer_functions.shade(az_el_canvas_points, cmap=cc.fire)\n",
    "az_el_points"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "7ac99032-fe37-4cff-9700-24cdf44b8593",
   "metadata": {},
   "outputs": [],
   "source": [
    "# interactive view\n",
    "az_el_projected.opts(projection=crs.AzimuthalEquidistant(), width=800, height=800)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "11ef2a7e-0d7b-47de-9339-d7f7980df592",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}