position_movies.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "1389efad-aacc-4eed-85cb-dfd8162a1d2b",
   "metadata": {},
   "source": [
    "### Spacecraft and planet trajectory movies\n",
    "\n",
    "\n",
    "https://github.com/cmoestl/heliocats\n",
    " \n",
    " Authors: C. Möstl, Eva Weiler, Emma Davies, Austrian Space Weather Office, GeoSphere Austria\n",
    " \n",
    " uses env helio4\n",
    " \n",
    " \n",
    " twitter @chrisoutofspace, https://github.com/cmoestl\n",
    " \n",
    " \n",
    " last update: April 2024\n",
    " \n",
    " needs the helio4 environment (see README.md)\n",
    " "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "d9011f96-a783-473b-b521-f602d0a5e3a3",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "time resolution in hours 4\n",
      "system is mac\n",
      "/Users/chris/python/data/insitu_python/\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "[NbConvertApp] Converting notebook position_movies.ipynb to script\n",
      "[NbConvertApp] Writing 23021 bytes to position_movies.py\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "0"
      ]
     },
     "execution_count": 21,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import os\n",
    "import datetime\n",
    "from datetime import datetime, timedelta\n",
    "\n",
    "\n",
    "############################ directories\n",
    "\n",
    "#animdirectory   = 'results/positions/movies_2024'\n",
    "#outputdirectory = 'results/positions/movies_2024/frames'\n",
    "#outputdirectory = 'results/positions/movies_2024/frames_zoom'\n",
    "\n",
    "#for the movie\n",
    "animdirectory   = 'icmecat'\n",
    "#outputdirectory = 'results/positions/movies_2024/frames'\n",
    "#for the frames\n",
    "outputdirectory = 'icmecat/anim_frames'\n",
    "\n",
    "\n",
    "if os.path.isdir(outputdirectory) == False: os.mkdir(outputdirectory)\n",
    "if os.path.isdir(animdirectory) == False: os.mkdir(animdirectory)\n",
    "\n",
    "#movie_filename='positions_punch_2025_2029'\n",
    "#movie_filename='positions_punch_2025_2029_zoom'\n",
    "#movie_filename='positions_2018_2030'\n",
    "movie_filename='icmecat_visual'\n",
    "\n",
    "\n",
    "####################### Time resolution\n",
    "\n",
    "#res_in_hours=24\n",
    "#res_in_hours=6\n",
    "res_in_hours=4\n",
    "\n",
    "\n",
    "print('time resolution in hours',res_in_hours)\n",
    "\n",
    "############### make time range\n",
    "\n",
    "##PUNCH 4.5 years\n",
    "t_start = datetime(2025,4,1)\n",
    "t_end   = datetime(2029,12,31)\n",
    "\n",
    "## Solar Orbiter 10 years\n",
    "#t_start = datetime(2020,2,11)\n",
    "#t_end   = datetime(2030,11,19)\n",
    "\n",
    "## from psp launch onwards\n",
    "#t_start = datetime(2018,10,1)\n",
    "#t_end   = datetime(2030,11,19)\n",
    "\n",
    "\n",
    "## ICME visuals\n",
    "#t_start = datetime(2023,3,10)\n",
    "#t_end   = datetime(2023,5,10)\n",
    "\n",
    "t_start = datetime(2020,4,1)\n",
    "t_end   = datetime(2024,4,1)\n",
    "\n",
    "\n",
    "########## plots\n",
    "\n",
    "#rmax=1.72\n",
    "rmax=1.1\n",
    "\n",
    "\n",
    "dpisave=100 #for 1080p\n",
    "#dpisave=200 #for 4K\n",
    "\n",
    "##multiprocessing\n",
    "\n",
    "#used=100\n",
    "used=8\n",
    "\n",
    "\n",
    "###########################################################\n",
    "\n",
    "ffmpeg_path=''\n",
    "\n",
    "\n",
    "from sunpy.time import parse_time\n",
    "\n",
    "import matplotlib\n",
    "import matplotlib.pyplot as plt\n",
    "import matplotlib.dates as mdates\n",
    "from scipy.signal import medfilt\n",
    "import numpy as np\n",
    "import pdb\n",
    "import pickle\n",
    "import seaborn as sns\n",
    "import sys\n",
    "import heliopy.data.spice as spicedata\n",
    "import heliopy.spice as spice\n",
    "import astropy\n",
    "import importlib    \n",
    "import time\n",
    "import numba\n",
    "from numba import jit\n",
    "import multiprocessing as mp\n",
    "\n",
    "#ignore warnings\n",
    "import warnings\n",
    "warnings.filterwarnings('ignore')\n",
    "\n",
    "#for server\n",
    "matplotlib.use('Agg')\n",
    "\n",
    "##### check for system type\n",
    "#server\n",
    "if sys.platform == 'linux': \n",
    "    print('system is linux')\n",
    "    matplotlib.use('Agg') \n",
    "    from config_server import data_path\n",
    "    \n",
    "    \n",
    "#mac - make sure the dpi is always set similar to plt.savefig\n",
    "if sys.platform =='darwin':  \n",
    "    print('system is mac')\n",
    "    #for testing\n",
    "    get_ipython().run_line_magic('matplotlib', 'inline')\n",
    "    from config_local import data_path\n",
    "    #matplotlib.use('Agg') \n",
    "\n",
    "\n",
    "print(data_path)\n",
    "\n",
    "\n",
    "\n",
    "#Convert this notebook to a script \n",
    "os.system('jupyter nbconvert --to script position_movies.ipynb')    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "7a87b502-8498-427e-a31f-75464ad73d01",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "load positions\n",
      "load icmecat\n",
      "done\n"
     ]
    }
   ],
   "source": [
    "print('load positions')\n",
    "[psp, solo, sta, stb, bepi, l1, juno, juice, uly, earth, mercury, venus, mars, jupiter, saturn, uranus, neptune]=pickle.load(open(data_path+'/positions_psp_solo_sta_bepi_wind_juno_juice_ulysses_planets_HEEQ_1hour_rad.p', \"rb\" ) )\n",
    "\n",
    "print('load icmecat')\n",
    "#load icmecat as numpy array \n",
    "file='icmecat/HELIO4CAST_ICMECAT_v22_numpy.p'\n",
    "[ic,ic_np,h,p]=pickle.load( open(file, 'rb'))   \n",
    "ic=ic.to_records()\n",
    "\n",
    "print('done')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "19944bd6-4ea9-4a59-9e38-ac89554afb0b",
   "metadata": {
    "tags": []
   },
   "outputs": [],
   "source": [
    "def make_frame(k):\n",
    "    '''\n",
    "    loop each frame in multiprocessing\n",
    "    '''\n",
    "     \n",
    "    \n",
    "    if not black: \n",
    "        fig=plt.figure(1, figsize=(19.2,10.8), dpi=100) #full hd\n",
    "        #fig=plt.figure(1, figsize=(19.2*2,10.8*2), dpi=100) #4k\n",
    "        ax = plt.subplot2grid((5,2), (0, 0), rowspan=5, projection='polar')\n",
    "        backcolor='black'\n",
    "        psp_color='black'\n",
    "        bepi_color='blue'\n",
    "        solo_color='green'\n",
    "\n",
    "    if black: #4k resolution        \n",
    "        fig=plt.figure(1, figsize=(19.2,10.8), dpi=dpisave, facecolor='black', edgecolor='black')\n",
    "        ax = plt.subplot(111,projection='polar',facecolor='black') \n",
    "        #ax = plt.subplot2grid((5,2), (0, 0), rowspan=5, projection='polar')\n",
    "        backcolor='white'\n",
    "        psp_color='white'\n",
    "        bepi_color='skyblue'\n",
    "        solo_color='springgreen'\n",
    "        sta_color='salmon'\n",
    "        juice_color='greenyellow'\n",
    "\n",
    "\n",
    "    frame_time_str=str(mdates.num2date(frame_time_num+k*res_in_hours/24))\n",
    "    \n",
    "    #print(frame_time_num)\n",
    "    print( 'current frame_time_num', frame_time_str, '     ',k)\n",
    "\n",
    "\n",
    "    #all same times\n",
    "    dct=frame_time_num+k*res_in_hours/24-earth.time\n",
    "    earth_timeind=np.argmin(abs(dct))\n",
    "    \n",
    "    \n",
    "    \n",
    "    #these have their own times\n",
    "    \n",
    "    #all same times\n",
    "    dct=frame_time_num+k*res_in_hours/24-sta.time\n",
    "    sta_timeind=np.argmin(abs(dct))\n",
    "    \n",
    "    dct=frame_time_num+k*res_in_hours/24-psp.time\n",
    "    psp_timeind=np.argmin(abs(dct))\n",
    "\n",
    "    dct=frame_time_num+k*res_in_hours/24-bepi.time\n",
    "    bepi_timeind=np.argmin(abs(dct))\n",
    "\n",
    "    dct=frame_time_num+k*res_in_hours/24-solo.time\n",
    "    solo_timeind=np.argmin(abs(dct))\n",
    "\n",
    "    dct=frame_time_num+k*res_in_hours/24-juice.time\n",
    "    juice_timeind=np.argmin(abs(dct))\n",
    "    \n",
    "    \n",
    "\n",
    "    #plot all positions including text R lon lat for some \n",
    "\n",
    "    #white background\n",
    "    if not black:\n",
    "        ax.scatter(venus.lon[earth_timeind], venus.r[earth_timeind]*np.cos(venus.lat[earth_timeind]), s=symsize_planet, c='orange', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(mercury.lon[earth_timeind], mercury.r[earth_timeind]*np.cos(mercury.lat[earth_timeind]), s=symsize_planet, c='dimgrey', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(earth.lon[earth_timeind], earth.r[earth_timeind]*np.cos(earth.lat[earth_timeind]), s=symsize_planet, c='mediumseagreen', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(sta.lon[earth_timeind], sta.r[earth_timeind]*np.cos(sta.lat[earth_timeind]), s=symsize_spacecraft, c='red', marker='s', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(mars.lon[earth_timeind], mars.r[earth_timeind]*np.cos(mars.lat[earth_timeind]), s=symsize_planet, c='orangered', alpha=1,lw=0,zorder=3)\n",
    "\n",
    "\n",
    "        plt.figtext(0.95,0.75,'PSP ', color='black', ha='center',fontsize=fsize+3)\n",
    "        plt.figtext(0.95,0.5,'Wind', color='mediumseagreen', ha='center',fontsize=fsize+3)\n",
    "        plt.figtext(0.95,0.25,'STEREO-A', color='red', ha='center',fontsize=fsize+3)\n",
    "        '''\n",
    "        plt.figtext(0.9,0.9,'Mercury', color='dimgrey', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9\t,0.8,'Venus', color='orange', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.7,'Earth', color='mediumseagreen', ha='center',fontsize=fsize+5)\n",
    "        #plt.figtext(0.9,0.7,'Mars', color='orangered', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.6,'STEREO-A', color='red', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.5,'Parker Solar Probe', color='black', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.4,'Bepi Colombo', color='blue', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.3,'Solar Orbiter', color='green', ha='center',fontsize=fsize+5)\n",
    "        '''\n",
    "\n",
    "    #black background\n",
    "    if black:\n",
    "        ax.scatter(venus.lon[earth_timeind], venus.r[earth_timeind]*np.cos(venus.lat[earth_timeind]), s=symsize_planet, c='orange', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(mercury.lon[earth_timeind], mercury.r[earth_timeind]*np.cos(mercury.lat[earth_timeind]), s=symsize_planet, c='grey', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(earth.lon[earth_timeind], earth.r[earth_timeind]*np.cos(earth.lat[earth_timeind]), s=symsize_planet, c='mediumseagreen', alpha=1,lw=0,zorder=3)\n",
    "        ax.scatter(mars.lon[earth_timeind], mars.r[earth_timeind]*np.cos(mars.lat[earth_timeind]), s=symsize_planet, c='orangered', alpha=1,lw=0,zorder=3)\n",
    "\n",
    "        ax.scatter(sta.lon[sta_timeind], sta.r[sta_timeind]*np.cos(sta.lat[sta_timeind]), s=symsize_spacecraft, c=sta_color, marker='s', alpha=1,lw=0,zorder=3)\n",
    "\n",
    "        plt.figtext(0.9,0.9,'Mercury', color='grey', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.8,'Venus', color='orange', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.7,'Earth', color='mediumseagreen', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.6,'Mars', color='orangered', ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.5,'STEREO-A', color=sta_color, ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.4,'Parker Solar Probe', color=psp_color, ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.3,'Bepi Colombo', color=bepi_color, ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.2,'Solar Orbiter', color=solo_color, ha='center',fontsize=fsize+5)\n",
    "        plt.figtext(0.9,0.1,'JUICE', color=juice_color, ha='center',fontsize=fsize+5)\n",
    "\n",
    "        \n",
    "    #get \n",
    "    if add_icmes: \n",
    "        #get all indices for icmes starting before ** days of current time\n",
    "        cur_icme=np.where(np.logical_and((frame_time_num+k*res_in_hours/24-ic.icme_start_time) > 0,(frame_time_num+k*res_in_hours/24-ic.icme_start_time) < time_diff_icme))[0]\n",
    "        #print(cur_icme)\n",
    "        \n",
    "\n",
    "\n",
    "    #positions text\n",
    "    f10=plt.figtext(0.01,0.93,'              R     lon     lat', fontsize=fsize+2, ha='left',color=backcolor)\n",
    "\n",
    "    if frame=='HEEQ': earth_text='Earth: '+str(f'{earth.r[earth_timeind]:6.2f}')+str(f'{0.0:8.1f}')+str(f'{np.rad2deg(earth.lat[earth_timeind]):8.1f}')\n",
    "    else: earth_text='Earth: '+str(f'{earth.r[earth_timeind]:6.2f}')+str(f'{np.rad2deg(earth.lon[earth_timeind]):8.1f}')+str(f'{np.rad2deg(earth.lat[earth_timeind]):8.1f}')\n",
    "\n",
    "    mars_text='Mars:  '+str(f'{mars.r[earth_timeind]:6.2f}')+str(f'{np.rad2deg(mars.lon[earth_timeind]):8.1f}')+str(f'{np.rad2deg(mars.lat[earth_timeind]):8.1f}')\n",
    "    sta_text='STA:   '+str(f'{sta.r[sta_timeind]:6.2f}')+str(f'{np.rad2deg(sta.lon[sta_timeind]):8.1f}')+str(f'{np.rad2deg(sta.lat[sta_timeind]):8.1f}')\n",
    "    \n",
    "    \n",
    "    \n",
    "    #ICMEs for Earth    \n",
    "    if add_icmes:                        \n",
    "            cur_icme_wind=np.where(ic.sc_insitu[cur_icme] == 'Wind')[0]            \n",
    "            for i in np.arange(len(cur_icme_wind)):\n",
    "                time_dist=frame_time_num+k*res_in_hours/24-ic.icme_start_time[cur_icme][cur_icme_wind]\n",
    "                fadedist=1-time_dist/time_diff_icme\n",
    "                ax.scatter(earth.lon[earth_timeind], earth.r[earth_timeind]*np.cos(earth.lat[earth_timeind]), s=symsize_icme, c='mediumseagreen', marker='o', alpha=fadedist,lw=0,zorder=3)\n",
    "\n",
    "    #ICMEs for STEREO-A\n",
    "    if add_icmes:                        \n",
    "            cur_icme_sta=np.where(ic.sc_insitu[cur_icme] == 'STEREO-A')[0]            \n",
    "            for i in np.arange(len(cur_icme_sta)):\n",
    "                #get difference of icme_start_time to current frame time\n",
    "                time_dist=frame_time_num+k*res_in_hours/24-ic.icme_start_time[cur_icme][cur_icme_sta]\n",
    "                #define fading\n",
    "                fadedist=1-time_dist/time_diff_icme\n",
    "                ax.scatter(sta.lon[sta_timeind], sta.r[sta_timeind]*np.cos(sta.lat[sta_timeind]), s=symsize_icme, c=sta_color, marker='o', alpha=fadedist,lw=0,zorder=3)\n",
    "\n",
    "                \n",
    "    \n",
    "\n",
    "    #position and text for PSP only before the kernel ends\n",
    "    if np.logical_and(psp_timeind > 0,(frame_time_num+k*res_in_hours/24) < psp.time[-1] ):\n",
    "                      \n",
    "        #plot trajectory\n",
    "        ax.scatter(psp.lon[psp_timeind], psp.r[psp_timeind]*np.cos(psp.lat[psp_timeind]), s=symsize_spacecraft, c=psp_color, marker='s', alpha=1,lw=0,zorder=3)\n",
    "        #plot position as text\n",
    "        psp_text='PSP:   '+str(f'{psp.r[psp_timeind]:6.2f}')+str(f'{np.rad2deg(psp.lon[psp_timeind]):8.1f}')+str(f'{np.rad2deg(psp.lat[psp_timeind]):8.1f}')\n",
    "        f5=plt.figtext(0.01,0.78,psp_text, fontsize=fsize, ha='left',color=psp_color)\n",
    "        if plot_orbit: \n",
    "            fadestart=psp_timeind-fadeind\n",
    "            if  fadestart < 0: fadestart=0\n",
    "            ax.plot(psp.lon[fadestart:psp_timeind+fadeind], psp.r[fadestart:psp_timeind+fadeind]*np.cos(psp.lat[fadestart:psp_timeind+fadeind]), c=psp_color, alpha=0.6,lw=1,zorder=3)\n",
    "            \n",
    "        if add_icmes:                        \n",
    "            cur_icme_psp=np.where(ic.sc_insitu[cur_icme] == 'PSP')[0]            \n",
    "            for i in np.arange(len(cur_icme_psp)):\n",
    "                time_dist=frame_time_num+k*res_in_hours/24-ic.icme_start_time[cur_icme][cur_icme_psp]\n",
    "                #define fading\n",
    "                fadedist=1-time_dist/time_diff_icme\n",
    "                ax.scatter(psp.lon[psp_timeind], psp.r[psp_timeind]*np.cos(psp.lat[psp_timeind]), s=symsize_icme, c=psp_color, marker='o', alpha=fadedist,lw=0,zorder=3)\n",
    "            \n",
    "            \n",
    "\n",
    "    if np.logical_and(bepi_timeind > 0,(frame_time_num+k*res_in_hours/24) < bepi.time[-1] ):\n",
    "        \n",
    "        ax.scatter(bepi.lon[bepi_timeind], bepi.r[bepi_timeind]*np.cos(bepi.lat[bepi_timeind]), s=symsize_spacecraft, c=bepi_color, marker='s', alpha=1,lw=0,zorder=3)\n",
    "        bepi_text='Bepi:   '+str(f'{bepi.r[bepi_timeind]:6.2f}')+str(f'{np.rad2deg(bepi.lon[bepi_timeind]):8.1f}')+str(f'{np.rad2deg(bepi.lat[bepi_timeind]):8.1f}')\n",
    "        f6=plt.figtext(0.01,0.74,bepi_text, fontsize=fsize, ha='left',color=bepi_color)\n",
    "        if plot_orbit: \n",
    "            fadestart=bepi_timeind-fadeind\n",
    "            if  fadestart < 0: fadestart=0            \n",
    "            ax.plot(bepi.lon[fadestart:bepi_timeind+fadeind], bepi.r[fadestart:bepi_timeind+fadeind]*np.cos(bepi.lat[fadestart:bepi_timeind+fadeind]), c=bepi_color, alpha=0.6,lw=1,zorder=3)            \n",
    "        if add_icmes:                        \n",
    "            cur_icme_bepi=np.where(ic.sc_insitu[cur_icme] == 'BepiColombo')[0]            \n",
    "            for i in np.arange(len(cur_icme_bepi)):\n",
    "                time_dist=frame_time_num+k*res_in_hours/24-ic.icme_start_time[cur_icme][cur_icme_bepi]\n",
    "                #define fading\n",
    "                fadedist=1-time_dist/time_diff_icme\n",
    "                ax.scatter(bepi.lon[bepi_timeind], bepi.r[bepi_timeind]*np.cos(bepi.lat[bepi_timeind]), s=symsize_icme, c=bepi_color, marker='o', alpha=fadedist,lw=0,zorder=3)\n",
    "            \n",
    "                        \n",
    "            \n",
    "            \n",
    "            \n",
    "    #after Bepi kernel is done, mercury position\n",
    "    if (frame_time_num+k*res_in_hours/24) > bepi.time[-1]:\n",
    "        \n",
    "        ax.scatter(mercury.lon[earth_timeind], mercury.r[earth_timeind]*np.cos(mercury.lat[earth_timeind]), s=symsize_spacecraft, c=bepi_color, marker='s', alpha=1,lw=0,zorder=3)\n",
    "        bepi_text='Bepi:   '+str(f'{mercury.r[earth_timeind]:6.2f}')+str(f'{np.rad2deg(mercury.lon[earth_timeind]):8.1f}')+str(f'{np.rad2deg(mercury.lat[earth_timeind]):8.1f}')\n",
    "        f6=plt.figtext(0.01,0.74,bepi_text, fontsize=fsize, ha='left',color=bepi_color)\n",
    "        if plot_orbit: \n",
    "            fadestart=earth_timeind-fadeind\n",
    "            if  fadestart < 0: fadestart=0            \n",
    "            ax.plot(mercury.lon[fadestart:earth_timeind+fadeind], mercury.r[fadestart:earth_timeind+fadeind]*np.cos(mercury.lat[fadestart:earth_timeind+fadeind]), c=bepi_color, alpha=0.6,lw=1,zorder=3)\n",
    "            \n",
    "            \n",
    "\n",
    "            \n",
    "            \n",
    "\n",
    "    if solo_timeind > 0:\n",
    "        ax.scatter(solo.lon[solo_timeind], solo.r[solo_timeind]*np.cos(solo.lat[solo_timeind]), s=symsize_spacecraft, c=solo_color, marker='s', alpha=1,lw=0,zorder=3)\n",
    "        solo_text='SolO:  '+str(f'{solo.r[solo_timeind]:6.2f}')+str(f'{np.rad2deg(solo.lon[solo_timeind]):8.1f}')+str(f'{np.rad2deg(solo.lat[solo_timeind]):8.1f}')\n",
    "        f7=plt.figtext(0.01,0.7,solo_text, fontsize=fsize, ha='left',color=solo_color)\n",
    "        if plot_orbit: \n",
    "            fadestart=solo_timeind-fadeind\n",
    "            if  fadestart < 0: fadestart=0            \n",
    "            ax.plot(solo.lon[fadestart:solo_timeind+fadeind], solo.r[fadestart:solo_timeind+fadeind]*np.cos(solo.lat[fadestart:solo_timeind+fadeind]), c=solo_color, alpha=0.6,lw=1,zorder=3)\n",
    "         \n",
    "        if add_icmes:                        \n",
    "            cur_icme_solo=np.where(ic.sc_insitu[cur_icme] == 'SolarOrbiter')[0]            \n",
    "            #plot current solo cmes\n",
    "            for i in np.arange(len(cur_icme_solo)):\n",
    "                time_dist=frame_time_num+k*res_in_hours/24-ic.icme_start_time[cur_icme][cur_icme_solo]\n",
    "                #define fading\n",
    "                fadedist=1-time_dist/time_diff_icme\n",
    "                ax.scatter(solo.lon[solo_timeind], solo.r[solo_timeind]*np.cos(solo.lat[solo_timeind]), s=symsize_icme, c=solo_color, marker='o', alpha=fadedist,lw=0,zorder=3)\n",
    "\n",
    "            \n",
    "\n",
    "            \n",
    "    if juice_timeind > 0:\n",
    "        ax.scatter(juice.lon[juice_timeind], juice.r[juice_timeind]*np.cos(juice.lat[juice_timeind]), s=symsize_spacecraft, c=juice_color, marker='s', alpha=1,lw=0,zorder=3)\n",
    "        juice_text='JUICE:  '+str(f'{juice.r[juice_timeind]:6.2f}')+str(f'{np.rad2deg(juice.lon[juice_timeind]):8.1f}')+str(f'{np.rad2deg(juice.lat[juice_timeind]):8.1f}')\n",
    "        f7=plt.figtext(0.01,0.66,juice_text, fontsize=fsize, ha='left',color=juice_color)\n",
    "        if plot_orbit: \n",
    "            fadestart=juice_timeind-fadeind\n",
    "            if  fadestart < 0: fadestart=0            \n",
    "            ax.plot(juice.lon[fadestart:juice_timeind+fadeind], juice.r[fadestart:juice_timeind+fadeind]*np.cos(juice.lat[fadestart:juice_timeind+fadeind]), c=juice_color, alpha=0.6,lw=1,zorder=3)\n",
    "\n",
    "            \n",
    "            \n",
    "    f10=plt.figtext(0.01,0.9,earth_text, fontsize=fsize, ha='left',color='mediumseagreen')\n",
    "    f9=plt.figtext(0.01,0.86,mars_text, fontsize=fsize, ha='left',color='orangered')\n",
    "    f8=plt.figtext(0.01,0.82,sta_text, fontsize=fsize, ha='left',color=sta_color)\n",
    "    \n",
    "   \n",
    "    \n",
    "    \n",
    "    \n",
    "    #parker spiral\n",
    "    if plot_parker:\n",
    "        for q in np.arange(0,12):\n",
    "            omega=2*np.pi/(sun_rot*60*60*24) #solar rotation in seconds\n",
    "            v=400/AUkm #km/s\n",
    "            r0=695000/AUkm\n",
    "            r=v/omega*theta+r0*7\n",
    "            if not black: \n",
    "                ax.plot(-theta+np.deg2rad(0+(360/24.47)*res_in_hours/24*k+360/12*q), r, alpha=0.4, lw=0.5,color='grey',zorder=2)\n",
    "            if black: \n",
    "                ax.plot(-theta+np.deg2rad(0+(360/24.47)*res_in_hours724*k+360/12*q), r, alpha=0.7, lw=0.7,color='grey',zorder=2)\n",
    "\n",
    "    #set axes and grid\n",
    "    ax.set_theta_zero_location('E')\n",
    "    #plt.thetagrids(range(0,360,45),(u'0\\u00b0 '+frame+' longitude',u'45\\u00b0',u'90\\u00b0',u'135\\u00b0',u'+/- 180\\u00b0',u'- 135\\u00b0',u'- 90\\u00b0',u'- 45\\u00b0'), ha='right', fmt='%d',fontsize=fsize-1,color=backcolor, alpha=0.9)\n",
    "    plt.thetagrids(range(0,360,45),(u'0\\u00b0',u'45\\u00b0',u'90\\u00b0',u'135\\u00b0',u'+/- 180\\u00b0',u'- 135\\u00b0',u'- 90\\u00b0',u'- 45\\u00b0'), ha='center', fmt='%d',fontsize=fsize-1,color=backcolor, alpha=0.9,zorder=4)\n",
    "\n",
    "\n",
    "    #plt.rgrids((0.10,0.39,0.72,1.00,1.52),('0.10','0.39','0.72','1.0','1.52 AU'),angle=125, fontsize=fsize,alpha=0.9, color=backcolor)\n",
    "    plt.rgrids((0.1,0.3,0.5,0.7,1.0,1.3),('0.10','0.3','0.5','0.7','1.0','1.3 AU'),angle=125, fontsize=fsize-3,alpha=0.5, color=backcolor)\n",
    "\n",
    "    #ax.set_ylim(0, 1.75) #with Mars\n",
    "    ax.set_ylim(0, rmax) #Mars at 1.66 max \n",
    "\n",
    "    #Sun\n",
    "    ax.scatter(0,0,s=100,c='yellow',alpha=1, edgecolors='black', linewidth=0.3)\n",
    "\n",
    "    \n",
    "    #signature\n",
    "    plt.figtext(0.01,0.02,'Austrian Space Weather Office, GeoSphere Austria', fontsize=fsize, ha='left',color=backcolor) \n",
    "    \n",
    "     \n",
    "    \n",
    "    logo = plt.imread('logo/GSA_Basislogo_NegativAufMidnightGreen_RGB_XXS.png')\n",
    "    newax = fig.add_axes([0.91,0.91,0.08,0.08], anchor='NE', zorder=1)\n",
    "    newax.imshow(logo)\n",
    "    newax.axis('off')\n",
    "    \n",
    "    \n",
    "    \n",
    "    ############ date\n",
    "    #plot text for date extra so it does not move    \n",
    "    xclock=0.65\n",
    "    yclock=0.06\n",
    "    f1=plt.figtext(xclock,yclock,frame_time_str[0:4],  ha='center',color=backcolor,fontsize=fsize+6)\n",
    "    #month\n",
    "    f2=plt.figtext(xclock+0.04,yclock,frame_time_str[5:7], ha='center',color=backcolor,fontsize=fsize+6)\n",
    "    #day\n",
    "    f3=plt.figtext(xclock+0.08,yclock,frame_time_str[8:10], ha='center',color=backcolor,fontsize=fsize+6)\n",
    "    #hours\n",
    "    f4=plt.figtext(xclock+0.12,yclock,frame_time_str[11:13], ha='center',color=backcolor,fontsize=fsize+6)\n",
    "    \n",
    "    \n",
    "    plt.tight_layout()\n",
    "\n",
    "\n",
    "    #save figure\n",
    "    framestr = '%05i' % (k)  \n",
    "    filename=outputdirectory+'/pos_anim_'+framestr+'.jpg'  \n",
    "    if k==0: print(filename)\n",
    "    plt.savefig(filename,dpi=dpisave,facecolor=fig.get_facecolor(), edgecolor='none')\n",
    "    #plt.clf()\n",
    "    plt.close('all')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "393c8433-8d2e-45fa-92b1-c2b0d3806a07",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "HEEQ\n",
      "2023-03-10 00:00:00\n",
      "2023-05-09 20:00:00\n",
      "number of frames: 366\n",
      "days for trajectory 60.0\n"
     ]
    }
   ],
   "source": [
    "plt.close('all')\n",
    "\n",
    "####################### SETTINGS\n",
    "\n",
    "#Coordinate System\n",
    "#frame='HCI'\n",
    "frame='HEEQ'\n",
    "print(frame)\n",
    "\n",
    "#sidereal solar rotation rate\n",
    "if frame=='HCI': sun_rot=24.47\n",
    "#synodic\n",
    "if frame=='HEEQ': sun_rot=26.24\n",
    "\n",
    "AUkm=149597870.7   \n",
    "\n",
    "#black background on or off\n",
    "#black=True\n",
    "black=True\n",
    "\n",
    "#animation settings\n",
    "plot_orbit=True\n",
    "#plot_orbit=False\n",
    "plot_parker=False\n",
    "#plot_parker=False\n",
    "\n",
    "high_res_mode=False\n",
    "\n",
    "\n",
    "# for adding ICMECAT indicators for ICMEs\n",
    "add_icmes=True\n",
    "#this is the time how long an ICME is visible as a faded circle\n",
    "time_diff_icme=7 \n",
    "#how big the symbol for the ICME is\n",
    "symsize_icme=450\n",
    "        \n",
    "\n",
    "\n",
    "def datetime_range(start, end, resolution_hours):\n",
    "    current = start\n",
    "    while current < end:\n",
    "        yield current\n",
    "        current += timedelta(hours=resolution_hours)\n",
    "\n",
    "time_array = [dt for dt in datetime_range(t_start, t_end, res_in_hours)]\n",
    "\n",
    "k_all=np.size(time_array)\n",
    "counter=[i for i in range(k_all)]\n",
    "\n",
    "\n",
    "\n",
    "print(time_array[0])\n",
    "print(time_array[-1])\n",
    "print('number of frames:',k_all)\n",
    "#print(time_array)\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "############################ \n",
    "\n",
    "#animation start time in matplotlib format\n",
    "frame_time_num=parse_time(t_start).plot_date\n",
    "\n",
    "sns.set_context('talk')\n",
    "\n",
    "if not black: sns.set_style('darkgrid'),#{'grid.linestyle': ':', 'grid.color': '.35'}) \n",
    "\n",
    "if black: sns.set_style('white',{'grid.linestyle': ':', 'grid.color': '.35'})   \n",
    "\n",
    "# animation settings \n",
    "fsize=13\n",
    "\n",
    "#how long the trajectory tracks are\n",
    "#1 hour resolution of the position file\n",
    "fadeind=int(24*60)\n",
    "\n",
    "print('days for trajectory',fadeind/24)\n",
    "\n",
    "symsize_planet=110\n",
    "symsize_spacecraft=80\n",
    "\n",
    "\n",
    "\n",
    "#for parker spiral   \n",
    "theta=np.arange(0,np.deg2rad(180),0.01)\n",
    "\n",
    "\n",
    "# ### single processing\n",
    "\n",
    "# In[11]:\n",
    "\n",
    "\n",
    "#print()\n",
    "#print('make animation')\n",
    "#print()\n",
    "\n",
    "#k_all=1\n",
    "\n",
    "#for debugging\n",
    "#k_all=1000\n",
    "#make_frame(1)\n",
    "#for i in np.arange(1,10,1):\n",
    "#    make_frame(i)\n",
    "    \n",
    "#os.system(ffmpeg_path+'ffmpeg -r 25 -i '+str(outputdirectory)+'/pos_anim_%05d.jpg -b 5000k \\\n",
    "#    -r 25 '+str(animdirectory)+'/positions_punch.mp4 -y -loglevel quiet')    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "f486749c-a2bd-4e5e-ad0c-49e437013476",
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "\n",
      "make animation\n",
      "\n",
      "Using multiprocessing, nr of cores 8 with nr of processes used:  8\n",
      "current frame_time_numcurrent frame_time_num current frame_time_numcurrent frame_time_num  2023-03-16 00:00:00+00:00current frame_time_num 2023-03-20 00:00:00+00:00 2023-03-18 00:00:00+00:00 2023-03-14 00:00:00+00:00current frame_time_numcurrent frame_time_num        2023-03-24 00:00:00+00:00                   2023-03-22 00:00:00+00:002023-03-10 00:00:00+00:0036          4860\n",
      "current frame_time_num      24     \n",
      "\n",
      " 84 2023-03-12 00:00:00+00:00\n",
      " \n",
      "72 0\n",
      "     \n",
      " 12\n",
      "icmecat/anim_frames/pos_anim_00000.jpg\n",
      "current frame_time_num 2023-03-16 04:00:00+00:00       37\n",
      "current frame_time_num 2023-03-20 04:00:00+00:00       61\n",
      "current frame_time_num 2023-03-22 04:00:00+00:00       73\n",
      "current frame_time_num 2023-03-24 04:00:00+00:00       85\n",
      "current frame_time_num 2023-03-18 04:00:00+00:00current frame_time_num       49\n",
      " 2023-03-10 04:00:00+00:00       1\n",
      "current frame_time_num 2023-03-14 04:00:00+00:00       25\n",
      "current frame_time_num 2023-03-12 04:00:00+00:00       13\n",
      "current frame_time_num 2023-03-16 08:00:00+00:00       38\n",
      "current frame_time_num 2023-03-22 08:00:00+00:00       74\n",
      "current frame_time_num 2023-03-20 08:00:00+00:00      current frame_time_num  2023-03-14 08:00:00+00:00       2662\n",
      "\n",
      "current frame_time_num 2023-03-10 08:00:00+00:00       2\n",
      "current frame_time_num current frame_time_num2023-03-12 08:00:00+00:00        142023-03-24 08:00:00+00:00\n",
      "       86\n",
      "current frame_time_num 2023-03-18 08:00:00+00:00       50\n",
      "current frame_time_num 2023-03-22 12:00:00+00:00       75\n",
      "current frame_time_num 2023-03-16 12:00:00+00:00       39\n",
      "current frame_time_num 2023-03-20 12:00:00+00:00       63\n",
      "current frame_time_num 2023-03-14 12:00:00+00:00       27\n",
      "current frame_time_num 2023-03-24 12:00:00+00:00       87\n",
      "current frame_time_num 2023-03-12 12:00:00+00:00current frame_time_num      current frame_time_num 15\n",
      "  2023-03-18 12:00:00+00:002023-03-10 12:00:00+00:00              513\n",
      "\n",
      "current frame_time_num 2023-03-16 16:00:00+00:00       40\n",
      "current frame_time_numcurrent frame_time_num 2023-03-22 16:00:00+00:00       2023-03-20 16:00:00+00:00 current frame_time_num      76\n",
      "  2023-03-18 16:00:00+00:0064 \n",
      "      current frame_time_num52\n",
      " 2023-03-14 16:00:00+00:00       28\n",
      "current frame_time_num 2023-03-10 16:00:00+00:00       4\n",
      "current frame_time_num current frame_time_num 2023-03-24 16:00:00+00:00       88\n",
      "2023-03-12 16:00:00+00:00       16\n",
      "current frame_time_num 2023-03-22 20:00:00+00:00      current frame_time_num 77\n",
      " 2023-03-16 20:00:00+00:00       41\n",
      "current frame_time_numcurrent frame_time_num 2023-03-20 20:00:00+00:00  2023-03-14 20:00:00+00:00            65 \n",
      "29\n",
      "current frame_time_num 2023-03-18 20:00:00+00:00       53\n",
      "current frame_time_num 2023-03-10 20:00:00+00:00       5\n",
      "current frame_time_numcurrent frame_time_num  2023-03-24 20:00:00+00:00 2023-03-12 20:00:00+00:00             8917\n",
      "\n",
      "current frame_time_num 2023-03-23 00:00:00+00:00       78\n",
      "current frame_time_num 2023-03-17 00:00:00+00:00       42\n",
      "current frame_time_num 2023-03-15 00:00:00+00:00       30\n",
      "current frame_time_num 2023-03-21 00:00:00+00:00       66\n",
      "current frame_time_numcurrent frame_time_num  2023-03-11 00:00:00+00:002023-03-25 00:00:00+00:00              690\n",
      "\n",
      "current frame_time_numcurrent frame_time_num  2023-03-19 00:00:00+00:00      2023-03-13 00:00:00+00:00  54      \n",
      "18\n",
      "current frame_time_num 2023-03-23 04:00:00+00:00       79\n",
      "current frame_time_num 2023-03-17 04:00:00+00:00       43\n",
      "current frame_time_num 2023-03-21 04:00:00+00:00       67\n",
      "current frame_time_num 2023-03-11 04:00:00+00:00       7\n",
      "current frame_time_num 2023-03-15 04:00:00+00:00       31\n",
      "current frame_time_num 2023-03-19 04:00:00+00:00       55\n",
      "current frame_time_num 2023-03-13 04:00:00+00:00       19\n",
      "current frame_time_num 2023-03-25 04:00:00+00:00       91\n",
      "current frame_time_num 2023-03-23 08:00:00+00:00       80\n",
      "current frame_time_num 2023-03-17 08:00:00+00:00       44\n",
      "current frame_time_num 2023-03-11 08:00:00+00:00current frame_time_num       2023-03-21 08:00:00+00:00       8 \n",
      "68\n",
      "current frame_time_num 2023-03-15 08:00:00+00:00       32\n",
      "current frame_time_num 2023-03-19 08:00:00+00:00       56\n",
      "current frame_time_num 2023-03-13 08:00:00+00:00       20\n",
      "current frame_time_num 2023-03-25 08:00:00+00:00       92\n",
      "current frame_time_num 2023-03-23 12:00:00+00:00       81\n",
      "current frame_time_num 2023-03-17 12:00:00+00:00       45\n",
      "current frame_time_num 2023-03-21 12:00:00+00:00       69\n",
      "current frame_time_num 2023-03-11 12:00:00+00:00       9\n",
      "current frame_time_num 2023-03-15 12:00:00+00:00       33\n",
      "current frame_time_num 2023-03-19 12:00:00+00:00       57\n",
      "current frame_time_num 2023-03-25 12:00:00+00:00       93\n",
      "current frame_time_num 2023-03-13 12:00:00+00:00       21\n",
      "current frame_time_num 2023-03-23 16:00:00+00:00       82\n",
      "current frame_time_num 2023-03-17 16:00:00+00:00       46\n",
      "current frame_time_num 2023-03-11 16:00:00+00:00       10current frame_time_num\n",
      " 2023-03-21 16:00:00+00:00       70\n",
      "current frame_time_num 2023-03-15 16:00:00+00:00       34\n",
      "current frame_time_numcurrent frame_time_num  2023-03-19 16:00:00+00:002023-03-25 16:00:00+00:00              9458\n",
      "\n",
      "current frame_time_num 2023-03-13 16:00:00+00:00       22\n",
      "current frame_time_num 2023-03-23 20:00:00+00:00       83\n",
      "current frame_time_num 2023-03-17 20:00:00+00:00       47\n",
      "current frame_time_num 2023-03-11 20:00:00+00:00       11\n",
      "current frame_time_num 2023-03-21 20:00:00+00:00       71\n",
      "current frame_time_num 2023-03-19 20:00:00+00:00       59\n",
      "current frame_time_num 2023-03-15 20:00:00+00:00       35\n",
      "current frame_time_num 2023-03-13 20:00:00+00:00       23\n",
      "current frame_time_num 2023-03-25 20:00:00+00:00       95\n",
      "current frame_time_num 2023-03-26 00:00:00+00:00       96\n",
      "current frame_time_num 2023-03-28 00:00:00+00:00       108\n",
      "current frame_time_num 2023-03-30 00:00:00+00:00       120\n",
      "current frame_time_num 2023-04-01 00:00:00+00:00       132\n",
      "current frame_time_num 2023-04-03 00:00:00+00:00       144\n",
      "current frame_time_numcurrent frame_time_num 2023-04-07 00:00:00+00:00       168\n",
      " 2023-04-05 00:00:00+00:00       156\n",
      "current frame_time_num 2023-04-09 00:00:00+00:00       180\n",
      "current frame_time_num 2023-03-26 04:00:00+00:00       97\n",
      "current frame_time_numcurrent frame_time_num  2023-04-01 04:00:00+00:00 2023-03-28 04:00:00+00:00            133 \n",
      "109\n",
      "current frame_time_num 2023-03-30 04:00:00+00:00       121\n",
      "current frame_time_num 2023-04-03 04:00:00+00:00       145\n",
      "current frame_time_num 2023-04-05 04:00:00+00:00       157\n",
      "current frame_time_num 2023-04-07 04:00:00+00:00       169\n",
      "current frame_time_num 2023-04-09 04:00:00+00:00       181\n",
      "current frame_time_num 2023-03-26 08:00:00+00:00       98\n",
      "current frame_time_num 2023-04-01 08:00:00+00:00       134\n",
      "current frame_time_num 2023-03-28 08:00:00+00:00       110\n",
      "current frame_time_num 2023-03-30 08:00:00+00:00       122\n",
      "current frame_time_num 2023-04-03 08:00:00+00:00      current frame_time_num  1462023-04-09 08:00:00+00:00 \n",
      "      182\n",
      "current frame_time_num 2023-04-05 08:00:00+00:00       158\n",
      "current frame_time_num 2023-04-07 08:00:00+00:00       170\n",
      "current frame_time_num 2023-04-01 12:00:00+00:00       135\n",
      "current frame_time_num 2023-03-26 12:00:00+00:00       99\n",
      "current frame_time_num 2023-03-28 12:00:00+00:00       111\n",
      "current frame_time_num 2023-04-09 12:00:00+00:00       183\n",
      "current frame_time_num 2023-03-30 12:00:00+00:00       123\n",
      "current frame_time_num current frame_time_num2023-04-03 12:00:00+00:00  2023-04-05 12:00:00+00:00            147 \n",
      "159\n",
      "current frame_time_num 2023-04-07 12:00:00+00:00       171\n",
      "current frame_time_num 2023-03-26 16:00:00+00:00       100\n",
      "current frame_time_num 2023-04-01 16:00:00+00:00current frame_time_num       136\n",
      " 2023-03-28 16:00:00+00:00       112\n",
      "current frame_time_num 2023-04-09 16:00:00+00:00current frame_time_num       2023-03-30 16:00:00+00:00  184      124\n",
      "\n",
      "current frame_time_numcurrent frame_time_num  2023-04-05 16:00:00+00:002023-04-03 16:00:00+00:00              148160\n",
      "\n",
      "current frame_time_num 2023-04-07 16:00:00+00:00       172\n",
      "current frame_time_num 2023-03-26 20:00:00+00:00       101\n",
      "current frame_time_num 2023-04-01 20:00:00+00:00       137\n",
      "current frame_time_num 2023-03-28 20:00:00+00:00       113\n",
      "current frame_time_num 2023-04-09 20:00:00+00:00       185\n",
      "current frame_time_num 2023-03-30 20:00:00+00:00       125\n",
      "current frame_time_num 2023-04-05 20:00:00+00:00       161\n",
      "current frame_time_num 2023-04-03 20:00:00+00:00       149\n",
      "current frame_time_num 2023-04-07 20:00:00+00:00       173\n",
      "current frame_time_num 2023-03-27 00:00:00+00:00       102\n",
      "current frame_time_num 2023-04-02 00:00:00+00:00       138\n",
      "current frame_time_num 2023-04-10 00:00:00+00:00       186\n",
      "current frame_time_num 2023-03-29 00:00:00+00:00       114\n",
      "current frame_time_num 2023-03-31 00:00:00+00:00       126\n",
      "current frame_time_num current frame_time_num 2023-04-06 00:00:00+00:00       162\n",
      "2023-04-04 00:00:00+00:00       150\n",
      "current frame_time_num 2023-04-08 00:00:00+00:00       174\n",
      "current frame_time_num 2023-03-27 04:00:00+00:00       103\n",
      "current frame_time_num 2023-04-10 04:00:00+00:00       187\n",
      "current frame_time_num 2023-04-02 04:00:00+00:00       139\n",
      "current frame_time_num 2023-03-31 04:00:00+00:00       127\n",
      "current frame_time_num 2023-03-29 04:00:00+00:00       115\n",
      "current frame_time_num 2023-04-04 04:00:00+00:00       151\n",
      "current frame_time_num 2023-04-06 04:00:00+00:00       163\n",
      "current frame_time_num 2023-04-08 04:00:00+00:00       175\n",
      "current frame_time_num 2023-03-27 08:00:00+00:00       104\n",
      "current frame_time_num 2023-04-10 08:00:00+00:00       188\n",
      "current frame_time_num current frame_time_num 2023-04-02 08:00:00+00:00       140\n",
      "2023-03-31 08:00:00+00:00       128\n",
      "current frame_time_numcurrent frame_time_num  2023-04-04 08:00:00+00:002023-03-29 08:00:00+00:00             152 \n",
      "116\n",
      "current frame_time_num 2023-04-06 08:00:00+00:00       164\n",
      "current frame_time_num 2023-04-08 08:00:00+00:00       176\n",
      "current frame_time_num 2023-04-10 12:00:00+00:00       189current frame_time_num\n",
      " 2023-03-27 12:00:00+00:00       105\n",
      "current frame_time_num 2023-03-31 12:00:00+00:00current frame_time_num 2023-04-02 12:00:00+00:00       141\n",
      "current frame_time_num 2023-03-29 12:00:00+00:00       117\n",
      "       129\n",
      "current frame_time_num 2023-04-04 12:00:00+00:00       153\n",
      "current frame_time_num 2023-04-06 12:00:00+00:00       165\n",
      "current frame_time_num 2023-04-08 12:00:00+00:00       177\n",
      "current frame_time_num 2023-04-10 16:00:00+00:00       190\n",
      "current frame_time_num 2023-03-27 16:00:00+00:00       106\n",
      "current frame_time_num 2023-04-02 16:00:00+00:00       142\n",
      "current frame_time_num 2023-03-31 16:00:00+00:00       130\n",
      "current frame_time_num 2023-03-29 16:00:00+00:00       118\n",
      "current frame_time_num 2023-04-08 16:00:00+00:00       178\n",
      "current frame_time_numcurrent frame_time_num  2023-04-06 16:00:00+00:002023-04-04 16:00:00+00:00              154166\n",
      "\n",
      "current frame_time_num 2023-04-10 20:00:00+00:00       191\n",
      "current frame_time_num 2023-03-27 20:00:00+00:00       107\n",
      "current frame_time_num 2023-04-02 20:00:00+00:00       143\n",
      "current frame_time_num 2023-03-31 20:00:00+00:00       131\n",
      "current frame_time_num 2023-04-08 20:00:00+00:00       179\n",
      "current frame_time_num 2023-04-06 20:00:00+00:00       167\n",
      "current frame_time_num 2023-04-04 20:00:00+00:00       155\n",
      "current frame_time_num 2023-03-29 20:00:00+00:00       119\n",
      "current frame_time_num 2023-04-13 00:00:00+00:00       204\n",
      "current frame_time_num 2023-04-11 00:00:00+00:00       192\n",
      "current frame_time_num 2023-04-15 00:00:00+00:00       216\n",
      "current frame_time_num 2023-04-17 00:00:00+00:00       228\n",
      "current frame_time_num 2023-04-19 00:00:00+00:00       240\n",
      "current frame_time_num 2023-04-23 00:00:00+00:00       264\n",
      "current frame_time_num 2023-04-21 00:00:00+00:00       252\n",
      "current frame_time_num 2023-04-25 00:00:00+00:00       276\n",
      "current frame_time_num 2023-04-13 04:00:00+00:00       205\n",
      "current frame_time_num 2023-04-11 04:00:00+00:00       193\n",
      "current frame_time_num 2023-04-15 04:00:00+00:00       217\n",
      "current frame_time_num 2023-04-17 04:00:00+00:00       229\n",
      "current frame_time_num 2023-04-19 04:00:00+00:00       241\n",
      "current frame_time_num 2023-04-21 04:00:00+00:00       253\n",
      "current frame_time_num 2023-04-23 04:00:00+00:00       265\n",
      "current frame_time_num 2023-04-25 04:00:00+00:00       277\n",
      "current frame_time_num 2023-04-13 08:00:00+00:00       206\n",
      "current frame_time_num 2023-04-15 08:00:00+00:00       218\n",
      "current frame_time_num 2023-04-11 08:00:00+00:00       194\n",
      "current frame_time_num 2023-04-17 08:00:00+00:00       230\n",
      "current frame_time_num 2023-04-21 08:00:00+00:00       254\n",
      "current frame_time_numcurrent frame_time_num  2023-04-19 08:00:00+00:002023-04-23 08:00:00+00:00             242 \n",
      "266\n",
      "current frame_time_num 2023-04-25 08:00:00+00:00       278\n",
      "current frame_time_num 2023-04-13 12:00:00+00:00       207\n",
      "current frame_time_num 2023-04-11 12:00:00+00:00       195\n",
      "current frame_time_num 2023-04-15 12:00:00+00:00       219\n",
      "current frame_time_num 2023-04-17 12:00:00+00:00       231\n",
      "current frame_time_num 2023-04-21 12:00:00+00:00       255\n",
      "current frame_time_num 2023-04-19 12:00:00+00:00 current frame_time_num      243\n",
      " 2023-04-25 12:00:00+00:00       279\n",
      "current frame_time_num 2023-04-23 12:00:00+00:00       267\n",
      "current frame_time_num 2023-04-13 16:00:00+00:00       208\n",
      "current frame_time_num 2023-04-15 16:00:00+00:00       220\n",
      "current frame_time_num 2023-04-11 16:00:00+00:00       196\n",
      "current frame_time_num 2023-04-17 16:00:00+00:00       232\n",
      "current frame_time_num 2023-04-21 16:00:00+00:00       256\n",
      "current frame_time_num 2023-04-19 16:00:00+00:00       244\n",
      "current frame_time_numcurrent frame_time_num  2023-04-23 16:00:00+00:002023-04-25 16:00:00+00:00              268280\n",
      "\n",
      "current frame_time_num 2023-04-13 20:00:00+00:00       209\n",
      "current frame_time_num 2023-04-15 20:00:00+00:00       221\n",
      "current frame_time_num 2023-04-11 20:00:00+00:00       197\n",
      "current frame_time_num 2023-04-17 20:00:00+00:00       233\n",
      "current frame_time_num 2023-04-19 20:00:00+00:00       245\n",
      "current frame_time_num 2023-04-21 20:00:00+00:00       257\n",
      "current frame_time_num 2023-04-23 20:00:00+00:00       269\n",
      "current frame_time_num 2023-04-25 20:00:00+00:00       281\n",
      "current frame_time_num 2023-04-14 00:00:00+00:00       210\n",
      "current frame_time_num 2023-04-16 00:00:00+00:00       222\n",
      "current frame_time_numcurrent frame_time_num  2023-04-12 00:00:00+00:00 2023-04-18 00:00:00+00:00            198 \n",
      "234\n",
      "current frame_time_num 2023-04-22 00:00:00+00:00       258\n",
      "current frame_time_num 2023-04-20 00:00:00+00:00       246\n",
      "current frame_time_num 2023-04-26 00:00:00+00:00       282\n",
      "current frame_time_num 2023-04-24 00:00:00+00:00       270\n",
      "current frame_time_num 2023-04-14 04:00:00+00:00       211\n",
      "current frame_time_num 2023-04-16 04:00:00+00:00       223\n",
      "current frame_time_num 2023-04-12 04:00:00+00:00       199\n",
      "current frame_time_num 2023-04-18 04:00:00+00:00       235\n",
      "current frame_time_num 2023-04-22 04:00:00+00:00       259\n",
      "current frame_time_num 2023-04-20 04:00:00+00:00       247\n",
      "current frame_time_num 2023-04-26 04:00:00+00:00       283\n",
      "current frame_time_num 2023-04-24 04:00:00+00:00       271\n",
      "current frame_time_numcurrent frame_time_num 2023-04-14 08:00:00+00:00        2023-04-16 08:00:00+00:00212\n",
      "       224\n",
      "current frame_time_num 2023-04-12 08:00:00+00:00       200\n",
      "current frame_time_num 2023-04-18 08:00:00+00:00       236\n",
      "current frame_time_num 2023-04-22 08:00:00+00:00       260\n",
      "current frame_time_num 2023-04-20 08:00:00+00:00       248\n",
      "current frame_time_num 2023-04-26 08:00:00+00:00       284\n",
      "current frame_time_num current frame_time_num2023-04-24 08:00:00+00:00  2023-04-14 12:00:00+00:00       272     \n",
      " 213\n",
      "current frame_time_num 2023-04-16 12:00:00+00:00       225\n",
      "current frame_time_num 2023-04-12 12:00:00+00:00       201\n",
      "current frame_time_num 2023-04-18 12:00:00+00:00current frame_time_num  2023-04-20 12:00:00+00:00            237 \n",
      "249\n",
      "current frame_time_num 2023-04-22 12:00:00+00:00       261\n",
      "current frame_time_num 2023-04-26 12:00:00+00:00       285\n",
      "current frame_time_num 2023-04-24 12:00:00+00:00       273\n",
      "current frame_time_numcurrent frame_time_num  2023-04-16 16:00:00+00:002023-04-14 16:00:00+00:00              214\n",
      "226\n",
      "current frame_time_num 2023-04-20 16:00:00+00:00       250\n",
      "current frame_time_num 2023-04-18 16:00:00+00:00       238\n",
      "current frame_time_num 2023-04-12 16:00:00+00:00       202\n",
      "current frame_time_num 2023-04-22 16:00:00+00:00       262\n",
      "current frame_time_num 2023-04-26 16:00:00+00:00       286\n",
      "current frame_time_num 2023-04-24 16:00:00+00:00       274\n",
      "current frame_time_num 2023-04-16 20:00:00+00:00current frame_time_num       2023-04-14 20:00:00+00:00  227      \n",
      "215\n",
      "current frame_time_num 2023-04-18 20:00:00+00:00       239\n",
      "current frame_time_num 2023-04-20 20:00:00+00:00       251\n",
      "current frame_time_num 2023-04-12 20:00:00+00:00       203\n",
      "current frame_time_num 2023-04-22 20:00:00+00:00       263\n",
      "current frame_time_num 2023-04-26 20:00:00+00:00       287\n",
      "current frame_time_num 2023-04-24 20:00:00+00:00       275\n",
      "current frame_time_num 2023-04-27 00:00:00+00:00       288\n",
      "current frame_time_num 2023-04-29 00:00:00+00:00       300\n",
      "current frame_time_numcurrent frame_time_num  2023-05-01 00:00:00+00:002023-05-03 00:00:00+00:00current frame_time_num   2023-05-05 00:00:00+00:00                  312 324\n",
      "336\n",
      "\n",
      "current frame_time_num 2023-05-07 00:00:00+00:00       348\n",
      "current frame_time_num 2023-05-09 00:00:00+00:00       360\n",
      "current frame_time_num 2023-04-27 04:00:00+00:00       289\n",
      "current frame_time_numcurrent frame_time_num  2023-04-29 04:00:00+00:002023-05-01 04:00:00+00:00              313301\n",
      "\n",
      "current frame_time_num 2023-05-03 04:00:00+00:00       325\n",
      "current frame_time_num 2023-05-05 04:00:00+00:00       337\n",
      "current frame_time_num 2023-05-07 04:00:00+00:00       349\n",
      "current frame_time_num 2023-05-09 04:00:00+00:00       361\n",
      "current frame_time_num 2023-04-27 08:00:00+00:00       290\n",
      "current frame_time_num 2023-05-01 08:00:00+00:00       314\n",
      "current frame_time_num 2023-05-03 08:00:00+00:00       326\n",
      "current frame_time_numcurrent frame_time_num  2023-05-05 08:00:00+00:002023-04-29 08:00:00+00:00              338302\n",
      "\n",
      "current frame_time_numcurrent frame_time_num  2023-05-07 08:00:00+00:002023-05-09 08:00:00+00:00             362\n",
      " 350\n",
      "current frame_time_num 2023-04-27 12:00:00+00:00       291\n",
      "current frame_time_num 2023-05-01 12:00:00+00:00       315\n",
      "current frame_time_num 2023-05-03 12:00:00+00:00       327\n",
      "current frame_time_num 2023-05-05 12:00:00+00:00       339\n",
      "current frame_time_num 2023-04-29 12:00:00+00:00       303\n",
      "current frame_time_num 2023-05-09 12:00:00+00:00       363\n",
      "current frame_time_num 2023-05-07 12:00:00+00:00       351\n",
      "current frame_time_num current frame_time_num2023-05-01 16:00:00+00:00  2023-04-27 16:00:00+00:00       316      \n",
      "292\n",
      "current frame_time_num 2023-05-03 16:00:00+00:00       328\n",
      "current frame_time_num 2023-05-05 16:00:00+00:00       340\n",
      "current frame_time_num 2023-04-29 16:00:00+00:00       304\n",
      "current frame_time_num 2023-05-09 16:00:00+00:00       364\n",
      "current frame_time_num 2023-05-07 16:00:00+00:00       352\n",
      "current frame_time_num 2023-05-01 20:00:00+00:00      current frame_time_num  2023-04-27 20:00:00+00:00317 \n",
      "      293\n",
      "current frame_time_num 2023-05-03 20:00:00+00:00       329\n",
      "current frame_time_num 2023-05-05 20:00:00+00:00       341\n",
      "current frame_time_num 2023-04-29 20:00:00+00:00       305\n",
      "current frame_time_num 2023-05-07 20:00:00+00:00       353\n",
      "current frame_time_num 2023-05-09 20:00:00+00:00       365\n",
      "current frame_time_num 2023-05-02 00:00:00+00:00       318\n",
      "current frame_time_num 2023-04-28 00:00:00+00:00       294\n",
      "current frame_time_num 2023-05-04 00:00:00+00:00       330\n",
      "current frame_time_num 2023-05-06 00:00:00+00:00       342\n",
      "current frame_time_num 2023-04-30 00:00:00+00:00       306\n",
      "current frame_time_num 2023-05-08 00:00:00+00:00       354\n",
      "current frame_time_num 2023-05-02 04:00:00+00:00       319\n",
      "current frame_time_num 2023-04-28 04:00:00+00:00       295\n",
      "current frame_time_numcurrent frame_time_num  2023-05-04 04:00:00+00:002023-05-06 04:00:00+00:00              331343\n",
      "\n",
      "current frame_time_num 2023-04-30 04:00:00+00:00       307\n",
      "current frame_time_num 2023-05-08 04:00:00+00:00       355\n",
      "current frame_time_num 2023-05-02 08:00:00+00:00       320\n",
      "current frame_time_num 2023-05-04 08:00:00+00:00       332\n",
      "current frame_time_num 2023-04-28 08:00:00+00:00       current frame_time_num296 \n",
      "2023-05-06 08:00:00+00:00       344\n",
      "current frame_time_num 2023-04-30 08:00:00+00:00       308\n",
      "current frame_time_num 2023-05-08 08:00:00+00:00       356\n",
      "current frame_time_num 2023-05-02 12:00:00+00:00       321\n",
      "current frame_time_num 2023-05-04 12:00:00+00:00       333\n",
      "current frame_time_numcurrent frame_time_numcurrent frame_time_num  2023-04-28 12:00:00+00:00        2972023-04-30 12:00:00+00:002023-05-06 12:00:00+00:00\n",
      "             309 345\n",
      "\n",
      "current frame_time_num 2023-05-08 12:00:00+00:00       357\n",
      "current frame_time_num 2023-05-02 16:00:00+00:00       322\n",
      "current frame_time_num 2023-05-04 16:00:00+00:00       334\n",
      "current frame_time_numcurrent frame_time_num  2023-04-30 16:00:00+00:002023-05-06 16:00:00+00:00             310 \n",
      "346\n",
      "current frame_time_num 2023-04-28 16:00:00+00:00       298\n",
      "current frame_time_num 2023-05-08 16:00:00+00:00       358\n",
      "current frame_time_num 2023-05-02 20:00:00+00:00       323\n",
      "current frame_time_num 2023-05-06 20:00:00+00:00       current frame_time_num347 2023-05-04 20:00:00+00:00\n",
      "       335\n",
      "current frame_time_num 2023-04-30 20:00:00+00:00       311\n",
      "current frame_time_num 2023-04-28 20:00:00+00:00       299\n",
      "current frame_time_num 2023-05-08 20:00:00+00:00       359\n",
      "time in min:  1.0\n",
      "plots done, frames saved in  icmecat/anim_frames\n",
      "movie done, saved in  icmecat\n"
     ]
    }
   ],
   "source": [
    "# ### Multiprocessing\n",
    "\n",
    "# In[ ]:\n",
    "\n",
    "\n",
    "print()\n",
    "print('make animation')\n",
    "print()\n",
    "\n",
    "#number of processes depends on your machines memory; check with command line \"top\"\n",
    "#how much memory is used by all your processes\n",
    "\n",
    "\n",
    "print('Using multiprocessing, nr of cores',mp.cpu_count(), \\\n",
    "      'with nr of processes used: ',used)\n",
    "\n",
    "\n",
    "#define pool using fork and number of processes\n",
    "pool=mp.get_context('fork').Pool(processes=used)\n",
    "# Map the worker function onto the parameters    \n",
    "t0 = time.time()\n",
    "pool.map(make_frame, counter) #or use apply_async?,imap\n",
    "pool.close()\n",
    "pool.join()     \n",
    "t1 = time.time()\n",
    "\n",
    "\n",
    "print('time in min: ',np.round((t1-t0)/60))\n",
    "print('plots done, frames saved in ',outputdirectory)\n",
    "\n",
    "\n",
    "os.system(ffmpeg_path+'ffmpeg -r 25 -i '+str(outputdirectory)+'/pos_anim_%05d.jpg -b 5000k \\\n",
    "    -r 30 '+str(animdirectory)+'/'+movie_filename+'.mp4 -y -loglevel quiet')    \n",
    "print('movie done, saved in ',animdirectory)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "aff7347b-0d50-4c53-a61b-8ca9da1e631a",
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "32e393a0-d102-449c-9264-e1abb2609922",
   "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.10.0"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}