Division-Average-Vectors.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 51,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "5410052d0ab0412dba73ca4747c186e3",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "VkJveChjaGlsZHJlbj0oSEJveChjaGlsZHJlbj0oU2VsZWN0TXVsdGlwbGUoZGVzY3JpcHRpb249dSdTZWxlY3Qgb25lIG9yIG1vcmUgYXJteSBncm91cChzKTonLCBvcHRpb25zPSgnSW5mIETigKY=\n"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import csv\n",
    "import math\n",
    "import pandas as pd\n",
    "import numpy as np\n",
    "import nvector as nv\n",
    "\n",
    "import folium\n",
    "import folium.plugins as plugins\n",
    "import ipywidgets as widgets\n",
    "from IPython.display import HTML\n",
    "from IPython.display import display\n",
    "\n",
    "reload(folium)\n",
    "reload(folium.plugins)\n",
    "\n",
    "#parse and load csv file into a Pandas DataFrame\n",
    "csvfile = 'Divisions.csv'\n",
    "df = pd.read_csv(csvfile,parse_dates=['DATE'])\n",
    "\n",
    "#group the data by Army group and num\n",
    "grouped_div = df.groupby(['Unit Type','Number'])\n",
    "\n",
    "#create a list of army groups to use in the selector\n",
    "army_groups = list(set(map((lambda key: key[0]),grouped_div.groups.keys())))\n",
    "\n",
    "# remove the Mass Unit type\n",
    "tmp = set(army_groups)\n",
    "tmp.remove('Mass')\n",
    "army_groups = list(tmp)\n",
    "\n",
    "out = widgets.Output()\n",
    "\n",
    "def get_div_origin_dest(div_data_df):\n",
    "    div_data = dict()\n",
    "    for row in div_data_df.itertuples():\n",
    "        # only consider non-detachment units\n",
    "        if math.isnan(row.Weight_D):\n",
    "            continue\n",
    "        if row.DATE not in div_data:\n",
    "            div_data[row.DATE] = dict()\n",
    "            div_data[row.DATE]['POINT_Y'] = row.POINT_Y\n",
    "            div_data[row.DATE]['POINT_X'] = row.POINT_X\n",
    "    div_locs = []\n",
    "    for key in sorted(div_data):\n",
    "        div_loc = dict()\n",
    "        div_loc['POINT_Y'] = div_data[key]['POINT_Y']\n",
    "        div_loc['POINT_X'] = div_data[key]['POINT_X']\n",
    "        div_locs.append(div_loc)\n",
    "    #there is an origin and dest\n",
    "    if len(div_locs) > 1:\n",
    "        origin = div_locs[0]\n",
    "        dest = div_locs[len(div_locs)-1]\n",
    "        return [[origin['POINT_Y'],origin['POINT_X']],\n",
    "                [dest['POINT_Y'],dest['POINT_X']]]\n",
    "    else:\n",
    "        # no non-detachment movement?\n",
    "        # look at just the detachments instead\n",
    "        div_data = dict()\n",
    "        for row in div_data_df.itertuples():\n",
    "            # only consider non-detachment units\n",
    "            if math.isnan(row.Weight_D):\n",
    "                if row.DATE not in div_data:\n",
    "                    div_data[row.DATE] = dict()\n",
    "                    div_data[row.DATE]['POINT_Y'] = row.POINT_Y\n",
    "                    div_data[row.DATE]['POINT_X'] = row.POINT_X\n",
    "        div_locs = []\n",
    "        for key in sorted(div_data):\n",
    "            div_loc = dict()\n",
    "            div_loc['POINT_Y'] = div_data[key]['POINT_Y']\n",
    "            div_loc['POINT_X'] = div_data[key]['POINT_X']\n",
    "            div_locs.append(div_loc)\n",
    "        #there is an origin and dest\n",
    "        if len(div_locs) > 1:\n",
    "            origin = div_locs[0]\n",
    "            dest = div_locs[len(div_locs)-1]\n",
    "            return [[origin['POINT_Y'],origin['POINT_X']],\n",
    "                    [dest['POINT_Y'],dest['POINT_X']]]\n",
    "            return None\n",
    "\n",
    "#gets the vector's bearing from North\n",
    "def get_bearing(orig, dest):\n",
    "    wgs84 = nv.FrameE(name='WGS84')\n",
    "    pointA = wgs84.GeoPoint(latitude=orig[0], longitude=orig[1], degrees=True)\n",
    "    pointB = wgs84.GeoPoint(latitude=dest[0], longitude=dest[1], degrees=True)\n",
    "    p_AB_E = nv.delta_E(pointA, pointB)\n",
    "    frame_N = nv.FrameN(pointA)\n",
    "    p_AB_N = p_AB_E.change_frame(frame_N)\n",
    "    p_AB_N = p_AB_N.pvector.ravel()\n",
    "    azimuth = np.arctan2(p_AB_N[1], p_AB_N[0])\n",
    "    return np.rad2deg(azimuth)\n",
    "\n",
    "# event handler for button click\n",
    "def plot_button_clicked(b):\n",
    "    if len(group_sel.value) == 0:\n",
    "        out.clear_output()\n",
    "        with out:\n",
    "            display(widgets.HTML(value=\"<HTML><font size='4' color='red'>Please select atleast one army group</font></HTML>\"))\n",
    "    else:\n",
    "        plot_average_vectors(group_sel.value)\n",
    "\n",
    "def plot_average_vectors(sel_groups):\n",
    "    origin_lats = []\n",
    "    origin_lons = []\n",
    "    dest_lats = []\n",
    "    dest_lons = []\n",
    "\n",
    "    #iterate over divisions\n",
    "    for group_key in grouped_div.groups.keys():\n",
    "        if group_key[0] in sel_groups:\n",
    "            div_df = grouped_div.get_group(group_key)[['POINT_X','POINT_Y','Weight_D','DATE']]\n",
    "            #get division's origin and destination\n",
    "            div_coords = get_div_origin_dest(div_df)\n",
    "\n",
    "            if div_coords is not None:\n",
    "                origin_lats.append(div_coords[0][0])\n",
    "                origin_lons.append(div_coords[0][1])\n",
    "                dest_lats.append(div_coords[1][0])\n",
    "                dest_lons.append(div_coords[1][1])\n",
    "\n",
    "    points = nv.GeoPoint(latitude=origin_lats,longitude=origin_lons,degrees=True)\n",
    "    nvectors = points.to_nvector()\n",
    "    n_EM_E = nvectors.mean()\n",
    "    g_EM_E = n_EM_E.to_geo_point()\n",
    "    lat, lon = g_EM_E.latitude_deg, g_EM_E.longitude_deg\n",
    "    mean_origin = [lat[0],lon[0]]\n",
    "\n",
    "    points = nv.GeoPoint(latitude=dest_lats,longitude=dest_lons,degrees=True)\n",
    "    nvectors = points.to_nvector()\n",
    "    n_EM_E = nvectors.mean()\n",
    "    g_EM_E = n_EM_E.to_geo_point()\n",
    "    lat, lon = g_EM_E.latitude_deg, g_EM_E.longitude_deg\n",
    "    mean_dest = [lat[0],lon[0]]\n",
    "\n",
    "    #now create the map\n",
    "    div_map = folium.Map([54.78, 32.04],zoom_start=6)\n",
    "\n",
    "    #first the markers\n",
    "    folium.CircleMarker(mean_origin,radius=4,color='green').add_to(div_map)\n",
    "    folium.CircleMarker(mean_dest,radius=4,color='red').add_to(div_map)\n",
    "    orig_dest_pair = [mean_origin,mean_dest]\n",
    "    vect_line = folium.PolyLine(locations=orig_dest_pair,weight=1,color='black')\n",
    "    div_map.add_child(vect_line)\n",
    "\n",
    "    #get vector direction\n",
    "    rotation = get_bearing(mean_origin,mean_dest) - 90 #to account for eastward initial rotation\n",
    "    #now display arrows in the direction\n",
    "    arrow_lats = np.linspace(mean_origin[0], mean_dest[0], 4)[1:3]\n",
    "    arrow_lons = np.linspace(mean_origin[1], mean_dest[1], 4)[1:3]\n",
    "    for points in zip(arrow_lats, arrow_lons):\n",
    "        folium.RegularPolygonMarker(location=points, \n",
    "                                    fill_color='black', number_of_sides=3, \n",
    "                                    radius=4, rotation=rotation).add_to(div_map)\n",
    "    # now display the map\n",
    "    out.clear_output()\n",
    "    with out:\n",
    "        display(div_map)\n",
    "\n",
    "style = {'description_width': 'initial'}\n",
    "group_sel = widgets.SelectMultiple(\n",
    "                options=army_groups,\n",
    "                rows=8,\n",
    "                description='Select one or more army group(s):',\n",
    "                disabled=False,\n",
    "                style=style\n",
    "            )\n",
    "plot_btn = widgets.Button(\n",
    "                description='Plot Average Vector',\n",
    "                disabled=False,\n",
    "                button_style='',\n",
    "                tooltip='Click to plot'\n",
    ")\n",
    "plot_btn.on_click(plot_button_clicked)\n",
    "sel_ui = widgets.HBox([group_sel,plot_btn])\n",
    "res = widgets.VBox([sel_ui,out])\n",
    "res"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}